io.deephaven.engine.table.impl.by.SumShortChunk Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation
Engine Table: Implementation and closely-coupled utilities
There is a newer version: 0.36.1
Show newest version
/**
 * Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
 */
/*
 * ---------------------------------------------------------------------------------------------------------------------
 * AUTO-GENERATED CLASS - DO NOT EDIT MANUALLY - for any changes edit SumCharChunk and regenerate
 * ---------------------------------------------------------------------------------------------------------------------
 */
package io.deephaven.engine.table.impl.by;

import io.deephaven.chunk.attributes.Any;
import io.deephaven.util.QueryConstants;
import io.deephaven.chunk.ShortChunk;
import org.apache.commons.lang3.mutable.MutableDouble;
import org.apache.commons.lang3.mutable.MutableInt;

public class SumShortChunk {
    private SumShortChunk() {} // static use only

    static long sumShortChunk(ShortChunk values, int chunkStart, int chunkSize, MutableInt nonNullCount) {
        final int end = chunkStart + chunkSize;
        long sum = 0;
        for (int ii = chunkStart; ii < end; ++ii) {
            final short value = values.get(ii);
            if (value != QueryConstants.NULL_SHORT) {
                sum += value;
                nonNullCount.increment();
            }
        }
        return sum;
    }

    public static long sumShortChunk(ShortChunk values, int chunkStart, int chunkSize) {
        final int end = chunkStart + chunkSize;
        long sum = 0;
        for (int ii = chunkStart; ii < end; ++ii) {
            final short value = values.get(ii);
            if (value != QueryConstants.NULL_SHORT) {
                sum += value;
            }
        }
        return sum;
    }

    /**
     * Produce the sum and sum of squares of a shortacter chunk, as doubles.
     */
    static double sum2ShortChunk(ShortChunk values, int chunkStart, int chunkSize, MutableInt nonNullCount, MutableDouble sum2out) {
        final int end = chunkStart + chunkSize;
        double sum = 0;
        double sum2 = 0;
        for (int ii = chunkStart; ii < end; ++ii) {
            final short value = values.get(ii);
            if (value != QueryConstants.NULL_SHORT) {
                //noinspection UnnecessaryLocalVariable
                final double doubleValue = value;
                sum += doubleValue;
                sum2 += doubleValue * doubleValue;
                nonNullCount.increment();
            }
        }

        sum2out.setValue(sum2);
        return sum;
    }

    static private short abs(short val) {
        if (val == QueryConstants.NULL_SHORT) {
            return val;
        } else if (val < 0) {
            return (short)-val;
        } else {
            return val;
        }
    }

    static long sumShortChunkAbs(ShortChunk values, int chunkStart, int chunkSize, MutableInt nonNullCount) {
        final int end = chunkStart + chunkSize;
        long sum = 0;
        for (int ii = chunkStart; ii < end; ++ii) {
            final short value = abs(values.get(ii));
            if (value != QueryConstants.NULL_SHORT) {
                sum += value;
                nonNullCount.increment();
            }
        }
        return sum;
    }
}