std::shared_ptr
[dcpomatic.git] / src / tools / dcpomatic_cli.cc
index 14b08d299621921cf93961e66ae6cfd68759e8ec..2d70fb578cd1b0629e904a1fb65702b3fa5c6b6f 100644 (file)
@@ -1,19 +1,20 @@
 /*
-    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2017 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
 #include "lib/config.h"
 #include "lib/log.h"
 #include "lib/signal_manager.h"
-#include "lib/server_finder.h"
+#include "lib/encode_server_finder.h"
 #include "lib/json_server.h"
 #include "lib/ratio.h"
 #include "lib/video_content.h"
 #include "lib/audio_content.h"
+#include "lib/dcpomatic_log.h"
 #include <dcp/version.h>
 #include <boost/foreach.hpp>
 #include <getopt.h>
@@ -45,23 +47,28 @@ using std::vector;
 using std::pair;
 using std::setw;
 using std::list;
-using boost::shared_ptr;
+using std::shared_ptr;
 using boost::optional;
-using boost::dynamic_pointer_cast;
+using std::dynamic_pointer_cast;
 
 static void
 help (string n)
 {
        cerr << "Syntax: " << n << " [OPTION] [<FILM>]\n"
-            << "  -v, --version      show DCP-o-matic version\n"
-            << "  -h, --help         show this help\n"
-            << "  -f, --flags        show flags passed to C++ compiler on build\n"
-            << "  -n, --no-progress  do not print progress to stdout\n"
-            << "  -r, --no-remote    do not use any remote servers\n"
-            << "  -j, --json <port>  run a JSON server on the specified port\n"
-            << "  -k, --keep-going   keep running even when the job is complete\n"
-            << "  -s, --servers      just display a list of encoding servers that DCP-o-matic is configured to use; don't encode\n"
-            << "      --dump         just dump a summary of the film's settings; don't encode\n"
+            << "  -v, --version        show DCP-o-matic version\n"
+            << "  -h, --help           show this help\n"
+            << "  -f, --flags          show flags passed to C++ compiler on build\n"
+            << "  -n, --no-progress    do not print progress to stdout\n"
+            << "  -r, --no-remote      do not use any remote servers\n"
+            << "  -t, --threads        specify number of local encoding threads (overriding configuration)\n"
+            << "  -j, --json <port>    run a JSON server on the specified port\n"
+            << "  -k, --keep-going     keep running even when the job is complete\n"
+            << "  -s, --servers <file> specify servers to use in a text file\n"
+            << "  -l, --list-servers   just display a list of encoding servers that DCP-o-matic is configured to use; don't encode\n"
+            << "  -d, --dcp-path       echo DCP's path to stdout on successful completion (implies -n)\n"
+            << "  -c, --config <dir>   directory containing config.xml and cinemas.xml\n"
+            << "      --dump           just dump a summary of the film's settings; don't encode\n"
+            << "      --no-check       don't check project's content files for changes before making the DCP\n"
             << "\n"
             << "<FILM> is the film directory.\n";
 }
@@ -70,7 +77,7 @@ static void
 print_dump (shared_ptr<Film> film)
 {
        cout << film->dcp_name (true) << "\n"
-            << film->container()->nickname() << " at " << ((film->resolution() == RESOLUTION_2K) ? "2K" : "4K") << "\n"
+            << film->container()->container_nickname() << " at " << ((film->resolution() == RESOLUTION_2K) ? "2K" : "4K") << "\n"
             << (film->j2k_bandwidth() / 1000000) << "Mbit/s" << "\n"
             << "Output " << film->video_frame_rate() << "fps " << (film->three_d() ? "3D" : "2D") << " " << (film->audio_frame_rate() / 1000) << "kHz\n"
             << (film->interop() ? "Inter-Op" : "SMPTE") << " " << (film->encrypted() ? "encrypted" : "unencrypted") << "\n";
@@ -79,23 +86,24 @@ print_dump (shared_ptr<Film> film)
                cout << "\n"
                     << c->path(0) << "\n"
                     << "\tat " << c->position().seconds ()
-                    << " length " << c->full_length().seconds ()
+                    << " length " << c->full_length(film).seconds ()
                     << " start trim " << c->trim_start().seconds ()
                     << " end trim " << c->trim_end().seconds () << "\n";
 
-               shared_ptr<VideoContent> video = dynamic_pointer_cast<VideoContent> (c);
-               if (video) {
-                       cout << "\t" << video->video_size().width << "x" << video->video_size().height << "\n"
-                            << "\t" << video->video_frame_rate() << "fps\n"
-                            << "\tcrop left " << video->left_crop()
-                            << " right " << video->right_crop()
-                            << " top " << video->top_crop()
-                            << " bottom " << video->bottom_crop() << "\n"
-                            << "\tscale " << video->scale().name() << "\n";
-                       if (video->colour_conversion()) {
-                               if (video->colour_conversion().get().preset()) {
+               if (c->video) {
+                       cout << "\t" << c->video->size().width << "x" << c->video->size().height << "\n"
+                            << "\t" << c->active_video_frame_rate(film) << "fps\n"
+                            << "\tcrop left " << c->video->left_crop()
+                            << " right " << c->video->right_crop()
+                            << " top " << c->video->top_crop()
+                            << " bottom " << c->video->bottom_crop() << "\n";
+                       if (c->video->custom_ratio()) {
+                               cout << "\tscale to custom ratio " << *c->video->custom_ratio() << ":1\n";
+                       }
+                       if (c->video->colour_conversion()) {
+                               if (c->video->colour_conversion().get().preset()) {
                                        cout << "\tcolour conversion "
-                                            << PresetColourConversion::all()[video->colour_conversion().get().preset().get()].name
+                                            << PresetColourConversion::all()[c->video->colour_conversion().get().preset().get()].name
                                             << "\n";
                                } else {
                                        cout << "\tcustom colour conversion\n";
@@ -106,61 +114,71 @@ print_dump (shared_ptr<Film> film)
 
                }
 
-               shared_ptr<AudioContent> audio = dynamic_pointer_cast<AudioContent> (c);
-               if (audio) {
-                       cout << "\t" << audio->audio_delay() << " delay\n"
-                            << "\t" << audio->audio_gain() << " gain\n";
+               if (c->audio) {
+                       cout << "\t" << c->audio->delay() << " delay\n"
+                            << "\t" << c->audio->gain() << " gain\n";
                }
        }
 }
 
 static void
-show_servers ()
+list_servers ()
 {
        while (true) {
                int N = 0;
-               list<ServerDescription> servers = ServerFinder::instance()->servers ();
+               list<EncodeServerDescription> servers = EncodeServerFinder::instance()->servers();
 
-               if (Config::instance()->use_any_servers ()) {
-                       if (servers.empty ()) {
-                               cout << "No encoding servers found.\n";
-                               ++N;
-                       } else {
-                               cout << std::left << setw(24) << "Host" << " Threads\n";
-                               ++N;
-                               BOOST_FOREACH (ServerDescription const & i, servers) {
-                                       cout << std::left << setw(24) << i.host_name() << " " << i.threads() << "\n";
-                                       ++N;
-                               }
-                       }
+               /* This is a bit fiddly because we want to list configured servers that are down as well
+                  as all those (configured and found by broadcast) that are up.
+               */
+
+               if (servers.empty() && Config::instance()->servers().empty()) {
+                       cout << "No encoding servers found or configured.\n";
+                       ++N;
                } else {
-                       vector<string> configured_servers = Config::instance()->servers();
-                       if (configured_servers.empty()) {
-                               cout << "No configured servers, and DCP-o-matic is not set to search for any server.\n";
-                               ++N;
-                       } else {
-                               cout << std::left << setw(24) << "Host" << " Status Threads\n";
-                               ++N;
-                               BOOST_FOREACH (string const & i, Config::instance()->servers()) {
-                                       cout << std::left << setw(24) << i << " ";
-                                       optional<int> threads;
-                                       BOOST_FOREACH (ServerDescription const & j, servers) {
-                                               if (i == j.host_name()) {
-                                                       threads = j.threads();
-                                               }
-                                       }
-                                       if (static_cast<bool>(threads)) {
-                                               cout << "UP     " << threads.get() << "\n";
+                       cout << std::left << setw(24) << "Host" << " Status Threads\n";
+                       ++N;
+
+                       /* Report the state of configured servers */
+                       BOOST_FOREACH (string i, Config::instance()->servers()) {
+                               cout << std::left << setw(24) << i << " ";
+
+                               /* See if this server is on the active list; if so, remove it and note
+                                  the number of threads it is offering.
+                               */
+                               optional<int> threads;
+                               list<EncodeServerDescription>::iterator j = servers.begin ();
+                               while (j != servers.end ()) {
+                                       if (i == j->host_name() && j->current_link_version()) {
+                                               threads = j->threads();
+                                               list<EncodeServerDescription>::iterator tmp = j;
+                                               ++tmp;
+                                               servers.erase (j);
+                                               j = tmp;
                                        } else {
-                                               cout << "DOWN\n";
+                                               ++j;
                                        }
-                                       ++N;
                                }
+                               if (static_cast<bool>(threads)) {
+                                       cout << "UP     " << threads.get() << "\n";
+                               } else {
+                                       cout << "DOWN\n";
+                               }
+                               ++N;
                        }
-               }
 
+                       /* Now report any left that have been found by broadcast */
+                       BOOST_FOREACH (EncodeServerDescription const & i, servers) {
+                               if (i.current_link_version()) {
+                                       cout << std::left << setw(24) << i.host_name() << " UP     " << i.threads() << "\n";
+                               } else {
+                                       cout << std::left << setw(24) << i.host_name() << " bad version\n";
+                               }
+                               ++N;
+                       }
+               }
 
-               dcpomatic_sleep (1);
+               dcpomatic_sleep_seconds (1);
 
                for (int i = 0; i < N; ++i) {
                        cout << "\033[1A\033[2K";
@@ -172,13 +190,18 @@ show_servers ()
 int
 main (int argc, char* argv[])
 {
-       string film_dir;
+       boost::filesystem::path film_dir;
        bool progress = true;
        bool no_remote = false;
+       optional<int> threads;
        optional<int> json_port;
        bool keep_going = false;
        bool dump = false;
-       bool servers = false;
+       optional<boost::filesystem::path> servers;
+       bool list_servers_ = false;
+       bool dcp_path = false;
+       optional<boost::filesystem::path> config;
+       bool check = true;
 
        int option_index = 0;
        while (true) {
@@ -188,15 +211,20 @@ main (int argc, char* argv[])
                        { "flags", no_argument, 0, 'f'},
                        { "no-progress", no_argument, 0, 'n'},
                        { "no-remote", no_argument, 0, 'r'},
+                       { "threads", required_argument, 0, 't'},
                        { "json", required_argument, 0, 'j'},
                        { "keep-going", no_argument, 0, 'k' },
-                       { "servers", no_argument, 0, 's' },
+                       { "servers", required_argument, 0, 's' },
+                       { "list-servers", no_argument, 0, 'l' },
+                       { "dcp-path", no_argument, 0, 'd' },
+                       { "config", required_argument, 0, 'c' },
                        /* Just using A, B, C ... from here on */
                        { "dump", no_argument, 0, 'A' },
+                       { "no-check", no_argument, 0, 'B' },
                        { 0, 0, 0, 0 }
                };
 
-               int c = getopt_long (argc, argv, "vhfnrj:kAs", long_options, &option_index);
+               int c = getopt_long (argc, argv, "vhfnrt:j:kAs:ldc:B", long_options, &option_index);
 
                if (c == -1) {
                        break;
@@ -218,6 +246,9 @@ main (int argc, char* argv[])
                case 'r':
                        no_remote = true;
                        break;
+               case 't':
+                       threads = atoi (optarg);
+                       break;
                case 'j':
                        json_port = atoi (optarg);
                        break;
@@ -228,13 +259,47 @@ main (int argc, char* argv[])
                        dump = true;
                        break;
                case 's':
-                       servers = true;
+                       servers = optarg;
+                       break;
+               case 'l':
+                       list_servers_ = true;
+                       break;
+               case 'd':
+                       dcp_path = true;
+                       progress = false;
+                       break;
+               case 'c':
+                       config = optarg;
+                       break;
+               case 'B':
+                       check = false;
                        break;
                }
        }
 
+       if (config) {
+               State::override_path = *config;
+       }
+
        if (servers) {
-               show_servers ();
+               FILE* f = fopen_boost (*servers, "r");
+               if (!f) {
+                       cerr << "Could not open servers list file " << *servers << "\n";
+                       exit (EXIT_FAILURE);
+               }
+               vector<string> servers;
+               while (!feof (f)) {
+                       char buffer[128];
+                       if (fscanf (f, "%s.127", buffer) == 1) {
+                               servers.push_back (buffer);
+                       }
+               }
+               fclose (f);
+               Config::instance()->set_servers (servers);
+       }
+
+       if (list_servers_) {
+               list_servers ();
                exit (EXIT_SUCCESS);
        }
 
@@ -250,19 +315,23 @@ main (int argc, char* argv[])
        signal_manager = new SignalManager ();
 
        if (no_remote) {
-               ServerFinder::instance()->disable ();
+               EncodeServerFinder::instance()->stop ();
        }
 
        if (json_port) {
                new JSONServer (json_port.get ());
        }
 
+       if (threads) {
+               Config::instance()->set_master_encoding_threads (threads.get ());
+       }
+
        shared_ptr<Film> film;
        try {
                film.reset (new Film (film_dir));
                film->read_metadata ();
        } catch (std::exception& e) {
-               cerr << argv[0] << ": error reading film `" << film_dir << "' (" << e.what() << ")\n";
+               cerr << argv[0] << ": error reading film `" << film_dir.string() << "' (" << e.what() << ")\n";
                exit (EXIT_FAILURE);
        }
 
@@ -271,6 +340,8 @@ main (int argc, char* argv[])
                exit (EXIT_SUCCESS);
        }
 
+       dcpomatic_log = film->log ();
+
        ContentList content = film->content ();
        for (ContentList::const_iterator i = content.begin(); i != content.end(); ++i) {
                vector<boost::filesystem::path> paths = (*i)->paths ();
@@ -286,65 +357,12 @@ main (int argc, char* argv[])
                cout << "\nMaking DCP for " << film->name() << "\n";
        }
 
-       film->make_dcp ();
-
-       bool should_stop = false;
-       bool first = true;
-       bool error = false;
-       while (!should_stop) {
-
-               dcpomatic_sleep (5);
-
-               list<shared_ptr<Job> > jobs = JobManager::instance()->get ();
-
-               if (!first && progress) {
-                       for (size_t i = 0; i < jobs.size(); ++i) {
-                               cout << "\033[1A\033[2K";
-                       }
-                       cout.flush ();
-               }
-
-               first = false;
-
-               int unfinished = 0;
-               int finished_in_error = 0;
-
-               for (list<shared_ptr<Job> >::iterator i = jobs.begin(); i != jobs.end(); ++i) {
-                       if (progress) {
-                               cout << (*i)->name() << ": ";
-
-                               if ((*i)->progress ()) {
-                                       cout << (*i)->status() << "                         \n";
-                               } else {
-                                       cout << ": Running           \n";
-                               }
-                       }
-
-                       if (!(*i)->finished ()) {
-                               ++unfinished;
-                       }
-
-                       if ((*i)->finished_in_error ()) {
-                               ++finished_in_error;
-                               error = true;
-                       }
-
-                       if (!progress && (*i)->finished_in_error ()) {
-                               /* We won't see this error if we haven't been showing progress,
-                                  so show it now.
-                               */
-                               cout << (*i)->status() << "\n";
-                       }
-               }
-
-               if (unfinished == 0 || finished_in_error != 0) {
-                       should_stop = true;
-               }
-       }
+       film->make_dcp (false, check);
+       bool const error = show_jobs_on_console (progress);
 
        if (keep_going) {
                while (true) {
-                       dcpomatic_sleep (3600);
+                       dcpomatic_sleep_seconds (3600);
                }
        }
 
@@ -353,7 +371,11 @@ main (int argc, char* argv[])
        */
        JobManager::drop ();
 
-       ServerFinder::drop ();
+       EncodeServerFinder::drop ();
+
+       if (dcp_path && !error) {
+               cout << film->dir (film->dcp_name (false)).string() << "\n";
+       }
 
        return error ? EXIT_FAILURE : EXIT_SUCCESS;
 }