1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276 |
- #include "mfPicoDstReader.h"
- #include "mfNamespace.h"
- #include "mfProcessEvent.h"
- #include "mfInputReader.h"
- #include "mfSelector.h"
- #include "mfRecentering.h"
- #include "mfFlattening.h"
- #include "mfAcceptanceFilter.h"
- #include <TStopwatch.h>
- #include <TProfile2D.h>
- #include <TClonesArray.h>
- #include <TFile.h>
- #include <iostream>
- #include <TMath.h>
- #include <TDatabasePDG.h>
- #include <PicoDstMCEvent.h>
- #include <PicoDstRecoEvent.h>
- #include <PicoDstMCTrack.h>
- #include <PicoDstRecoTrack.h>
- #include <PicoDstFHCal.h>
- using mfdca::Ndim;
- using mfEp::EpNames;
- using mfEp::NetaGaps;
- using mfEp::Nflattening;
- using mfEp::NsubEvents;
- using mfEp::SpTpcLEp;
- using mfEp::SpTpcREp;
- using mfEp::TpcEtaGap;
- using mfEp::TpcLEp;
- using mfEp::TpcREp;
- using mfFlow::MinV1PtEta;
- using mfPid::Mass2;
- using mfPid::MinPidProbability;
- using mfPid::Npid;
- using mfRes::NresBin;
- using TMath::Abs;
- using TMath::ATan2;
- using TMath::Cos;
- using TMath::Log;
- using TMath::Pi;
- using TMath::Power;
- using TMath::Sin;
- using TMath::Sqrt;
- int main(int argc, char **argv)
- {
- Bool_t isAcceptanceFilterOn = false;
- TString iFileName, oFileName, dcaFileName, resFileName, recenteringFileName = "", flatteningFileName = "";
- if (argc < 9)
- {
- std::cerr << "./get-flow -i INPUTFILE/LIST -o OUTPUT.root -dca DCA_FIT.root -res RES_FIT.root [OPTIONAL: -recentering RECENTERING.root -flattening FLATTENING.root --bad-acceptance]" << std::endl;
- return 1;
- }
- for (int i = 1; i < argc; i++)
- {
- if (std::string(argv[i]) != "-i" &&
- std::string(argv[i]) != "-o" &&
- std::string(argv[i]) != "-dca" &&
- std::string(argv[i]) != "-res" &&
- std::string(argv[i]) != "-recentering" &&
- std::string(argv[i]) != "-flattening" &&
- std::string(argv[i]) != "--bad-acceptance")
- {
- std::cerr << "\n[ERROR]: Unknown parameter " << i << ": " << argv[i] << std::endl;
- return 2;
- }
- else
- {
- if (std::string(argv[i]) == "-i" && i != argc - 1)
- {
- iFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-i" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "-o" && i != argc - 1)
- {
- oFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-o" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Output file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "-dca" && i != argc - 1)
- {
- dcaFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-dca" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "-res" && i != argc - 1)
- {
- resFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-res" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "-recentering" && i != argc - 1)
- {
- recenteringFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-recentering" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "-flattening" && i != argc - 1)
- {
- flatteningFileName = argv[++i];
- continue;
- }
- if (std::string(argv[i]) == "-flattening" && i == argc - 1)
- {
- std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
- return 1;
- }
- if (std::string(argv[i]) == "--bad-acceptance")
- {
- std::cout << "Message: Acceptance filter is ON." << std::endl;
- isAcceptanceFilterOn = true;
- continue;
- }
- }
- }
- TStopwatch timer;
- timer.Start();
- mfPicoDstReader *input = new mfPicoDstReader();
- TChain *inChain = (TChain *)input->Open(iFileName);
- if (!inChain)
- {
- std::cerr << "[ERROR]: nothing to read. Maybe the input TTree(s) is(are) empty?" << std::endl;
- return 10;
- }
- // Read recentering
- mfRecentering *recentering = new mfRecentering();
- recentering->ReadRecenteringFile(recenteringFileName);
- // Read flattening
- mfFlattening *flattening = new mfFlattening();
- flattening->ReadFlatteningFile(flatteningFileName);
- // Init output profiles
- TProfile2D *pv2PtTpc[NetaGaps][Npid];
- TProfile2D *pv2PtSpTpc[NetaGaps][Npid];
- TProfile2D *pv1PtFHCal[Npid];
- TProfile2D *pv2PtFHCal[Npid];
- TProfile2D *pv2YTpc[NetaGaps][Npid];
- TProfile2D *pv2YSpTpc[NetaGaps][Npid];
- TProfile2D *pv1YFHCal[Npid];
- TProfile2D *pv2YFHCal[Npid];
- TProfile2D *pv2PdgPtTpc[NetaGaps][Npid];
- TProfile2D *pv2PdgPtSpTpc[NetaGaps][Npid];
- TProfile2D *pv1PdgPtFHCal[Npid];
- TProfile2D *pv2PdgPtFHCal[Npid];
- TProfile2D *pv2PdgYTpc[NetaGaps][Npid];
- TProfile2D *pv2PdgYSpTpc[NetaGaps][Npid];
- TProfile2D *pv1PdgYFHCal[Npid];
- TProfile2D *pv2PdgYFHCal[Npid];
- TProfile2D *pv2PtMcTpc[NetaGaps][Npid];
- TProfile2D *pv1PtRp[Npid];
- TProfile2D *pv2PtRp[Npid];
- TProfile2D *pv2YMcTpc[NetaGaps][Npid];
- TProfile2D *pv1YRp[Npid];
- TProfile2D *pv2YRp[Npid];
- TProfile2D *pv2PtRecoMcTpc[NetaGaps][Npid];
- TProfile2D *pv1PtRecoRp[Npid];
- TProfile2D *pv2PtRecoRp[Npid];
- TProfile2D *pv2YRecoMcTpc[NetaGaps][Npid];
- TProfile2D *pv1YRecoRp[Npid];
- TProfile2D *pv2YRecoRp[Npid];
- TProfile2D *pv2PdgPtRecoMcTpc[NetaGaps][Npid];
- TProfile2D *pv1PdgPtRecoRp[Npid];
- TProfile2D *pv2PdgPtRecoRp[Npid];
- TProfile2D *pv2PdgYRecoMcTpc[NetaGaps][Npid];
- TProfile2D *pv1PdgYRecoRp[Npid];
- TProfile2D *pv2PdgYRecoRp[Npid];
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtTpc[iGap][iPid] = new TProfile2D(Form("pv2PtTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PtTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2YTpc[iGap][iPid] = new TProfile2D(Form("pv2YTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2YTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PtSpTpc[iGap][iPid] = new TProfile2D(Form("pv2PtSpTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PtSpTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2YSpTpc[iGap][iPid] = new TProfile2D(Form("pv2YSpTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2YSpTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PtMcTpc[iGap][iPid] = new TProfile2D(Form("pv2PtMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PtMcTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2YMcTpc[iGap][iPid] = new TProfile2D(Form("pv2YMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2YMcTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PtRecoMcTpc[iGap][iPid] = new TProfile2D(Form("pv2PtRecoMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PtRecoMcTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2YRecoMcTpc[iGap][iPid] = new TProfile2D(Form("pv2YRecoMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2YRecoMcTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PdgPtTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgPtTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgPtTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PdgYTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgYTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgYTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PdgPtSpTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgPtSpTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgPtSpTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PdgYSpTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgYSpTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgYSpTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PdgPtRecoMcTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgPtRecoMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgPtRecoMcTpc_gap%i_pid%i", iGap, iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PdgYRecoMcTpc[iGap][iPid] = new TProfile2D(Form("pv2PdgYRecoMcTpc_gap%i_pid%i", iGap, iPid),
- Form("pv2PdgYRecoMcTpc_gap%i_pid%i", iGap, iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- }
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PtFHCal[iPid] = new TProfile2D(Form("pv1PtFHCal_pid%i", iPid),
- Form("pv1PtFHCal_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PtFHCal[iPid] = new TProfile2D(Form("pv2PtFHCal_pid%i", iPid),
- Form("pv2PtFHCal_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv1YFHCal[iPid] = new TProfile2D(Form("pv1YFHCal_pid%i", iPid),
- Form("pv1YFHCal_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2YFHCal[iPid] = new TProfile2D(Form("pv2YFHCal_pid%i", iPid),
- Form("pv2YFHCal_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv1PdgPtFHCal[iPid] = new TProfile2D(Form("pv1PdgPtFHCal_pid%i", iPid),
- Form("pv1PdgPtFHCal_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PdgPtFHCal[iPid] = new TProfile2D(Form("pv2PdgPtFHCal_pid%i", iPid),
- Form("pv2PdgPtFHCal_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv1PdgYFHCal[iPid] = new TProfile2D(Form("pv1PdgYFHCal_pid%i", iPid),
- Form("pv1PdgYFHCal_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PdgYFHCal[iPid] = new TProfile2D(Form("pv2PdgYFHCal_pid%i", iPid),
- Form("pv2PdgYFHCal_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv1PtRp[iPid] = new TProfile2D(Form("pv1PtRp_pid%i", iPid),
- Form("pv1PtRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PtRp[iPid] = new TProfile2D(Form("pv2PtRp_pid%i", iPid),
- Form("pv2PtRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv1YRp[iPid] = new TProfile2D(Form("pv1YRp_pid%i", iPid),
- Form("pv1YRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2YRp[iPid] = new TProfile2D(Form("pv2YRp_pid%i", iPid),
- Form("pv2YRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv1PtRecoRp[iPid] = new TProfile2D(Form("pv1PtRecoRp_pid%i", iPid),
- Form("pv1PtRecoRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PtRecoRp[iPid] = new TProfile2D(Form("pv2PtRecoRp_pid%i", iPid),
- Form("pv2PtRecoRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv1YRecoRp[iPid] = new TProfile2D(Form("pv1YRecoRp_pid%i", iPid),
- Form("pv1YRecoRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2YRecoRp[iPid] = new TProfile2D(Form("pv2YRecoRp_pid%i", iPid),
- Form("pv2YRecoRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv1PdgPtRecoRp[iPid] = new TProfile2D(Form("pv1PdgPtRecoRp_pid%i", iPid),
- Form("pv1PdgPtRecoRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv2PdgPtRecoRp[iPid] = new TProfile2D(Form("pv2PdgPtRecoRp_pid%i", iPid),
- Form("pv2PdgPtRecoRp_pid%i", iPid),
- 300, 0., 3., NresBin, 0., 100.);
- pv1PdgYRecoRp[iPid] = new TProfile2D(Form("pv1PdgYRecoRp_pid%i", iPid),
- Form("pv1PdgYRecoRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- pv2PdgYRecoRp[iPid] = new TProfile2D(Form("pv2PdgYRecoRp_pid%i", iPid),
- Form("pv2PdgYRecoRp_pid%i", iPid),
- 320, -1.6, 1.6, NresBin, 0., 100.);
- }
- PicoDstMCEvent *mcEvent = nullptr;
- // PicoDstRecoEvent *recoEvent = nullptr;
- TClonesArray *recoTracks = nullptr;
- TClonesArray *mcTracks = nullptr;
- TClonesArray *fhcalmodules = nullptr;
- inChain->SetBranchAddress("mcevent.", &mcEvent);
- // inChain->SetBranchAddress("recoevent.", &recoEvent);
- inChain->SetBranchAddress("mctracks", &mcTracks);
- inChain->SetBranchAddress("recotracks", &recoTracks);
- inChain->SetBranchAddress("FHCalModules", &fhcalmodules);
- TF1 *fResTpc[NetaGaps];
- TF1 *fResSpTpc[NetaGaps];
- TF1 *fResMcTpc[NetaGaps];
- TF1 *fResv1FHCal, *fResv2FHCal;
- TFile *fiRes = new TFile(resFileName.Data(), "READ");
- fiRes->cd();
- for (int i = 0; i < NetaGaps; i++)
- {
- fResTpc[i] = (TF1 *)fiRes->Get(Form("fResTpc_gap%i", i));
- fResSpTpc[i] = (TF1 *)fiRes->Get(Form("fResSpTpc_gap%i", i));
- fResMcTpc[i] = (TF1 *)fiRes->Get(Form("fResMcTpc_gap%i", i));
- }
- fResv1FHCal = (TF1 *)fiRes->Get(Form("fResV1FHCal"));
- fResv2FHCal = (TF1 *)fiRes->Get(Form("fResV2FHCal"));
- std::vector<float> vPsiTpcR, vPsiTpcL;
- std::vector<float> vPsiMcTpcR, vPsiMcTpcL;
- float PsiFHCal;
- std::vector<TVector2> Q_R_SP, Q_L_SP;
- TVector2 u_R_SP, u_L_SP;
- // Start event loop
- int n_entries = inChain->GetEntries();
- mfSelector *selector = new mfSelector();
- mfInputReader *inputReader = new mfInputReader();
- inputReader->SetDcaFitFile(dcaFileName);
- for (int iEv = 0; iEv < n_entries; iEv++)
- {
- if (iEv % 1000 == 0)
- std::cout << "Event [" << iEv << "/" << n_entries << "]" << std::endl;
- inChain->GetEntry(iEv);
- mfProcessEvent *procEvent = new mfProcessEvent();
- float cent = procEvent->GetCentB(mcEvent);
- if (cent == -1)
- continue;
- procEvent->SetInputReader(inputReader);
- mfAcceptanceFilter *AccFilter = new mfAcceptanceFilter();
- if (isAcceptanceFilterOn)
- AccFilter->Activate();
- procEvent->SetAcceptaceFilter(AccFilter);
- mfQvector qv;
- mfEventInfo evInfoTpcRaw = procEvent->ProcessRecoTpcEP(recoTracks, 2.);
- vPsiTpcR.clear();
- vPsiTpcL.clear();
- Q_R_SP.clear();
- Q_L_SP.clear();
- PsiFHCal = 0.;
- if (evInfoTpcRaw.GetQvectors().size() == 0)
- continue;
- mfEventInfo evInfoTpcRec = recentering->ProcessRecentering(evInfoTpcRaw, cent);
- mfEventInfo evInfoTpc = flattening->ProcessFlattening(evInfoTpcRec, cent);
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- // Ntracks >= 4 for TpcR and TpcL EP
- if (selector->isGoodTpcEp(evInfoTpc.GetEpMult(TpcREp[iGap]), evInfoTpc.GetEpMult(TpcLEp[iGap])))
- {
- qv.SetHarmonic(evInfoTpc.GetQvector(TpcREp[iGap]).GetHarmonic());
- qv.Set(evInfoTpc.GetQvector(TpcREp[iGap]).X(),
- evInfoTpc.GetQvector(TpcREp[iGap]).Y());
- vPsiTpcR.push_back(qv.GetPhi());
- qv.SetHarmonic(evInfoTpc.GetQvector(TpcLEp[iGap]).GetHarmonic());
- qv.Set(evInfoTpc.GetQvector(TpcLEp[iGap]).X(),
- evInfoTpc.GetQvector(TpcLEp[iGap]).Y());
- vPsiTpcL.push_back(qv.GetPhi());
- }
- else
- {
- vPsiTpcR.push_back(-999.);
- vPsiTpcL.push_back(-999.);
- }
- if (selector->isGoodTpcEp(evInfoTpc.GetEpMult(SpTpcREp[iGap]), evInfoTpc.GetEpMult(SpTpcLEp[iGap])))
- {
- Q_R_SP.push_back((TVector2)evInfoTpc.GetQvector(SpTpcREp[iGap]));
- Q_L_SP.push_back((TVector2)evInfoTpc.GetQvector(SpTpcLEp[iGap]));
- }
- else
- {
- Q_R_SP.push_back({0., 0.});
- Q_L_SP.push_back({0., 0.});
- }
- }
- mfEventInfo evInfoFHCalRaw = procEvent->ProcessRecoFHCalEP(fhcalmodules, 1.);
- mfEventInfo evInfoFHCalRec = recentering->ProcessRecentering(evInfoFHCalRaw, cent);
- mfEventInfo evInfoFHCal = flattening->ProcessFlattening(evInfoFHCalRec, cent);
- // FHCal energy > 0 in FHCal_L and FHCal_R
- if (selector->isGoodFHCalEp(evInfoFHCal.GetEpWeight(EpNames::kFHCal_R), evInfoFHCal.GetEpWeight(EpNames::kFHCal_R)))
- {
- qv.SetHarmonic(evInfoFHCal.GetQvector(EpNames::kFHCal).GetHarmonic());
- qv.Set(evInfoFHCal.GetQvector(EpNames::kFHCal).X(),
- evInfoFHCal.GetQvector(EpNames::kFHCal).Y());
- PsiFHCal = qv.GetPhi();
- }
- else
- PsiFHCal = -999.;
- mfEventInfo evInfoMcTpc = procEvent->ProcessMcTpcEP(mcTracks, 2.);
- vPsiMcTpcR.clear();
- vPsiMcTpcL.clear();
- if (evInfoMcTpc.GetQvectors().size() == 0)
- continue;
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- // Ntracks >= 4 for TpcR and TpcL EP
- if (selector->isGoodTpcEp(evInfoMcTpc.GetEpMult(TpcREp[iGap]), evInfoMcTpc.GetEpMult(TpcLEp[iGap])))
- {
- vPsiMcTpcR.push_back(evInfoMcTpc.GetQvector(TpcREp[iGap]).GetPhi());
- vPsiMcTpcL.push_back(evInfoMcTpc.GetQvector(TpcLEp[iGap]).GetPhi());
- }
- else
- {
- vPsiMcTpcR.push_back(-999.);
- vPsiMcTpcL.push_back(-999.);
- }
- }
- // Starting flow calculation
- for (int iTrack = 0; iTrack < recoTracks->GetEntriesFast(); iTrack++)
- {
- auto recoTrack = (PicoDstRecoTrack *)recoTracks->UncheckedAt(iTrack);
- if (!recoTrack)
- continue;
- if (!AccFilter->isGoodTpc(recoTrack->GetPhi()))
- continue;
- // Primary track selection
- if (!selector->isRecoPrimary(recoTrack->GetDCAx(), inputReader->GetDcaSigmaX(recoTrack)))
- continue;
- if (!selector->isRecoPrimary(recoTrack->GetDCAy(), inputReader->GetDcaSigmaY(recoTrack)))
- continue;
- if (!selector->isRecoPrimary(recoTrack->GetDCAz(), inputReader->GetDcaSigmaZ(recoTrack)))
- continue;
- // Good reco track selection
- if (!selector->isGoodRecoTrack(recoTrack))
- continue;
- // Only tracks mathed with Tof will be selected
- if (!selector->isGoodPid(recoTrack))
- continue;
- auto mcTrack = (PicoDstMCTrack *)mcTracks->UncheckedAt(recoTrack->GetMcId());
- double Rapidity, RapidityMc, Energy = -1;
- if (selector->isPion(recoTrack))
- Energy = Sqrt(Power(recoTrack->GetP(), 2) + Mass2[0]);
- if (selector->isKaon(recoTrack))
- Energy = Sqrt(Power(recoTrack->GetP(), 2) + Mass2[1]);
- if (selector->isProton(recoTrack))
- Energy = Sqrt(Power(recoTrack->GetP(), 2) + Mass2[2]);
- if (Energy != -1)
- {
- Rapidity = 0.5 * Log((Energy + recoTrack->GetPz()) / (Energy - recoTrack->GetPz()));
- RapidityMc = 0.5 * Log((mcTrack->GetEnergy() + mcTrack->GetPz()) / (mcTrack->GetEnergy() - mcTrack->GetPz()));
- }
- else
- {
- Rapidity = -999.;
- RapidityMc = -999.;
- }
- // v2 (TPC EP)
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- double v2Tpc, v2TpcMc, v2SpTpc;
- if (vPsiTpcR.at(iGap) == -999. || (Q_R_SP.at(iGap).X() == 0. && Q_R_SP.at(iGap).Y() == 0.))
- continue;
- if (vPsiTpcL.at(iGap) == -999. || (Q_L_SP.at(iGap).X() == 0. && Q_L_SP.at(iGap).Y() == 0.))
- continue;
- if (recoTrack->GetEta() < -1. * TpcEtaGap[iGap])
- {
- u_L_SP.Set(Cos(2. * recoTrack->GetPhi()), Sin(2. * recoTrack->GetPhi()));
- v2SpTpc = (double)(u_L_SP * Q_R_SP.at(iGap)) / fResSpTpc[iGap]->Eval(cent);
- v2Tpc = Cos(2. * (recoTrack->GetPhi() - vPsiTpcR.at(iGap))) / fResTpc[iGap]->Eval(cent);
- v2TpcMc = Cos(2. * (mcTrack->GetPhi() - vPsiTpcR.at(iGap))) / fResTpc[iGap]->Eval(cent);
- }
- if (recoTrack->GetEta() > TpcEtaGap[iGap])
- {
- u_R_SP.Set(Cos(2. * recoTrack->GetPhi()), Sin(2. * recoTrack->GetPhi()));
- v2SpTpc = (double)(u_R_SP * Q_L_SP.at(iGap)) / fResSpTpc[iGap]->Eval(cent);
- v2Tpc = Cos(2. * (recoTrack->GetPhi() - vPsiTpcL.at(iGap))) / fResTpc[iGap]->Eval(cent);
- v2TpcMc = Cos(2. * (mcTrack->GetPhi() - vPsiTpcL.at(iGap))) / fResTpc[iGap]->Eval(cent);
- }
- if (Abs(recoTrack->GetEta()) > TpcEtaGap[iGap])
- {
- if (recoTrack->GetCharge() > 0)
- pv2PtTpc[iGap][0]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtTpc[iGap][1]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtTpc[iGap][2]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtTpc[iGap][3]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YTpc[iGap][1]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YTpc[iGap][2]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YTpc[iGap][3]->Fill(Rapidity, cent, v2Tpc);
- if (recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][0]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][1]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][2]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][3]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][1]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][2]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][3]->Fill(Rapidity, cent, v2Tpc);
- if (recoTrack->GetCharge() > 0)
- pv2PtRecoMcTpc[iGap][0]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoMcTpc[iGap][1]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoMcTpc[iGap][2]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoMcTpc[iGap][3]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoMcTpc[iGap][1]->Fill(RapidityMc, cent, v2TpcMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoMcTpc[iGap][2]->Fill(RapidityMc, cent, v2TpcMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoMcTpc[iGap][3]->Fill(RapidityMc, cent, v2TpcMc);
- if (recoTrack->GetCharge() > 0)
- pv2PtSpTpc[iGap][0]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtSpTpc[iGap][1]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtSpTpc[iGap][2]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtSpTpc[iGap][3]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YSpTpc[iGap][1]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YSpTpc[iGap][2]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YSpTpc[iGap][3]->Fill(Rapidity, cent, v2SpTpc);
- if (recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][0]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][1]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][2]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][3]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][1]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][2]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][3]->Fill(Rapidity, cent, v2SpTpc);
- if (recoTrack->GetCharge() < 0)
- pv2PtTpc[iGap][4]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtTpc[iGap][5]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtTpc[iGap][6]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtTpc[iGap][7]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][5]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][6]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][7]->Fill(Rapidity, cent, v2Tpc);
- if (recoTrack->GetCharge() < 0)
- pv2PdgPtTpc[iGap][4]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][5]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][6]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtTpc[iGap][7]->Fill(recoTrack->GetPt(), cent, v2Tpc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][5]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][6]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYTpc[iGap][7]->Fill(Rapidity, cent, v2Tpc);
- if (recoTrack->GetCharge() < 0)
- pv2PtRecoMcTpc[iGap][4]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoMcTpc[iGap][5]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoMcTpc[iGap][6]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoMcTpc[iGap][7]->Fill(mcTrack->GetPt(), cent, v2TpcMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoMcTpc[iGap][5]->Fill(RapidityMc, cent, v2TpcMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoMcTpc[iGap][6]->Fill(RapidityMc, cent, v2TpcMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoMcTpc[iGap][7]->Fill(RapidityMc, cent, v2TpcMc);
- if (recoTrack->GetCharge() < 0)
- pv2PtSpTpc[iGap][4]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtSpTpc[iGap][5]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtSpTpc[iGap][6]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtSpTpc[iGap][7]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][5]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][6]->Fill(Rapidity, cent, v2Tpc);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YTpc[iGap][7]->Fill(Rapidity, cent, v2Tpc);
- if (recoTrack->GetCharge() < 0)
- pv2PdgPtSpTpc[iGap][4]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][5]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][6]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtSpTpc[iGap][7]->Fill(recoTrack->GetPt(), cent, v2SpTpc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][5]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][6]->Fill(Rapidity, cent, v2SpTpc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYSpTpc[iGap][7]->Fill(Rapidity, cent, v2SpTpc);
- }
- }
- // v1, v2 FHCal
- double v1FHCal, v1FHCalY, v2FHCal, w_sign;
- double v1FHCalMc, v1FHCalYMc, v2FHCalMc;
- if (PsiFHCal != -999.)
- {
- w_sign = (recoTrack->GetEta() > 0) ? 1 : -1;
- v1FHCal = w_sign * Cos(1. * (recoTrack->GetPhi() - PsiFHCal)) / fResv1FHCal->Eval(cent);
- v1FHCalY = Cos(1. * (recoTrack->GetPhi() - PsiFHCal)) / fResv1FHCal->Eval(cent);
- v2FHCal = Cos(2. * (recoTrack->GetPhi() - PsiFHCal)) / fResv2FHCal->Eval(cent);
- v1FHCalMc = w_sign * Cos(1. * (mcTrack->GetPhi() - mcEvent->GetPhiRP()));
- v1FHCalYMc = Cos(1. * (mcTrack->GetPhi() - mcEvent->GetPhiRP()));
- v2FHCalMc = Cos(2. * (mcTrack->GetPhi() - mcEvent->GetPhiRP()));
- if (Abs(recoTrack->GetEta()) > MinV1PtEta)
- {
- if (recoTrack->GetCharge() > 0)
- pv1PtFHCal[0]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtFHCal[1]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtFHCal[2]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtFHCal[3]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (recoTrack->GetCharge() > 0)
- pv1PtRecoRp[0]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtRecoRp[1]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtRecoRp[2]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PtRecoRp[3]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (recoTrack->GetCharge() > 0)
- pv1PdgPtFHCal[0]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgPtFHCal[1]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgPtFHCal[2]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgPtFHCal[3]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (recoTrack->GetCharge() < 0)
- pv1PtFHCal[4]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtFHCal[5]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtFHCal[6]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtFHCal[7]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (recoTrack->GetCharge() < 0)
- pv1PtRecoRp[4]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtRecoRp[5]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtRecoRp[6]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PtRecoRp[7]->Fill(mcTrack->GetPt(), cent, v1FHCalMc);
- if (recoTrack->GetCharge() < 0)
- pv1PdgPtFHCal[4]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgPtFHCal[5]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgPtFHCal[6]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgPtFHCal[7]->Fill(recoTrack->GetPt(), cent, v1FHCal);
- }
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YFHCal[1]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YFHCal[2]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YFHCal[3]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YRecoRp[1]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YRecoRp[2]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1YRecoRp[3]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgYFHCal[1]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgYFHCal[2]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv1PdgYFHCal[3]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YFHCal[5]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YFHCal[6]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YFHCal[7]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YRecoRp[5]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YRecoRp[6]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1YRecoRp[7]->Fill(RapidityMc, cent, v1FHCalYMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgYFHCal[5]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgYFHCal[6]->Fill(Rapidity, cent, v1FHCalY);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv1PdgYFHCal[7]->Fill(Rapidity, cent, v1FHCalY);
- if (recoTrack->GetCharge() > 0)
- pv2PtFHCal[0]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtFHCal[1]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtFHCal[2]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtFHCal[3]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (recoTrack->GetCharge() > 0)
- pv2PtRecoRp[0]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoRp[1]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoRp[2]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PtRecoRp[3]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (recoTrack->GetCharge() > 0)
- pv2PdgPtFHCal[0]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtFHCal[1]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtFHCal[2]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgPtFHCal[3]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (recoTrack->GetCharge() < 0)
- pv2PtFHCal[4]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtFHCal[5]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtFHCal[6]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtFHCal[7]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (recoTrack->GetCharge() < 0)
- pv2PtRecoRp[4]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoRp[5]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoRp[6]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PtRecoRp[7]->Fill(mcTrack->GetPt(), cent, v2FHCalMc);
- if (recoTrack->GetCharge() < 0)
- pv2PdgPtFHCal[4]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgPtFHCal[5]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgPtFHCal[6]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgPtFHCal[7]->Fill(recoTrack->GetPt(), cent, v2FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YFHCal[1]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YFHCal[2]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YFHCal[3]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoRp[1]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoRp[2]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2YRecoRp[3]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYFHCal[1]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYFHCal[2]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() > 0)
- pv2PdgYFHCal[3]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YFHCal[5]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YFHCal[6]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YFHCal[7]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoRp[5]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoRp[6]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2YRecoRp[7]->Fill(RapidityMc, cent, v2FHCalMc);
- if (selector->isPdgAntiPion(mcTrack)) //if (selector->isPion(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgYFHCal[5]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgAntiKaon(mcTrack)) //if (selector->isKaon(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgYFHCal[6]->Fill(Rapidity, cent, v2FHCal);
- if (selector->isPdgAntiProton(mcTrack)) //if (selector->isProton(recoTrack) && recoTrack->GetCharge() < 0)
- pv2PdgYFHCal[7]->Fill(Rapidity, cent, v2FHCal);
- }
- }
- // Starting Mc flow calculation
- for (int iTrack = 0; iTrack < mcTracks->GetEntriesFast(); iTrack++)
- {
- auto mcTrack = (PicoDstMCTrack *)mcTracks->UncheckedAt(iTrack);
- if (!mcTrack)
- continue;
- // Primary track selection
- if (!selector->isMcPrimary(mcTrack->GetMotherId()))
- continue;
- // Good reco track selection
- if (!selector->isGoodMcTrack(mcTrack))
- continue;
- double Rapidity, Energy = -1;
- if (selector->isPdgPion(mcTrack))
- Energy = Sqrt(Power(mcTrack->GetP(), 2) + Mass2[0]);
- if (selector->isPdgKaon(mcTrack))
- Energy = Sqrt(Power(mcTrack->GetP(), 2) + Mass2[1]);
- if (selector->isPdgProton(mcTrack))
- Energy = Sqrt(Power(mcTrack->GetP(), 2) + Mass2[2]);
- if (Energy != -1)
- Rapidity = 0.5 * Log((Energy + mcTrack->GetPz()) / (Energy - mcTrack->GetPz()));
- else
- Rapidity = -999.;
- double charge = 0.;
- if (TDatabasePDG::Instance()->GetParticle(mcTrack->GetPdg()))
- charge = TDatabasePDG::Instance()->GetParticle(mcTrack->GetPdg())->Charge() / 3.;
- else
- charge = 0.;
- // v2 (TPC EP)
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- double v2McTpc;
- if (vPsiMcTpcR.at(iGap) == -999.)
- continue;
- if (mcTrack->GetEta() < -1. * TpcEtaGap[iGap])
- {
- v2McTpc = Cos(2. * (mcTrack->GetPhi() - vPsiMcTpcR.at(iGap))) / fResMcTpc[iGap]->Eval(cent);
- if (charge > 0.)
- pv2PtMcTpc[iGap][0]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgPion(mcTrack))
- pv2PtMcTpc[iGap][1]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgKaon(mcTrack))
- pv2PtMcTpc[iGap][2]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgProton(mcTrack))
- pv2PtMcTpc[iGap][3]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgPion(mcTrack))
- pv2YMcTpc[iGap][1]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgKaon(mcTrack))
- pv2YMcTpc[iGap][2]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgProton(mcTrack))
- pv2YMcTpc[iGap][3]->Fill(Rapidity, cent, v2McTpc);
- if (charge < 0.)
- pv2PtMcTpc[iGap][4]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiPion(mcTrack))
- pv2PtMcTpc[iGap][5]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2PtMcTpc[iGap][6]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiProton(mcTrack))
- pv2PtMcTpc[iGap][7]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiPion(mcTrack))
- pv2YMcTpc[iGap][5]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2YMcTpc[iGap][6]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgAntiProton(mcTrack))
- pv2YMcTpc[iGap][7]->Fill(Rapidity, cent, v2McTpc);
- }
- if (mcTrack->GetEta() > TpcEtaGap[iGap])
- {
- v2McTpc = Cos(2. * (mcTrack->GetPhi() - vPsiMcTpcL.at(iGap))) / fResMcTpc[iGap]->Eval(cent);
- if (charge > 0.)
- pv2PtMcTpc[iGap][0]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgPion(mcTrack))
- pv2PtMcTpc[iGap][1]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgKaon(mcTrack))
- pv2PtMcTpc[iGap][2]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgProton(mcTrack))
- pv2PtMcTpc[iGap][3]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgPion(mcTrack))
- pv2YMcTpc[iGap][1]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgKaon(mcTrack))
- pv2YMcTpc[iGap][2]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgProton(mcTrack))
- pv2YMcTpc[iGap][3]->Fill(Rapidity, cent, v2McTpc);
- if (charge < 0.)
- pv2PtMcTpc[iGap][4]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiPion(mcTrack))
- pv2PtMcTpc[iGap][5]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2PtMcTpc[iGap][6]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiProton(mcTrack))
- pv2PtMcTpc[iGap][7]->Fill(mcTrack->GetPt(), cent, v2McTpc);
- if (selector->isPdgAntiPion(mcTrack))
- pv2YMcTpc[iGap][5]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2YMcTpc[iGap][6]->Fill(Rapidity, cent, v2McTpc);
- if (selector->isPdgAntiProton(mcTrack))
- pv2YMcTpc[iGap][7]->Fill(Rapidity, cent, v2McTpc);
- }
- }
- // v1, v2 Rp
- double v1Rp, v1RpY, v2Rp, w_sign;
- double PsiRp = mcEvent->GetPhiRP();
- if (PsiRp != -999.)
- {
- w_sign = (mcTrack->GetEta() > 0) ? 1 : -1;
- v1Rp = w_sign * Cos(1. * (mcTrack->GetPhi() - PsiRp));
- v1RpY = Cos(1. * (mcTrack->GetPhi() - PsiRp));
- v2Rp = Cos(2. * (mcTrack->GetPhi() - PsiRp));
- if (Abs(mcTrack->GetEta()) > MinV1PtEta)
- {
- if (charge > 0.)
- pv1PtRp[0]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgPion(mcTrack))
- pv1PtRp[1]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgKaon(mcTrack))
- pv1PtRp[2]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgProton(mcTrack))
- pv1PtRp[3]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (charge < 0.)
- pv1PtRp[4]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgAntiPion(mcTrack))
- pv1PtRp[5]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgAntiKaon(mcTrack))
- pv1PtRp[6]->Fill(mcTrack->GetPt(), cent, v1Rp);
- if (selector->isPdgAntiProton(mcTrack))
- pv1PtRp[7]->Fill(mcTrack->GetPt(), cent, v1Rp);
- }
- if (selector->isPdgPion(mcTrack))
- pv1YRp[1]->Fill(Rapidity, cent, v1RpY);
- if (selector->isPdgKaon(mcTrack))
- pv1YRp[2]->Fill(Rapidity, cent, v1RpY);
- if (selector->isPdgProton(mcTrack))
- pv1YRp[3]->Fill(Rapidity, cent, v1RpY);
- if (selector->isPdgAntiPion(mcTrack))
- pv1YRp[5]->Fill(Rapidity, cent, v1RpY);
- if (selector->isPdgAntiKaon(mcTrack))
- pv1YRp[6]->Fill(Rapidity, cent, v1RpY);
- if (selector->isPdgAntiProton(mcTrack))
- pv1YRp[7]->Fill(Rapidity, cent, v1RpY);
- if (charge > 0.)
- pv2PtRp[0]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgPion(mcTrack))
- pv2PtRp[1]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgKaon(mcTrack))
- pv2PtRp[2]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgProton(mcTrack))
- pv2PtRp[3]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgPion(mcTrack))
- pv2YRp[1]->Fill(Rapidity, cent, v2Rp);
- if (selector->isPdgKaon(mcTrack))
- pv2YRp[2]->Fill(Rapidity, cent, v2Rp);
- if (selector->isPdgProton(mcTrack))
- pv2YRp[3]->Fill(Rapidity, cent, v2Rp);
- if (charge < 0.)
- pv2PtRp[4]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgAntiPion(mcTrack))
- pv2PtRp[5]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2PtRp[6]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgAntiProton(mcTrack))
- pv2PtRp[7]->Fill(mcTrack->GetPt(), cent, v2Rp);
- if (selector->isPdgAntiPion(mcTrack))
- pv2YRp[5]->Fill(Rapidity, cent, v2Rp);
- if (selector->isPdgAntiKaon(mcTrack))
- pv2YRp[6]->Fill(Rapidity, cent, v2Rp);
- if (selector->isPdgAntiProton(mcTrack))
- pv2YRp[7]->Fill(Rapidity, cent, v2Rp);
- }
- }
- }
- // Init output file
- TFile *fo = new TFile(oFileName.Data(), "recreate");
- fo->cd();
- fo->mkdir("TpcEp");
- fo->mkdir("TpcSp");
- fo->mkdir("TpcEpPdg");
- fo->mkdir("TpcSpPdg");
- fo->mkdir("McTpcEp");
- fo->mkdir("RecoMcTpcEp");
- fo->mkdir("FHCalEp");
- fo->mkdir("FHCalEpPdg");
- fo->mkdir("FHCalRp");
- fo->mkdir("Rp");
- fo->cd("TpcEp");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("TpcSp");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtSpTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YSpTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("FHCalEp");
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PtFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1YFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YFHCal[iPid]->Write();
- }
- fo->cd("FHCalRp");
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PtRecoRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtRecoRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1YRecoRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YRecoRp[iPid]->Write();
- }
- fo->cd("McTpcEp");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtMcTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YMcTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("RecoMcTpcEp");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtRecoMcTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YRecoMcTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("Rp");
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PtRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PtRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1YRp[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2YRp[iPid]->Write();
- }
- fo->cd("TpcEpPdg");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgPtTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgYTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("TpcSpPdg");
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgPtSpTpc[iGap][iPid]->Write();
- }
- }
- for (int iGap = 0; iGap < NetaGaps; iGap++)
- {
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgYSpTpc[iGap][iPid]->Write();
- }
- }
- fo->cd("FHCalEpPdg");
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PdgPtFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgPtFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv1PdgYFHCal[iPid]->Write();
- }
- for (int iPid = 0; iPid < Npid; iPid++)
- {
- pv2PdgYFHCal[iPid]->Write();
- }
- fo->Close();
- timer.Stop();
- timer.Print();
- return 0;
- }
|