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

  *     before starting the Application.

  * 

  * See the Basic and Advanced demos in the examples section for more

  * information.  If you have any trouble, please PM me on the jMonkeyEngine

  * forums.  My username is bortreb.

  * 

  * @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:

      * 

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

      *   frames-per-second.

      *

      *  2.) 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);

     }		

 }