rootana/src/framework/CommandLine.h File Reference

#include <string>

Go to the source code of this file.

Classes
struct	ARGUMENTS
Functions
void	help_command_line (const char *my_name)
void	print_arguments (const ARGUMENTS &args)
bool	isNumber (const char *c)
int	GetRunNumber (const std::string &file_name)
	Obtains the run number from a file name.
int	check_arguments (ARGUMENTS &arguments)
int	analyze_command_line (int argc, char **argv, ARGUMENTS &arguments)
int	load_config_file (const char *filename)

---

Function Documentation

int analyze_command_line	(	int	argc,
		char **	argv,
		ARGUMENTS &	arguments
	)

Definition at line 118 of file CommandLine.cpp.

References ARGUMENTS::calib, check_arguments(), ARGUMENTS::correction_file, help_command_line(), ARGUMENTS::infile, isNumber(), ARGUMENTS::mod_file, ARGUMENTS::outfile, ARGUMENTS::run, ARGUMENTS::start, and ARGUMENTS::stop.

Referenced by main().

{
  // Inialise the arguments to be consistently meaningless
  arguments.infile="";
  arguments.outfile="";
  arguments.correction_file="";
  arguments.start=0;
  arguments.stop=0;
  arguments.run=-1;
  arguments.calib=false;

  // Now loop over all the arguments
  // There are a minimum of seven arguments:
  // -i, -m, -o, and corresponding files, and
  // program name.
  if(argc < 7 || std::string(argv[1]) == "--help") {
    help_command_line(argv[0]);
    return 1;
  }

  for(int i=1; i<argc; /* incrementing of i done in the loop! */){
    if(argv[i][0] != '-'){
      std::cerr << "ERROR: Wrong argument " << argv[i] << std::endl;
      help_command_line(argv[0]);
      return 1;
    }
    
   if(strlen(&argv[i][1]) != 1){ 
     std::cerr << "ERROR: All options must be single characters, "
               << "separated with a space and prefixed with a '-'"
               << std::endl;
      help_command_line(argv[0]);
      return 1;
   }

   switch(argv[i][1]){
   case 'm':
     if(i+1 < argc){
       arguments.mod_file = argv[i+1];
       i+=2;
     }
     else {
       std::cerr << "ERROR: No argument for module file specified\n";
       help_command_line(argv[0]);   return 1;
     }
     break;

     //----------
   case 'i':
     if(i+1 < argc){
       arguments.infile = argv[i+1];
       i+=2;
     }
     else{
       std::cerr << "ERROR: No argument for input file specified\n";
       help_command_line(argv[0]);   return 1;
     }
     break;
     
     //----------
   case 'o':
     if(i+1 < argc){
       arguments.outfile = argv[i+1];
       i+=2;
     }
     else{
       std::cerr << "ERROR: No argument for input file specified\n";
       help_command_line(argv[0]);   return 1;
     }
     break;

     //----------
   case 's':
     if(i+1 < argc){
       arguments.correction_file = argv[i+1];
       i+=2;
     }
     else{
       std::cerr << "ERROR: No argument for TSetupData correction file specified\n";
       help_command_line(argv[0]);   return 1;
     }
     break;

     //----------
   case 'n': 
     {
       //detemine how many integer arguments are associated with the -n
       int nArgs = 0;
       for (int j = 1; j < argc-i; ++j) {
         if ( !isNumber(argv[i+j]) ) break;
         ++nArgs;
       }
       if ( nArgs > 2 || nArgs < 1) {
         std::cerr << "ERROR: " << nArgs 
                   << " non-negative integer arguments passed to"
                   << " -n option, which accepts only one or two";
         help_command_line(argv[0]);   return 1;        
       }
       arguments.start = (nArgs==2) ? atoi(argv[i+1]) : 0; 
       arguments.stop = (nArgs==2) ? atoi(argv[i+2]) : atoi(argv[i+1]);
       i += (nArgs+1);
     }
     break;
     //----------
   case 'r':
     if(i+1 < argc){
       if(isNumber(argv[i+1])){
         arguments.run = atoi(argv[i+1]);
         i+=2;
       }
       else{
         std::cerr << "ERROR: Argument " << argv[i+1]
                   << " for option -r is not a number\n";
         help_command_line(argv[0]);   return 1;
       }
     }
     break;
   case 'c':
     arguments.calib = true;
     i+=1;
     break;
     //----------
   default:
     std::cerr << "ERROR: Argument " << argv[i] << " not recognized\n";
     help_command_line(argv[0]);   return 1;
   } // End switch block
  } // End for loop over all args

  // Everything looks ok, so we now check the arguments
  return check_arguments(arguments);
}

int check_arguments

(

ARGUMENTS &

arguments

)

Definition at line 71 of file CommandLine.cpp.

References ARGUMENTS::correction_file, GetRunNumber(), ARGUMENTS::infile, ARGUMENTS::mod_file, ARGUMENTS::outfile, ARGUMENTS::run, ARGUMENTS::start, and ARGUMENTS::stop.

