Fix DCP content font ID allocation to cope with DCPs that have multiple fonts
[dcpomatic.git] / src / lib / writer.cc
index 9e998a8307c1d9a4937343b88761a99bbd2320d8..9ab3d4e1edaf3be63d2f335b16677b96f09685b4 100644 (file)
@@ -72,12 +72,12 @@ using dcp::ArrayData;
 using namespace dcpomatic;
 
 
-/** @param j Job to report progress to, or 0.
+/** @param weak_job Job to report progress to, or 0.
  *  @param text_only true to enable only the text (subtitle/ccap) parts of the writer.
  */
-Writer::Writer (weak_ptr<const Film> weak_film, weak_ptr<Job> j, bool text_only)
+Writer::Writer(weak_ptr<const Film> weak_film, weak_ptr<Job> weak_job, bool text_only)
        : WeakConstFilm (weak_film)
-       , _job (j)
+       , _job(weak_job)
        /* These will be reset to sensible values when J2KEncoder is created */
        , _maximum_frames_in_memory (8)
        , _maximum_queue_size (8)
@@ -454,12 +454,12 @@ try
 
                        /* Find one from the back of the queue */
                        _queue.sort ();
-                       auto i = _queue.rbegin ();
-                       while (i != _queue.rend() && (i->type != QueueItem::Type::FULL || !i->encoded)) {
-                               ++i;
+                       auto item = _queue.rbegin();
+                       while (item != _queue.rend() && (item->type != QueueItem::Type::FULL || !item->encoded)) {
+                               ++item;
                        }
 
-                       DCPOMATIC_ASSERT (i != _queue.rend());
+                       DCPOMATIC_ASSERT(item != _queue.rend());
                        ++_pushed_to_disk;
                        /* For the log message below */
                        int const awaiting = _last_written[_queue.front().reel].frame() + 1;
@@ -470,15 +470,15 @@ try
                           thread could erase the last item in the list.
                        */
 
-                       LOG_GENERAL ("Writer full; pushes %1 to disk while awaiting %2", i->frame, awaiting);
+                       LOG_GENERAL("Writer full; pushes %1 to disk while awaiting %2", item->frame, awaiting);
 
-                       i->encoded->write_via_temp (
-                               film()->j2c_path(i->reel, i->frame, i->eyes, true),
-                               film()->j2c_path(i->reel, i->frame, i->eyes, false)
+                       item->encoded->write_via_temp(
+                               film()->j2c_path(item->reel, item->frame, item->eyes, true),
+                               film()->j2c_path(item->reel, item->frame, item->eyes, false)
                                );
 
                        lock.lock ();
-                       i->encoded.reset ();
+                       item->encoded.reset();
                        --_queued_full_in_memory;
                        _full_condition.notify_all ();
                }
@@ -572,9 +572,9 @@ Writer::finish (boost::filesystem::path output_dcp)
 
        LOG_GENERAL_NC ("Finishing ReelWriters");
 
-       for (auto& i: _reels) {
-               write_hanging_text (i);
-               i.finish (output_dcp);
+       for (auto& reel: _reels) {
+               write_hanging_text(reel);
+               reel.finish(output_dcp);
        }
 
        LOG_GENERAL_NC ("Writing XML");
@@ -616,7 +616,10 @@ Writer::finish (boost::filesystem::path output_dcp)
 
        vector<dcp::ContentVersion> cv;
        for (auto i: film()->content_versions()) {
-               cv.push_back (dcp::ContentVersion(i));
+               /* Make sure we don't end up writing an empty <LabelText> node as some validators
+                * complain about that.
+                */
+               cv.push_back(!i.empty() ? dcp::ContentVersion(i) : dcp::ContentVersion("1"));
        }
        if (cv.empty()) {
                cv = { dcp::ContentVersion("1") };
@@ -653,9 +656,12 @@ Writer::finish (boost::filesystem::path output_dcp)
                field = dcp::MCASoundField::FIVE_POINT_ONE;
        }
 
-       dcp::MainSoundConfiguration msc(field, MAX_DCP_AUDIO_CHANNELS);
+       auto const audio_channels = film()->audio_channels();
+       dcp::MainSoundConfiguration msc(field, audio_channels);
        for (auto i: film()->mapped_audio_channels()) {
-               msc.set_mapping(i, static_cast<dcp::Channel>(i));
+               if (i < audio_channels) {
+                       msc.set_mapping(i, static_cast<dcp::Channel>(i));
+               }
        }
 
        cpl->set_main_sound_configuration(msc);
@@ -698,8 +704,8 @@ void
 Writer::write_cover_sheet (boost::filesystem::path output_dcp)
 {
        auto const cover = film()->file("COVER_SHEET.txt");
-       dcp::File f(cover, "w");
-       if (!f) {
+       dcp::File file(cover, "w");
+       if (!file) {
                throw OpenFileError (cover, errno, OpenFileError::WRITE);
        }
 
@@ -728,11 +734,11 @@ Writer::write_cover_sheet (boost::filesystem::path output_dcp)
 
        boost::uintmax_t size = 0;
        for (
-               auto i = boost::filesystem::recursive_directory_iterator(output_dcp);
-               i != boost::filesystem::recursive_directory_iterator();
+               auto i = dcp::filesystem::recursive_directory_iterator(output_dcp);
+               i != dcp::filesystem::recursive_directory_iterator();
                ++i) {
-               if (boost::filesystem::is_regular_file (i->path())) {
-                       size += boost::filesystem::file_size (i->path());
+               if (dcp::filesystem::is_regular_file(i->path())) {
+                       size += dcp::filesystem::file_size(i->path());
                }
        }
 
@@ -766,7 +772,7 @@ Writer::write_cover_sheet (boost::filesystem::path output_dcp)
 
        boost::algorithm::replace_all (text, "$LENGTH", length);
 
-       f.checked_write(text.c_str(), text.length());
+       file.checked_write(text.c_str(), text.length());
 }
 
 
@@ -869,53 +875,9 @@ Writer::write (vector<shared_ptr<Font>> fonts)
                }
                _chosen_interop_font = fonts[0];
        } else {
-               set<string> used_ids;
-
-               /* Return the index of a _N at the end of a string, or string::npos */
-               auto underscore_number_position = [](string s) {
-                       auto last_underscore = s.find_last_of("_");
-                       if (last_underscore == string::npos) {
-                               return string::npos;
-                       }
-
-                       for (auto i = last_underscore + 1; i < s.size(); ++i) {
-                               if (!isdigit(s[i])) {
-                                       return string::npos;
-                               }
-                       }
-
-                       return last_underscore;
-               };
-
-               /* Write fonts to _fonts, changing any duplicate IDs so that they are unique */
                for (auto font: fonts) {
-                       auto id = fix_id(font->id());
-                       if (used_ids.find(id) == used_ids.end()) {
-                               /* This ID is unique so we can just use it as-is */
-                               _fonts.put(font, id);
-                               used_ids.insert(id);
-                       } else {
-                               auto end = underscore_number_position(id);
-                               if (end == string::npos) {
-                                       /* This string has no _N suffix, so add one */
-                                       id += "_0";
-                                       end = underscore_number_position(id);
-                               }
-
-                               ++end;
-
-                               /* Increment the suffix until we find a unique one */
-                               auto number = dcp::raw_convert<int>(id.substr(end));
-                               while (used_ids.find(id) != used_ids.end()) {
-                                       ++number;
-                                       id = String::compose("%1_%2", id.substr(0, end - 1), number);
-                               }
-                               used_ids.insert(id);
-                       }
-                       _fonts.put(font, id);
+                       _fonts.put(font, fix_id(font->id()));
                }
-
-               DCPOMATIC_ASSERT(_fonts.map().size() == used_ids.size());
        }
 }
 
@@ -962,13 +924,13 @@ size_t
 Writer::video_reel (int frame) const
 {
        auto t = DCPTime::from_frames (frame, film()->video_frame_rate());
-       size_t i = 0;
-       while (i < _reels.size() && !_reels[i].period().contains (t)) {
-               ++i;
+       size_t reel_index = 0;
+       while (reel_index < _reels.size() && !_reels[reel_index].period().contains(t)) {
+               ++reel_index;
        }
 
-       DCPOMATIC_ASSERT (i < _reels.size ());
-       return i;
+       DCPOMATIC_ASSERT(reel_index < _reels.size ());
+       return reel_index;
 }