public DateScalarSubtractDateColumn(long value, int colNum, int outputColumn) {
this.colNum = colNum;
this.value = new Timestamp(0);
this.value.setTime(DateWritable.daysToMillis((int) value));
this.outputColumn = outputColumn;
scratchTimestamp2 = new Timestamp(0);
}
/* copied over from VectorUDFTimestampFieldLong */
private TimestampWritable toTimestampWritable(long daysSinceEpoch) {
Timestamp ts = new Timestamp(DateWritable.daysToMillis((int) daysSinceEpoch));
return new TimestampWritable(ts);
}
@Override
/**
* Method to evaluate scalar-column operation in vectorized fashion.
*
* @batch a package of rows with each column stored in a vector
*/
public void evaluate(VectorizedRowBatch batch) {
if (childExpressions != null) {
super.evaluateChildren(batch);
}
// Input #2 is type date (epochDays).
LongColumnVector inputColVector2 = (LongColumnVector) batch.cols[colNum];
// Output is type HiveIntervalDayTime.
IntervalDayTimeColumnVector outputColVector =
(IntervalDayTimeColumnVector) batch.cols[outputColumn];
int[] sel = batch.selected;
boolean[] inputIsNull = inputColVector2.isNull;
boolean[] outputIsNull = outputColVector.isNull;
outputColVector.noNulls = inputColVector2.noNulls;
outputColVector.isRepeating = inputColVector2.isRepeating;
int n = batch.size;
long[] vector2 = inputColVector2.vector;
// return immediately if batch is empty
if (n == 0) {
return;
}
if (inputColVector2.isRepeating) {
scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[0]));
dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime());
outputColVector.setFromScratchIntervalDayTime(0);
// Even if there are no nulls, we always copy over entry 0. Simplifies code.
outputIsNull[0] = inputIsNull[0];
} else if (inputColVector2.noNulls) {
if (batch.selectedInUse) {
for (int j = 0; j != n; j++) {
int i = sel[j];
scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i]));
dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime());
outputColVector.setFromScratchIntervalDayTime(i);
}
} else {
for (int i = 0; i != n; i++) {
scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i]));
dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime());
outputColVector.setFromScratchIntervalDayTime(i);
}
}
} else {
/* there are nulls */
if (batch.selectedInUse) {
for (int j = 0; j != n; j++) {
int i = sel[j];
scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i]));
dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime());
outputColVector.setFromScratchIntervalDayTime(i);
outputIsNull[i] = inputIsNull[i];
}
} else {
for (int i = 0; i != n; i++) {
scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i]));
dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime());
outputColVector.setFromScratchIntervalDayTime(i);
}
System.arraycopy(inputIsNull, 0, outputIsNull, 0, n);
}
}
NullUtil.setNullOutputEntriesColScalar(outputColVector, batch.selectedInUse, sel, n);
}
