package com.aurellem.capture.video;

 

 import java.awt.image.BufferedImage;

 import java.io.File;

 import java.io.IOException;

 import java.util.concurrent.TimeUnit;

 

 import com.xuggle.mediatool.IMediaWriter;

 import com.xuggle.mediatool.ToolFactory;

 import com.xuggle.xuggler.IRational;

 

 

 /**

  * Handles writing video files using Xuggle.

  *

  * @author Robert McIntyre

  *

  */

 

 public  class XuggleVideoRecorder extends AbstractVideoRecorder{

 

 	IMediaWriter writer;

 	BufferedImage frame;

 	int videoChannel = 0;

 	long currentTimeStamp = 0;

 	boolean videoReady = false;

 	

 	

 	public XuggleVideoRecorder(File output) throws IOException {super(output);}

 	

 	public void initVideo(){

 		this.frame = new BufferedImage(

 				width, height,

 				BufferedImage.TYPE_3BYTE_BGR);

 		this.writer = ToolFactory.makeWriter(this.targetFileName);

 		writer.addVideoStream(videoChannel, 

 				0, IRational.make(fps), 

 				width, height);

 		this.videoReady = true;

 	}

 	

 	public void record(BufferedImage rawFrame) {

 		if (!this.videoReady){initVideo();}

 		// convert the Image into the form that Xuggle likes.

 		this.frame.getGraphics().drawImage(rawFrame, 0, 0, null);

 		writer.encodeVideo(videoChannel, 

 			frame,

 			currentTimeStamp, TimeUnit.NANOSECONDS);

 		

 		currentTimeStamp += (long) (1000000000.0 / fps);

 	}

 

 	public void finish() {

 		writer.close();

 	}

 	

 }