Replace CinemaKDMs class with a list<KDMWithMetadata>
[dcpomatic.git] / src / lib / cinema_kdms.cc
index 99d1511520b1bbb3d1e1a73b5d01a6190c7ac07c..94e83f8077de30baf735722c2dd696a4c2e3d608 100644 (file)
@@ -29,6 +29,7 @@
 #include "log.h"
 #include "zipper.h"
 #include "dcpomatic_log.h"
+#include "kdm_with_metadata.h"
 #include <boost/foreach.hpp>
 
 #include "i18n.h"
@@ -39,16 +40,14 @@ using std::string;
 using std::runtime_error;
 using boost::shared_ptr;
 using boost::function;
+using boost::optional;
 
 void
-CinemaKDMs::make_zip_file (boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values) const
+make_zip_file (list<KDMWithMetadataPtr> kdms, boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values)
 {
        Zipper zipper (zip_file);
 
-       name_values['c'] = cinema->name;
-
-       BOOST_FOREACH (shared_ptr<KDMWithMetadata> i, screen_kdms) {
-               name_values['s'] = i->screen->name;
+       BOOST_FOREACH (KDMWithMetadataPtr i, kdms) {
                name_values['i'] = i->kdm_id ();
                string const name = careful_string_filter(name_format.get(name_values, ".xml"));
                zipper.add (name, i->kdm_as_xml());
@@ -57,31 +56,30 @@ CinemaKDMs::make_zip_file (boost::filesystem::path zip_file, dcp::NameFormat nam
        zipper.close ();
 }
 
-/** Collect a list of KDMWithMetadatas into a list of CinemaKDMs so that each
+/** Collect a list of KDMWithMetadatas into a list of list<KDMWithMetadata> so that each
  *  CinemaKDM contains the KDMs for its cinema.
  */
-list<CinemaKDMs>
-CinemaKDMs::collect (list<shared_ptr<KDMWithMetadata> > screen_kdms)
+list<list<KDMWithMetadataPtr> >
+collect (list<KDMWithMetadataPtr> screen_kdms)
 {
-       list<CinemaKDMs> cinema_kdms;
+       list<list<KDMWithMetadataPtr> > cinema_kdms;
 
        while (!screen_kdms.empty ()) {
 
                /* Get all the screens from a single cinema */
 
-               CinemaKDMs ck;
+               list<KDMWithMetadataPtr> ck;
 
-               list<shared_ptr<KDMWithMetadata> >::iterator i = screen_kdms.begin ();
-               ck.cinema = (*i)->screen->cinema;
-               ck.screen_kdms.push_back (*i);
-               list<shared_ptr<KDMWithMetadata> >::iterator j = i;
+               list<KDMWithMetadataPtr>::iterator i = screen_kdms.begin ();
+               ck.push_back (*i);
+               list<KDMWithMetadataPtr>::iterator j = i;
                ++i;
                screen_kdms.remove (*j);
 
                while (i != screen_kdms.end ()) {
-                       if ((*i)->screen->cinema == ck.cinema) {
-                               ck.screen_kdms.push_back (*i);
-                               list<shared_ptr<KDMWithMetadata> >::iterator j = i;
+                       if ((*i)->cinema() == ck.front()->cinema()) {
+                               ck.push_back (*i);
+                               list<KDMWithMetadataPtr>::iterator j = i;
                                ++i;
                                screen_kdms.remove (*j);
                        } else {
@@ -97,8 +95,8 @@ CinemaKDMs::collect (list<shared_ptr<KDMWithMetadata> > screen_kdms)
 
 /** Write one directory per cinema into another directory */
 int
-CinemaKDMs::write_directories (
-       list<CinemaKDMs> cinema_kdms,
+write_directories (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        boost::filesystem::path directory,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
@@ -111,15 +109,14 @@ CinemaKDMs::write_directories (
 
        int written = 0;
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
                boost::filesystem::path path = directory;
-               name_values['c'] = i.cinema->name;
                path /= container_name_format.get(name_values, "");
                if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
                        boost::filesystem::create_directories (path);
-                       KDMWithMetadata::write_files (i.screen_kdms, path, filename_format, name_values, confirm_overwrite);
+                       write_files (i, path, filename_format, name_values, confirm_overwrite);
                }
-               written += i.screen_kdms.size();
+               written += i.size();
        }
 
        return written;
@@ -127,8 +124,8 @@ CinemaKDMs::write_directories (
 
 /** Write one ZIP file per cinema into a directory */
 int
-CinemaKDMs::write_zip_files (
-       list<CinemaKDMs> cinema_kdms,
+write_zip_files (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        boost::filesystem::path directory,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
@@ -141,17 +138,16 @@ CinemaKDMs::write_zip_files (
 
        int written = 0;
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
                boost::filesystem::path path = directory;
-               name_values['c'] = i.cinema->name;
                path /= container_name_format.get(name_values, ".zip");
                if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
                        if (boost::filesystem::exists (path)) {
                                /* Creating a new zip file over an existing one is an error */
                                boost::filesystem::remove (path);
                        }
-                       i.make_zip_file (path, filename_format, name_values);
-                       written += i.screen_kdms.size();
+                       make_zip_file (i, path, filename_format, name_values);
+                       written += i.size();
                }
        }
 
@@ -166,8 +162,8 @@ CinemaKDMs::write_zip_files (
  *  @param cpl_name Name of the CPL that the KDMs are for.
  */
 void
-CinemaKDMs::email (
-       list<CinemaKDMs> cinema_kdms,
+email (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
        dcp::NameFormat::Map name_values,
@@ -183,38 +179,39 @@ CinemaKDMs::email (
        /* No specific screen */
        name_values['s'] = "";
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
 
-               if (i.cinema->emails.empty()) {
+               if (i.front()->cinema()->emails.empty()) {
                        continue;
                }
 
-               name_values['c'] = i.cinema->name;
-
                boost::filesystem::path zip_file = boost::filesystem::temp_directory_path() / boost::filesystem::unique_path();
                boost::filesystem::create_directories (zip_file);
                zip_file /= container_name_format.get(name_values, ".zip");
-               i.make_zip_file (zip_file, filename_format, name_values);
+               make_zip_file (i, zip_file, filename_format, name_values);
 
                string subject = config->kdm_subject();
                boost::algorithm::replace_all (subject, "$CPL_NAME", cpl_name);
                boost::algorithm::replace_all (subject, "$START_TIME", name_values['b']);
                boost::algorithm::replace_all (subject, "$END_TIME", name_values['e']);
-               boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.cinema->name);
+               boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.front()->cinema()->name);
 
                string body = config->kdm_email().c_str();
                boost::algorithm::replace_all (body, "$CPL_NAME", cpl_name);
                boost::algorithm::replace_all (body, "$START_TIME", name_values['b']);
                boost::algorithm::replace_all (body, "$END_TIME", name_values['e']);
-               boost::algorithm::replace_all (body, "$CINEMA_NAME", i.cinema->name);
+               boost::algorithm::replace_all (body, "$CINEMA_NAME", i.front()->cinema()->name);
 
                string screens;
-               BOOST_FOREACH (shared_ptr<KDMWithMetadata> j, i.screen_kdms) {
-                       screens += j->screen->name + ", ";
+               BOOST_FOREACH (KDMWithMetadataPtr j, i) {
+                       optional<string> screen_name = j->get('n');
+                       if (screen_name) {
+                               screens += *screen_name + ", ";
+                       }
                }
                boost::algorithm::replace_all (body, "$SCREENS", screens.substr (0, screens.length() - 2));
 
-               Emailer email (config->kdm_from(), i.cinema->emails, subject, body);
+               Emailer email (config->kdm_from(), i.front()->cinema()->emails, subject, body);
 
                BOOST_FOREACH (string i, config->kdm_cc()) {
                        email.add_cc (i);