| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 at |
| * |
| * 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 org.apache.carbondata.geo.scan.expression; |
| |
| import java.io.IOException; |
| import java.io.ObjectInputStream; |
| import java.io.ObjectOutputStream; |
| import java.util.ArrayList; |
| import java.util.Arrays; |
| import java.util.List; |
| |
| import org.apache.carbondata.common.annotations.InterfaceAudience; |
| import org.apache.carbondata.core.datastore.block.SegmentProperties; |
| import org.apache.carbondata.core.metadata.datatype.DataTypes; |
| import org.apache.carbondata.core.scan.expression.ColumnExpression; |
| import org.apache.carbondata.core.scan.expression.Expression; |
| import org.apache.carbondata.core.scan.expression.ExpressionResult; |
| import org.apache.carbondata.core.scan.expression.UnknownExpression; |
| import org.apache.carbondata.core.scan.expression.conditional.ConditionalExpression; |
| import org.apache.carbondata.core.scan.filter.executer.FilterExecutor; |
| import org.apache.carbondata.core.scan.filter.intf.ExpressionType; |
| import org.apache.carbondata.core.scan.filter.intf.RowIntf; |
| import org.apache.carbondata.core.scan.filter.resolver.FilterResolverIntf; |
| import org.apache.carbondata.core.scan.filter.resolver.RowLevelFilterResolverImpl; |
| import org.apache.carbondata.core.util.CustomIndex; |
| import org.apache.carbondata.geo.GeoHashUtils; |
| import org.apache.carbondata.geo.scan.filter.executor.PolygonFilterExecutorImpl; |
| |
| /** |
| * InPolygon expression processor. It inputs the InPolygon string to the Geo implementation's |
| * query method, gets the list of ranges of IDs to filter as an output. And then, build |
| * InExpression with list of all the IDs present in those list of ranges. |
| */ |
| @InterfaceAudience.Internal |
| public class PolygonExpression extends UnknownExpression implements ConditionalExpression { |
| public String polygon; |
| public CustomIndex<List<Long[]>> instance; |
| public List<Long[]> ranges = new ArrayList<Long[]>(); |
| private ColumnExpression column; |
| private static final ExpressionResult trueExpRes = |
| new ExpressionResult(DataTypes.BOOLEAN, true); |
| private static final ExpressionResult falseExpRes = |
| new ExpressionResult(DataTypes.BOOLEAN, false); |
| |
| public PolygonExpression(String polygon, String columnName, CustomIndex indexInstance) { |
| this.polygon = polygon; |
| this.instance = indexInstance; |
| this.column = new ColumnExpression(columnName, DataTypes.LONG); |
| } |
| |
| /** |
| * This method calls the query processor and gets the list of ranges of IDs. |
| */ |
| public void processExpression() { |
| try { |
| ranges = instance.query(polygon); |
| GeoHashUtils.validateRangeList(ranges); |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } |
| } |
| |
| public List<Long[]> getRanges() { |
| return ranges; |
| } |
| |
| @Override |
| public ExpressionResult evaluate(RowIntf value) { |
| if (GeoHashUtils.rangeBinarySearch(ranges, (Long) value.getVal(0))) { |
| return trueExpRes; |
| } |
| return falseExpRes; |
| } |
| |
| @Override |
| public ExpressionType getFilterExpressionType() { |
| return ExpressionType.UNKNOWN; |
| } |
| |
| @Override |
| public List<Expression> getChildren() { |
| if (ranges.isEmpty()) { |
| processExpression(); |
| } |
| return super.getChildren(); |
| } |
| |
| @Override |
| public void findAndSetChild(Expression oldExpr, Expression newExpr) { |
| } |
| |
| @Override |
| public String getString() { |
| return getStatement(); |
| } |
| |
| @Override |
| public String getStatement() { |
| return "IN_POLYGON('" + polygon + "')"; |
| } |
| |
| private void writeObject(ObjectOutputStream out) throws IOException { |
| out.writeObject(polygon); |
| out.writeObject(instance); |
| out.writeObject(column); |
| out.writeObject(ranges); |
| } |
| |
| private void readObject(ObjectInputStream in) throws IOException, ClassNotFoundException { |
| polygon = (String) in.readObject(); |
| instance = (CustomIndex<List<Long[]>>) in.readObject(); |
| column = (ColumnExpression) in.readObject(); |
| ranges = (List<Long[]>) in.readObject(); |
| } |
| |
| @Override |
| public List<ColumnExpression> getAllColumnList() { |
| return new ArrayList<ColumnExpression>(Arrays.asList(column)); |
| } |
| |
| @Override |
| public List<ColumnExpression> getColumnList() { |
| return getAllColumnList(); |
| } |
| |
| @Override |
| public boolean isSingleColumn() { |
| return true; |
| } |
| |
| @Override |
| public List<ExpressionResult> getLiterals() { |
| return null; |
| } |
| |
| @Override |
| public FilterExecutor getFilterExecutor(FilterResolverIntf resolver, |
| SegmentProperties segmentProperties) { |
| assert (resolver instanceof RowLevelFilterResolverImpl); |
| RowLevelFilterResolverImpl rowLevelResolver = (RowLevelFilterResolverImpl) resolver; |
| return new PolygonFilterExecutorImpl(rowLevelResolver.getDimColEvaluatorInfoList(), |
| rowLevelResolver.getMsrColEvalutorInfoList(), rowLevelResolver.getFilterExpresion(), |
| rowLevelResolver.getTableIdentifier(), segmentProperties, null); |
| } |
| } |