/*

 * Copyright (c) 2001, 2014, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.imageio.plugins.jpeg;

import javax.imageio.metadata.IIOInvalidTreeException;

import javax.imageio.metadata.IIOMetadataNode;

import javax.imageio.stream.ImageOutputStream;

import javax.imageio.plugins.jpeg.JPEGHuffmanTable;

import java.io.IOException;

import java.util.List;

import java.util.ArrayList;

import java.util.Iterator;

import org.w3c.dom.Node;

import org.w3c.dom.NodeList;

import org.w3c.dom.NamedNodeMap;

/**

 * A DHT (Define Huffman Table) marker segment.

 */

class DHTMarkerSegment extends MarkerSegment {

    DHTMarkerSegment(boolean needFour) {

        super(JPEG.DHT);

        tables.add(new Htable(JPEGHuffmanTable.StdDCLuminance, true, 0));

        if (needFour) {

            tables.add(new Htable(JPEGHuffmanTable.StdDCChrominance, true, 1));

        }

        tables.add(new Htable(JPEGHuffmanTable.StdACLuminance, false, 0));

        if (needFour) {

            tables.add(new Htable(JPEGHuffmanTable.StdACChrominance, false, 1));

        }

    }

    DHTMarkerSegment(JPEGBuffer buffer) throws IOException {

        super(buffer);

        int count = length;

        while (count > 0) {

            Htable newGuy = new Htable(buffer);

            tables.add(newGuy);

            count -= 1 + 16 + newGuy.values.length;

        }

        buffer.bufAvail -= length;

    }

    DHTMarkerSegment(JPEGHuffmanTable[] dcTables,

                     JPEGHuffmanTable[] acTables) {

        super(JPEG.DHT);

        for (int i = 0; i < dcTables.length; i++) {

            tables.add(new Htable(dcTables[i], true, i));

        }

        for (int i = 0; i < acTables.length; i++) {

            tables.add(new Htable(acTables[i], false, i));

        }

    }

    DHTMarkerSegment(Node node) throws IIOInvalidTreeException {

        super(JPEG.DHT);

        NodeList children = node.getChildNodes();

        int size = children.getLength();

        if ((size < 1) || (size > 4)) {

            throw new IIOInvalidTreeException("Invalid DHT node", node);

        }

        for (int i = 0; i < size; i++) {

            tables.add(new Htable(children.item(i)));

        }

    }

    protected Object clone() {

        DHTMarkerSegment newGuy = (DHTMarkerSegment) super.clone();

        while (iter.hasNext()) {

        }

        return newGuy;

    }

    IIOMetadataNode getNativeNode() {

        IIOMetadataNode node = new IIOMetadataNode("dht");

        for (int i= 0; i<tables.size(); i++) {

            node.appendChild(table.getNativeNode());

        }

        return node;

    }

    /**

     * Writes the data for this segment to the stream in

     * valid JPEG format.

     */

    void write(ImageOutputStream ios) throws IOException {

        // We don't write DHT segments; the IJG library does.

    }

    void print() {

        printTag("DHT");

        System.out.println("Num tables: "

                           + Integer.toString(tables.size()));

        for (int i= 0; i<tables.size(); i++) {

            table.print();

        }

        System.out.println();

    }

    Htable getHtableFromNode(Node node) throws IIOInvalidTreeException {

        return new Htable(node);

    }

    void addHtable(JPEGHuffmanTable table, boolean isDC, int id) {

        tables.add(new Htable(table, isDC, id));

    }

    /**

     * A Huffman table within a DHT marker segment.

     */

    class Htable implements Cloneable {

        int tableClass;  // 0 == DC, 1 == AC

        int tableID; // 0 - 4

        private static final int NUM_LENGTHS = 16;

        // # of codes of each length

        short [] numCodes = new short[NUM_LENGTHS];

        short [] values;

        Htable(JPEGBuffer buffer) {

            tableClass = buffer.buf[buffer.bufPtr] >>> 4;

            tableID = buffer.buf[buffer.bufPtr++] & 0xf;

            for (int i = 0; i < NUM_LENGTHS; i++) {

                numCodes[i] = (short) (buffer.buf[buffer.bufPtr++] & 0xff);

            }

            int numValues = 0;

            for (int i = 0; i < NUM_LENGTHS; i++) {

                numValues += numCodes[i];

            }

            values = new short[numValues];

            for (int i = 0; i < numValues; i++) {

                values[i] = (short) (buffer.buf[buffer.bufPtr++] & 0xff);

            }

        }

        Htable(JPEGHuffmanTable table, boolean isDC, int id) {

            tableClass = isDC ? 0 : 1;

            tableID = id;

            numCodes = table.getLengths();

            values = table.getValues();

        }

        Htable(Node node) throws IIOInvalidTreeException {

            if (node.getNodeName().equals("dhtable")) {

                NamedNodeMap attrs = node.getAttributes();

                int count = attrs.getLength();

                if (count != 2) {

                    throw new IIOInvalidTreeException

                        ("dhtable node must have 2 attributes", node);

                }

                tableClass = getAttributeValue(node, attrs, "class", 0, 1, true);

                tableID = getAttributeValue(node, attrs, "htableId", 0, 3, true);

                if (node instanceof IIOMetadataNode) {

                    IIOMetadataNode ourNode = (IIOMetadataNode) node;

                    JPEGHuffmanTable table =

                        (JPEGHuffmanTable) ourNode.getUserObject();

                    if (table == null) {

                        throw new IIOInvalidTreeException

                            ("dhtable node must have user object", node);

                    }

                    numCodes = table.getLengths();

                    values = table.getValues();

                } else {

                    throw new IIOInvalidTreeException

                        ("dhtable node must have user object", node);

                }

            } else {

                throw new IIOInvalidTreeException

                    ("Invalid node, expected dqtable", node);

            }

        }

        protected Object clone() {

            Htable newGuy = null;

            try {

                newGuy = (Htable) super.clone();

            } catch (CloneNotSupportedException e) {} // won't happen

            if (numCodes != null) {

                newGuy.numCodes = numCodes.clone();

            }

            if (values != null) {

                newGuy.values = values.clone();

            }

            return newGuy;

        }

        IIOMetadataNode getNativeNode() {

            IIOMetadataNode node = new IIOMetadataNode("dhtable");

            node.setAttribute("class", Integer.toString(tableClass));

            node.setAttribute("htableId", Integer.toString(tableID));

            node.setUserObject(new JPEGHuffmanTable(numCodes, values));

            return node;

        }

        void print() {

            System.out.println("Huffman Table");

            System.out.println("table class: "

                               + ((tableClass == 0) ? "DC":"AC"));

            System.out.println("table id: " + Integer.toString(tableID));

            (new JPEGHuffmanTable(numCodes, values)).toString();

            /*

              System.out.print("Lengths:");

              for (int i=0; i<16; i++) {

              System.out.print(" " + Integer.toString(numCodes[i]));

              }

              int count = 0;

              if (values.length > 16) {

              System.out.println("\nFirst 16 Values:");

              count = 16;

              } else {

              System.out.println("\nValues:");

              count = values.length;

              }

              for (int i=0; i<count; i++) {

              System.out.println(Integer.toString(values[i]&0xff));

              }

            */

        }

    }

}