blob: a76237b418d4348ef5be102518f63f941e6ca5d6 [file] [log] [blame]
/*
* Copyright 2009-2010 by The Regents of the University of California
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* you may obtain a copy of the License from
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package edu.uci.ics.hyracks.algebricks.core.algebra.operators.logical;
import java.util.ArrayList;
import org.apache.commons.lang3.mutable.Mutable;
import edu.uci.ics.hyracks.algebricks.core.algebra.base.ILogicalExpression;
import edu.uci.ics.hyracks.algebricks.core.algebra.base.LogicalOperatorTag;
import edu.uci.ics.hyracks.algebricks.core.algebra.base.LogicalVariable;
import edu.uci.ics.hyracks.algebricks.core.algebra.expressions.IVariableTypeEnvironment;
import edu.uci.ics.hyracks.algebricks.core.algebra.properties.VariablePropagationPolicy;
import edu.uci.ics.hyracks.algebricks.core.algebra.typing.ITypingContext;
import edu.uci.ics.hyracks.algebricks.core.algebra.visitors.ILogicalExpressionReferenceTransform;
import edu.uci.ics.hyracks.algebricks.core.algebra.visitors.ILogicalOperatorVisitor;
import edu.uci.ics.hyracks.algebricks.core.api.exceptions.AlgebricksException;
public class PartitioningSplitOperator extends AbstractLogicalOperator {
private Mutable<ILogicalExpression>[] expressions;
private boolean hasDefault;
public PartitioningSplitOperator(Mutable<ILogicalExpression>[] exprList, boolean defaultBranchDefined) {
expressions = exprList;
hasDefault = defaultBranchDefined;
}
public Mutable<ILogicalExpression>[] getExpressions() {
return expressions;
}
public boolean hasDefault() {
return hasDefault;
}
@Override
public LogicalOperatorTag getOperatorTag() {
return LogicalOperatorTag.PARTITIONINGSPLIT;
}
@Override
public void recomputeSchema() {
schema = new ArrayList<LogicalVariable>();
schema.addAll(inputs.get(0).getValue().getSchema());
}
@Override
public VariablePropagationPolicy getVariablePropagationPolicy() {
return VariablePropagationPolicy.ALL;
}
@Override
public boolean acceptExpressionTransform(ILogicalExpressionReferenceTransform visitor) throws AlgebricksException {
boolean b = false;
for (int i = 0; i < expressions.length; i++) {
if (visitor.transform(expressions[i])) {
b = true;
}
}
return b;
}
@Override
public <R, T> R accept(ILogicalOperatorVisitor<R, T> visitor, T arg) throws AlgebricksException {
return visitor.visitPartitioningSplitOperator(this, arg);
}
@Override
public boolean isMap() {
return false;
}
@Override
public IVariableTypeEnvironment computeOutputTypeEnvironment(ITypingContext ctx) throws AlgebricksException {
return createPropagatingAllInputsTypeEnvironment(ctx);
}
}