void analyzeTrackingCorrection(
TString jetAlgo = "ak3PF",
TString trkCol = "ppTrack",
// TString trkCol = "anaTrack",
// TString inname="/mnt/hadoop/cms/store/user/yenjie/HiForest_v27/Dijet${pthat}_HydjetDrum_v27_mergedV1.root",
// TString inname="/mnt/hadoop/cms/store/user/yenjie/HiForest_v27/Dijet80_HydjetDrum_v27_mergedV1.root",
// TString outname="trkcorr/test/TrkCorrtest_hy18dj100.root",
TString coll = "PbP",
// double samplePtHat=0,
// double ptHatMax=9999,
// double sampleWeight = 1,
double cutPtTrk=0.5,
double vzMax = 15,
int maxEntries = -1,
double leadingJetPtMin=100,
double subleadingJetPtMin=-1,
double sigDPhi=-1,
bool genJetMode=false
) {
samplePtHat = atoi(getenv("PTHAT")) ;
cout <<"PTHAT =" <<samplePtHat<<endl ;
// samplePtHat = pthat ;
double xsection=0;
double xup=0;
double xsub=0;
TString COM = "5TeV" ;
if(coll=="PP" || coll=="PbPb"){
if(samplePtHat==15){
ptHatMax=30;
xup =2.034E-01;
xsub=1.075E-02;
}
else if(samplePtHat==30){
ptHatMax=50;
xup =1.075E-02;
xsub=1.025E-03;
}
else if(samplePtHat==50){
ptHatMax=80;
xup =1.025E-03;
xsub=9.865E-05;
}
else if(samplePtHat==80){
ptHatMax=120;
xup =9.865E-05;
xsub=1.129E-05;
}
else if(samplePtHat==120){
ptHatMax=170;
xup=1.129E-05;
xsub=1.465E-06;
}else if(samplePtHat==170){
ptHatMax=200;
xup=1.465E-06;
xsub=5.310E-07;
}else if(samplePtHat==200){
ptHatMax=250;
xup =5.310E-07;
xsub=1.192E-07;
}else if(samplePtHat==250){
ptHatMax=300;
xup =1.192E-07;
xsub=3.176E-08;
}else if(samplePtHat==300){
ptHatMax=9999;
xup =3.176E-08;
xsub=0;
}
}
else { //for pPb at 5 TeV cross section
if(samplePtHat==15){
ptHatMax=30;
xup =5.335E-01;
xsub=3.378E-02;
}
else if(samplePtHat==30){
ptHatMax=50;
xup =3.378E-02;
xsub=3.778E-03;
}
else if(samplePtHat==50){
ptHatMax=80;
xup =3.778E-03;
xsub=4.412E-04;
}
else if(samplePtHat==80){
ptHatMax=120;
xup =4.412E-04;
xsub=6.147E-05;
}
else if(samplePtHat==120){
ptHatMax=170;
xup=6.147E-05;
xsub=1.018E-05;
}else if(samplePtHat==170){
ptHatMax=220;
xup=1.018E-05;
xsub=2.477E-06;
}else if(samplePtHat==220){
ptHatMax=280;
xup =2.477E-06;
xsub=6.160E-07;
}else if(samplePtHat==280){
// ptHatMax=9999;
ptHatMax=370;
xup =6.160E-07;
xsub=1.088E-07;
// xsub=0;
}else if(samplePtHat==370){
ptHatMax=9999;
xup=1.088E-07;
xsub=0;
}
}
xsection = xup-xsub;
std::cout<<std::endl;
std::cout<<std::endl;
if(coll=="PP"){
// dataPath= Form("/net/hisrv0001/home/zhukova/scratch/HIHighPt/forest/pthat%d", samplePtHat); //2012 pp MC with HI tracking
// dataPath= Form("/mnt/hadoop/cms/store/user/dgulhan/ppHiIterativeTrack/P01/prod24/Signal_Pythia_pt%d/HiForest_v84_merged01", samplePtHat); //MIT path for 2013 pp 2.76 TeV with HI tracking
dataPath= Form("/mnt/hadoop/cms/store/user/dgulhan/pPb/HP04/prod16/Signal_Pythia_pt%d/HiForest_v77_v2_merged01", samplePtHat); //MIT path for 2013 pp 5 TeV with pp tracking
// dataPath= Form("/mnt/hadoop/cms/store/user/dgulhan/pp2013/P01/prod22/Signal_Pythia_pt%d/HiForest_v81_merged01", samplePtHat); //lxplus path for 2013 pp with pp tracking
// intputFile=Form("mergedFile.root"); //! 2012 pp MC with HI tracking
// intputFile=Form("pt%d_JEC_ppHiIterativeTrack_P01_prod24_v84_merged_forest_0.root", samplePtHat); //! 2013 pp 2.76 TeV with HI tracking
intputFile=Form("pt%d_HP04_hiforest77_hiSignal.root", samplePtHat); //! 2013 pp 5 TeV with pp tracking
// intputFile=Form("pt%d_pp2013_P01_prod22_v81_merged_forest_0.root", samplePtHat); //pp tracking
}
else if(coll=="PbPb"){
if(samplePtHat==50||samplePtHat==80||samplePtHat==100||samplePtHat==170){
dataPath= Form("/mnt/hadoop/cms/store/user/yenjie/HiForest_v27/"); //MIT MC normial
intputFile=Form("Dijet%d_HydjetDrum_v27_mergedV1.root", samplePtHat);
}
else {
dataPath= Form("/mnt/hadoop/cms/store/user/yenjie/HiForest_v28/"); //MIT MC normial
intputFile=Form("Dijet%d_HydjetDrum_v28_mergedV1.root", samplePtHat);
}
}
else {
// dataPath= Form("/mnt/hadoop/cms/store/user/dgulhan/pPb/HP04/prod16/Hijing_Pythia_pt%d/HiForest_v77_merged01", samplePtHat); //MC pPb sample
// intputFile=Form("pt%d_HP04_prod16_v77_merged_forest_0.root",samplePtHat);
//for Pbp MC input
dataPath=Form("/mnt/hadoop/cms/store/user/dgulhan/Pbp/HP05/prod24/Hijing_Pythia_pt%d/HiForest_v84_merged02", samplePtHat); //! MC Pbp sample
intputFile=Form("pt%d_HP05_prod24_v84_merged_forest_0.root",samplePtHat);
}
TString inname=Form("%s/%s", dataPath.Data(),intputFile.Data());
// TString outname=Form("TrkCorrtest_pp2013HITracking_Sigdj%d.root",samplePtHat);
// TString outname=Form("TrkCorr_pythia5TeV_Sigdj%d.root",samplePtHat);
TString outname=Form("TrkCorr_%s%s%sEff_dj%d.root",COM.Data(), coll.Data(),trkCol.Data(),samplePtHat);
TH1::SetDefaultSumw2();
TString tag=Form("%s_%.0f_%.0f_%.0f_genJetMode%d",jetAlgo.Data(),leadingJetPtMin,subleadingJetPtMin,sigDPhi*1000,genJetMode);
outname.ReplaceAll(".root",Form("_%s.root",tag.Data()));
cout << "Input: " << inname << endl;
cout << "Sample pthat = " << samplePtHat << " ptHatMax = " << ptHatMax << endl;
cout << "Track pt min = " << cutPtTrk << endl;
cout << "skim: leading Jet > " << leadingJetPtMin << " subleading > " << subleadingJetPtMin << " dphi > " << sigDPhi << endl;
cout << "Genjet mode: " << genJetMode << endl;
cout << "Output: " << outname << endl;
///////////////////////////////////////////////////
// Setup Analysis
///////////////////////////////////////////////////
double cutjetPt = 40;
double cutjetEta = 2;
double cutEtaTrk = 2.4;
// Define the input file and HiForest
// HiForest * c = new HiForest(inname,"mergedTrack");
// HiForest * c = new HiForest(inname,trkCol.Data());
HiForest * c ;
if(COM=="5TeV")
c = new HiForest(inname.Data(),trkCol.Data(),cPPb);
else if(coll=="PP")
c = new HiForest(inname.Data(),trkCol.Data(),cPP);
else
c = new HiForest(inname.Data(),trkCol.Data(),cPbPb);
c->doTrackCorrections = false;
c->InitTree();
// intialize jet variables
Jets * anajet = 0;
if (jetAlgo=="akPu3PF") anajet = &(c->akPu3PF);
else if (jetAlgo=="ak3PF") anajet = &(c->ak3PF);
else {
cout << "Fatal Error: jetalgo " << jetAlgo << " not defined" << endl;
exit(1);
}
// Output file
cout << "Output: " << outname << endl;
TFile *output = new TFile(outname,"recreate");
EvtSel evt;
DiJet gj;
// TrkReso trkRes1(1.6);
///////////////////////////////////////////////////
// Book Histograms
///////////////////////////////////////////////////
// Tracking Corrections
TrkCorrHisAna effMergedGeneral("Forest2_MergedGeneral",output,40,coll.Data());
effMergedGeneral.DeclareHistograms();
TrkCorrHisAna effMergedGeneral_j1("Forest2_MergedGeneral_j1",output,40,coll.Data());
effMergedGeneral_j1.DeclareHistograms();
TrkCorrHisAna effMergedGeneral_j2("Forest2_MergedGeneral_j2",output,40,coll.Data());
effMergedGeneral_j2.DeclareHistograms();
TrkCorrHisAna effMergedGeneral_jetfine("Forest2_MergedGeneral_jetfine",output,40,coll.Data());
effMergedGeneral_jetfine.jetBins.clear();
// float fineJetBins[10] = {0,40,80,100,120,160,200,250,500,1000};
float fineJetBins[] = {0,40, 60, 80,100, 120, 160, 200, 300, 500, 1000};
effMergedGeneral_jetfine.jetBins.insert(effMergedGeneral_jetfine.jetBins.begin(),fineJetBins,fineJetBins+9);
effMergedGeneral_jetfine.DeclareHistograms();
// basics
output->cd();
TH1D * hCent = new TH1D("hCent","",100,0,100);
TH1D * hVz = new TH1D("hVz","",60,-30,30);
TH1D * hPtHatBeforeSel = new TH1D("hPtHatBeforeSel","",200,0,1000);
TH1D * hPtHat = new TH1D("hPtHat","",200,0,1000);
TH2D * hJetPt2D = new TH2D("hJetPt2D","",100,0,500,100,0,500);
TH1D * hJDPhi = new TH1D("hJDPhi","",40,0,Pi());
TH1D * hAj = new TH1D("hAj","",32,0,0.8);
// In separate centrality bins
vector<TH1D*> vhCent;
vector<TH1D*> vhPtHat;
vector<TH1D*> vhPtHatBeforeSel;
vector<TH2D*> vhJetPt2D;
for (int ib=0; ib<(Int_t)effMergedGeneral.centBins.size()-1; ++ib) {
vhCent.push_back(new TH1D(Form("hCent_c%d",ib),
// Form("cbin%dto%d;Centrality Bin;",effMergedGeneral.centBins[ib],effMergedGeneral.centBins[ib+1]),40,0,40));
Form("cbin%dto%d;Centrality Bin;",effMergedGeneral.centBins[ib],effMergedGeneral.centBins[ib+1]),100,0,100));
vhPtHat.push_back(new TH1D(Form("hPtHat_c%d",ib),"",200,0,1000));
vhPtHatBeforeSel.push_back(new TH1D(Form("hPtHatBeforeSel_c%d",ib),"",200,0,1000));
vhJetPt2D.push_back(new TH2D(Form("hJetPt2D_c%d",ib),"",100,0,500,100,0,500));
}
///////////////////////////////////////////////////
// Main loop
///////////////////////////////////////////////////
if (maxEntries<0) maxEntries = c->GetEntries();
for (int i=0;i<maxEntries;i++) {
c->GetEntry(i);
evt.cBin = c->evt.hiBin;
evt.evtPlane = c->evt.hiEvtPlanes[21];
evt.nJ = anajet->nref;
evt.nT = c->track.nTrk;
evt.offlSel = (c->skim.pcollisionEventSelection > 0);
if (!c->hasSkimTree) evt.offlSel = (c->evt.hiNtracks>0 && c->evt.hiHFplus>=4 && c->evt.hiHFminus>=4);
evt.vz = c->track.vz[1];
// evt.sampleWeight = sampleWeight/maxEntries; // for different mc sample, 1 for data
sampleWeight = xsection/(maxEntries/100000.);
evt.sampleWeight = sampleWeight; // for different mc sample, 1 for data
evt.pthat = anajet->pthat;
evt.samplePtHat = samplePtHat;
if (i%1000==0) cout <<i<<" / "<< maxEntries << " pthat: " << evt.pthat << " cBin: " << evt.cBin << " nT: " << evt.nT <<endl;
// initialize
int leadingIndex=-1,genLeadingIndex=-1;
int awayIndex=-1,genAwayIndex=-1;
gj.clear();
///////////////////////////////////////////////////////
// Find Leading jets
///////////////////////////////////////////////////////
for (int j=0;j<anajet->nref;j++) {
if (anajet->jtpt[j]<cutjetPt) continue;
if (fabs(anajet->jteta[j])>cutjetEta) continue;
if(anajet->trackMax[j]/anajet->jtpt[j] <=0.01) continue ;
if (anajet->jtpt[j] > gj.pt1) {
gj.pt1 = anajet->jtpt[j];
leadingIndex = j;
}
}
// Found a leading jet which passed basic quality cut!
if (leadingIndex!=-1) {
// set leading jet
gj.pt1raw=anajet->rawpt[leadingIndex];
gj.eta1=anajet->jteta[leadingIndex];
gj.phi1=anajet->jtphi[leadingIndex];
gj.ref1pt = anajet->refpt[leadingIndex];
gj.ref1eta = anajet->refeta[leadingIndex];
gj.ref1phi = anajet->refphi[leadingIndex];
gj.ref1partonpt = anajet->refparton_pt[leadingIndex];
gj.ref1partonflavor = anajet->refparton_flavor[leadingIndex];
// Loop over jet tree to find a away side leading jet
gj.nJet=0;
for (int j=0;j<anajet->nref;j++) {
if (anajet->jtpt[j]<cutjetPt) continue;
if (fabs(anajet->jteta[j])>cutjetEta) continue;
if(anajet->trackMax[j]/anajet->jtpt[j] <=0.01) continue ;
gj.inclJetPt[gj.nJet] = anajet->jtpt[j];
gj.inclJetEta[gj.nJet] = anajet->jteta[j];
gj.inclJetPhi[gj.nJet] = anajet->jtphi[j];
gj.inclJetRefPt[gj.nJet] = anajet->refpt[j];
gj.inclJetRefPartonPt[gj.nJet] = anajet->refparton_pt[j];
if (j!=leadingIndex&&anajet->jtpt[j] > gj.pt2) {
gj.pt2 = anajet->jtpt[j];
awayIndex = j;
}
++gj.nJet;
}
if (awayIndex !=-1) { // Found an away jet!
gj.pt2raw = anajet->rawpt[awayIndex];
gj.eta2 = anajet->jteta[awayIndex];
gj.phi2 = anajet->jtphi[awayIndex];
gj.deta = gj.eta2 - gj.eta1;
gj.dphi = deltaPhi(gj.phi2,gj.phi1);
gj.Aj = (gj.pt1-gj.pt2)/(gj.pt1+gj.pt2);
gj.ref2pt = anajet->refpt[awayIndex];
gj.ref2eta = anajet->refeta[awayIndex];
gj.ref2phi = anajet->refphi[awayIndex];
gj.ref2partonpt = anajet->refparton_pt[awayIndex];
gj.ref2partonflavor = anajet->refparton_flavor[awayIndex];
}
} // end of if leadingIndex
////////////////////////////////////////
// Find Leading Genjets
////////////////////////////////////////
// Loop over genjets to look for leading genjet candidate in the event
for (int j=0;j<anajet->ngen;j++) {
if (anajet->genpt[j]<cutjetPt) continue;
if (fabs(anajet->geneta[j])>cutjetEta) continue;
if (anajet->genpt[j] > gj.genjetpt1) {
gj.genjetpt1=anajet->genpt[j];
genLeadingIndex=j;
}
}
if (genLeadingIndex!=-1) {
gj.genjeteta1=anajet->geneta[genLeadingIndex];
gj.genjetphi1=anajet->genphi[genLeadingIndex];
}
// subleading
gj.nGenJet=0;
for (int j=0;j<anajet->ngen;j++) {
if (anajet->genpt[j]<cutjetPt) continue;
if (fabs(anajet->geneta[j])>cutjetEta) continue;
gj.inclGenJetPt[gj.nGenJet] = anajet->genpt[j];
gj.inclGenJetEta[gj.nGenJet] = anajet->geneta[j];
gj.inclGenJetPhi[gj.nGenJet] = anajet->genphi[j];
if (j!=genLeadingIndex && anajet->genpt[j]>gj.genjetpt2) {
gj.genjetpt2=anajet->genpt[j];
gj.genjeteta2=anajet->geneta[j];
gj.genjetphi2=anajet->genphi[j];
}
++gj.nGenJet;
}
if (genJetMode) {
gj.pt1 = gj.genjetpt1;
gj.eta1 = gj.genjeteta1;
gj.phi1 = gj.genjetphi1;
gj.pt2 = gj.genjetpt2;
gj.eta2 = gj.genjeteta2;
gj.phi2 = gj.genjetphi2;
gj.dphi = deltaPhi(gj.phi2,gj.phi1);
}
///////////////////////////////////////////////////////
// Skim
///////////////////////////////////////////////////////
if (!evt.offlSel) continue;
// if (samplePtHat>0 && evt.pthat>=ptHatMax) continue;
if (evt.pthat>=ptHatMax) continue;
if (vzMax>0 && fabs(evt.vz)>vzMax) continue;
// protection against high pt jet from background event
if (leadingIndex>=0&&anajet->subid[leadingIndex]>0) continue;
hPtHatBeforeSel->Fill(evt.pthat);
for (int ib=0; ib<(Int_t)effMergedGeneral.centBins.size(); ++ib) {
if(evt.cBin>=effMergedGeneral.centBins[ib] && evt.cBin<effMergedGeneral.centBins[ib+1]){
vhPtHatBeforeSel[ib]->Fill(evt.pthat);
}
}
// ensure jet distribution unbiased by pthat turn on
if (leadingJetPtMin>0&&gj.pt1<leadingJetPtMin) continue;
if (subleadingJetPtMin>0&&gj.pt2<subleadingJetPtMin) continue;
if (sigDPhi>0&&fabs(gj.dphi)<sigDPhi) continue;
// Fill Baisc Event info
hCent->Fill(evt.cBin);
hVz->Fill(evt.vz);
hPtHat->Fill(evt.pthat);
hJetPt2D->Fill(gj.pt1,gj.pt2);
hJDPhi->Fill(fabs(gj.dphi));
hAj->Fill(gj.Aj);
for (int ib=0; ib<(Int_t)effMergedGeneral.centBins.size(); ++ib) {
if(evt.cBin>=effMergedGeneral.centBins[ib] && evt.cBin<effMergedGeneral.centBins[ib+1]){
vhCent[ib]->Fill(evt.cBin);
vhPtHat[ib]->Fill(evt.pthat);
vhJetPt2D[ib]->Fill(gj.pt1,gj.pt2);
}
}
///////////////////////////////////////////////////////
// Tracks
///////////////////////////////////////////////////////
// Full Tracks, Pixel Tracks
Tracks * anaTrks[1] = {&(c->track)};
for (int it=0; it<anaTrks[0]->nTrk; ++it) {
// Kinematic Selection
if (anaTrks[0]->trkPt[it] < cutPtTrk) continue;
if (fabs(anaTrks[0]->trkEta[it]) > cutEtaTrk) continue;
// if (anaTrks[0]->trkAlgo[it]>=4&&!anaTrks[0]->highPurity[it]) continue; // quality selection
if (!anaTrks[0]->highPurity[it]) continue; // quality selection
if(anaTrks[0]->trkPtError[it]/anaTrks[0]->trkPt[it]>=0.1 || TMath::Abs(anaTrks[0]->trkDz1[it]/anaTrks[0]->trkDzError1[it])>=3.0 || TMath::Abs(anaTrks[0]->trkDxy1[it]/anaTrks[0]->trkDxyError1[it])>=3.0) continue ; //ridge cut for tracks
RecTrack_t r;
r.ptr = anaTrks[0]->trkPt[it];
r.etar = anaTrks[0]->trkEta[it];
r.phir = anaTrks[0]->trkPhi[it];
r.algo = anaTrks[0]->trkAlgo[it];
r.nsim = !anaTrks[0]->trkFake[it];
r.status = 1; // for now correct all tracks
float dr1 = deltaR(r.etar,r.phir,gj.eta1,gj.phi1);
float dr2 = deltaR(r.etar,r.phir,gj.eta2,gj.phi2);
r.jet = 0;
r.jeta = -99;
r.jdr = -99;
if (samplePtHat>0) {
if (dr1<0.5&&gj.pt1>=40) {
r.jet = gj.pt1;
r.jeta = gj.eta1;
r.jdr = dr1;
effMergedGeneral_j1.FillRecHistograms(evt,gj,r);
} else if (dr2<0.5&&gj.pt2>=40) {
r.jet = gj.pt2;
r.jeta = gj.eta2;
r.jdr = dr2;
effMergedGeneral_j2.FillRecHistograms(evt,gj,r);
}
}
// Fill
effMergedGeneral.FillRecHistograms(evt,gj,r);
effMergedGeneral_jetfine.FillRecHistograms(evt,gj,r);
// effMergedGeneral_trkPhi.FillRecHistograms(evt,gj,r);
// if (r.jet>=120) effMergedGeneral_trkPhi_jet_120to999.FillRecHistograms(evt,gj,r);
// else if (r.jet>=cutjetPt) effMergedGeneral_trkPhi_jet_50to120.FillRecHistograms(evt,gj,r);
// else effMergedGeneral_trkPhi_noJet.FillRecHistograms(evt,gj,r);
// effMergedGeneral_SimSmear.FillRecHistograms(evt,gj,r);
}
///////////////////////////////////////////////////////
// SimTracks
///////////////////////////////////////////////////////
for (int ip=0; ip<anaTrks[0]->nParticle; ++ip) {
if (anaTrks[0]->pPt[ip] < cutPtTrk) continue;
if (fabs(anaTrks[0]->pEta[ip]) > cutEtaTrk) continue;
SimTrack_t s;
s.pts = anaTrks[0]->pPt[ip];
s.etas = anaTrks[0]->pEta[ip];
s.phis = anaTrks[0]->pPhi[ip];
s.status = 1; // for now assume all simtracks as signal tracks
// s.nrec = (anaTrks[0]->pNRec[ip])*(int)(anaTrks[0]->mtrkAlgo[ip]<4||anaTrks[0]->mtrkQual[ip]>0);
// s.nrec = (anaTrks[0]->pNRec[ip])*(int)(anaTrks[0]->mtrkAlgo[ip]<4||anaTrks[0]->mtrkQual[ip]>0)*(anaTrks[0]->mtrkPtError[ip]/anaTrks[0]->mtrkPt[ip]<0.1 && anaTrks[0]->mtrkDz1[ip]/anaTrks[0]->mtrkDzError1[ip]<3.0 && anaTrks[0]->mtrkDxy1[ip]/anaTrks[0]->mtrkDxyError1[ip]<3.0);
s.nrec = (anaTrks[0]->pNRec[ip])*(int)(anaTrks[0]->mtrkQual[ip]>0)*(anaTrks[0]->mtrkPtError[ip]/anaTrks[0]->mtrkPt[ip]<0.1 && TMath::Abs(anaTrks[0]->mtrkDz1[ip]/anaTrks[0]->mtrkDzError1[ip])<3.0 && TMath::Abs(anaTrks[0]->mtrkDxy1[ip]/anaTrks[0]->mtrkDxyError1[ip])<3.0);
// s.nrec = (int)(anaTrks[0]->mtrkAlgo[ip]<4||anaTrks[0]->mtrkQual[ip]>0)*(anaTrks[0]->mtrkPtError[ip]/anaTrks[0]->mtrkPt[ip]<0.1 && TMath::Abs(anaTrks[0]->mtrkDz1[ip]/anaTrks[0]->mtrkDzError1[ip])<3.0 && TMath::Abs(anaTrks[0]->mtrkDxy1[ip]/anaTrks[0]->mtrkDxyError1[ip])<3.0);
// if (s.pts<1.5) s.nrec = anaTrks[0]->pNRec[ip];
// else s.nrec = (anaTrks[0]->pNRec[ip])*(anaTrks[0]->mtrkQual[ip]);
s.acc = anaTrks[0]->pAcc[ip];
s.jet = gj.pt1;
s.jeta = gj.eta1;
float dr1 = deltaR(s.etas,s.phis,gj.eta1,gj.phi1);
float dr2 = deltaR(s.etas,s.phis,gj.eta2,gj.phi2);
s.jet = 0;
s.jeta = -99;
s.jdr = -99;
if (samplePtHat>0) {
if (dr1<0.5&&gj.pt1>=40) {
s.jet = gj.pt1;
s.jeta = gj.eta1;
s.jdr = dr1;
effMergedGeneral_j1.FillSimHistograms(evt,gj,s);
} else if (dr2<0.5&&gj.pt2>=40) {
s.jet = gj.pt2;
s.jeta = gj.eta2;
s.jdr = dr2;
effMergedGeneral_j2.FillSimHistograms(evt,gj,s);
}
}
// Fill
effMergedGeneral.FillSimHistograms(evt,gj,s);
effMergedGeneral_jetfine.FillSimHistograms(evt,gj,s);
// effMergedGeneral_trkPhi.FillSimHistograms(evt,gj,s);
// if (s.jet>=120) effMergedGeneral_trkPhi_jet_120to999.FillSimHistograms(evt,gj,s);
// else if (s.jet>=cutjetPt) effMergedGeneral_trkPhi_jet_50to120.FillSimHistograms(evt,gj,s);
// else effMergedGeneral_trkPhi_noJet.FillSimHistograms(evt,gj,s);
// Sim Smearing
// if (s.pts>=1&&s.pts<2) {
// float trksm = trkRes1.GetSmear(s.pts);
// s.pts*=trksm;
// }
// effMergedGeneral_SimSmear.FillSimHistograms(evt,gj,s);
}
// All done
}
// effMergedGeneral.WriteHistograms();
output->Write();
output->Close();
delete c;
}
void analyzeDiJetMPT(
TString jetAlgo = "akPu3PF",
TString trkCol = "anaTrack",
TString inname="/mnt/hadoop/cms/store/user/yenjie/HiForest_v27/Dijet${pthat}_HydjetDrum_v27_mergedV1.root",
TString outname="output.root",
bool isPP=false,
int dataSrcType = 1, // 0 mc, 1 hi, 2 pp 2.76 TeV, 3 pp 7TeV
double samplePtHat=0,
double ptHatMax=9999,
double sampleWeight = 1, // data: 1, mc: s = 0.62, b = 0.38
double vzMax = 15,
int maxEntries = -1,
double leadingJetPtMin=-1,
double subleadingJetPtMin=-1,
double sigDPhi=-1,
bool genJetMode=false,
int smearCentBin = 0,
bool doCentReWeight=false,
TString mcfname="",
TString datafname="output-data-Photon-v7_v30.root"
)
{
///////////////////////////////////////////////////
// Setup Analysis
///////////////////////////////////////////////////
int saveTracks = 1; // 0=none, 1=all, 10=cone
double cutjetPt = 10;
double cutjetEta = 2;
double cutPtTrk=1.;
double cutEtaTrk = 2.4;
double trkJetAssoR=0.3;
TString tag=Form("%s_%.0f_%.0f_%.0f_saveTrk%d_jmin%.0f_tmin%.0f_genj%d_sm%d",jetAlgo.Data(),leadingJetPtMin,subleadingJetPtMin,sigDPhi*1000,saveTracks,cutjetPt,cutPtTrk,genJetMode,smearCentBin);
outname.ReplaceAll(".root",Form("_%s.root",tag.Data()));
cout << "Input: " << inname << " isPP: " << isPP << endl;
cout << "Sample pthat = " << samplePtHat << " ptHatMax = " << ptHatMax << endl;
cout << "Track pt min = " << cutPtTrk << endl;
cout << "skim: leading Jet > " << leadingJetPtMin << " subleading > " << subleadingJetPtMin << " dphi > " << sigDPhi << endl;
cout << "Genjet mode: " << genJetMode << endl;
cout << "SmearCentBin = " << smearCentBin << endl;
cout << "Output: " << outname << endl;
// Centrality reweiting
CentralityReWeight cw(datafname,mcfname,"offlSel&&pt1>100&&pt2>0&&acos(cos(phi2-phi1))>2./3*3.14159");
// Define the input file and HiForest
HiForest * c = new HiForest(inname,jetAlgo.Data(),isPP);
c->doTrackCorrections = true;
c->doTrackingSeparateLeadingSubleading = false;
c->InitTree();
// intialize jet variables
Jets* anajet = &(c->akPu3PF) ;
// Output file
cout << "Output: " << outname << endl;
TFile *output = new TFile(outname,"recreate");
if (doCentReWeight&&mcfname!="") {
cw.Init(); //cw.hCentData->Draw(); cw.hCentMc->Draw("same");
}
// basics
output->cd();
TH1D * hCent = new TH1D("hCent","",40,0,40);
TH1D * hVz = new TH1D("hVz","",60,-30,30);
TH1D * hPtHatBeforeSel = new TH1D("hPtHatBeforeSel","",200,0,1000);
TH1D * hPtHat = new TH1D("hPtHat","",200,0,1000);
TH2D * hJetPt2D = new TH2D("hJetPt2D","",100,0,500,100,0,500);
TH1D * hJDPhi = new TH1D("hJDPhi","",40,0,Pi());
TH1D * hAj = new TH1D("hAj","",32,0,0.8);
TH1D * hTrkPt = new TH1D("hTrkPt",";p_{T} (GeV/c)",200,0,200);
TH1D * hTrkCorrPt = new TH1D("hTrkCorrPt",";p_{T} (GeV/c)",200,0,200);
TH1D * hGenpPt = new TH1D("hGenpPt",";p_{T} (GeV/c)",200,0,200);
TH1D* smearingHist=0;
if ( smearCentBin > 0 ) {
smearingHist = new TH1D("smearingH","",100,-2,2);
}
EvtSel evt;
DiJet gj;
TTree * tgj = new TTree("tgj","dijet jet tree");
BookGJBranches(tgj,evt,gj);
// pp triggers
int HLT_Jet40_v1=0;
if (dataSrcType==2) {
c->hltTree->SetBranchAddress("HLT_Jet40_v1",&HLT_Jet40_v1);
} else if (dataSrcType==3) {
c->hltTree->SetBranchAddress("HLT_Jet40_v1",&HLT_Jet40_v1);
}
////////////////////////
// Smearing Setup
////////////////////////
if (smearCentBin>0) {
LoadParameters();
}
///////////////////////////////////////////////////
// Main loop
///////////////////////////////////////////////////
if (maxEntries<0||maxEntries > c->GetEntries()) maxEntries = c->GetEntries();
for (int i=0;i<maxEntries;i++) {
c->GetEntry(i);
// Event Info
evt.run = c->hlt.Run;
evt.evt = c->hlt.Event;
evt.cBin = c->evt.hiBin;
if (isPP) evt.cBin = 39;
evt.evtPlane = c->evt.hiEvtPlanes[21];
evt.nJ = anajet->nref;
evt.nT = c->track.nTrk;
evt.trig = (c->hlt.HLT_HIJet80_v1 > 0);
evt.offlSel = (c->skim.pcollisionEventSelection > 0);
if (!c->hasSkimTree) evt.offlSel = (c->evt.hiNtracks>0 && c->evt.hiHFplus>=4 && c->evt.hiHFminus>=4);
evt.noiseFilt = (c->skim.pHBHENoiseFilter > 0);
if (dataSrcType>1) {
if (dataSrcType==2) {
evt.trig = (HLT_Jet40_v1>0);
} else if (dataSrcType==3) {
evt.trig = (HLT_Jet40_v1>0);
}
// evt.offlSel = (c->skim.phfCoincFilter && c->skim.ppurityFractionFilter);
}
if (!c->hasHltTree) evt.trig = true;
evt.anaEvtSel = evt.offlSel;
if (dataSrcType>0) evt.anaEvtSel = evt.anaEvtSel && evt.trig && evt.noiseFilt;
evt.vz = c->track.vz[1];
// Get Centrality Weight
if (doCentReWeight) evt.weight = cw.GetWeight(evt.cBin);
else evt.weight = 1;
evt.npart = getNpart(evt.cBin);
evt.ncoll = getNcoll(evt.cBin);
evt.sampleWeight = sampleWeight/maxEntries; // for different mc sample, 1 for data
evt.pthat = anajet->pthat;
evt.samplePtHat = samplePtHat;
if (i%1000==0) cout <<i<<" / "<<maxEntries << " run: " << evt.run << " evt: " << evt.evt << " bin: " << evt.cBin << " nT: " << evt.nT << " trig: " << evt.trig << " anaEvtSel: " << evt.anaEvtSel <<endl;
// initialize
int leadingIndex=-1,genLeadingIndex=-1;
int awayIndex=-1,genAwayIndex=-1;
gj.clear();
// smearing for pp
gj.nUnSmJet=0;
if (smearCentBin>0) {
for (int j=0;j<anajet->nref;j++) {
if (anajet->jtpt[j]<cutjetPt) continue;
if (fabs(anajet->jteta[j])>cutjetEta) continue;
if ((anajet->trackMax[j]/anajet->jtpt[j])<0.01) continue;
//if (anajet->jtpt[j]>200) cout << "Before smearing: " << anajet->jtpt[j] << endl;
int smCentBin = GetCBin( centBin[smearCentBin] );
//cout << "smCentBin: " << smCentBin << endl;
gj.inclJetUnSmPt[gj.nUnSmJet] = anajet->jtpt[j];
anajet->jtpt[j] = GetSmearedPtData(2,smCentBin,anajet->jtpt[j],0,"");
gj.inclJetSm[gj.nUnSmJet] = anajet->jtpt[j]/gj.inclJetUnSmPt[gj.nUnSmJet];
//if (anajet->jtpt[j]>200) cout << "After smearing: " << anajet->jtpt[j] << endl;
if ( (gj.inclJetUnSmPt[gj.nUnSmJet] > 100) && (fabs(anajet->jteta[j])<2.0))
smearingHist->Fill( (anajet->jtpt[j] - gj.inclJetUnSmPt[gj.nUnSmJet] )/gj.inclJetUnSmPt[gj.nUnSmJet] );
++gj.nUnSmJet;
}
}
///////////////////////////////////////////////////////
// Find Leading jets
///////////////////////////////////////////////////////
for (int j=0;j<anajet->nref;j++) {
if (anajet->jtpt[j]<cutjetPt) continue;
if (fabs(anajet->jteta[j])>cutjetEta) continue;
if ((anajet->trackMax[j]/anajet->jtpt[j])<0.01) continue;
if (anajet->jtpt[j] > gj.pt1) {
gj.pt1 = anajet->jtpt[j];
leadingIndex = j;
}
}
// Found a leading jet which passed basic quality cut!
if (leadingIndex!=-1) {
// set leading jet
gj.pt1raw=anajet->rawpt[leadingIndex];
gj.eta1=anajet->jteta[leadingIndex];
gj.phi1=anajet->jtphi[leadingIndex];
gj.ref1pt = anajet->refpt[leadingIndex];
gj.ref1eta = anajet->refeta[leadingIndex];
gj.ref1phi = anajet->refphi[leadingIndex];
gj.ref1partonpt = anajet->refparton_pt[leadingIndex];
gj.ref1partonflavor = anajet->refparton_flavor[leadingIndex];
// Loop over jet tree to find a away side leading jet
gj.nJet=0;
for (int j=0;j<anajet->nref;j++) {
if (anajet->jtpt[j]<cutjetPt) continue;
if (fabs(anajet->jteta[j])>cutjetEta) continue;
if ((anajet->trackMax[j]/anajet->jtpt[j])<0.01) continue;
gj.inclJetPt[gj.nJet] = anajet->jtpt[j];
gj.inclJetEta[gj.nJet] = anajet->jteta[j];
gj.inclJetPhi[gj.nJet] = anajet->jtphi[j];
gj.inclJetRefPt[gj.nJet] = anajet->refpt[j];
gj.inclJetRefPartonPt[gj.nJet] = anajet->refparton_pt[j];
if (j!=leadingIndex&&anajet->jtpt[j] > gj.pt2) {
gj.pt2 = anajet->jtpt[j];
awayIndex = j;
}
++gj.nJet;
}
if (awayIndex !=-1) { // Found an away jet!
gj.pt2raw = anajet->rawpt[awayIndex];
gj.eta2 = anajet->jteta[awayIndex];
gj.phi2 = anajet->jtphi[awayIndex];
gj.deta = gj.eta2 - gj.eta1;
gj.dphi = deltaPhi(gj.phi2,gj.phi1);
gj.Aj = (gj.pt1-gj.pt2)/(gj.pt1+gj.pt2);
gj.ref2pt = anajet->refpt[awayIndex];
gj.ref2eta = anajet->refeta[awayIndex];
gj.ref2phi = anajet->refphi[awayIndex];
gj.ref2partonpt = anajet->refparton_pt[awayIndex];
gj.ref2partonflavor = anajet->refparton_flavor[awayIndex];
}
} // end of if leadingIndex
////////////////////////////////////////
// Find Leading Genjets
////////////////////////////////////////
// Loop over genjets to look for leading genjet candidate in the event
for (int j=0;j<anajet->ngen;j++) {
if (anajet->genpt[j]<cutjetPt) continue;
if (fabs(anajet->geneta[j])>cutjetEta) continue;
if (anajet->genpt[j] > gj.genjetpt1) {
gj.genjetpt1=anajet->genpt[j];
genLeadingIndex=j;
}
}
if (genLeadingIndex!=-1) {
gj.genjeteta1=anajet->geneta[genLeadingIndex];
gj.genjetphi1=anajet->genphi[genLeadingIndex];
}
// subleading
gj.nGenJet=0;
for (int j=0;j<anajet->ngen;j++) {
if (anajet->genpt[j]<cutjetPt) continue;
if (fabs(anajet->geneta[j])>cutjetEta) continue;
gj.inclGenJetPt[gj.nGenJet] = anajet->genpt[j];
gj.inclGenJetEta[gj.nGenJet] = anajet->geneta[j];
gj.inclGenJetPhi[gj.nGenJet] = anajet->genphi[j];
if (j!=genLeadingIndex && anajet->genpt[j]>gj.genjetpt2) {
gj.genjetpt2=anajet->genpt[j];
gj.genjeteta2=anajet->geneta[j];
gj.genjetphi2=anajet->genphi[j];
}
++gj.nGenJet;
}
if (genJetMode) {
gj.pt1 = gj.genjetpt1;
gj.eta1 = gj.genjeteta1;
gj.phi1 = gj.genjetphi1;
gj.pt2 = gj.genjetpt2;
gj.eta2 = gj.genjeteta2;
gj.phi2 = gj.genjetphi2;
gj.dphi = deltaPhi(gj.phi2,gj.phi1);
}
///////////////////////////////////////////////////////
// Skim
///////////////////////////////////////////////////////
if (!evt.anaEvtSel) continue;
if (samplePtHat>0 && evt.pthat>=ptHatMax) continue;
if (vzMax>0 && fabs(evt.vz)>vzMax) continue;
hPtHatBeforeSel->Fill(evt.pthat);
// ensure jet distribution unbiased by pthat turn on
if (leadingJetPtMin>0&&gj.pt1<leadingJetPtMin) continue;
if (subleadingJetPtMin>0&&gj.pt2<subleadingJetPtMin) continue;
if (sigDPhi>0&&fabs(gj.dphi)<sigDPhi) continue;
// Fill Baisc Event info
hCent->Fill(evt.cBin);
hVz->Fill(evt.vz);
hPtHat->Fill(evt.pthat);
hJetPt2D->Fill(gj.pt1,gj.pt2);
hJDPhi->Fill(fabs(gj.dphi));
hAj->Fill(gj.Aj);
///////////////////////////////////////////////////////
// Tracks
///////////////////////////////////////////////////////
bool doTracks = false;
if (doTracks) {
double trkcorr[4];
gj.nTrk=0;
const int nTrkSet=1;
Tracks * anaTrks[nTrkSet] = {&(c->track)};
// Full Tracks, Pixel Tracks
for (int iset=0; iset<nTrkSet; ++iset) {
for (int it=0; it<anaTrks[iset]->nTrk; ++it) {
// Kinematic Selection
if (anaTrks[iset]->trkPt[it] < cutPtTrk) continue;
if (fabs(anaTrks[iset]->trkEta[it]) > cutEtaTrk) continue;
if (!anaTrks[iset]->highPurity[it]) continue;
float trkPt = anaTrks[iset]->trkPt[it];
float trkEta = anaTrks[iset]->trkEta[it];
float trkPhi = anaTrks[iset]->trkPhi[it];
float trkPtErrorNorm = anaTrks[iset]->trkPtError[it]/anaTrks[iset]->trkPt[it];
float trkNHit = anaTrks[iset]->trkNHit[it];
float trkChi2Norm = anaTrks[iset]->trkChi2[it]/anaTrks[iset]->trkNlayer[it]/anaTrks[iset]->trkNdof[it];
float trkDzNorm = anaTrks[iset]->trkDz1[it]/anaTrks[iset]->trkDzError1[it];
float trkDxyNorm = anaTrks[iset]->trkDxy1[it]/anaTrks[iset]->trkDxyError1[it];
int trkAlgo = anaTrks[iset]->trkAlgo[it];
bool trkHP = (trkPtErrorNorm<0.06&&trkNHit>=13&&trkChi2Norm<0.15&&trkDzNorm<3&&trkDxyNorm<3);
gj.trkPt[gj.nTrk] = trkPt;
gj.trkEta[gj.nTrk] = trkEta;
gj.trkPhi[gj.nTrk] = trkPhi;
gj.trkWt[gj.nTrk] = c->getTrackCorrection(it);
float dr1 = deltaR(trkEta,trkPhi,gj.eta1,gj.phi1);
float dr2 = deltaR(trkEta,trkPhi,gj.eta2,gj.phi2);
float dr1bkg = deltaR(trkEta,trkPhi,-gj.eta1,gj.phi1);
float dr2bkg = deltaR(trkEta,trkPhi,-gj.eta2,gj.phi2);
gj.trkAsso[gj.nTrk] = 0;
if (dr1<trkJetAssoR) gj.trkAsso[gj.nTrk] = 1;
else if (dr1bkg<trkJetAssoR) gj.trkAsso[gj.nTrk] = 11;
else if (dr2<trkJetAssoR) gj.trkAsso[gj.nTrk] = 2;
else if (dr2bkg<trkJetAssoR) gj.trkAsso[gj.nTrk] = 12;
// Histogram
hTrkPt->Fill(trkPt);
hTrkCorrPt->Fill(trkPt,gj.trkWt[gj.nTrk]);
// Select what to save
if (saveTracks==0) continue;
if (saveTracks==10&&gj.trkAsso[gj.nTrk]==0) continue;
// Fill
gj.trkNHit[gj.nTrk] = trkNHit;
gj.trkChi2Norm[gj.nTrk] = trkChi2Norm;
gj.trkAlgo[gj.nTrk] = trkAlgo;
gj.trkHP[gj.nTrk] = trkHP;
gj.trkIsFake[gj.nTrk] = anaTrks[iset]->trkFake[it];
++gj.nTrk;
}
}
gj.nSim=0;
for (int ip=0; ip<anaTrks[0]->nParticle; ++ip) {
if (anaTrks[0]->pPt[ip] < cutPtTrk) continue;
if (fabs(anaTrks[0]->pEta[ip]) > cutEtaTrk) continue;
float simPt = anaTrks[0]->pPt[ip];
float simEta = anaTrks[0]->pEta[ip];
float simPhi = anaTrks[0]->pPhi[ip];
float dr1 = deltaR(simEta,simPhi,gj.eta1,gj.phi1);
float dr2 = deltaR(simEta,simPhi,gj.eta2,gj.phi2);
float dr1bkg = deltaR(simEta,simPhi,-gj.eta1,gj.phi1);
float dr2bkg = deltaR(simEta,simPhi,-gj.eta2,gj.phi2);
gj.simAsso[gj.nSim] = 0;
if (dr1<trkJetAssoR) gj.simAsso[gj.nSim] = 1;
else if (dr1bkg<trkJetAssoR) gj.simAsso[gj.nSim] = 11;
else if (dr2<trkJetAssoR) gj.simAsso[gj.nSim] = 2;
else if (dr2bkg<trkJetAssoR) gj.simAsso[gj.nSim] = 12;
// Select what to save
if (saveTracks==0) continue;
if (saveTracks==10&&gj.simAsso[gj.nSim]==0) continue;
// Fill
gj.simPt[gj.nSim] = simPt;
gj.simEta[gj.nSim] = simEta;
gj.simPhi[gj.nSim] = simPhi;
gj.simHasRec[gj.nSim] = (anaTrks[0]->pNRec[ip]==1&&(anaTrks[0]->mtrkAlgo[ip]<4||anaTrks[0]->mtrkQual[ip]>0));
++gj.nSim;
}
}
// All done
tgj->Fill();
}
output->Write();
output->Close();
delete c;
}