MyCHiggsAnalysis/exe/AAN_analysis.cxx

//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 8.11.2006, AUTHOR: OLIVER KORTNER
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

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

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

// standard C++ //
#include <iostream>
#include <fstream>

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

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

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

// user analysis //
//#include "MyAnalysisExample/MyAnalysisExample.h"  //example
//#include "HiggsAnalysis/MyJetAnalysis.h"    // steffen
//#include "MyHtautauAnalysis/MyHtautauAnalysis.h" // manfred
#include "MyZTauTauAnalysis/MyZTauTauAnalysis.h"

// TMVA stuff
//#include "/.at/mdtraid01/mgroh/tmva/3.8.6/TMVA/include/TMVA/TMVA_Dict.h"
//#include "TMVA/Reader.h"
//#include "/.at/mdtraid01/mgroh/tmva/3.8.6/TMVA/include/TMVA/Reader.h"
//#include "TTreeFormula.h"
//#include "TMultiLayerPerceptron.h"
//#include "HiggsAnalysis/Ttbar_Analyse.h"  // thies

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

using namespace std;


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

void help(const char *argv0) {
    
    cerr << "Usage: " << argv0 << " [options] <cut_values file> <infile> <outfile>"<< 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
         << "-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 --
   //----------------------------
        
        cout<<"argc "<<argc<<endl;
        
    extern char *optarg;
    //extern int optind, optopt;
    char c;
    string infile(""), outputpath(""), addon(""), outfile("");
    string settingsFile("");
    bool overwrite(true);
    int  nb_events(-1);
    int first_event(0);

    while((c = getopt(argc, argv, "-hio:a:e:s:")) !=-1)
    {
        //cout << c <<endl;
        switch(c)
        {
        case 'i':
            {
                overwrite = false;
                break;
            }
        case 'o':
            {
                outputpath = optarg;
                break;
            }
        case 'a':
            {
                addon = optarg;
                break;
            }
        case 'e':
            {
                nb_events = atoi(optarg);
                break;
            } 
        case 's':
            {
                first_event = atoi(optarg);
                break;
            } 
        case 1:
            {   
                if(settingsFile==""){
                    settingsFile = optarg;
                }
                                else if(infile==""){
                                        infile = optarg;
                                }
                else{
                    outfile = optarg;
                }
                break;
            }
        default:
            cerr << "Help or unknown argument!" << endl;
            help(argv[0]);
            break;
        }
                
    }
        if(infile=="" || outfile==""){
                cerr<<"Provide a infile and a outfile"<<endl;
                help(argv[0]);
                exit(1);
        }
    
    cout << endl
         << "command line arguments: "     << endl
         << "  outputpath:  " << outputpath << endl
         << "  cut_values:  " << settingsFile << endl
         << "  infile:      " << infile     << endl
         << "  outfile:     " << outfile    << endl
         << "  addon:       " << addon      << endl
         << "  nb_events:   " << nb_events  << endl 
         << "  first event: " << first_event  << endl 
         << "  overwrite:   " << overwrite  << endl
         << endl;
   
   
   //---------------
   //-- VARIABLES --
   //---------------
   
   
   //TSystem *gSystem=new TSystem;
   //gSystem->Load("libAnalysisTools.so");
   
   MyZTauTauAnalysis my_analysis; 


   string sdummy; // auxiliary string
   vector<string> filename; // vector containing the names of the 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 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");
    TTree* Fasttree = (TTree*)testfile.Get("FastTree");
   
    bool CBNT_flag = false;
   
    if(Colltree!=NULL && CBNTtree==NULL && Fasttree==NULL){
        CBNT_flag = false;
    }
    else if(Colltree!=NULL && CBNTtree==NULL  && Fasttree!=NULL){
        CBNT_flag = true;
    }
    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   = outfile;
    my_analysis.settingsFilename   = settingsFile;
        //TOstring->GetString();
//    my_analysis.outputpath = TString(outputpath);
//    my_analysis.addon      = TString(addon);
//    my_analysis.overwrite  = overwrite;

    if(!CBNT_flag){
        cout<<"NtupleReader "<<endl;
        NtupleReader reader(filename, string("CollectionTree"), &my_analysis);
        reader.event_loop(nb_events, first_event);
    }
//    else{
//        cout<<"NtupleReader "<<endl;
//        NtupleReaderFast reader(filename, string("FastTree"),  &my_analysis);
//        reader.event_loop(nb_events, first_event);
//    }
      
   
    return 0;
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   
   // 
//    
//    
//    
//    
//    bool multifiles;
//    multifiles = true;
//    
//    if (!multifiles){
//        filename.push_back(argv[1]);
//    }
//    
//    if (multifiles){  
//      ifstream settings_file(argv[1]); // file containing the names of the
//                                      // AAN files
//      if (settings_file.fail()) {
//              cerr << endl
//                      << "Could not open settings file " << argv[1] << "."
//                      << 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);
//      }
//    }
//    //----------------------------------------------------------
//    //-- INITIALIZE THE EVENT READER AND LOOP OVER THE EVENTS --
//    //----------------------------------------------------------
// 
//    //my_analysis.filename = filename[0].c_str();
//    my_analysis.filename = argv[2];
// 
//  // fuer ntuple vom NtupleWriter
//    NtupleReader reader(filename, string("CollectionTree"), &my_analysis);
//  
//  // fuer CBNTs von Makis
//  //  NtupleReaderFast reader(filename, string("CBNT/t3333"),  &my_analysis);
// 
//  // fuer die selbst geschriebenen:
//  //  NtupleReaderFast reader(filename, string("CBNT/T3333"),  &my_analysis);
//    
//    reader.event_loop();
//    
//    return 0;
//    //-----------------------------
//    //-- CHECK FOR CORRECT USAGE --
//    //-----------------------------
// 
//    if (argc!=3) {
//       cerr << endl
//         << "Incorrect usage!\n"
//         << "Correct usage:\n"
//         << "AAN_analysis.exe <file of input files> <outputfilename>\n";
//       return 1;
//    }
   
}

---