RootTreeHandler Class Reference

#include <RootTreeHandler.h>

Public Member Functions
void	readTree (const int maxEvents, const TString &fileName, MuonPairVector *savedPair, const int muonType, std::vector< std::pair< unsigned int, unsigned long long > > *evtRun, MuonPairVector *genPair=nullptr)
void	readTree (const int maxEvents, const TString &fileName, MuonPairExtendedVector *savedPair, const int muonType, std::vector< std::pair< unsigned int, unsigned long long > > *evtRun, MuonPairExtendedVector *genPair=nullptr)
void	readTree (const int maxEvents, const TString &fileName, std::vector< MuonPair > *savedPair, const int muonType, std::vector< GenMuonPair > *genPair=nullptr)
	Used to read the external trees. More...
void	writeTree (const TString &fileName, const std::vector< MuonPair > *savedPair, const int muonType=0, const std::vector< GenMuonPair > *genPair=nullptr, const bool saveAll=false)

Detailed Description

This class can be used to save all the muon pairs (and gen muon pairs if any) to a root tree.
The writeTree method gets the name of the file to store the tree and the savedPair (and possibly genPair) vector of muon pairs.
Likewise, the readTree method takes the same arguments. It reads back from the file with the given name the pairs and stores them in the given savedPair (and genPair) vector.

Definition at line 24 of file RootTreeHandler.h.

Member Function Documentation

void RootTreeHandler::readTree ( const int  maxEvents, const TString &  fileName, MuonPairVector *  savedPair, const int  muonType, std::vector< std::pair< unsigned int, unsigned long long > > *  evtRun, MuonPairVector *  genPair = nullptr  )

inline

Definition at line 88 of file RootTreeHandler.h.

References gather_cfg::cout, MuScleFitEvent::event(), MuonPair::event, cmsRelvalreport::exit, FrontierConditions_GlobalTag_cff::file, mps_fire::i, Data_TkAlMinBias_Run2018C_PromptReco_v3_cff::maxEvents, MuonPair::mu1, GenMuonPair::mu1, MuonPair::mu2, GenMuonPair::mu2, MuScleFitMuon::p4(), and MuScleFitEvent::run().

Referenced by ResolutionAnalyzer::analyze(), TreeSplitter::endJob(), ErrorsAnalyzer::fillHistograms(), ErrorsPropagationAnalyzer::fillHistograms(), main(), and MuScleFit::selectMuons().

  {
    TFile * file = TFile::Open(fileName, "READ");
    if (file->IsOpen()) {
      TTree * tree = (TTree*)file->Get("T");
      MuonPair * muonPair = nullptr;
      GenMuonPair * genMuonPair = nullptr;
      // MuonPair * genMuonPair = 0;
      tree->SetBranchAddress("event", &muonPair);
      if (genPair != nullptr) {
        tree->SetBranchAddress("genEvent", &genMuonPair);
      }

      Long64_t nentries = tree->GetEntries();
      if ((maxEvents != -1) && (nentries > maxEvents)) nentries = maxEvents;
      for (Long64_t i=0; i<nentries; ++i) {
        tree->GetEntry(i);
        //std::cout << "Reco muon1, pt = " << muonPair->mu1 << "; Reco muon2, pt = " << muonPair->mu2 << std::endl;
        savedPair->push_back(std::make_pair(muonPair->mu1.p4(), muonPair->mu2.p4()));
        evtRun->push_back(std::make_pair(muonPair->event.event(), muonPair->event.run()));
        // savedPair->push_back(muonPair->getPair(muonType));
        if (genPair != nullptr) {
          genPair->push_back(std::make_pair(genMuonPair->mu1.p4(), genMuonPair->mu2.p4()));
          //std::cout << "Gen muon1, pt = " << genMuonPair->mu1 << "; Gen muon2, pt = " << genMuonPair->mu2 << std::endl;
          // genPair->push_back(genMuonPair->getPair(muonId));
        }
      }
    }
    else {
      std::cout << "ERROR: no file " << fileName << " found. Please, correct the file name or specify an empty field in the InputRootTreeFileName parameter to read events from the edm source." << std::endl;
      exit(1);
    }
    file->Close();
  }

void RootTreeHandler::readTree ( const int  maxEvents, const TString &  fileName, MuonPairExtendedVector *  savedPair, const int  muonType, std::vector< std::pair< unsigned int, unsigned long long > > *  evtRun, MuonPairExtendedVector *  genPair = nullptr  )

inline

Definition at line 124 of file RootTreeHandler.h.

