CharEncoder.avt

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

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

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

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

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

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

package avt.io.charset;

import avt.io.*;
import avt.io.extension.*;
import avt.lang.array.*;

public abstract class CharEncoder(Object)
{
    private static CharSource toStream(CharArray src, int offset, int length) {
        char[] array = new char[length];
        for(int i = 0; i < length; offset++, i++) array[i] = src[offset];
        return new CharSource(array);
    }

    private static CharSource toStream(CharReader src, int length) throws IOException {
        char[] array = new char[length];
        src.read(array, 0, length);
        return new CharSource(array);
    }


    private long fldGivedBytes;
    private long fldAcceptedChars;
    private EncodingErrorHandler fldHandler;
    private ErrorAction fldErrorAction;
    private final Charset fldCharset;
    private final Object fldMonitor;

    protected (Charset charset) {
        fldErrorAction = ErrorAction.replace;
        fldCharset = charset;
        fldMonitor = new Object();
    }

    public boolean canEncode(char src) { return true; }

    public final void encodeToWriter(ByteWriter dst, CharArray src) throws CharacterEncodingException, IOException {
        if(dst == null)
        {
            throw new NullPointerException("аргумент dst равен нулевой ссылке");
        }
        dst.write(encode(src));
    }

    public final void encodeToWriter(ByteWriter dst, CharArray src, int offset, int length) throws CharacterEncodingException, IOException {
        if(dst == null)
        {
            throw new NullPointerException("аргумент dst равен нулевой ссылке");
        }
        dst.write(encode(src, offset, length));
    }

    public final void encodeToWriter(ByteWriter dst, CharReader src, int length) throws CharacterEncodingException, IOException {
        if(dst == null)
        {
            throw new NullPointerException("аргумент dst равен нулевой ссылке");
        }
        dst.write(encode(src, length));
    }

    public final void reset() { synchronized(fldMonitor) { fldGivedBytes = fldAcceptedChars = 0L; } }

    public final boolean canEncode(CharArray src) { return canEncode(src, 0, src == null ? 0 : src.length); }

    public final boolean canEncode(CharArray src, int offset, int length) {
        if(src == null)
        {
            throw new NullPointerException("аргумент src равен нулевой ссылке");
        }
        Array.checkBounds(src, offset, length);
        while(length-- > 0)
        {
            if(!canEncode(src[offset++])) return false;
        }
        return true;
    }

    public final long givedBytes() { return fldGivedBytes; }

    public final long acceptedChars() { return fldAcceptedChars; }

    public final byte[] encode(CharArray src) throws CharacterEncodingException { return encode(src, 0, src == null ? 0 : src.length); }

    public final byte[] encode(CharArray src, int offset, int length) throws CharacterEncodingException {
        if(src == null)
        {
            throw new NullPointerException("аргумент src равен нулевой ссылке");
        }
        Array.checkBounds(src, offset, length);
        if(length <= 0) return new byte[0];
        long deltaAcceptedChars = 0L;
        ByteArrayOutputStream result = new ByteArrayOutputStream();
        try
        {
            if(src instanceof char[])
            {
                CharSource buffer = new CharSource((char[]) src, offset + length, offset);
                encode(buffer, length, result);
                deltaAcceptedChars = buffer.position - offset;
            } else
            {
                CharSource buffer = toStream(src, offset, length);
                encode(buffer, length, result);
                deltaAcceptedChars = buffer.position;
            }
        }
        catch(IOException e) {  }
        long deltaGivedBytes = result.position();
        synchronized(fldMonitor)
        {
            fldAcceptedChars += deltaAcceptedChars;
            fldGivedBytes += deltaGivedBytes;
        }
        return result.toByteArray();
    }

    public final byte[] encode(CharReader src, int length) throws CharacterEncodingException, IOException {
        if(src == null)
        {
            throw new NullPointerException("аргумент src равен нулевой ссылке");
        }
        if(length <= 0) return new byte[0];
        long deltaAcceptedChars = 0L;
        ByteArrayOutputStream result = new ByteArrayOutputStream();
        SeekExtension seekable = null;
        if(src instanceof CharSource)
        {
            CharSource buffer = (CharSource) src;
            long offset = buffer.position;
            encode(src, length, result);
            deltaAcceptedChars = buffer.position - offset;
        }
        else if((seekable = (SeekExtension) src.getExtension(SeekExtension.class)) != null)
        {
            long offset = seekable.position();
            encode(src, length, result);
            deltaAcceptedChars = seekable.position() - offset;
        }
        else
        {
            CharSource buffer = toStream(src, length);
            encode(buffer, length, result);
            deltaAcceptedChars = buffer.position;
        }
        long deltaGivedBytes = result.position();
        synchronized(fldMonitor)
        {
            fldAcceptedChars += deltaAcceptedChars;
            fldGivedBytes += deltaGivedBytes;
        }
        return result.toByteArray();
    }

    public final Charset charset { read = fldCharset }

    public final ErrorAction errorAction { read = fldErrorAction, write = setErrorAction }

    public final EncodingErrorHandler encodingErrorHandler { read = fldHandler, write = fldHandler }

    protected abstract void encode(CharReader src, int length, ByteWriter dst) throws CharacterEncodingException, IOException;

    protected final void report() {
        EncodingErrorHandler handler = fldHandler;
        if(handler != null) handler.encodingError(this);
    }

    private void setErrorAction(ErrorAction errorAction) { fldErrorAction = errorAction == null ? ErrorAction.report : errorAction; }
}