io.deephaven.engine.table.impl.sources.ring.CharacterRingChunkSource 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.ring;

import io.deephaven.chunk.ChunkType;
import io.deephaven.chunk.WritableCharChunk;
import io.deephaven.chunk.WritableChunk;
import io.deephaven.chunk.attributes.Values;
import io.deephaven.engine.rowset.RowSet;
import io.deephaven.util.type.TypeUtils;
import org.jetbrains.annotations.NotNull;

import java.util.Objects;

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

final class CharacterRingChunkSource extends AbstractRingChunkSource {
    public static RingColumnSource columnSource(int n) {
        return new RingColumnSource<>(char.class, new CharacterRingChunkSource(n), new CharacterRingChunkSource(n));
    }

    public CharacterRingChunkSource(int capacity) {
        super(char.class, capacity);
    }

    @Override
    public ChunkType getChunkType() {
        return ChunkType.Char;
    }

    @Override
    Character get(long key) {
        return TypeUtils.box(getChar(key));
    }

    @Override
    char getChar(long key) {
        if (key == RowSet.NULL_ROW_KEY) {
            return NULL_CHAR;
        }
        if (STRICT_KEYS && !containsKey(key)) {
            throw new IllegalArgumentException(String.format("Invalid key %d. available=[%d, %d]", key, firstKey(), lastKey()));
        }
        return ring[keyToRingIndex(key)];
    }

    @Override
    Filler filler(@NotNull WritableChunk destination) {
        return new FillerImpl(destination.asWritableCharChunk());
    }

    private class FillerImpl extends Filler {
        private final WritableCharChunk dest;

        FillerImpl(WritableCharChunk dest) {
            this.dest = Objects.requireNonNull(dest);
        }

        @Override
        protected void copyFromRing(int srcRingIx, int destOffset) {
            dest.set(destOffset, ring[srcRingIx]);
        }

        @Override
        protected void copyFromRing(int srcRingIx, int destOffset, int size) {
            dest.copyFromTypedArray(ring, srcRingIx, destOffset, size);
        }

        @Override
        protected void setSize(int size) {
            dest.setSize(size);
        }
    }
}