MyHtoWWAnalysis/exe/AAN_analysis.cxx

//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 03/2007, AUTHOR: STEFFEN KAISER
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

//::::::::::::::::::::::::::::::::::::::::::::::::::::
//:: MAIN PROGRAMME TO RUN THE ANALYSIS OF THE AAN. ::
//::::::::::::::::::::::::::::::::::::::::::::::::::::

//:::::::::::::::::
//:: HEADER FILE ::
//:::::::::::::::::

// standard C++ //
#include <iostream>
#include <fstream>
#include "unistd.h"

// STL //
#include <string>
#include <vector>

// AAN reader //
#include "NtupleReader/NtupleReader.h"
//#include "NtupleReader/NtupleReaderFast.h"

// ROOT //
#include "TSystem.h"
#include "TObjString.h"

// user analysis //
#include "MyHtoWWAnalysis/MyHtoWWAnalysis.h"

//::::::::::::::::::::::::
//:: NAMESPACE SETTINGS ::
//::::::::::::::::::::::::

using namespace std;


//:::::::::::::::::
//:: METHOD help ::
//:::::::::::::::::

void help(const char *argv0) {
    
    cerr << "Usage: " << argv0 << " [options] <infile>"                 << endl
         << "       <infile> can be <textfile>.txt or <rootfile>.root " << endl
         << "Options:"                                                  << endl
         << "-o <path>:\tWrite outputfile in directory <path>"          << endl
         << "-a <addon>:\tWrite outputfile as filename_<addon>.root"    << endl
         << "-d <decay>:\tRun analysis in the decay mode <decay>"       << endl
         << "-c <v1=x,v2=y>:\tUse these cut values "                    << endl
         << "-f <file>:\tUse cut values defined in <file>"              << endl
         << "-e <events>:\tRun analysis on <events> events"             << endl
         << "-s <events>:\tSkip the first <events> events"              << endl
         << "-i:\t\tQuit when outputfile already exits"                 << endl
         << "-h:\t\tShow this message and quit."                        << endl;
    exit(0);
}

//::::::::::
//:: MAIN ::
//::::::::::

