org.jcodec.common.ByteBufferSeekableByteChannel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jcodec Show documentation

Pure Java implementation of video/audio codecs and formats

There is a newer version: 0.2.5

package org.jcodec.common;

import java.io.IOException;
import java.nio.ByteBuffer;

/**
 * This class is part of JCodec ( www.jcodec.org ) This software is distributed
 * under FreeBSD License
 * 
 * @author Jay Codec
 * 
 */
public class ByteBufferSeekableByteChannel implements SeekableByteChannel {

    private ByteBuffer backing;
    private boolean open;
    private int contentLength;

    public ByteBufferSeekableByteChannel(ByteBuffer backing) {
        this.backing = backing;
        this.open = true;
    }

    public boolean isOpen() {
        return open;
    }

    public void close() throws IOException {
        open = false;
    }

    public int read(ByteBuffer dst) throws IOException {
        if (!backing.hasRemaining()) {
            return -1;
        }
        int toRead = Math.min(backing.remaining(), dst.remaining());
        dst.put(NIOUtils.read(backing, toRead));
        contentLength = Math.max(contentLength, backing.position());
        return toRead;
    }

    public int write(ByteBuffer src) throws IOException {
        int toWrite = Math.min(backing.remaining(), src.remaining());
        backing.put(NIOUtils.read(src, toWrite));
        contentLength = Math.max(contentLength, backing.position());
        return toWrite;
    }

    public long position() throws IOException {
        return backing.position();
    }

    public SeekableByteChannel position(long newPosition) throws IOException {
        backing.position((int) newPosition);
        contentLength = Math.max(contentLength, backing.position());
        return this;
    }

    public long size() throws IOException {
        return contentLength;
    }

    public SeekableByteChannel truncate(long size) throws IOException {
        contentLength = (int) size;
        return this;
    }

    public ByteBuffer getContents() {
        ByteBuffer contents = backing.duplicate();
        contents.position(0);
        contents.limit(contentLength);
        return contents;
    }
}