Encoder.java

package cn.edu.ustc.aaron.encoder;

import java.util.*;
import cn.edu.ustc.aaron.common.*;
import Jama.*;
import nayuki.arithcode.*;
import java.io.IOException;

public class Encoder {
    

    public static void encode (String paraFile) throws IOException {
        // timing
        long startMili = System.currentTimeMillis();
        // read paras
        DomXmlDocument paraXml = new DomXmlDocument();
        HashMap<String, String> paraMap = paraXml.parseXml(paraFile);
        int width = Integer.parseInt(paraMap.get("Width"));
        int height = Integer.parseInt(paraMap.get("Height"));
        int totalFrames = Integer.parseInt(paraMap.get("TotalFrames"));
        int gopSize = Integer.parseInt(paraMap.get("GopSize"));
        boolean codeCbCr = Boolean.parseBoolean(paraMap.get("CodeCbCr"));
        boolean acEncoding = Boolean.parseBoolean(paraMap.get("ArithmeticEncoding"));
        boolean diagOutput = Boolean.parseBoolean(paraMap.get("DiagData"));

        // prepare to read org pics
        String rFilename = paraMap.get("OrgYuv");
        ArrayList<Picture> picList = new ArrayList<>();
        ReadYCbCr rYCbCr = ReadYCbCr.getInstance();

        // read org pics
        rYCbCr.startReading(rFilename, width, width >> 1);
        rYCbCr.readPic(width, height, 0, gopSize, picList);


        /* start the encoding process */

        // 1st, stack gopSize pictures into 'stackedPicMatBase' matrix with (width*height) rows and gopSize cols
        StackGops stackedGopBase = new StackGops(width * height, gopSize, codeCbCr);
        stackedGopBase.matrixLineByLine(picList, 0);  // use the line by line method so that writing yuv at the end will be easy

        // 2nd, svd this stackedPicMatBase matrix
        Svd svdResultBase = new Svd(stackedGopBase.getMatrix(), codeCbCr);

        // 3rd, reshape U and V matrix into 'reshapedProductUVBase' with (width*height*gopSize) rows and gopSize cols
        MatrixCreationAndOperation reshapedProductUVBase = new ReshapeProductUV(width * height * gopSize, gopSize, codeCbCr);
        reshapedProductUVBase.operateMatrix(svdResultBase.getMatU(), svdResultBase.getMatV());
        // { NOMORLIAZATION
        double normalization = 255.0; // use this coz U and V are both divided by 255 for Psvd
        reshapedProductUVBase.getMatrix(0).timesEquals(1.0/normalization/normalization);
        if (codeCbCr) {
            reshapedProductUVBase.getMatrix(1).timesEquals(1.0/normalization/normalization);
            reshapedProductUVBase.getMatrix(2).timesEquals(1.0/normalization/normalization);
        }
        // }

        // 4th, inverse reshapedProductUVBase for Psvd's use
        MatrixCreationAndOperation inverseProductUVBase = new Inverse(gopSize, gopSize, codeCbCr);
        // inverseProductUVBase.operateMatrix(reshapedProductUVBase.getMatrix());
        // Since inverseProductUVBase is an identity matrix, here we directly generate it.
        // { NOMORLIAZATION
        Matrix[] coeffMat = new Matrix[1]; // use this because U and V are both divided by 255 for Psvd
        coeffMat[0] = new Matrix(1, 1, normalization*normalization*normalization*normalization);
        // }
        inverseProductUVBase.operateMatrix(coeffMat);

        // test
        // int testColor = 2;
        // WriteYCbCr wYCbCr = WriteYCbCr.getInstance();
        // String wFilename = new String("stackedGopBase.txt");
        // wYCbCr.writeTxt(wFilename, stackedGopBase.getMatrix()[testColor]);
        // wFilename = new String("U.txt");
        // wYCbCr.writeTxt(wFilename, svdResultBase.getMatU()[testColor]);
        // wFilename = new String("V.txt");
        // wYCbCr.writeTxt(wFilename, svdResultBase.getMatV()[testColor]);
        // wFilename = new String("D.txt");
        // wYCbCr.writeTxt(wFilename, svdResultBase.getMatS()[testColor]);

        // 5th, preparation for pseudo svd
        // prepare to read the following gops' YCbCr
        StackGops stackedGop = new StackGops(width * height, gopSize, codeCbCr);
        // prepare to stack gop matrix to one-col vector with (width*height*gopSize) rows
        MatrixCreationAndOperation stackedToOneColVector = new StackToOneCol(width * height * gopSize, 1, codeCbCr);
        // prepare to psvd
        Psvd psvdOperation = new Psvd(stackedToOneColVector.getMatrix(), reshapedProductUVBase.getMatrix(), inverseProductUVBase.getMatrix(), codeCbCr, 1.0, 1.0e-7);

        // prepare to write Residue
        WriteYCbCr wYCbCr = WriteYCbCr.getInstance();
        String wResidueFilename = paraMap.get("ResYuv");
        int residueWidth = width * height;
        wYCbCr.startWriting(wResidueFilename, residueWidth, residueWidth >> 2);

        for (int gopNo = 1; gopNo < totalFrames / gopSize; gopNo++) {
            // read following YCbCr and stack Line By Line
            rYCbCr.readPic(width, height, gopNo * gopSize, gopSize, picList);
            stackedGop.matrixLineByLine(picList, gopNo * gopSize);
            // stack gop matrix to one-col vector
            stackedToOneColVector.operateMatrix(stackedGop.getMatrix());
            // { NOMORLIAZATION
            stackedToOneColVector.getMatrix(0).timesEquals(1.0/normalization);
            if (codeCbCr) {
                stackedToOneColVector.getMatrix(1).timesEquals(1.0/normalization);
                stackedToOneColVector.getMatrix(2).timesEquals(1.0/normalization);
            }
            // }

            // test
            // int testColor = 0;
            // WriteYCbCr wYCbCr = WriteYCbCr.getInstance();
            // wFilename = new String("inverseProductUVBase.txt");
            // wYCbCr.writeTxt(wFilename, inverseProductUVBase.getMatrix()[testColor]);
            // wFilename = new String("stackedGop.txt");
            // wYCbCr.writeTxt(wFilename, stackedGop.getMatrix()[testColor]);
            // wFilename = new String("stackedToOneColVector.txt");
            // wYCbCr.writeTxt(wFilename, stackedToOneColVector.getMatrix()[testColor]);
            // wFilename = new String("reshapedProductUVBase.txt");
            // wYCbCr.writeTxt(wFilename, reshapedProductUVBase.getMatrix()[testColor]);

            // psvd
            System.out.println("gopNo: "+gopNo);
            psvdOperation.psvdIteration();

            // write Residue
            wYCbCr.writeMatColByCol(psvdOperation.getResidue(), gopSize, width * height, codeCbCr);

            // test (getResidue() can be used only once!)
            // String wFilename = new String("ResidueRefined.txt");
            // wYCbCr.writeTxt(wFilename, psvdOperation.getResidue()[0]);
        }
        // end reading Org Pics
        rYCbCr.endReading();
        // end writing Residue
        wYCbCr.endWriting();

        // test
        // Iterator diagListIt = psvdOperation.getDiag().get(0).iterator();
        // while(diagListIt.hasNext()) {
        //     int elem = (int) diagListIt.next();
        //     System.out.println(elem);
        // }

        // 6th diag.mat Binarization
        if (acEncoding) {
            long psvdTime = System.currentTimeMillis();
            System.out.println("Psvd time: " + ((psvdTime - startMili)/1000) + "s");
            try {
                Binarization diagBinarization = new Binarization(psvdOperation.getDiag(), codeCbCr);
                diagBinarization.binarizeDiag();

                // // test
                // // int testColor = 0;
                // // WriteYCbCr wYCbCr = WriteYCbCr.getInstance();
                // // String wFilename = new String("Binarization.txt");
                // // wYCbCr.writeTxt(wFilename, diagBinarization.getbinArr(testColor), 1, diagBinarization.getbinArr(testColor).length);

                // 7th arithmetic encoding
                ArithmeticCompress arithmeticEncoder = new ArithmeticCompress(diagBinarization.getbinArr(), paraMap.get("DiagBinFile"), codeCbCr);

                // // arithmetic decoding
                // ArithmeticDecompress arithmeticDecoder = new ArithmeticDecompress("binDiag", codeCbCr);

                // // diag.mat invBinarization
                // Binarization invBinarization = new Binarization(arithmeticDecoder.getbinArr(), codeCbCr);
                // invBinarization.invBinarizeDiag();

                // test
                // diagListIt = invBinarization.getDiag().get(0).iterator();
                // while(diagListIt.hasNext()) {
                //     int elem = (int) diagListIt.next();
                //     System.out.println(elem);
                // }
                System.out.println("AC time: " + ((System.currentTimeMillis() - psvdTime)/1000) + "s");
            } catch (Exception e)
            {
                System.out.println(e.getMessage());
            }
        }
        if (diagOutput) {
            wYCbCr.writeData(paraMap.get("DiagDataFile"), psvdOperation.getDiag(), codeCbCr);
        }
        
        System.out.println("Total time: " + ((System.currentTimeMillis() - startMili)/1000) + "s");
    }
}