BfldBenchmark.cc

Go to the documentation of this file.

//
//
// A program to test various methods of interpolating the B field.
//
//

#include <iostream> //System includes
#include <fstream>
#include <unistd.h>
#include <string.h>

#include "TROOT.h" //ROOT includes

#include "MessageService/MsgService.h" //MINOS includes

// --- Constants ---
const char kDefaultAccTestOutFileName[20] = "accresults.root";
const char kDefaultPerfTestOutFileName[20] = "perfresults.bfb";
const char kDefaultConfigFileName[20] = "config.bfb";

const char kProgramOptions[] = "pahdf:"; //To be passed to getopt()

// --- Function Prototypes ---
void BenchmarkPerformance(const char *outFileName,const char *rectMapFileName,
                          const char *vorMapFileName,
                          const char *vorMeshFileName,
                          const char *vorBFldFileName,
                          const char *testPtsFileName);
void BenchmarkAccuracy(const char *outFileName,const char *rectMapFileName,
                       const char *vorMapFileName,const char *vorMeshFileName,
                       const char *vorBFldFileName,
                       const char *rectAccTestPtsFileName,
                       const char *vorAccTestPtsFileName);

// --- Global Variables ---

TROOT root("root1","root1");

// --- main ---

int main(int argc, char** argv) {
  bool isPerf = true;
  bool isAcc = true;

  bool isConfigFileArg = false;
  char configFileName[255];

  bool isDebug = false;

  char opt = getopt(argc,argv,kProgramOptions);
  if(opt == EOF) {
    isDebug = false;
    isConfigFileArg = false;
    isPerf = true;
    isAcc = true;
    cout << "Running both performance and accuracy tests..." << endl;
  }
  else {
    isPerf = false;
    isAcc = false;

    while(opt != EOF) {
      switch(opt) {
      case EOF:
        break;
      case '?':
        cout << "Invalid option. Use BfldBenchmark -h for help." << endl;
        exit(1);
        break;
      case 'h':
        cout << "usage: BfldBenchmark [-pahd] [-f configfile] [[file1] file2]"
             << endl
             << "Benchmarks the BField package." << endl;
        cout << "By default, does both performance and accuracy tests."
             << endl;
        cout << "-p: Do performance tests. Output to file1 or to the" << endl
             << "    default, " << kDefaultPerfTestOutFileName << endl;
        cout << "-a: Do accuracy tests. Can be used together with -p or alone."
             << endl
             << "    If alone, file1 specifies where to write results; "
             << endl << "    otherwise, file2 does. If the appropriate file " 
             << endl << "    is not specified, the default, "
             << kDefaultAccTestOutFileName << ", is used." << endl;
        cout << "-f  configfile: Use the configuration file given instead of"
             << endl
             << "                the default file " << kDefaultConfigFileName
             << endl;
        cout << "-d: Print out lots of debugging information." << endl;
        cout << "-h: Display this help." << endl << endl;
        cout << "Config file format:" << endl
             << "perfTestPtsFileName" << endl
             << "rectAccTestPtsName" << endl
             << "vorAccTestPtsName" << endl
             << "rectMapName (currently ignored)" << endl
             << "vorMapName (currently ignored)" << endl
             << "vorMeshName (currently ignored)" << endl
             << "vorBFldName (currently ignored)" << endl;
        exit(0);
        break;
      case 'p':
        isPerf = true;
        break;
      case 'a':
        isAcc = true;
        break;
      case 'f':
        isConfigFileArg = true;
        if(optarg == 0) {
          cout << "You must specify a configuration file name with" << endl
               << "the -f option. Use the -h option for help." << endl;
          cout << "This run will use the default configuration file." << endl;
          isConfigFileArg = false;
        }
        else
          strcpy(configFileName,optarg);
        break;
      case 'd':
        isDebug = true;
        break;
      default:
        cout << "Error processing arguments. Bail." << endl;
        exit(1);
      }
    
      opt = getopt(argc,argv,kProgramOptions);
    }
  }

  if(isDebug) {
    MsgService::Instance()->GetStream("Bfldtest")->SetLogLevel(Msg::kDebug);
    //We'll also hear from the package itself, but without lots of extra
    //information appended by the MessageService.
    MsgService::Instance()->GetStream("Bfld")->SetFormat(Msg::kInfo,
                                                         Msg::kName);
    MsgService::Instance()->GetStream("Bfld")->SetFormat(Msg::kWarning,
                                                         Msg::kName
                                                         & Msg::kPriority);
    MsgService::Instance()->GetStream("BFL")->SetFormat(Msg::kInfo,
                                                        Msg::kName);
    MsgService::Instance()->GetStream("BFL")->SetFormat(Msg::kWarning,
                                                        Msg::kName
                                                        & Msg::kPriority);
  }
  else {
    //We leave Bfldtest at kInfo level. In addition, we want to quiet
    //chatter from the BField package itself.
    MsgService::Instance()->GetStream("BFL")->SetLogLevel(Msg::kFatal);
    MsgService::Instance()->GetStream("Bfld")->SetLogLevel(Msg::kFatal);
  }
  //We don't want all the information Msg normally prints out.
  MsgService::Instance()->GetStream("Bfldtest")->SetFormat(Msg::kDebug,0);
  MsgService::Instance()->GetStream("Bfldtest")->SetFormat(Msg::kWarning,0);

  if(!isConfigFileArg) { //Use default config file
    strcpy(configFileName,kDefaultConfigFileName);
  }

  ifstream configFile(configFileName,ios::in);
  if(!configFile.good()) {
    cout << "Couldn't open configuration file " << configFileName << endl;
    exit(1);
  }

  char *rectMapName,*vorMapName,*vorMeshName,*vorBFldName,
    *perfTestPtsName,*rectAccTestPtsName,*vorAccTestPtsName;

  configFile.gets(&perfTestPtsName);
  configFile.gets(&rectAccTestPtsName);
  configFile.gets(&vorAccTestPtsName);

  configFile.gets(&rectMapName);
  configFile.gets(&vorMapName);
  configFile.gets(&vorMeshName);
  configFile.gets(&vorBFldName);

  if(isPerf) {
    cout << endl << "Running performance tests with test point file "
         << perfTestPtsName << endl;

    if(argc - optind == 0)
      BenchmarkPerformance(kDefaultPerfTestOutFileName,rectMapName,vorMapName,
                           vorMeshName,vorBFldName,perfTestPtsName);
    else
      BenchmarkPerformance(argv[optind],rectMapName,vorMapName,
                           vorMeshName,vorBFldName,perfTestPtsName);

  }

  if(isAcc) {
    cout << endl << "Running accuracy tests with test point files:" << endl
         << "rect2d:  " << rectAccTestPtsName << endl
         << "Voronoi: " << vorAccTestPtsName << endl;

    int reqArgCount = (isPerf) ? 1 : 0;

    if(argc - optind == reqArgCount)
      BenchmarkAccuracy(kDefaultAccTestOutFileName,rectMapName,vorMapName,
                        vorMeshName,vorBFldName,rectAccTestPtsName,
                        vorAccTestPtsName);
    else
      BenchmarkAccuracy(argv[optind + reqArgCount],rectMapName,vorMapName,
                        vorMeshName,vorBFldName,rectAccTestPtsName,
                        vorAccTestPtsName);
  }
}

---