Rename Server -> EncodeServer, ServerFinder -> EncodeServerFinder, ServerDescription...
[dcpomatic.git] / src / lib / encoder.h
index 902dae2d5e1f113ea8ebdaa0e9e4cb40a30a8779..20f2951b64cbb63ab4d1020ed1a60cba99bb5b6d 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
 #ifndef DCPOMATIC_ENCODER_H
 #define DCPOMATIC_ENCODER_H
 
-/** @file src/encoder.h
- *  @brief Encoder to J2K and WAV for DCP.
+/** @file  src/encoder.h
+ *  @brief Encoder class.
  */
 
+#include "util.h"
+#include "cross.h"
+#include "exception_store.h"
 #include <boost/shared_ptr.hpp>
 #include <boost/thread/mutex.hpp>
 #include <boost/thread/condition.hpp>
 #include <boost/thread.hpp>
 #include <boost/optional.hpp>
+#include <boost/signals2.hpp>
 #include <list>
 #include <stdint.h>
-extern "C" {
-#include <libavutil/samplefmt.h>
-#include <libswresample/swresample.h>
-}
-#include "util.h"
-#include "config.h"
 
-class Image;
-class AudioBuffers;
 class Film;
-class ServerDescription;
-class DCPVideoFrame;
-class EncodedData;
+class EncodeServerDescription;
+class DCPVideo;
 class Writer;
 class Job;
-class ServerFinder;
+class PlayerVideo;
 
 /** @class Encoder
- *  @brief Encoder to J2K and WAV for DCP.
+ *  @brief Class to manage encoding to JPEG2000.
  *
- *  Video is supplied to process_video as RGB frames, and audio
- *  is supplied as uncompressed PCM in blocks of various sizes.
+ *  This class keeps a queue of frames to be encoded and distributes
+ *  the work around threads and encoding servers.
  */
 
-class Encoder : public boost::noncopyable
+class Encoder : public boost::noncopyable, public ExceptionStore
 {
 public:
-       Encoder (boost::shared_ptr<const Film> f, boost::shared_ptr<Job>);
+       Encoder (boost::shared_ptr<const Film>, boost::shared_ptr<Writer>);
        virtual ~Encoder ();
 
        /** Called to indicate that a processing run is about to begin */
-       void process_begin ();
+       void 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.
+       /** Called to pass in zero or more bits of video to be encoded
+        *  as the next DCP frame.
         */
-       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<const AudioBuffers>);
+       void encode (std::list<boost::shared_ptr<PlayerVideo> > f);
 
        /** Called when a processing run has finished */
-       void process_end ();
+       void end ();
 
        float current_encoding_rate () const;
        int video_frames_out () const;
 
 private:
-       
+
+       void enqueue (boost::shared_ptr<PlayerVideo> f);
        void frame_done ();
-       
-       void encoder_thread (boost::optional<ServerDescription>);
+
+       void encoder_thread (boost::optional<EncodeServerDescription>);
        void terminate_threads ();
-       void add_worker_threads (ServerDescription);
-       void server_found (ServerDescription);
+       void servers_list_changed ();
 
        /** Film that we are encoding */
        boost::shared_ptr<const Film> _film;
-       boost::shared_ptr<Job> _job;
 
-       /** Mutex for _time_history and _last_frame */
+       /** Mutex for _time_history and _video_frames_enqueued */
        mutable boost::mutex _state_mutex;
        /** List of the times of completion of the last _history_size frames;
            first is the most recently completed.
@@ -100,20 +90,28 @@ private:
        std::list<struct timeval> _time_history;
        /** Number of frames that we should keep history for */
        static int const _history_size;
+       /** Current DCP frame index */
+       Frame _position;
+
+       /* XXX: probably should be atomic */
+       bool _terminate_enqueue;
+       bool _terminate_encoding;
+       /** Mutex for _threads */
+       mutable boost::mutex _threads_mutex;
+       std::list<boost::thread *> _threads;
+       mutable boost::mutex _queue_mutex;
+       std::list<boost::shared_ptr<DCPVideo> > _queue;
+       /** condition to manage thread wakeups when we have nothing to do */
+       boost::condition _empty_condition;
+       /** condition to manage thread wakeups when we have too much to do */
+       boost::condition _full_condition;
 
-       /** Number of video frames written for the DCP so far */
-       int _video_frames_out;
+       boost::shared_ptr<Writer> _writer;
+       Waker _waker;
 
-       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;
+       boost::shared_ptr<PlayerVideo> _last_player_video;
 
-       boost::shared_ptr<Writer> _writer;
-       boost::shared_ptr<ServerFinder> _server_finder;
+       boost::signals2::scoped_connection _server_found_connection;
 };
 
 #endif