Add --channel option to dcpomatic_create.
[dcpomatic.git] / src / lib / create_cli.cc
index e1c5f1233cc45790fa12e918c23109cdd0cf6fca..8d7c5928758720e50e2ce491aa4638617e08c860 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2019 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2019-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
 
 */
 
+
+#include "compose.hpp"
+#include "config.h"
 #include "create_cli.h"
 #include "dcp_content_type.h"
 #include "ratio.h"
-#include "compose.hpp"
 #include <dcp/raw_convert.h>
+#include <iostream>
 #include <string>
 
-#include <iostream>
 
 using std::string;
 using std::cout;
 using boost::optional;
 
+
 string CreateCLI::_help =
        "\nSyntax: %1 [OPTION] <CONTENT> [OPTION] [<CONTENT> ...]\n"
        "  -v, --version                 show DCP-o-matic version\n"
@@ -41,7 +44,6 @@ string CreateCLI::_help =
        "  -c, --dcp-content-type <type> FTR, SHR, TLR, TST, XSN, RTG, TSR, POL, PSA or ADV\n"
        "  -f, --dcp-frame-rate <rate>   set DCP video frame rate (otherwise guessed from content)\n"
        "      --container-ratio <ratio> 119, 133, 137, 138, 166, 178, 185 or 239\n"
-       "      --content-ratio <ratio>   119, 133, 137, 138, 166, 178, 185 or 239\n"
        "  -s, --still-length <n>        number of seconds that still content should last\n"
        "      --standard <standard>     SMPTE or interop (default SMPTE)\n"
        "      --no-use-isdcf-name       do not use an ISDCF name; use the specified name unmodified\n"
@@ -50,8 +52,11 @@ string CreateCLI::_help =
        "      --fourk                   make a 4K DCP rather than a 2K one\n"
        "  -o, --output <dir>            output directory\n"
        "      --threed                  make a 3D DCP\n"
+       "      --j2k-bandwidth <Mbit/s>  J2K bandwidth in Mbit/s\n"
        "      --left-eye                next piece of content is for the left eye\n"
-       "      --right-eye               next piece of content is for the right eye\n";
+       "      --right-eye               next piece of content is for the right eye\n"
+       "      --channel <channel>       next piece of content should be mapped to audio channel L, R, C, Lfe, Ls or Rs\n";
+
 
 template <class T>
 void
@@ -71,29 +76,63 @@ argument_option (int& n, int argc, char* argv[], string short_name, string long_
        *claimed = true;
 }
 
+
+template <class T>
+void
+argument_option (
+       int& n,
+       int argc,
+       char* argv[],
+       string short_name,
+       string long_name,
+       bool* claimed,
+       optional<string>* error,
+       boost::optional<T>* out,
+       std::function<boost::optional<T> (string s)> convert = [](string s) { return dcp::raw_convert<T>(s); }
+       )
+{
+       string const a = argv[n];
+       if (a != short_name && a != long_name) {
+               return;
+       }
+
+       if ((n + 1) >= argc) {
+               **error = String::compose("%1: option %2 requires an argument", argv[0], long_name);
+               return;
+       }
+
+       auto const arg = argv[++n];
+       auto const value = convert(arg);
+       if (!value) {
+               *error = String::compose("%1: %2 is not valid for %3", argv[0], arg, long_name);
+               *claimed = true;
+               return;
+       }
+
+       *out = value;
+       *claimed = true;
+}
+
+
 CreateCLI::CreateCLI (int argc, char* argv[])
        : version (false)
        , encrypt (false)
        , threed (false)
-       , dcp_content_type (0)
-       , dcp_frame_rate (24)
-       , container_ratio (0)
-       , content_ratio (0)
+       , dcp_content_type (nullptr)
+       , container_ratio (nullptr)
        , still_length (10)
-       , standard (dcp::SMPTE)
+       , standard (dcp::Standard::SMPTE)
        , no_use_isdcf_name (false)
