X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Ftools%2Fdcpomatic_cli.cc;h=52a113cbb4c72d6693faf84677a430cc3fc81e3e;hp=d95b2ed99f63fac0a2c0e923c4272bedf1fa69ae;hb=ed0b3ee0c5a0ba11d3a1a1dfee8e71238bcab4bd;hpb=854f2e5bbb7ffb9758b823af87034033033f3cb8 diff --git a/src/tools/dcpomatic_cli.cc b/src/tools/dcpomatic_cli.cc index d95b2ed99..52a113cbb 100644 --- a/src/tools/dcpomatic_cli.cc +++ b/src/tools/dcpomatic_cli.cc @@ -1,90 +1,222 @@ /* - Copyright (C) 2012 Carl Hetherington + Copyright (C) 2012-2017 Carl Hetherington - 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 . */ -#include -#include -#include -#include #include "lib/film.h" #include "lib/filter.h" #include "lib/transcode_job.h" #include "lib/job_manager.h" #include "lib/util.h" -#include "lib/scaler.h" #include "lib/version.h" #include "lib/cross.h" #include "lib/config.h" #include "lib/log.h" -#include "lib/ui_signaller.h" -#include "lib/server_finder.h" +#include "lib/signal_manager.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 +#include +#include +#include +#include using std::string; using std::cerr; using std::cout; using std::vector; using std::pair; +using std::setw; using std::list; using boost::shared_ptr; +using boost::optional; +using boost::dynamic_pointer_cast; static void help (string n) { - cerr << "Syntax: " << n << " [OPTION] \n" - << " -v, --version show DCP-o-matic version\n" - << " -h, --help show this help\n" - << " -d, --deps list DCP-o-matic dependency details and quit\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 run a JSON server on the specified port\n" - << " -k, --keep-going keep running even when the job is complete\n" + cerr << "Syntax: " << n << " [OPTION] []\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 run a JSON server on the specified port\n" + << " -k, --keep-going keep running even when the job is complete\n" + << " -s, --servers 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" << " is the film directory.\n"; } +static void +print_dump (shared_ptr film) +{ + cout << film->dcp_name (true) << "\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"; + + BOOST_FOREACH (shared_ptr c, film->content ()) { + cout << "\n" + << c->path(0) << "\n" + << "\tat " << c->position().seconds () + << " length " << c->full_length().seconds () + << " start trim " << c->trim_start().seconds () + << " end trim " << c->trim_end().seconds () << "\n"; + + 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()[c->video->colour_conversion().get().preset().get()].name + << "\n"; + } else { + cout << "\tcustom colour conversion\n"; + } + } else { + cout << "\tno colour conversion\n"; + } + + } + + if (c->audio) { + cout << "\t" << c->audio->delay() << " delay\n" + << "\t" << c->audio->gain() << " gain\n"; + } + } +} + +static void +list_servers () +{ + while (true) { + int N = 0; + list servers = EncodeServerFinder::instance()->good_servers (); + + /* 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 { + 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 threads; + list::iterator j = servers.begin (); + while (j != servers.end ()) { + if (i == j->host_name()) { + threads = j->threads(); + list::iterator tmp = j; + ++tmp; + servers.erase (j); + j = tmp; + } else { + ++j; + } + } + if (static_cast(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) { + cout << std::left << setw(24) << i.host_name() << " UP " << i.threads() << "\n"; + ++N; + } + + /* And those that have a bad version */ + BOOST_FOREACH (EncodeServerDescription i, EncodeServerFinder::instance()->good_servers()) { + cout << std::left << setw(24) << i.host_name() << " bad version\n"; + } + } + + dcpomatic_sleep (1); + + for (int i = 0; i < N; ++i) { + cout << "\033[1A\033[2K"; + } + } +} + + int main (int argc, char* argv[]) { - string film_dir; + boost::filesystem::path film_dir; bool progress = true; bool no_remote = false; - int log_level = 0; - int json_port = 0; + optional threads; + optional json_port; bool keep_going = false; + bool dump = false; + optional servers; + bool list_servers_ = false; + bool dcp_path = false; int option_index = 0; - while (1) { + while (true) { static struct option long_options[] = { { "version", no_argument, 0, 'v'}, { "help", no_argument, 0, 'h'}, - { "deps", no_argument, 0, 'd'}, { "flags", no_argument, 0, 'f'}, { "no-progress", no_argument, 0, 'n'}, { "no-remote", no_argument, 0, 'r'}, - { "log-level", required_argument, 0, 'l' }, - { "json", required_argument, 0, 'j' }, + { "threads", required_argument, 0, 't'}, + { "json", required_argument, 0, 'j'}, { "keep-going", no_argument, 0, 'k' }, + { "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, "vhdfnrl:j:k", long_options, &option_index); + int c = getopt_long (argc, argv, "vhfnrt:j:kAs:ld", long_options, &option_index); if (c == -1) { break; @@ -97,9 +229,6 @@ main (int argc, char* argv[]) case 'h': help (argv[0]); exit (EXIT_SUCCESS); - case 'd': - cout << dependency_version_summary () << "\n"; - exit (EXIT_SUCCESS); case 'f': cout << dcpomatic_cxx_flags << "\n"; exit (EXIT_SUCCESS); @@ -109,8 +238,8 @@ main (int argc, char* argv[]) case 'r': no_remote = true; break; - case 'l': - log_level = atoi (optarg); + case 't': + threads = atoi (optarg); break; case 'j': json_port = atoi (optarg); @@ -118,7 +247,42 @@ main (int argc, char* argv[]) case 'k': keep_going = true; break; + case 'A': + dump = true; + break; + case 's': + servers = optarg; + break; + case 'l': + list_servers_ = true; + break; + case 'd': + dcp_path = true; + progress = false; + break; + } + } + + if (servers) { + FILE* f = fopen_boost (*servers, "r"); + if (!f) { + cerr << "Could not open servers list file " << *servers << "\n"; + exit (EXIT_FAILURE); + } + vector 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); } if (optind >= argc) { @@ -127,40 +291,51 @@ main (int argc, char* argv[]) } film_dir = argv[optind]; - + + dcpomatic_setup_path_encoding (); dcpomatic_setup (); - ui_signaller = new UISignaller (); + signal_manager = new SignalManager (); if (no_remote) { - ServerFinder::instance()->disable (); + EncodeServerFinder::instance()->stop (); } if (json_port) { - new JSONServer (json_port); + new JSONServer (json_port.get ()); } - cout << "DCP-o-matic " << dcpomatic_version << " git " << dcpomatic_git_commit; - char buf[256]; - if (gethostname (buf, 256) == 0) { - cout << " on " << buf; + if (threads) { + Config::instance()->set_master_encoding_threads (threads.get ()); } - cout << "\n"; shared_ptr 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); } - film->log()->set_level ((Log::Level) log_level); + if (dump) { + print_dump (film); + exit (EXIT_SUCCESS); + } + + ContentList content = film->content (); + for (ContentList::const_iterator i = content.begin(); i != content.end(); ++i) { + vector paths = (*i)->paths (); + for (vector::const_iterator j = paths.begin(); j != paths.end(); ++j) { + if (!boost::filesystem::exists (*j)) { + cerr << argv[0] << ": content file " << *j << " not found.\n"; + exit (EXIT_FAILURE); + } + } + } - cout << "\nMaking DCP for " << film->name() << "\n"; -// cout << "Content: " << film->content() << "\n"; -// pair const f = Filter::ffmpeg_strings (film->filters ()); -// cout << "Filters: " << f.first << " " << f.second << "\n"; + if (progress) { + cout << "\nMaking DCP for " << film->name() << "\n"; + } film->make_dcp (); @@ -174,7 +349,9 @@ main (int argc, char* argv[]) list > jobs = JobManager::instance()->get (); if (!first && progress) { - cout << "\033[" << jobs.size() << "A"; + for (size_t i = 0; i < jobs.size(); ++i) { + cout << "\033[1A\033[2K"; + } cout.flush (); } @@ -183,33 +360,35 @@ main (int argc, char* argv[]) int unfinished = 0; int finished_in_error = 0; - for (list >::iterator i = jobs.begin(); i != jobs.end(); ++i) { + BOOST_FOREACH (shared_ptr i, jobs) { if (progress) { - cout << (*i)->name() << ": "; - - float const p = (*i)->progress (); - - if (p >= 0) { - cout << (*i)->status() << " \n"; + cout << i->name(); + if (!i->sub_name().empty()) { + cout << "; " << i->sub_name(); + } + cout << ": "; + + 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"; } } @@ -219,7 +398,7 @@ main (int argc, char* argv[]) } if (keep_going) { - while (1) { + while (true) { dcpomatic_sleep (3600); } } @@ -228,8 +407,12 @@ main (int argc, char* argv[]) indirectly holding onto codecs. */ JobManager::drop (); - + + EncodeServerFinder::drop (); + + if (dcp_path && !error) { + cout << film->dir (film->dcp_name (false)).string() << "\n"; + } + return error ? EXIT_FAILURE : EXIT_SUCCESS; } - -