How to use

getInput

method

in

org.apache.calcite.rel.core.Calc

  @Override
  public RelNode convert(RelNode rel) {
    final Calc calc = (Calc) rel;
    final RelNode input = calc.getInput();

    return new StreamsCalcRel(calc.getCluster(), calc.getTraitSet().replace(StreamsLogicalConvention.INSTANCE),
                 convert(input, input.getTraitSet().replace(StreamsLogicalConvention.INSTANCE)),
                 calc.getProgram());
  }
}

@Override public double estimateRowCount(RelMetadataQuery mq) {
 return RelMdUtil.estimateFilteredRows(getInput(), program, mq);
}

@Override public double estimateRowCount(RelMetadataQuery mq) {
 return RelMdUtil.estimateFilteredRows(getInput(), program, mq);
}

public ImmutableList<RelCollation> collations(Calc calc,
  RelMetadataQuery mq) {
 return ImmutableList.copyOf(calc(mq, calc.getInput(), calc.getProgram()));
}

public ImmutableList<RelCollation> collations(Calc calc,
  RelMetadataQuery mq) {
 return ImmutableList.copyOf(calc(mq, calc.getInput(), calc.getProgram()));
}

public Double getRowCount(Calc rel, RelMetadataQuery mq) {
 return RelMdUtil.estimateFilteredRows(rel.getInput(), rel.getProgram(), mq);
}

public Double getRowCount(Calc rel, RelMetadataQuery mq) {
 return RelMdUtil.estimateFilteredRows(rel.getInput(), rel.getProgram(), mq);
}

@Override public RelOptCost computeSelfCost(RelOptPlanner planner,
  RelMetadataQuery mq) {
 double dRows = mq.getRowCount(this);
 double dCpu = mq.getRowCount(getInput())
   * program.getExprCount();
 double dIo = 0;
 return planner.getCostFactory().makeCost(dRows, dCpu, dIo);
}

@Override public RelOptCost computeSelfCost(RelOptPlanner planner,
  RelMetadataQuery mq) {
 double dRows = mq.getRowCount(this);
 double dCpu = mq.getRowCount(getInput())
   * program.getExprCount();
 double dIo = 0;
 return planner.getCostFactory().makeCost(dRows, dCpu, dIo);
}

  public void onMatch(RelOptRuleCall call) {
    final Aggregate aggregate = call.rel(0);
    final Calc calc = call.rel(1);
    final RexProgram program = calc.getProgram();
    final List<RexNode> projects = new ArrayList<>();
    for (RexLocalRef localRef : program.getProjectList()) {
      projects.add(program.expandLocalRef(localRef));
    }
    final Project project = LogicalProject.create(calc.getInput(), projects, calc.getRowType());
    RelNode x = AggregateProjectMergeRule.apply(call, aggregate, project);
    if (x != null) {
      call.transformTo(x);
    }
  }
}

/**
 * Constructs a CalcRelSplitter.
 *
 * @param calc     Calc to split
 * @param relTypes Array of rel types, e.g. {Java, Fennel}. Must be
 *                 distinct.
 */
CalcRelSplitter(Calc calc, RelBuilder relBuilder, RelType[] relTypes) {
 this.relBuilder = relBuilder;
 for (int i = 0; i < relTypes.length; i++) {
  assert relTypes[i] != null;
  for (int j = 0; j < i; j++) {
   assert relTypes[i] != relTypes[j]
     : "Rel types must be distinct";
  }
 }
 this.program = calc.getProgram();
 this.cluster = calc.getCluster();
 this.traits = calc.getTraitSet();
 this.typeFactory = calc.getCluster().getTypeFactory();
 this.child = calc.getInput();
 this.relTypes = relTypes;
}

/**
 * Constructs a CalcRelSplitter.
 *
 * @param calc     Calc to split
 * @param relTypes Array of rel types, e.g. {Java, Fennel}. Must be
 *                 distinct.
 */
