io.deephaven.engine.table.impl.sources.UngroupedShortVectorColumnSource 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.37.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 UngroupedCharVectorColumnSource and regenerate
 * ---------------------------------------------------------------------------------------------------------------------
 */
package io.deephaven.engine.table.impl.sources;

import io.deephaven.engine.table.ColumnSource;
import io.deephaven.engine.table.impl.MutableColumnSourceGetDefaults;
import io.deephaven.vector.ShortVector;

import static io.deephaven.util.QueryConstants.NULL_SHORT;

public class UngroupedShortVectorColumnSource extends UngroupedColumnSource implements MutableColumnSourceGetDefaults.ForShort {
    private ColumnSource innerSource;
    private final boolean isUngroupable;

    @Override
    public Class getComponentType() {
        return null;
    }


    public UngroupedShortVectorColumnSource(ColumnSource innerSource) {
        super(Short.class);
        this.innerSource = innerSource;
        this.isUngroupable = innerSource instanceof UngroupableColumnSource && ((UngroupableColumnSource)innerSource).isUngroupable();
    }

    @Override
    public Short get(long rowKey) {
        if (rowKey < 0) {
            return null;
        }
        long segment = rowKey >>base;
        int offset = (int) (rowKey & ((1<>base;
        int offset = (int) (rowKey & ((1<> getPrevBase();
        int offset = (int) (rowKey & ((1<< getPrevBase()) - 1));
        final Short result;
        if (isUngroupable) {
            result = (Short)((UngroupableColumnSource)innerSource).getUngroupedPrev(segment, offset);
            if (result == null) {
                return null;
            }
        } else {
            final ShortVector segmentArray = innerSource.getPrev(segment);
            result = segmentArray == null ? NULL_SHORT : segmentArray.get(offset);
        }

        return (result == NULL_SHORT ? null : result);
    }

    @Override
    public short getPrevShort(long rowKey) {
        if (rowKey < 0) {
            return NULL_SHORT;
        }

        long segment = rowKey >> getPrevBase();
        int offset = (int) (rowKey & ((1<< getPrevBase()) - 1));

        if (isUngroupable) {
            return ((UngroupableColumnSource)innerSource).getUngroupedPrevShort(segment, offset);
        }

        final ShortVector segmentArray = innerSource.getPrev(segment);
        return segmentArray == null ? NULL_SHORT : segmentArray.get(offset);
    }

    @Override
    public boolean isImmutable() {
        return false;
    }

    @Override
    public boolean isStateless() {
        return innerSource.isStateless();
    }
}