Java Code Examples for org.apache.calcite.rex.RexCall#isA()

The following examples show how to use org.apache.calcite.rex.RexCall#isA() . These examples are extracted from open source projects. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source Project: Bats   File: RelStructuredTypeFlattener.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public RexNode visitCall(RexCall rexCall) {
    if (rexCall.isA(SqlKind.CAST)) {
        RexNode input = rexCall.getOperands().get(0).accept(this);
        RelDataType targetType = removeDistinct(rexCall.getType());
        return rexBuilder.makeCast(targetType, input);
    }
    if (!rexCall.isA(SqlKind.COMPARISON)) {
        return super.visitCall(rexCall);
    }
    RexNode lhs = rexCall.getOperands().get(0);
    if (!lhs.getType().isStruct()) {
        // NOTE jvs 9-Mar-2005: Calls like IS NULL operate
        // on the representative null indicator. Since it comes
        // first, we don't have to do any special translation.
        return super.visitCall(rexCall);
    }

    // NOTE jvs 22-Mar-2005: Likewise, the null indicator takes
    // care of comparison null semantics without any special casing.
    return flattenComparison(rexBuilder, rexCall.getOperator(), rexCall.getOperands());
}
 
Example 2
Source Project: Bats   File: RelMdUtil.java    License: Apache License 2.0 5 votes vote down vote up
public Double visitCall(RexCall call) {
  Double distinctRowCount;
  Double rowCount = mq.getRowCount(rel);
  if (call.isA(SqlKind.MINUS_PREFIX)) {
    distinctRowCount = cardOfProjExpr(mq, rel, call.getOperands().get(0));
  } else if (call.isA(ImmutableList.of(SqlKind.PLUS, SqlKind.MINUS))) {
    Double card0 = cardOfProjExpr(mq, rel, call.getOperands().get(0));
    if (card0 == null) {
      return null;
    }
    Double card1 = cardOfProjExpr(mq, rel, call.getOperands().get(1));
    if (card1 == null) {
      return null;
    }
    distinctRowCount = Math.max(card0, card1);
  } else if (call.isA(ImmutableList.of(SqlKind.TIMES, SqlKind.DIVIDE))) {
    distinctRowCount =
        NumberUtil.multiply(
            cardOfProjExpr(mq, rel, call.getOperands().get(0)),
            cardOfProjExpr(mq, rel, call.getOperands().get(1)));

    // TODO zfong 6/21/06 - Broadbase has code to handle date
    // functions like year, month, day; E.g., cardinality of Month()
    // is 12
  } else {
    if (call.getOperands().size() == 1) {
      distinctRowCount = cardOfProjExpr(mq, rel, call.getOperands().get(0));
    } else {
      distinctRowCount = rowCount / 10;
    }
  }

  return numDistinctVals(distinctRowCount, rowCount);
}
 
Example 3
Source Project: Bats   File: RelStructuredTypeFlattener.java    License: Apache License 2.0 5 votes vote down vote up
private boolean isConstructor(RexNode rexNode) {
    // TODO jvs 11-Feb-2005: share code with SqlToRelConverter
    if (!(rexNode instanceof RexCall)) {
        return false;
    }
    RexCall call = (RexCall) rexNode;
    return call.getOperator().getName().equalsIgnoreCase("row") || (call.isA(SqlKind.NEW_SPECIFICATION));
}
 
