de.jarnbjo.jmf.TheoraDecoder Maven / Gradle / Ivy

Go to download
/*
 * $ProjectName$
 * $ProjectRevision$
 * -----------------------------------------------------------
 * $Id: TheoraDecoder.java,v 1.1 2003/03/03 22:06:12 jarnbjo Exp $
 * -----------------------------------------------------------
 *
 * $Author: jarnbjo $
 *
 * Description:
 *
 * Copyright 2002-2003 Tor-Einar Jarnbjo
 * -----------------------------------------------------------
 *
 * Change History
 * -----------------------------------------------------------
 * $Log: TheoraDecoder.java,v $
 * Revision 1.1  2003/03/03 22:06:12  jarnbjo
 * no message
 */
package de.jarnbjo.jmf;

import de.jarnbjo.theora.Header;
import de.jarnbjo.theora.TheoraStream;
import java.awt.Dimension;
import java.io.IOException;
import javax.media.Buffer;
import javax.media.Codec;
import javax.media.Format;
import javax.media.PlugIn;
import javax.media.format.RGBFormat;
import javax.media.format.VideoFormat;
import javax.media.format.YUVFormat;

public class TheoraDecoder implements Codec {
    private static final String CODEC_NAME = "Theora decoder";

    private static final Format[] supportedInputFormats = {
        new VideoFormat("THEORA")
    };

    private static final Format[] supportedOutputFormats = {
        new RGBFormat(null, -1, VideoFormat.intArray, -1.0f,
                32, 0xff0000, 0xff00, 0xff)
        //new YUVFormat(null, -1, VideoFormat.byteArray, -1.0f,
        //YUVFormat.YUV_422, -1, -1, -1, -1, -1)
    };

    final private TheoraStream theoraStream = new TheoraStream();
    private Format currentFormat;

    public TheoraDecoder() {
    }

    @Override
    public Format[] getSupportedInputFormats() {
        return supportedInputFormats;
    }

    @Override
    public Format[] getSupportedOutputFormats(Format input) {
        System.out.println("input: " + input);
        if (input == null) {
            return supportedOutputFormats;
        } else {
            VideoFormat vf = (VideoFormat) input;
            Format[] res = new Format[1];
            res[0] = new RGBFormat(vf.getSize(), vf.getMaxDataLength(),
                    Format.intArray, vf.getFrameRate(), 32,
                    0xff0000, 0xff00, 0xff);
            return res;
        }
    }

    boolean first = true;
    long sequence = 0;

    @Override
    public int process(Buffer in, Buffer out) {
        if (1 == 1) {
            return PlugIn.OUTPUT_BUFFER_NOT_FILLED;
        }

        int[] data = (int[]) out.getData();
        if (data == null || data.length < 352 * 208) {
            data = new int[352 * 208 * 3];
            java.util.Arrays.fill(data, 4711);
            out.setData(data);
        }

        out.setSequenceNumber(sequence);
        long time = 1000000000L * sequence / 24L;
        System.out.println("v: " + time);
        out.setTimeStamp(time);
        out.setFlags(Buffer.FLAG_NO_DROP);
        sequence++;
        out.setOffset(0);
        out.setLength(data.length);

        if (1 == 1) {
            return PlugIn.BUFFER_PROCESSED_OK;
        }

        try {
            if (first) {
                first = false;
                byte[] res = theoraStream.decodePacket((byte[]) in.getData());
                if (res == null) {
                    return PlugIn.OUTPUT_BUFFER_NOT_FILLED;
                }
            }
            Header header = theoraStream.getHeader();
            byte[] res = new byte[119808];
            currentFormat = new YUVFormat(
                    new Dimension(header.getWidth(), header.getHeight()),
                    res.length,
                    YUVFormat.byteArray,
                    (float) theoraStream.getHeader().getFrameRate(),
                    YUVFormat.YUV_420,
                    theoraStream.getPbi().getYStride(),
                    theoraStream.getPbi().getUvStride(),
                    theoraStream.getPbi().getYOffset(),
                    header.getHeight() * theoraStream.getPbi().getYStride(),
                    header.getHeight() * theoraStream.getPbi().getYStride()
                            * 5 / 4);
            //   theoraStream.getPbi().getUOffset(),
            //   theoraStream.getPbi().getVOffset());

            System.out.println("--frame info--");
            System.out.println("length: " + res.length);
            System.out.println(theoraStream.getPbi().getYStride());
            System.out.println(theoraStream.getPbi().getUvStride());
            System.out.println(theoraStream.getPbi().getYOffset());
            System.out.println(theoraStream.getPbi().getUOffset());
            System.out.println(theoraStream.getPbi().getVOffset());
            System.out.println("--frame info--");

            for (int i = 0; i < res.length; i++) {
                res[i] = (byte) (i);
            }

            out.setData(res);
            out.setOffset(0);
            out.setLength(res.length);
            out.setFormat(currentFormat);
            return PlugIn.BUFFER_PROCESSED_OK;
        } catch (IOException e) {
            e.printStackTrace();
            return PlugIn.BUFFER_PROCESSED_FAILED;
        }
    }

    @Override
    public Format setInputFormat(Format format) {
        return format;
    }

    @Override
    public Format setOutputFormat(Format format) {
        return format;
    }

    @Override
    public void open() {
    }

    @Override
    public void close() {
    }

    @Override
    public void reset() {
    }

    @Override
    public String getName() {
        return CODEC_NAME;
    }

    @Override
    public Object getControl(String controlType) {
        return null;
    }

    @Override
    public Object[] getControls() {
        return null;
    }
}