package gov.nasa.gsfc.spdf.cdfj;
import java.nio.*;
import java.lang.reflect.*;

/**
 *
 * @author nand
 */
public final class StringVarContainer extends ByteVarContainer implements 
    VDataContainer.CString {

    /**
     *
     * @param cdfi
     * @param vrbl
     * @param ints
     * @throws IllegalAccessException
     * @throws InvocationTargetException
     * @throws Throwable
     */
    public StringVarContainer(CDFImpl thisCDF, Variable var, int[] pt) throws
        IllegalAccessException, InvocationTargetException, Throwable {
        super(thisCDF, var, pt);
    }

    /**
     *
     * @param type
     * @param preserve
     * @return
     */
    public static boolean isCompatible(int type, boolean preserve) {
        if  (isCompatible(type, preserve, Byte.TYPE)) {
            boolean stringType = DataTypes.isStringType(type);
            return stringType;
        }
        return false;
    }

    /**
     *
     * @return
     * @throws Throwable
     */
    @Override
    public Object _asArray() throws Throwable {
        int rank = var.getEffectiveRank();
        if (rank > 2) {
            throw new Throwable("Rank > 2 not supported for strings.");
        }
        ByteBuffer buf = getBuffer();
        if (buf == null) return null;
        int words = buf.remaining();
        int records = -1;
        int len = var.getNumberOfElements();
        byte[] ba = new byte[len];
        switch (rank) {
        case 0:
            records = words/len;
            String[] sa = new String[records];
            for (int r = 0; r < records; r++) {
                buf.get(ba);
                sa[r] = new String(ba);
            }
            return sa;
        case 1:
            int n0 =
                    (((Integer)var.getElementCount().elementAt(0)));
            records = words/(n0*len);
            String[][] sa1 = new String[records][n0];
            for (int r = 0; r < records; r++) {
                for (int e = 0; e < n0; e++) {
                    buf.get(ba);
                    sa1[r][e] = new String(ba);
                }
            }
            return sa1;
        case 2:
            n0 = (Integer)var.getElementCount().elementAt(0);
            int n1= (Integer)var.getElementCount().elementAt(1);
            records = words/(n0*n1*len);
            String[][][] sa11 = new String[records][n0][n1];
            for (int r = 0; r < records; r++) {
                for (int e1 = 0; e1 < n0; e1++) {
                    for (int e2 = 0; e2 < n1; e2++) {
                        buf.get(ba);
                        sa11[r][e1][e2] = new String(ba);
                    }
                }
            }
            return sa11;            
        default:
            throw new Throwable("Internal error");
        }
    }

    @Override
    public byte[] as1DArray() {return (byte[])super.as1DArray();}

    /**
     *
     * @return
     */
    @Override
    public byte[] asOneDArray() {return (byte[])super.asOneDArray();}

    /**
     *
     * @param cmtarget
     * @return
     */
    @Override
    public byte[] asOneDArray(boolean cmtarget) {
        return (byte[])super.asOneDArray(cmtarget);
    }

    /**
     *
     * @return
     * @throws Throwable
     */
    @Override
    public AArray asArray() throws Throwable {
        return new StringArray(_asArray());
    }
}