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

/* ****************************************************************
 ******************************************************************
 ******************************************************************
 *** COPYRIGHT (c) Eastman Kodak Company, 1997
 *** As  an unpublished  work pursuant to Title 17 of the United
 *** States Code.  All rights reserved.
 ******************************************************************
 ******************************************************************
 ******************************************************************/

package java.awt.image;

import java.util.Objects;

import static sun.java2d.StateTrackable.State.STABLE;
import static sun.java2d.StateTrackable.State.UNTRACKABLE;

/**
 * This class extends {@code DataBuffer} and stores data internally as
 * shorts.  Values stored in the short array(s) of this {@code DataBuffer}
 * are treated as unsigned values.
 * <p>
 * <a id="optimizations">
 * Note that some implementations may function more efficiently
 * if they can maintain control over how the data for an image is
 * stored.
 * For example, optimizations such as caching an image in video
 * memory require that the implementation track all modifications
 * to that data.
 * Other implementations may operate better if they can store the
 * data in locations other than a Java array.
 * To maintain optimum compatibility with various optimizations
 * it is best to avoid constructors and methods which expose the
 * underlying storage as a Java array as noted below in the
 * documentation for those methods.
 * </a>
 */
public final class DataBufferUShort extends DataBuffer
{
    /** The default data bank. */
    short[] data;

    /** All data banks */
    short[][] bankdata;

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with a single bank and the
     * specified size.
     *
     * @param size The size of the {@code DataBuffer}.
     * @throws IllegalArgumentException if {@code size} is less than or equal to zero.
     */
    public DataBufferUShort(int size) {
        super(STABLE, TYPE_USHORT, size);
        data = new short[size];
        bankdata = new short[1][];
        bankdata[0] = data;
    }

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with the specified number of
     * banks, all of which are the specified size.
     *
     * @param size The size of the banks in the {@code DataBuffer}.
     * @param numBanks The number of banks in the {@code DataBuffer}.
     * @throws IllegalArgumentException if {@code size} or {@code numBanks}
     *         is less than or equal to zero.
     */
    public DataBufferUShort(int size, int numBanks) {
        super(STABLE, TYPE_USHORT, size, numBanks);
        bankdata = new short[numBanks][];
        for (int i= 0; i < numBanks; i++) {
            bankdata[i] = new short[size];
        }
        data = bankdata[0];
    }

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with a single bank
     * using the specified array.
     * Only the first {@code size} elements should be used by accessors of
     * this {@code DataBuffer}.  {@code dataArray} must be large enough to
     * hold {@code size} elements.
     * <p>
     * Note that {@code DataBuffer} objects created by this constructor
     * may be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @param dataArray The unsigned-short array for the {@code DataBuffer}.
     * @param size The size of the {@code DataBuffer} bank.
     * @throws NullPointerException if {@code dataArray} is {@code null}.
     * @throws IllegalArgumentException if {@code size} is less than or equal
     *         to zero, or is greater than the length of {@code dataArray}.
     */
    public DataBufferUShort(short[] dataArray, int size) {
        super(UNTRACKABLE, TYPE_USHORT, size);
        Objects.requireNonNull(dataArray, "dataArray must not be null");
        checkArraySize(size, dataArray.length);
        data = dataArray;
        bankdata = new short[1][];
        bankdata[0] = data;
    }

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with a single bank
     * using the specified array, size, and offset.  {@code dataArray} must have at
     * least {@code offset} + {@code size} elements.  Only elements
     * {@code offset} through {@code offset} + {@code size} - 1 should
     * be used by accessors of this {@code DataBuffer}.
     * <p>
     * Note that {@code DataBuffer} objects created by this constructor
     * may be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @param dataArray The unsigned-short array for the {@code DataBuffer}.
     * @param size The size of the {@code DataBuffer} bank.
     * @param offset The offset into the {@code dataArray}.
     * @throws NullPointerException if {@code dataArray} is {@code null}.
     * @throws IllegalArgumentException
     *         if {@code size} is less than or equal to zero,
     *         or {@code offset} is less than zero,
     *         or {@code (offset + size)} is greater than the length of {@code dataArray}.
     */
    public DataBufferUShort(short[] dataArray, int size, int offset) {
        super(UNTRACKABLE, TYPE_USHORT, size, 1, offset);
        Objects.requireNonNull(dataArray, "dataArray must not be null");
        checkArraySize(size, offset, dataArray.length);
        data = dataArray;
        bankdata = new short[1][];
        bankdata[0] = data;
    }

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with the specified arrays.
     * The number of banks will be equal to {@code dataArray.length}.
     * Only the first {@code size} elements of each array should be used by
     * accessors of this {@code DataBuffer}.
     * <p>
     * Note that {@code DataBuffer} objects created by this constructor
     * may be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @param dataArray The unsigned-short arrays for the {@code DataBuffer}.
     * @param size The size of the banks in the {@code DataBuffer}.
     * @throws NullPointerException if {@code dataArray} is {@code null}.
     * @throws IllegalArgumentException if {@code size} is less than or equal to zero.
     * @throws IllegalArgumentException if {@code dataArray} does not have at least one bank.
     * @throws NullPointerException if any bank of {@code dataArray} is {@code null}.
     * @throws IllegalArgumentException if the length of any bank of {@code dataArray}
     *         is less than {@code size}.
     */
    public DataBufferUShort(short[][] dataArray, int size) {
        super(UNTRACKABLE, TYPE_USHORT, size, dataArray.length);
        for (int b = 0; b < dataArray.length; b++) {
            Objects.requireNonNull(dataArray[b], "bank must not be null");
            checkBankSize(b, size, 0, dataArray[b].length);
        }
        bankdata = dataArray.clone();
        data = bankdata[0];
    }