int main(int argc, char * argv[]) {

   //----------------------------
   //-- PARSE THE COMMAND LINE --
   //----------------------------

    extern char *optarg;
    //extern int optind, optopt;
    char c;
    string infile(""), outputpath(""), addon(""), decay_mode(""), cutstring(""), cutfile("");
    bool overwrite(true);
    int  nb_events(-1), first_event(0);
    
    while((c = getopt(argc, argv, "-hio:a:d:c:f:e:s:")) !=-1)
    {
        //cout << c <<endl;
        switch(c)
        {
        case 'i':
            {
                overwrite = false;
                break;
            }
        case 'o':
            {
                outputpath = optarg;
                break;
            }
        case 'a':
            {
                addon = optarg;
                break;
            }
        case 'd':
            {
                decay_mode = optarg;
                break;
            }
        case 'c':
            {
                cutstring = optarg;
                break;
            } 
        case 'f':
            {
                cutfile = optarg;
                break;
            } 
        case 'e':
            {
                nb_events = atoi(optarg);
                break;
            } 
        case 's':
            {
                first_event = atoi(optarg);
                break;
            } 
        case 1:
            {
                infile = optarg;
                break;
            }
        default:
            cerr << "Help or unknown argument!" << endl;
            help(argv[0]);
            break;
        }
    }

    if(infile == ""){
        cerr << "Error: Missing input file!" << endl;
        help(argv[0]);
    }
    
    cout << endl
         << "command line arguments: "     << endl
         << "  outputpath:  " << outputpath  << endl
         << "  infile:      " << infile      << endl
         << "  addon:       " << addon       << endl
         << "  decay_mode:  " << decay_mode  << endl
         << "  cutfile:     " << cutfile     << endl
         << "  cutstring:   " << cutstring   << endl
         << "  nb_events:   " << nb_events   << endl 
         << "  first event: " << first_event << endl 
         << "  overwrite:   " << overwrite   << endl
         << endl;
   

    //---------------
    //-- VARIABLES --
    //---------------
   
    //TSystem *gSystem=new TSystem;
    //gSystem->Load("libAnalysisTools.so");
   
    MyHtoWWAnalysis my_analysis;
   

    string sdummy; // auxiliary string
    vector<string> filename; // vector containing names of AAN files


    //-------------------------
    //-- CHECK THE INPUTFILE --
    //-------------------------

    if(TString(infile).Contains(".root")){
       
        filename.push_back(infile);
    }
    else if(TString(infile).Contains(".txt")){
       
        ifstream settings_file(infile.c_str()); // file containing names of AAN files
        if (settings_file.fail()) {
            cerr << endl
                 << "ERROR: Could not open settings file " << infile << endl
                 << endl;
            return 1;
        }
       
        //-------------------------------------
        //-- READ THE NAMES OF THE AAN FILES --
        //-------------------------------------
       
        while (!settings_file.eof()) {
            settings_file >> sdummy;
            if (settings_file.eof()) {
                break;
            }
            filename.push_back(sdummy);
        }
       
    }else{
        cerr << endl
             << "ERROR: could not read inputfile " << infile << "," << endl
             << "       please provide '.root' or '.txt' file." << endl
             << endl;
        return 1;
    }
 

    //------------------------------------------------------------
    //-- CHECK WHETHER THE FILE COMES FROM CBNT OR NTUPLEWRITER --
    //------------------------------------------------------------

    TFile testfile(filename[0].c_str());
    
    TTree* Colltree = (TTree*)testfile.Get("CollectionTree");
    TTree* CBNTtree = (TTree*)testfile.Get("CBNT/T3333");
   
    bool CBNT_flag = false;
   
    if(Colltree!=NULL && CBNTtree==NULL){
        CBNT_flag = false;
    }
    else if(Colltree==NULL && CBNTtree!=NULL){
        CBNT_flag = true;
    }
    else{
        cerr << endl
             << "ERROR: could not read tree" << endl
             << "       please provide 'CollectionTree' or 'CBNT/T3333'." << endl
             << endl;
        return 1;  
    }

    testfile.Close();
   
      
    //----------------------------------------------------------
    //-- INITIALIZE THE EVENT READER AND LOOP OVER THE EVENTS --
    //----------------------------------------------------------
   
    //set the output filename
    //my_analysis.filename = filename[0].c_str();
   
    TObjArray*  TOarray  = ((TString)filename[0].c_str()).Tokenize("/");
    TObjString* TOstring = (TObjString*)TOarray->Last();
     
    my_analysis.filename     = TOstring->GetString();
    my_analysis.outputpath   = TString(outputpath);
    my_analysis.addon        = TString(addon);
    my_analysis.overwrite    = overwrite;
    my_analysis.m_decay_mode = decay_mode;
    my_analysis.m_cutstring  = cutstring;
    my_analysis.m_cutfile    = cutfile;

    if(!CBNT_flag){
        NtupleReader reader(filename, string("CollectionTree"), &my_analysis);
        reader.event_loop(nb_events, first_event);
    }
    else{
        cerr << "NtupleReaderFast has been removed from the Makefile" << endl;
        exit(1);
        //NtupleReaderFast reader(filename, string("CBNT/T3333"),  &my_analysis);
        //reader.event_loop();
    }
      
   
    return 0;
   
}


   //-----------------------------
   //-- CHECK FOR CORRECT USAGE --
   //-----------------------------
 
    // if (argc<2 || argc>3) {
//         //if (argc!=2) {
//         cerr << endl
//              << "Incorrect usage!\n"
//              << "Correct usage:\n"
//              << "AAN_analysis.exe <textfile.txt> [<outputpath>], or\n"
//              << "AAN_analysis.exe <rootfile.root> [<outputpath>]\n" 
//              << endl;
//         return 1;
//     }

---