public override NodeInfo Visit(ProjectOp op, Node n) { ExtendedNodeInfo extendedNodeInfo1 = this.InitExtendedNodeInfo(n); ExtendedNodeInfo extendedNodeInfo2 = this.GetExtendedNodeInfo(n.Child0); foreach (Var output in op.Outputs) { if (extendedNodeInfo2.Definitions.IsSet(output)) { extendedNodeInfo1.Definitions.Set(output); } else { extendedNodeInfo1.ExternalReferences.Set(output); } } extendedNodeInfo1.NonNullableDefinitions.InitFrom(extendedNodeInfo2.NonNullableDefinitions); extendedNodeInfo1.NonNullableDefinitions.And(op.Outputs); extendedNodeInfo1.NonNullableVisibleDefinitions.InitFrom(extendedNodeInfo2.NonNullableDefinitions); foreach (Node child in n.Child1.Children) { VarDefOp op1 = child.Op as VarDefOp; NodeInfo nodeInfo = this.GetNodeInfo(child.Child0); extendedNodeInfo1.LocalDefinitions.Set(op1.Var); extendedNodeInfo1.ExternalReferences.Clear(op1.Var); extendedNodeInfo1.Definitions.Set(op1.Var); extendedNodeInfo1.ExternalReferences.Or(nodeInfo.ExternalReferences); if (NodeInfoVisitor.IsDefinitionNonNullable(child.Child0, extendedNodeInfo1.NonNullableVisibleDefinitions)) { extendedNodeInfo1.NonNullableDefinitions.Set(op1.Var); } } extendedNodeInfo1.ExternalReferences.Minus(extendedNodeInfo2.Definitions); extendedNodeInfo1.ExternalReferences.Or(extendedNodeInfo2.ExternalReferences); extendedNodeInfo1.Keys.NoKeys = true; if (!extendedNodeInfo2.Keys.NoKeys) { VarVec varVec1 = this.m_command.CreateVarVec(extendedNodeInfo2.Keys.KeyVars).Remap(NodeInfoVisitor.ComputeVarRemappings(n.Child1)); VarVec varVec2 = varVec1.Clone(); VarVec varVec3 = this.m_command.CreateVarVec(op.Outputs); varVec1.Minus(varVec3); if (varVec1.IsEmpty) { extendedNodeInfo1.Keys.InitFrom((IEnumerable <Var>)varVec2); } } extendedNodeInfo1.InitRowCountFrom(extendedNodeInfo2); return((NodeInfo)extendedNodeInfo1); }
protected override NodeInfo VisitSortOp(SortBaseOp op, Node n) { ExtendedNodeInfo extendedNodeInfo1 = this.InitExtendedNodeInfo(n); ExtendedNodeInfo extendedNodeInfo2 = this.GetExtendedNodeInfo(n.Child0); extendedNodeInfo1.Definitions.Or(extendedNodeInfo2.Definitions); extendedNodeInfo1.ExternalReferences.Or(extendedNodeInfo2.ExternalReferences); extendedNodeInfo1.ExternalReferences.Minus(extendedNodeInfo2.Definitions); extendedNodeInfo1.Keys.InitFrom(extendedNodeInfo2.Keys); extendedNodeInfo1.NonNullableDefinitions.InitFrom(extendedNodeInfo2.NonNullableDefinitions); extendedNodeInfo1.NonNullableVisibleDefinitions.InitFrom(extendedNodeInfo2.NonNullableDefinitions); extendedNodeInfo1.InitRowCountFrom(extendedNodeInfo2); if (OpType.ConstrainedSort == op.OpType && n.Child2.Op.OpType == OpType.Constant && !((ConstrainedSortOp)op).WithTies) { ConstantBaseOp op1 = (ConstantBaseOp)n.Child2.Op; if (TypeHelpers.IsIntegerConstant(op1.Type, op1.Value, 1L)) { extendedNodeInfo1.SetRowCount(RowCount.Zero, RowCount.One); } } return((NodeInfo)extendedNodeInfo1); }
public override NodeInfo Visit(DistinctOp op, Node n) { ExtendedNodeInfo extendedNodeInfo1 = this.InitExtendedNodeInfo(n); extendedNodeInfo1.Keys.InitFrom((IEnumerable <Var>)op.Keys, true); ExtendedNodeInfo extendedNodeInfo2 = this.GetExtendedNodeInfo(n.Child0); extendedNodeInfo1.ExternalReferences.InitFrom(extendedNodeInfo2.ExternalReferences); foreach (Var key in op.Keys) { if (extendedNodeInfo2.Definitions.IsSet(key)) { extendedNodeInfo1.Definitions.Set(key); } else { extendedNodeInfo1.ExternalReferences.Set(key); } } extendedNodeInfo1.NonNullableDefinitions.InitFrom(extendedNodeInfo2.NonNullableDefinitions); extendedNodeInfo1.NonNullableDefinitions.And(op.Keys); extendedNodeInfo1.InitRowCountFrom(extendedNodeInfo2); return((NodeInfo)extendedNodeInfo1); }