cli_options.h

Go to the documentation of this file.

#include <TROOT.h>
#include <argp.h>

#ifndef _PARSER_
#define _PARSER_

static int parse_opt(int key, char *arg, struct argp_state *state);

// -- Global variables
struct argp_option options[] = {{"overwrite", 'w', 0, 0, "Overwrite results"},
                                {"verbosity", 'v', "LEVEL", 0, "Verbosity level [Info, Debug, Trace]"},
                                {"input-file", 'i', "FILE", 0, "Input file"},
                                {"output-file", 'o', "FILE", 0, "Output file"},
                                {"event", 'E', "INT", 0, "Process Single Event"},
                                {"from-event", 'e', "INT", 0, "Process from Event"},
                                {"nevents", 'n', "INT", 0, "Process # events"},
                                {0}};
struct argp argp = {options, parse_opt, 0, 0};

struct arguments {
  const char *infilename;
  const char *outfilename;
  const char *recotype;
  const char *debug_level;
  bool flag_overwrite;
  bool flag_input;
  bool flag_output;
  int event;
  int from_event;
  int nevents;
};

static int parse_opt(int key, char *arg, struct argp_state *state) {

  struct arguments *a = (struct arguments *)state->input;

  switch (key) {
  case 'w': {
    a->flag_overwrite = kTRUE;
    break;
  }
  case 'v': {
    a->debug_level = arg;
    break;
  }
  case 'i': {
    a->flag_input = kTRUE;
    a->infilename = arg;
    break;
  }
  case 'o': {
    a->flag_output = kTRUE;
    a->outfilename = arg;
    break;
  }
  case 'E': {
    a->event = atoi(arg);
    break;
  }
  case 'e': {
    a->from_event = atoi(arg);
    break;
  }
  case 'n': {
    a->nevents = atoi(arg);
    break;
  }
  case ARGP_KEY_ARG: {
    argp_failure(state, 1, 0, "No arguments requested");
    break;
  }
  case ARGP_KEY_INIT: {
    a->infilename = nullptr;
    a->outfilename = nullptr;
    a->debug_level = "Info";
    a->flag_overwrite = kFALSE;
    a->flag_input = kFALSE;
    a->flag_output = kFALSE;
    a->event = -1;
    a->from_event = -1;
    a->nevents = -1;
    break;
  }
  case ARGP_KEY_END: {
    break;
  }
  }

  return 0;
}

#endif