CmdLine.cxx

Go to the documentation of this file.

// $Id: CmdLine.cxx,v 1.2 2008/08/29 22:18:11 fmwk Exp $
//
//
// messier@indiana.edu
#include "JobControl/CmdLine.h"
#include <iostream>
#include <climits>
extern "C" {
#include "getopt.h"
}
#include "IoModules/ReadWriteModule.h"
#include "IoModules/EventFilter.h"
using namespace jobc;

CmdLine* CmdLine::fInstance = 0;

void CmdLine::Usage() const
{
  std::cerr <<
    "usage: ana [options] file1.root file2.root ...\n" <<
    "options are:\n" <<
    "  -i file.root,--input=file.root  : Add input data file\n"
    "  -o file.root,--output=file.root : Set output data file \n"
    "  -g file.root,--histo=file.root  : Set output histogram file"
    " (0 for no file)\n"
    "  -n #, --nevt=#                  : Set number of events to process\n"
    "  -s #, --nskip=#                 : Skip number of events at start\n"
    "  -r #, --run=#                   : Skip to run number at start\n"
    "  -e #, --event=#                 : Skip to event number at start\n"
    "  -x #, --xml=file.xml            : Parse XML file\n"
    "  -p #, --xmlpath=dir             : Add directory to XML serach path\n"
    "  -d #, --xmldir=dir              : Read all XML files in directory\n"
    "  -m #, --memory-report=#         : Summarize resources used energy n events\n"
    "  -l file.txt,--evtent-list=file  : Read event list from text file\n"
    "  -L #, --size-limit=#            : Largest acceptable file size (in MB)\n"
    "  -D #, --dumpcore                : If 0 stop cleanly, if 1 dump core on assert_jobc's\n";
}


//......................................................................

CmdLine::CmdLine() :
  fOutFileName(""),
  fFirstRun   ( -1 ),
  fFirstEvent ( -1 ),
  fNevt       ( INT_MAX ),
  fNskip      ( 0  ),
  fNmemReport (-1  ),
  fSIGHUP     (false),
  fDumpOnException(0),
  fSizeLimit  ( 0 )
{ }

//......................................................................

CmdLine& CmdLine::Instance()
{
  if (fInstance==0) fInstance = new CmdLine();
  return *fInstance;
}

//......................................................................

void CmdLine::Parse(int argc, char** argv) 
{
  static const int kXMLDirOpt    = 'd';
  static const int kEvtOpt       = 'e';
  static const int kHFileOpt     = 'g';
  static const int kHelpOpt      = 'h';
  static const int kInputOpt     = 'i';
  static const int kNevtOpt      = 'n';
  static const int kOutputOpt    = 'o';
  static const int kXMLPathOpt   = 'p';
  static const int kRunOpt       = 'r';
  static const int kNskipOpt     = 's';
  static const int kXMLOpt       = 'x';
  static const int kMemReportOpt = 'm';
  static const int kEvtListOpt   = 'l';
  static const int kSizeLimitOpt = 'L';
  static const int kCoreDumpOpt  = 'D';
  static struct option long_options[] = {
    {"input",        1, 0, kInputOpt},
    {"output",       1, 0, kOutputOpt},
    {"help",         0, 0, kHelpOpt},
    {"histo",        1, 0, kHFileOpt},
    {"nevt",         1, 0, kNevtOpt},
    {"nskip",        1, 0, kNskipOpt},
    {"run",          1, 0, kRunOpt},
    {"evt",          1, 0, kEvtOpt},
    {"xml",          1, 0, kXMLOpt},
    {"xmlpath",      1, 0, kXMLPathOpt},
    {"xmldir",       1, 0, kXMLDirOpt},
    {"memory-report",1, 0, kMemReportOpt},
    {"event-list",   1, 0, kEvtListOpt},
    {"size-limit",   1, 0, kSizeLimitOpt},
    {"dump",         1, 0, kCoreDumpOpt},
    {0,0,0,0}
  };
  while (1) {
    int c;
    int optindx;
    c = getopt_long(argc, argv, "hg:i:o:n:s:r:e:x:p:d:m:l:L:D:", 
                    long_options, &optindx);
    if (c==-1) break;
    switch (c) {
    case kInputOpt:     fInFileList.push_back(std::string(optarg));  break;
    case kOutputOpt:    fOutFileName     = optarg;                   break;
    case kHelpOpt:      this->Usage(); exit(0);                      break;
    case kHFileOpt:     fHistoFileName   = optarg;                   break;
    case kNevtOpt:      fNevt            = atoi(optarg);             break;
    case kNskipOpt:     fNskip           = atoi(optarg);             break;
    case kRunOpt:       fFirstRun        = atoi(optarg);             break;
    case kEvtOpt:       fFirstEvent      = atoi(optarg);             break;
    case kXMLOpt:       fXMLFileList.push_back(std::string(optarg)); break;
    case kXMLPathOpt:   fXMLPath         = std::string(optarg);      break;
    case kXMLDirOpt:    fXMLDir.push_back(std::string(optarg));      break;
    case kMemReportOpt: fNmemReport      = atoi(optarg);             break;
    case kEvtListOpt:   fEvtList.push_back(std::string(optarg));     break;
    case kSizeLimitOpt: fSizeLimit       = atoi(optarg);             break;
    case kCoreDumpOpt:  fDumpOnException = atoi(optarg) + 1;         break;
    default: 
      std::cerr << "Unknown option: " << optind << argv[optind] << std::endl;
      this->Usage();
      exit(1);
    }
  }
  for (; optind<argc; ++optind) {
    fInFileList.push_back(std::string(argv[optind]));
  }
};

