io.deephaven.engine.table.impl.asofjoin.RightIncrementalHashedAsOfJoinStateManager 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.asofjoin;

import io.deephaven.engine.rowset.*;
import io.deephaven.engine.table.ColumnSource;
import io.deephaven.engine.table.impl.RightIncrementalAsOfJoinStateManager;
import io.deephaven.engine.table.impl.sources.*;
import io.deephaven.engine.table.impl.ssa.SegmentedSortedArray;
import org.apache.commons.lang3.mutable.MutableObject;
import org.jetbrains.annotations.NotNull;

import java.util.function.Function;

public abstract class RightIncrementalHashedAsOfJoinStateManager extends RightIncrementalAsOfJoinStateManager {
    protected RightIncrementalHashedAsOfJoinStateManager(ColumnSource[] keySourcesForErrorMessages) {
        super(keySourcesForErrorMessages);
    }

    public static final byte ENTRY_RIGHT_MASK = 0x3;
    public static final byte ENTRY_RIGHT_IS_EMPTY = 0x0;
    public static final byte ENTRY_RIGHT_IS_BUILDER = 0x1;
    public static final byte ENTRY_RIGHT_IS_SSA = 0x2;
    public static final byte ENTRY_RIGHT_IS_INDEX = 0x3;

    public static final byte ENTRY_LEFT_MASK = 0x30;
    public static final byte ENTRY_LEFT_IS_EMPTY = 0x00;
    public static final byte ENTRY_LEFT_IS_BUILDER = 0x10;
    public static final byte ENTRY_LEFT_IS_SSA = 0x20;
    public static final byte ENTRY_LEFT_IS_INDEX = 0x30;

    protected void addToSequentialBuilder(long slot, @NotNull ObjectArraySource sequentialBuilders, long indexKey) {
        RowSetBuilderSequential builder = sequentialBuilders.getUnsafe(slot);
        if (builder == null) {
            builder = RowSetFactory.builderSequential();
            sequentialBuilders.set(slot, builder);
        }
        builder.appendKey(indexKey);
    }

    protected byte leftEntryAsRightType(byte entryType) {
        return (byte)((entryType & ENTRY_LEFT_MASK) >> 4);
    }

    protected byte getRightEntryType(byte entryType) {
        return (byte)(entryType & ENTRY_RIGHT_MASK);
    }

    public abstract int buildFromLeftSide(RowSequence leftRowSet, ColumnSource[] leftSources, @NotNull IntegerArraySource addedSlots);
    public abstract int buildFromRightSide(RowSequence rightRowSet, ColumnSource[] rightSources, @NotNull IntegerArraySource addedSlots, int usedSlots);

    public abstract void probeRightInitial(RowSequence rightIndex, ColumnSource[] rightSources);
    public abstract int probeAdditions(RowSet restampAdditions, ColumnSource[] sources, IntegerArraySource slots, ObjectArraySource sequentialBuilders);
    public abstract int buildAdditions(boolean isLeftSide, RowSet additions, ColumnSource[] sources, IntegerArraySource slots, ObjectArraySource sequentialBuilders);

    public abstract SegmentedSortedArray getRightSsa(int slot);
    public abstract SegmentedSortedArray getRightSsa(int slot, Function ssaFactory);
    public abstract SegmentedSortedArray getLeftSsa(int slot);
    public abstract SegmentedSortedArray getLeftSsa(int slot, Function ssaFactory);

    public abstract SegmentedSortedArray getLeftSsaOrIndex(int slot, MutableObject indexOutput);
    public abstract SegmentedSortedArray getRightSsaOrIndex(int slot, MutableObject indexOutput);
    public abstract void setRightIndex(int slot, RowSet rowSet);
    public abstract void setLeftIndex(int slot, RowSet rowSet);
    public abstract WritableRowSet getLeftIndex(int slot);
    public abstract WritableRowSet getRightIndex(int slot);

    public abstract WritableRowSet getAndClearLeftIndex(int slot);

    public abstract int markForRemoval(RowSet restampRemovals, ColumnSource[] sources, IntegerArraySource slots, ObjectArraySource sequentialBuilders);

    public abstract int gatherShiftIndex(RowSet restampAdditions, ColumnSource[] sources, IntegerArraySource slots, ObjectArraySource sequentialBuilders);

    public abstract int gatherModifications(RowSet restampAdditions, ColumnSource[] sources, IntegerArraySource slots, ObjectArraySource sequentialBuilders);

    public abstract byte getState(int slot);

    public abstract int getTableSize();
}