ByteArrayOutputStream.avt

/*
    Исходный код среды исполнения ПВТ-ОО.

    Этот исходный код является частью проекта ПВТ-ОО.

    Copyright © 2021 Малик Разработчик

    Это свободная программа: вы можете перераспространять её и/или
    изменять её на условиях Меньшей Стандартной общественной лицензии GNU в том виде,
    в каком она была опубликована Фондом свободного программного обеспечения;
    либо версии 3 лицензии, либо (по вашему выбору) любой более поздней версии.

    Эта программа распространяется в надежде, что она может быть полезна,
    но БЕЗО ВСЯКИХ ГАРАНТИЙ; даже без неявной гарантии ТОВАРНОГО ВИДА
    или ПРИГОДНОСТИ ДЛЯ ОПРЕДЕЛЁННЫХ ЦЕЛЕЙ. Подробнее см. в Меньшей Стандартной
    общественной лицензии GNU.

    Вы должны были получить копию Меньшей Стандартной общественной лицензии GNU
    вместе с этой программой. Если это не так, см.
    <http://www.gnu.org/licenses/>.
*/

package avt.io;

import avt.io.extension.*;

public class ByteArrayOutputStream(Object, Closeable, Extension, MarkExtension, ByteWriter, TruncateExtension, LimitedSizeExtension, SeekExtension, Extendable)
{
    package int fldLength;
    package int fldPosition;
    package byte[] fldArray;
    package final ByteArrayInputStream fldSource;
    private int fldMarkedPosition;

    public () {
        fldArray = new byte[127];
    }

    public (byte[] array) {
        int slen = fldLength = fldPosition = array == null ? 0 : array.length;
        Array.copy(array, 0, fldArray = new byte[slen >= Int.MAX_VALUE - 127 ? Int.MAX_VALUE : slen + 127], 0, slen);
    }

    public (byte[] array, int length) {
        int alen = array == null ? 0 : array.length;
        int slen = fldLength = fldPosition = length < 0 ? 0 : length > alen ? alen : length;
        Array.copy(array, 0, fldArray = new byte[slen >= Int.MAX_VALUE - 127 ? Int.MAX_VALUE : slen + 127], 0, slen);
    }

    public (byte[] array, int length, int position) {
        int alen = array == null ? 0 : array.length;
        int slen = fldLength = length < 0 ? 0 : length > alen ? alen : length;
        fldPosition = position < 0 ? 0 : position > slen ? slen : position;
        Array.copy(array, 0, fldArray = new byte[slen >= Int.MAX_VALUE - 127 ? Int.MAX_VALUE : slen + 127], 0, slen);
    }

    protected (ByteArrayInputStream source, int position) {
        if(source == null)
        {
            fldArray = new byte[127];
            return;
        }
        int slen = fldLength = source.fldLength;
        fldPosition = position < 0 ? 0 : position > slen ? slen : position;
        fldArray = source.fldArray;
        fldSource = source;
    }

    public String toString() { return new String(fldArray, 0, fldLength); }

    public void close() {  }

    public void reset() { fldPosition = fldMarkedPosition; }

    public void mark(int readLimit) { fldMarkedPosition = fldPosition; }

    public void write(int byteData) {
        int curlen = fldLength;
        int curpos = fldPosition;
        int newpos = curpos + 1;
        byte[] array = newpos >= 0 && newpos <= curlen ? fldArray : madeLarger(curlen, newpos);
        array[curpos] = (byte) byteData;
        fldPosition = newpos;
    }

    public void write(byte[] src) { write(src, 0, src == null ? 0 : src.length); }

    public void write(byte[] src, int offset, int length) {
        if(src == null)
        {
            throw new NullPointerException("аргумент src равен нулевой ссылке");
        }
        Array.checkBounds(src, offset, length);
        if(length <= 0) return;
        int curlen = fldLength;
        int curpos = fldPosition;
        int newpos = curpos + length;
        byte[] array = newpos >= 0 && newpos <= curlen ? fldArray : madeLarger(curlen, newpos);
        Array.copy(src, offset, array, curpos, length);
        fldPosition = newpos;
    }

    public void flush() {  }

    public void truncate() {
        int length = fldLength = fldPosition;
        if(fldMarkedPosition > length) fldMarkedPosition = length;
        ByteArrayInputStream source = fldSource;
        if(source != null)
        {
            source.fldLength = length;
            source.notifyLengthChanged(length);
        }
    }

    public long available() { return (long) (fldLength - fldPosition); }

    public long position() { return (long) fldPosition; }

    public long size() { return (long) fldLength; }

    public long seek(long offset, int from) {
        long pos;
        long length = (long) fldLength;
        switch(from)
        {
        case BEGIN:
            pos = offset;
            break;
        case END:
            pos = offset + length;
            break;
        case CURRENT:
            pos = offset + (long) fldPosition;
            break;
        default:
            throw new IllegalArgumentException("аргумент from имеет недопустимое значение");
        }
        if(pos < 0L) pos = 0L;
        if(pos > length) pos = length;
        return (long) (fldPosition = (int) pos);
    }

    public Extension[] getExtensions() { return new Extension[] { this }; }

    public byte[] toByteArray() {
        int length = fldLength;
        byte[] result = new byte[length];
        Array.copy(fldArray, 0, result, 0, length);
        return result;
    }

    package final void notifyLengthChanged(int length) {
        if(fldPosition > length) fldPosition = length;
        if(fldMarkedPosition > length) fldMarkedPosition = length;
    }

    private byte[] madeLarger(int curLength, int newLength) {
        if(newLength < 0)
        {
            throw new BufferTooLargeError("объём буфера очень велик");
        }
        ByteArrayInputStream source = fldSource;
        byte[] array = fldArray;
        if(array == null)
        {
            fldArray = array = new byte[newLength >= Int.MAX_VALUE - 127 ? Int.MAX_VALUE : newLength + 127];
            if(source != null) source.fldArray = array;
        } else
        {
            int curCapacity = array.length;
            if(newLength > curCapacity)
            {
                int newCapacity = (curCapacity << 1) + 1;
                if(newCapacity < 0) newCapacity = Int.MAX_VALUE;
                if(newCapacity < newLength) newCapacity = newLength >= Int.MAX_VALUE - 127 ? Int.MAX_VALUE : newLength + 127;
                Array.copy(array, 0, fldArray = array = new byte[newCapacity], 0, curLength);
                if(source != null) source.fldArray = array;
            }
        }
        fldLength = newLength;
        if(source != null) source.fldLength = newLength;
        return array;
    }
}