//......................................................................

void CmdLine::Print() const 
{
  std::cerr <<
    "Command line options:" << std::endl   <<
    "  fFirstRun   = "      << fFirstRun   << std::endl <<
    "  fFirstEvent = "      << fFirstEvent << std::endl <<
    "  fNevt       = "      << fNevt       << std::endl <<
    "  fNskip      = "      << fNskip      << std::endl <<
    "  fNmemReport = "      << fNmemReport << std::endl;

  std::list<std::string>::const_iterator itr;
  std::list<std::string>::const_iterator itrEnd;
  
  std::cerr << "  Input files = " << std::endl;
  itr    = fInFileList.begin();
  itrEnd = fInFileList.end();
  for (; itr!=itrEnd; ++itr) std::cerr << "    " << (*itr) << std::endl;

  std::cerr << "  Output file = " << fOutFileName << std::endl;
  std::cerr << "  XML path  = " << fXMLPath << std::endl;

  std::cerr << "  XML files = " << std::endl;
  itr    = fXMLFileList.begin();
  itrEnd = fXMLFileList.end();
  for (; itr!=itrEnd; ++itr) std::cerr << "    " << (*itr) << std::endl;

  std::cerr << "  XML dirs  = " << std::endl;
  itr    = fXMLDir.begin();
  itrEnd = fXMLDir.end();
  for (; itr!=itrEnd; ++itr) std::cerr << "    " << (*itr) << std::endl;
}

//......................................................................

void CmdLine::SetupIoModule(io::ReadWriteModule& iomod, int* nevent) const 
{
//======================================================================
// Setup and I/O module give parameters specified on the command line
//======================================================================
  const std::list<std::string>& fileList = this->InFileList();
  std::list<std::string>::const_iterator itr(fileList.begin());
  std::list<std::string>::const_iterator itrEnd(fileList.end());
  for (; itr!=itrEnd; ++itr) iomod.AddFile(itr->c_str());

  // Set output file limit
  iomod.SetOutSizeLimit(fSizeLimit);

  // Add input file filters
  const std::list<std::string>& evtList = this->EvtListFiles();
  itr    = evtList.begin();
  itrEnd = evtList.end();
  for (; itr!=itrEnd; ++itr) {
    io::EventFilter* iof = new io::EventFilter();
    iof->ReadEventFile(itr->c_str());
    iomod.AdoptInputFilter(iof);
  }

  // Setup the output streams
  std::string outfile = this->OutFileName();
  if (outfile!="") iomod.SetOutputFileName(outfile.c_str());

  int firstRun   = this->FirstRun();
  int firstEvent = this->FirstEvent();
  int nevt       = (*nevent);
  if (nevt==-1) nevt = this->Nevt(); // Fall back on command line
  if (nevt==-1) nevt = INT_MAX;    // Then fall back on all events
  int nskip      = this->NSkip();
  if (firstRun>=0) {
    if (firstEvent>=0) {
      iomod.GoTo(firstRun,firstEvent);
    }
    else {
      iomod.GoTo(firstRun,0);
    }
  }
  else {
    if (firstEvent>=0) {
      iomod.GoTo(iomod.RunNumber(),firstEvent);
    }
  }
  if (nskip>0) iomod.Advance(nskip);
  *nevent = nevt;
}

//......................................................................
void CmdLine::SetSIGHUP()
{
  if (fSIGHUP) {
    std::cout << "jobc::CmdLine: aborting anamipp!" << std::endl;
    abort();
  }
  std::cout << "Stopping job..."
            << "This may take a minute..." << std::endl;
  fSIGHUP = true; 
}

---