// 
// Decompiled by Procyon v0.6.0
// 

package org.bouncycastle.pqc.crypto.mldsa;

class PolyVecK
{
    private final Poly[] vec;
    
    PolyVecK(final MLDSAEngine mldsaEngine) {
        final int dilithiumK = mldsaEngine.getDilithiumK();
        this.vec = new Poly[dilithiumK];
        for (int i = 0; i < dilithiumK; ++i) {
            this.vec[i] = new Poly(mldsaEngine);
        }
    }
    
    Poly getVectorIndex(final int n) {
        return this.vec[n];
    }
    
    void setVectorIndex(final int n, final Poly poly) {
        this.vec[n] = poly;
    }
    
    public void uniformEta(final byte[] array, final short n) {
        short n2 = n;
        for (int i = 0; i < this.vec.length; ++i) {
            final Poly poly = this.vec[i];
            final short n3 = n2;
            ++n2;
            poly.uniformEta(array, n3);
        }
    }
    
    public void reduce() {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).reduce();
        }
    }
    
    public void invNttToMont() {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).invNttToMont();
        }
    }
    
    public void addPolyVecK(final PolyVecK polyVecK) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).addPoly(polyVecK.getVectorIndex(i));
        }
    }
    
    public void conditionalAddQ() {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).conditionalAddQ();
        }
    }
    
    public void power2Round(final PolyVecK polyVecK) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).power2Round(polyVecK.getVectorIndex(i));
        }
    }
    
    public void polyVecNtt() {
        for (int i = 0; i < this.vec.length; ++i) {
            this.vec[i].polyNtt();
        }
    }
    
    public void decompose(final PolyVecK polyVecK) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).decompose(polyVecK.getVectorIndex(i));
        }
    }
    
    public void packW1(final MLDSAEngine mldsaEngine, final byte[] array, final int n) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).packW1(array, n + i * mldsaEngine.getDilithiumPolyW1PackedBytes());
        }
    }
    
    public void pointwisePolyMontgomery(final Poly poly, final PolyVecK polyVecK) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).pointwiseMontgomery(poly, polyVecK.getVectorIndex(i));
        }
    }
    
    public void subtract(final PolyVecK polyVecK) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).subtract(polyVecK.getVectorIndex(i));
        }
    }
    
    public boolean checkNorm(final int n) {
        for (int i = 0; i < this.vec.length; ++i) {
            if (this.getVectorIndex(i).checkNorm(n)) {
                return true;
            }
        }
        return false;
    }
    
    public int makeHint(final PolyVecK polyVecK, final PolyVecK polyVecK2) {
        int n = 0;
        for (int i = 0; i < this.vec.length; ++i) {
            n += this.getVectorIndex(i).polyMakeHint(polyVecK.getVectorIndex(i), polyVecK2.getVectorIndex(i));
        }
        return n;
    }
    
    public void useHint(final PolyVecK polyVecK, final PolyVecK polyVecK2) {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).polyUseHint(polyVecK.getVectorIndex(i), polyVecK2.getVectorIndex(i));
        }
    }
    
    public void shiftLeft() {
        for (int i = 0; i < this.vec.length; ++i) {
            this.getVectorIndex(i).shiftLeft();
        }
    }
    
    @Override
    public String toString() {
        String str = "[";
        for (int i = 0; i < this.vec.length; ++i) {
            str = str + i + " " + this.getVectorIndex(i).toString();
            if (i != this.vec.length - 1) {
                str += ",\n";
            }
        }
        return str + "]";
    }
    
    public String toString(final String str) {
        return str + ": " + this.toString();
    }
}