X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fencoder.h;h=86880bc34942561ba12a1488feb00663c2c9d6d8;hb=db468a15e50c8491d4b8462ad0676be905f49065;hp=561e419017b1e75488edf9cf0f55bdb85114bc7a;hpb=572f030509c2b0c1f662a54624ab0be6efdd9024;p=dcpomatic.git

diff --git a/src/lib/encoder.h b/src/lib/encoder.h
index 561e41901..86880bc34 100644
--- a/src/lib/encoder.h
+++ b/src/lib/encoder.h
@@ -21,74 +21,82 @@
 #define DVDOMATIC_ENCODER_H
 
 /** @file src/encoder.h
- *  @brief Parent class for classes which can encode video and audio frames.
+ *  @brief Encoder to J2K and WAV for DCP.
  */
 
 #include <boost/shared_ptr.hpp>
 #include <boost/thread/mutex.hpp>
+#include <boost/thread/condition.hpp>
+#include <boost/thread.hpp>
+#include <boost/optional.hpp>
 #include <list>
 #include <stdint.h>
 extern "C" {
 #include <libavutil/samplefmt.h>
 }
+#ifdef HAVE_SWRESAMPLE
+extern "C" {
+#include <libswresample/swresample.h>
+}
+#endif
+#include "util.h"
+#include "video_sink.h"
+#include "audio_sink.h"
 
-class FilmState;
-class Options;
 class Image;
-class Log;
 class Subtitle;
 class AudioBuffers;
+class Film;
+class ServerDescription;
+class DCPVideoFrame;
+class EncodedData;
+class Writer;
 
 /** @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
+class Encoder : public VideoSink, public AudioSink
 {
 public:
-	Encoder (boost::shared_ptr<const FilmState> s, boost::shared_ptr<const Options> o, Log* l);
+	Encoder (boost::shared_ptr<Film> f);
+	virtual ~Encoder ();
 
 	/** Called to indicate that a processing run is about to begin */
-	virtual void process_begin (int64_t audio_channel_layout) = 0;
+	virtual void process_begin ();
 
-	/** Called with a frame of video.
+	/** Call with a frame of video.
 	 *  @param i Video frame image.
-	 *  @param f Frame number within the film.
+	 *  @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.
 	 */
-	virtual void process_video (boost::shared_ptr<Image> i, int f, boost::shared_ptr<Subtitle> s) = 0;
+	void process_video (boost::shared_ptr<Image> i, bool same, boost::shared_ptr<Subtitle> s);
 
-	/** Called with some audio data.
-	 *  @param d Array of pointers to floating point sample data for each channel.
-	 *  @param s Number of frames (ie number of samples in each channel)
-	 */
-	virtual void process_audio (boost::shared_ptr<const AudioBuffers>) = 0;
+	/** Call with some audio data */
+	void process_audio (boost::shared_ptr<AudioBuffers>);
 
 	/** Called when a processing run has finished */
-	virtual void process_end () = 0;
+	virtual void process_end ();
 
 	float current_frames_per_second () const;
-	bool skipping () const;
-	int last_frame () const;
+	int video_frames_out () const;
 
-protected:
-	void frame_done (int n);
-	void frame_skipped ();
+private:
+	
+	void frame_done ();
 	
-	/** FilmState of the film that we are encoding */
-	boost::shared_ptr<const FilmState> _fs;
-	/** Options */
-	boost::shared_ptr<const Options> _opt;
-	/** Log */
-	Log* _log;
-
-	/** Mutex for _time_history, _just_skipped and _last_frame */
+	void write_audio (boost::shared_ptr<const AudioBuffers> data);
+
+	void encoder_thread (ServerDescription *);
+	void terminate_threads ();
+
+	/** Film that we are encoding */
+	boost::shared_ptr<Film> _film;
+
+	/** Mutex for _time_history and _last_frame */
 	mutable boost::mutex _history_mutex;
 	/** List of the times of completion of the last _history_size frames;
 	    first is the most recently completed.
@@ -96,10 +104,24 @@ protected:
 	std::list<struct timeval> _time_history;
 	/** Number of frames that we should keep history for */
 	static int const _history_size;
-	/** true if the last frame we processed was skipped (because it was already done) */
-	bool _just_skipped;
-	/** Index of the last frame to be processed */
-	int _last_frame;
+
+	/** Number of video frames received so far */
+	SourceFrame _video_frames_in;
+	/** Number of video frames written for the DCP so far */
+	int _video_frames_out;
+
+#if HAVE_SWRESAMPLE	
+	SwrContext* _swr_context;
+#endif
+
+	bool _have_a_real_frame;
+	bool _terminate;
+	std::list<boost::shared_ptr<DCPVideoFrame> > _queue;
+	std::list<boost::thread *> _threads;
+	mutable boost::mutex _mutex;
+	boost::condition _condition;
+
+	boost::shared_ptr<Writer> _writer;
 };
 
 #endif