Tidy up to use one list of servers.
[dcpomatic.git] / src / tools / dcpomatic_cli.cc
index 80e10daaf08299a7483ff551266b2c0522563dce..788594f0ca6273b509ee3641e13f0d25dbb70808 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/>.
 
 */
 
@@ -53,15 +54,18 @@ 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"
+            << "      --dump           just dump a summary of the film's settings; don't encode\n"
             << "\n"
             << "<FILM> is the film directory.\n";
 }
@@ -70,7 +74,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";
@@ -83,19 +87,18 @@ print_dump (shared_ptr<Film> film)
                     << " 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() << "fps\n"
+                            << "\tcrop left " << c->video->left_crop()
+                            << " right " << c->video->right_crop()
+                            << " top " << c->video->top_crop()
+                            << " bottom " << c->video->bottom_crop() << "\n"
+                            << "\tscale " << c->video->scale().name() << "\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,59 +109,69 @@ 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<EncodeServerDescription> servers = EncodeServerFinder::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 (EncodeServerDescription 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 (EncodeServerDescription 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);
 
@@ -172,13 +185,16 @@ 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;
 
        int option_index = 0;
        while (true) {
@@ -188,15 +204,18 @@ 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' },
                        /* Just using A, B, C ... from here on */
                        { "dump", no_argument, 0, 'A' },
                        { 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:ld", long_options, &option_index);
 
                if (c == -1) {
                        break;
@@ -218,6 +237,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 +250,37 @@ 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;
                }
        }
 
        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 +296,23 @@ main (int argc, char* argv[])
        signal_manager = new SignalManager ();
 
        if (no_remote) {
-               EncodeServerFinder::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);
        }
 
@@ -309,31 +359,35 @@ main (int argc, char* argv[])
                int unfinished = 0;
                int finished_in_error = 0;
 
-               for (list<shared_ptr<Job> >::iterator i = jobs.begin(); i != jobs.end(); ++i) {
+               BOOST_FOREACH (shared_ptr<Job> i, jobs) {
                        if (progress) {
-                               cout << (*i)->name() << ": ";
+                               cout << i->name();
+                               if (!i->sub_name().empty()) {
+                                       cout << "; " << i->sub_name();
+                               }
+                               cout << ": ";
 
-                               if ((*i)->progress ()) {
-                                       cout << (*i)->status() << "                         \n";
+                               if (i->progress ()) {
+                                       cout << i->status() << "                            \n";
                                } else {
                                        cout << ": Running           \n";
                                }
                        }
 
-                       if (!(*i)->finished ()) {
+                       if (!i->finished ()) {
                                ++unfinished;
                        }
 
-                       if ((*i)->finished_in_error ()) {
+                       if (i->finished_in_error ()) {
                                ++finished_in_error;
                                error = true;
                        }
 
-                       if (!progress && (*i)->finished_in_error ()) {
+                       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";
+                               cout << i->status() << "\n";
                        }
                }
 
@@ -355,5 +409,9 @@ main (int argc, char* argv[])
 
        EncodeServerFinder::drop ();
 
+       if (dcp_path && !error) {
+               cout << film->dir (film->dcp_name (false)).string() << "\n";
+       }
+
        return error ? EXIT_FAILURE : EXIT_SUCCESS;
 }