package com.aurellem.capture;

 

 import java.io.File;

 import java.io.IOException;

 import java.util.concurrent.Callable;

 

 import com.aurellem.capture.audio.MultiListener;

 import com.aurellem.capture.audio.WaveFileWriter;

 import com.aurellem.capture.video.AVIVideoRecorder;

 import com.aurellem.capture.video.AbstractVideoRecorder;

 import com.aurellem.capture.video.FileVideoRecorder;

 import com.aurellem.capture.video.XuggleVideoRecorder;

 import com.jme3.app.Application;

 import com.jme3.audio.AudioRenderer;

 import com.jme3.renderer.ViewPort;

 import com.jme3.scene.Spatial;

 import com.jme3.system.AppSettings;

 import com.jme3.system.JmeSystem;

 

 /**

  * Use the methods in this class for capturing consistent, high quality video and audio from a 

  * jMonkeyEngine3 application.  To capture audio or video do the following:

  * 

  * 1.) Set your application's timer to an IsoTimer. Create the IsoTimer with the desired video

  *     frames-per-second.

  * 2.) Call captureAudio and/or captureVideo on the Application as desired.

  *

  * @author Robert McIntyre

  */

 

 public class Capture {

 

 	/**

 	 * Use this function to capture video from your application.  You specify the framerate at which

 	 * the video will be recording by setting the Application's timer to an IsoTimer created with the 

 	 * desired frames-per-second. 

 	 * 

 	 * There are three ways to record and they are selected by the properties of the file that you 

 	 * specify.

 	 * 

 	 * 1.) (Preferred)

 	 *     If you supply an empty directory as the file, then the video will

 	 *     be saved as a sequence of .png files, one file per frame.  The files start at

 	 *     0000000.png and increment from there.  You can then combine the frames into your  

 	 *     preferred container/codec. If the directory is not empty, then writing video frames

 	 *     to it will fail, and nothing will be written.

 	 *     

 	 * 2.) If the filename ends in ".avi" then the frames will be encoded as a RAW stream

 	 *     inside an AVI 1.0 container.  The resulting file will be quite large and you will

 	 *     probably want to re-encode it to your preferred container/codec format.  Be advised

 	 *     that some video payers cannot process AVI with a RAW stream, and that AVI 1.0 files

 	 *     generated by this method that exceed 2.0GB are invalid according to the AVI 1.0 spec

 	 *     (but many programs can still deal with them.)  Thanks to Werner Randelshofer for his 

 	 *     excellent work which made AVI file writer option possible.

 	 *  

 	 * 3.) Any non-directory file ending in anything other than ".avi" will be processed through 

 	 *     Xuggle.  Xuggle provides the option to use many codecs/containers, but you will have to 

 	 *     install it on your system yourself in order to use this option. Please visit 

 	 *     http://www.xuggle.com/ to learn how to do this.

 	 * 

 	 * @param app The Application from which you wish to record Video.

 	 * @param file The file to which the video will be captured. 

 	 * @throws IOException

 	 */

 	

 	public static void captureVideo(final Application app, final File file) throws IOException{

 		final AbstractVideoRecorder videoRecorder;

 

 		if (file.getCanonicalPath().endsWith(".avi")){

 			videoRecorder = new AVIVideoRecorder(file);}

 		else if (file.isDirectory()){

 			videoRecorder = new FileVideoRecorder(file);}

 		else { videoRecorder = new XuggleVideoRecorder(file);}

 

 		Callable<Object> thunk = new Callable<Object>(){

 			public Object call(){

 

 				ViewPort viewPort = 

 						app.getRenderManager()

 						.createPostView("aurellem video record", app.getCamera());

 

 				viewPort.setClearFlags(false, false, false);

 

 				// get GUI node stuff

 				for (Spatial s : app.getGuiViewPort().getScenes()){

 					viewPort.attachScene(s);

 				}

 

 				app.getStateManager().attach(videoRecorder);

 				viewPort.addProcessor(videoRecorder);

 				return null;

 			}

 		};

 		app.enqueue(thunk);

 	}

 

 

 	/**

 	 * Use this function to capture audio from your application.  Audio data will be saved in linear

 	 * PCM format at 44,100 hertz in the wav container format to the file that you specify.  

 	 * 

 	 * Note that you *have* to use an IsoTimer for your Application's timer while recording audio or 

 	 * the recording will fail. Also ensure that your IsoTimer obeys the following constraints: 

 	 * 

 	 *  - The frames-per-second value of the IsoTimer cannot be lower than 10 frames-per-second.

 	 *  - The frames-per-second value of the IsoTimer must evenly divide 44,100.  

 	 * 

 	 * @param app The Application from which you wish to record Audio.

 	 * @param file the target file to which you want to record audio data.

 	 * @throws IOException

 	 */

 	

 	public static void captureAudio(final Application app, final File file) throws IOException{

 		AppSettings settings = null;

 		if (app.getContext() != null){settings = app.getContext().getSettings();}

 		if (settings == null){settings = new AppSettings(true);}

 		settings.setAudioRenderer("Send");

 		app.setSettings(settings);

 

 		JmeSystem.setSystemDelegate(new AurellemSystemDelegate());

 		

 		final WaveFileWriter writer = new WaveFileWriter(file);

 

 		Callable<Object> thunk = new Callable<Object>(){

 			public Object call(){

 				AudioRenderer ar = app.getAudioRenderer();

 				if (ar instanceof MultiListener){

 					MultiListener ml = (MultiListener)ar;

 					ml.registerSoundProcessor(writer);

 				}

 				return null;

 			}

 		};

 		app.enqueue(thunk);

 	}		

 }