Remove ref_write mechanism and instead maintain state for each
[dcpomatic.git] / src / lib / writer.h
index 90b36962cf0acf3637b3bf948c4464ceac247413..71ab4dc1b5128eec759e7df5e8515e70412dd9af 100644 (file)
  *  @brief Writer class.
  */
 
-#include "exceptions.h"
 #include "types.h"
 #include "player_subtitles.h"
 #include "data.h"
+#include "exception_store.h"
 #include <dcp/picture_asset_writer.h>
 #include <boost/shared_ptr.hpp>
 #include <boost/weak_ptr.hpp>
@@ -37,6 +37,7 @@ class Data;
 class AudioBuffers;
 class Job;
 class Font;
+class ReferencedReelAsset;
 
 namespace dcp {
        class MonoPictureAsset;
@@ -48,6 +49,7 @@ namespace dcp {
        class SoundAsset;
        class SoundAssetWriter;
        class SubtitleAsset;
+       class ReelAsset;
 }
 
 struct QueueItem
@@ -56,7 +58,7 @@ public:
        QueueItem ()
                : size (0)
        {}
-       
+
        enum Type {
                /** a normal frame with some JPEG200 data */
                FULL,
@@ -72,8 +74,11 @@ public:
        boost::optional<Data> encoded;
        /** size of data for FAKE */
        int size;
-       /** frame index */
+       /** reel index */
+       size_t reel;
+       /** frame index within the reel */
        int frame;
+       /** eyes for FULL, FAKE and REPEAT */
        Eyes eyes;
 };
 
@@ -87,7 +92,7 @@ bool operator== (QueueItem const & a, QueueItem const & b);
  *  or AudioBuffers objects (containing image or sound data respectively)
  *  and writes them to the assets.
  *
- *  ::write() for Data can be called out of order, and the Writer
+ *  ::write() for Data (picture) can be called out of order, and the Writer
  *  will sort it out.  write() for AudioBuffers must be called in order.
  */
 
@@ -97,34 +102,65 @@ public:
        Writer (boost::shared_ptr<const Film>, boost::weak_ptr<Job>);
        ~Writer ();
 
-       bool can_fake_write (int) const;
-       
-       void write (Data, int, Eyes);
-       void fake_write (int, Eyes);
-       void repeat (int, Eyes);
+       void start ();
+
+       bool can_fake_write (Frame) const;
+
+       void write (Data, Frame, Eyes);
+       void fake_write (Frame, Eyes);
+       void repeat (Frame, Eyes);
        void write (boost::shared_ptr<const AudioBuffers>);
        void write (PlayerSubtitles subs);
        void write (std::list<boost::shared_ptr<Font> > fonts);
+       void write (ReferencedReelAsset asset);
        void finish ();
 
        void set_encoder_threads (int threads);
 
 private:
 
+       class Reel {
+       public:
+               Reel ()
+                       : first_nonexistant_frame (0)
+                       , last_written_video_frame (-1)
+                       , last_written_eyes (EYES_RIGHT)
+                       , total_written_audio_frames (0)
+               {}
+
+               DCPTimePeriod period;
+               /** the first frame index that does not already exist in our MXF */
+               int first_nonexistant_frame;
+               /** the data of the last written frame, if there is one */
+               boost::optional<Data> last_written[EYES_COUNT];
+               /** the index of the last written video frame within the reel */
+               int last_written_video_frame;
+               Eyes last_written_eyes;
+               /** the number of audio frames that have been written to the reel */
+               int total_written_audio_frames;
+
+               boost::shared_ptr<dcp::PictureAsset> picture_asset;
+               boost::shared_ptr<dcp::PictureAssetWriter> picture_asset_writer;
+               boost::shared_ptr<dcp::SoundAsset> sound_asset;
+               boost::shared_ptr<dcp::SoundAssetWriter> sound_asset_writer;
+               boost::shared_ptr<dcp::SubtitleAsset> subtitle_asset;
+       };
+
        void thread ();
        void terminate_thread (bool);
-       void check_existing_picture_asset ();
-       bool check_existing_picture_asset_frame (FILE *, int, Eyes);
+       void check_existing_picture_asset (Reel& reel);
        bool have_sequenced_image_at_queue_head ();
-       void write_frame_info (int frame, Eyes eyes, dcp::FrameInfo info) const;
+       void write_frame_info (Reel const & reel, int frame, Eyes eyes, dcp::FrameInfo info) const;
        long frame_info_position (int frame, Eyes eyes) const;
        dcp::FrameInfo read_frame_info (FILE* file, int frame, Eyes eyes) const;
+       size_t video_reel (int frame) const;
 
        /** our Film */
        boost::shared_ptr<const Film> _film;
        boost::weak_ptr<Job> _job;
-       /** the first frame index that does not already exist in our MXF */
-       int _first_nonexistant_frame;
+       std::vector<Reel> _reels;
+       std::vector<Reel>::iterator _audio_reel;
+       std::vector<Reel>::iterator _subtitle_reel;
 
        /** our thread, or 0 */
        boost::thread* _thread;
@@ -135,21 +171,16 @@ private:
        /** number of FULL frames whose JPEG200 data is currently held in RAM */
        int _queued_full_in_memory;
        /** mutex for thread state */
-       mutable boost::mutex _mutex;
+       mutable boost::mutex _state_mutex;
        /** 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;
-       /** the data of the last written frame, if there is one */
-       boost::optional<Data> _last_written[EYES_COUNT];
-       /** the index of the last written frame */
-       int _last_written_frame;
-       Eyes _last_written_eyes;
        /** maximum number of frames to hold in memory, for when we are managing
         *  ordering
         */
        int _maximum_frames_in_memory;
-       
+
        /** number of FULL written frames */
        int _full_written;
        /** number of FAKE written frames */
@@ -159,12 +190,10 @@ private:
            due to the limit of frames to be held in memory.
        */
        int _pushed_to_disk;
-       
-       boost::shared_ptr<dcp::PictureAsset> _picture_asset;
-       boost::shared_ptr<dcp::PictureAssetWriter> _picture_asset_writer;
-       boost::shared_ptr<dcp::SoundAsset> _sound_asset;
-       boost::shared_ptr<dcp::SoundAssetWriter> _sound_asset_writer;
-       boost::shared_ptr<dcp::SubtitleAsset> _subtitle_asset;
+
+       std::list<ReferencedReelAsset> _reel_assets;
 
        std::list<boost::shared_ptr<Font> > _fonts;
+
+       static int const _info_size;
 };