src/Lucene.Net/Util/Packed/PackedReaderIterator.cs

using Lucene.Net.Diagnostics;
using Lucene.Net.Support;
using System;
using System.IO;
using System.Runtime.CompilerServices;

namespace Lucene.Net.Util.Packed
{
    /*
     * Licensed to the Apache Software Foundation (ASF) under one or more
     * contributor license agreements.  See the NOTICE file distributed with
     * this work for additional information regarding copyright ownership.
     * The ASF licenses this file to You under the Apache License, Version 2.0
     * (the "License"); you may not use this file except in compliance with
     * the License.  You may obtain a copy of the License at
     *
     *     http://www.apache.org/licenses/LICENSE-2.0
     *
     * Unless required by applicable law or agreed to in writing, software
     * distributed under the License is distributed on an "AS IS" BASIS,
     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     * See the License for the specific language governing permissions and
     * limitations under the License.
     */

    using DataInput = Lucene.Net.Store.DataInput;

    internal sealed class PackedReaderIterator : PackedInt32s.ReaderIterator
    {
        internal readonly int packedIntsVersion;
        internal readonly PackedInt32s.Format format;
        internal readonly BulkOperation bulkOperation;
        internal readonly byte[] nextBlocks;
        internal readonly Int64sRef nextValues;
        internal readonly int iterations;
        internal int position;

        internal PackedReaderIterator(PackedInt32s.Format format, int packedIntsVersion, int valueCount, int bitsPerValue, DataInput @in, int mem)
            : base(valueCount, bitsPerValue, @in)
        {
            this.format = format;
            this.packedIntsVersion = packedIntsVersion;
            bulkOperation = BulkOperation.Of(format, bitsPerValue);
            iterations = Iterations(mem);
            if (Debugging.AssertsEnabled) Debugging.Assert(valueCount == 0 || iterations > 0);
            nextBlocks = new byte[iterations * bulkOperation.ByteBlockCount];
            nextValues = new Int64sRef(new long[iterations * bulkOperation.ByteValueCount], 0, 0);
            nextValues.Offset = nextValues.Int64s.Length;
            position = -1;
        }

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        private int Iterations(int mem)
        {
            int iterations = bulkOperation.ComputeIterations(m_valueCount, mem);
            if (packedIntsVersion < PackedInt32s.VERSION_BYTE_ALIGNED)
            {
                // make sure iterations is a multiple of 8
                iterations = (iterations + 7) & unchecked((int)0xFFFFFFF8);
            }
            return iterations;
        }

        public override Int64sRef Next(int count)
        {
            if (Debugging.AssertsEnabled)
            {
                Debugging.Assert(nextValues.Length >= 0);
                Debugging.Assert(count > 0);
                Debugging.Assert(nextValues.Offset + nextValues.Length <= nextValues.Int64s.Length);
            }

            nextValues.Offset += nextValues.Length;

            int remaining = m_valueCount - position - 1;
            if (remaining <= 0)
            {
                throw new EndOfStreamException();
            }
            count = Math.Min(remaining, count);

            if (nextValues.Offset == nextValues.Int64s.Length)
            {
                long remainingBlocks = format.ByteCount(packedIntsVersion, remaining, m_bitsPerValue);
                int blocksToRead = (int)Math.Min(remainingBlocks, nextBlocks.Length);
                m_in.ReadBytes(nextBlocks, 0, blocksToRead);
                if (blocksToRead < nextBlocks.Length)
                {
                    Arrays.Fill(nextBlocks, blocksToRead, nextBlocks.Length, (byte)0);
                }

                bulkOperation.Decode(nextBlocks, 0, nextValues.Int64s, 0, iterations);
                nextValues.Offset = 0;
            }

            nextValues.Length = Math.Min(nextValues.Int64s.Length - nextValues.Offset, count);
            position += nextValues.Length;
            return nextValues;
        }

        public override int Ord => position;
    }
}