git.sesse.net Git - nageru/blob - futatabi/flags.cpp

   1 #include "flags.h"
   2
   3 #include <getopt.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <utility>
   8
   9 using namespace std;
  10
  11 Flags global_flags;
  12 int flow_initialized_interpolation_quality;
  13
  14 // Long options that have no corresponding short option.
  15 enum LongOption {
  16         OPTION_HELP = 1000,
  17         OPTION_SLOW_DOWN_INPUT = 1001,
  18         OPTION_HTTP_PORT = 1002,
  19         OPTION_TALLY_URL = 1003,
  20         OPTION_CUE_IN_POINT_PADDING = 1004,
  21         OPTION_CUE_OUT_POINT_PADDING = 1005,
  22         OPTION_MIDI_MAPPING = 1006
  23 };
  24
  25 void usage()
  26 {
  27         fprintf(stderr, "Usage: futatabi [OPTION]... SOURCE_URL\n");
  28         fprintf(stderr, "\n");
  29         fprintf(stderr, "      --help                      print usage information\n");
  30         fprintf(stderr, "  -w, --width                     output width in pixels (default 1280)\n");
  31         fprintf(stderr, "  -h, --height                    output height in pixels (default 720)\n");
  32         fprintf(stderr, "  -r, --frame-rate NUM[/NUM]      output frame rate, as a float or fraction\n");
  33         fprintf(stderr, "                                    (default 60000/1001 ~= 59.94)\n");
  34         fprintf(stderr, "      --slow-down-input           slow down input to realtime (default on if no\n");
  35         fprintf(stderr, "                                    source URL given)\n");
  36         fprintf(stderr, "  -q, --interpolation-quality N   0 = off\n");
  37         fprintf(stderr, "                                  1 = fastest\n");
  38         fprintf(stderr, "                                  2 = default (realtime 720p on fast embedded GPUs)\n");
  39         fprintf(stderr, "                                  3 = good (realtime 720p on GTX 970 or so)\n");
  40         fprintf(stderr, "                                  4 = best (not realtime on any current GPU)\n");
  41         fprintf(stderr, "      --cue-in-point-padding SECS   move cue-in N seconds earlier on set\n");
  42         fprintf(stderr, "      --cue-out-point-padding SECS  move cue-out N seconds later on set\n");
  43         fprintf(stderr, "  -d, --working-directory DIR     where to store frames and database\n");
  44         fprintf(stderr, "      --http-port PORT            which port to listen on for output\n");
  45         fprintf(stderr, "      --tally-url URL             URL to get tally color from (polled every 100 ms)\n");
  46         fprintf(stderr, "      --midi-mapping=FILE         start with the given MIDI controller mapping\n");
  47 }
  48
  49 void parse_flags(int argc, char *const argv[])
  50 {
  51         static const option long_options[] = {
  52                 { "help", no_argument, 0, OPTION_HELP },
  53                 { "width", required_argument, 0, 'w' },
  54                 { "height", required_argument, 0, 'h' },
  55                 { "frame-rate", required_argument, 0, 'r' },
  56                 { "slow-down-input", no_argument, 0, OPTION_SLOW_DOWN_INPUT },
  57                 { "interpolation-quality", required_argument, 0, 'q' },
  58                 { "working-directory", required_argument, 0, 'd' },
  59                 { "http-port", required_argument, 0, OPTION_HTTP_PORT },
  60                 { "tally-url", required_argument, 0, OPTION_TALLY_URL },
  61                 { "cue-in-point-padding", required_argument, 0, OPTION_CUE_IN_POINT_PADDING },
  62                 { "cue-out-point-padding", required_argument, 0, OPTION_CUE_OUT_POINT_PADDING },
  63                 { "midi-mapping", required_argument, 0, OPTION_MIDI_MAPPING },
  64                 { 0, 0, 0, 0 }
  65         };
  66         for (;;) {
  67                 int option_index = 0;
  68                 int c = getopt_long(argc, argv, "w:h:r:q:d:", long_options, &option_index);
  69
  70                 if (c == -1) {
  71                         break;
  72                 }
  73                 switch (c) {
  74                 case 'w':
  75                         global_flags.width = atoi(optarg);
  76                         break;
  77                 case 'h':
  78                         global_flags.height = atoi(optarg);
  79                         break;
  80                 case 'r': {
  81                         double num, den;
  82                         if (sscanf(optarg, "%lf/%lf", &num, &den) == 2) {
  83                                 global_flags.output_framerate = num / den;
  84                         } else if (sscanf(optarg, "%lf", &num) == 1) {
  85                                 global_flags.output_framerate = num;
  86                         } else {
  87                                 fprintf(stderr, "Invalid frame rate given (must be on the form N or N/M)\n");
  88                                 exit(1);
  89                         }
  90                         break;
  91                 }
  92                 case OPTION_SLOW_DOWN_INPUT:
  93                         global_flags.slow_down_input = true;
  94                         break;
  95                 case 'q':
  96                         global_flags.interpolation_quality = atoi(optarg);
  97                         global_flags.interpolation_quality_set = true;
  98                         break;
  99                 case 'd':
 100                         global_flags.working_directory = optarg;
 101                         break;
 102                 case OPTION_HTTP_PORT:
 103                         global_flags.http_port = atoi(optarg);
 104                         break;
 105                 case OPTION_TALLY_URL:
 106                         global_flags.tally_url = optarg;
 107                         break;
 108                 case OPTION_CUE_IN_POINT_PADDING:
 109                         global_flags.cue_in_point_padding_seconds = atof(optarg);
 110                         global_flags.cue_in_point_padding_set = true;
 111                         break;
 112                 case OPTION_CUE_OUT_POINT_PADDING:
 113                         global_flags.cue_out_point_padding_seconds = atof(optarg);
 114                         global_flags.cue_out_point_padding_set = true;
 115                         break;
 116                 case OPTION_MIDI_MAPPING:
 117                         global_flags.midi_mapping_filename = optarg;
 118                         break;
 119                 case OPTION_HELP:
 120                         usage();
 121                         exit(0);
 122                 default:
 123                         fprintf(stderr, "Unknown option '%s'\n", argv[option_index]);
 124                         fprintf(stderr, "\n");
 125                         usage();
 126                         exit(1);
 127                 }
 128         }
 129
 130         if (global_flags.interpolation_quality < 0 || global_flags.interpolation_quality > 4) {
 131                 fprintf(stderr, "Interpolation quality must be 0, 1, 2, 3 or 4.\n");
 132                 usage();
 133                 exit(1);
 134         }
 135         if (global_flags.cue_in_point_padding_seconds < 0.0 ||
 136             global_flags.cue_out_point_padding_seconds < 0.0) {
 137                 fprintf(stderr, "Cue point padding cannot be negative.\n");
 138                 usage();
 139                 exit(1);
 140         }
 141 }