summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCarl Hetherington <cth@carlh.net>2025-03-02 23:14:26 +0100
committerCarl Hetherington <cth@carlh.net>2025-03-02 23:21:32 +0100
commit91fa9976a5927d8d42cce92f4ce202a542e568d3 (patch)
tree69b05c78709f7d7d6a42466f9f69aec1f823fb64
parent5c14c5e61d71fd4a249587835db36225e0f20190 (diff)
Move CLI encoder to a vaguely-testable method.
-rw-r--r--src/lib/encode_cli.cc542
-rw-r--r--src/lib/encode_cli.h27
-rw-r--r--src/lib/wscript1
-rw-r--r--src/tools/dcpomatic_cli.cc529
4 files changed, 582 insertions, 517 deletions
diff --git a/src/lib/encode_cli.cc b/src/lib/encode_cli.cc
new file mode 100644
index 000000000..62218e2b2
--- /dev/null
+++ b/src/lib/encode_cli.cc
@@ -0,0 +1,542 @@
+/*
+ Copyright (C) 2012-2022 Carl Hetherington <cth@carlh.net>
+
+ 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.
+
+ 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 DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+
+#include "ansi.h"
+#include "audio_content.h"
+#include "config.h"
+#include "cross.h"
+#include "dcpomatic_log.h"
+#include "encode_server_finder.h"
+#include "ffmpeg_film_encoder.h"
+#include "film.h"
+#include "filter.h"
+#ifdef DCPOMATIC_GROK
+#include "grok/context.h"
+#endif
+#include "hints.h"
+#include "job_manager.h"
+#include "json_server.h"
+#include "log.h"
+#include "make_dcp.h"
+#include "ratio.h"
+#include "signal_manager.h"
+#include "transcode_job.h"
+#include "util.h"
+#include "variant.h"
+#include "version.h"
+#include "video_content.h"
+#include <dcp/filesystem.h>
+#include <dcp/version.h>
+#include <fmt/format.h>
+#include <getopt.h>
+#include <iostream>
+#include <iomanip>
+
+
+using std::cout;
+using std::dynamic_pointer_cast;
+using std::function;
+using std::list;
+using std::pair;
+using std::runtime_error;
+using std::shared_ptr;
+using std::string;
+using std::vector;
+using boost::optional;
+
+
+static void
+help(function <void (string)> out)
+{
+ out(fmt::format("Syntax: {} [OPTION] [<FILM>]\n", program_name));
+ out(variant::insert_dcpomatic(" -v, --version show %1 version\n"));
+ out(" -h, --help show this help\n");
+ out(" -f, --flags show flags passed to C++ compiler on build\n");
+ out(" -n, --no-progress do not print progress to stdout\n");
+ out(" -r, --no-remote do not use any remote servers\n");
+ out(" -t, --threads specify number of local encoding threads (overriding configuration)\n");
+ out(" -j, --json <port> run a JSON server on the specified port\n");
+ out(" -k, --keep-going keep running even when the job is complete\n");
+ out(" -s, --servers <file> specify servers to use in a text file\n");
+ out(variant::insert_dcpomatic(" -l, --list-servers just display a list of encoding servers that %1 is configured to use; don't encode\n"));
+ out(" -d, --dcp-path echo DCP's path to stdout on successful completion (implies -n)\n");
+ out(" -c, --config <dir> directory containing config.xml and cinemas.xml\n");
+ out(" --dump just dump a summary of the film's settings; don't encode\n");
+ out(" --no-check don't check project's content files for changes before making the DCP\n");
+ out(" --export-format <format> export project to a file, rather than making a DCP: specify mov or mp4\n");
+ out(" --export-filename <filename> filename to export to with --export-format\n");
+ out(" --hints analyze film for hints before encoding and abort if any are found\n");
+ out("\n");
+ out("<FILM> is the film directory.\n");
+}
+
+
+static void
+print_dump(function<void (string)> out, shared_ptr<Film> film)
+{
+ out(fmt::format("{}\n", film->dcp_name(true)));
+ out(fmt::format("{} at {}\n", film->container()->container_nickname(), film->resolution() == Resolution::TWO_K ? "2K" : "4K"));
+ out(fmt::format("{}Mbit/s\n", film->video_bit_rate(film->video_encoding()) / 1000000));
+ out(fmt::format("Duration {}\n", film->length().timecode(film->video_frame_rate())));
+ out(fmt::format("Output {}fps {} {}kHz\n", film->video_frame_rate(), film->three_d() ? "3D" : "2D", film->audio_frame_rate() / 1000));
+ out(fmt::format("{} {}\n", film->interop() ? "Inter-Op" : "SMPTE", film->encrypted() ? "encrypted" : "unencrypted"));
+
+ for (auto c: film->content()) {
+ out(fmt::format("\n{}\n", c->path(0).string()));
+ out(fmt::format("\tat {} length {} start trim {} end trim {}\n", c->position().seconds(), c->full_length(film).seconds(), c->trim_start().seconds(), c->trim_end().seconds()));
+
+ if (c->video && c->video->size()) {
+ out(fmt::format("\t{}x{}\n", c->video->size()->width, c->video->size()->height));
+ out(fmt::format("\t{}fps\n", c->active_video_frame_rate(film)));
+ out(fmt::format("\tcrop left {} right {} top {} bottom {}\n", c->video->requested_left_crop(), c->video->requested_right_crop(), c->video->requested_top_crop(), c->video->requested_bottom_crop()));
+ if (c->video->custom_ratio()) {
+ out(fmt::format("\tscale to custom ratio {}:1\n", *c->video->custom_ratio()));
+ }
+ if (c->video->colour_conversion()) {
+ if (c->video->colour_conversion().get().preset()) {
+ out(fmt::format("\tcolour conversion {}\n", PresetColourConversion::all()[c->video->colour_conversion().get().preset().get()].name));
+ } else {
+ out("\tcustom colour conversion\n");
+ }
+ } else {
+ out("\tno colour conversion\n");
+ }
+
+ }
+
+ if (c->audio) {
+ out(fmt::format("\t{} delay\n", c->audio->delay()));
+ out(fmt::format("\t{} gain\n", c->audio->gain()));
+ }
+ }
+}
+
+
+static void
+list_servers(function <void (string)> out)
+{
+ while (true) {
+ int N = 0;
+ auto servers = EncodeServerFinder::instance()->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()) {
+ out("No encoding servers found or configured.\n");
+ ++N;
+ } else {
+ out(fmt::format("{:24} Status Threads\n", "Host"));
+ ++N;
+
+ /* Report the state of configured servers */
+ for (auto i: Config::instance()->servers()) {
+ /* 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;
+ auto j = servers.begin();
+ while (j != servers.end()) {
+ if (i == j->host_name() && j->current_link_version()) {
+ threads = j->threads();
+ auto tmp = j;
+ ++tmp;
+ servers.erase(j);
+ j = tmp;
+ } else {
+ ++j;
+ }
+ }
+ if (static_cast<bool>(threads)) {
+ out(fmt::format("{:24} UP {}\n", i, threads.get()));
+ } else {
+ out(fmt::format("{:24} DOWN\n", i));
+ }
+ ++N;
+ }
+
+ /* Now report any left that have been found by broadcast */
+ for (auto const& i: servers) {
+ if (i.current_link_version()) {
+ out(fmt::format("{:24} UP {}\n", i.host_name(), i.threads()));
+ } else {
+ out(fmt::format("{:24} bad version\n", i.host_name()));
+ }
+ ++N;
+ }
+ }
+
+ dcpomatic_sleep_seconds(1);
+
+ for (int i = 0; i < N; ++i) {
+ out("\033[1A\033[2K");
+ }
+ }
+}
+
+
+bool
+show_jobs_on_console(function<void (string)> out, function<void ()> flush, bool progress)
+{
+ bool first = true;
+ bool error = false;
+ while (true) {
+
+ dcpomatic_sleep_seconds(5);
+
+ auto jobs = JobManager::instance()->get();
+
+ if (!first && progress) {
+ for (size_t i = 0; i < jobs.size(); ++i) {
+ out(UP_ONE_LINE_AND_ERASE);
+ }
+ flush();
+ }
+
+ first = false;
+
+ for (auto i: jobs) {
+ if (progress) {
+ out(i->name());
+ if (!i->sub_name().empty()) {
+ out(fmt::format("; {}", i->sub_name()));
+ }
+ out(": ");
+
+ if (i->progress()) {
+ out(fmt::format("{} \n", i->status()));
+ } else {
+ out(": Running \n");
+ }
+ }
+
+ if (!progress && i->finished_in_error()) {
+ /* We won't see this error if we haven't been showing progress,
+ so show it now.
+ */
+ out(fmt::format("{}\n", i->status()));
+ }
+
+ if (i->finished_in_error()) {
+ error = true;
+ }
+ }
+
+ if (!JobManager::instance()->work_to_do()) {
+ break;
+ }
+ }
+
+ return error;
+}
+
+
+optional<string>
+encode_cli(int argc, char* argv[], function<void (string)> out, function<void ()> flush)
+{
+ auto const program_name = argv[0];
+
+ 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;
+ optional<boost::filesystem::path> servers;
+ bool list_servers_ = false;
+ bool dcp_path = false;
+ optional<boost::filesystem::path> config;
+ bool check = true;
+ optional<string> export_format;
+ optional<boost::filesystem::path> export_filename;
+ bool hints = false;
+
+ int option_index = 0;
+ while (true) {
+ static struct option long_options[] = {
+ { "version", no_argument, 0, 'v'},
+ { "help", no_argument, 0, 'h'},
+ { "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", 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' },
+ { "export-format", required_argument, 0, 'C' },
+ { "export-filename", required_argument, 0, 'D' },
+ { "hints", no_argument, 0, 'E' },
+ { 0, 0, 0, 0 }
+ };
+
+ int c = getopt_long(argc, argv, "vhfnrt:j:kAs:ldc:BC:D:E", long_options, &option_index);
+
+ if (c == -1) {
+ break;
+ }
+
+ switch (c) {
+ case 'v':
+ out(fmt::format("dcpomatic version {} {}\n", dcpomatic_version, dcpomatic_git_commit));
+ exit(EXIT_SUCCESS);
+ case 'h':
+ help(out);
+ exit(EXIT_SUCCESS);
+ case 'f':
+ out(fmt::format("{}\n", dcpomatic_cxx_flags));
+ exit(EXIT_SUCCESS);
+ case 'n':
+ progress = false;
+ break;
+ case 'r':
+ no_remote = true;
+ break;
+ case 't':
+ threads = atoi(optarg);
+ break;
+ case 'j':
+ json_port = atoi(optarg);
+ break;
+ 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;
+ case 'c':
+ config = optarg;
+ break;
+ case 'B':
+ check = false;
+ break;
+ case 'C':
+ export_format = optarg;
+ break;
+ case 'D':
+ export_filename = optarg;
+ break;
+ case 'E':
+ hints = true;
+ break;
+ }
+ }
+
+ if (config) {
+ State::override_path = *config;
+ }
+
+ if (servers) {
+ dcp::File f(*servers, "r");
+ if (!f) {
+ return fmt::format("Could not open servers list file {}\n", servers->string());
+ }
+ vector<string> servers;
+ while (!f.eof()) {
+ char buffer[128];
+ if (fscanf(f.get(), "%s.127", buffer) == 1) {
+ servers.push_back(buffer);
+ }
+ }
+ Config::instance()->set_servers(servers);
+ }
+
+ if (list_servers_) {
+ list_servers(out);
+ exit(EXIT_SUCCESS);
+ }
+
+ if (optind >= argc) {
+ help(out);
+ exit(EXIT_FAILURE);
+ }
+
+ if (export_format && !export_filename) {
+ return string{"Argument --export-filename is required with --export-format\n"};
+ }
+
+ if (!export_format && export_filename) {
+ return string{"Argument --export-format is required with --export-filename\n"};
+ }
+
+ if (export_format && *export_format != "mp4" && *export_format != "mov") {
+ return string{"Unrecognised export format: must be mp4 or mov\n"};
+ }
+
+ film_dir = argv[optind];
+
+ signal_manager = new SignalManager();
+
+ if (no_remote || export_format) {
+ EncodeServerFinder::drop();
+ }
+
+ 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) {
+ return fmt::format("{}: error reading film `{}' ({})\n", program_name, film_dir.string(), e.what());
+ }
+
+ if (dump) {
+ print_dump(out, film);
+ exit(EXIT_SUCCESS);
+ }
+
+ dcpomatic_log = film->log();
+
+ for (auto i: film->content()) {
+ auto paths = i->paths();
+ for (auto j: paths) {
+ if (!dcp::filesystem::exists(j)) {
+ return fmt::format("{}: content file {} not found.\n", program_name, j.string());
+ }
+ }
+ }
+
+ if (!export_format && hints) {
+ string const prefix = "Checking project for hints";
+ bool pulse_phase = false;
+ vector<string> hints;
+ bool finished = false;
+
+ Hints hint_finder(film);
+ hint_finder.Progress.connect([prefix, &out, &flush](string progress) {
+ out(fmt::format("{}{}: {}\n", UP_ONE_LINE_AND_ERASE, prefix, progress));
+ flush();
+ });
+ hint_finder.Pulse.connect([prefix, &pulse_phase, &out, &flush]() {
+ out(fmt::format("{}{}: {}\n", UP_ONE_LINE_AND_ERASE, prefix, pulse_phase ? "X" : "x"));
+ flush();
+ pulse_phase = !pulse_phase;
+ });
+ hint_finder.Hint.connect([&hints](string hint) {
+ hints.push_back(hint);
+ });
+ hint_finder.Finished.connect([&finished]() {
+ finished = true;
+ });
+
+ out(fmt::format("{}:\n", prefix));
+ flush();
+
+ hint_finder.start();
+ while (!finished) {
+ signal_manager->ui_idle();
+ dcpomatic_sleep_milliseconds(200);
+ }
+
+ out(UP_ONE_LINE_AND_ERASE);
+
+ if (!hints.empty()) {
+ string error = "Hints:\n\n";
+ for (auto hint: hints) {
+ error += word_wrap("* " + hint, 70) + "\n";
+ }
+ error += "*** Encoding aborted because hints were found ***\n\n";
+ error += "Modify your settings and run the command again, or run without\n";
+ error += "the `--hints' option to ignore these hints and encode anyway.\n";
+ return error;
+ }
+ }
+
+#ifdef DCPOMATIC_GROK
+ grk_plugin::setMessengerLogger(new grk_plugin::GrokLogger("[GROK] "));
+ setup_grok_library_path();
+#endif
+
+ if (progress) {
+ if (export_format) {
+ out(fmt::format("\nExporting {}\n", film->name()));
+ } else {
+ out(fmt::format("\nMaking DCP for {}\n", film->name()));
+ }
+ }
+
+ TranscodeJob::ChangedBehaviour const behaviour = check ? TranscodeJob::ChangedBehaviour::STOP : TranscodeJob::ChangedBehaviour::IGNORE;
+
+ if (export_format) {
+ auto job = std::make_shared<TranscodeJob>(film, behaviour);
+ job->set_encoder(
+ std::make_shared<FFmpegFilmEncoder>(
+ film, job, *export_filename, *export_format == "mp4" ? ExportFormat::H264_AAC : ExportFormat::PRORES_HQ, false, false, false, 23
+ )
+ );
+ JobManager::instance()->add(job);
+ } else {
+ try {
+ make_dcp(film, behaviour);
+ } catch (runtime_error& e) {
+ return fmt::format("Could not make DCP: {}\n", e.what());
+ }
+ }
+
+ bool const error = show_jobs_on_console(out, flush, progress);
+
+ if (keep_going) {
+ while (true) {
+ dcpomatic_sleep_seconds(3600);
+ }
+ }
+
+ /* This is just to stop valgrind reporting leaks due to JobManager
+ indirectly holding onto codecs.
+ */
+ JobManager::drop();
+
+ EncodeServerFinder::drop();
+
+ if (dcp_path && !error) {
+ out(fmt::format("{}\n", film->dir(film->dcp_name(false)).string()));
+ }
+
+ if (error) {
+ return string{"Error during encoding"};
+ }
+
+ return {};
+}
diff --git a/src/lib/encode_cli.h b/src/lib/encode_cli.h
new file mode 100644
index 000000000..bbfce16af
--- /dev/null
+++ b/src/lib/encode_cli.h
@@ -0,0 +1,27 @@
+/*
+ Copyright (C) 2025 Carl Hetherington <cth@carlh.net>
+
+ 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.
+
+ 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 DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+
+#include <boost/optional.hpp>
+#include <string>
+
+
+extern boost::optional<std::string> encode_cli(int argc, char* argv[], std::function<void (std::string)> out, std::function<void ()> flush);
+
diff --git a/src/lib/wscript b/src/lib/wscript
index 2855041b2..79f0d563e 100644
--- a/src/lib/wscript
+++ b/src/lib/wscript
@@ -90,6 +90,7 @@ sources = """
dolby_cp750.cc
email.cc
empty.cc
+ encode_cli.cc
encode_server.cc
encode_server_finder.cc
encoded_log_entry.cc
diff --git a/src/tools/dcpomatic_cli.cc b/src/tools/dcpomatic_cli.cc
index a5d3f1fa7..178cd9123 100644
--- a/src/tools/dcpomatic_cli.cc
+++ b/src/tools/dcpomatic_cli.cc
@@ -1,5 +1,5 @@
/*
- Copyright (C) 2012-2022 Carl Hetherington <cth@carlh.net>
+ Copyright (C) 2023 Carl Hetherington <cth@carlh.net>
This file is part of DCP-o-matic.
@@ -19,536 +19,31 @@
*/
-#include "lib/ansi.h"
-#include "lib/audio_content.h"
-#include "lib/config.h"
+/** @file src/tools/dcpomatic_cli.cc
+ * @brief Command-line program to encode DCPs (and do some other utility bits).
+ */
+
+
#include "lib/cross.h"
-#include "lib/dcpomatic_log.h"
-#include "lib/encode_server_finder.h"
-#include "lib/ffmpeg_film_encoder.h"
-#include "lib/film.h"
-#include "lib/filter.h"
-#ifdef DCPOMATIC_GROK
-#include "lib/grok/context.h"
-#endif
-#include "lib/hints.h"
-#include "lib/job_manager.h"
-#include "lib/json_server.h"
-#include "lib/log.h"
-#include "lib/make_dcp.h"
-#include "lib/ratio.h"
-#include "lib/signal_manager.h"
-#include "lib/transcode_job.h"
+#include "lib/encode_cli.h"
#include "lib/util.h"
-#include "lib/variant.h"
-#include "lib/version.h"
-#include "lib/video_content.h"
-#include <dcp/filesystem.h>
-#include <dcp/version.h>
-#include <fmt/format.h>
-#include <getopt.h>
#include <iostream>
-#include <iomanip>
-
-
-using std::cerr;
-using std::cout;
-using std::dynamic_pointer_cast;
-using std::function;
-using std::list;
-using std::pair;
-using std::runtime_error;
-using std::shared_ptr;
-using std::string;
-using std::vector;
-using boost::optional;
-
-
-static void
-help(function <void (string)> out)
-{
- out(fmt::format("Syntax: {} [OPTION] [<FILM>]\n", program_name));
- out(variant::insert_dcpomatic(" -v, --version show %1 version\n"));
- out(" -h, --help show this help\n");
- out(" -f, --flags show flags passed to C++ compiler on build\n");
- out(" -n, --no-progress do not print progress to stdout\n");
- out(" -r, --no-remote do not use any remote servers\n");
- out(" -t, --threads specify number of local encoding threads (overriding configuration)\n");
- out(" -j, --json <port> run a JSON server on the specified port\n");
- out(" -k, --keep-going keep running even when the job is complete\n");
- out(" -s, --servers <file> specify servers to use in a text file\n");
- out(variant::insert_dcpomatic(" -l, --list-servers just display a list of encoding servers that %1 is configured to use; don't encode\n"));
- out(" -d, --dcp-path echo DCP's path to stdout on successful completion (implies -n)\n");
- out(" -c, --config <dir> directory containing config.xml and cinemas.xml\n");
- out(" --dump just dump a summary of the film's settings; don't encode\n");
- out(" --no-check don't check project's content files for changes before making the DCP\n");
- out(" --export-format <format> export project to a file, rather than making a DCP: specify mov or mp4\n");
- out(" --export-filename <filename> filename to export to with --export-format\n");
- out(" --hints analyze film for hints before encoding and abort if any are found\n");
- out("\n");
- out("<FILM> is the film directory.\n");
-}
-
-
-static void
-print_dump(function<void (string)> out, shared_ptr<Film> film)
-{
- out(fmt::format("{}\n", film->dcp_name(true)));
- out(fmt::format("{} at {}\n", film->container()->container_nickname(), film->resolution() == Resolution::TWO_K ? "2K" : "4K"));
- out(fmt::format("{}Mbit/s\n", film->video_bit_rate(film->video_encoding()) / 1000000));
- out(fmt::format("Duration {}\n", film->length().timecode(film->video_frame_rate())));
- out(fmt::format("Output {}fps {} {}kHz\n", film->video_frame_rate(), film->three_d() ? "3D" : "2D", film->audio_frame_rate() / 1000));
- out(fmt::format("{} {}\n", film->interop() ? "Inter-Op" : "SMPTE", film->encrypted() ? "encrypted" : "unencrypted"));
-
- for (auto c: film->content()) {
- out(fmt::format("\n{}\n", c->path(0).string()));
- out(fmt::format("\tat {} length {} start trim {} end trim {}\n", c->position().seconds(), c->full_length(film).seconds(), c->trim_start().seconds(), c->trim_end().seconds()));
-
- if (c->video && c->video->size()) {
- out(fmt::format("\t{}x{}\n", c->video->size()->width, c->video->size()->height));
- out(fmt::format("\t{}fps\n", c->active_video_frame_rate(film)));
- out(fmt::format("\tcrop left {} right {} top {} bottom {}\n", c->video->requested_left_crop(), c->video->requested_right_crop(), c->video->requested_top_crop(), c->video->requested_bottom_crop()));
- if (c->video->custom_ratio()) {
- out(fmt::format("\tscale to custom ratio {}:1\n", *c->video->custom_ratio()));
- }
- if (c->video->colour_conversion()) {
- if (c->video->colour_conversion().get().preset()) {
- out(fmt::format("\tcolour conversion {}\n", PresetColourConversion::all()[c->video->colour_conversion().get().preset().get()].name));
- } else {
- out("\tcustom colour conversion\n");
- }
- } else {
- out("\tno colour conversion\n");
- }
-
- }
-
- if (c->audio) {
- out(fmt::format("\t{} delay\n", c->audio->delay()));
- out(fmt::format("\t{} gain\n", c->audio->gain()));
- }
- }
-}
-
-
-static void
-list_servers(function <void (string)> out)
-{
- while (true) {
- int N = 0;
- auto servers = EncodeServerFinder::instance()->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()) {
- out("No encoding servers found or configured.\n");
- ++N;
- } else {
- out(fmt::format("{:24} Status Threads\n", "Host"));
- ++N;
-
- /* Report the state of configured servers */
- for (auto i: Config::instance()->servers()) {
- /* 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;
- auto j = servers.begin();
- while (j != servers.end()) {
- if (i == j->host_name() && j->current_link_version()) {
- threads = j->threads();
- auto tmp = j;
- ++tmp;
- servers.erase(j);
- j = tmp;
- } else {
- ++j;
- }
- }
- if (static_cast<bool>(threads)) {
- out(fmt::format("{:24} UP {}\n", i, threads.get()));
- } else {
- out(fmt::format("{:24} DOWN\n", i));
- }
- ++N;
- }
-
- /* Now report any left that have been found by broadcast */
- for (auto const& i: servers) {
- if (i.current_link_version()) {
- out(fmt::format("{:24} UP {}\n", i.host_name(), i.threads()));
- } else {
- out(fmt::format("{:24} bad version\n", i.host_name()));
- }
- ++N;
- }
- }
-
- dcpomatic_sleep_seconds(1);
-
- for (int i = 0; i < N; ++i) {
- out("\033[1A\033[2K");
- }
- }
-}
-
-
-bool
-show_jobs_on_console(function<void (string)> out, function<void ()> flush, bool progress)
-{
- bool first = true;
- bool error = false;
- while (true) {
-
- dcpomatic_sleep_seconds(5);
-
- auto jobs = JobManager::instance()->get();
-
- if (!first && progress) {
- for (size_t i = 0; i < jobs.size(); ++i) {
- out(UP_ONE_LINE_AND_ERASE);
- }
- flush();
- }
-
- first = false;
-
- for (auto i: jobs) {
- if (progress) {
- out(i->name());
- if (!i->sub_name().empty()) {
- out(fmt::format("; {}", i->sub_name()));
- }
- out(": ");
-
- if (i->progress()) {
- out(fmt::format("{} \n", i->status()));
- } else {
- out(": Running \n");
- }
- }
-
- if (!progress && i->finished_in_error()) {
- /* We won't see this error if we haven't been showing progress,
- so show it now.
- */
- out(fmt::format("{}\n", i->status()));
- }
-
- if (i->finished_in_error()) {
- error = true;
- }
- }
-
- if (!JobManager::instance()->work_to_do()) {
- break;
- }
- }
-
- return error;
-}
int
main(int argc, char* argv[])
{
- auto out = [](string s) { cout << s; };
- auto flush = []() { cout.flush(); };
-
ArgFixer fixer(argc, argv);
- auto const program_name = fixer.argv()[0];
dcpomatic_setup_path_encoding();
dcpomatic_setup();
- 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;
- optional<boost::filesystem::path> servers;
- bool list_servers_ = false;
- bool dcp_path = false;
- optional<boost::filesystem::path> config;
- bool check = true;
- optional<string> export_format;
- optional<boost::filesystem::path> export_filename;
- bool hints = false;
-
- int option_index = 0;
- while (true) {
- static struct option long_options[] = {
- { "version", no_argument, 0, 'v'},
- { "help", no_argument, 0, 'h'},
- { "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", 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' },
- { "export-format", required_argument, 0, 'C' },
- { "export-filename", required_argument, 0, 'D' },
- { "hints", no_argument, 0, 'E' },
- { 0, 0, 0, 0 }
- };
-
- int c = getopt_long(fixer.argc(), fixer.argv(), "vhfnrt:j:kAs:ldc:BC:D:E", long_options, &option_index);
-
- if (c == -1) {
- break;
- }
-
- switch (c) {
- case 'v':
- out(fmt::format("dcpomatic version {} {}\n", dcpomatic_version, dcpomatic_git_commit));
- exit(EXIT_SUCCESS);
- case 'h':
- help(out);
- exit(EXIT_SUCCESS);
- case 'f':
- out(fmt::format("{}\n", dcpomatic_cxx_flags));
- exit(EXIT_SUCCESS);
- case 'n':
- progress = false;
- break;
- case 'r':
- no_remote = true;
- break;
- case 't':
- threads = atoi(optarg);
- break;
- case 'j':
- json_port = atoi(optarg);
- break;
- 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;
- case 'c':
- config = optarg;
- break;
- case 'B':
- check = false;
- break;
- case 'C':
- export_format = optarg;
- break;
- case 'D':
- export_filename = optarg;
- break;
- case 'E':
- hints = true;
- break;
- }
- }
-
- if (config) {
- State::override_path = *config;
- }
-
- if (servers) {
- dcp::File f(*servers, "r");
- if (!f) {
- cerr << fmt::format("Could not open servers list file {}\n", servers->string());
- exit(EXIT_FAILURE);
- }
- vector<string> servers;
- while (!f.eof()) {
- char buffer[128];
- if (fscanf(f.get(), "%s.127", buffer) == 1) {
- servers.push_back(buffer);
- }
- }
- Config::instance()->set_servers(servers);
- }
-
- if (list_servers_) {
- list_servers(out);
- exit(EXIT_SUCCESS);
- }
-
- if (optind >= fixer.argc()) {
- help(out);
- exit(EXIT_FAILURE);
- }
-
- if (export_format && !export_filename) {
- cerr << "Argument --export-filename is required with --export-format\n";
- exit(EXIT_FAILURE);
- }
-
- if (!export_format && export_filename) {
- cerr << "Argument --export-format is required with --export-filename\n";
- exit(EXIT_FAILURE);
- }
-
- if (export_format && *export_format != "mp4" && *export_format != "mov") {
- cerr << "Unrecognised export format: must be mp4 or mov\n";
+ auto error = encode_cli(fixer.argc(), fixer.argv(), [](std::string s) { std::cout << s; }, []() { std::cout.flush(); });
+ if (error) {
+ std::cerr << *error << "\n";
exit(EXIT_FAILURE);
}
- film_dir = fixer.argv()[optind];
-
- signal_manager = new SignalManager();
-
- if (no_remote || export_format) {
- EncodeServerFinder::drop();
- }
-
- 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 << fmt::format("{}: error reading film `{}' ({})\n", program_name, film_dir.string(), e.what());
- exit(EXIT_FAILURE);
- }
-
- if (dump) {
- print_dump(out, film);
- exit(EXIT_SUCCESS);
- }
-
- dcpomatic_log = film->log();
-
- for (auto i: film->content()) {
- auto paths = i->paths();
- for (auto j: paths) {
- if (!dcp::filesystem::exists(j)) {
- cerr << fmt::format("{}: content file {} not found.\n", program_name, j.string());
- exit(EXIT_FAILURE);
- }
- }
- }
-
- if (!export_format && hints) {
- string const prefix = "Checking project for hints";
- bool pulse_phase = false;
- vector<string> hints;
- bool finished = false;
-
- Hints hint_finder(film);
- hint_finder.Progress.connect([prefix, &out, &flush](string progress) {
- out(fmt::format("{}{}: {}\n", UP_ONE_LINE_AND_ERASE, prefix, progress));
- flush();
- });
- hint_finder.Pulse.connect([prefix, &pulse_phase, &out, &flush]() {
- out(fmt::format("{}{}: {}\n", UP_ONE_LINE_AND_ERASE, prefix, pulse_phase ? "X" : "x"));
- flush();
- pulse_phase = !pulse_phase;
- });
- hint_finder.Hint.connect([&hints](string hint) {
- hints.push_back(hint);
- });
- hint_finder.Finished.connect([&finished]() {
- finished = true;
- });
-
- out(fmt::format("{}:\n", prefix));
- flush();
-
- hint_finder.start();
- while (!finished) {
- signal_manager->ui_idle();
- dcpomatic_sleep_milliseconds(200);
- }
-
- out(UP_ONE_LINE_AND_ERASE);
-
- if (!hints.empty()) {
- string error = "Hints:\n\n";
- for (auto hint: hints) {
- error += word_wrap("* " + hint, 70) + "\n";
- }
- error += "*** Encoding aborted because hints were found ***\n\n";
- error += "Modify your settings and run the command again, or run without\n";
- error += "the `--hints' option to ignore these hints and encode anyway.\n";
- std::cerr << error;
- exit(EXIT_FAILURE);
- }
- }
-
-#ifdef DCPOMATIC_GROK
- grk_plugin::setMessengerLogger(new grk_plugin::GrokLogger("[GROK] "));
- setup_grok_library_path();
-#endif
-
- if (progress) {
- if (export_format) {
- out(fmt::format("\nExporting {}\n", film->name()));
- } else {
- out(fmt::format("\nMaking DCP for {}\n", film->name()));
- }
- }
-
- TranscodeJob::ChangedBehaviour const behaviour = check ? TranscodeJob::ChangedBehaviour::STOP : TranscodeJob::ChangedBehaviour::IGNORE;
-
- if (export_format) {
- auto job = std::make_shared<TranscodeJob>(film, behaviour);
- job->set_encoder(
- std::make_shared<FFmpegFilmEncoder>(
- film, job, *export_filename, *export_format == "mp4" ? ExportFormat::H264_AAC : ExportFormat::PRORES_HQ, false, false, false, 23
- )
- );
- JobManager::instance()->add(job);
- } else {
- try {
- make_dcp(film, behaviour);
- } catch (runtime_error& e) {
- std::cerr << fmt::format("Could not make DCP: {}\n", e.what());
- exit(EXIT_FAILURE);
- }
- }
-
- bool const error = show_jobs_on_console(out, flush, progress);
-
- if (keep_going) {
- while (true) {
- dcpomatic_sleep_seconds(3600);
- }
- }
-
- /* This is just to stop valgrind reporting leaks due to JobManager
- indirectly holding onto codecs.
- */
- JobManager::drop();
-
- EncodeServerFinder::drop();
-
- if (dcp_path && !error) {
- out(fmt::format("{}\n", film->dir(film->dcp_name(false)).string()));
- }
-
- return error ? EXIT_FAILURE : EXIT_SUCCESS;
+ return 0;
}
+