Referenced by analyze_command_line().

                                         {
  if(arguments.stop > 0){
    if (arguments.stop <= arguments.start){
      std::cerr << "ERROR: Cannot process events from "
                << arguments.start << " to " << arguments.stop 
                << " because " << arguments.start << ">=" << arguments.stop 
                << std::endl;
      return 1;
    }
  }
  
  if(arguments.infile.size() == 0){
    std::cerr << "ERROR: Empty input file name."
              <<"  Did you specify the -i option?"  << std::endl;
    return 2;
  }
  
  if(arguments.outfile.size() == 0){
    std::cerr << "ERROR: Empty output file name."
              << "  Did you specify the -o option?\n" << std::endl;
    return 3;
  }  

  if(arguments.run == -1){
    // No run number has been set, obtain it from the filename
    arguments.run = GetRunNumber(arguments.infile);
  }

  if(arguments.mod_file.size() == 0){
    arguments.mod_file = "testModule.txt";
  }

  if(arguments.correction_file.size() == 0){
    std::stringstream buff("");
    buff << "wiremap_corrections/correct" << arguments.run << ".dat";
    arguments.correction_file=buff.str();
  }
  return 0; //success
}

int GetRunNumber

(

const std::string &

file_name

)

Obtains the run number from a file name.

Assumes the file name is of the format aaaaaaSnnnnnn.xxx where: 'a' is any character 'S' is any non-numeric character 'n' is the run number (includes '+' '-' '.') 'x' is any character excluding '.'

Assumes the file name is of the format aaaaaaSnnnnnn.xxx where 'a' is any character 'S' is any non-numeric character 'n' is the run number (includes '+' '-' '.') 'x' is any character excluding '.'

Definition at line 45 of file CommandLine.cpp.

Referenced by check_arguments().

{

  //remove the extension
  std::string name = file_name.substr(0,file_name.find_last_of('.'));

  //find the last numbers
  name = name.substr(name.find_last_not_of("01234566789")+1);

  //Convert the rest
  int number (-1);
  std::stringstream(name) >> number;
  if (number == -1) {
    std::cerr << "Unable to deduce run number from \"" << name 
              << "\" in \"" << file_name << "\"" << std::endl;
    throw std::invalid_argument("not a valid int");
  }
  return number;
}

void help_command_line

(

const char *

my_name

)

Definition at line 12 of file CommandLine.cpp.

Referenced by analyze_command_line().

{
  std::cerr << "\nUsage: " << my_name << "  [options]\n"
            << "    Positional arguments: None\n\n"
            << "Valid options are:\n"
            << "  -i <filename>          Input root tree file.\n"
            << "  -o <filename>          Output root tree file.\n"
            << "  -n <count>             Analyze only <count> events.\n"
            << "  -n <first> <last>      Analyze only events from "
            << "<first> to <last>.\n"
            << "  -r <PSI run number>    Run number specification for the shrubs.\n"
            << "  -s <correction file>   Name of the correction file to be used.\n"
            << "  -m <modules file>      Name of the MODULES file to be used.\n"
            << "  -c                     Tell the setup navigator that this is a calibration\n"
            << "                         run, and it might not find all of the tables it needs\n"
            << "                         in the calibration database\nbut to continue all the\n"
            << "                         same."
            << std::endl;
  return;
}

bool isNumber

(

const char *

c

)

Definition at line 36 of file CommandLine.cpp.

Referenced by analyze_command_line().

{
  for ( std::size_t i=0; i<strlen(c); i++){
    if(!isdigit(c[i])) return false;
  }
  return true;
}

int load_config_file

(

const char *

filename

)

Definition at line 251 of file CommandLine.cpp.

References TemplateFactory< BaseModule, OptionsType >::createModule(), modules::options::DumpOptions(), modules::reader::GetModule(), modules::reader::GetNumModules(), modules::reader::GetOptions(), modules::factory::Instance(), modules::reader::PrintAllOptions(), and modules::reader::ReadFile().

                                          {
  modules::reader modules_file;
  modules_file.ReadFile(filename);
  modules_file.PrintAllOptions();

  modules::factory* mgr = modules::factory::Instance();
  size_t num_modules = modules_file.GetNumModules();
  std::cout << "number of modules requested: " << num_modules << std::endl;
  std::string name;
  modules::options* opts;
  modules::BaseModule** mods=NULL;
  for(unsigned i = 0; i < num_modules; i++){
    name = modules_file.GetModule(i);
    opts = modules_file.GetOptions(i);
    std::cout << "Creating module: " << name
              << "\nWith options:" << std::endl;
    opts->DumpOptions("  ");
    mods[i] = mgr->createModule(name,opts);
  }
  
  return 0;
}

void print_arguments

(

const ARGUMENTS &

args

)

Definition at line 275 of file CommandLine.cpp.

References ARGUMENTS::calib, ARGUMENTS::correction_file, ARGUMENTS::infile, ARGUMENTS::mod_file, ARGUMENTS::outfile, ARGUMENTS::run, ARGUMENTS::start, and ARGUMENTS::stop.

                                           {
  std::cout << "ARGUMENTS struct:"
            << "\n    infile:\t\t" << args.infile
            << "\n    outfile:\t\t" << args.outfile
            << "\n    correction file:\t" << args.correction_file
            << "\n    mod file:\t\t" << args.mod_file
            << "\n    start event:\t" << args.start
            << "\n    stop event:\t\t" << args.stop
            << "\n    run number:\t\t" << args.run
            << "\n    calibration:\t " << args.calib
            << std::endl;
}