org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet Class Reference

Inheritance diagram for org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet:

Collaboration diagram for org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet:

Public Member Functions
RexNode	convertCall (SqlRexContext cx, SqlCall call)

Package Functions
	RegrCovarianceConvertlet (SqlKind kind)

Private Member Functions
SqlNode	expandRegrSzz (final SqlNode arg1, final SqlNode arg2, final RelDataType avgType, final SqlRexContext cx, boolean variance)
SqlNode	expandCovariance (final SqlNode arg0Input, final SqlNode arg1Input, final SqlNode dependent, final RelDataType varType, final SqlRexContext cx, boolean biased)
SqlNode	getCastedSqlNode (SqlNode argInput, RelDataType varType, SqlParserPos pos, RexNode argRex)

Private Attributes
final SqlKind	kind

Detailed Description

Convertlet that handles

COVAR_POP 

,

COVAR_SAMP 

,

REGR_SXX 

,

REGR_SYY 

windowed aggregate functions.

Definition at line 1043 of file StandardConvertletTable.java.

Constructor & Destructor Documentation

org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.RegrCovarianceConvertlet ( SqlKind  kind )

inlinepackage

Definition at line 1046 of file StandardConvertletTable.java.

                                           {
      this.kind = kind;
    }

Member Function Documentation

RexNode org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.convertCall ( SqlRexContext  cx, SqlCall  call  )

inline

Definition at line 1050 of file StandardConvertletTable.java.

References run_benchmark_import.type.

                                                               {
      assert call.operandCount() == 2;
      final SqlNode arg1 = call.operand(0);
      final SqlNode arg2 = call.operand(1);
      final SqlNode expr;
      final RelDataType type = cx.getValidator().getValidatedNodeType(call);
      switch (kind) {
        case COVAR_POP:
          expr = expandCovariance(arg1, arg2, null, type, cx, true);
          break;
        case COVAR_SAMP:
          expr = expandCovariance(arg1, arg2, null, type, cx, false);
          break;
        case REGR_SXX:
          expr = expandRegrSzz(arg2, arg1, type, cx, true);
          break;
        case REGR_SYY:
          expr = expandRegrSzz(arg1, arg2, type, cx, true);
          break;
        default:
          throw Util.unexpected(kind);
      }
      RexNode rex = cx.convertExpression(expr);
      return cx.getRexBuilder().ensureType(type, rex, true);
    }

SqlNode org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.expandCovariance ( final SqlNode  arg0Input, final SqlNode  arg1Input, final SqlNode  dependent, final RelDataType  varType, final SqlRexContext  cx, boolean  biased  )

inlineprivate

