/*
 * Copyright (c) 2005, 2017, 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.
 *
 * 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.
 */

/*
 * @test
 * @bug 6275211 6276621
 * @summary Tests that GIF writer plugin is able to write indexed images if
 *          palette size is not a power of two
 */

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.awt.image.IndexColorModel;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

import javax.imageio.ImageIO;
import javax.imageio.ImageWriter;
import javax.imageio.stream.ImageOutputStream;

public class OddPaletteTest {

    private static int w = 100;
    private static int h = 100;

    public static void main(String[] args) {
        BufferedImage[] srcs = new BufferedImage[2];
        srcs[0] = createTestImage(7); // bug 6275211
        srcs[1] = createTestImage(1); // bug 6276621

        for (int i = 0; i < srcs.length; i++) {
            doTest(srcs[i]);
        }
    }

    private static void doTest(BufferedImage src) {
        ImageWriter w = ImageIO.getImageWritersByFormatName("GIF").next();
        if (w == null) {
            throw new RuntimeException("No writer available!");
        }

        try {
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageOutputStream ios = ImageIO.createImageOutputStream(baos);
            w.setOutput(ios);
            w.write(src);
        } catch (IOException e) {
            throw new RuntimeException("Test failed.", e);
        } catch (IllegalArgumentException e) {
            throw new RuntimeException("Test failed.", e);
        }
    }

    private static BufferedImage createTestImage(int paletteSize) {
        byte[] r = new byte[paletteSize];
        byte[] g = new byte[paletteSize];
        byte[] b = new byte[paletteSize];

        int shift = 256 / paletteSize;
        for (int i = 0; i < paletteSize; i++) {
            r[i] = g[i] = b[i] = (byte)(shift * i);
        }

        int numBits = getNumBits(paletteSize);

        System.out.println("num of bits " + numBits);

        IndexColorModel icm =
            new IndexColorModel(numBits, paletteSize,  r, g, b);

        BufferedImage img = new BufferedImage(w, h,
                                              BufferedImage.TYPE_BYTE_INDEXED,
                                              icm);
        Graphics2D  g2d = img.createGraphics();
        g2d.setColor(Color.white);
        g2d.fillRect(0, 0, w, h);
        g2d.setColor(Color.black);
        g2d.drawLine(0, 0, w, h);
        g2d.drawLine(0, h, w, 0);

        return img;
    }

    private static int getNumBits(int paletteSize) {
        if (paletteSize < 0) {
            throw new IllegalArgumentException("negative palette size: " +
                                               paletteSize);
        }
        if (paletteSize < 2) {
            return 1;
        }
        int numBits = 0;

        paletteSize--;

        while (paletteSize > 0) {
            numBits++;
            paletteSize = paletteSize >> 1;
        }
        return numBits;
    }
}