io.deephaven.engine.table.impl.sources.ByteAsBooleanColumnSource 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
 */
package io.deephaven.engine.table.impl.sources;

import io.deephaven.engine.rowset.chunkattributes.RowKeys;
import io.deephaven.engine.table.SharedContext;
import io.deephaven.engine.rowset.RowSequence;
import io.deephaven.engine.table.ColumnSource;
import io.deephaven.engine.table.impl.AbstractColumnSource;
import io.deephaven.engine.table.impl.MutableColumnSourceGetDefaults;
import io.deephaven.util.BooleanUtils;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.chunk.*;
import org.jetbrains.annotations.NotNull;

/**
 * Reinterpret result {@link ColumnSource} implementations that translates {@link byte} to {@code Boolean} values.
 */
public class ByteAsBooleanColumnSource extends AbstractColumnSource implements MutableColumnSourceGetDefaults.ForBoolean, FillUnordered {

    private final ColumnSource alternateColumnSource;

    public ByteAsBooleanColumnSource(@NotNull final ColumnSource alternateColumnSource) {
        super(Boolean.class);
        this.alternateColumnSource = alternateColumnSource;
    }

    @Override
    public Boolean get(final long rowKey) {
        return BooleanUtils.byteAsBoolean(alternateColumnSource.getByte(rowKey));
    }

    @Override
    public Boolean getPrev(final long rowKey) {
        return BooleanUtils.byteAsBoolean(alternateColumnSource.getPrevByte(rowKey));
    }

    @Override
    public boolean isImmutable() {
        return alternateColumnSource.isImmutable();
    }

    @Override
    public  boolean allowsReinterpret(@NotNull final Class alternateDataType) {
        return alternateDataType == byte.class || alternateDataType == Byte.class;
    }

    @Override
    public  ColumnSource doReinterpret(@NotNull final Class alternateDataType) throws IllegalArgumentException {
        //noinspection unchecked
        return (ColumnSource) alternateColumnSource;
    }

    private class ToBooleanFillContext implements FillContext {
        final GetContext alternateGetContext;
        final FillContext alternateFillContext;
        final WritableByteChunk byteChunk;

        private ToBooleanFillContext(final int chunkCapacity, final SharedContext sharedContext) {
            alternateGetContext = alternateColumnSource.makeGetContext(chunkCapacity, sharedContext);
            if (providesFillUnordered()) {
                alternateFillContext = alternateColumnSource.makeFillContext(chunkCapacity, sharedContext);
                byteChunk = WritableByteChunk.makeWritableChunk(chunkCapacity);
            } else {
                alternateFillContext = null;
                byteChunk = null;
            }
        }

        @Override
        public void close() {
            alternateGetContext.close();
            if (alternateFillContext != null) {
                alternateFillContext.close();
                byteChunk.close();
            }
        }
    }

    @Override
    public FillContext makeFillContext(final int chunkCapacity, final SharedContext sharedContext) {
        return new ToBooleanFillContext(chunkCapacity, sharedContext);
    }

    @Override
    public void fillChunk(@NotNull final FillContext context, @NotNull final WritableChunk destination, @NotNull final RowSequence rowSequence) {
        final ToBooleanFillContext toBooleanFillContext = (ToBooleanFillContext) context;
        final ByteChunk byteChunk = alternateColumnSource.getChunk(toBooleanFillContext.alternateGetContext, rowSequence).asByteChunk();
        convertToBoolean(destination, byteChunk);
    }

    @Override
    public void fillPrevChunk(@NotNull final FillContext context, @NotNull final WritableChunk destination, @NotNull final RowSequence rowSequence) {
        final ToBooleanFillContext toBooleanFillContext = (ToBooleanFillContext) context;
        final ByteChunk byteChunk = alternateColumnSource.getPrevChunk(toBooleanFillContext.alternateGetContext, rowSequence).asByteChunk();
        convertToBoolean(destination, byteChunk);
    }

    private static void convertToBoolean(@NotNull final WritableChunk destination, @NotNull final ByteChunk byteChunk) {
        final WritableObjectChunk booleanObjectDestination = destination.asWritableObjectChunk();
        for (int ii = 0; ii < byteChunk.size(); ++ii) {
            booleanObjectDestination.set(ii, BooleanUtils.byteAsBoolean(byteChunk.get(ii)));
        }
        booleanObjectDestination.setSize(byteChunk.size());
    }

    @Override
    public boolean providesFillUnordered() {
        return FillUnordered.providesFillUnordered(alternateColumnSource);
    }

    @Override
    public void fillChunkUnordered(@NotNull FillContext context, @NotNull WritableChunk dest, @NotNull LongChunk keys) {
        final ToBooleanFillContext toBooleanFillContext = (ToBooleanFillContext) context;
        if (toBooleanFillContext.byteChunk == null) {
            throw new UnsupportedOperationException("Unordered fill is not supported by this column source!");
        }
        toBooleanFillContext.byteChunk.setSize(keys.size());
        // noinspection unchecked
        ((FillUnordered) alternateColumnSource).fillChunkUnordered(toBooleanFillContext.alternateFillContext, toBooleanFillContext.byteChunk, keys);
        convertToBoolean(dest, toBooleanFillContext.byteChunk);
    }

    @Override
    public void fillPrevChunkUnordered(@NotNull FillContext context, @NotNull WritableChunk dest, @NotNull LongChunk keys) {
        final ToBooleanFillContext toBooleanFillContext = (ToBooleanFillContext) context;
        if (toBooleanFillContext.byteChunk == null) {
            throw new UnsupportedOperationException("Unordered fill is not supported by this column source!");
        }
        toBooleanFillContext.byteChunk.setSize(keys.size());
        // noinspection unchecked
        ((FillUnordered) alternateColumnSource).fillPrevChunkUnordered(toBooleanFillContext.alternateFillContext, toBooleanFillContext.byteChunk, keys);
        convertToBoolean(dest, toBooleanFillContext.byteChunk);
    }

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