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 // Long options that have no corresponding short option.
  15 enum LongOption {
  16         OPTION_HELP = 1000,
  17         OPTION_SLOW_DOWN_INPUT = 1001
  18 };
  19
  20 void usage()
  21 {
  22         fprintf(stderr, "Usage: futatabi [OPTION]... SOURCE_URL\n");
  23         fprintf(stderr, "\n");
  24         fprintf(stderr, "      --help                      print usage information\n");
  25         fprintf(stderr, "      --slow-down-input           slow down input to realtime (default on if no\n");
  26         fprintf(stderr, "                                    source URL given)");
  27         fprintf(stderr, "  -q, --interpolation-quality N   1 = fastest\n");
  28         fprintf(stderr, "                                  2 = default (realtime 720p on fast embedded GPUs)\n");
  29         fprintf(stderr, "                                  3 = good (realtime 720p on GTX 970 or so)\n");
  30         fprintf(stderr, "                                  4 = best (not realtime on any current GPU)\n");
  31         fprintf(stderr, "  -d, --working-directory DIR     where to store frames and database\n");
  32 }
  33
  34 void parse_flags(int argc, char * const argv[])
  35 {
  36         static const option long_options[] = {
  37                 { "help", no_argument, 0, OPTION_HELP },
  38                 { "slow-down-input", no_argument, 0, OPTION_SLOW_DOWN_INPUT },
  39                 { "interpolation-quality", required_argument, 0, 'q' },
  40                 { "working-directory", required_argument, 0, 'd' },
  41                 { 0, 0, 0, 0 }
  42         };
  43         for ( ;; ) {
  44                 int option_index = 0;
  45                 int c = getopt_long(argc, argv, "q:d:", long_options, &option_index);
  46
  47                 if (c == -1) {
  48                         break;
  49                 }
  50                 switch (c) {
  51                 case OPTION_SLOW_DOWN_INPUT:
  52                         global_flags.slow_down_input = true;
  53                         break;
  54                 case 'q':
  55                         global_flags.interpolation_quality = atoi(optarg);
  56                         break;
  57                 case 'd':
  58                         global_flags.working_directory = optarg;
  59                         break;
  60                 case OPTION_HELP:
  61                         usage();
  62                         exit(0);
  63                 default:
  64                         fprintf(stderr, "Unknown option '%s'\n", argv[option_index]);
  65                         fprintf(stderr, "\n");
  66                         usage();
  67                         exit(1);
  68                 }
  69         }
  70
  71         if (global_flags.interpolation_quality < 1 || global_flags.interpolation_quality > 4) {
  72                 fprintf(stderr, "Interpolation quality must be 1, 2, 3 or 4.\n");
  73                 usage();
  74                 exit(1);
  75         }
  76 }