Example 4
Source Project: calcite   File: RelMdUtil.java    License: Apache License 2.0 5 votes vote down vote up
public Double visitCall(RexCall call) {
  Double distinctRowCount;
  Double rowCount = mq.getRowCount(rel);
  if (call.isA(SqlKind.MINUS_PREFIX)) {
    distinctRowCount = cardOfProjExpr(mq, rel, call.getOperands().get(0));
  } else if (call.isA(ImmutableList.of(SqlKind.PLUS, SqlKind.MINUS))) {
    Double card0 = cardOfProjExpr(mq, rel, call.getOperands().get(0));
    if (card0 == null) {
      return null;
    }
    Double card1 = cardOfProjExpr(mq, rel, call.getOperands().get(1));
    if (card1 == null) {
      return null;
    }
    distinctRowCount = Math.max(card0, card1);
  } else if (call.isA(ImmutableList.of(SqlKind.TIMES, SqlKind.DIVIDE))) {
    distinctRowCount =
        NumberUtil.multiply(
            cardOfProjExpr(mq, rel, call.getOperands().get(0)),
            cardOfProjExpr(mq, rel, call.getOperands().get(1)));

    // TODO zfong 6/21/06 - Broadbase has code to handle date
    // functions like year, month, day; E.g., cardinality of Month()
    // is 12
  } else {
    if (call.getOperands().size() == 1) {
      distinctRowCount = cardOfProjExpr(mq, rel, call.getOperands().get(0));
    } else {
      distinctRowCount = rowCount / 10;
    }
  }

  return numDistinctVals(distinctRowCount, rowCount);
}
 
Example 5
private boolean isConstructor(RexNode rexNode) {
  // TODO jvs 11-Feb-2005:  share code with SqlToRelConverter
  if (!(rexNode instanceof RexCall)) {
    return false;
  }
  RexCall call = (RexCall) rexNode;
  return call.getOperator().getName().equalsIgnoreCase("row")
      || (call.isA(SqlKind.NEW_SPECIFICATION));
}
 
Example 6
Source Project: Bats   File: ReduceDecimalsRule.java    License: Apache License 2.0 4 votes vote down vote up
@Override
public boolean canExpand(RexCall call) {
    return call.isA(SqlKind.REINTERPRET) && call.getOperands().get(0).isA(SqlKind.REINTERPRET);
}
 
Example 7
@Override
public RexNode visitCall(RexCall call, final Pointer<SimplifyNLJConditionRule.Side> sidePointer) {
  List<RexNode> newOperands = new ArrayList<>();
  List<SimplifyNLJConditionRule.Side> sides = new ArrayList<>(); call.getOperands().forEach(o -> { Pointer<SimplifyNLJConditionRule.Side> s = new Pointer<>();
    RexNode newOperand = o.accept(this, s);
    sides.add(s.value);
    newOperands.add(newOperand);
  });
  RexNode special = handleSpecial(call, sides, sidePointer, rexBuilder);
  if (special != null) {
    return special.accept(this, sidePointer);
  }
  if (SimplifyNLJConditionRule.Side.shouldPushDown(sides)) {
    List<RexNode> finalOperands = new ArrayList<>();
    for (Pair<RexNode, SimplifyNLJConditionRule.Side> pair : Pair.zip(newOperands, sides)) {
      if (pair.right == SimplifyNLJConditionRule.Side.LEFT || pair.right == SimplifyNLJConditionRule.Side.RIGHT) {
        String digest = pair.left.toString();
        Integer index = exprMap.get(digest);
        if (index == null) {
          index = exprs.size();
          exprs.add(pair.left);
          exprMap.put(digest, index);
          if (pair.right == SimplifyNLJConditionRule.Side.LEFT) {
            leftExprs.add(index);
          } else {
            rightExprs.add(index);
          }
        }
        finalOperands.add(new RexInputRef(index, pair.left.getType()));
      } else {
        finalOperands.add(pair.left);
      }
    }
    sidePointer.value = SimplifyNLJConditionRule.Side.BOTH;
    return rexBuilder.makeCall(call.getOperator(), finalOperands);
  }
  sidePointer.value = SimplifyNLJConditionRule.Side.merge(sides);
  if (call.isA(SqlKind.CAST)) {
    return rexBuilder.makeCast(call.getType(), newOperands.get(0));
  }
  return rexBuilder.makeCall(call.getOperator(), newOperands);
}
 
Example 8
Source Project: calcite   File: ReduceDecimalsRule.java    License: Apache License 2.0 4 votes vote down vote up
public boolean canExpand(RexCall call) {
  return call.isA(SqlKind.REINTERPRET)
      && call.operands.get(0).isA(SqlKind.REINTERPRET);
}