    /**
     * Constructs an unsigned-short based {@code DataBuffer} with specified arrays,
     * size, and offsets.
     * The number of banks is equal to {@code dataArray.length}.  Each array must
     * be at least as large as {@code size} + the corresponding offset.   There must
     * be an entry in the offset array for each {@code dataArray} entry.  For each
     * bank, only elements {@code offset} through
     * {@code offset} + {@code size} - 1 should be
     * used by accessors of this {@code DataBuffer}.
     * <p>
     * Note that {@code DataBuffer} objects created by this constructor
     * may be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @param dataArray The unsigned-short arrays for the {@code DataBuffer}.
     * @param size The size of the banks in the {@code DataBuffer}.
     * @param offsets The offsets into each array.
     * @throws IllegalArgumentException if {@code size} is less than or equal to zero.
     * @throws NullPointerException if {@code dataArray} is {@code null}.
     * @throws NullPointerException if any bank of {@code dataArray} is {@code null}.
     * @throws IllegalArgumentException if {@code dataArray} does not have at least one bank.
     * @throws NullPointerException if {@code offsets} is {@code null}.
     * @throws IllegalArgumentException if any element of {@code offsets} is less than zero.
     * @throws ArrayIndexOutOfBoundsException if the lengths of {@code dataArray} and {@code offsets} differ.
     * @throws IllegalArgumentException if the length of any bank of {@code dataArray}
     *         is less than {@code (size + offsets[bankIndex])}.
     */
    public DataBufferUShort(short[][] dataArray, int size, int[] offsets) {
        super(UNTRACKABLE, TYPE_USHORT, size, dataArray.length, offsets);
        for (int b = 0; b < dataArray.length; b++) {
            Objects.requireNonNull(dataArray[b], "bank must not be null");
            checkBankSize(b, size, offsets[b], dataArray[b].length);
        }
        bankdata = dataArray.clone();
        data = bankdata[0];
    }

    /**
     * Returns the default (first) unsigned-short data array.
     * <p>
     * Note that calling this method may cause this {@code DataBuffer}
     * object to be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @return The first unsigned-short data array.
     */
    public short[] getData() {
        theTrackable.setUntrackable();
        return data;
    }

    /**
     * Returns the data array for the specified bank.
     * <p>
     * Note that calling this method may cause this {@code DataBuffer}
     * object to be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @param bank The bank whose data array you want to get.
     * @return The data array for the specified bank.
     * @throws ArrayIndexOutOfBoundsException if {@code bank} is not a valid bank index.
     */
    public short[] getData(int bank) {
        checkBank(bank);
        theTrackable.setUntrackable();
        return bankdata[bank];
    }

    /**
     * Returns the data arrays for all banks.
     * <p>
     * Note that calling this method may cause this {@code DataBuffer}
     * object to be incompatible with <a href="#optimizations">performance
     * optimizations</a> used by some implementations (such as caching
     * an associated image in video memory).
     *
     * @return All of the data arrays.
     */
    public short[][] getBankData() {
        theTrackable.setUntrackable();
        return bankdata.clone();
    }

    /**
     * Returns the requested data array element from the first (default) bank.
     *
     * @param i The data array element you want to get.
     * @return The requested data array element as an integer.
     * @throws ArrayIndexOutOfBoundsException if {@code (i + getOffset())} is not a valid index.
     * @see #setElem(int, int)
     * @see #setElem(int, int, int)
     */
    public int getElem(int i) {
        checkIndex(i);
        return data[i+offset]&0xffff;
    }

    /**
     * Returns the requested data array element from the specified bank.
     *
     * @param bank The bank from which you want to get a data array element.
     * @param i The data array element you want to get.
     * @return The requested data array element as an integer.
     * @throws ArrayIndexOutOfBoundsException if {@code bank} is not a valid bank index,
     *         or {@code (i + getOffsets()[bank])} is not a valid index into the bank.
     * @see #setElem(int, int)
     * @see #setElem(int, int, int)
     */
    public int getElem(int bank, int i) {
        checkIndex(bank, i);
        return bankdata[bank][i+offsets[bank]]&0xffff;
    }

    /**
     * Sets the requested data array element in the first (default) bank
     * to the specified value.
     *
     * @param i The data array element you want to set.
     * @param val The integer value to which you want to set the data array element.
     * @throws ArrayIndexOutOfBoundsException if {@code (i + getOffset())} is not a valid index.
     * @see #getElem(int)
     * @see #getElem(int, int)
     */
    public void setElem(int i, int val) {
        checkIndex(i);
        data[i+offset] = (short)(val&0xffff);
        theTrackable.markDirty();
    }

    /**
     * Sets the requested data array element in the specified bank
     * from the given integer.
     * @param bank The bank in which you want to set the data array element.
     * @param i The data array element you want to set.
     * @param val The integer value to which you want to set the specified data array element.
     * @throws ArrayIndexOutOfBoundsException if {@code bank} is not a valid bank index,
     *         or {@code (i + getOffsets()[bank])} is not a valid index into the bank.
     * @see #getElem(int)
     * @see #getElem(int, int)
     */
    public void setElem(int bank, int i, int val) {
        checkIndex(bank, i);
        bankdata[bank][i+offsets[bank]] = (short)(val&0xffff);
        theTrackable.markDirty();
    }
}
