Rename video/audio/subtitle part methods.
[dcpomatic.git] / src / lib / ffmpeg_content.cc
index f2fe06f36aca3b182b892fc719654aeda96bcb82..d9941685d08d43cbbf6eb523c34d8049aeaeb2af 100644 (file)
@@ -18,6 +18,8 @@
 */
 
 #include "ffmpeg_content.h"
+#include "video_content.h"
+#include "audio_content.h"
 #include "ffmpeg_examiner.h"
 #include "ffmpeg_subtitle_stream.h"
 #include "ffmpeg_audio_stream.h"
@@ -31,6 +33,7 @@
 #include "frame_rate_change.h"
 #include "safe_stringstream.h"
 #include "raw_convert.h"
+#include "subtitle_content.h"
 #include <libcxml/cxml.h>
 extern "C" {
 #include <libavformat/avformat.h>
@@ -60,19 +63,21 @@ int const FFmpegContentProperty::FILTERS = 102;
 
 FFmpegContent::FFmpegContent (shared_ptr<const Film> film, boost::filesystem::path p)
        : Content (film, p)
-       , VideoContent (film, p)
-       , AudioContent (film, p)
-       , SubtitleContent (film, p)
 {
+       video.reset (new VideoContent (this, film));
+       audio.reset (new AudioContent (this, film));
+       subtitle.reset (new SubtitleContent (this, film));
 
+       set_default_colour_conversion ();
 }
 
 FFmpegContent::FFmpegContent (shared_ptr<const Film> film, cxml::ConstNodePtr node, int version, list<string>& notes)
        : Content (film, node)
-       , VideoContent (film, node, version)
-       , AudioContent (film, node)
-       , SubtitleContent (film, node, version)
 {
+       video.reset (new VideoContent (this, film, node, version));
+       audio.reset (new AudioContent (this, film, node));
+       subtitle.reset (new SubtitleContent (this, film, node, version));
+
        list<cxml::NodePtr> c = node->node_children ("SubtitleStream");
        for (list<cxml::NodePtr>::const_iterator i = c.begin(); i != c.end(); ++i) {
                _subtitle_streams.push_back (shared_ptr<FFmpegSubtitleStream> (new FFmpegSubtitleStream (*i, version)));
@@ -83,10 +88,11 @@ FFmpegContent::FFmpegContent (shared_ptr<const Film> film, cxml::ConstNodePtr no
 
        c = node->node_children ("AudioStream");
        for (list<cxml::NodePtr>::const_iterator i = c.begin(); i != c.end(); ++i) {
-               _audio_streams.push_back (shared_ptr<FFmpegAudioStream> (new FFmpegAudioStream (*i, version)));
+               shared_ptr<FFmpegAudioStream> as (new FFmpegAudioStream (*i, version));
+               audio->add_stream (as);
                if (version < 11 && !(*i)->optional_node_child ("Selected")) {
                        /* This is an old file and this stream is not selected, so un-map it */
-                       _audio_streams.back()->set_mapping (AudioMapping (_audio_streams.back()->channels (), MAX_DCP_AUDIO_CHANNELS));
+                       as->set_mapping (AudioMapping (_audio_streams.back()->channels (), MAX_DCP_AUDIO_CHANNELS));
                }
        }
 
@@ -117,16 +123,17 @@ FFmpegContent::FFmpegContent (shared_ptr<const Film> film, cxml::ConstNodePtr no
 
 FFmpegContent::FFmpegContent (shared_ptr<const Film> film, vector<boost::shared_ptr<Content> > c)
        : Content (film, c)
-       , VideoContent (film, c)
-       , AudioContent (film, c)
-       , SubtitleContent (film, c)
 {
+       video.reset (new VideoContent (this, film, c));
+       audio.reset (new AudioContent (this, film, c));
+       subtitle.reset (new SubtitleContent (this, film, c));
+
        shared_ptr<FFmpegContent> ref = dynamic_pointer_cast<FFmpegContent> (c[0]);
        DCPOMATIC_ASSERT (ref);
 
        for (size_t i = 0; i < c.size(); ++i) {
                shared_ptr<FFmpegContent> fc = dynamic_pointer_cast<FFmpegContent> (c[i]);
-               if (fc->use_subtitles() && *(fc->_subtitle_stream.get()) != *(ref->_subtitle_stream.get())) {
+               if (fc->subtitle->use() && *(fc->_subtitle_stream.get()) != *(ref->_subtitle_stream.get())) {
                        throw JoinError (_("Content to be joined must use the same subtitle stream."));
                }
        }
@@ -135,7 +142,6 @@ FFmpegContent::FFmpegContent (shared_ptr<const Film> film, vector<boost::shared_
 
        _subtitle_streams = ref->subtitle_streams ();
        _subtitle_stream = ref->subtitle_stream ();
-       _audio_streams = ref->ffmpeg_audio_streams ();
        _first_video = ref->_first_video;
        _filters = ref->_filters;
        _color_range = ref->_color_range;
@@ -150,9 +156,9 @@ FFmpegContent::as_xml (xmlpp::Node* node) const
 {
        node->add_child("Type")->add_child_text ("FFmpeg");
        Content::as_xml (node);
-       VideoContent::as_xml (node);
-       AudioContent::as_xml (node);
-       SubtitleContent::as_xml (node);
+       video->as_xml (node);
+       audio->as_xml (node);
+       subtitle->as_xml (node);
 
        boost::mutex::scoped_lock lm (_mutex);
 
@@ -164,8 +170,10 @@ FFmpegContent::as_xml (xmlpp::Node* node) const
                (*i)->as_xml (t);
        }
 
-       for (vector<shared_ptr<FFmpegAudioStream> >::const_iterator i = _audio_streams.begin(); i != _audio_streams.end(); ++i) {
-               (*i)->as_xml (node->add_child("AudioStream"));
+       BOOST_FOREACH (AudioStreamPtr i, audio->streams ()) {
+               shared_ptr<FFmpegAudioStream> f = dynamic_pointer_cast<FFmpegAudioStream> (i);
+               DCPOMATIC_ASSERT (f);
+               f->as_xml (node->add_child("AudioStream"));
        }
 
        for (vector<Filter const *>::const_iterator i = _filters.begin(); i != _filters.end(); ++i) {
@@ -193,7 +201,8 @@ FFmpegContent::examine (shared_ptr<Job> job)
        Content::examine (job);
 
        shared_ptr<FFmpegExaminer> examiner (new FFmpegExaminer (shared_from_this (), job));
-       take_from_video_examiner (examiner);
+       video->take_from_examiner (examiner);
+       set_default_colour_conversion ();
 
        {
                boost::mutex::scoped_lock lm (_mutex);
@@ -203,12 +212,15 @@ FFmpegContent::examine (shared_ptr<Job> job)
                        _subtitle_stream = _subtitle_streams.front ();
                }
 
-               _audio_streams = examiner->audio_streams ();
+               BOOST_FOREACH (shared_ptr<FFmpegAudioStream> i, examiner->audio_streams ()) {
+                       audio->add_stream (i);
+               }
 
-               if (!_audio_streams.empty ()) {
-                       AudioMapping m = _audio_streams.front()->mapping ();
+               if (!audio->streams().empty ()) {
+                       AudioStreamPtr as = audio->streams().front();
+                       AudioMapping m = as->mapping ();
                        film()->make_audio_mapping_default (m);
-                       _audio_streams.front()->set_mapping (m);
+                       as->set_mapping (m);
                }
 
                _first_video = examiner->first_video ();
@@ -222,7 +234,6 @@ FFmpegContent::examine (shared_ptr<Job> job)
 
        signal_changed (FFmpegContentProperty::SUBTITLE_STREAMS);
        signal_changed (FFmpegContentProperty::SUBTITLE_STREAM);
-       signal_changed (AudioContentProperty::AUDIO_STREAMS);
 }
 
 string
@@ -252,8 +263,8 @@ FFmpegContent::technical_summary () const
        string filt = Filter::ffmpeg_string (_filters);
 
        return Content::technical_summary() + " - "
-               + VideoContent::technical_summary() + " - "
-               + AudioContent::technical_summary() + " - "
+               + video->technical_summary() + " - "
+               + audio->technical_summary() + " - "
                + String::compose (
                        "ffmpeg: audio %1 subtitle %2 filters %3", as, ss, filt
                        );
@@ -285,8 +296,8 @@ operator!= (FFmpegStream const & a, FFmpegStream const & b)
 DCPTime
 FFmpegContent::full_length () const
 {
-       FrameRateChange const frc (video_frame_rate (), film()->video_frame_rate ());
-       return DCPTime::from_frames (llrint (video_length_after_3d_combine() * frc.factor()), film()->video_frame_rate());
+       FrameRateChange const frc (video->frame_rate (), film()->video_frame_rate ());
+       return DCPTime::from_frames (llrint (video->length_after_3d_combine() * frc.factor()), film()->video_frame_rate());
 }
 
 void
@@ -305,8 +316,9 @@ FFmpegContent::identifier () const
 {
        SafeStringStream s;
 
-       s << VideoContent::identifier() << "_"
-         << SubtitleContent::identifier();
+       s << Content::identifier() << "_"
+         << video->identifier() << "_"
+         << subtitle->identifier();
 
        boost::mutex::scoped_lock lm (_mutex);
 
@@ -370,14 +382,14 @@ FFmpegContent::has_text_subtitles () const
 void
 FFmpegContent::set_default_colour_conversion ()
 {
-       dcp::Size const s = video_size ();
+       dcp::Size const s = video->size ();
 
        boost::mutex::scoped_lock lm (_mutex);
 
        if (s.width < 1080) {
-               _colour_conversion = PresetColourConversion::from_id ("rec601").conversion;
+               video->set_colour_conversion (PresetColourConversion::from_id ("rec601").conversion);
        } else {
-               _colour_conversion = PresetColourConversion::from_id ("rec709").conversion;
+               video->set_colour_conversion (PresetColourConversion::from_id ("rec709").conversion);
        }
 }
 
@@ -392,10 +404,11 @@ FFmpegContent::audio_streams () const
 }
 
 void
-FFmpegContent::add_properties (list<pair<string, string> >& p) const
+FFmpegContent::add_properties (list<UserProperty>& p) const
 {
-       VideoContent::add_properties (p);
-       AudioContent::add_properties (p);
+       Content::add_properties (p);
+       video->add_properties (p);
+       audio->add_properties (p);
 
        if (_bits_per_pixel) {
                int const sub = 219 * pow (2, _bits_per_pixel.get() - 8);
@@ -405,17 +418,21 @@ FFmpegContent::add_properties (list<pair<string, string> >& p) const
                case AVCOL_RANGE_UNSPECIFIED:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is unknown (not specified in the file).
-                       p.push_back (make_pair (_("Colour range"), _("Unspecified")));
+                       p.push_back (UserProperty (_("Video"), _("Colour range"), _("Unspecified")));
                        break;
                case AVCOL_RANGE_MPEG:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is limited, so that not all possible values are valid.
-                       p.push_back (make_pair (_("Colour range"), String::compose (_("Limited (%1-%2)"), (total - sub) / 2, (total + sub) / 2)));
+                       p.push_back (
+                               UserProperty (
+                                       _("Video"), _("Colour range"), String::compose (_("Limited (%1-%2)"), (total - sub) / 2, (total + sub) / 2)
+                                       )
+                               );
                        break;
                case AVCOL_RANGE_JPEG:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is full, so that all possible pixel values are valid.
-                       p.push_back (make_pair (_("Colour range"), String::compose (_("Full (0-%1)"), total)));
+                       p.push_back (UserProperty (_("Video"), _("Colour range"), String::compose (_("Full (0-%1)"), total)));
                        break;
                default:
                        DCPOMATIC_ASSERT (false);
@@ -425,17 +442,17 @@ FFmpegContent::add_properties (list<pair<string, string> >& p) const
                case AVCOL_RANGE_UNSPECIFIED:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is unknown (not specified in the file).
-                       p.push_back (make_pair (_("Colour range"), _("Unspecified")));
+                       p.push_back (UserProperty (_("Video"), _("Colour range"), _("Unspecified")));
                        break;
                case AVCOL_RANGE_MPEG:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is limited, so that not all possible values are valid.
-                       p.push_back (make_pair (_("Colour range"), _("Limited")));
+                       p.push_back (UserProperty (_("Video"), _("Colour range"), _("Limited")));
                        break;
                case AVCOL_RANGE_JPEG:
                        /// TRANSLATORS: this means that the range of pixel values used in this
                        /// file is full, so that all possible pixel values are valid.
-                       p.push_back (make_pair (_("Colour range"), _("Full")));
+                       p.push_back (UserProperty (_("Video"), _("Colour range"), _("Full")));
                        break;
                default:
                        DCPOMATIC_ASSERT (false);
@@ -452,11 +469,12 @@ FFmpegContent::add_properties (list<pair<string, string> >& p) const
                _("SMPTE 170M (BT601)"),
                _("SMPTE 240M"),
                _("Film"),
-               _("BT2020")
+               _("BT2020"),
+               _("SMPTE ST 428-1 (CIE 1931 XYZ)")
        };
 
-       DCPOMATIC_ASSERT (AVCOL_PRI_NB == 10);
-       p.push_back (make_pair (_("Colour primaries"), primaries[_color_primaries]));
+       DCPOMATIC_ASSERT (AVCOL_PRI_NB == 11);
+       p.push_back (UserProperty (_("Video"), _("Colour primaries"), primaries[_color_primaries]));
 
        char const * transfers[] = {
                _("Unspecified"),
@@ -474,11 +492,13 @@ FFmpegContent::add_properties (list<pair<string, string> >& p) const
                _("BT1361 extended colour gamut"),
                _("IEC61966-2-1 (sRGB or sYCC)"),
                _("BT2020 for a 10-bit system"),
-               _("BT2020 for a 12-bit system")
+               _("BT2020 for a 12-bit system"),
+               _("SMPTE ST 2084 for 10, 12, 14 and 16 bit systems"),
+               _("SMPTE ST 428-1")
        };
 
-       DCPOMATIC_ASSERT (AVCOL_TRC_NB == 16);
-       p.push_back (make_pair (_("Colour transfer characteristic"), transfers[_color_trc]));
+       DCPOMATIC_ASSERT (AVCOL_TRC_NB == 18);
+       p.push_back (UserProperty (_("Video"), _("Colour transfer characteristic"), transfers[_color_trc]));
 
        char const * spaces[] = {
                _("RGB / sRGB (IEC61966-2-1)"),
@@ -495,9 +515,28 @@ FFmpegContent::add_properties (list<pair<string, string> >& p) const
        };
 
        DCPOMATIC_ASSERT (AVCOL_SPC_NB == 11);
-       p.push_back (make_pair (_("Colourspace"), spaces[_colorspace]));
+       p.push_back (UserProperty (_("Video"), _("Colourspace"), spaces[_colorspace]));
 
        if (_bits_per_pixel) {
-               p.push_back (make_pair (_("Bits per pixel"), raw_convert<string> (_bits_per_pixel.get ())));
+               p.push_back (UserProperty (_("Video"), _("Bits per pixel"), raw_convert<string> (_bits_per_pixel.get ())));
+       }
+}
+
+/** Our subtitle streams have colour maps, which can be changed, but
+ *  they have no way of signalling that change.  As a hack, we have this
+ *  method which callers can use when they've modified one of our subtitle
+ *  streams.
+ */
+void
+FFmpegContent::signal_subtitle_stream_changed ()
+{
+       signal_changed (FFmpegContentProperty::SUBTITLE_STREAM);
+}
+
+void
+FFmpegContent::changed (int property)
+{
+       if (property == VideoContentProperty::FRAME_RATE && subtitle) {
+               subtitle->set_video_frame_rate (video->frame_rate ());
        }
 }