io.deephaven.chunk.util.LongChunkRangeIterator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-chunk Show documentation

Engine Chunks: Array-like data structures for dense, efficient data movement

The newest version!

//
// Copyright (c) 2016-2024 Deephaven Data Labs and Patent Pending
//
package io.deephaven.chunk.util;

import io.deephaven.chunk.LongChunk;
import io.deephaven.chunk.attributes.Any;
import io.deephaven.util.datastructures.LongRangeAbortableConsumer;
import io.deephaven.util.datastructures.LongRangeIterator;

public class LongChunkRangeIterator implements LongRangeIterator {

    private final LongChunk ck;
    private final int lastRangeStart;

    private int previousRangeStart;

    public LongChunkRangeIterator(final LongChunk ck) {
        this.ck = ck;
        lastRangeStart = ck.size() - 2;
        previousRangeStart = -2;
    }

    @Override
    public boolean hasNext() {
        return previousRangeStart < lastRangeStart;
    }

    @Override
    public void next() {
        previousRangeStart += 2;
    }

    @Override
    public long start() {
        return ck.get(previousRangeStart);
    }

    @Override
    public long end() {
        return ck.get(previousRangeStart + 1);
    }

    @Override
    public boolean forEachLongRange(final LongRangeAbortableConsumer lrc) {
        while (previousRangeStart < lastRangeStart) {
            previousRangeStart += 2;
            final long s = ck.get(previousRangeStart);
            final long e = ck.get(previousRangeStart + 1);
            if (!lrc.accept(s, e)) {
                return false;
            }
        }
        return true;
    }
}