package net.minecraft.util;

import java.util.function.IntConsumer;
import javax.annotation.Nullable;
import net.minecraft.SystemUtils;
import org.apache.commons.lang3.Validate;

public class DataBits {

    private static final int[] a = new int[]{-1, -1, 0, Integer.MIN_VALUE, 0, 0, 1431655765, 1431655765, 0, Integer.MIN_VALUE, 0, 1, 858993459, 858993459, 0, 715827882, 715827882, 0, 613566756, 613566756, 0, Integer.MIN_VALUE, 0, 2, 477218588, 477218588, 0, 429496729, 429496729, 0, 390451572, 390451572, 0, 357913941, 357913941, 0, 330382099, 330382099, 0, 306783378, 306783378, 0, 286331153, 286331153, 0, Integer.MIN_VALUE, 0, 3, 252645135, 252645135, 0, 238609294, 238609294, 0, 226050910, 226050910, 0, 214748364, 214748364, 0, 204522252, 204522252, 0, 195225786, 195225786, 0, 186737708, 186737708, 0, 178956970, 178956970, 0, 171798691, 171798691, 0, 165191049, 165191049, 0, 159072862, 159072862, 0, 153391689, 153391689, 0, 148102320, 148102320, 0, 143165576, 143165576, 0, 138547332, 138547332, 0, Integer.MIN_VALUE, 0, 4, 130150524, 130150524, 0, 126322567, 126322567, 0, 122713351, 122713351, 0, 119304647, 119304647, 0, 116080197, 116080197, 0, 113025455, 113025455, 0, 110127366, 110127366, 0, 107374182, 107374182, 0, 104755299, 104755299, 0, 102261126, 102261126, 0, 99882960, 99882960, 0, 97612893, 97612893, 0, 95443717, 95443717, 0, 93368854, 93368854, 0, 91382282, 91382282, 0, 89478485, 89478485, 0, 87652393, 87652393, 0, 85899345, 85899345, 0, 84215045, 84215045, 0, 82595524, 82595524, 0, 81037118, 81037118, 0, 79536431, 79536431, 0, 78090314, 78090314, 0, 76695844, 76695844, 0, 75350303, 75350303, 0, 74051160, 74051160, 0, 72796055, 72796055, 0, 71582788, 71582788, 0, 70409299, 70409299, 0, 69273666, 69273666, 0, 68174084, 68174084, 0, Integer.MIN_VALUE, 0, 5};
    private final long[] b;
    private final int c;
    private final long d;
    private final int e;
    private final int f;
    private final int g;
    private final int h;
    private final int i;

    public DataBits(int i, int j) {
        this(i, j, (long[]) null);
    }

    public DataBits(int i, int j, @Nullable long[] along) {
        Validate.inclusiveBetween(1L, 32L, (long) i);
        this.e = j;
        this.c = i;
        this.d = (1L << i) - 1L;
        this.f = (char) (64 / i);
        int k = 3 * (this.f - 1);

        this.g = DataBits.a[k + 0];
        this.h = DataBits.a[k + 1];
        this.i = DataBits.a[k + 2];
        int l = (j + this.f - 1) / this.f;

        if (along != null) {
            if (along.length != l) {
                throw (RuntimeException) SystemUtils.c((Throwable) (new RuntimeException("Invalid length given for storage, got: " + along.length + " but expected: " + l)));
            }

            this.b = along;
        } else {
            this.b = new long[l];
        }

    }

    private int b(int i) {
        long j = Integer.toUnsignedLong(this.g);
        long k = Integer.toUnsignedLong(this.h);

        return (int) ((long) i * j + k >> 32 >> this.i);
    }

    public int a(int i, int j) {
        Validate.inclusiveBetween(0L, (long) (this.e - 1), (long) i);
        Validate.inclusiveBetween(0L, this.d, (long) j);
        int k = this.b(i);
        long l = this.b[k];
        int i1 = (i - k * this.f) * this.c;
        int j1 = (int) (l >> i1 & this.d);

        this.b[k] = l & ~(this.d << i1) | ((long) j & this.d) << i1;
        return j1;
    }

    public void b(int i, int j) {
        Validate.inclusiveBetween(0L, (long) (this.e - 1), (long) i);
        Validate.inclusiveBetween(0L, this.d, (long) j);
        int k = this.b(i);
        long l = this.b[k];
        int i1 = (i - k * this.f) * this.c;

        this.b[k] = l & ~(this.d << i1) | ((long) j & this.d) << i1;
    }

    public int a(int i) {
        Validate.inclusiveBetween(0L, (long) (this.e - 1), (long) i);
        int j = this.b(i);
        long k = this.b[j];
        int l = (i - j * this.f) * this.c;

        return (int) (k >> l & this.d);
    }

    public long[] a() {
        return this.b;
    }

    public int b() {
        return this.e;
    }

    public void a(IntConsumer intconsumer) {
        int i = 0;
        long[] along = this.b;
        int j = along.length;

        for (int k = 0; k < j; ++k) {
            long l = along[k];

            for (int i1 = 0; i1 < this.f; ++i1) {
                intconsumer.accept((int) (l & this.d));
                l >>= this.c;
                ++i;
                if (i >= this.e) {
                    return;
                }
            }
        }

    }
}