X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fencoder.h;h=9fcba560f23085988ec4ca9ce8c51b8906ec5d68;hb=e8a6e330570b817c6c5fdfef0f508f2cbe0168ef;hp=e5916ad3a022bc8f2f9c8c117e182f339d7f3def;hpb=6a8f6bb77ab0a0e0b44bf3bd1cadbe7d55b1b632;p=dcpomatic.git

diff --git a/src/lib/encoder.h b/src/lib/encoder.h
index e5916ad3a..9fcba560f 100644
--- a/src/lib/encoder.h
+++ b/src/lib/encoder.h
@@ -17,11 +17,11 @@
 
 */
 
-#ifndef DVDOMATIC_ENCODER_H
-#define DVDOMATIC_ENCODER_H
+#ifndef DCPOMATIC_ENCODER_H
+#define DCPOMATIC_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>
@@ -33,105 +33,90 @@
 #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"
+#include "config.h"
 
-class EncodeOptions;
 class Image;
-class Subtitle;
 class AudioBuffers;
 class Film;
 class ServerDescription;
 class DCPVideoFrame;
+class EncodedData;
+class Writer;
+class Job;
 
 /** @class Encoder
  *  @brief Encoder to J2K and WAV for DCP.
  *
- *  Video is supplied to process_video as YUV frames, and audio
+ *  Video is supplied to process_video as RGB frames, and audio
  *  is supplied as uncompressed PCM in blocks of various sizes.
  */
 
-class Encoder : public VideoSink, public AudioSink
+class Encoder : public boost::noncopyable
 {
 public:
-	Encoder (boost::shared_ptr<const Film> f, boost::shared_ptr<const EncodeOptions> o);
+	Encoder (boost::shared_ptr<const Film> f, boost::shared_ptr<Job>);
 	virtual ~Encoder ();
 
 	/** Called to indicate that a processing run is about to begin */
-	virtual void process_begin ();
+	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, bool same, boost::shared_ptr<Subtitle> s);
+	void process_video (boost::shared_ptr<const Image> i, Eyes eyes, ColourConversion, bool same);
 
 	/** Call with some audio data */
-	void process_audio (boost::shared_ptr<AudioBuffers>);
+	void process_audio (boost::shared_ptr<const AudioBuffers>);
 
 	/** Called when a processing run has finished */
-	virtual void process_end ();
+	void process_end ();
 
-	float current_frames_per_second () const;
-	bool skipping () const;
-	SourceFrame video_frame () const;
+	float current_encoding_rate () const;
+	int video_frames_out () const;
 
-protected:
+private:
 	
 	void frame_done ();
-	void frame_skipped ();
 	
+	void encoder_thread (boost::optional<ServerDescription>);
+	void terminate_threads ();
+	void broadcast_thread ();
+	void listen_thread ();
+	void add_worker_threads (ServerDescription);
+
 	/** Film that we are encoding */
 	boost::shared_ptr<const Film> _film;
-	/** Options */
-	boost::shared_ptr<const EncodeOptions> _opt;
+	boost::shared_ptr<Job> _job;
 
-	/** Mutex for _time_history, _just_skipped and _last_frame */
-	mutable boost::mutex _history_mutex;
+	/** Mutex for _time_history and _last_frame */
+	mutable boost::mutex _state_mutex;
 	/** List of the times of completion of the last _history_size frames;
 	    first is the most recently completed.
 	*/
 	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;
 
-	/** Number of video frames received so far */
-	SourceFrame _video_frame;
-	/** Number of audio frames received so far */
-	int64_t _audio_frame;
+	/** Number of video frames written for the DCP so far */
+	int _video_frames_out;
 
-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	
+	bool _have_a_real_frame[EYES_COUNT];
+	bool _terminate;
+	std::list<boost::shared_ptr<DCPVideoFrame> > _queue;
+	typedef std::list<std::pair<boost::optional<ServerDescription>, boost::thread *> > ThreadList;
+	ThreadList _threads;
+	mutable boost::mutex _mutex;
+	boost::condition _condition;
 
-	std::vector<SNDFILE*> _sound_files;
-	int64_t _audio_frames_written;
+	boost::shared_ptr<Writer> _writer;
 
-	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;
+	/** A thread to periodically issue broadcasts to find encoding servers */
+	boost::thread* _broadcast_thread;
+	boost::thread* _listen_thread;
 };
 
 #endif