X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fwriter.h;h=1e25c3bdf644d455434622e761629af9fcc3cb90;hb=ccd69987ab861ea4521836fa2a5227e2dfac3479;hp=484ca1285043073f9f2a64af74fd2c91d10244c3;hpb=8855efa208776139bf9ce5a188db080e1e136e9b;p=dcpomatic.git diff --git a/src/lib/writer.h b/src/lib/writer.h index 484ca1285..1e25c3bdf 100644 --- a/src/lib/writer.h +++ b/src/lib/writer.h @@ -1,5 +1,5 @@ /* - Copyright (C) 2012-2018 Carl Hetherington + Copyright (C) 2012-2021 Carl Hetherington This file is part of DCP-o-matic. @@ -18,41 +18,45 @@ */ + /** @file src/lib/writer.h * @brief Writer class. */ + +#include "atmos_metadata.h" #include "types.h" #include "player_text.h" #include "exception_store.h" -#include -#include +#include "dcp_text_track.h" +#include "weak_film.h" +#include #include #include #include + namespace dcp { class Data; } +namespace dcpomatic { + class FontData; +} + class Film; class AudioBuffers; class Job; -class Font; class ReferencedReelAsset; class ReelWriter; + struct QueueItem { public: - QueueItem () - : size (0) - , reel (0) - , frame (0) - , eyes (EYES_BOTH) - {} - - enum Type { + QueueItem () {} + + enum class Type { /** a normal frame with some JPEG200 data */ FULL, /** a frame whose data already exists in the MXF, @@ -64,20 +68,22 @@ public: } type; /** encoded data for FULL */ - boost::optional encoded; + std::shared_ptr encoded; /** size of data for FAKE */ - int size; + int size = 0; /** reel index */ - size_t reel; + size_t reel = 0; /** frame index within the reel */ - int frame; + int frame = 0; /** eyes for FULL, FAKE and REPEAT */ - Eyes eyes; + Eyes eyes = Eyes::BOTH; }; + bool operator< (QueueItem const & a, QueueItem const & b); bool operator== (QueueItem const & a, QueueItem const & b); + /** @class Writer * @brief Class to manage writing JPEG2000 and audio data to assets on disk. * @@ -89,25 +95,29 @@ bool operator== (QueueItem const & a, QueueItem const & b); * will sort it out. write() for AudioBuffers must be called in order. */ -class Writer : public ExceptionStore, public boost::noncopyable +class Writer : public ExceptionStore, public WeakConstFilm { public: - Writer (boost::shared_ptr, boost::weak_ptr); + Writer (std::weak_ptr, std::weak_ptr, bool text_only = false); ~Writer (); + Writer (Writer const &) = delete; + Writer& operator= (Writer const &) = delete; + void start (); bool can_fake_write (Frame) const; - void write (dcp::Data, Frame, Eyes); + void write (std::shared_ptr, Frame, Eyes); void fake_write (Frame, Eyes); bool can_repeat (Frame) const; void repeat (Frame, Eyes); - void write (boost::shared_ptr, DCPTime time); - void write (PlayerText text, TextType type, DCPTimePeriod period); - void write (std::list > fonts); + void write (std::shared_ptr, dcpomatic::DCPTime time); + void write (PlayerText text, TextType type, boost::optional, dcpomatic::DCPTimePeriod period); + void write (std::vector fonts); void write (ReferencedReelAsset asset); - void finish (); + void write (std::shared_ptr atmos, dcpomatic::DCPTime time, AtmosMetadata metadata); + void finish (boost::filesystem::path output_dcp); void set_encoder_threads (int threads); @@ -117,23 +127,26 @@ private: bool have_sequenced_image_at_queue_head (); size_t video_reel (int frame) const; void set_digest_progress (Job* job, float progress); - void write_cover_sheet (); + void write_cover_sheet (boost::filesystem::path output_dcp); + void calculate_referenced_digests (std::function set_progress); + void write_hanging_text (ReelWriter& reel); + void calculate_digests (); - /** our Film */ - boost::shared_ptr _film; - boost::weak_ptr _job; + std::weak_ptr _job; std::vector _reels; std::vector::iterator _audio_reel; - std::vector::iterator _text_reel[TEXT_COUNT]; + std::vector::iterator _subtitle_reel; + std::map::iterator> _caption_reels; + std::vector::iterator _atmos_reel; - /** our thread, or 0 */ - boost::thread* _thread; + /** our thread */ + boost::thread _thread; /** true if our thread should finish */ - bool _finish; + bool _finish = false; /** queue of things to write to disk */ std::list _queue; /** number of FULL frames whose JPEG200 data is currently held in RAM */ - int _queued_full_in_memory; + int _queued_full_in_memory = 0; /** mutex for thread state */ mutable boost::mutex _state_mutex; /** condition to manage thread wakeups when we have nothing to do */ @@ -146,20 +159,60 @@ private: int _maximum_frames_in_memory; unsigned int _maximum_queue_size; + class LastWritten + { + public: + LastWritten() + : _frame(-1) + , _eyes(Eyes::RIGHT) + {} + + /** @return true if qi is the next item after this one */ + bool next (QueueItem qi) const; + void update (QueueItem qi); + + int frame () const { + return _frame; + } + + private: + int _frame; + Eyes _eyes; + }; + + /** The last frame written to each reel */ + std::vector _last_written; + /** number of FULL written frames */ - int _full_written; + int _full_written = 0; /** number of FAKE written frames */ - int _fake_written; - int _repeat_written; + int _fake_written = 0; + int _repeat_written = 0; /** number of frames pushed to disk and then recovered due to the limit of frames to be held in memory. */ - int _pushed_to_disk; + int _pushed_to_disk = 0; + + bool _text_only; boost::mutex _digest_progresses_mutex; std::map _digest_progresses; std::list _reel_assets; - std::list > _fonts; + std::vector _fonts; + + /** true if any reel has any subtitles */ + bool _have_subtitles = false; + /** all closed caption tracks that we have on any reel */ + std::set _have_closed_captions; + + struct HangingText { + PlayerText text; + TextType type; + boost::optional track; + dcpomatic::DCPTimePeriod period; + }; + + std::vector _hanging_texts; };