void dimuonSkim(const TString configFile, const TString inputFile, const TString outputFile)
{
std::cout<<"running dimuonSkim()" <<std::endl;
std::cout<<"configFile = "<< configFile.Data() <<std::endl;
std::cout<<"inputFile = "<< inputFile.Data() <<std::endl;
std::cout<<"outputFile = "<< outputFile.Data() <<std::endl;
InputConfiguration configInput = InputConfigurationParser::Parse(configFile.Data());
CutConfiguration configCuts = CutConfigurationParser::Parse(configFile.Data());
if (!configInput.isValid) {
std::cout << "Input configuration is invalid." << std::endl;
std::cout << "exiting" << std::endl;
return;
}
if (!configCuts.isValid) {
std::cout << "Cut configuration is invalid." << std::endl;
std::cout << "exiting" << std::endl;
return;
}
// input configuration
int collisionType = configInput.proc[INPUT::kSKIM].i[INPUT::k_collisionType];
std::string treePath = configInput.proc[INPUT::kSKIM].s[INPUT::k_treePath];
// set default values
if (treePath.size() == 0) treePath = "ggHiNtuplizer/EventTree";
// verbose about input configuration
std::cout<<"Input Configuration :"<<std::endl;
std::cout << "collisionType = " << collisionType << std::endl;
const char* collisionName = getCollisionTypeName((COLL::TYPE)collisionType).c_str();
std::cout << "collision = " << collisionName << std::endl;
std::cout << "treePath = " << treePath.c_str() << std::endl;
// cut configuration
float cut_vz = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].f[CUTS::EVT::k_vz];
int cut_pcollisionEventSelection = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pcollisionEventSelection];
int cut_pPAprimaryVertexFilter = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pPAprimaryVertexFilter];
int cut_pBeamScrapingFilter = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pBeamScrapingFilter];
int cut_nMu = configCuts.proc[CUTS::kSKIM].obj[CUTS::kMUON].i[CUTS::MUO::k_nMu];
// bool isMC = collisionIsMC((COLL::TYPE)collisionType);
bool isHI = collisionIsHI((COLL::TYPE)collisionType);
bool isPP = collisionIsPP((COLL::TYPE)collisionType);
// verbose about cut configuration
std::cout<<"Cut Configuration :"<<std::endl;
std::cout<<"cut_vz = "<< cut_vz <<std::endl;
if (isHI) {
std::cout<<"cut_pcollisionEventSelection = "<< cut_pcollisionEventSelection <<std::endl;
}
else { // PP
std::cout<<"cut_pPAprimaryVertexFilter = "<< cut_pPAprimaryVertexFilter <<std::endl;
std::cout<<"cut_pBeamScrapingFilter = "<< cut_pBeamScrapingFilter <<std::endl;
}
std::cout<<"cut_nMu = "<<cut_nMu<<std::endl;
std::vector<std::string> inputFiles = InputConfigurationParser::ParseFiles(inputFile.Data());
std::cout<<"input ROOT files : num = "<<inputFiles.size()<< std::endl;
std::cout<<"#####"<< std::endl;
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
std::cout<<(*it).c_str()<< std::endl;
}
std::cout<<"##### END #####"<< std::endl;
TChain* treeHLT = new TChain("hltanalysis/HltTree");
TChain* treeggHiNtuplizer = new TChain("ggHiNtuplizer/EventTree");
TChain* treeHiEvt = new TChain("hiEvtAnalyzer/HiTree");
TChain* treeSkim = new TChain("skimanalysis/HltTree");
TChain* treeHiForestInfo = new TChain("HiForest/HiForestInfo");
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
treeHLT->Add((*it).c_str());
treeggHiNtuplizer->Add((*it).c_str());
treeHiEvt->Add((*it).c_str());
treeSkim->Add((*it).c_str());
treeHiForestInfo->Add((*it).c_str());
}
HiForestInfoController hfic(treeHiForestInfo);
std::cout<<"### HiForestInfo Tree ###"<< std::endl;
hfic.printHiForestInfo();
std::cout<<"###"<< std::endl;
treeHLT->SetBranchStatus("*",0); // disable all branches
treeHLT->SetBranchStatus("HLT_HI*SinglePhoton*Eta*",1); // enable photon branches
treeHLT->SetBranchStatus("HLT_HI*DoublePhoton*Eta*",1); // enable photon branches
treeHLT->SetBranchStatus("*DoubleMu*",1); // enable muon branches
treeHLT->SetBranchStatus("HLT_HIL1Mu*",1); // enable muon branches
treeHLT->SetBranchStatus("HLT_HIL2Mu*",1); // enable muon branches
treeHLT->SetBranchStatus("HLT_HIL3Mu*",1); // enable muon branches
// specify explicitly which branches to store, do not use wildcard
treeHiEvt->SetBranchStatus("*",1);
// specify explicitly which branches to store, do not use wildcard
treeSkim->SetBranchStatus("*",0);
Int_t pcollisionEventSelection; // this filter is used for HI.
if (isHI) {
treeSkim->SetBranchStatus("pcollisionEventSelection",1);
if (treeSkim->GetBranch("pcollisionEventSelection")) {
treeSkim->SetBranchAddress("pcollisionEventSelection",&pcollisionEventSelection);
}
else { // overwrite to default
pcollisionEventSelection = 1;
std::cout<<"could not get branch : pcollisionEventSelection"<<std::endl;
std::cout<<"set to default value : pcollisionEventSelection = "<<pcollisionEventSelection<<std::endl;
}
}
else {
pcollisionEventSelection = 0; // default value if the collision is not HI, will not be used anyway.
}
Int_t pPAprimaryVertexFilter; // this filter is used for PP.
if (isPP) {
treeSkim->SetBranchStatus("pPAprimaryVertexFilter",1);
if (treeSkim->GetBranch("pPAprimaryVertexFilter")) {
treeSkim->SetBranchAddress("pPAprimaryVertexFilter",&pPAprimaryVertexFilter);
}
else { // overwrite to default
pPAprimaryVertexFilter = 1;
std::cout<<"could not get branch : pPAprimaryVertexFilter"<<std::endl;
std::cout<<"set to default value : pPAprimaryVertexFilter = "<<pPAprimaryVertexFilter<<std::endl;
}
}
else {
pPAprimaryVertexFilter = 0; // default value if the collision is not PP, will not be used anyway.
}
Int_t pBeamScrapingFilter; // this filter is used for PP.
if (isPP) {
treeSkim->SetBranchStatus("pBeamScrapingFilter",1);
if (treeSkim->GetBranch("pBeamScrapingFilter")) {
treeSkim->SetBranchAddress("pBeamScrapingFilter",&pBeamScrapingFilter);
}
else { // overwrite to default
pBeamScrapingFilter = 1;
std::cout<<"could not get branch : pBeamScrapingFilter"<<std::endl;
std::cout<<"set to default value : pBeamScrapingFilter = "<<pBeamScrapingFilter<<std::endl;
}
}
else {
pBeamScrapingFilter = 0; // default value if the collision is not PP, will not be used anyway.
}
ggHiNtuplizer ggHi;
ggHi.setupTreeForReading(treeggHiNtuplizer);
hiEvt hiEvt;
hiEvt.setupTreeForReading(treeHiEvt);
TFile* output = new TFile(outputFile,"RECREATE");
TTree *configTree = setupConfigurationTreeForWriting(configCuts);
// output tree variables
TTree *outputTreeHLT = treeHLT->CloneTree(0);
outputTreeHLT->SetName("hltTree");
outputTreeHLT->SetTitle("subbranches of hltanalysis/HltTree");
TTree *outputTreeggHiNtuplizer = treeggHiNtuplizer->CloneTree(0);
TTree *outputTreeHiEvt = treeHiEvt->CloneTree(0);
outputTreeHiEvt->SetName("HiEvt");
outputTreeHiEvt->SetTitle("subbranches of hiEvtAnalyzer/HiTree");
TTree* outputTreeSkim = treeSkim->CloneTree(0);
outputTreeSkim->SetName("skim");
outputTreeSkim->SetTitle("subbranches of skimanalysis/HltTree");
TTree* outputTreeHiForestInfo = treeHiForestInfo->CloneTree(0);
outputTreeHiForestInfo->SetName("HiForestInfo");
outputTreeHiForestInfo->SetTitle("first entry of HiForest/HiForestInfo");
outputTreeHLT->SetMaxTreeSize(MAXTREESIZE);
outputTreeggHiNtuplizer->SetMaxTreeSize(MAXTREESIZE);
outputTreeHiEvt->SetMaxTreeSize(MAXTREESIZE);
outputTreeHiForestInfo->SetMaxTreeSize(MAXTREESIZE);
// write HiForestInfo
treeHiForestInfo->GetEntry(0);
outputTreeHiForestInfo->Fill();
TTree *diMuonTree = new TTree("dimuon","muon pairs");
diMuonTree->SetMaxTreeSize(MAXTREESIZE);
dimuon diMu;
diMu.branchDiMuonTree(diMuonTree);
EventMatcher* em = new EventMatcher();
Long64_t duplicateEntries = 0;
Long64_t entries = treeggHiNtuplizer->GetEntries();
Long64_t entriesPassedEventSelection = 0;
Long64_t entriesAnalyzed = 0;
std::cout << "entries = " << entries << std::endl;
std::cout<< "Loop : " << treePath.c_str() <<std::endl;
for (Long64_t j_entry=0; j_entry<entries; ++j_entry)
{
if (j_entry % 20000 == 0) {
std::cout << "current entry = " <<j_entry<<" out of "<<entries<<" : "<<std::setprecision(2)<<(double)j_entry/entries*100<<" %"<<std::endl;
}
treeHLT->GetEntry(j_entry);
treeggHiNtuplizer->GetEntry(j_entry);
treeHiEvt->GetEntry(j_entry);
treeSkim->GetEntry(j_entry);
bool eventAdded = em->addEvent(ggHi.run,ggHi.lumis,ggHi.event,j_entry);
if(!eventAdded) // this event is duplicate, skip this one.
{
duplicateEntries++;
continue;
}
// event selection
if (!(TMath::Abs(hiEvt.vz) < cut_vz)) continue;
if (isHI) {
if ((pcollisionEventSelection < cut_pcollisionEventSelection)) continue;
}
else {
if (pPAprimaryVertexFilter < cut_pPAprimaryVertexFilter || pBeamScrapingFilter < cut_pBeamScrapingFilter) continue;
}
entriesPassedEventSelection++;
// skip if there are no muon pairs to study
if(ggHi.nMu < cut_nMu) continue;
entriesAnalyzed++;
diMu.makeDiMuonPairs(ggHi);
outputTreeHLT->Fill();
outputTreeggHiNtuplizer->Fill();
outputTreeHiEvt->Fill();
outputTreeSkim->Fill();
diMuonTree->Fill();
}
std::cout<< "Loop ENDED : " << treePath.c_str() <<std::endl;
std::cout << "entries = " << entries << std::endl;
std::cout << "duplicateEntries = " << duplicateEntries << std::endl;
std::cout << "entriesPassedEventSelection = " << entriesPassedEventSelection << std::endl;
std::cout << "entriesAnalyzed = " << entriesAnalyzed << std::endl;
std::cout << "outputTreeHLT->GetEntries() = " << outputTreeHLT->GetEntries() << std::endl;
std::cout << "outputTreeggHiNtuplizer->GetEntries() = " << outputTreeggHiNtuplizer->GetEntries() << std::endl;
std::cout << "outputTreeHiEvt->GetEntries() = " << outputTreeHiEvt->GetEntries() << std::endl;
std::cout << "outputTreeSkim->GetEntries() = " << outputTreeSkim->GetEntries() << std::endl;
std::cout << "diMuonTree->GetEntries() = " << diMuonTree->GetEntries() << std::endl;
// overwrite existing trees
outputTreeHLT->Write("", TObject::kOverwrite);
outputTreeggHiNtuplizer->Write("", TObject::kOverwrite);
outputTreeHiEvt->Write("", TObject::kOverwrite);
diMuonTree->Write("", TObject::kOverwrite);
configTree->Write("", TObject::kOverwrite);
output->Write("", TObject::kOverwrite);
output->Close();
std::cout<<"dimuonSkim() - END" <<std::endl;
}
void photonRaaSkim(const TString configFile, const TString inputFile, const TString outputFile, COLL::TYPE colli)
{
std::cout<<"running photonRaaSkim()"<<std::endl;
std::cout<<"configFile = "<< configFile.Data() <<std::endl;
std::cout<<"inputFile = "<< inputFile.Data() <<std::endl;
std::cout<<"outputFile = "<< outputFile.Data() <<std::endl;
InputConfiguration configInput = InputConfigurationParser::Parse(configFile.Data());
CutConfiguration configCuts = CutConfigurationParser::Parse(configFile.Data());
// input configuration
/* int collisionType;
if (configInput.isValid) {
collisionType = configInput.proc[INPUT::kSKIM].i[INPUT::k_collisionType];
}
else {
collisionType = COLL::kPP;
}
*/
// verbose about input configuration
int collisionType = colli;
std::cout<<"Input Configuration :"<<std::endl;
std::cout << "collisionType = " << collisionType << std::endl;
const char* collisionName = getCollisionTypeName((COLL::TYPE)collisionType).c_str();
std::cout << "collision = " << collisionName << std::endl;
// cut configuration
float cut_vz;
int cut_pcollisionEventSelection;
int cut_pPAprimaryVertexFilter;
int cut_pBeamScrapingFilter;
float cutPhoEt;
float cutPhoEta;
if (configCuts.isValid) {
cut_vz = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].f[CUTS::EVT::k_vz];
cut_pcollisionEventSelection = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pcollisionEventSelection];
cut_pPAprimaryVertexFilter = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pPAprimaryVertexFilter];
cut_pBeamScrapingFilter = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pBeamScrapingFilter];
cutPhoEt = configCuts.proc[CUTS::kSKIM].obj[CUTS::kPHOTON].f[CUTS::PHO::k_et];
cutPhoEta = configCuts.proc[CUTS::kSKIM].obj[CUTS::kPHOTON].f[CUTS::PHO::k_eta];
}
else {
cut_vz = 15;
cut_pcollisionEventSelection = 1;
cut_pPAprimaryVertexFilter = 1;
cut_pBeamScrapingFilter = 1;
cutPhoEt = 15;
cutPhoEta = 1.44;
}
bool isMC = collisionIsMC((COLL::TYPE)collisionType);
bool isHI = collisionIsHI((COLL::TYPE)collisionType);
bool isPP = collisionIsPP((COLL::TYPE)collisionType);
// verbose about cut configuration
std::cout<<"Cut Configuration :"<<std::endl;
std::cout<<"cut_vz = "<< cut_vz <<std::endl;
if (isHI) {
std::cout<<"cut_pcollisionEventSelection = "<< cut_pcollisionEventSelection <<std::endl;
}
else { // PP
std::cout<<"cut_pPAprimaryVertexFilter = "<< cut_pPAprimaryVertexFilter <<std::endl;
std::cout<<"cut_pBeamScrapingFilter = "<< cut_pBeamScrapingFilter <<std::endl;
}
std::cout<<"cutPhoEt = "<<cutPhoEt<<std::endl;
std::cout<<"cutPhoEta = "<<cutPhoEta<<std::endl;
std::vector<std::string> inputFiles = InputConfigurationParser::ParseFiles(inputFile.Data());
std::cout<<"input ROOT files : num = "<<inputFiles.size()<< std::endl;
std::cout<<"#####"<< std::endl;
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
std::cout<<(*it).c_str()<< std::endl;
}
std::cout<<"##### END #####"<< std::endl;
TChain* treeHLT = new TChain("hltanalysis/HltTree");
TChain* treeggHiNtuplizer = new TChain("ggHiNtuplizer/EventTree");
TChain* treeHiEvt = new TChain("hiEvtAnalyzer/HiTree");
TChain* treeSkim = new TChain("skimanalysis/HltTree");
TChain* treeHiForestInfo = new TChain("HiForest/HiForestInfo");
TChain* treeGen=0;
if(isMC) treeGen = new TChain("HiGenParticleAna/hi");
// pthatWeight Calculation block!
int nPthat = 5;
float pthatCut[nPthat+1];
const char* lowestPthatFileName="";
int nfiles = 0;
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
treeHLT->Add((*it).c_str());
treeggHiNtuplizer->Add((*it).c_str());
treeHiEvt->Add((*it).c_str());
treeSkim->Add((*it).c_str());
treeHiForestInfo->Add((*it).c_str());
if(isMC) treeGen->Add((*it).c_str());
if(isMC && (nfiles==0)) {
lowestPthatFileName = (*it).c_str();
TString str(lowestPthatFileName);
cout << "lowestPthatFileName = " << lowestPthatFileName << endl;
if(str.Contains("15")) {
float temp[] = {15,30,50,80,120,9999};
for(int j=0;j<nPthat+1;j++){
pthatCut[j] = temp[j];
}
} else if(str.Contains("30")) {
float temp[] = {30,50,80,120,170,9999};
for(int j=0;j<nPthat+1;j++){
pthatCut[j] = temp[j];
}
}
}
nfiles++;
}
float tmpWeight[nPthat];
if(isMC) {
for(int j=0; j<nPthat ; j++){
tmpWeight[j] = xSecCal(lowestPthatFileName,treeHiEvt, pthatCut[j], pthatCut[j+1]);
cout << collisionName << ", pthatWeight of " << pthatCut[j] << " to " << pthatCut[j+1] << " = " << tmpWeight[j] << endl;
}
}
/*
HiForestInfoController hfic(treeHiForestInfo);
std::cout<<"### HiForestInfo Tree ###"<< std::endl;
hfic.printHiForestInfo();
std::cout<<"###"<< std::endl;
*/
treeHLT->SetBranchStatus("*",0); // disable all branches
treeHLT->SetBranchStatus("HLT_HI*SinglePhoton*Eta*",1); // enable photon branches
treeHLT->SetBranchStatus("HLT_HI*DoublePhoton*Eta*",1); // enable photon branches
float vz;
Int_t hiBin;
UInt_t run, lumis;
ULong64_t event;
float pthat, pthatWeight;
treeHiEvt->SetBranchAddress("vz",&vz);
treeHiEvt->SetBranchAddress("hiBin",&hiBin);
treeHiEvt->SetBranchAddress("run", &run);
treeHiEvt->SetBranchAddress("evt", &event);
treeHiEvt->SetBranchAddress("lumi", &lumis);
if(isMC) {
//treeHiEvt->Branch("pthatWeight", &pthatWeight, "pthatWeight/F");
treeHiEvt->SetBranchAddress("pthat", &pthat);
}
// specify explicitly which branches to store, do not use wildcard
treeSkim->SetBranchStatus("*",0);
Int_t pcollisionEventSelection; // this filter is used for HI.
if (isHI) {
treeSkim->SetBranchStatus("pcollisionEventSelection",1);
if (treeSkim->GetBranch("pcollisionEventSelection")) {
treeSkim->SetBranchAddress("pcollisionEventSelection",&pcollisionEventSelection);
}
else { // overwrite to default
pcollisionEventSelection = 1;
std::cout<<"could not get branch : pcollisionEventSelection"<<std::endl;
std::cout<<"set to default value : pcollisionEventSelection = "<<pcollisionEventSelection<<std::endl;
}
}
else {
pcollisionEventSelection = 0; // default value if the collision is not HI, will not be used anyway.
}
Int_t pPAprimaryVertexFilter; // this filter is used for PP.
if (isPP) {
treeSkim->SetBranchStatus("pPAprimaryVertexFilter",1);
if (treeSkim->GetBranch("pPAprimaryVertexFilter")) {
treeSkim->SetBranchAddress("pPAprimaryVertexFilter",&pPAprimaryVertexFilter);
}
else { // overwrite to default
pPAprimaryVertexFilter = 1;
std::cout<<"could not get branch : pPAprimaryVertexFilter"<<std::endl;
std::cout<<"set to default value : pPAprimaryVertexFilter = "<<pPAprimaryVertexFilter<<std::endl;
}
}
else {
pPAprimaryVertexFilter = 0; // default value if the collision is not PP, will not be used anyway.
}
Int_t pBeamScrapingFilter; // this filter is used for PP.
if (isPP) {
treeSkim->SetBranchStatus("pBeamScrapingFilter",1);
if (treeSkim->GetBranch("pBeamScrapingFilter")) {
treeSkim->SetBranchAddress("pBeamScrapingFilter",&pBeamScrapingFilter);
}
else { // overwrite to default
pBeamScrapingFilter = 1;
std::cout<<"could not get branch : pBeamScrapingFilter"<<std::endl;
std::cout<<"set to default value : pBeamScrapingFilter = "<<pBeamScrapingFilter<<std::endl;
}
}
else {
pBeamScrapingFilter = 0; // default value if the collision is not PP, will not be used anyway.
}
// objects for z-jet correlations
ggHiNtuplizer ggHi;
ggHi.setupTreeForReading(treeggHiNtuplizer); // treeggHiNtuplizer is input
treeggHiNtuplizer->SetBranchStatus("*",0);
treeggHiNtuplizer->SetBranchStatus("run",1);
treeggHiNtuplizer->SetBranchStatus("event",1);
treeggHiNtuplizer->SetBranchStatus("lumis",1);
treeggHiNtuplizer->SetBranchStatus("nPho",1);
treeggHiNtuplizer->SetBranchStatus("pho*",1);
treeggHiNtuplizer->SetBranchStatus("pf*",1);
//treeggHiNtuplizer->SetBranchStatus("tower*",1);
if(isMC) treeggHiNtuplizer->SetBranchStatus("mc*",1);
TFile* output = TFile::Open(outputFile,"RECREATE");
TTree* configTree = setupConfigurationTreeForWriting(configCuts);
// output tree variables
TTree *outputTreeHLT=0, *outputTreeggHiNtuplizer=0,
*outputTreeHiEvt=0, *outputTreeSkim=0, *outputTreeHiForestInfo=0, *outputTreeGen=0;
// output tree variables
outputTreeHLT = treeHLT->CloneTree(0);
outputTreeggHiNtuplizer = treeggHiNtuplizer->CloneTree(0);
outputTreeHiEvt = treeHiEvt->CloneTree(0);
outputTreeSkim = treeSkim->CloneTree(0);
outputTreeHiForestInfo = treeHiForestInfo->CloneTree(0);
if(isMC) outputTreeGen = treeGen ->CloneTree(0);
outputTreeSkim->SetName("skimTree");
outputTreeHLT->SetMaxTreeSize(MAXTREESIZE);
outputTreeggHiNtuplizer->SetMaxTreeSize(MAXTREESIZE);
outputTreeHiEvt->SetMaxTreeSize(MAXTREESIZE);
outputTreeSkim->SetMaxTreeSize(MAXTREESIZE);
outputTreeHiForestInfo->SetMaxTreeSize(MAXTREESIZE);
if(isMC) outputTreeGen->SetMaxTreeSize(MAXTREESIZE);
if(isMC) outputTreeHiEvt->Branch("pthatWeight", &pthatWeight, "pthatWeight/F");
/////// Event Matching for DATA ///////
EventMatcher* em = new EventMatcher();
Long64_t duplicateEntries = 0;
Long64_t entriesPassedEventSelection =0;
Long64_t entriesAnalyzed =0;
Long64_t entriesSpikeRejected=0;
Long64_t entries = treeggHiNtuplizer->GetEntries();
std::cout << "entries = " << entries << std::endl;
std::cout<< "Loop : ggHiNtuplizer/EventTree" <<std::endl;
//for (Long64_t j_entry=10000; j_entry<10100; ++j_entry)
for (Long64_t j_entry=0; j_entry<entries; ++j_entry)
{
if (j_entry % 2000 == 0) {
std::cout << "current entry = " <<j_entry<<" out of "<<entries<<" : "<<std::setprecision(2)<<(double)j_entry/entries*100<<" %"<<std::endl;
}
treeHLT->GetEntry(j_entry);
treeggHiNtuplizer->GetEntry(j_entry);
treeSkim->GetEntry(j_entry);
treeHiEvt->GetEntry(j_entry);
treeHiForestInfo->GetEntry(j_entry);
if(isMC) treeGen->GetEntry(j_entry);
bool eventAdded = em->addEvent(run,lumis,event,j_entry);
//std::cout << run << " " << lumis << " " << event << " " << j_entry << std::endl;
if(!eventAdded) // this event is duplicate, skip this one.
{
duplicateEntries++;
continue;
}
if(isMC) {
if((pthat>=pthatCut[0]) && (pthat<pthatCut[1])) pthatWeight = tmpWeight[0];
else if((pthat>=pthatCut[1]) && (pthat<pthatCut[2])) pthatWeight = tmpWeight[1];
else if((pthat>=pthatCut[2]) && (pthat<pthatCut[3])) pthatWeight = tmpWeight[2];
else if((pthat>=pthatCut[3]) && (pthat<pthatCut[4])) pthatWeight = tmpWeight[3];
else if((pthat>=pthatCut[4]) && (pthat<pthatCut[5])) pthatWeight = tmpWeight[4];
else continue;
}
// event selection
if (!(TMath::Abs(vz) < cut_vz)) continue;
if (isHI) {
if ((pcollisionEventSelection < cut_pcollisionEventSelection)) continue;
}
else {
if (pPAprimaryVertexFilter < cut_pPAprimaryVertexFilter || pBeamScrapingFilter < cut_pBeamScrapingFilter) continue;
}
entriesPassedEventSelection++;
// photon block
// find leading photon
int phoIdx = -1; // index of the leading photon
double maxPhoEt = -1;
for(int i=0; i<ggHi.nPho; ++i)
{
bool failedEtCut = (ggHi.phoEt->at(i) < cutPhoEt) ;
bool failedEtaCut = (TMath::Abs(ggHi.phoEta->at(i)) > cutPhoEta) ;
bool failedSpikeRejection;
bool failedHotSpotRejection;
//if (isHI) {
failedSpikeRejection =( (ggHi.phoEta->at(i)<1.44) &&
(ggHi.phoSigmaIEtaIEta->at(i) < 0.002 ||
ggHi.pho_swissCrx->at(i) > 0.9 ||
TMath::Abs(ggHi.pho_seedTime->at(i)) > 3) );
// }
// else {
// failedSpikeRejection = (ggHi.phoSigmaIEtaIEta->at(i) < 0.002);
// }
failedHotSpotRejection = (
(ggHi.phoE3x3->at(i)/ggHi.phoE5x5->at(i) > 2./3.-0.03 && ggHi.phoE3x3->at(i)/ggHi.phoE5x5->at(i) < 2./3.+0.03) &&
(ggHi.phoE1x5->at(i)/ggHi.phoE5x5->at(i) > 1./3.-0.03 && ggHi.phoE1x5->at(i)/ggHi.phoE5x5->at(i) < 1./3.+0.03) &&
(ggHi.phoE2x5->at(i)/ggHi.phoE5x5->at(i) > 2./3.-0.03 && ggHi.phoE2x5->at(i)/ggHi.phoE5x5->at(i) < 2./3.+0.03) );
bool failedHoverE = (ggHi.phoHoverE->at(i) > 0.2); // <0.1 cut is applied after corrections
// bool failedEnergyRatio = ((float)ggHi.phoSCRawE->at(i)/ggHi.phoE->at(i) < 0.5);
if (failedEtCut) continue;
if (failedEtaCut) continue;
if (failedSpikeRejection) continue;
if (failedHotSpotRejection) {entriesSpikeRejected++; continue;}
if (failedHoverE) continue;
// if (failedEnergyRatio) continue; // actually applied after corrections
if (ggHi.phoEt->at(i) > maxPhoEt)
{
maxPhoEt = ggHi.phoEt->at(i);
phoIdx = i;
}
}
if (phoIdx == -1) continue;
entriesAnalyzed++;
outputTreeHLT->Fill();
outputTreeggHiNtuplizer->Fill();
outputTreeHiEvt->Fill();
outputTreeSkim->Fill();
outputTreeHiForestInfo->Fill();
if(isMC) outputTreeGen->Fill();
}// event loop closed here
std::cout<< "Loop ENDED : ggHiNtuplizer/EventTree" <<std::endl;
std::cout << "entries = " << entries << std::endl;
std::cout << "duplicateEntries = " << duplicateEntries << std::endl;
std::cout << "entriesPassedEventSelection = " << entriesPassedEventSelection << std::endl;
std::cout << "entriesAnalyzed = " << entriesAnalyzed << std::endl;
std::cout << "entriesSpikeRejected = " << entriesSpikeRejected << std::endl;
std::cout << "outputTreeHLT->GetEntries() = " << outputTreeHLT->GetEntries() << std::endl;
std::cout << "outputTreeggHiNtuplizer->GetEntries() = " << outputTreeggHiNtuplizer->GetEntries() << std::endl;
std::cout << "outputTreeSkim->GetEntries() = " << outputTreeSkim->GetEntries() << std::endl;
std::cout << "outputTreeHiEvt->GetEntries() = " << outputTreeHiEvt->GetEntries() << std::endl;
std::cout << "outputTreeHiForestInfo->GetEntries() = " << outputTreeHiForestInfo->GetEntries() << std::endl;
if(isMC) std::cout << "outputTreeGen->GetEntries() = " << outputTreeGen->GetEntries() << std::endl;
output->cd();
configTree->Write("",TObject::kOverwrite);
output->Write("",TObject::kOverwrite);
output->Close();
}
void minBiasJetSkim(const TString configFile, const TString inputFile, const TString outputFile)
{
std::cout<<"running minBiasJetSkim()"<<std::endl;
std::cout<<"configFile = "<< configFile.Data() <<std::endl;
std::cout<<"inputFile = "<< inputFile.Data() <<std::endl;
std::cout<<"outputFile = "<< outputFile.Data() <<std::endl;
InputConfiguration configInput = InputConfigurationParser::Parse(configFile.Data());
CutConfiguration configCuts = CutConfigurationParser::Parse(configFile.Data());
if (!configInput.isValid) {
std::cout << "Input configuration is invalid." << std::endl;
std::cout << "exiting" << std::endl;
return;
}
if (!configCuts.isValid) {
std::cout << "Cut configuration is invalid." << std::endl;
std::cout << "exiting" << std::endl;
return;
}
// cut configuration
float cut_vz = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].f[CUTS::EVT::k_vz];
int cut_pcollisionEventSelection = configCuts.proc[CUTS::kSKIM].obj[CUTS::kEVENT].i[CUTS::EVT::k_pcollisionEventSelection];
std::vector<std::string> jetCollections = ConfigurationParser::ParseList(configCuts.proc[CUTS::kSKIM].obj[CUTS::kJET].s[CUTS::JET::k_jetCollection]);
int nMaxEvents_minBiasMixing = configCuts.proc[CUTS::kSKIM].obj[CUTS::kGAMMAJET].i[CUTS::GJT::k_nMaxEvents_minBiasMixing];
int nCentralityBins = configCuts.proc[CUTS::kSKIM].obj[CUTS::kGAMMAJET].i[CUTS::GJT::k_nCentralityBins];
int nVertexBins = configCuts.proc[CUTS::kSKIM].obj[CUTS::kGAMMAJET].i[CUTS::GJT::k_nVertexBins];
int nEventPlaneBins = configCuts.proc[CUTS::kSKIM].obj[CUTS::kGAMMAJET].i[CUTS::GJT::k_nEventPlaneBins];
int nJetCollections = jetCollections.size();
// verbose about cut configuration
std::cout<<"Cut Configuration :"<<std::endl;
std::cout<<"cut_vz = "<< cut_vz <<std::endl;
std::cout<<"cut_pcollisionEventSelection = "<< cut_pcollisionEventSelection <<std::endl;
std::cout<<"nJetCollections = "<< nJetCollections <<std::endl;
for (int i=0; i<nJetCollections; ++i) {
std::cout << Form("jetCollections[%d] = %s", i, jetCollections.at(i).c_str()) << std::endl;
}
std::cout<<"nMaxEvents_minBiasMixing = "<< nMaxEvents_minBiasMixing <<std::endl;
std::cout<<"nCentralityBins = "<< nCentralityBins <<std::endl;
std::cout<<"nVertexBins = "<< nVertexBins <<std::endl;
std::cout<<"nEventPlaneBins = "<< nEventPlaneBins <<std::endl;
std::vector<std::string> inputFiles = InputConfigurationParser::ParseFiles(inputFile.Data());
std::cout<<"input ROOT files : num = "<<inputFiles.size()<< std::endl;
std::cout<<"#####"<< std::endl;
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
std::cout<<(*it).c_str()<< std::endl;
}
std::cout<<"##### END #####"<< std::endl;
TChain* treeHLT = new TChain("hltanalysis/HltTree");
TChain* treeEvent = new TChain("ggHiNtuplizer/EventTree");
TChain* treeJets[nJetCollections];
for (int i=0; i<nJetCollections; ++i) {
treeJets[i] = new TChain(Form("%s/t", jetCollections.at(i).c_str()));
}
TChain* treeHiEvt = new TChain("hiEvtAnalyzer/HiTree");
TChain* treeSkim = new TChain("skimanalysis/HltTree");
TChain* treeHiForestInfo = new TChain("HiForest/HiForestInfo");
for (std::vector<std::string>::iterator it = inputFiles.begin() ; it != inputFiles.end(); ++it) {
treeHLT->Add((*it).c_str());
treeEvent->Add((*it).c_str());
for (int i=0; i<nJetCollections; ++i) {
treeJets[i]->Add((*it).c_str());
}
treeHiEvt->Add((*it).c_str());
treeSkim->Add((*it).c_str());
if (it == inputFiles.begin()) treeHiForestInfo->Add((*it).c_str());
}
HiForestInfoController hfic(treeHiForestInfo);
std::cout<<"### HiForestInfo Tree ###"<< std::endl;
hfic.printHiForestInfo();
std::cout<<"###"<< std::endl;
treeHLT->SetBranchStatus("*",0); // disable all branches
treeEvent->SetBranchStatus("*",0);
treeEvent->SetBranchStatus("run",1);
treeEvent->SetBranchStatus("event",1);
treeEvent->SetBranchStatus("lumis",1);
for (int i=0; i<nJetCollections; ++i) {
treeJets[i]->SetBranchStatus("*",0); // disable all branches
treeJets[i]->SetBranchStatus("nref",1); // enable jet branches
treeJets[i]->SetBranchStatus("rawpt",1); // enable jet branches
treeJets[i]->SetBranchStatus("jtpt",1); // enable jet branches
treeJets[i]->SetBranchStatus("jteta",1); // enable jet branches
treeJets[i]->SetBranchStatus("jtphi",1); // enable jet branches
treeJets[i]->SetBranchStatus("jty",1); // enable jet branches
treeJets[i]->SetBranchStatus("jtpu",1); // enable jet branches
treeJets[i]->SetBranchStatus("jtm",1); // enable jet branches
treeJets[i]->SetBranchStatus("track*",1);
treeJets[i]->SetBranchStatus("charged*",1);
treeJets[i]->SetBranchStatus("photon*",1);
treeJets[i]->SetBranchStatus("neutral*",1);
treeJets[i]->SetBranchStatus("eMax*",1);
treeJets[i]->SetBranchStatus("eSum*",1);
treeJets[i]->SetBranchStatus("eN*",1);
treeJets[i]->SetBranchStatus("muMax*",1);
treeJets[i]->SetBranchStatus("muSum*",1);
treeJets[i]->SetBranchStatus("muN*",1);
}
// specify explicitly which branches to store, do not use wildcard
treeHiEvt->SetBranchStatus("*",0);
treeHiEvt->SetBranchStatus("run",1);
treeHiEvt->SetBranchStatus("evt",1);
treeHiEvt->SetBranchStatus("lumi",1);
treeHiEvt->SetBranchStatus("vz",1);
treeHiEvt->SetBranchStatus("hiBin",1);
treeHiEvt->SetBranchStatus("hiHF",1);
treeHiEvt->SetBranchStatus("hiHFplus",1);
treeHiEvt->SetBranchStatus("hiHFminus",1);
treeHiEvt->SetBranchStatus("hiHFplusEta4",1);
treeHiEvt->SetBranchStatus("hiHFminusEta4",1);
// treeHiEvt->SetBranchStatus("hiNevtPlane",1);
float vz;
Int_t hiBin;
Float_t hiEvtPlanes[29]; //[hiNevtPlane]
treeHiEvt->SetBranchAddress("hiEvtPlanes",&hiEvtPlanes);
treeHiEvt->SetBranchAddress("vz",&vz);
treeHiEvt->SetBranchAddress("hiBin",&hiBin);
// specify explicitly which branches to store, do not use wildcard
treeSkim->SetBranchStatus("*",0);
treeSkim->SetBranchStatus("ana_step",1);
treeSkim->SetBranchStatus("pcollisionEventSelection",1);
treeSkim->SetBranchStatus("pHBHENoiseFilterResultProducer",1);
treeSkim->SetBranchStatus("HBHENoiseFilterResultRun1",1);
treeSkim->SetBranchStatus("HBHENoiseFilterResultRun2Loose",1);
treeSkim->SetBranchStatus("HBHENoiseFilterResultRun2Tight",1);
treeSkim->SetBranchStatus("HBHENoiseFilterResult",1);
treeSkim->SetBranchStatus("HBHEIsoNoiseFilterResult",1);
Int_t pcollisionEventSelection;
if (treeSkim->GetBranch("pcollisionEventSelection")) {
treeSkim->SetBranchAddress("pcollisionEventSelection",&pcollisionEventSelection);
}
else { // overwrite to default
pcollisionEventSelection = 1;
std::cout<<"could not get branch : pcollisionEventSelection"<<std::endl;
std::cout<<"set to default value : pcollisionEventSelection = "<<pcollisionEventSelection<<std::endl;
}
// event information
UInt_t run, lumis;
ULong64_t event;
treeEvent->SetBranchAddress("run", &run);
treeEvent->SetBranchAddress("event", &event);
treeEvent->SetBranchAddress("lumis", &lumis);
TFile* output = new TFile(outputFile,"RECREATE");
TTree *configTree = setupConfigurationTreeForWriting(configCuts);
int centBinWidth = 200/nCentralityBins; // number of "hiBin"s that a centrality bin covers
int vertexBinWidth = 30/nVertexBins; // number of "vz"s that a vertex bin covers
float eventPlaneBinWidth = TMath::Pi()/nEventPlaneBins; // number of "vz"s that a vertex bin covers
// accepted vz range is -15 to 15.
TTree *outputTreesHLT[nCentralityBins][nVertexBins][nEventPlaneBins];
TTree *outputTreesEvent[nCentralityBins][nVertexBins][nEventPlaneBins];
TTree *outputTreesJet[nCentralityBins][nVertexBins][nEventPlaneBins][nJetCollections];
TTree *outputTreesHiEvt[nCentralityBins][nVertexBins][nEventPlaneBins];
TTree *outputTreesSkim[nCentralityBins][nVertexBins][nEventPlaneBins];
Long64_t nEntriesFilled[nCentralityBins][nVertexBins][nEventPlaneBins]; // number of events read for a centrality bin
for(int i=0; i<nCentralityBins; ++i)
{
for(int j=0; j<nVertexBins; ++j){
for(int l=0; l<nEventPlaneBins; ++l) {
nEntriesFilled[i][j][l] = 0;
outputTreesHLT[i][j][l] = treeHLT->CloneTree(0);
outputTreesHLT[i][j][l]->SetName(Form("hltTree_centBin%d_vzBin%d_evPlaneBin%d", i, j, l));
outputTreesHLT[i][j][l]->SetTitle(Form("subbranches of hltanalysis/HltTree for centrality bin %d vz bin %d evPlane Bin %d", i, j, l));
outputTreesHLT[i][j][l]->SetMaxTreeSize(MAXTREESIZE);
outputTreesEvent[i][j][l] = treeEvent->CloneTree(0);
outputTreesEvent[i][j][l]->SetName(Form("EventTree_centBin%d_vzBin%d_evPlaneBin%d", i, j, l));
outputTreesEvent[i][j][l]->SetTitle(Form("subbranches of ggHiNtuplizer/EventTree for centrality bin %d vz bin %d evPlane Bin %d", i, j, l));
outputTreesEvent[i][j][l]->SetMaxTreeSize(MAXTREESIZE);
for(int k=0; k<nJetCollections; ++k){
std::string jetCollectionName = jetCollections.at(k).c_str();
outputTreesJet[i][j][l][k] = treeJets[k]->CloneTree(0);
outputTreesJet[i][j][l][k]->SetName(Form("%s_centBin%d_vzBin%d_evPlaneBin%d", jetCollectionName.c_str(), i, j, l));
outputTreesJet[i][j][l][k]->SetTitle(Form("subbranches of %s/t for centrality bin %d vz bin %d evPlane Bin %d", jetCollectionName.c_str(), i, j, l));
outputTreesJet[i][j][l][k]->SetMaxTreeSize(MAXTREESIZE);
}
outputTreesHiEvt[i][j][l] = treeHiEvt->CloneTree(0);
outputTreesHiEvt[i][j][l]->SetName(Form("HiEvt_centBin%d_vzBin%d_evPlaneBin%d", i, j, l));
outputTreesHiEvt[i][j][l]->SetTitle(Form("subbranches of hiEvtAnalyzer/HiTree for centrality bin %d vz bin %d evPlane Bin %d", i, j, l));
outputTreesHiEvt[i][j][l]->SetMaxTreeSize(MAXTREESIZE);
outputTreesSkim[i][j][l] = treeSkim->CloneTree(0);
outputTreesSkim[i][j][l]->SetName(Form("skim_centBin%d_vzBin%d_evPlaneBin%d", i, j, l));
outputTreesSkim[i][j][l]->SetTitle(Form("subbranches of skimanalysis/HltTree for centrality bin %d vz bin %d evPlane Bin %d", i, j, l));
outputTreesSkim[i][j][l]->SetMaxTreeSize(MAXTREESIZE);
}
}
}
// TTree* outputTreeHiForestInfo = treeHiForestInfo->CloneTree(0);
// outputTreeHiForestInfo->SetName("HiForestInfo");
// outputTreeHiForestInfo->SetTitle("first entry of HiForest/HiForestInfo");
//
// outputTreeHiForestInfo->SetMaxTreeSize(MAXTREESIZE);
//
// // write HiForestInfo
// treeHiForestInfo->GetEntry(0);
// outputTreeHiForestInfo->Fill();
EventMatcher* em = new EventMatcher();
Long64_t duplicateEntries = 0;
Long64_t entries = treeEvent->GetEntries();
Long64_t entriesAnalyzed = 0;
std::cout << "entries = " << entries << std::endl;
std::cout<< "Loop : ggHiNtuplizer/EventTree" <<std::endl;
for (Long64_t j_entry=0; j_entry<entries; ++j_entry)
{
if (j_entry % 20000 == 0) {
std::cout << "current entry = " <<j_entry<<" out of "<<entries<<" : "<<std::setprecision(2)<<(double)j_entry/entries*100<<" %"<<std::endl;
}
treeHLT->GetEntry(j_entry);
treeEvent->GetEntry(j_entry);
for (int i = 0; i<nJetCollections; ++i) {
treeJets[i]->GetEntry(j_entry);
}
treeHiEvt->GetEntry(j_entry);
treeSkim->GetEntry(j_entry);
bool eventAdded = em->addEvent(run,lumis,event,j_entry);
if(!eventAdded) // this event is duplicate, skip this one.
{
duplicateEntries++;
continue;
}
// event selection
if(!(TMath::Abs(vz) < cut_vz)) continue;
if(pcollisionEventSelection < cut_pcollisionEventSelection) continue;
entriesAnalyzed++;
int centBin = hiBin / centBinWidth;
int vzBin = (vz+15) / vertexBinWidth;
int evplaneBin = (hiEvtPlanes[8]+(TMath::Pi()/2.)) / eventPlaneBinWidth;
// std::cout<<evplaneBin<< " bin " << eventPlaneBinWidth << " value " << hiEvtPlanes[8]+(TMath::Pi()/2.) << std::endl;
if (nEntriesFilled[centBin][vzBin][evplaneBin] > nMaxEvents_minBiasMixing) continue;
outputTreesHLT[centBin][vzBin][evplaneBin]->Fill();
outputTreesEvent[centBin][vzBin][evplaneBin]->Fill();
for (int i = 0; i<nJetCollections; ++i) {
outputTreesJet[centBin][vzBin][evplaneBin][i]->Fill();
}
outputTreesHiEvt[centBin][vzBin][evplaneBin]->Fill();
outputTreesSkim[centBin][vzBin][evplaneBin]->Fill();
nEntriesFilled[centBin][vzBin][evplaneBin]++;
}
std::cout<< "Loop ENDED : ggHiNtuplizer/EventTree" <<std::endl;
std::cout << "entries = " << entries << std::endl;
std::cout << "duplicateEntries = " << duplicateEntries << std::endl;
std::cout << "entriesAnalyzed = " << entriesAnalyzed << std::endl;
for(int i=0; i<nCentralityBins; ++i){
for(int j=0; j<nVertexBins; ++j){
for(int l=0; l<nEventPlaneBins; ++l){
std::cout<<Form("outputTreesJet[%d][%d][%d]->GetEntries() = %lld", i, j, l, outputTreesEvent[i][j][l]->GetEntries())<<std::endl;
if (outputTreesEvent[i][j][l]->GetEntries() < 100) {
std::cout<< "Be careful : less than 100 events were put into centBin = "<<i<<" , vertexBin = "<<j<<" , EventPlaneBin = "<<l<<std::endl;
}
}
}
}
configTree->Write("",TObject::kOverwrite);
output->Write("",TObject::kOverwrite);
output->Close();
std::cout<<"minBiasJetSkim() - END"<<std::endl;
}
