MyAnalysisExample/src/MyAnalysisExample.cxx

//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// 09.11.2006, AUTHOR: OLIVER KORTNER
//+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
//:: IMPLEMENTATION OF METHODS DEFINED IN THE CLASS MyAnalysisExample ::
//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::

//::::::::::::::::::
//:: HEADER FILES ::
//::::::::::::::::::

#include "MyAnalysisExample/MyAnalysisExample.h"

//:::::::::::::::::::::::::
//:: NAME SPACE SETTINGS ::
//:::::::::::::::::::::::::

using namespace std;

//*****************************************************************************

//:::::::::::::::::
//:: METHOD init ::
//:::::::::::::::::

void MyAnalysisExample::init(void) {
   
   filename = "testfile";
   return;

}

//*****************************************************************************

//:::::::::::::::::::::
//:: METHOD destruct ::
//:::::::::::::::::::::

void MyAnalysisExample::destruct(void) {

        return;

}

//*****************************************************************************

//:::::::::::::::::::::::
//:: METHOD initialize ::
//:::::::::::::::::::::::

void MyAnalysisExample::initialize(void) {

// CREATE THE ROOT FILE //

   //m_tfile = new TFile("tmp.root", "RECREATE");
   m_tfile = new TFile( filename.ReplaceAll(".aan",".selection") , "RECREATE");


// BOOK THE HISTOGRAM //

        m_nb_electrons = new TH1F("nb_electrons",
                                "NUMBER OF RECONSTRUCTED ELECTRONS / EVENT",
                                11, -0.5, 10.5);
        m_e_energy = new TH1F("e_energy", "ELECTRON ENERGY (GeV)", 
                                201, -0.5, 200.5);
        m_e_pt = new TH1F("e_pt", "ELECTRON PT (GeV)", 
                                201, -0.5, 200.5);
        m_nb_muons = new TH1F("nb_muons",
                                "NUMBER OF RECONSTRUCTED MUONS / EVENT",
                                11, -0.5, 10.5);
        m_nb_jets = new TH1F("nb_jets",
                                "NUMBER OF RECONSTRUCTED JETS / EVENT",
                                11, -0.5, 10.5);
        
        m_missinget_x = new TH1F("missinget_x",
                                "MISSING ET IN X DIRECTION",
                                1001, -500.5, 500.5);

        h_nb_matchedjets = new TH1F("nb_matchedjets",
                                  "NUMBER OF MATCHED JETS / EVENT",
                                  11, -0.5, 10.5);

        h_nb_fakedjets = new TH1F("nb_fakedjets",
                                  "NUMBER OF FAKED JETS / EVENT",
                                  11, -0.5, 10.5);



        return;

}

//*****************************************************************************

//::::::::::::::::::::::::::
//:: METHOD analyse_event ::
//::::::::::::::::::::::::::

void MyAnalysisExample::analyse_event(const MyEvent & event) {

// FILL HISTOGRAMS //

    // number of reconstructed electrons, muons, and jets //
    m_nb_electrons->Fill(event.reco_electrons().size(), 1.0);
    m_nb_muons->Fill(event.reco_muons().size(), 1.0);
    m_nb_jets->Fill(event.reco_jets().size(), 1.0);
    
    // electron energy distribution //
        //for (unsigned int k=0; k<event.reco_electrons().size(); k++) {
        //m_e_energy->Fill(0.001*
        //              event.reco_electrons()[k]->tlv()[3],
    //                                                  1.0);
                // get index in original vector of electrons
                //cout<<"elctron index: "
                //  <<event.reco_electrons()[k]->index()<<endl;
                // get type
                //cout<<"elctron type: "
                //  <<event.reco_electrons()[k]->type()<<endl;
                // get charge
                //cout<<"elctron charge: "
                //  <<event.reco_electrons()[k]->charge()<<endl;
//}
        
// electron pt distribution //
    for (unsigned int k=0; k<event.reco_electrons().size(); k++) {
        m_e_pt->Fill(event.reco_electrons()[k]->tlv().Pt(),1.0);
    }
    
// missing et in x direction
    //cout <<"missinget_x: "<< event.missinget().missinget_x() << endl; 
    
    m_missinget_x->Fill(event.met_manager().get_missinget("MET_RefFinal").met_x(),1.0);
    
// matching between reco and truth jets //

    int nb_matchedjets = 0;
    int nb_fakedjets = 0;

    std::vector<MyParticle*> 
        selected_truthjets = select_jets( event.truth_jets() );

    for (unsigned int k=0; k<event.reco_jets().size(); k++) {
        
        MyParticle* tmp_jet = NULL;
         
        tmp_jet = 
            (MyParticle*)m_tools.find_matching_particle( event.reco_jets()[k], 
                                                         selected_truthjets
                                                         );
        
        if(tmp_jet!=NULL) nb_matchedjets++;
        else              nb_fakedjets++;
        
    }

    h_nb_matchedjets->Fill(nb_matchedjets);
    h_nb_fakedjets->Fill(nb_fakedjets);

    
    //for (unsigned int k=0; k<selected_truthjets.size(); k++){
    //cout << "matchingflag: " << selected_truthjets[k]->matchingflag() << endl;
    //}

   
    //m_tools.test();
    
   return;

}

//*****************************************************************************

//::::::::::::::::::::::::
//:: METHOD select_jets ::
//::::::::::::::::::::::::

std::vector<MyParticle*>
MyAnalysisExample::select_jets( std::vector<MyParticle*> jet_vector){

  std::vector<MyParticle*> selected_jets(0);
  
  for (unsigned int k=0; k<jet_vector.size(); k++) {
  
      //if( !(jet_vector[k]->tlv().Pt() > cutvalue_jet_pt_min) )      continue;

      selected_jets.push_back( jet_vector[k] );
      
      
  }
  
  return selected_jets;
  
}



//*****************************************************************************

//::::::::::::::::::::::::::::
//:: METHOD end_of_analysis ::
//::::::::::::::::::::::::::::

void MyAnalysisExample::end_of_analysis(void) {

    //----------------------
    //-- MULTIPLE MATCHES --
    //----------------------
    
    Int_t multiplematches1 = m_tools.nb_multiple_matches();
    m_tools.clear_nb_multiple_matches();
    Int_t multiplematches2 = m_tools.nb_multiple_matches();
    
    cout << "multiplematches1: " <<  multiplematches1 << endl;
    cout << "multiplematches2: " <<  multiplematches2 << endl;

        
    //------------------------
    //-- SAVE THE ROOT FILE --
    //------------------------
    
    m_tfile->Write();
    
    return;
    
}

---