if (nullConstraintBitSet.indexOf(constraintIdx) != -1) { ExprNodeDesc currExpr = TypeCheckProcFactory.toExprNodeDesc(colInfos.get(colExprIdx)); ExprNodeDesc isNotNullUDF = TypeCheckProcFactory.DefaultExprProcessor.
List<String> fieldNames = new LinkedList<String>(); for (RelDataTypeField rdtf : rowType.getFieldList()) { if (bitSet.indexOf(rdtf.getIndex()) != -1) { fieldTypes.add(dtFactory.createTypeWithNullability(rdtf.getType(), false)); } else {
private static int remap(ImmutableBitSet groupSet, int arg) { return arg < 0 ? -1 : groupSet.indexOf(arg); }
private static int remap(ImmutableBitSet groupSet, int arg) { return arg < 0 ? -1 : groupSet.indexOf(arg); }
private static int remap(ImmutableBitSet groupSet, int arg) { return arg < 0 ? -1 : groupSet.indexOf(arg); }
private static Set<RexNode> createAllPossibleExpressions(RexBuilder rexBuilder, RexNode expr, ImmutableBitSet predFieldsUsed, Map<RexInputRef, Set<RexNode>> mapping, Map<RexInputRef, RexNode> singleMapping) { final RexInputRef inputRef = mapping.keySet().iterator().next(); final Set<RexNode> replacements = mapping.remove(inputRef); Set<RexNode> result = new HashSet<>(); assert !replacements.isEmpty(); if (predFieldsUsed.indexOf(inputRef.getIndex()) != -1) { for (RexNode replacement : replacements) { singleMapping.put(inputRef, replacement); createExpressions(rexBuilder, expr, predFieldsUsed, mapping, singleMapping, result); singleMapping.remove(inputRef); } } else { createExpressions(rexBuilder, expr, predFieldsUsed, mapping, singleMapping, result); } mapping.put(inputRef, replacements); return result; }
private static Set<RexNode> createAllPossibleExpressions(RexBuilder rexBuilder, RexNode expr, ImmutableBitSet predFieldsUsed, Map<RexInputRef, Set<RexNode>> mapping, Map<RexInputRef, RexNode> singleMapping) { final RexInputRef inputRef = mapping.keySet().iterator().next(); final Set<RexNode> replacements = mapping.remove(inputRef); Set<RexNode> result = new HashSet<>(); assert !replacements.isEmpty(); if (predFieldsUsed.indexOf(inputRef.getIndex()) != -1) { for (RexNode replacement : replacements) { singleMapping.put(inputRef, replacement); createExpressions(rexBuilder, expr, predFieldsUsed, mapping, singleMapping, result); singleMapping.remove(inputRef); } } else { createExpressions(rexBuilder, expr, predFieldsUsed, mapping, singleMapping, result); } mapping.put(inputRef, replacements); return result; }
@Test public void testIndexOf() { assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(0), equalTo(0)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(2), equalTo(1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(3), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(4), equalTo(2)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(5), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(-1), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(-2), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(-1), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(-2), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(0), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(1000), equalTo(-1)); }
@Test public void testIndexOf() { assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(0), equalTo(0)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(2), equalTo(1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(3), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(4), equalTo(2)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(5), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(-1), equalTo(-1)); assertThat(ImmutableBitSet.of(0, 2, 4).indexOf(-2), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(-1), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(-2), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(0), equalTo(-1)); assertThat(ImmutableBitSet.of().indexOf(1000), equalTo(-1)); }
public int getSourceOpt(int source) { assert aggregate.getIndicatorCount() == 0; if (source < aggregate.getGroupCount()) { int in = tileKey.dimensions.nth(source); return aggregate.getGroupSet().indexOf(in); } Lattice.Measure measure = measures.get(source - aggregate.getGroupCount()); int i = tileKey.measures.indexOf(measure); assert i >= 0; return tileKey.dimensions.cardinality() + i; } } .inverse()));
public int getSourceOpt(int source) { assert aggregate.getIndicatorCount() == 0; if (source < aggregate.getGroupCount()) { int in = tileKey.dimensions.nth(source); return aggregate.getGroupSet().indexOf(in); } Lattice.Measure measure = measures.get(source - aggregate.getGroupCount()); int i = tileKey.measures.indexOf(measure); assert i >= 0; return tileKey.dimensions.cardinality() + i; } } .inverse()));
private static boolean validateTraits(RelTraitSet traitSet, RelNode left, RelNode right) { ImmutableBitSet bitSet = ImmutableBitSet.range(left.getRowType().getFieldCount(), left.getRowType().getFieldCount() + right.getRowType().getFieldCount()); for (RelTrait trait: traitSet) { if (trait.getTraitDef().getTraitClass().equals(RelCollation.class)) { RelCollation collationTrait = (RelCollation)trait; for (RelFieldCollation field : collationTrait.getFieldCollations()) { if (bitSet.indexOf(field.getFieldIndex()) > 0) { return false; } } } else if (trait.getTraitDef().getTraitClass().equals(DrillDistributionTrait.class)) { DrillDistributionTrait distributionTrait = (DrillDistributionTrait) trait; for (DrillDistributionTrait.DistributionField field : distributionTrait.getFields()) { if (bitSet.indexOf(field.getFieldId()) > 0) { return false; } } } } return true; }
for (int i = 0; i < fullGroupSet.length; ++i) { int group = fullGroupSet[i]; int index = newGroupSet.indexOf(group); projectAfterAgg.add(new RexInputRef(index, aggFields.get(i).getType()));
ImmutableBitSet.Builder groupSet = ImmutableBitSet.builder(); for (int key : mergedAggregate.getGroupSet()) { groupSet.set(quarkTile.bitSet().indexOf(key));
int z = tileKey.dimensions.indexOf(arg); if (z < 0) { break tryGroup;
int z = tileKey.dimensions.indexOf(arg); if (z < 0) { break tryGroup;