1 files changed, 38 insertions, 17 deletions

diff --git a/src/lib/encoder.h b/src/lib/encoder.h
index 64f113d74..8ab4479ef 100644
--- a/src/lib/encoder.h
+++ b/src/lib/encoder.h
@@ -26,11 +26,19 @@
 
 #include <boost/shared_ptr.hpp>
 #include <boost/thread/mutex.hpp>
+#include <boost/thread/condition.hpp>
+#include <boost/thread.hpp>
 #include <list>
 #include <stdint.h>
 extern "C" {
 #include <libavutil/samplefmt.h>
 }
+#ifdef HAVE_SWRESAMPLE
+extern "C" {
+#include <libswresample/swresample.h>
+}
+#endif
+#include <sndfile.h>
 #include "util.h"
 #include "video_sink.h"
 #include "audio_sink.h"
@@ -40,52 +48,43 @@ class Image;
 class Subtitle;
 class AudioBuffers;
 class Film;
+class ServerDescription;
+class DCPVideoFrame;
 
 /** @class Encoder
- *  @brief Parent class for classes which can encode video and audio frames.
+ *  @brief Encoder to J2K and WAV for DCP.
  *
  *  Video is supplied to process_video as YUV frames, and audio
  *  is supplied as uncompressed PCM in blocks of various sizes.
- *
- *  The subclass is expected to encode the video and/or audio in
- *  some way and write it to disk.
  */
 
 class Encoder : public VideoSink, public AudioSink
 {
 public:
 	Encoder (boost::shared_ptr<const Film> f, boost::shared_ptr<const EncodeOptions> o);
-	virtual ~Encoder () {}
+	virtual ~Encoder ();
 
 	/** Called to indicate that a processing run is about to begin */
-	virtual void process_begin () {}
+	virtual void process_begin ();
 
 	/** Call with a frame of video.
 	 *  @param i Video frame image.
+	 *  @param same true if i is the same as the last time we were called.
 	 *  @param s A subtitle that should be on this frame, or 0.
 	 */
-	void process_video (boost::shared_ptr<Image> i, boost::shared_ptr<Subtitle> s);
+	void process_video (boost::shared_ptr<Image> i, bool same, boost::shared_ptr<Subtitle> s);
 
 	/** Call with some audio data */
 	void process_audio (boost::shared_ptr<AudioBuffers>);
 
 	/** Called when a processing run has finished */
-	virtual void process_end () {}
+	virtual void process_end ();
 
 	float current_frames_per_second () const;
 	bool skipping () const;
 	SourceFrame video_frame () const;
 
 protected:
-
-	/** Called with a frame of video.
-	 *  @param i Video frame image.
-	 *  @param s A subtitle that should be on this frame, or 0.
-	 */
-	virtual void do_process_video (boost::shared_ptr<Image> i, boost::shared_ptr<Subtitle> s) = 0;
-	
-	/** Called with some audio data */
-	virtual void do_process_audio (boost::shared_ptr<AudioBuffers>) = 0;
 	
 	void frame_done ();
 	void frame_skipped ();
@@ -110,6 +109,28 @@ protected:
 	SourceFrame _video_frame;
 	/** Number of audio frames received so far */
 	int64_t _audio_frame;
+
+private:
+	void close_sound_files ();
+	void write_audio (boost::shared_ptr<const AudioBuffers> audio);
+
+	void encoder_thread (ServerDescription *);
+	void terminate_worker_threads ();
+	void link (std::string, std::string) const;
+
+#if HAVE_SWRESAMPLE	
+	SwrContext* _swr_context;
+#endif	
+
+	std::vector<SNDFILE*> _sound_files;
+	int64_t _audio_frames_written;
+
+	boost::optional<int> _last_real_frame;
+	bool _process_end;
+	std::list<boost::shared_ptr<DCPVideoFrame> > _queue;
+	std::list<boost::thread *> _worker_threads;
+	mutable boost::mutex _worker_mutex;
+	boost::condition _worker_condition;
 };
 
 #endif