-       , no_sign (false)
        , fourk (false)
 {
        string dcp_content_type_string = "TST";
-       string content_ratio_string;
        string container_ratio_string;
        string standard_string = "SMPTE";
        int dcp_frame_rate_int = 0;
        string template_name_string;
-       string config_dir_string;
-       string output_dir_string;
-       VideoFrameType next_frame_type = VIDEO_FRAME_TYPE_2D;
+       int j2k_bandwidth_int = 0;
+       auto next_frame_type = VideoFrameType::TWO_D;
+       optional<dcp::Channel> channel;
 
        int i = 1;
        while (i < argc) {
@@ -114,31 +153,53 @@ CreateCLI::CreateCLI (int argc, char* argv[])
                        encrypt = claimed = true;
                } else if (a == "--no-use-isdcf-name") {
                        no_use_isdcf_name = claimed = true;
-               } else if (a == "--no-sign") {
-                       no_sign = claimed = true;
                } else if (a == "--threed") {
                        threed = claimed = true;
                } else if (a == "--left-eye") {
-                       next_frame_type = VIDEO_FRAME_TYPE_3D_LEFT;
+                       next_frame_type = VideoFrameType::THREE_D_LEFT;
                        claimed = true;
                } else if (a == "--right-eye") {
-                       next_frame_type = VIDEO_FRAME_TYPE_3D_RIGHT;
+                       next_frame_type = VideoFrameType::THREE_D_RIGHT;
                        claimed = true;
                } else if (a == "--fourk") {
                        fourk = true;
                        claimed = true;
                }
 
+               std::function<optional<boost::filesystem::path> (string s)> string_to_path = [](string s) {
+                       return boost::filesystem::path(s);
+               };
+
                argument_option(i, argc, argv, "-n", "--name",             &claimed, &error, &name);
                argument_option(i, argc, argv, "-t", "--template",         &claimed, &error, &template_name_string);
                argument_option(i, argc, argv, "-c", "--dcp-content-type", &claimed, &error, &dcp_content_type_string);
                argument_option(i, argc, argv, "-f", "--dcp-frame-rate",   &claimed, &error, &dcp_frame_rate_int);
                argument_option(i, argc, argv, "",   "--container-ratio",  &claimed, &error, &container_ratio_string);
-               argument_option(i, argc, argv, "",   "--content-ratio",    &claimed, &error, &content_ratio_string);
                argument_option(i, argc, argv, "-s", "--still-length",     &claimed, &error, &still_length);
                argument_option(i, argc, argv, "",   "--standard",         &claimed, &error, &standard_string);
-               argument_option(i, argc, argv, "",   "--config",           &claimed, &error, &config_dir_string);
-               argument_option(i, argc, argv, "-o", "--output",           &claimed, &error, &output_dir_string);
+               argument_option(i, argc, argv, "",   "--config",           &claimed, &error, &config_dir, string_to_path);
+               argument_option(i, argc, argv, "-o", "--output",           &claimed, &error, &output_dir, string_to_path);
+               argument_option(i, argc, argv, "",   "--j2k-bandwidth",    &claimed, &error, &j2k_bandwidth_int);
+
+               std::function<optional<dcp::Channel> (string)> convert_channel = [](string channel) -> optional<dcp::Channel>{
+                       if (channel == "L") {
+                               return dcp::Channel::LEFT;
+                       } else if (channel == "R") {
+                               return dcp::Channel::RIGHT;
+                       } else if (channel == "C") {
+                               return dcp::Channel::CENTRE;
+                       } else if (channel == "Lfe") {
+                               return dcp::Channel::LFE;
+                       } else if (channel == "Ls") {
+                               return dcp::Channel::LS;
+                       } else if (channel == "Rs") {
+                               return dcp::Channel::RS;
+                       } else {
+                               return {};
+                       }
+               };
+
+               argument_option(i, argc, argv, "", "--channel", &claimed, &error, &channel, convert_channel);
 
                if (!claimed) {
                        if (a.length() > 2 && a.substr(0, 2) == "--") {
@@ -148,22 +209,16 @@ CreateCLI::CreateCLI (int argc, char* argv[])
                                Content c;
                                c.path = a;
                                c.frame_type = next_frame_type;
+                               c.channel = channel;
                                content.push_back (c);
-                               next_frame_type = VIDEO_FRAME_TYPE_2D;
+                               next_frame_type = VideoFrameType::TWO_D;
+                               channel = {};
                        }
                }
 
                ++i;
        }
 
-       if (!config_dir_string.empty()) {
-               config_dir = config_dir_string;
-       }
-
-       if (!output_dir_string.empty()) {
-               output_dir = output_dir_string;
-       }
-
        if (!template_name_string.empty()) {
                template_name = template_name_string;
        }
@@ -172,31 +227,22 @@ CreateCLI::CreateCLI (int argc, char* argv[])
                dcp_frame_rate = dcp_frame_rate_int;
        }
 
+       if (j2k_bandwidth_int) {
+               j2k_bandwidth = j2k_bandwidth_int * 1000000;
+       }
+
        dcp_content_type = DCPContentType::from_isdcf_name(dcp_content_type_string);
        if (!dcp_content_type) {
                error = String::compose("%1: unrecognised DCP content type '%2'", argv[0], dcp_content_type_string);
                return;
        }
 
-       if (content_ratio_string.empty()) {
-               error = String::compose("%1: missing required option --content-ratio", argv[0]);
-               return;
-       }
-
-       content_ratio = Ratio::from_id (content_ratio_string);
-       if (!content_ratio) {
-               error = String::compose("%1: unrecognised content ratio %2", content_ratio_string);
-               return;
-       }
-
-       if (container_ratio_string.empty()) {
-               container_ratio_string = content_ratio_string;
-       }
-
-       container_ratio = Ratio::from_id (container_ratio_string);
-       if (!container_ratio) {
-               error = String::compose("%1: unrecognised container ratio %2", argv[0], container_ratio_string);
-               return;
+       if (!container_ratio_string.empty()) {
+               container_ratio = Ratio::from_id (container_ratio_string);
+               if (!container_ratio) {
+                       error = String::compose("%1: unrecognised container ratio %2", argv[0], container_ratio_string);
+                       return;
+               }
        }
 
        if (standard_string != "SMPTE" && standard_string != "interop") {
@@ -204,6 +250,10 @@ CreateCLI::CreateCLI (int argc, char* argv[])
                return;
        }
 
+       if (standard_string == "interop") {
+               standard = dcp::Standard::INTEROP;
+       }
+
        if (content.empty()) {
                error = String::compose("%1: no content specified", argv[0]);
                return;
@@ -212,4 +262,9 @@ CreateCLI::CreateCLI (int argc, char* argv[])
        if (name.empty()) {
                name = content[0].path.leaf().string();
        }
+
+       if (j2k_bandwidth && (*j2k_bandwidth < 10000000 || *j2k_bandwidth > Config::instance()->maximum_j2k_bandwidth())) {
+               error = String::compose("%1: j2k-bandwidth must be between 10 and %2 Mbit/s", argv[0], (Config::instance()->maximum_j2k_bandwidth() / 1000000));
+               return;
+       }
 }