git.sesse.net Git - nageru/blob - 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 void usage()
  15 {
  16         fprintf(stderr, "Usage: nageru [OPTION]...\n");
  17         fprintf(stderr, "\n");
  18         fprintf(stderr, "  -h, --help                      print usage information\n");
  19         fprintf(stderr, "  -c, --num-cards                 set number of input cards, including fake cards (default 2)\n");
  20         fprintf(stderr, "  -C, --num-fake-cards            set number of fake cards (default 0)\n");
  21         fprintf(stderr, "  -t, --theme=FILE                choose theme (default theme.lua)\n");
  22         fprintf(stderr, "  -v, --va-display=SPEC           VA-API device for H.264 encoding\n");
  23         fprintf(stderr, "                                    ($DISPLAY spec or /dev/dri/render* path)\n");
  24         fprintf(stderr, "  -m, --map-signal=SIGNAL,CARD    set a default card mapping (can be given multiple times)\n");
  25         fprintf(stderr, "      --http-uncompressed-video   send uncompressed NV12 video to HTTP clients\n");
  26         fprintf(stderr, "      --http-x264-video           send x264-compressed video to HTTP clients\n");
  27         fprintf(stderr, "      --x264-preset               x264 quality preset (default " X264_DEFAULT_PRESET ")\n");
  28         fprintf(stderr, "      --x264-tune                 x264 tuning (default " X264_DEFAULT_TUNE ", can be blank)\n");
  29         fprintf(stderr, "      --x264-speedcontrol         try to match x264 preset to available CPU speed\n");
  30         fprintf(stderr, "      --x264-speedcontrol-verbose  output speedcontrol debugging statistics\n");
  31         fprintf(stderr, "      --x264-bitrate              x264 bitrate (in kilobit/sec, default %d)\n",
  32                 DEFAULT_X264_OUTPUT_BIT_RATE);
  33         fprintf(stderr, "      --x264-vbv-bufsize          x264 VBV size (in kilobits, 0 = one-frame VBV,\n");
  34         fprintf(stderr, "                                  default: same as --x264-bitrate, that is, one-second VBV)\n");
  35         fprintf(stderr, "      --x264-vbv-max-bitrate      x264 local max bitrate (in kilobit/sec per --vbv-bufsize,\n");
  36         fprintf(stderr, "                                  0 = no limit, default: same as --x264-bitrate, i.e., CBR)\n");
  37         fprintf(stderr, "      --x264-param=NAME[,VALUE]   set any x264 parameter, for fine tuning\n");
  38         fprintf(stderr, "      --http-mux=NAME             mux to use for HTTP streams (default " DEFAULT_STREAM_MUX_NAME ")\n");
  39         fprintf(stderr, "      --http-audio-codec=NAME     audio codec to use for HTTP streams\n");
  40         fprintf(stderr, "                                  (default is to use the same as for the recording)\n");
  41         fprintf(stderr, "      --http-audio-bitrate=KBITS  audio codec bit rate to use for HTTP streams\n");
  42         fprintf(stderr, "                                  (default is %d, ignored unless --http-audio-codec is set)\n",
  43                 DEFAULT_AUDIO_OUTPUT_BIT_RATE / 1000);
  44         fprintf(stderr, "      --http-coarse-timebase      use less timebase for HTTP (recommended for muxers\n");
  45         fprintf(stderr, "                                  that handle large pts poorly, like e.g. MP4)\n");
  46         fprintf(stderr, "      --flat-audio                start with most audio processing turned off\n");
  47         fprintf(stderr, "                                    (can be overridden by e.g. --enable-limiter)\n");
  48         fprintf(stderr, "      --gain-staging=DB           set initial gain staging to the given value\n");
  49         fprintf(stderr, "                                    (--disable-gain-staging-auto)\n");
  50         fprintf(stderr, "      --disable-locut             turn off locut filter (also --enable)\n");
  51         fprintf(stderr, "      --disable-gain-staging-auto  turn off automatic gain staging (also --enable)\n");
  52         fprintf(stderr, "      --disable-compressor        turn off regular compressor (also --enable)\n");
  53         fprintf(stderr, "      --disable-limiter           turn off limiter (also --enable)\n");
  54         fprintf(stderr, "      --disable-makeup-gain-auto  turn off auto-adjustment of final makeup gain (also --enable)\n");
  55         fprintf(stderr, "      --disable-alsa-output       disable audio monitoring via ALSA\n");
  56         fprintf(stderr, "      --no-flush-pbos             do not explicitly signal texture data uploads\n");
  57         fprintf(stderr, "                                    (will give display corruption, but makes it\n");
  58         fprintf(stderr, "                                    possible to run with apitrace in real time)\n");
  59 }
  60
  61 void parse_flags(int argc, char * const argv[])
  62 {
  63         static const option long_options[] = {
  64                 { "help", no_argument, 0, 'h' },
  65                 { "num-cards", required_argument, 0, 'c' },
  66                 { "num-fake-cards", required_argument, 0, 'C' },
  67                 { "theme", required_argument, 0, 't' },
  68                 { "map-signal", required_argument, 0, 'm' },
  69                 { "va-display", required_argument, 0, 1000 },
  70                 { "http-uncompressed-video", no_argument, 0, 1001 },
  71                 { "http-x264-video", no_argument, 0, 1008 },
  72                 { "x264-preset", required_argument, 0, 1009 },
  73                 { "x264-tune", required_argument, 0, 1010 },
  74                 { "x264-speedcontrol", no_argument, 0, 1015 },
  75                 { "x264-speedcontrol-verbose", no_argument, 0, 1016 },
  76                 { "x264-bitrate", required_argument, 0, 1011 },
  77                 { "x264-vbv-bufsize", required_argument, 0, 1012 },
  78                 { "x264-vbv-max-bitrate", required_argument, 0, 1013 },
  79                 { "x264-param", required_argument, 0, 1017 },
  80                 { "http-mux", required_argument, 0, 1004 },
  81                 { "http-coarse-timebase", no_argument, 0, 1005 },
  82                 { "http-audio-codec", required_argument, 0, 1006 },
  83                 { "http-audio-bitrate", required_argument, 0, 1007 },
  84                 { "flat-audio", no_argument, 0, 1002 },
  85                 { "gain-staging", required_argument, 0, 1018 },
  86                 { "disable-locut", no_argument, 0, 1019 },
  87                 { "enable-locut", no_argument, 0, 1020 },
  88                 { "disable-gain-staging-auto", no_argument, 0, 1021 },
  89                 { "enable-gain-staging-auto", no_argument, 0, 1022 },
  90                 { "disable-compressor", no_argument, 0, 1023 },
  91                 { "enable-compressor", no_argument, 0, 1024 },
  92                 { "disable-limiter", no_argument, 0, 1025 },
  93                 { "enable-limiter", no_argument, 0, 1026 },
  94                 { "disable-makeup-gain-auto", no_argument, 0, 1027 },
  95                 { "enable-makeup-gain-auto", no_argument, 0, 1028 },
  96                 { "disable-alsa-output", no_argument, 0, 1014 },
  97                 { "no-flush-pbos", no_argument, 0, 1003 },
  98                 { 0, 0, 0, 0 }
  99         };
 100         for ( ;; ) {
 101                 int option_index = 0;
 102                 int c = getopt_long(argc, argv, "c:C:t:v:m:", long_options, &option_index);
 103
 104                 if (c == -1) {
 105                         break;
 106                 }
 107                 switch (c) {
 108                 case 'c':
 109                         global_flags.num_cards = atoi(optarg);
 110                         break;
 111                 case 'C':
 112                         global_flags.num_fake_cards = atoi(optarg);
 113                         break;
 114                 case 't':
 115                         global_flags.theme_filename = optarg;
 116                         break;
 117                 case 'm': {
 118                         char *ptr = strchr(optarg, ',');
 119                         if (ptr == nullptr) {
 120                                 fprintf(stderr, "ERROR: Invalid argument '%s' to --map-signal (needs a signal and a card number, separated by comma)\n", optarg);
 121                                 exit(1);
 122                         }
 123                         *ptr = '\0';
 124                         const int signal_num = atoi(optarg);
 125                         const int card_num = atoi(ptr + 1);
 126                         if (global_flags.default_stream_mapping.count(signal_num)) {
 127                                 fprintf(stderr, "ERROR: Signal %d already mapped to card %d\n",
 128                                         signal_num, global_flags.default_stream_mapping[signal_num]);
 129                                 exit(1);
 130                         }
 131                         global_flags.default_stream_mapping[signal_num] = card_num;
 132                         break;
 133                 }
 134                 case 1000:
 135                         global_flags.va_display = optarg;
 136                         break;
 137                 case 1001:
 138                         global_flags.uncompressed_video_to_http = true;
 139                         break;
 140                 case 1004:
 141                         global_flags.stream_mux_name = optarg;
 142                         break;
 143                 case 1005:
 144                         global_flags.stream_coarse_timebase = true;
 145                         break;
 146                 case 1006:
 147                         global_flags.stream_audio_codec_name = optarg;
 148                         break;
 149                 case 1007:
 150                         global_flags.stream_audio_codec_bitrate = atoi(optarg) * 1000;
 151                         break;
 152                 case 1008:
 153                         global_flags.x264_video_to_http = true;
 154                         break;
 155                 case 1009:
 156                         global_flags.x264_preset = optarg;
 157                         break;
 158                 case 1010:
 159                         global_flags.x264_tune = optarg;
 160                         break;
 161                 case 1015:
 162                         global_flags.x264_speedcontrol = true;
 163                         break;
 164                 case 1016:
 165                         global_flags.x264_speedcontrol_verbose = true;
 166                         break;
 167                 case 1011:
 168                         global_flags.x264_bitrate = atoi(optarg);
 169                         break;
 170                 case 1012:
 171                         global_flags.x264_vbv_buffer_size = atoi(optarg);
 172                         break;
 173                 case 1013:
 174                         global_flags.x264_vbv_max_bitrate = atoi(optarg);
 175                         break;
 176                 case 1017:
 177                         global_flags.x264_extra_param.push_back(optarg);
 178                         break;
 179                 case 1002:
 180                         // If --flat-audio is given, turn off everything that messes with the sound,
 181                         // except the final makeup gain.
 182                         global_flags.locut_enabled = false;
 183                         global_flags.gain_staging_auto = false;
 184                         global_flags.compressor_enabled = false;
 185                         global_flags.limiter_enabled = false;
 186                         break;
 187                 case 1018:
 188                         global_flags.initial_gain_staging_db = atof(optarg);
 189                         global_flags.gain_staging_auto = false;
 190                         break;
 191                 case 1019:
 192                         global_flags.locut_enabled = false;
 193                         break;
 194                 case 1020:
 195                         global_flags.locut_enabled = true;
 196                         break;
 197                 case 1021:
 198                         global_flags.gain_staging_auto = false;
 199                         break;
 200                 case 1022:
 201                         global_flags.gain_staging_auto = true;
 202                         break;
 203                 case 1023:
 204                         global_flags.compressor_enabled = false;
 205                         break;
 206                 case 1024:
 207                         global_flags.compressor_enabled = true;
 208                         break;
 209                 case 1025:
 210                         global_flags.limiter_enabled = false;
 211                         break;
 212                 case 1026:
 213                         global_flags.limiter_enabled = true;
 214                         break;
 215                 case 1027:
 216                         global_flags.final_makeup_gain_auto = false;
 217                         break;
 218                 case 1028:
 219                         global_flags.final_makeup_gain_auto = true;
 220                         break;
 221                 case 1014:
 222                         global_flags.enable_alsa_output = false;
 223                         break;
 224                 case 1003:
 225                         global_flags.flush_pbos = false;
 226                         break;
 227                 case 'h':
 228                         usage();
 229                         exit(0);
 230                 default:
 231                         fprintf(stderr, "Unknown option '%s'\n", argv[option_index]);
 232                         fprintf(stderr, "\n");
 233                         usage();
 234                         exit(1);
 235                 }
 236         }
 237
 238         if (global_flags.uncompressed_video_to_http &&
 239             global_flags.x264_video_to_http) {
 240                 fprintf(stderr, "ERROR: --http-uncompressed-video and --http-x264-video are mutually incompatible\n");
 241                 exit(1);
 242         }
 243         if (global_flags.num_fake_cards > global_flags.num_cards) {
 244                 fprintf(stderr, "ERROR: More fake cards then total cards makes no sense\n");
 245                 exit(1);
 246         }
 247         if (global_flags.num_cards <= 0) {
 248                 fprintf(stderr, "ERROR: --num-cards must be at least 1\n");
 249                 exit(1);
 250         }
 251         if (global_flags.num_fake_cards < 0) {
 252                 fprintf(stderr, "ERROR: --num-fake-cards cannot be negative\n");
 253                 exit(1);
 254         }
 255         if (global_flags.x264_speedcontrol) {
 256                 if (!global_flags.x264_preset.empty() && global_flags.x264_preset != "faster") {
 257                         fprintf(stderr, "WARNING: --x264-preset is overridden by --x264-speedcontrol (implicitly uses \"faster\" as base preset)\n");
 258                 }
 259                 global_flags.x264_preset = "faster";
 260         } else if (global_flags.x264_preset.empty()) {
 261                 global_flags.x264_preset = X264_DEFAULT_PRESET;
 262         }
 263
 264         for (pair<int, int> mapping : global_flags.default_stream_mapping) {
 265                 if (mapping.second >= global_flags.num_cards) {
 266                         fprintf(stderr, "ERROR: Signal %d mapped to card %d, which doesn't exist (try adjusting --num-cards)\n",
 267                                 mapping.first, mapping.second);
 268                         exit(1);
 269                 }
 270         }
 271 }