]> git.sesse.net Git - nageru/blob - nageru/flags.cpp
887cd9603278ad478cac9a2babf3201bbb0b01cb
[nageru] / nageru / flags.cpp
1 #include "flags.h"
2
3 #include <getopt.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <string.h>
7
8 #include <utility>
9
10 using namespace std;
11
12 Flags global_flags;
13
14 // Long options that have no corresponding short option.
15 enum LongOption {
16         OPTION_HELP = 1000,
17         OPTION_FULLSCREEN,
18         OPTION_MULTICHANNEL,
19         OPTION_MIDI_MAPPING,
20         OPTION_DEFAULT_HDMI_INPUT,
21         OPTION_FAKE_CARDS_AUDIO,
22         OPTION_V4L_OUTPUT,
23         OPTION_HTTP_UNCOMPRESSED_VIDEO,
24         OPTION_HTTP_X264_VIDEO,
25         OPTION_RECORD_X264_VIDEO,
26         OPTION_X264_PRESET,
27         OPTION_X264_TUNE,
28         OPTION_X264_SPEEDCONTROL,
29         OPTION_X264_SPEEDCONTROL_VERBOSE,
30         OPTION_X264_BITRATE,
31         OPTION_X264_CRF,
32         OPTION_X264_VBV_BUFSIZE,
33         OPTION_X264_VBV_MAX_BITRATE,
34         OPTION_X264_PARAM,
35         OPTION_HTTP_MUX,
36         OPTION_HTTP_COARSE_TIMEBASE,
37         OPTION_HTTP_AUDIO_CODEC,
38         OPTION_HTTP_AUDIO_BITRATE,
39         OPTION_HTTP_PORT,
40         OPTION_NO_TRANSCODE_AUDIO,
41         OPTION_DISABLE_AUDIO,
42         OPTION_FLAT_AUDIO,
43         OPTION_GAIN_STAGING,
44         OPTION_DISABLE_LOCUT,
45         OPTION_ENABLE_LOCUT,
46         OPTION_DISABLE_GAIN_STAGING_AUTO,
47         OPTION_ENABLE_GAIN_STAGING_AUTO,
48         OPTION_DISABLE_COMPRESSOR,
49         OPTION_ENABLE_COMPRESSOR,
50         OPTION_DISABLE_LIMITER,
51         OPTION_ENABLE_LIMITER,
52         OPTION_DISABLE_MAKEUP_GAIN_AUTO,
53         OPTION_ENABLE_MAKEUP_GAIN_AUTO,
54         OPTION_DISABLE_ALSA_OUTPUT,
55         OPTION_NO_FLUSH_PBOS,
56         OPTION_PRINT_VIDEO_LATENCY,
57         OPTION_MAX_INPUT_QUEUE_FRAMES,
58         OPTION_AUDIO_QUEUE_LENGTH_MS,
59         OPTION_OUTPUT_YCBCR_COEFFICIENTS,
60         OPTION_OUTPUT_BUFFER_FRAMES,
61         OPTION_OUTPUT_SLOP_FRAMES,
62         OPTION_TIMECODE_STREAM,
63         OPTION_TIMECODE_STDOUT,
64         OPTION_QUICK_CUT_KEYS,
65         OPTION_10_BIT_INPUT,
66         OPTION_10_BIT_OUTPUT,
67         OPTION_INPUT_YCBCR_INTERPRETATION,
68         OPTION_MJPEG_EXPORT_CARDS,
69 };
70
71 map<unsigned, unsigned> parse_mjpeg_export_cards(char *optarg)
72 {
73         map<unsigned, unsigned> ret;
74         if (optarg[0] == '\0') {
75                 return ret;
76         }
77
78         unsigned stream_idx = 0;
79         char *start = optarg;
80         for ( ;; ) {
81                 char *end = strchr(start, ',');
82                 if (end != nullptr) {
83                         *end = '\0';
84                 }
85
86                 unsigned range_begin, range_end;
87                 if (sscanf(start, "%u-%u", &range_begin, &range_end) != 2) {
88                         range_begin = range_end = atoi(start);
89                 }
90                 if (range_end < range_begin) {
91                         fprintf(stderr, "ERROR: Invalid range %u-%u in --mjpeg-export-cards=\n", range_begin, range_end);
92                         exit(1);
93                 }
94                 if (range_end >= unsigned(global_flags.num_cards)) {
95                         // There are situations where we could possibly want to
96                         // include FFmpeg inputs (CEF inputs are unlikely),
97                         // but they're not necessarily in 4:2:2 Y'CbCr, so it would
98                         // require more functionality the the JPEG encoder.
99                         fprintf(stderr, "ERROR: Asked for (zero-indexed) card %u in --mjpeg-export-cards=, but there are only %u cards\n",
100                                 range_end, global_flags.num_cards);
101                         exit(1);
102                 }
103                 for (unsigned card_idx = range_begin; card_idx <= range_end; ++card_idx) {
104                         if (ret.count(card_idx)) {
105                                 fprintf(stderr, "ERROR: Card %u was given twice in --mjpeg-export-cards=\n", card_idx);
106                                 exit(1);
107                         }
108                         ret[card_idx] = stream_idx++;
109                 }
110                 if (end == nullptr) {
111                         break;
112                 } else {
113                         start = end + 1;
114                 }
115         }
116         return ret;
117 }
118
119 void usage(Program program)
120 {
121         if (program == PROGRAM_KAERU) {
122                 fprintf(stderr, "Usage: kaeru [OPTION]... SOURCE_URL\n");
123         } else {
124                 fprintf(stderr, "Usage: nageru [OPTION]...\n");
125         }
126         fprintf(stderr, "\n");
127         fprintf(stderr, "      --help                      print usage information\n");
128         if (program == PROGRAM_NAGERU) {
129                 fprintf(stderr, "      --fullscreen                run in full screen, with no decorations\n");
130         }
131         fprintf(stderr, "  -w, --width                     output width in pixels (default 1280)\n");
132         fprintf(stderr, "  -h, --height                    output height in pixels (default 720)\n");
133         if (program == PROGRAM_NAGERU) {
134                 fprintf(stderr, "  -c, --num-cards                 set number of input cards (default 2)\n");
135                 fprintf(stderr, "  -o, --output-card=CARD          also output signal to the given card (default none)\n");
136                 fprintf(stderr, "  -t, --theme=FILE                choose theme (default theme.lua)\n");
137                 fprintf(stderr, "  -I, --theme-dir=DIR             search for theme in this directory (can be given multiple times)\n");
138                 fprintf(stderr, "  -r, --recording-dir=DIR         where to store disk recording\n");
139                 fprintf(stderr, "  -v, --va-display=SPEC           VA-API device for H.264 encoding\n");
140                 fprintf(stderr, "                                    ($DISPLAY spec or /dev/dri/render* path)\n");
141                 fprintf(stderr, "  -m, --map-signal=SIGNAL,CARD    set a default card mapping (can be given multiple times)\n");
142                 fprintf(stderr, "  -M, --input-mapping=FILE        start with the given audio input mapping (implies --multichannel)\n");
143                 fprintf(stderr, "      --multichannel              start in multichannel audio mapping mode\n");
144                 fprintf(stderr, "      --midi-mapping=FILE         start with the given MIDI controller mapping (implies --multichannel)\n");
145                 fprintf(stderr, "      --default-hdmi-input        default to HDMI over SDI inputs for cards that have both\n");
146                 fprintf(stderr, "      --fake-cards-audio          make fake (disconnected) cards output a simple tone\n");
147                 fprintf(stderr, "      --http-uncompressed-video   send uncompressed NV12 video to HTTP clients\n");
148                 fprintf(stderr, "      --http-x264-video           send x264-compressed video to HTTP clients\n");
149                 fprintf(stderr, "      --record-x264-video         store x264-compressed video to disk (implies --http-x264-video,\n");
150                 fprintf(stderr, "                                    removes the need for working VA-API encoding)\n");
151         }
152         fprintf(stderr, "      --x264-preset               x264 quality preset (default " X264_DEFAULT_PRESET ")\n");
153         fprintf(stderr, "      --x264-tune                 x264 tuning (default " X264_DEFAULT_TUNE ", can be blank)\n");
154         fprintf(stderr, "      --x264-speedcontrol         try to match x264 preset to available CPU speed\n");
155         fprintf(stderr, "      --x264-speedcontrol-verbose  output speedcontrol debugging statistics\n");
156         fprintf(stderr, "      --x264-bitrate              x264 bitrate (in kilobit/sec, default %d)\n",
157                 DEFAULT_X264_OUTPUT_BIT_RATE);
158         fprintf(stderr, "      --x264-crf=VALUE            quality-based VBR (-12 to 51), incompatible with --x264-bitrate and VBV\n");
159         fprintf(stderr, "      --x264-vbv-bufsize          x264 VBV size (in kilobits, 0 = one-frame VBV,\n");
160         fprintf(stderr, "                                  default: same as --x264-bitrate, that is, one-second VBV)\n");
161         fprintf(stderr, "      --x264-vbv-max-bitrate      x264 local max bitrate (in kilobit/sec per --vbv-bufsize,\n");
162         fprintf(stderr, "                                  0 = no limit, default: same as --x264-bitrate, i.e., CBR)\n");
163         fprintf(stderr, "      --x264-param=NAME[,VALUE]   set any x264 parameter, for fine tuning\n");
164         fprintf(stderr, "      --http-mux=NAME             mux to use for HTTP streams (default " DEFAULT_STREAM_MUX_NAME ")\n");
165         fprintf(stderr, "      --http-audio-codec=NAME     audio codec to use for HTTP streams\n");
166         fprintf(stderr, "                                  (default is to use the same as for the recording)\n");
167         fprintf(stderr, "      --http-audio-bitrate=KBITS  audio codec bit rate to use for HTTP streams\n");
168         fprintf(stderr, "                                  (default is %d, ignored unless --http-audio-codec is set)\n",
169                 DEFAULT_AUDIO_OUTPUT_BIT_RATE / 1000);
170         fprintf(stderr, "      --http-port=PORT            which port to use for the built-in HTTP server\n");
171         fprintf(stderr, "                                  (default is %d)\n", DEFAULT_HTTPD_PORT);
172         if (program == PROGRAM_KAERU) {
173                 fprintf(stderr, "      --no-transcode-audio        copy encoded audio raw from the source stream\n");
174                 fprintf(stderr, "                                    (requires --http-audio-codec= to be set)\n");
175                 fprintf(stderr, "      --disable-audio             do not include any audio in the stream\n");
176         }
177         if (program == PROGRAM_NAGERU) {
178                 fprintf(stderr, "      --flat-audio                start with most audio processing turned off\n");
179                 fprintf(stderr, "                                    (can be overridden by e.g. --enable-limiter)\n");
180                 fprintf(stderr, "      --gain-staging=DB           set initial gain staging to the given value\n");
181                 fprintf(stderr, "                                    (--disable-gain-staging-auto)\n");
182                 fprintf(stderr, "      --disable-locut             turn off locut filter (also --enable)\n");
183                 fprintf(stderr, "      --disable-gain-staging-auto  turn off automatic gain staging (also --enable)\n");
184                 fprintf(stderr, "      --disable-compressor        turn off regular compressor (also --enable)\n");
185                 fprintf(stderr, "      --disable-limiter           turn off limiter (also --enable)\n");
186                 fprintf(stderr, "      --disable-makeup-gain-auto  turn off auto-adjustment of final makeup gain (also --enable)\n");
187                 fprintf(stderr, "      --disable-alsa-output       disable audio monitoring via ALSA\n");
188                 fprintf(stderr, "      --no-flush-pbos             do not explicitly signal texture data uploads\n");
189                 fprintf(stderr, "                                    (will give display corruption, but makes it\n");
190                 fprintf(stderr, "                                    possible to run with apitrace in real time)\n");
191                 fprintf(stderr, "      --print-video-latency       print out measurements of video latency on stdout\n");
192                 fprintf(stderr, "      --max-input-queue-frames=FRAMES  never keep more than FRAMES frames for each card\n");
193                 fprintf(stderr, "                                    (default 6, minimum 1)\n");
194                 fprintf(stderr, "      --audio-queue-length-ms=MS  length of audio resampling queue (default 100.0)\n");
195                 fprintf(stderr, "      --output-ycbcr-coefficients={rec601,rec709,auto}\n");
196                 fprintf(stderr, "                                  Y'CbCr coefficient standard of output (default auto)\n");
197                 fprintf(stderr, "                                    auto is rec601, unless --output-card is used\n");
198                 fprintf(stderr, "                                    and a Rec. 709 mode (typically HD modes) is in use\n");
199                 fprintf(stderr, "      --output-buffer-frames=NUM  number of frames in output buffer for --output-card,\n");
200                 fprintf(stderr, "                                    can be fractional (default 6.0); note also\n");
201                 fprintf(stderr, "                                    the audio queue can't be much longer than this\n");
202                 fprintf(stderr, "      --output-slop-frames=NUM    if more less than this number of frames behind for\n");
203                 fprintf(stderr, "                                    --output-card, try to submit anyway instead of\n");
204                 fprintf(stderr, "                                    dropping the frame (default 0.5)\n");
205                 fprintf(stderr, "      --timecode-stream           show timestamp and timecode in stream\n");
206                 fprintf(stderr, "      --timecode-stdout           show timestamp and timecode on standard output\n");
207                 fprintf(stderr, "      --quick-cut-keys            enable direct cutting by Q, W, E, ... keys\n");
208                 fprintf(stderr, "      --10-bit-input              use 10-bit video input (requires compute shaders)\n");
209                 fprintf(stderr, "      --10-bit-output             use 10-bit video output (requires compute shaders,\n");
210                 fprintf(stderr, "                                    implies --record-x264-video)\n");
211                 fprintf(stderr, "      --input-ycbcr-interpretation=CARD,{rec601,rec709,auto}[,{limited,full}]\n");
212                 fprintf(stderr, "                                  Y'CbCr coefficient standard of card CARD (default auto)\n");
213                 fprintf(stderr, "                                    auto is rec601 for SD, rec709 for HD, always limited\n");
214                 fprintf(stderr, "                                    limited means standard 0-240/0-235 input range (for 8-bit)\n");
215                 fprintf(stderr, "      --mjpeg-export-cards=RANGE[,RANGE...]\n");
216                 fprintf(stderr, "                                  export the given cards in MJPEG format to /multicam.mp4,\n");
217                 fprintf(stderr, "                                    in the given order (ranges can be either single card indexes\n");
218                 fprintf(stderr, "                                    or pairs like 1-3 for camera 1,2,3; default is all cards)\n");
219         }
220 }
221
222 void parse_flags(Program program, int argc, char * const argv[])
223 {
224         static const option long_options[] = {
225                 { "help", no_argument, 0, OPTION_HELP },
226                 { "fullscreen", no_argument, 0, OPTION_FULLSCREEN },
227                 { "width", required_argument, 0, 'w' },
228                 { "height", required_argument, 0, 'h' },
229                 { "num-cards", required_argument, 0, 'c' },
230                 { "output-card", required_argument, 0, 'o' },
231                 { "theme", required_argument, 0, 't' },
232                 { "theme-dir", required_argument, 0, 'I' },
233                 { "recording-dir", required_argument, 0, 'r' },
234                 { "map-signal", required_argument, 0, 'm' },
235                 { "input-mapping", required_argument, 0, 'M' },
236                 { "va-display", required_argument, 0, 'v' },
237                 { "multichannel", no_argument, 0, OPTION_MULTICHANNEL },
238                 { "midi-mapping", required_argument, 0, OPTION_MIDI_MAPPING },
239                 { "default-hdmi-input", no_argument, 0, OPTION_DEFAULT_HDMI_INPUT },
240                 { "fake-cards-audio", no_argument, 0, OPTION_FAKE_CARDS_AUDIO },
241                 { "v4l-output", required_argument, 0, OPTION_V4L_OUTPUT },
242                 { "http-uncompressed-video", no_argument, 0, OPTION_HTTP_UNCOMPRESSED_VIDEO },
243                 { "http-x264-video", no_argument, 0, OPTION_HTTP_X264_VIDEO },
244                 { "record-x264-video", no_argument, 0, OPTION_RECORD_X264_VIDEO },
245                 { "x264-preset", required_argument, 0, OPTION_X264_PRESET },
246                 { "x264-tune", required_argument, 0, OPTION_X264_TUNE },
247                 { "x264-speedcontrol", no_argument, 0, OPTION_X264_SPEEDCONTROL },
248                 { "x264-speedcontrol-verbose", no_argument, 0, OPTION_X264_SPEEDCONTROL_VERBOSE },
249                 { "x264-bitrate", required_argument, 0, OPTION_X264_BITRATE },
250                 { "x264-crf", required_argument, 0, OPTION_X264_CRF },
251                 { "x264-vbv-bufsize", required_argument, 0, OPTION_X264_VBV_BUFSIZE },
252                 { "x264-vbv-max-bitrate", required_argument, 0, OPTION_X264_VBV_MAX_BITRATE },
253                 { "x264-param", required_argument, 0, OPTION_X264_PARAM },
254                 { "http-mux", required_argument, 0, OPTION_HTTP_MUX },
255                 { "http-audio-codec", required_argument, 0, OPTION_HTTP_AUDIO_CODEC },
256                 { "http-audio-bitrate", required_argument, 0, OPTION_HTTP_AUDIO_BITRATE },
257                 { "http-port", required_argument, 0, OPTION_HTTP_PORT },
258                 { "no-transcode-audio", no_argument, 0, OPTION_NO_TRANSCODE_AUDIO },
259                 { "disable-audio", no_argument, 0, OPTION_DISABLE_AUDIO },
260                 { "flat-audio", no_argument, 0, OPTION_FLAT_AUDIO },
261                 { "gain-staging", required_argument, 0, OPTION_GAIN_STAGING },
262                 { "disable-locut", no_argument, 0, OPTION_DISABLE_LOCUT },
263                 { "enable-locut", no_argument, 0, OPTION_ENABLE_LOCUT },
264                 { "disable-gain-staging-auto", no_argument, 0, OPTION_DISABLE_GAIN_STAGING_AUTO },
265                 { "enable-gain-staging-auto", no_argument, 0, OPTION_ENABLE_GAIN_STAGING_AUTO },
266                 { "disable-compressor", no_argument, 0, OPTION_DISABLE_COMPRESSOR },
267                 { "enable-compressor", no_argument, 0, OPTION_ENABLE_COMPRESSOR },
268                 { "disable-limiter", no_argument, 0, OPTION_DISABLE_LIMITER },
269                 { "enable-limiter", no_argument, 0, OPTION_ENABLE_LIMITER },
270                 { "disable-makeup-gain-auto", no_argument, 0, OPTION_DISABLE_MAKEUP_GAIN_AUTO },
271                 { "enable-makeup-gain-auto", no_argument, 0, OPTION_ENABLE_MAKEUP_GAIN_AUTO },
272                 { "disable-alsa-output", no_argument, 0, OPTION_DISABLE_ALSA_OUTPUT },
273                 { "no-flush-pbos", no_argument, 0, OPTION_NO_FLUSH_PBOS },
274                 { "print-video-latency", no_argument, 0, OPTION_PRINT_VIDEO_LATENCY },
275                 { "max-input-queue-frames", required_argument, 0, OPTION_MAX_INPUT_QUEUE_FRAMES },
276                 { "audio-queue-length-ms", required_argument, 0, OPTION_AUDIO_QUEUE_LENGTH_MS },
277                 { "output-ycbcr-coefficients", required_argument, 0, OPTION_OUTPUT_YCBCR_COEFFICIENTS },
278                 { "output-buffer-frames", required_argument, 0, OPTION_OUTPUT_BUFFER_FRAMES },
279                 { "output-slop-frames", required_argument, 0, OPTION_OUTPUT_SLOP_FRAMES },
280                 { "timecode-stream", no_argument, 0, OPTION_TIMECODE_STREAM },
281                 { "timecode-stdout", no_argument, 0, OPTION_TIMECODE_STDOUT },
282                 { "quick-cut-keys", no_argument, 0, OPTION_QUICK_CUT_KEYS },
283                 { "10-bit-input", no_argument, 0, OPTION_10_BIT_INPUT },
284                 { "10-bit-output", no_argument, 0, OPTION_10_BIT_OUTPUT },
285                 { "input-ycbcr-interpretation", required_argument, 0, OPTION_INPUT_YCBCR_INTERPRETATION },
286                 { "mjpeg-export-cards", required_argument, 0, OPTION_MJPEG_EXPORT_CARDS },
287                 { 0, 0, 0, 0 }
288         };
289         vector<string> theme_dirs;
290         string output_ycbcr_coefficients = "auto";
291         bool card_to_mjpeg_stream_export_set = false;
292         for ( ;; ) {
293                 int option_index = 0;
294                 int c = getopt_long(argc, argv, "c:o:t:I:r:v:m:M:w:h:", long_options, &option_index);
295
296                 if (c == -1) {
297                         break;
298                 }
299                 switch (c) {
300                 case 'w':
301                         global_flags.width = atoi(optarg);
302                         break;
303                 case 'h':
304                         global_flags.height = atoi(optarg);
305                         break;
306                 case 'c':
307                         global_flags.num_cards = atoi(optarg);
308                         break;
309                 case 'o':
310                         global_flags.output_card = atoi(optarg);
311                         break;
312                 case 't':
313                         global_flags.theme_filename = optarg;
314                         break;
315                 case 'I':
316                         theme_dirs.push_back(optarg);
317                         break;
318                 case 'r':
319                         global_flags.recording_dir = optarg;
320                         break;
321                 case 'm': {
322                         char *ptr = strchr(optarg, ',');
323                         if (ptr == nullptr) {
324                                 fprintf(stderr, "ERROR: Invalid argument '%s' to --map-signal (needs a signal and a card number, separated by comma)\n", optarg);
325                                 exit(1);
326                         }
327                         *ptr = '\0';
328                         const int signal_num = atoi(optarg);
329                         const int card_num = atoi(ptr + 1);
330                         if (global_flags.default_stream_mapping.count(signal_num)) {
331                                 fprintf(stderr, "ERROR: Signal %d already mapped to card %d\n",
332                                         signal_num, global_flags.default_stream_mapping[signal_num]);
333                                 exit(1);
334                         }
335                         global_flags.default_stream_mapping[signal_num] = card_num;
336                         break;
337                 }
338                 case 'M':
339                         global_flags.input_mapping_filename = optarg;
340                         break;
341                 case OPTION_MULTICHANNEL:
342                         global_flags.multichannel_mapping_mode = true;
343                         break;
344                 case 'v':
345                         global_flags.va_display = optarg;
346                         break;
347                 case OPTION_MIDI_MAPPING:
348                         global_flags.midi_mapping_filename = optarg;
349                         global_flags.multichannel_mapping_mode = true;
350                         break;
351                 case OPTION_DEFAULT_HDMI_INPUT:
352                         global_flags.default_hdmi_input = true;
353                         break;
354                 case OPTION_FAKE_CARDS_AUDIO:
355                         global_flags.fake_cards_audio = true;
356                         break;
357                 case OPTION_V4L_OUTPUT:
358                         global_flags.v4l_output_device = optarg;
359                         break;
360                 case OPTION_HTTP_UNCOMPRESSED_VIDEO:
361                         global_flags.uncompressed_video_to_http = true;
362                         break;
363                 case OPTION_HTTP_MUX:
364                         global_flags.stream_mux_name = optarg;
365                         break;
366                 case OPTION_HTTP_AUDIO_CODEC:
367                         global_flags.stream_audio_codec_name = optarg;
368                         break;
369                 case OPTION_HTTP_AUDIO_BITRATE:
370                         global_flags.stream_audio_codec_bitrate = atoi(optarg) * 1000;
371                         break;
372                 case OPTION_HTTP_PORT:
373                         global_flags.http_port = atoi(optarg);
374                         break;
375                 case OPTION_NO_TRANSCODE_AUDIO:
376                         global_flags.transcode_audio = false;
377                         break;
378                 case OPTION_DISABLE_AUDIO:
379                         global_flags.transcode_audio = false;
380                         global_flags.enable_audio = false;
381                         break;
382                 case OPTION_HTTP_X264_VIDEO:
383                         global_flags.x264_video_to_http = true;
384                         break;
385                 case OPTION_RECORD_X264_VIDEO:
386                         global_flags.x264_video_to_disk = true;
387                         global_flags.x264_video_to_http = true;
388                         break;
389                 case OPTION_X264_PRESET:
390                         global_flags.x264_preset = optarg;
391                         break;
392                 case OPTION_X264_TUNE:
393                         global_flags.x264_tune = optarg;
394                         break;
395                 case OPTION_X264_SPEEDCONTROL:
396                         global_flags.x264_speedcontrol = true;
397                         break;
398                 case OPTION_X264_SPEEDCONTROL_VERBOSE:
399                         global_flags.x264_speedcontrol_verbose = true;
400                         break;
401                 case OPTION_X264_BITRATE:
402                         global_flags.x264_bitrate = atoi(optarg);
403                         break;
404                 case OPTION_X264_CRF:
405                         global_flags.x264_crf = atof(optarg);
406                         break;
407                 case OPTION_X264_VBV_BUFSIZE:
408                         global_flags.x264_vbv_buffer_size = atoi(optarg);
409                         break;
410                 case OPTION_X264_VBV_MAX_BITRATE:
411                         global_flags.x264_vbv_max_bitrate = atoi(optarg);
412                         break;
413                 case OPTION_X264_PARAM:
414                         global_flags.x264_extra_param.push_back(optarg);
415                         break;
416                 case OPTION_FLAT_AUDIO:
417                         // If --flat-audio is given, turn off everything that messes with the sound,
418                         // except the final makeup gain.
419                         global_flags.locut_enabled = false;
420                         global_flags.gain_staging_auto = false;
421                         global_flags.compressor_enabled = false;
422                         global_flags.limiter_enabled = false;
423                         break;
424                 case OPTION_GAIN_STAGING:
425                         global_flags.initial_gain_staging_db = atof(optarg);
426                         global_flags.gain_staging_auto = false;
427                         break;
428                 case OPTION_DISABLE_LOCUT:
429                         global_flags.locut_enabled = false;
430                         break;
431                 case OPTION_ENABLE_LOCUT:
432                         global_flags.locut_enabled = true;
433                         break;
434                 case OPTION_DISABLE_GAIN_STAGING_AUTO:
435                         global_flags.gain_staging_auto = false;
436                         break;
437                 case OPTION_ENABLE_GAIN_STAGING_AUTO:
438                         global_flags.gain_staging_auto = true;
439                         break;
440                 case OPTION_DISABLE_COMPRESSOR:
441                         global_flags.compressor_enabled = false;
442                         break;
443                 case OPTION_ENABLE_COMPRESSOR:
444                         global_flags.compressor_enabled = true;
445                         break;
446                 case OPTION_DISABLE_LIMITER:
447                         global_flags.limiter_enabled = false;
448                         break;
449                 case OPTION_ENABLE_LIMITER:
450                         global_flags.limiter_enabled = true;
451                         break;
452                 case OPTION_DISABLE_MAKEUP_GAIN_AUTO:
453                         global_flags.final_makeup_gain_auto = false;
454                         break;
455                 case OPTION_ENABLE_MAKEUP_GAIN_AUTO:
456                         global_flags.final_makeup_gain_auto = true;
457                         break;
458                 case OPTION_DISABLE_ALSA_OUTPUT:
459                         global_flags.enable_alsa_output = false;
460                         break;
461                 case OPTION_NO_FLUSH_PBOS:
462                         global_flags.flush_pbos = false;
463                         break;
464                 case OPTION_PRINT_VIDEO_LATENCY:
465                         global_flags.print_video_latency = true;
466                         break;
467                 case OPTION_MAX_INPUT_QUEUE_FRAMES:
468                         global_flags.max_input_queue_frames = atoi(optarg);
469                         break;
470                 case OPTION_AUDIO_QUEUE_LENGTH_MS:
471                         global_flags.audio_queue_length_ms = atof(optarg);
472                         break;
473                 case OPTION_OUTPUT_YCBCR_COEFFICIENTS:
474                         output_ycbcr_coefficients = optarg;
475                         break;
476                 case OPTION_OUTPUT_BUFFER_FRAMES:
477                         global_flags.output_buffer_frames = atof(optarg);
478                         break;
479                 case OPTION_OUTPUT_SLOP_FRAMES:
480                         global_flags.output_slop_frames = atof(optarg);
481                         break;
482                 case OPTION_TIMECODE_STREAM:
483                         global_flags.display_timecode_in_stream = true;
484                         break;
485                 case OPTION_TIMECODE_STDOUT:
486                         global_flags.display_timecode_on_stdout = true;
487                         break;
488                 case OPTION_QUICK_CUT_KEYS:
489                         global_flags.enable_quick_cut_keys = true;
490                         break;
491                 case OPTION_10_BIT_INPUT:
492                         global_flags.ten_bit_input = true;
493                         break;
494                 case OPTION_10_BIT_OUTPUT:
495                         global_flags.ten_bit_output = true;
496                         global_flags.x264_video_to_disk = true;
497                         global_flags.x264_video_to_http = true;
498                         global_flags.x264_bit_depth = 10;
499                         break;
500                 case OPTION_INPUT_YCBCR_INTERPRETATION: {
501                         char *ptr = strchr(optarg, ',');
502                         if (ptr == nullptr) {
503                                 fprintf(stderr, "ERROR: Invalid argument '%s' to --input-ycbcr-interpretation (needs a card and an interpretation, separated by comma)\n", optarg);
504                                 exit(1);
505                         }
506                         *ptr = '\0';
507                         const int card_num = atoi(optarg);
508                         if (card_num < 0 || card_num >= MAX_VIDEO_CARDS) {
509                                 fprintf(stderr, "ERROR: Invalid card number %d\n", card_num);
510                                 exit(1);
511                         }
512
513                         YCbCrInterpretation interpretation;
514                         char *interpretation_str = ptr + 1;
515                         ptr = strchr(interpretation_str, ',');
516                         if (ptr != nullptr) {
517                                 *ptr = '\0';
518                                 const char *range = ptr + 1;
519                                 if (strcmp(range, "full") == 0) {
520                                         interpretation.full_range = true;
521                                 } else if (strcmp(range, "limited") == 0) {
522                                         interpretation.full_range = false;
523                                 } else {
524                                         fprintf(stderr, "ERROR: Invalid Y'CbCr range '%s' (must be “full” or “limited”)\n", range);
525                                         exit(1);
526                                 }
527                         }
528
529                         if (strcmp(interpretation_str, "rec601") == 0) {
530                                 interpretation.ycbcr_coefficients_auto = false;
531                                 interpretation.ycbcr_coefficients = movit::YCBCR_REC_601;
532                         } else if (strcmp(interpretation_str, "rec709") == 0) {
533                                 interpretation.ycbcr_coefficients_auto = false;
534                                 interpretation.ycbcr_coefficients = movit::YCBCR_REC_709;
535                         } else if (strcmp(interpretation_str, "auto") == 0) {
536                                 interpretation.ycbcr_coefficients_auto = true;
537                                 if (interpretation.full_range) {
538                                         fprintf(stderr, "ERROR: Cannot use “auto” Y'CbCr coefficients with full range\n");
539                                         exit(1);
540                                 }
541                         } else {
542                                 fprintf(stderr, "ERROR: Invalid Y'CbCr coefficients '%s' (must be “rec601”, “rec709” or “auto”)\n", interpretation_str);
543                                 exit(1);
544                         }
545                         global_flags.ycbcr_interpretation[card_num] = interpretation;
546                         break;
547                 }
548                 case OPTION_FULLSCREEN:
549                         global_flags.fullscreen = true;
550                         break;
551                 case OPTION_MJPEG_EXPORT_CARDS: {
552                         if (card_to_mjpeg_stream_export_set) {
553                                 fprintf(stderr, "ERROR: --mjpeg-export-cards given twice\n");
554                                 exit(1);
555                         }
556                         global_flags.card_to_mjpeg_stream_export = parse_mjpeg_export_cards(optarg);    
557                         card_to_mjpeg_stream_export_set = true;
558                         break;
559                 }
560                 case OPTION_HELP:
561                         usage(program);
562                         exit(0);
563                 default:
564                         fprintf(stderr, "Unknown option '%s'\n", argv[option_index]);
565                         fprintf(stderr, "\n");
566                         usage(program);
567                         exit(1);
568                 }
569         }
570
571         if (global_flags.uncompressed_video_to_http &&
572             global_flags.x264_video_to_http) {
573                 fprintf(stderr, "ERROR: --http-uncompressed-video and --http-x264-video are mutually incompatible\n");
574                 exit(1);
575         }
576         if (global_flags.num_cards <= 0) {
577                 fprintf(stderr, "ERROR: --num-cards must be at least 1\n");
578                 exit(1);
579         }
580         if (global_flags.output_card < -1 ||
581             global_flags.output_card >= global_flags.num_cards) {
582                 fprintf(stderr, "ERROR: --output-card points to a nonexistant card\n");
583                 exit(1);
584         }
585         if (global_flags.enable_audio && !global_flags.transcode_audio && global_flags.stream_audio_codec_name.empty()) {
586                 fprintf(stderr, "ERROR: If not transcoding audio, you must specify ahead-of-time what audio codec is in use\n");
587                 fprintf(stderr, "       (using --http-audio-codec).\n");
588                 exit(1);
589         }
590         if (global_flags.x264_speedcontrol) {
591                 if (!global_flags.x264_preset.empty() && global_flags.x264_preset != "faster") {
592                         fprintf(stderr, "WARNING: --x264-preset is overridden by --x264-speedcontrol (implicitly uses \"faster\" as base preset)\n");
593                 }
594                 global_flags.x264_preset = "faster";
595         } else if (global_flags.x264_preset.empty()) {
596                 global_flags.x264_preset = X264_DEFAULT_PRESET;
597         }
598         if (!theme_dirs.empty()) {
599                 global_flags.theme_dirs = theme_dirs;
600         }
601
602         // In reality, we could probably do with any even value (we subsample
603         // by two in some places), but it's better to be on the safe side
604         // wrt. video codecs and such. (I'd set 16 if I could, but 1080 isn't
605         // divisible by 16.)
606         if (global_flags.width <= 0 || (global_flags.width % 8) != 0 ||
607             global_flags.height <= 0 || (global_flags.height % 8) != 0) {
608                 fprintf(stderr, "ERROR: --width and --height must be positive integers divisible by 8\n");
609                 exit(1);
610         }
611
612         for (pair<int, int> mapping : global_flags.default_stream_mapping) {
613                 if (mapping.second >= global_flags.num_cards) {
614                         fprintf(stderr, "ERROR: Signal %d mapped to card %d, which doesn't exist (try adjusting --num-cards)\n",
615                                 mapping.first, mapping.second);
616                         exit(1);
617                 }
618         }
619
620         // Rec. 709 would be the sane thing to do, but it seems many players
621         // just default to BT.601 coefficients no matter what. We _do_ set
622         // the right flags, so that a player that works properly doesn't have
623         // to guess, but it's frequently ignored. See discussions
624         // in e.g. https://trac.ffmpeg.org/ticket/4978; the situation with
625         // browsers is complicated and depends on things like hardware acceleration
626         // (https://bugs.chromium.org/p/chromium/issues/detail?id=333619 for
627         // extensive discussion). VLC generally fixed this as part of 3.0.0
628         // (see e.g. https://github.com/videolan/vlc/commit/bc71288b2e38c07d6921472824b92eef1aa85f7e
629         // and https://github.com/videolan/vlc/commit/c3fc2683a9cde1d42674ebf9935dced05733a215),
630         // but earlier versions were pretty random.
631         //
632         // On the other hand, HDMI/SDI output typically requires Rec. 709 for
633         // HD resolutions (with no way of signaling anything else), which is
634         // a conflicting demand. In this case, we typically let the HDMI/SDI
635         // output win if it is active, but the user can override this.
636         if (output_ycbcr_coefficients == "auto") {
637                 // Essentially: BT.709 if HDMI/SDI output is on, otherwise BT.601.
638                 global_flags.ycbcr_rec709_coefficients = false;
639                 global_flags.ycbcr_auto_coefficients = true;
640         } else if (output_ycbcr_coefficients == "rec709") {
641                 global_flags.ycbcr_rec709_coefficients = true;
642                 global_flags.ycbcr_auto_coefficients = false;
643         } else if (output_ycbcr_coefficients == "rec601") {
644                 global_flags.ycbcr_rec709_coefficients = false;
645                 global_flags.ycbcr_auto_coefficients = false;
646         } else {
647                 fprintf(stderr, "ERROR: --output-ycbcr-coefficients must be “rec601”, “rec709” or “auto”\n");
648                 exit(1);
649         }
650
651         if (global_flags.output_buffer_frames < 0.0f) {
652                 // Actually, even zero probably won't make sense; there is some internal
653                 // delay to the card.
654                 fprintf(stderr, "ERROR: --output-buffer-frames can't be negative.\n");
655                 exit(1);
656         }
657         if (global_flags.output_slop_frames < 0.0f) {
658                 fprintf(stderr, "ERROR: --output-slop-frames can't be negative.\n");
659                 exit(1);
660         }
661         if (global_flags.max_input_queue_frames < 1) {
662                 fprintf(stderr, "ERROR: --max-input-queue-frames must be at least 1.\n");
663                 exit(1);
664         }
665         if (global_flags.max_input_queue_frames > 10) {
666                 fprintf(stderr, "WARNING: --max-input-queue-frames has little effect over 10.\n");
667         }
668
669         if (!isinf(global_flags.x264_crf)) {  // CRF mode is selected.
670                 if (global_flags.x264_bitrate != -1) {
671                         fprintf(stderr, "ERROR: --x264-bitrate and --x264-crf are mutually incompatible.\n");
672                         exit(1);
673                 }
674                 if (global_flags.x264_vbv_max_bitrate != -1 && global_flags.x264_vbv_buffer_size != -1) {
675                         fprintf(stderr, "WARNING: VBV settings are ignored with --x264-crf.\n");
676                 }
677         } else if (global_flags.x264_bitrate == -1) {
678                 global_flags.x264_bitrate = DEFAULT_X264_OUTPUT_BIT_RATE;
679         }
680
681         if (!card_to_mjpeg_stream_export_set) {
682                 // Fill in the default mapping (export all cards, in order).
683                 for (unsigned card_idx = 0; card_idx < unsigned(global_flags.num_cards); ++card_idx) {
684                         global_flags.card_to_mjpeg_stream_export[card_idx] = card_idx;
685                 }
686         }
687 }