References gather_cfg::cout, MuScleFitEvent::event(), MuonPair::event, cmsRelvalreport::exit, FrontierConditions_GlobalTag_cff::file, mps_fire::i, Data_TkAlMinBias_Run2018C_PromptReco_v3_cff::maxEvents, MuonPair::mu1, GenMuonPair::mu1, MuonPair::mu2, GenMuonPair::mu2, and MuScleFitEvent::run().

  {
    TFile * file = TFile::Open(fileName, "READ");
    if (file->IsOpen()) {
      TTree * tree = (TTree*)file->Get("T");
      MuonPair * muonPair = nullptr;
      GenMuonPair * genMuonPair = nullptr;
      tree->SetBranchAddress("event", &muonPair);
      if (genPair != nullptr) {
        tree->SetBranchAddress("genEvent", &genMuonPair);
      }
      Long64_t nentries = tree->GetEntries();
      if ((maxEvents != -1) && (nentries > maxEvents)) nentries = maxEvents;
      for (Long64_t i=0; i<nentries; ++i) {
        tree->GetEntry(i);
        //std::cout << "Reco muon1, pt = " << muonPair->mu1 << "; Reco muon2, pt = " << muonPair->mu2 << std::endl;
        savedPair->push_back(std::make_pair(muonPair->mu1, muonPair->mu2));
        evtRun->push_back(std::make_pair(muonPair->event.event(), muonPair->event.run()));
        // savedPair->push_back(muonPair->getPair(muonType));
        if (genPair != nullptr) {
          genPair->push_back(std::make_pair(genMuonPair->mu1, genMuonPair->mu2));
          //std::cout << "Gen muon1, pt = " << genMuonPair->mu1 << "; Gen muon2, pt = " << genMuonPair->mu2 << std::endl;
          // genPair->push_back(genMuonPair->getPair(muonId));
        }
      }
    }
    else {
      std::cout << "ERROR: no file " << fileName << " found. Please, correct the file name or specify an empty field in the InputRootTreeFileName parameter to read events from the edm source." << std::endl;
      exit(1);
    }
    file->Close();
  }

void RootTreeHandler::readTree ( const int  maxEvents, const TString &  fileName, std::vector< MuonPair > *  savedPair, const int  muonType, std::vector< GenMuonPair > *  genPair = nullptr  )

inline

Used to read the external trees.

Definition at line 160 of file RootTreeHandler.h.

References gather_cfg::cout, cmsRelvalreport::exit, FrontierConditions_GlobalTag_cff::file, mps_fire::i, and Data_TkAlMinBias_Run2018C_PromptReco_v3_cff::maxEvents.

  {
    TFile * file = TFile::Open(fileName, "READ");
    if (file->IsOpen()) {
      TTree * tree = (TTree*)file->Get("T");
      MuonPair * muonPair = nullptr;
      GenMuonPair * genMuonPair = nullptr;
      tree->SetBranchAddress("event", &muonPair);
      if (genPair != nullptr) {
        tree->SetBranchAddress("genEvent", &genMuonPair);
      }

      Long64_t nentries = tree->GetEntries();
      if ((maxEvents != -1) && (nentries > maxEvents)) nentries = maxEvents;
      for (Long64_t i=0; i<nentries; ++i) {
        tree->GetEntry(i);
        savedPair->push_back(*muonPair);
        if (genPair != nullptr) {
          genPair->push_back(*genMuonPair);
        }
      }
    }
    else {
      std::cout << "ERROR: no file " << fileName << " found. Please, correct the file name or specify an empty field in the InputRootTreeFileName parameter to read events from the edm source." << std::endl;
      exit(1);
    }
    file->Close();
  }

void RootTreeHandler::writeTree ( const TString &  fileName, const std::vector< MuonPair > *  savedPair, const int  muonType = 0, const std::vector< GenMuonPair > *  genPair = nullptr, const bool  saveAll = false  )

inline

Definition at line 29 of file RootTreeHandler.h.

References MuonPair::copy(), GenMuonPair::copy(), gather_cfg::cout, cmsRelvalreport::exit, and connectstrParser::f1.

Referenced by TreeSplitter::endJob(), main(), and MuScleFit::~MuScleFit().

  {
    lorentzVector emptyLorentzVector(0,0,0,0);
    TFile * f1 = new TFile(fileName, "RECREATE");
    TTree * tree = new TTree("T", "Muon pairs");
    MuonPair * muonPair = new MuonPair;
    GenMuonPair * genMuonPair = new GenMuonPair;
    // MuonPair * genMuonPair = new MuonPair;
    tree->Branch("event", "MuonPair", &muonPair);
    if( genPair != nullptr ) {
      tree->Branch("genEvent", "GenMuonPair", &genMuonPair);
      // tree->Branch("genEvent", "MuonPair", &genMuonPair);

      if( savedPair->size() != genPair->size() ) {
    std::cout << "Error: savedPair size ("
    << savedPair->size() <<") and genPair size ("
    << genPair->size() <<") are different. This is severe and I will not write the tree." << std::endl;
    exit(1);
      }
    }
    std::cout << "savedPair->size() is "<<savedPair->size()<< std::endl;
    std::vector<MuonPair>::const_iterator muonPairIt = savedPair->begin();
    unsigned int iev = 0;
    for( ; muonPairIt != savedPair->end(); ++muonPairIt, ++iev ) {

      if( saveAll || ( (muonPairIt->mu1.p4() != emptyLorentzVector) && (muonPairIt->mu2.p4() != emptyLorentzVector) ) ) {

    // muonPair->setPair(muonType, std::make_pair(muonPairIt->first, muonPairIt->second));
    muonPair->copy(*muonPairIt);

    // if( genPair != 0 && genPair->size() != 0 ) {
    //   genMuonPair->setPair(muonId, std::make_pair((*genPair)[iev].first, (*genPair)[iev].second));
    //   genMuonPair->mu1 = ((*genPair)[iev].first);
    //   genMuonPair->mu2 = ((*genPair)[iev].second);
        // }
    if( genPair != nullptr ) {
      genMuonPair->copy((*genPair)[iev]);
    }

    tree->Fill();
      }
      // // Tree filled. Clear the map for the next event.
      // muonPair->muonPairs.clear();
    }

    // Save provenance information in the TFile
    TH1F muonTypeHisto("MuonType", "MuonType", 40, -20, 20);
    muonTypeHisto.Fill(muonType);
    muonTypeHisto.Write();
    MuScleFitProvenance provenance(muonType);
    provenance.Write();

    f1->Write();
    f1->Close();
  }