io.deephaven.engine.table.impl.util.freezeby.FreezeByCountOperator 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.util.freezeby;

import io.deephaven.chunk.attributes.ChunkLengths;
import io.deephaven.chunk.attributes.ChunkPositions;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.engine.rowset.chunkattributes.RowKeys;
import io.deephaven.util.QueryConstants;
import io.deephaven.engine.table.impl.by.IterativeChunkedAggregationOperator;
import io.deephaven.engine.table.impl.sources.ByteArraySource;
import io.deephaven.engine.table.ColumnSource;
import io.deephaven.chunk.*;

import java.util.Collections;
import java.util.Map;

public class FreezeByCountOperator implements IterativeChunkedAggregationOperator {
    private final ByteArraySource rowCount;
    private boolean inInitialization;

    public FreezeByCountOperator() {
        rowCount = new ByteArraySource();
        inInitialization = true;
    }

    @Override
    public void addChunk(BucketedContext context, Chunk values,
            LongChunk inputRowKeys, IntChunk destinations,
            IntChunk startPositions, IntChunk length,
            WritableBooleanChunk stateModified) {
        for (int ii = 0; ii < startPositions.size(); ++ii) {
            final int position = startPositions.get(ii);
            final int destination = destinations.get(position);
            if (length.get(ii) != 1) {
                throw new IllegalStateException("FreezeBy only allows one row per state!");
            }
            setFilled(destination);
        }
    }

    @Override
    public void removeChunk(BucketedContext context, Chunk values,
            LongChunk inputRowKeys, IntChunk destinations,
            IntChunk startPositions, IntChunk length,
            WritableBooleanChunk stateModified) {
        for (int ii = 0; ii < startPositions.size(); ++ii) {
            final int position = startPositions.get(ii);
            final int destination = destinations.get(position);
            setEmpty(destination);
        }
    }

    @Override
    public boolean addChunk(SingletonContext context, int chunkSize, Chunk values,
            LongChunk inputRowKeys, long destination) {
        if (chunkSize != 1) {
            throw new IllegalStateException("FreezeBy only allows one row per state!");
        }
        setFilled(destination);
        return false;
    }

    @Override
    public boolean removeChunk(SingletonContext context, int chunkSize, Chunk values,
            LongChunk inputRowKeys, long destination) {
        setEmpty(destination);
        return false;
    }

    @Override
    public void ensureCapacity(long tableSize) {
        rowCount.ensureCapacity(tableSize);
    }

    @Override
    public Map> getResultColumns() {
        return Collections.emptyMap();
    }

    @Override
    public void startTrackingPrevValues() {
        rowCount.startTrackingPrevValues();
        inInitialization = false;
    }

    private void setFilled(long destination) {
        final byte oldCount = rowCount.getAndSetUnsafe(destination, (byte) 1);
        if (oldCount != 0 && oldCount != QueryConstants.NULL_BYTE) {
            throw new IllegalStateException("FreezeBy only allows one row per state!");
        }
    }

    private void setEmpty(long destination) {
        final byte count = rowCount.getAndSetUnsafe(destination, (byte) 0);
        if (count != 1) {
            throw new IllegalStateException("FreezeBy only allows one row per state, old count: " + count);
        }
    }

    boolean wasDestinationEmpty(long destination) {
        if (inInitialization) {
            return true;
        }
        final byte prevByte = rowCount.getPrevByte(destination);
        return prevByte == 0 || prevByte == QueryConstants.NULL_BYTE;
    }
}