/*

 * Copyright (c) 2018, 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 8194489

 * @summary The test checks whether BandedSampleModel computes appropriate

 *          size for allocating memory in DataBuffer.

 * @run main BandedSampleModelSizeTest

 */

import java.awt.image.BandedSampleModel;

import java.awt.image.DataBuffer;

import java.util.Arrays;

public class BandedSampleModelSizeTest {

    // Constants

    private static final int TEST_NUM_BANDS = 3;

    private static final int TEST_SRC_IMG_DIM = 10;

    // Required sample models

    private static BandedSampleModel singleBankModel = null;

    private static BandedSampleModel multiBankModel = null;

    private static void initTest() {

        int[] bandOffsets = new int[TEST_NUM_BANDS];

        int[] bankIndices = new int[TEST_NUM_BANDS];

        /*

         * Create a BandedSampleModel to store samples of all bands in one

         * bank of DataBuffer.

         */

        bandOffsets[0] = 0;

        bandOffsets[1] = 120;

        bandOffsets[2] = 240;

        bankIndices[0] = 0;

        bankIndices[1] = 0;

        bankIndices[2] = 0;

        singleBankModel = new BandedSampleModel(DataBuffer.TYPE_BYTE,

                TEST_SRC_IMG_DIM, TEST_SRC_IMG_DIM, TEST_SRC_IMG_DIM,

                bankIndices, bandOffsets);

        /*

         * Create a BandedSampleModel to store samples of all bands in

         * different banks of DataBuffer.

         */

        bandOffsets[0] = 0;

        bandOffsets[1] = 20;

        bandOffsets[2] = 40;

        bankIndices[0] = 0;

        bankIndices[1] = 1;

        bankIndices[2] = 2;

        multiBankModel = new BandedSampleModel(DataBuffer.TYPE_BYTE,

                TEST_SRC_IMG_DIM, TEST_SRC_IMG_DIM, TEST_SRC_IMG_DIM,

                bankIndices, bandOffsets);

    }

    private static void testSingleBankModel() {

        int[] srcSamples = new int[TEST_NUM_BANDS];

        int[] resSamples = new int[TEST_NUM_BANDS];

        // Create image buffer for the requried sample model

        DataBuffer imgBuffer = singleBankModel.createDataBuffer();

        // Test the sample model by setting a pixel value & inspecting the same.

        Arrays.fill(srcSamples, 125);

        singleBankModel.setPixel(0, 0, srcSamples, imgBuffer);

        singleBankModel.getPixel(0, 0, resSamples, imgBuffer);

        if (!Arrays.equals(srcSamples, resSamples)) {

            throw new RuntimeException("Test Failed. Incorrect samples found"

                    + " in the image");

        }

        Arrays.fill(srcSamples, 250);

        singleBankModel.setPixel(9, 9, srcSamples, imgBuffer);

        singleBankModel.getPixel(9, 9, resSamples, imgBuffer);

        if (!Arrays.equals(srcSamples, resSamples)) {

            throw new RuntimeException("Test Failed. Incorrect samples found"

                    + " in the image");

        }

    }

    private static void testMultiBankModel() {

        int[] srcSamples = new int[TEST_NUM_BANDS];

        int[] resSamples = new int[TEST_NUM_BANDS];

        // Create image buffer for the required sample model

        DataBuffer imgBuffer = multiBankModel.createDataBuffer();

        // Test the sample model by setting a pixel value & inspecting the same.

        Arrays.fill(srcSamples, 125);

        multiBankModel.setPixel(0, 0, srcSamples, imgBuffer);

        multiBankModel.getPixel(0, 0, resSamples, imgBuffer);

        if (!Arrays.equals(srcSamples, resSamples)) {

            throw new RuntimeException("Test Failed. Incorrect samples found"

                    + " in the image");

        }

        Arrays.fill(srcSamples, 250);

        multiBankModel.setPixel(9, 9, srcSamples, imgBuffer);

        multiBankModel.getPixel(9, 9, resSamples, imgBuffer);

        if (!Arrays.equals(srcSamples, resSamples)) {

            throw new RuntimeException("Test Failed. Incorrect samples found"

                    + " in the image");

        }

    }

    public static void main(String args[]) {

        // Initialize the test

        initTest();

        // Test banded sample model with single bank of data buffer

        testSingleBankModel();

        // Test banded sample model with multiple banks of data buffer

        testMultiBankModel();

    }

}