CalcRelSplitter(Calc calc, RelBuilder relBuilder, RelType[] relTypes) {
 this.relBuilder = relBuilder;
 for (int i = 0; i < relTypes.length; i++) {
  assert relTypes[i] != null;
  for (int j = 0; j < i; j++) {
   assert relTypes[i] != relTypes[j]
     : "Rel types must be distinct";
  }
 }
 this.program = calc.getProgram();
 this.cluster = calc.getCluster();
 this.traits = calc.getTraitSet();
 this.typeFactory = calc.getCluster().getTypeFactory();
 this.child = calc.getInput();
 this.relTypes = relTypes;
}

 public RelNode convert(RelNode rel) {
  final Calc calc = (Calc) rel;
  // If there's a multiset, let FarragoMultisetSplitter work on it
  // first.
  if (RexMultisetUtil.containsMultiset(calc.getProgram())) {
   return null;
  }
  return new JdbcCalc(rel.getCluster(), rel.getTraitSet().replace(out),
    convert(calc.getInput(), calc.getTraitSet().replace(out)),
    calc.getProgram());
 }
}

 public RelNode convert(RelNode rel) {
  final Calc calc = (Calc) rel;
  // If there's a multiset, let FarragoMultisetSplitter work on it
  // first.
  if (RexMultisetUtil.containsMultiset(calc.getProgram())) {
   return null;
  }
  return new JdbcCalc(rel.getCluster(), rel.getTraitSet().replace(out),
    convert(calc.getInput(), calc.getTraitSet().replace(out)),
    calc.getProgram());
 }
}

 @Override public void onMatch(RelOptRuleCall call) {
  final Calc calc = call.rel(0);
  final Pair<ImmutableList<RexNode>, ImmutableList<RexNode>> projectFilter =
    calc.getProgram().split();
  final RelBuilder relBuilder = call.builder();
  relBuilder.push(calc.getInput());
  relBuilder.filter(projectFilter.right);
  relBuilder.project(projectFilter.left, calc.getRowType().getFieldNames());
  call.transformTo(relBuilder.build());
 }
}

 @Override public void onMatch(RelOptRuleCall call) {
  final Calc calc = call.rel(0);
  final Pair<ImmutableList<RexNode>, ImmutableList<RexNode>> projectFilter =
    calc.getProgram().split();
  final RelBuilder relBuilder = call.builder();
  relBuilder.push(calc.getInput());
  relBuilder.filter(projectFilter.right);
  relBuilder.project(projectFilter.left, calc.getRowType().getFieldNames());
  call.transformTo(relBuilder.build());
 }
}

public boolean isValid(Litmus litmus, Context context) {
 if (!RelOptUtil.equal(
   "program's input type",
   program.getInputRowType(),
   "child's output type",
   getInput().getRowType(), litmus)) {
  return litmus.fail(null);
 }
 if (!program.isValid(litmus, context)) {
  return litmus.fail(null);
 }
 if (!program.isNormalized(litmus, getCluster().getRexBuilder())) {
  return litmus.fail(null);
 }
 return litmus.succeed();
}

public boolean isValid(Litmus litmus, Context context) {
 if (!RelOptUtil.equal(
   "program's input type",
   program.getInputRowType(),
   "child's output type",
   getInput().getRowType(), litmus)) {
  return litmus.fail(null);
 }
 if (!program.isValid(litmus, context)) {
  return litmus.fail(null);
 }
 if (!program.isNormalized(litmus, getCluster().getRexBuilder())) {
  return litmus.fail(null);
 }
 return litmus.succeed();
}

public Result visitCalc(Calc e) {
 Result x = visitChild(0, e.getInput());
 final RexProgram program = e.getProgram();
 Builder builder =
   program.getCondition() != null
     ? x.builder(e, Clause.WHERE)
     : x.builder(e);
 if (!isStar(program)) {
  final List<SqlNode> selectList = new ArrayList<>();
  for (RexLocalRef ref : program.getProjectList()) {
   SqlNode sqlExpr = builder.context.toSql(program, ref);
   addSelect(selectList, sqlExpr, e.getRowType());
  }
  builder.setSelect(new SqlNodeList(selectList, POS));
 }
 if (program.getCondition() != null) {
  builder.setWhere(
    builder.context.toSql(program, program.getCondition()));
 }
 return builder.result();
}

/** @see #dispatch */
public Result visit(Calc e) {
 Result x = visitChild(0, e.getInput());
 parseCorrelTable(e, x);
 final RexProgram program = e.getProgram();
 Builder builder =
   program.getCondition() != null
     ? x.builder(e, Clause.WHERE)
     : x.builder(e);
 if (!isStar(program)) {
  final List<SqlNode> selectList = new ArrayList<>();
  for (RexLocalRef ref : program.getProjectList()) {
   SqlNode sqlExpr = builder.context.toSql(program, ref);
   addSelect(selectList, sqlExpr, e.getRowType());
  }
  builder.setSelect(new SqlNodeList(selectList, POS));
 }
 if (program.getCondition() != null) {
  builder.setWhere(
    builder.context.toSql(program, program.getCondition()));
 }
 return builder.result();
}