Definition at line 1091 of file StandardConvertletTable.java.

                            {
      // covar_pop(x1, x2) ==>
      //     (sum(x1 * x2) - sum(x2) * sum(x1) / count(x1, x2))
      //     / count(x1, x2)
      //
      // covar_samp(x1, x2) ==>
      //     (sum(x1 * x2) - sum(x1) * sum(x2) / count(x1, x2))
      //     / (count(x1, x2) - 1)
      final SqlParserPos pos = SqlParserPos.ZERO;
      final SqlLiteral nullLiteral = SqlLiteral.createNull(SqlParserPos.ZERO);

      final RexNode arg0Rex = cx.convertExpression(arg0Input);
      final RexNode arg1Rex = cx.convertExpression(arg1Input);

      final SqlNode arg0 = getCastedSqlNode(arg0Input, varType, pos, arg0Rex);
      final SqlNode arg1 = getCastedSqlNode(arg1Input, varType, pos, arg1Rex);
      final SqlNode argSquared = SqlStdOperatorTable.MULTIPLY.createCall(pos, arg0, arg1);
      final SqlNode sumArgSquared;
      final SqlNode sum0;
      final SqlNode sum1;
      final SqlNode count;
      if (dependent == null) {
        sumArgSquared = SqlStdOperatorTable.SUM.createCall(pos, argSquared);
        sum0 = SqlStdOperatorTable.SUM.createCall(pos, arg0, arg1);
        sum1 = SqlStdOperatorTable.SUM.createCall(pos, arg1, arg0);
        count = SqlStdOperatorTable.REGR_COUNT.createCall(pos, arg0, arg1);
      } else {
        sumArgSquared = SqlStdOperatorTable.SUM.createCall(pos, argSquared, dependent);
        sum0 = SqlStdOperatorTable.SUM.createCall(
                pos, arg0, Objects.equals(dependent, arg0Input) ? arg1 : dependent);
        sum1 = SqlStdOperatorTable.SUM.createCall(
                pos, arg1, Objects.equals(dependent, arg1Input) ? arg0 : dependent);
        count = SqlStdOperatorTable.REGR_COUNT.createCall(
                pos, arg0, Objects.equals(dependent, arg0Input) ? arg1 : dependent);
      }

      final SqlNode sumSquared = SqlStdOperatorTable.MULTIPLY.createCall(pos, sum0, sum1);
      final SqlNode countCasted =
              getCastedSqlNode(count, varType, pos, cx.convertExpression(count));

      final SqlNode avgSumSquared =
              SqlStdOperatorTable.DIVIDE.createCall(pos, sumSquared, countCasted);
      final SqlNode diff =
              SqlStdOperatorTable.MINUS.createCall(pos, sumArgSquared, avgSumSquared);
      SqlNode denominator;
      if (biased) {
        denominator = countCasted;
      } else {
        final SqlNumericLiteral one = SqlLiteral.createExactNumeric("1", pos);
        denominator = new SqlCase(SqlParserPos.ZERO,
                countCasted,
                SqlNodeList.of(
                        SqlStdOperatorTable.EQUALS.createCall(pos, countCasted, one)),
                SqlNodeList.of(getCastedSqlNode(nullLiteral, varType, pos, null)),
                SqlStdOperatorTable.MINUS.createCall(pos, countCasted, one));
      }

      return SqlStdOperatorTable.DIVIDE.createCall(pos, diff, denominator);
    }

SqlNode org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.expandRegrSzz ( final SqlNode  arg1, final SqlNode  arg2, final RelDataType  avgType, final SqlRexContext  cx, boolean  variance  )

inlineprivate

Definition at line 1076 of file StandardConvertletTable.java.

                              {
      final SqlParserPos pos = SqlParserPos.ZERO;
      final SqlNode count = SqlStdOperatorTable.REGR_COUNT.createCall(pos, arg1, arg2);
      final SqlNode varPop =
              expandCovariance(arg1, variance ? arg1 : arg2, arg2, avgType, cx, true);
      final RexNode varPopRex = cx.convertExpression(varPop);
      final SqlNode varPopCast;
      varPopCast = getCastedSqlNode(varPop, avgType, pos, varPopRex);
      return SqlStdOperatorTable.MULTIPLY.createCall(pos, varPopCast, count);
    }

SqlNode org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.getCastedSqlNode ( SqlNode  argInput, RelDataType  varType, SqlParserPos  pos, RexNode  argRex  )

inlineprivate

Definition at line 1156 of file StandardConvertletTable.java.

                                                                                     {
      SqlNode arg;
      if (argRex != null && !argRex.getType().equals(varType)) {
        arg = SqlStdOperatorTable.CAST.createCall(
                pos, argInput, SqlTypeUtil.convertTypeToSpec(varType));
      } else {
        arg = argInput;
      }
      return arg;
    }

Member Data Documentation

final SqlKind org.apache.calcite.sql2rel.StandardConvertletTable.RegrCovarianceConvertlet.kind

private

Definition at line 1044 of file StandardConvertletTable.java.

Referenced by TableFunctionsFactory_linker.BaseGenerateFiles._get_filename(), heavydb.serialized_result_set.ttypes.TTargetInfo.read(), and heavydb.serialized_result_set.ttypes.TTargetInfo.write().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/java/calcite/src/main/java/org/apache/calcite/sql2rel/StandardConvertletTable.java