org.jcodec.codecs.raw.RAWVideoEncoder 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.codecs.raw;

import java.nio.ByteBuffer;

import org.jcodec.common.VideoEncoder;
import org.jcodec.common.model.ColorSpace;
import org.jcodec.common.model.Picture;

/**
 * This class is part of JCodec ( www.jcodec.org ) This software is distributed
 * under FreeBSD License
 * 
 * Encodes image data into raw bytes according to the color space.
 * 
 * @author The JCodec project
 * 
 */
public class RAWVideoEncoder extends VideoEncoder {

	@Override
	public EncodedFrame encodeFrame(Picture pic, ByteBuffer _out) {
		ByteBuffer dup = _out.duplicate();

		ColorSpace color = pic.getColor();
		if (color.planar) {
			for (int plane = 0; plane < color.nComp; plane++) {
				int width = pic.getWidth() >> color.compWidth[plane];
				int startX = pic.getStartX();
				int startY = pic.getStartY();
				int cropW = pic.getCroppedWidth() >> color.compWidth[plane];
				int cropH = pic.getCroppedHeight() >> color.compHeight[plane];
				byte[] planeData = pic.getPlaneData(plane);

				int pos = width * startY + startX;
				for (int y = 0; y < cropH; y++) {
					for (int x = 0; x < cropW; x++) {
						dup.put((byte) (planeData[pos + x] + 128));
					}
					pos += width;
				}
			}
		} else {
			int bytesPerPixel = (color.bitsPerPixel + 7) >> 3;
			int stride = pic.getWidth() * bytesPerPixel;
			int startX = pic.getStartX();
			int startY = pic.getStartY();
			int cropW = pic.getCroppedWidth();
			int cropH = pic.getCroppedHeight();
			byte[] planeData = pic.getPlaneData(0);

			int pos = stride * startY + startX * bytesPerPixel;
			for (int y = 0; y < cropH; y++) {
				for (int x = 0, off = 0; x < cropW; x++, off += bytesPerPixel) {
					for (int b = 0; b < bytesPerPixel; b++)
						dup.put((byte) (planeData[pos + off + b] + 128));
				}
				pos += stride;
			}
		}
		dup.flip();
		return new EncodedFrame(dup, true);
	}

	@Override
	public ColorSpace[] getSupportedColorSpaces() {
		return null;
	}

	@Override
	public int estimateBufferSize(Picture frame) {
		int fullPlaneSize = frame.getWidth() * frame.getCroppedHeight();
		ColorSpace color = frame.getColor();
		int totalSize = 0;
		for (int i = 0; i < color.nComp; i++) {
			totalSize += (fullPlaneSize >> color.compWidth[i]) >> color.compHeight[i];
		}
		return totalSize;
	}

}