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

package org.bouncycastle.asn1.nist;

import org.bouncycastle.asn1.DERSequence;
import org.bouncycastle.asn1.DEROctetString;
import org.bouncycastle.asn1.ASN1Encodable;
import org.bouncycastle.asn1.ASN1EncodableVector;
import org.bouncycastle.asn1.ASN1Primitive;
import org.bouncycastle.asn1.ASN1OctetString;
import org.bouncycastle.asn1.ASN1Integer;
import org.bouncycastle.asn1.ASN1Sequence;
import org.bouncycastle.util.Arrays;
import org.bouncycastle.asn1.ASN1Object;

public class KMACwithSHAKE256_params extends ASN1Object
{
    private static final byte[] EMPTY_STRING;
    private static final int DEF_LENGTH = 512;
    private final int outputLength;
    private final byte[] customizationString;
    
    public KMACwithSHAKE256_params(final int outputLength) {
        this.outputLength = outputLength;
        this.customizationString = KMACwithSHAKE256_params.EMPTY_STRING;
    }
    
    public KMACwithSHAKE256_params(final int outputLength, final byte[] array) {
        this.outputLength = outputLength;
        this.customizationString = Arrays.clone(array);
    }
    
    public static KMACwithSHAKE256_params getInstance(final Object o) {
        if (o instanceof KMACwithSHAKE256_params) {
            return (KMACwithSHAKE256_params)o;
        }
        if (o != null) {
            return new KMACwithSHAKE256_params(ASN1Sequence.getInstance(o));
        }
        return null;
    }
    
    private KMACwithSHAKE256_params(final ASN1Sequence asn1Sequence) {
        if (asn1Sequence.size() > 2) {
            throw new IllegalArgumentException("sequence size greater than 2");
        }
        if (asn1Sequence.size() == 2) {
            this.outputLength = ASN1Integer.getInstance(asn1Sequence.getObjectAt(0)).intValueExact();
            this.customizationString = Arrays.clone(ASN1OctetString.getInstance(asn1Sequence.getObjectAt(1)).getOctets());
        }
        else if (asn1Sequence.size() == 1) {
            if (asn1Sequence.getObjectAt(0) instanceof ASN1Integer) {
                this.outputLength = ASN1Integer.getInstance(asn1Sequence.getObjectAt(0)).intValueExact();
                this.customizationString = KMACwithSHAKE256_params.EMPTY_STRING;
            }
            else {
                this.outputLength = 512;
                this.customizationString = Arrays.clone(ASN1OctetString.getInstance(asn1Sequence.getObjectAt(0)).getOctets());
            }
        }
        else {
            this.outputLength = 512;
            this.customizationString = KMACwithSHAKE256_params.EMPTY_STRING;
        }
    }
    
    public int getOutputLength() {
        return this.outputLength;
    }
    
    public byte[] getCustomizationString() {
        return Arrays.clone(this.customizationString);
    }
    
    @Override
    public ASN1Primitive toASN1Primitive() {
        final ASN1EncodableVector asn1EncodableVector = new ASN1EncodableVector();
        if (this.outputLength != 512) {
            asn1EncodableVector.add(new ASN1Integer(this.outputLength));
        }
        if (this.customizationString.length != 0) {
            asn1EncodableVector.add(new DEROctetString(this.getCustomizationString()));
        }
        return new DERSequence(asn1EncodableVector);
    }
    
    static {
        EMPTY_STRING = new byte[0];
    }
}