b56d4519fbd1e0fa6c58d7d11cf8fc858ff095c4
[hive.git] / ql / src / gen / vectorization / ExpressionTemplates / FilterStringGroupColumnCompareStringGroupScalarBase.txt
1 /**
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements.  See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership.  The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License.  You may obtain a copy of the License at
9  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  */
18  
19 package org.apache.hadoop.hive.ql.exec.vector.expressions.gen;
20
21 import java.nio.charset.StandardCharsets;
22
23 import org.apache.hadoop.hive.ql.exec.vector.expressions.VectorExpression;
24 import org.apache.hadoop.hive.ql.exec.vector.expressions.StringExpr;
25 import org.apache.hadoop.hive.ql.exec.vector.BytesColumnVector;
26 import org.apache.hadoop.hive.ql.exec.vector.VectorizedRowBatch;
27 import org.apache.hadoop.hive.ql.exec.vector.VectorExpressionDescriptor;
28
29 /**
30  * This is a generated class to evaluate a <OperatorSymbol> comparison on a vector of strings.
31  */
32 public abstract class <ClassName> extends VectorExpression {
33
34   private static final long serialVersionUID = 1L;
35
36   protected int colNum;
37   protected byte[] value;
38
39   @Override
40   public void evaluate(VectorizedRowBatch batch) {
41     if (childExpressions != null) {
42       super.evaluateChildren(batch);
43     }
44     BytesColumnVector inputColVector = (BytesColumnVector) batch.cols[colNum];
45     int[] sel = batch.selected;
46     boolean[] nullPos = inputColVector.isNull;
47     int n = batch.size;
48     byte[][] vector = inputColVector.vector;
49     int[] length = inputColVector.length;
50     int[] start = inputColVector.start;
51     
52
53     // return immediately if batch is empty
54     if (n == 0) {
55       return;
56     }
57     
58     if (inputColVector.noNulls) {
59       if (inputColVector.isRepeating) {
60       
61         // All must be selected otherwise size would be zero. Repeating property will not change.
62         if (!(<CompareOrEqual>(vector[0], start[0], length[0], value, 0, value.length)<OptionalCompare>)) {
63
64           //Entire batch is filtered out.
65           batch.size = 0;
66         }
67       } else if (batch.selectedInUse) {
68         int newSize = 0;
69         for(int j=0; j != n; j++) {
70           int i = sel[j];
71           if (<CompareOrEqual>(vector[i], start[i], length[i], value, 0, value.length)<OptionalCompare>) {
72             sel[newSize++] = i;
73           }
74         }
75         batch.size = newSize;
76       } else {
77         int newSize = 0;
78         for(int i = 0; i != n; i++) {
79           if (<CompareOrEqual>(vector[i], start[i], length[i], value, 0, value.length)<OptionalCompare>) {
80             sel[newSize++] = i;
81           }
82         }
83         if (newSize < n) {
84           batch.size = newSize;
85           batch.selectedInUse = true;
86         }
87       }
88     } else {
89       if (inputColVector.isRepeating) {
90       
91         // All must be selected otherwise size would be zero. Repeating property will not change.
92         if (!nullPos[0]) {
93           if (!(<CompareOrEqual>(vector[0], start[0], length[0], value, 0, value.length)<OptionalCompare>)) {
94
95             //Entire batch is filtered out.
96             batch.size = 0;
97           }
98         } else {
99           batch.size = 0;
100         }
101       } else if (batch.selectedInUse) {
102         int newSize = 0;
103         for(int j=0; j != n; j++) {
104           int i = sel[j];
105           if (!nullPos[i]) {
106            if (<CompareOrEqual>(vector[i], start[i], length[i], value, 0, value.length)<OptionalCompare>) {
107              sel[newSize++] = i;
108            }
109           }
110         }
111         
112         //Change the selected vector
113         batch.size = newSize;
114       } else {
115         int newSize = 0;
116         for(int i = 0; i != n; i++) {
117           if (!nullPos[i]) {
118             if (<CompareOrEqual>(vector[i], start[i], length[i], value, 0, value.length)<OptionalCompare>) {
119               sel[newSize++] = i;
120             }
121           }
122         }
123         if (newSize < n) {
124           batch.size = newSize;
125           batch.selectedInUse = true;
126         }
127       }
128     }
129   }
130
131   @Override
132   public int getOutputColumn() {
133     return -1;
134   }
135
136   @Override
137   public String getOutputType() {
138     return "boolean";
139   }
140   
141   public int getColNum() {
142     return colNum;
143   }
144
145   public void setColNum(int colNum) {
146     this.colNum = colNum;
147   }
148
149   public byte[] getValue() {
150     return value;
151   }
152
153   public void setValue(byte[] value) {
154     this.value = value;
155   }
156
157   @Override
158   public String vectorExpressionParameters() {
159     return "col " + colNum + ", val " + new String(value, StandardCharsets.UTF_8);
160   }
161
162 }