#include "AliAODEvent.h"\r
#include "AliAODTrack.h"\r
#include "AliAODInputHandler.h"\r
+#include "AliCollisionGeometry.h"\r
#include "AliGenEventHeader.h"\r
-#include "AliGenHijingEventHeader.h"\r
#include "AliMCEventHandler.h"\r
#include "AliMCEvent.h"\r
#include "AliStack.h"\r
#include "AliESDtrackCuts.h"\r
#include "AliEventplane.h"\r
-#include "AliTHn.h" \r
+#include "AliTHn.h" \r
+\r
+#include "AliEventPoolManager.h" \r
\r
#include "AliPID.h" \r
#include "AliPIDResponse.h" \r
\r
#include "AliAnalysisTaskBFPsi.h"\r
#include "AliBalancePsi.h"\r
+#include "AliAnalysisTaskTriggeredBF.h"\r
\r
\r
// Analysis task for the BF vs Psi code\r
// Authors: Panos.Christakoglou@nikhef.nl\r
\r
+using std::cout;\r
+using std::endl;\r
+\r
ClassImp(AliAnalysisTaskBFPsi)\r
\r
//________________________________________________________________________\r
fBalance(0),\r
fRunShuffling(kFALSE),\r
fShuffledBalance(0),\r
+ fRunMixing(kFALSE),\r
+ fRunMixingEventPlane(kFALSE),\r
+ fMixingTracks(50000),\r
+ fMixedBalance(0),\r
+ fPoolMgr(0),\r
fList(0),\r
fListBF(0),\r
fListBFS(0),\r
+ fListBFM(0),\r
fHistListPIDQA(0),\r
fHistEventStats(0),\r
fHistCentStats(0),\r
nAODtrackCutBit(128),\r
fPtMin(0.3),\r
fPtMax(1.5),\r
+ fPtMinForCorrections(0.3),//=================================correction\r
+ fPtMaxForCorrections(1.5),//=================================correction\r
+ fPtBinForCorrections(36), //=================================correction\r
fEtaMin(-0.8),\r
fEtaMax(-0.8),\r
+ fEtaMinForCorrections(-0.8),//=================================correction\r
+ fEtaMaxForCorrections(0.8),//=================================correction\r
+ fEtaBinForCorrections(16), //=================================correction\r
+ fPhiMin(0.),\r
+ fPhiMax(360.),\r
+ fPhiMinForCorrections(0.),//=================================correction\r
+ fPhiMaxForCorrections(360.),//=================================correction\r
+ fPhiBinForCorrections(100), //=================================correction\r
fDCAxyCut(-1),\r
fDCAzCut(-1),\r
fTPCchi2Cut(-1),\r
fUseFlowAfterBurner(kFALSE),\r
fExcludeResonancesInMC(kFALSE),\r
fUseMCPdgCode(kFALSE),\r
- fPDGCodeToBeAnalyzed(-1) {\r
+ fPDGCodeToBeAnalyzed(-1),\r
+ fEventClass("EventPlane") \r
+{\r
// Constructor\r
// Define input and output slots here\r
// Input slot #0 works with a TChain\r
+\r
+ //======================================================correction\r
+ for (Int_t i=0; i<=kCENTRALITY; i++){\r
+ fHistMatrixCorrectionPlus[i] = NULL; \r
+ fHistMatrixCorrectionMinus[i] = NULL; \r
+ }\r
+ //=====================================================correction\r
+\r
DefineInput(0, TChain::Class());\r
// Output slot #0 writes into a TH1 container\r
DefineOutput(1, TList::Class());\r
DefineOutput(2, TList::Class());\r
DefineOutput(3, TList::Class());\r
DefineOutput(4, TList::Class());\r
+ DefineOutput(5, TList::Class());\r
}\r
\r
//________________________________________________________________________\r
void AliAnalysisTaskBFPsi::UserCreateOutputObjects() {\r
// Create histograms\r
// Called once\r
+\r
+ // global switch disabling the reference \r
+ // (to avoid "Replacing existing TH1" if several wagons are created in train)\r
+ Bool_t oldStatus = TH1::AddDirectoryStatus();\r
+ TH1::AddDirectory(kFALSE);\r
+\r
if(!fBalance) {\r
fBalance = new AliBalancePsi();\r
fBalance->SetAnalysisLevel("ESD");\r
+ fBalance->SetEventClass(fEventClass);\r
//fBalance->SetNumberOfBins(-1,16);\r
//fBalance->SetInterval(-1,-0.8,0.8,16,0.,1.6,15.);\r
}\r
if(!fShuffledBalance) {\r
fShuffledBalance = new AliBalancePsi();\r
fShuffledBalance->SetAnalysisLevel("ESD");\r
+ fShuffledBalance->SetEventClass(fEventClass);\r
//fShuffledBalance->SetNumberOfBins(-1,16);\r
//fShuffledBalance->SetInterval(-1,-0.8,0.8,16,0.,1.6,15.);\r
}\r
}\r
+ if(fRunMixing) {\r
+ if(!fMixedBalance) {\r
+ fMixedBalance = new AliBalancePsi();\r
+ fMixedBalance->SetAnalysisLevel("ESD");\r
+ fMixedBalance->SetEventClass(fEventClass);\r
+ }\r
+ }\r
\r
//QA list\r
fList = new TList();\r
fListBFS->SetOwner();\r
}\r
\r
+ if(fRunMixing) {\r
+ fListBFM = new TList();\r
+ fListBFM->SetName("listTriggeredBFMixed");\r
+ fListBFM->SetOwner();\r
+ }\r
+\r
//PID QA list\r
if(fUsePID) {\r
fHistListPIDQA = new TList();\r
}\r
\r
//Event stats.\r
- TString gCutName[4] = {"Total","Offline trigger",\r
- "Vertex","Analyzed"};\r
+ TString gCutName[5] = {"Total","Offline trigger",\r
+ "Vertex","Analyzed","sel. Centrality"};\r
fHistEventStats = new TH2F("fHistEventStats",\r
"Event statistics;;Centrality percentile;N_{events}",\r
- 4,0.5,4.5,220,-5,105);\r
- for(Int_t i = 1; i <= 4; i++)\r
+ 5,0.5,5.5,220,-5,105);\r
+ for(Int_t i = 1; i <= 5; i++)\r
fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1].Data());\r
fList->Add(fHistEventStats);\r
\r
fHistCentStats->GetXaxis()->SetBinLabel(i,gCentName[i-1].Data());\r
fList->Add(fHistCentStats);\r
\r
- fHistTriggerStats = new TH1F("fHistTriggerStats","Trigger statistics;TriggerBit;N_{events}",130,0,130);\r
+ fHistTriggerStats = new TH1F("fHistTriggerStats","Trigger statistics;TriggerBit;N_{events}",1025,0,1025);\r
fList->Add(fHistTriggerStats);\r
\r
- fHistTrackStats = new TH1F("fHistTrackStats","Event statistics;TrackFilterBit;N_{events}",130,0,130);\r
+ fHistTrackStats = new TH1F("fHistTrackStats","Event statistics;TrackFilterBit;N_{events}",16,0,16);\r
fList->Add(fHistTrackStats);\r
\r
fHistNumberOfAcceptedTracks = new TH2F("fHistNumberOfAcceptedTracks",";N_{acc.};Centrality percentile;Entries",4001,-0.5,4000.5,220,-5,105);\r
fList->Add(fHistEta);\r
fHistRapidity = new TH2F("fHistRapidity","y distribution;y;Centrality percentile",200,-2,2,220,-5,105);\r
fList->Add(fHistRapidity);\r
- fHistPhi = new TH2F("fHistPhi","#phi distribution;#phi;Centrality percentile",200,-20,380,220,-5,105);\r
+ fHistPhi = new TH2F("fHistPhi","#phi distribution;#phi (rad);Centrality percentile",200,0.0,2.*TMath::Pi(),220,-5,105);\r
fList->Add(fHistPhi);\r
fHistPhiBefore = new TH2F("fHistPhiBefore","#phi distribution;#phi;Centrality percentile",200,0.,2*TMath::Pi(),220,-5,105);\r
fList->Add(fHistPhiBefore);\r
fHistPhiAfter = new TH2F("fHistPhiAfter","#phi distribution;#phi;Centrality percentile",200,0.,2*TMath::Pi(),220,-5,105);\r
fList->Add(fHistPhiAfter);\r
- fHistPhiPos = new TH2F("fHistPhiPos","#phi distribution for positive particles;#phi;Centrality percentile",200,-20,380,220,-5,105);\r
+ fHistPhiPos = new TH2F("fHistPhiPos","#phi distribution for positive particles;#phi;Centrality percentile",200,0.,2*TMath::Pi(),220,-5,105);\r
fList->Add(fHistPhiPos);\r
- fHistPhiNeg = new TH2F("fHistPhiNeg","#phi distribution for negative particles;#phi;Centrality percentile",200,-20,380,220,-5,105);\r
+ fHistPhiNeg = new TH2F("fHistPhiNeg","#phi distribution for negative particles;#phi;Centrality percentile",200,0.,2.*TMath::Pi(),220,-5,105);\r
fList->Add(fHistPhiNeg);\r
fHistV0M = new TH2F("fHistV0M","V0 Multiplicity C vs. A",500, 0, 20000, 500, 0, 20000);\r
fList->Add(fHistV0M);\r
fHistRefTracks->GetXaxis()->SetBinLabel(i,gRefTrackName[i-1].Data());\r
fList->Add(fHistRefTracks);\r
\r
+ // QA histograms for HBTinspired and Conversion cuts\r
+ fList->Add(fBalance->GetQAHistHBTbefore());\r
+ fList->Add(fBalance->GetQAHistHBTafter());\r
+ fList->Add(fBalance->GetQAHistConversionbefore());\r
+ fList->Add(fBalance->GetQAHistConversionafter());\r
+ fList->Add(fBalance->GetQAHistPsiMinusPhi());\r
+\r
// Balance function histograms\r
// Initialize histograms if not done yet\r
if(!fBalance->GetHistNp()){\r
fListBFS->Add(fShuffledBalance->GetHistNpp());\r
fListBFS->Add(fShuffledBalance->GetHistNnp());\r
} \r
+\r
+ if(fRunMixing) {\r
+ fListBFM->Add(fMixedBalance->GetHistNp());\r
+ fListBFM->Add(fMixedBalance->GetHistNn());\r
+ fListBFM->Add(fMixedBalance->GetHistNpn());\r
+ fListBFM->Add(fMixedBalance->GetHistNnn());\r
+ fListBFM->Add(fMixedBalance->GetHistNpp());\r
+ fListBFM->Add(fMixedBalance->GetHistNnp());\r
+ }\r
//}\r
\r
+\r
+ // Event Mixing\r
+ if(fRunMixing){\r
+ Int_t trackDepth = fMixingTracks; \r
+ Int_t poolsize = 1000; // Maximum number of events, ignored in the present implemented of AliEventPoolManager\r
+ \r
+ // centrality bins\r
+ Double_t centralityBins[] = {0.,1.,2.,3.,4.,5.,6.,7.,8.,9.,10.,15.,20.,25.,30.,35.,40.,45.,50.,55.,60.,65.,70.,75.,80.,90.,100.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+ Double_t* centbins = centralityBins;\r
+ Int_t nCentralityBins = sizeof(centralityBins) / sizeof(Double_t) - 1;\r
+ \r
+ // Zvtx bins\r
+ Double_t vertexBins[] = {-10., -7., -5., -3., -1., 1., 3., 5., 7., 10.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+ Double_t* vtxbins = vertexBins;\r
+ Int_t nVertexBins = sizeof(vertexBins) / sizeof(Double_t) - 1;\r
+ \r
+ // Event plane angle (Psi) bins\r
+ Double_t psiBins[] = {0.,45.,135.,215.,305.,360.}; // SHOULD BE DEDUCED FROM CREATED ALITHN!!!\r
+ Double_t* psibins = psiBins;\r
+ Int_t nPsiBins = sizeof(psiBins) / sizeof(Double_t) - 1;\r
+ \r
+ // run the event mixing also in bins of event plane (statistics!)\r
+ if(fRunMixingEventPlane){\r
+ fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centbins, nVertexBins, vtxbins, nPsiBins, psibins);\r
+ }\r
+ else{\r
+ fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centbins, nVertexBins, vtxbins);\r
+ }\r
+ }\r
+\r
if(fESDtrackCuts) fList->Add(fESDtrackCuts);\r
\r
//====================PID========================//\r
PostData(1, fList);\r
PostData(2, fListBF);\r
if(fRunShuffling) PostData(3, fListBFS);\r
- if(fUsePID) PostData(4, fHistListPIDQA); //PID\r
+ if(fRunMixing) PostData(4, fListBFM);\r
+ if(fUsePID) PostData(5, fHistListPIDQA); //PID\r
+\r
+ TH1::AddDirectory(oldStatus);\r
+}\r
+\r
+\r
+//________________________________________________________________________\r
+void AliAnalysisTaskBFPsi::SetInputCorrection(const char* filename, \r
+ const char* gCollSystem) {\r
+ //Open files that will be used for correction\r
+ TString gCollidingSystem = gCollSystem;\r
+\r
+ //Open the input file\r
+ TFile *f = TFile::Open(filename);\r
+ if(!f->IsOpen()) {\r
+ Printf("File not found!!!");\r
+ return;\r
+ }\r
+ \r
+ //Get the TDirectoryFile and TList\r
+ TDirectoryFile *dir = dynamic_cast<TDirectoryFile *>(f->Get("PWGCFEbyE.outputBalanceFunctionEfficiencyAnalysis"));\r
+ if(!dir) {\r
+ Printf("TDirectoryFile not found!!!");\r
+ return;\r
+ }\r
+ \r
+ TString listEffName = "";\r
+ for (Int_t iCent = 0; iCent < kCENTRALITY; iCent++) {\r
+ listEffName = "listEfficiencyBF_"; \r
+ listEffName += (TString)((Int_t)(centralityArrayForPbPb[iCent]));\r
+ listEffName += "_";\r
+ listEffName += (TString)((Int_t)(centralityArrayForPbPb[iCent+1]));\r
+ TList *list = (TList*)dir->Get(listEffName.Data());\r
+ if(!list) {\r
+ Printf("TList Efficiency not found!!!");\r
+ return;\r
+ }\r
+ \r
+ TString histoName = "fHistMatrixCorrectionPlus";\r
+ fHistMatrixCorrectionPlus[iCent]= dynamic_cast<TH3D *>(list->FindObject(histoName.Data()));\r
+ if(!fHistMatrixCorrectionPlus[iCent]) {\r
+ Printf("fHist not found!!!");\r
+ return;\r
+ }\r
+ \r
+ histoName = "fHistMatrixCorrectionMinus";\r
+ fHistMatrixCorrectionMinus[iCent] = dynamic_cast<TH3D *>(list->FindObject(histoName.Data())); \r
+ if(!fHistMatrixCorrectionMinus[iCent]) {\r
+ Printf("fHist not found!!!");\r
+ return; \r
+ }\r
+ }//loop over centralities: ONLY the PbPb case is covered\r
+\r
+ if(fHistMatrixCorrectionPlus[0]){\r
+ fEtaMinForCorrections = fHistMatrixCorrectionPlus[0]->GetXaxis()->GetXmin();\r
+ fEtaMaxForCorrections = fHistMatrixCorrectionPlus[0]->GetXaxis()->GetXmax();\r
+ fEtaBinForCorrections = fHistMatrixCorrectionPlus[0]->GetNbinsX();\r
+ \r
+ fPtMinForCorrections = fHistMatrixCorrectionPlus[0]->GetYaxis()->GetXmin();\r
+ fPtMaxForCorrections = fHistMatrixCorrectionPlus[0]->GetYaxis()->GetXmax();\r
+ fPtBinForCorrections = fHistMatrixCorrectionPlus[0]->GetNbinsY();\r
+ \r
+ fPhiMinForCorrections = fHistMatrixCorrectionPlus[0]->GetZaxis()->GetXmin();\r
+ fPhiMaxForCorrections = fHistMatrixCorrectionPlus[0]->GetZaxis()->GetXmax();\r
+ fPhiBinForCorrections = fHistMatrixCorrectionPlus[0]->GetNbinsZ();\r
+ }\r
}\r
\r
//________________________________________________________________________\r
void AliAnalysisTaskBFPsi::UserExec(Option_t *) {\r
// Main loop\r
// Called for each event\r
- TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
-\r
- AliESDtrack *track_TPC = NULL;\r
\r
+ TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
Int_t gNumberOfAcceptedTracks = 0;\r
- Float_t fCentrality = 0.;\r
- Double_t gReactionPlane = 0.;\r
-\r
- // vector holding the charges/kinematics of all tracks (charge,y,eta,phi,p0,p1,p2,pt,E)\r
- vector<Double_t> *chargeVectorShuffle[9]; // this will be shuffled\r
- vector<Double_t> *chargeVector[9]; // original charge\r
- for(Int_t i = 0; i < 9; i++){\r
- chargeVectorShuffle[i] = new vector<Double_t>;\r
- chargeVector[i] = new vector<Double_t>;\r
+ Double_t gCentrality = -1.;\r
+ Double_t gReactionPlane = -1.; \r
+ Float_t bSign = 0.;\r
+ \r
+ // get the event (for generator level: MCEvent())\r
+ AliVEvent* eventMain = NULL;\r
+ if(gAnalysisLevel == "MC") {\r
+ eventMain = dynamic_cast<AliVEvent*>(MCEvent()); \r
}\r
-\r
- Double_t v_charge;\r
- Double_t v_y;\r
- Double_t v_eta;\r
- Double_t v_phi;\r
- Double_t v_p[3];\r
- Double_t v_pt;\r
- Double_t v_E;\r
-\r
+ else{\r
+ eventMain = dynamic_cast<AliVEvent*>(InputEvent()); \r
+ \r
+ // for HBT like cuts need magnetic field sign\r
+ bSign = (eventMain->GetMagneticField() > 0) ? 1 : -1;\r
+ }\r
+ if(!eventMain) {\r
+ AliError("eventMain not available");\r
+ return;\r
+ }\r
+ \r
+ // PID Response task active?\r
if(fUsePID) {\r
fPIDResponse = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->GetPIDResponse();\r
if (!fPIDResponse) AliFatal("This Task needs the PID response attached to the inputHandler");\r
}\r
- \r
- //ESD analysis\r
- if(gAnalysisLevel == "ESD") {\r
- AliESDEvent* gESD = dynamic_cast<AliESDEvent*>(InputEvent()); // from TaskSE\r
- if (!gESD) {\r
- Printf("ERROR: gESD not available");\r
- return;\r
- }\r
+ \r
+ // check event cuts and fill event histograms\r
+ if((gCentrality = IsEventAccepted(eventMain)) < 0){\r
+ return;\r
+ }\r
+ \r
+ //Compute Multiplicity or Centrality variable\r
+ Double_t lMultiplicityVar = GetRefMultiOrCentrality( eventMain );\r
\r
- // store offline trigger bits\r
- fHistTriggerStats->Fill(((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected());\r
+ // get the reaction plane\r
+ gReactionPlane = GetEventPlane(eventMain);\r
+ fHistEventPlane->Fill(gReactionPlane,gCentrality);\r
+ if(gReactionPlane < 0){\r
+ return;\r
+ }\r
+ \r
+ // get the accepted tracks in main event\r
+ TObjArray *tracksMain = GetAcceptedTracks(eventMain,gCentrality,gReactionPlane);\r
+ gNumberOfAcceptedTracks = tracksMain->GetEntriesFast();\r
\r
- // event selection done in AliAnalysisTaskSE::Exec() --> this is not used\r
- fHistEventStats->Fill(1,fCentrality); //all events\r
- Bool_t isSelected = kTRUE;\r
- if(fUseOfflineTrigger)\r
- isSelected = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();\r
- if(isSelected) {\r
- fHistEventStats->Fill(2,fCentrality); //triggered events\r
+ //multiplicity cut (used in pp)\r
+ fHistNumberOfAcceptedTracks->Fill(gNumberOfAcceptedTracks,gCentrality);\r
+ if(fUseMultiplicity) {\r
+ if((gNumberOfAcceptedTracks < fNumberOfAcceptedTracksMin)||(gNumberOfAcceptedTracks > fNumberOfAcceptedTracksMax))\r
+ return;\r
+ }\r
\r
- if(fUseCentrality) {\r
- //Centrality stuff\r
- AliCentrality *centrality = gESD->GetCentrality();\r
+ // store charges of all accepted tracks, shuffle and reassign (two extra loops!)\r
+ TObjArray* tracksShuffled = NULL;\r
+ if(fRunShuffling){\r
+ tracksShuffled = GetShuffledTracks(tracksMain,gCentrality);\r
+ }\r
+ \r
+ // Event mixing \r
+ if (fRunMixing)\r
+ {\r
+ // 1. First get an event pool corresponding in mult (cent) and\r
+ // zvertex to the current event. Once initialized, the pool\r
+ // should contain nMix (reduced) events. This routine does not\r
+ // pre-scan the chain. The first several events of every chain\r
+ // will be skipped until the needed pools are filled to the\r
+ // specified depth. If the pool categories are not too rare, this\r
+ // should not be a problem. If they are rare, you could lose`\r
+ // statistics.\r
+ \r
+ // 2. Collect the whole pool's content of tracks into one TObjArray\r
+ // (bgTracks), which is effectively a single background super-event.\r
+ \r
+ // 3. The reduced and bgTracks arrays must both be passed into\r
+ // FillCorrelations(). Also nMix should be passed in, so a weight\r
+ // of 1./nMix can be applied.\r
+ \r
+ AliEventPool* pool = fPoolMgr->GetEventPool(gCentrality, eventMain->GetPrimaryVertex()->GetZ(),gReactionPlane);\r
+ \r
+ if (!pool){\r
+ AliFatal(Form("No pool found for centrality = %f, zVtx = %f, psi = %f", gCentrality, eventMain->GetPrimaryVertex()->GetZ(),gReactionPlane));\r
+ }\r
+ else{\r
\r
- fCentrality = centrality->GetCentralityPercentile(fCentralityEstimator.Data());\r
+ //pool->SetDebug(1);\r
+\r
+ if (pool->IsReady() || pool->NTracksInPool() > fMixingTracks / 10 || pool->GetCurrentNEvents() >= 5){ \r
+ \r
+ \r
+ Int_t nMix = pool->GetCurrentNEvents();\r
+ //cout << "nMix = " << nMix << " tracks in pool = " << pool->NTracksInPool() << endl;\r
+ \r
+ //((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(2);\r
+ //((TH2F*) fListOfHistos->FindObject("mixedDist"))->Fill(centrality, pool->NTracksInPool());\r
+ //if (pool->IsReady())\r
+ //((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(3);\r
+ \r
+ // Fill mixed-event histos here \r
+ for (Int_t jMix=0; jMix<nMix; jMix++) \r
+ {\r
+ TObjArray* tracksMixed = pool->GetEvent(jMix);\r
+ fMixedBalance->CalculateBalance(gReactionPlane,tracksMain,tracksMixed,bSign,lMultiplicityVar);\r
+ }\r
+ }\r
\r
- // take only events inside centrality class\r
- if(!centrality->IsEventInCentralityClass(fCentralityPercentileMin,\r
- fCentralityPercentileMax,\r
- fCentralityEstimator.Data()))\r
- return;\r
+ // Update the Event pool\r
+ pool->UpdatePool(tracksMain);\r
+ //pool->PrintInfo();\r
\r
+ }//pool NULL check \r
+ }//run mixing\r
+ \r
+ // calculate balance function\r
+ fBalance->CalculateBalance(gReactionPlane,tracksMain,NULL,bSign,lMultiplicityVar);\r
+ \r
+ // calculate shuffled balance function\r
+ if(fRunShuffling && tracksShuffled != NULL) {\r
+ fShuffledBalance->CalculateBalance(gReactionPlane,tracksShuffled,NULL,bSign,lMultiplicityVar);\r
+ }\r
+} \r
+\r
+//________________________________________________________________________\r
+Double_t AliAnalysisTaskBFPsi::IsEventAccepted(AliVEvent *event){\r
+ // Checks the Event cuts\r
+ // Fills Event statistics histograms\r
+ \r
+ Bool_t isSelectedMain = kTRUE;\r
+ Float_t gCentrality = -1.;\r
+ TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
+\r
+ fHistEventStats->Fill(1,gCentrality); //all events\r
+\r
+ // Event trigger bits\r
+ fHistTriggerStats->Fill(((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected());\r
+ if(fUseOfflineTrigger)\r
+ isSelectedMain = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();\r
+ \r
+ if(isSelectedMain) {\r
+ fHistEventStats->Fill(2,gCentrality); //triggered events\r
+ \r
+ //Centrality stuff \r
+ if(fUseCentrality) {\r
+ if(gAnalysisLevel == "AOD") { //centrality in AOD header\r
+ AliAODHeader *header = (AliAODHeader*) event->GetHeader();\r
+ if(header){\r
+ gCentrality = header->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data());\r
+ \r
+ // QA for centrality estimators\r
+ fHistCentStats->Fill(0.,header->GetCentralityP()->GetCentralityPercentile("V0M"));\r
+ fHistCentStats->Fill(1.,header->GetCentralityP()->GetCentralityPercentile("FMD"));\r
+ fHistCentStats->Fill(2.,header->GetCentralityP()->GetCentralityPercentile("TRK"));\r
+ fHistCentStats->Fill(3.,header->GetCentralityP()->GetCentralityPercentile("TKL"));\r
+ fHistCentStats->Fill(4.,header->GetCentralityP()->GetCentralityPercentile("CL0"));\r
+ fHistCentStats->Fill(5.,header->GetCentralityP()->GetCentralityPercentile("CL1"));\r
+ fHistCentStats->Fill(6.,header->GetCentralityP()->GetCentralityPercentile("V0MvsFMD"));\r
+ fHistCentStats->Fill(7.,header->GetCentralityP()->GetCentralityPercentile("TKLvsV0M"));\r
+ fHistCentStats->Fill(8.,header->GetCentralityP()->GetCentralityPercentile("ZEMvsZDC"));\r
+ \r
+ // centrality QA (V0M)\r
+ fHistV0M->Fill(event->GetVZEROData()->GetMTotV0A(), event->GetVZEROData()->GetMTotV0C());\r
+ \r
+ // centrality QA (reference tracks)\r
+ fHistRefTracks->Fill(0.,header->GetRefMultiplicity());\r
+ fHistRefTracks->Fill(1.,header->GetRefMultiplicityPos());\r
+ fHistRefTracks->Fill(2.,header->GetRefMultiplicityNeg());\r
+ fHistRefTracks->Fill(3.,header->GetTPConlyRefMultiplicity());\r
+ fHistRefTracks->Fill(4.,header->GetNumberOfITSClusters(0));\r
+ fHistRefTracks->Fill(5.,header->GetNumberOfITSClusters(1));\r
+ fHistRefTracks->Fill(6.,header->GetNumberOfITSClusters(2));\r
+ fHistRefTracks->Fill(7.,header->GetNumberOfITSClusters(3));\r
+ fHistRefTracks->Fill(8.,header->GetNumberOfITSClusters(4));\r
+ }//AOD header\r
+ }//AOD\r
+\r
+ else if(gAnalysisLevel == "ESD" || gAnalysisLevel == "MCESD"){ // centrality class for ESDs or MC-ESDs\r
+ AliCentrality *centrality = event->GetCentrality();\r
+ gCentrality = centrality->GetCentralityPercentile(fCentralityEstimator.Data());\r
+\r
+ // QA for centrality estimators\r
+ fHistCentStats->Fill(0.,centrality->GetCentralityPercentile("V0M"));\r
+ fHistCentStats->Fill(1.,centrality->GetCentralityPercentile("FMD"));\r
+ fHistCentStats->Fill(2.,centrality->GetCentralityPercentile("TRK"));\r
+ fHistCentStats->Fill(3.,centrality->GetCentralityPercentile("TKL"));\r
+ fHistCentStats->Fill(4.,centrality->GetCentralityPercentile("CL0"));\r
+ fHistCentStats->Fill(5.,centrality->GetCentralityPercentile("CL1"));\r
+ fHistCentStats->Fill(6.,centrality->GetCentralityPercentile("V0MvsFMD"));\r
+ fHistCentStats->Fill(7.,centrality->GetCentralityPercentile("TKLvsV0M"));\r
+ fHistCentStats->Fill(8.,centrality->GetCentralityPercentile("ZEMvsZDC"));\r
+\r
// centrality QA (V0M)\r
- fHistV0M->Fill(gESD->GetVZEROData()->GetMTotV0A(), gESD->GetVZEROData()->GetMTotV0C());\r
+ fHistV0M->Fill(event->GetVZEROData()->GetMTotV0A(), event->GetVZEROData()->GetMTotV0C());\r
+ }//ESD\r
+ else if(gAnalysisLevel == "MC"){\r
+ Double_t gImpactParameter = 0.;\r
+ if(dynamic_cast<AliMCEvent*>(event)){\r
+ AliCollisionGeometry* headerH = dynamic_cast<AliCollisionGeometry*>(dynamic_cast<AliMCEvent*>(event)->GenEventHeader());\r
+ if(headerH){\r
+ gImpactParameter = headerH->ImpactParameter();\r
+ gCentrality = gImpactParameter;\r
+ }//MC header\r
+ }//MC event cast\r
+ }//MC\r
+ else{\r
+ gCentrality = -1.;\r
}\r
- \r
- const AliESDVertex *vertex = gESD->GetPrimaryVertex();\r
+ }\r
+ \r
+ // Event Vertex MC\r
+ if(gAnalysisLevel == "MC"){\r
+ AliGenEventHeader *header = dynamic_cast<AliMCEvent*>(event)->GenEventHeader();\r
+ if(header){ \r
+ TArrayF gVertexArray;\r
+ header->PrimaryVertex(gVertexArray);\r
+ //Printf("Vertex: %lf (x) - %lf (y) - %lf (z)",\r
+ //gVertexArray.At(0),\r
+ //gVertexArray.At(1),\r
+ //gVertexArray.At(2));\r
+ fHistEventStats->Fill(3,gCentrality); //events with a proper vertex\r
+ if(TMath::Abs(gVertexArray.At(0)) < fVxMax) {\r
+ if(TMath::Abs(gVertexArray.At(1)) < fVyMax) {\r
+ if(TMath::Abs(gVertexArray.At(2)) < fVzMax) {\r
+ fHistEventStats->Fill(4,gCentrality); //analayzed events\r
+ fHistVx->Fill(gVertexArray.At(0));\r
+ fHistVy->Fill(gVertexArray.At(1));\r
+ fHistVz->Fill(gVertexArray.At(2),gCentrality);\r
+ \r
+ // take only events inside centrality class\r
+ if((fImpactParameterMin < gCentrality) && (fImpactParameterMax > gCentrality)){\r
+ fHistEventStats->Fill(5,gCentrality); //events with correct centrality\r
+ return gCentrality; \r
+ }//centrality class\r
+ }//Vz cut\r
+ }//Vy cut\r
+ }//Vx cut\r
+ }//header \r
+ }//MC\r
+ \r
+ // Event Vertex AOD, ESD, ESDMC\r
+ else{\r
+ const AliVVertex *vertex = event->GetPrimaryVertex();\r
+ \r
if(vertex) {\r
+ Double32_t fCov[6];\r
+ vertex->GetCovarianceMatrix(fCov);\r
if(vertex->GetNContributors() > 0) {\r
- if(vertex->GetZRes() != 0) {\r
- fHistEventStats->Fill(3,fCentrality); //events with a proper vertex\r
- if(TMath::Abs(vertex->GetXv()) < fVxMax) {\r
- if(TMath::Abs(vertex->GetYv()) < fVyMax) {\r
- if(TMath::Abs(vertex->GetZv()) < fVzMax) {\r
- fHistEventStats->Fill(4,fCentrality); //analayzed events\r
- fHistVx->Fill(vertex->GetXv());\r
- fHistVy->Fill(vertex->GetYv());\r
- fHistVz->Fill(vertex->GetZv(),fCentrality);\r
+ if(fCov[5] != 0) {\r
+ fHistEventStats->Fill(3,gCentrality); //events with a proper vertex\r
+ if(TMath::Abs(vertex->GetX()) < fVxMax) {\r
+ if(TMath::Abs(vertex->GetY()) < fVyMax) {\r
+ if(TMath::Abs(vertex->GetZ()) < fVzMax) {\r
+ fHistEventStats->Fill(4,gCentrality); //analyzed events\r
+ fHistVx->Fill(vertex->GetX());\r
+ fHistVy->Fill(vertex->GetY());\r
+ fHistVz->Fill(vertex->GetZ(),gCentrality);\r
\r
- //========Get the VZERO event plane========//\r
- Double_t gVZEROEventPlane = -10.0;\r
- Double_t qxTot = 0.0, qyTot = 0.0;\r
- AliEventplane *ep = gESD->GetEventplane();\r
- if(ep) \r
- gVZEROEventPlane = ep->CalculateVZEROEventPlane(gESD,10,2,qxTot,qyTot);\r
- if(gVZEROEventPlane < 0.) gVZEROEventPlane += TMath::Pi();\r
- gReactionPlane = gVZEROEventPlane*TMath::RadToDeg();\r
- fHistEventPlane->Fill(gReactionPlane,fCentrality);\r
- //========Get the VZERO event plane========//\r
-\r
- //Printf("There are %d tracks in this event", gESD->GetNumberOfTracks());\r
- for (Int_t iTracks = 0; iTracks < gESD->GetNumberOfTracks(); iTracks++) {\r
- AliESDtrack* track = dynamic_cast<AliESDtrack *>(gESD->GetTrack(iTracks));\r
- if (!track) {\r
- Printf("ERROR: Could not receive track %d", iTracks);\r
- continue;\r
- } \r
- \r
- // take only TPC only tracks\r
- track_TPC = new AliESDtrack();\r
- if(!track->FillTPCOnlyTrack(*track_TPC)) continue;\r
- \r
- //ESD track cuts\r
- if(fESDtrackCuts) \r
- if(!fESDtrackCuts->AcceptTrack(track_TPC)) continue;\r
- \r
- // fill QA histograms\r
- Float_t b[2];\r
- Float_t bCov[3];\r
- track_TPC->GetImpactParameters(b,bCov);\r
- if (bCov[0]<=0 || bCov[2]<=0) {\r
- AliDebug(1, "Estimated b resolution lower or equal zero!");\r
- bCov[0]=0; bCov[2]=0;\r
- }\r
- \r
- Int_t nClustersTPC = -1;\r
- nClustersTPC = track_TPC->GetTPCNclsIter1(); // TPC standalone\r
- //nClustersTPC = track->GetTPCclusters(0); // global track\r
- Float_t chi2PerClusterTPC = -1;\r
- if (nClustersTPC!=0) {\r
- chi2PerClusterTPC = track_TPC->GetTPCchi2Iter1()/Float_t(nClustersTPC); // TPC standalone\r
- //chi2PerClusterTPC = track->GetTPCchi2()/Float_t(nClustersTPC); // global track\r
- }\r
-\r
- //===========================PID===============================// \r
- if(fUsePID) {\r
- Double_t prob[AliPID::kSPECIES]={0.};\r
- Double_t probTPC[AliPID::kSPECIES]={0.};\r
- Double_t probTOF[AliPID::kSPECIES]={0.};\r
- Double_t probTPCTOF[AliPID::kSPECIES]={0.};\r
-\r
- Double_t nSigma = 0.;\r
- UInt_t detUsedTPC = 0;\r
- UInt_t detUsedTOF = 0;\r
- UInt_t detUsedTPCTOF = 0;\r
-\r
- //Decide what detector configuration we want to use\r
- switch(fPidDetectorConfig) {\r
- case kTPCpid:\r
- fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTPC);\r
- nSigma = TMath::Abs(fPIDResponse->NumberOfSigmasTPC(track,(AliPID::EParticleType)fParticleOfInterest));\r
- detUsedTPC = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTPC);\r
- for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
- prob[iSpecies] = probTPC[iSpecies];\r
- break;\r
- case kTOFpid:\r
- fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTOF);\r
- nSigma = TMath::Abs(fPIDResponse->NumberOfSigmasTOF(track,(AliPID::EParticleType)fParticleOfInterest));\r
- detUsedTOF = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTOF);\r
- for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
- prob[iSpecies] = probTOF[iSpecies];\r
- break;\r
- case kTPCTOF:\r
- fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTOF|AliPIDResponse::kDetTPC);\r
- detUsedTPCTOF = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTPCTOF);\r
- for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
- prob[iSpecies] = probTPCTOF[iSpecies];\r
- break;\r
- default:\r
- break;\r
- }//end switch: define detector mask\r
- \r
- //Filling the PID QA\r
- Double_t tofTime = -999., length = 999., tof = -999.;\r
- Double_t c = TMath::C()*1.E-9;// m/ns\r
- Double_t beta = -999.;\r
- Double_t nSigmaTOFForParticleOfInterest = -999.;\r
- if ( (track->IsOn(AliESDtrack::kTOFin)) &&\r
- (track->IsOn(AliESDtrack::kTIME)) ) { \r
- tofTime = track->GetTOFsignal();//in ps\r
- length = track->GetIntegratedLength();\r
- tof = tofTime*1E-3; // ns \r
- \r
- if (tof <= 0) {\r
- //Printf("WARNING: track with negative TOF time found! Skipping this track for PID checks\n");\r
- continue;\r
- }\r
- if (length <= 0){\r
- //printf("WARNING: track with negative length found!Skipping this track for PID checks\n");\r
- continue;\r
- }\r
- \r
- length = length*0.01; // in meters\r
- tof = tof*c;\r
- beta = length/tof;\r
- \r
- nSigmaTOFForParticleOfInterest = fPIDResponse->NumberOfSigmasTOF(track,(AliPID::EParticleType)fParticleOfInterest);\r
- fHistBetavsPTOFbeforePID ->Fill(track->P()*track->Charge(),beta);\r
- fHistProbTOFvsPtbeforePID ->Fill(track->Pt(),probTOF[fParticleOfInterest]);\r
- fHistNSigmaTOFvsPtbeforePID ->Fill(track->Pt(),nSigmaTOFForParticleOfInterest);\r
- }//TOF signal \r
- \r
- \r
- Double_t nSigmaTPCForParticleOfInterest = fPIDResponse->NumberOfSigmasTPC(track,(AliPID::EParticleType)fParticleOfInterest);\r
- fHistdEdxVsPTPCbeforePID -> Fill(track->P()*track->Charge(),track->GetTPCsignal());\r
- fHistProbTPCvsPtbeforePID -> Fill(track->Pt(),probTPC[fParticleOfInterest]); \r
- fHistNSigmaTPCvsPtbeforePID -> Fill(track->Pt(),nSigmaTPCForParticleOfInterest); \r
- fHistProbTPCTOFvsPtbeforePID -> Fill(track->Pt(),probTPCTOF[fParticleOfInterest]);\r
- //end of QA-before pid\r
- \r
- if ((detUsedTPC != 0)||(detUsedTOF != 0)||(detUsedTPCTOF != 0)) {\r
- //Make the decision based on the n-sigma\r
- if(fUsePIDnSigma) {\r
- if(nSigma > fPIDNSigma) continue;}\r
- \r
- //Make the decision based on the bayesian\r
- else if(fUsePIDPropabilities) {\r
- if(fParticleOfInterest != TMath::LocMax(AliPID::kSPECIES,prob)) continue;\r
- if (prob[fParticleOfInterest] < fMinAcceptedPIDProbability) continue; \r
- }\r
- \r
- //Fill QA after the PID\r
- fHistBetavsPTOFafterPID ->Fill(track->P()*track->Charge(),beta);\r
- fHistProbTOFvsPtafterPID ->Fill(track->Pt(),probTOF[fParticleOfInterest]);\r
- fHistNSigmaTOFvsPtafterPID ->Fill(track->Pt(),nSigmaTOFForParticleOfInterest);\r
- \r
- fHistdEdxVsPTPCafterPID -> Fill(track->P()*track->Charge(),track->GetTPCsignal());\r
- fHistProbTPCvsPtafterPID -> Fill(track->Pt(),probTPC[fParticleOfInterest]); \r
- fHistProbTPCTOFvsPtafterPID -> Fill(track->Pt(),probTPCTOF[fParticleOfInterest]);\r
- fHistNSigmaTPCvsPtafterPID -> Fill(track->Pt(),nSigmaTPCForParticleOfInterest); \r
- }\r
- \r
- PostData(4, fHistListPIDQA);\r
- }\r
- //===========================PID===============================//\r
- v_charge = track_TPC->Charge();\r
- v_y = track_TPC->Y();\r
- v_eta = track_TPC->Eta();\r
- v_phi = track_TPC->Phi() * TMath::RadToDeg();\r
- v_E = track_TPC->E();\r
- v_pt = track_TPC->Pt();\r
- track_TPC->PxPyPz(v_p);\r
- fHistClus->Fill(track_TPC->GetITSclusters(0),nClustersTPC);\r
- fHistDCA->Fill(b[1],b[0]);\r
- fHistChi2->Fill(chi2PerClusterTPC,fCentrality);\r
- fHistPt->Fill(v_pt,fCentrality);\r
- fHistEta->Fill(v_eta,fCentrality);\r
- fHistPhi->Fill(v_phi,fCentrality);\r
- fHistRapidity->Fill(v_y,fCentrality);\r
- if(v_charge > 0) fHistPhiPos->Fill(v_phi,fCentrality);\r
- else if(v_charge < 0) fHistPhiNeg->Fill(v_phi,fCentrality);\r
-\r
- // fill charge vector\r
- chargeVector[0]->push_back(v_charge);\r
- chargeVector[1]->push_back(v_y);\r
- chargeVector[2]->push_back(v_eta);\r
- chargeVector[3]->push_back(v_phi);\r
- chargeVector[4]->push_back(v_p[0]);\r
- chargeVector[5]->push_back(v_p[1]);\r
- chargeVector[6]->push_back(v_p[2]);\r
- chargeVector[7]->push_back(v_pt);\r
- chargeVector[8]->push_back(v_E);\r
-\r
- if(fRunShuffling) {\r
- chargeVectorShuffle[0]->push_back(v_charge);\r
- chargeVectorShuffle[1]->push_back(v_y);\r
- chargeVectorShuffle[2]->push_back(v_eta);\r
- chargeVectorShuffle[3]->push_back(v_phi);\r
- chargeVectorShuffle[4]->push_back(v_p[0]);\r
- chargeVectorShuffle[5]->push_back(v_p[1]);\r
- chargeVectorShuffle[6]->push_back(v_p[2]);\r
- chargeVectorShuffle[7]->push_back(v_pt);\r
- chargeVectorShuffle[8]->push_back(v_E);\r
- }\r
- \r
- delete track_TPC;\r
- \r
- } //track loop\r
+ // take only events inside centrality class\r
+ if((gCentrality > fCentralityPercentileMin) && (gCentrality < fCentralityPercentileMax)){\r
+ fHistEventStats->Fill(5,gCentrality); //events with correct centrality\r
+ return gCentrality; \r
+ }//centrality class\r
}//Vz cut\r
}//Vy cut\r
}//Vx cut\r
}//proper number of contributors\r
}//vertex object valid\r
}//triggered event \r
- }//ESD analysis\r
+ }//AOD,ESD,ESDMC\r
\r
- //AOD analysis (vertex and track cuts also here!!!!)\r
- else if(gAnalysisLevel == "AOD") {\r
- AliAODEvent* gAOD = dynamic_cast<AliAODEvent*>(InputEvent()); // from TaskSE\r
- if(!gAOD) {\r
- Printf("ERROR: gAOD not available");\r
- return;\r
- }\r
+ // in all other cases return -1 (event not accepted)\r
+ return -1;\r
+}\r
\r
- AliAODHeader *aodHeader = gAOD->GetHeader();\r
\r
- // store offline trigger bits\r
- fHistTriggerStats->Fill(aodHeader->GetOfflineTrigger());\r
+//________________________________________________________________________\r
+Double_t AliAnalysisTaskBFPsi::GetRefMultiOrCentrality(AliVEvent *event){\r
+ // Checks the Event cuts\r
+ // Fills Event statistics histograms\r
+ \r
+ Float_t gCentrality = -1.;\r
+ Double_t fMultiplicity = -100.;\r
+ TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
+ if(fEventClass == "Centrality"){\r
\r
- // event selection done in AliAnalysisTaskSE::Exec() --> this is not used\r
- fHistEventStats->Fill(1,fCentrality); //all events\r
- Bool_t isSelected = kTRUE;\r
- if(fUseOfflineTrigger)\r
- isSelected = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();\r
- if(isSelected) {\r
- fHistEventStats->Fill(2,fCentrality); //triggered events\r
- \r
- //Centrality stuff (centrality in AOD header)\r
- if(fUseCentrality) {\r
- fCentrality = aodHeader->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data());\r
- \r
- // QA for centrality estimators\r
- fHistCentStats->Fill(0.,aodHeader->GetCentralityP()->GetCentralityPercentile("V0M"));\r
- fHistCentStats->Fill(1.,aodHeader->GetCentralityP()->GetCentralityPercentile("FMD"));\r
- fHistCentStats->Fill(2.,aodHeader->GetCentralityP()->GetCentralityPercentile("TRK"));\r
- fHistCentStats->Fill(3.,aodHeader->GetCentralityP()->GetCentralityPercentile("TKL"));\r
- fHistCentStats->Fill(4.,aodHeader->GetCentralityP()->GetCentralityPercentile("CL0"));\r
- fHistCentStats->Fill(5.,aodHeader->GetCentralityP()->GetCentralityPercentile("CL1"));\r
- fHistCentStats->Fill(6.,aodHeader->GetCentralityP()->GetCentralityPercentile("V0MvsFMD"));\r
- fHistCentStats->Fill(7.,aodHeader->GetCentralityP()->GetCentralityPercentile("TKLvsV0M"));\r
- fHistCentStats->Fill(8.,aodHeader->GetCentralityP()->GetCentralityPercentile("ZEMvsZDC"));\r
- \r
- // take only events inside centrality class\r
- if((fCentrality < fCentralityPercentileMin) || (fCentrality > fCentralityPercentileMax)) \r
- return;\r
- \r
- // centrality QA (V0M)\r
- fHistV0M->Fill(gAOD->GetVZEROData()->GetMTotV0A(), gAOD->GetVZEROData()->GetMTotV0C());\r
- \r
- // centrality QA (reference tracks)\r
- fHistRefTracks->Fill(0.,aodHeader->GetRefMultiplicity());\r
- fHistRefTracks->Fill(1.,aodHeader->GetRefMultiplicityPos());\r
- fHistRefTracks->Fill(2.,aodHeader->GetRefMultiplicityNeg());\r
- fHistRefTracks->Fill(3.,aodHeader->GetTPConlyRefMultiplicity());\r
- fHistRefTracks->Fill(4.,aodHeader->GetNumberOfITSClusters(0));\r
- fHistRefTracks->Fill(5.,aodHeader->GetNumberOfITSClusters(1));\r
- fHistRefTracks->Fill(6.,aodHeader->GetNumberOfITSClusters(2));\r
- fHistRefTracks->Fill(7.,aodHeader->GetNumberOfITSClusters(3));\r
- fHistRefTracks->Fill(8.,aodHeader->GetNumberOfITSClusters(4));\r
- }\r
\r
- const AliAODVertex *vertex = gAOD->GetPrimaryVertex();\r
- \r
- if(vertex) {\r
- Double32_t fCov[6];\r
- vertex->GetCovarianceMatrix(fCov);\r
- \r
- if(vertex->GetNContributors() > 0) {\r
- if(fCov[5] != 0) {\r
- fHistEventStats->Fill(3,fCentrality); //events with a proper vertex\r
- if(TMath::Abs(vertex->GetX()) < fVxMax) {\r
- if(TMath::Abs(vertex->GetY()) < fVyMax) {\r
- if(TMath::Abs(vertex->GetZ()) < fVzMax) {\r
- fHistEventStats->Fill(4,fCentrality); //analyzed events\r
- fHistVx->Fill(vertex->GetX());\r
- fHistVy->Fill(vertex->GetY());\r
- fHistVz->Fill(vertex->GetZ(),fCentrality);\r
- \r
- //========Get the VZERO event plane========//\r
- Double_t gVZEROEventPlane = -10.0;\r
- Double_t qxTot = 0.0, qyTot = 0.0;\r
- AliEventplane *ep = gAOD->GetEventplane();\r
- if(ep) \r
- gVZEROEventPlane = ep->CalculateVZEROEventPlane(gAOD,10,2,qxTot,qyTot);\r
- if(gVZEROEventPlane < 0.) gVZEROEventPlane += TMath::Pi();\r
- gReactionPlane = gVZEROEventPlane*TMath::RadToDeg();\r
- fHistEventPlane->Fill(gReactionPlane,fCentrality);\r
- //========Get the VZERO event plane========//\r
-\r
- //Printf("There are %d tracks in this event", gAOD->GetNumberOfTracks());\r
- for (Int_t iTracks = 0; iTracks < gAOD->GetNumberOfTracks(); iTracks++) {\r
- AliAODTrack* aodTrack = dynamic_cast<AliAODTrack *>(gAOD->GetTrack(iTracks));\r
- if (!aodTrack) {\r
- Printf("ERROR: Could not receive track %d", iTracks);\r
- continue;\r
- }\r
- \r
- // AOD track cuts\r
- \r
- // For ESD Filter Information: ANALYSIS/macros/AddTaskESDfilter.C\r
- // take only TPC only tracks \r
- fHistTrackStats->Fill(aodTrack->GetFilterMap());\r
- if(!aodTrack->TestFilterBit(nAODtrackCutBit)) continue;\r
- \r
- v_charge = aodTrack->Charge();\r
- v_y = aodTrack->Y();\r
- v_eta = aodTrack->Eta();\r
- v_phi = aodTrack->Phi() * TMath::RadToDeg();\r
- v_E = aodTrack->E();\r
- v_pt = aodTrack->Pt();\r
- aodTrack->PxPyPz(v_p);\r
- \r
- Float_t DCAxy = aodTrack->DCA(); // this is the DCA from global track (not exactly what is cut on)\r
- Float_t DCAz = aodTrack->ZAtDCA(); // this is the DCA from global track (not exactly what is cut on)\r
- \r
- \r
- // Kinematics cuts from ESD track cuts\r
- if( v_pt < fPtMin || v_pt > fPtMax) continue;\r
- if (!fUsePID) {\r
- if( v_eta < fEtaMin || v_eta > fEtaMax) continue;\r
- }\r
- else if (fUsePID){\r
- if( v_y < fEtaMin || v_y > fEtaMax) continue;\r
- }\r
-\r
- // Extra DCA cuts (for systematic studies [!= -1])\r
- if( fDCAxyCut != -1 && fDCAzCut != -1){\r
- if(TMath::Sqrt((DCAxy*DCAxy)/(fDCAxyCut*fDCAxyCut)+(DCAz*DCAz)/(fDCAzCut*fDCAzCut)) > 1 ){\r
- continue; // 2D cut\r
- }\r
- }\r
- \r
- // Extra TPC cuts (for systematic studies [!= -1])\r
- if( fTPCchi2Cut != -1 && aodTrack->Chi2perNDF() > fTPCchi2Cut){\r
- continue;\r
- }\r
- if( fNClustersTPCCut != -1 && aodTrack->GetTPCNcls() < fNClustersTPCCut){\r
- continue;\r
- }\r
- \r
- // fill QA histograms\r
- fHistClus->Fill(aodTrack->GetITSNcls(),aodTrack->GetTPCNcls());\r
- fHistDCA->Fill(DCAz,DCAxy);\r
- fHistChi2->Fill(aodTrack->Chi2perNDF(),fCentrality);\r
- fHistPt->Fill(v_pt,fCentrality);\r
- fHistEta->Fill(v_eta,fCentrality);\r
- fHistPhi->Fill(v_phi,fCentrality);\r
- fHistRapidity->Fill(v_y,fCentrality);\r
- if(v_charge > 0) fHistPhiPos->Fill(v_phi,fCentrality);\r
- else if(v_charge < 0) fHistPhiNeg->Fill(v_phi,fCentrality);\r
-\r
- // fill charge vector\r
- chargeVector[0]->push_back(v_charge);\r
- chargeVector[1]->push_back(v_y);\r
- chargeVector[2]->push_back(v_eta);\r
- chargeVector[3]->push_back(v_phi);\r
- chargeVector[4]->push_back(v_p[0]);\r
- chargeVector[5]->push_back(v_p[1]);\r
- chargeVector[6]->push_back(v_p[2]);\r
- chargeVector[7]->push_back(v_pt);\r
- chargeVector[8]->push_back(v_E);\r
-\r
- if(fRunShuffling) {\r
- chargeVectorShuffle[0]->push_back(v_charge);\r
- chargeVectorShuffle[1]->push_back(v_y);\r
- chargeVectorShuffle[2]->push_back(v_eta);\r
- chargeVectorShuffle[3]->push_back(v_phi);\r
- chargeVectorShuffle[4]->push_back(v_p[0]);\r
- chargeVectorShuffle[5]->push_back(v_p[1]);\r
- chargeVectorShuffle[6]->push_back(v_p[2]);\r
- chargeVectorShuffle[7]->push_back(v_pt);\r
- chargeVectorShuffle[8]->push_back(v_E);\r
- }\r
- \r
- gNumberOfAcceptedTracks += 1;\r
- \r
- } //track loop\r
- }//Vz cut\r
- }//Vy cut\r
- }//Vx cut\r
- }//proper vertex resolution\r
- }//proper number of contributors\r
- }//vertex object valid\r
- }//triggered event \r
- }//AOD analysis\r
+ if(gAnalysisLevel == "AOD") { //centrality in AOD header\r
+ AliAODHeader *header = (AliAODHeader*) event->GetHeader();\r
+ if(header){\r
+ gCentrality = header->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data());\r
+ }//AOD header\r
+ }//AOD\r
+ \r
+ else if(gAnalysisLevel == "ESD" || gAnalysisLevel == "MCESD"){ // centrality class for ESDs or MC-ESDs\r
+ AliCentrality *centrality = event->GetCentrality();\r
+ gCentrality = centrality->GetCentralityPercentile(fCentralityEstimator.Data());\r
+ }//ESD\r
+ else if(gAnalysisLevel == "MC"){\r
+ Double_t gImpactParameter = 0.;\r
+ if(dynamic_cast<AliMCEvent*>(event)){\r
+ AliCollisionGeometry* headerH = dynamic_cast<AliCollisionGeometry*>(dynamic_cast<AliMCEvent*>(event)->GenEventHeader()); \r
+ if(headerH){\r
+ gImpactParameter = headerH->ImpactParameter();\r
+ gCentrality = gImpactParameter;\r
+ }//MC header\r
+ }//MC event cast\r
+ }//MC\r
+ else{\r
+ gCentrality = -1.;\r
+ }\r
+ }//End if "Centrality"\r
+ if(fEventClass=="Multiplicity"&&gAnalysisLevel == "ESD"){\r
+ if(dynamic_cast<AliESDEvent*>(event)){\r
+ fMultiplicity = fESDtrackCuts->GetReferenceMultiplicity(dynamic_cast<AliESDEvent*>(event), AliESDtrackCuts::kTrackletsITSTPC,0.5);\r
+ }//AliESDevent cast\r
+ }\r
+ if(fEventClass=="Multiplicity"&&gAnalysisLevel != "ESD"){\r
+ AliAODHeader *header = (AliAODHeader*) event->GetHeader();\r
+ if(header){\r
+ fMultiplicity = header->GetRefMultiplicity();\r
+ }//AOD header\r
+ }\r
+ Double_t lReturnVal = -100;\r
+ if(fEventClass=="Multiplicity"){\r
+ lReturnVal = fMultiplicity;\r
+ }else if(fEventClass=="Centrality"){\r
+ lReturnVal = gCentrality;\r
+ }\r
+ return lReturnVal;\r
+}\r
\r
- //MC-ESD analysis\r
- if(gAnalysisLevel == "MCESD") {\r
- AliMCEvent* mcEvent = MCEvent(); \r
- if (!mcEvent) {\r
- Printf("ERROR: mcEvent not available");\r
- return;\r
+//________________________________________________________________________\r
+Double_t AliAnalysisTaskBFPsi::GetEventPlane(AliVEvent *event){\r
+ // Get the event plane\r
+\r
+ TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
+\r
+ Float_t gVZEROEventPlane = -10.;\r
+ Float_t gReactionPlane = -10.;\r
+ Double_t qxTot = 0.0, qyTot = 0.0;\r
+\r
+ //MC: from reaction plane\r
+ if(gAnalysisLevel == "MC"){\r
+ if(dynamic_cast<AliMCEvent*>(event)){\r
+ AliCollisionGeometry* headerH = dynamic_cast<AliCollisionGeometry*>(dynamic_cast<AliMCEvent*>(event)->GenEventHeader()); \r
+ if (headerH) {\r
+ gReactionPlane = headerH->ReactionPlaneAngle();\r
+ //gReactionPlane *= TMath::RadToDeg();\r
+ }//MC header\r
+ }//MC event cast\r
+ }//MC\r
+ \r
+ // AOD,ESD,ESDMC: from VZERO Event Plane\r
+ else{\r
+ \r
+ AliEventplane *ep = event->GetEventplane();\r
+ if(ep){ \r
+ gVZEROEventPlane = ep->CalculateVZEROEventPlane(event,10,2,qxTot,qyTot);\r
+ if(gVZEROEventPlane < 0.) gVZEROEventPlane += TMath::Pi();\r
+ //gReactionPlane = gVZEROEventPlane*TMath::RadToDeg();\r
+ gReactionPlane = gVZEROEventPlane;\r
}\r
+ }//AOD,ESD,ESDMC\r
\r
- AliESDEvent* gESD = dynamic_cast<AliESDEvent*>(InputEvent()); // from TaskSE\r
- if (!gESD) {\r
- Printf("ERROR: gESD not available");\r
- return;\r
+ return gReactionPlane;\r
+}\r
+\r
+//________________________________________________________________________\r
+Double_t AliAnalysisTaskBFPsi::GetTrackbyTrackCorrectionMatrix( Double_t vEta, \r
+ Double_t vPhi, \r
+ Double_t vPt, \r
+ Short_t vCharge, \r
+ Double_t gCentrality) {\r
+ // -- Get efficiency correction of particle dependent on (eta, phi, pt, charge, centrality) \r
+\r
+ Double_t correction = 1.;\r
+ Int_t binEta = 0, binPt = 0, binPhi = 0;\r
+\r
+ //Printf("EtaMAx: %lf - EtaMin: %lf - EtaBin: %lf", fEtaMaxForCorrections,fEtaMinForCorrections,fEtaBinForCorrections);\r
+ if(fEtaBinForCorrections != 0) {\r
+ Double_t widthEta = (fEtaMaxForCorrections - fEtaMinForCorrections)/fEtaBinForCorrections;\r
+ if(fEtaMaxForCorrections != fEtaMinForCorrections) \r
+ binEta = (Int_t)(vEta/widthEta)+1;\r
+ }\r
+\r
+ if(fPtBinForCorrections != 0) {\r
+ Double_t widthPt = (fPtMaxForCorrections - fPtMinForCorrections)/fPtBinForCorrections;\r
+ if(fPtMaxForCorrections != fPtMinForCorrections) \r
+ binPt = (Int_t)(vPt/widthPt) + 1;\r
+ }\r
+ \r
+ if(fPhiBinForCorrections != 0) {\r
+ Double_t widthPhi = (fPhiMaxForCorrections - fPhiMinForCorrections)/fPhiBinForCorrections;\r
+ if(fPhiMaxForCorrections != fPhiMinForCorrections) \r
+ binPhi = (Int_t)(vPhi/widthPhi)+ 1;\r
+ }\r
+\r
+ Int_t gCentralityInt = 1;\r
+ for (Int_t i=0; i<=kCENTRALITY; i++){\r
+ if((centralityArrayForPbPb[i] <= gCentrality)&&(gCentrality <= centralityArrayForPbPb[i+1]))\r
+ gCentralityInt = i;\r
+ }\r
+ \r
+ if(fHistMatrixCorrectionPlus[gCentralityInt]){\r
+ if (vCharge > 0) {\r
+ correction = fHistMatrixCorrectionPlus[gCentralityInt]->GetBinContent(fHistMatrixCorrectionPlus[gCentralityInt]->GetBin(binEta, binPt, binPhi));\r
+ }\r
+ if (vCharge < 0) {\r
+ correction = fHistMatrixCorrectionMinus[gCentralityInt]->GetBinContent(fHistMatrixCorrectionMinus[gCentralityInt]->GetBin(binEta, binPt, binPhi));\r
}\r
+ }\r
+ else {\r
+ correction = 1.;\r
+ }\r
+ if (correction == 0.) { \r
+ AliError(Form("Should not happen : bin content = 0. >> eta: %.2f | phi : %.2f | pt : %.2f | cent %d",vEta, vPhi, vPt, gCentralityInt)); \r
+ correction = 1.; \r
+ } \r
+ \r
+ return correction;\r
+}\r
\r
- // store offline trigger bits\r
- fHistTriggerStats->Fill(((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected());\r
+//________________________________________________________________________\r
+TObjArray* AliAnalysisTaskBFPsi::GetAcceptedTracks(AliVEvent *event, Double_t gCentrality, Double_t gReactionPlane){\r
+ // Returns TObjArray with tracks after all track cuts (only for AOD!)\r
+ // Fills QA histograms\r
\r
- // event selection done in AliAnalysisTaskSE::Exec() --> this is not used\r
- fHistEventStats->Fill(1,fCentrality); //all events\r
- Bool_t isSelected = kTRUE;\r
- if(fUseOfflineTrigger)\r
- isSelected = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();\r
- if(isSelected) {\r
- fHistEventStats->Fill(2,fCentrality); //triggered events\r
+ TString gAnalysisLevel = fBalance->GetAnalysisLevel();\r
\r
- if(fUseCentrality) {\r
- //Centrality stuff\r
- AliCentrality *centrality = gESD->GetCentrality();\r
+ //output TObjArray holding all good tracks\r
+ TObjArray* tracksAccepted = new TObjArray;\r
+ tracksAccepted->SetOwner(kTRUE);\r
\r
- fCentrality = centrality->GetCentralityPercentile(fCentralityEstimator.Data());\r
- \r
- // take only events inside centrality class\r
- if(!centrality->IsEventInCentralityClass(fCentralityPercentileMin,\r
- fCentralityPercentileMax,\r
- fCentralityEstimator.Data()))\r
- return;\r
- \r
- // centrality QA (V0M)\r
- fHistV0M->Fill(gESD->GetVZEROData()->GetMTotV0A(), gESD->GetVZEROData()->GetMTotV0C());\r
- }\r
- \r
- const AliESDVertex *vertex = gESD->GetPrimaryVertex();\r
- if(vertex) {\r
- if(vertex->GetNContributors() > 0) {\r
- if(vertex->GetZRes() != 0) {\r
- fHistEventStats->Fill(3,fCentrality); //events with a proper vertex\r
- if(TMath::Abs(vertex->GetXv()) < fVxMax) {\r
- if(TMath::Abs(vertex->GetYv()) < fVyMax) {\r
- if(TMath::Abs(vertex->GetZv()) < fVzMax) {\r
- fHistEventStats->Fill(4,fCentrality); //analayzed events\r
- fHistVx->Fill(vertex->GetXv());\r
- fHistVy->Fill(vertex->GetYv());\r
- fHistVz->Fill(vertex->GetZv(),fCentrality);\r
- \r
- //========Get the VZERO event plane========//\r
- Double_t gVZEROEventPlane = -10.0;\r
- Double_t qxTot = 0.0, qyTot = 0.0;\r
- AliEventplane *ep = gESD->GetEventplane();\r
- if(ep) \r
- gVZEROEventPlane = ep->CalculateVZEROEventPlane(gESD,10,2,qxTot,qyTot);\r
- if(gVZEROEventPlane < 0.) gVZEROEventPlane += TMath::Pi();\r
- gReactionPlane = gVZEROEventPlane*TMath::RadToDeg();\r
- fHistEventPlane->Fill(gReactionPlane,fCentrality);\r
- //========Get the VZERO event plane========//\r
-\r
- //Printf("There are %d tracks in this event", gESD->GetNumberOfTracks());\r
- for (Int_t iTracks = 0; iTracks < gESD->GetNumberOfTracks(); iTracks++) {\r
- AliESDtrack* track = dynamic_cast<AliESDtrack *>(gESD->GetTrack(iTracks));\r
- if (!track) {\r
- Printf("ERROR: Could not receive track %d", iTracks);\r
- continue;\r
- } \r
- \r
- Int_t label = TMath::Abs(track->GetLabel());\r
- if(label > mcEvent->GetNumberOfTracks()) continue;\r
- if(label > mcEvent->GetNumberOfPrimaries()) continue;\r
- \r
- AliMCParticle* mcTrack = dynamic_cast<AliMCParticle *>(mcEvent->GetTrack(label));\r
- if(!mcTrack) continue;\r
-\r
- // take only TPC only tracks\r
- track_TPC = new AliESDtrack();\r
- if(!track->FillTPCOnlyTrack(*track_TPC)) continue;\r
- \r
- //ESD track cuts\r
- if(fESDtrackCuts) \r
- if(!fESDtrackCuts->AcceptTrack(track_TPC)) continue;\r
- \r
- // fill QA histograms\r
- Float_t b[2];\r
- Float_t bCov[3];\r
- track_TPC->GetImpactParameters(b,bCov);\r
- if (bCov[0]<=0 || bCov[2]<=0) {\r
- AliDebug(1, "Estimated b resolution lower or equal zero!");\r
- bCov[0]=0; bCov[2]=0;\r
- }\r
- \r
- Int_t nClustersTPC = -1;\r
- nClustersTPC = track_TPC->GetTPCNclsIter1(); // TPC standalone\r
- //nClustersTPC = track->GetTPCclusters(0); // global track\r
- Float_t chi2PerClusterTPC = -1;\r
- if (nClustersTPC!=0) {\r
- chi2PerClusterTPC = track_TPC->GetTPCchi2Iter1()/Float_t(nClustersTPC); // TPC standalone\r
- //chi2PerClusterTPC = track->GetTPCchi2()/Float_t(nClustersTPC); // global track\r
- }\r
- \r
- v_charge = track_TPC->Charge();\r
- v_y = track_TPC->Y();\r
- v_eta = track_TPC->Eta();\r
- v_phi = track_TPC->Phi() * TMath::RadToDeg();\r
- v_E = track_TPC->E();\r
- v_pt = track_TPC->Pt();\r
- track_TPC->PxPyPz(v_p);\r
-\r
- fHistClus->Fill(track_TPC->GetITSclusters(0),nClustersTPC);\r
- fHistDCA->Fill(b[1],b[0]);\r
- fHistChi2->Fill(chi2PerClusterTPC,fCentrality);\r
- fHistPt->Fill(v_pt,fCentrality);\r
- fHistEta->Fill(v_eta,fCentrality);\r
- fHistPhi->Fill(v_phi,fCentrality);\r
- fHistRapidity->Fill(v_y,fCentrality);\r
- if(v_charge > 0) fHistPhiPos->Fill(v_phi,fCentrality);\r
- else if(v_charge < 0) fHistPhiNeg->Fill(v_phi,fCentrality);\r
-\r
- // fill charge vector\r
- chargeVector[0]->push_back(v_charge);\r
- chargeVector[1]->push_back(v_y);\r
- chargeVector[2]->push_back(v_eta);\r
- chargeVector[3]->push_back(v_phi);\r
- chargeVector[4]->push_back(v_p[0]);\r
- chargeVector[5]->push_back(v_p[1]);\r
- chargeVector[6]->push_back(v_p[2]);\r
- chargeVector[7]->push_back(v_pt);\r
- chargeVector[8]->push_back(v_E);\r
-\r
- if(fRunShuffling) {\r
- chargeVectorShuffle[0]->push_back(v_charge);\r
- chargeVectorShuffle[1]->push_back(v_y);\r
- chargeVectorShuffle[2]->push_back(v_eta);\r
- chargeVectorShuffle[3]->push_back(v_phi);\r
- chargeVectorShuffle[4]->push_back(v_p[0]);\r
- chargeVectorShuffle[5]->push_back(v_p[1]);\r
- chargeVectorShuffle[6]->push_back(v_p[2]);\r
- chargeVectorShuffle[7]->push_back(v_pt);\r
- chargeVectorShuffle[8]->push_back(v_E);\r
- }\r
- \r
- delete track_TPC;\r
- gNumberOfAcceptedTracks += 1;\r
- \r
- } //track loop\r
- }//Vz cut\r
- }//Vy cut\r
- }//Vx cut\r
- }//proper vertex resolution\r
- }//proper number of contributors\r
- }//vertex object valid\r
- }//triggered event \r
- }//MC-ESD analysis\r
+ Short_t vCharge;\r
+ Double_t vEta;\r
+ Double_t vY;\r
+ Double_t vPhi;\r
+ Double_t vPt;\r
\r
- //MC analysis\r
- else if(gAnalysisLevel == "MC") {\r
- AliMCEvent* mcEvent = MCEvent(); \r
- if (!mcEvent) {\r
- Printf("ERROR: mcEvent not available");\r
- return;\r
- }\r
- fHistEventStats->Fill(1,fCentrality); //total events\r
- fHistEventStats->Fill(2,fCentrality); //offline trigger\r
\r
- Double_t gImpactParameter = 0.;\r
- if(fUseCentrality) {\r
- //Get the MC header\r
- AliGenHijingEventHeader* headerH = dynamic_cast<AliGenHijingEventHeader*>(mcEvent->GenEventHeader());\r
- if (headerH) {\r
- //Printf("=====================================================");\r
- //Printf("Reaction plane angle: %lf",headerH->ReactionPlaneAngle());\r
- //Printf("Impact parameter: %lf",headerH->ImpactParameter());\r
- //Printf("=====================================================");\r
- gReactionPlane = headerH->ReactionPlaneAngle();\r
- gImpactParameter = headerH->ImpactParameter();\r
- fCentrality = gImpactParameter;\r
+ if(gAnalysisLevel == "AOD") { // handling of TPC only tracks different in AOD and ESD\r
+ // Loop over tracks in event\r
+ for (Int_t iTracks = 0; iTracks < event->GetNumberOfTracks(); iTracks++) {\r
+ AliAODTrack* aodTrack = dynamic_cast<AliAODTrack *>(event->GetTrack(iTracks));\r
+ if (!aodTrack) {\r
+ AliError(Form("Could not receive track %d", iTracks));\r
+ continue;\r
+ }\r
+ \r
+ // AOD track cuts\r
+ \r
+ // For ESD Filter Information: ANALYSIS/macros/AddTaskESDfilter.C\r
+ // take only TPC only tracks \r
+ //fHistTrackStats->Fill(aodTrack->GetFilterMap());\r
+ for(Int_t iTrackBit = 0; iTrackBit < 16; iTrackBit++){\r
+ fHistTrackStats->Fill(iTrackBit,aodTrack->TestFilterBit(1<<iTrackBit));\r
+ }\r
+ if(!aodTrack->TestFilterBit(nAODtrackCutBit)) continue;\r
+ \r
+ vCharge = aodTrack->Charge();\r
+ vEta = aodTrack->Eta();\r
+ vY = aodTrack->Y();\r
+ vPhi = aodTrack->Phi();// * TMath::RadToDeg();\r
+ vPt = aodTrack->Pt();\r
+ \r
+ Float_t dcaXY = aodTrack->DCA(); // this is the DCA from global track (not exactly what is cut on)\r
+ Float_t dcaZ = aodTrack->ZAtDCA(); // this is the DCA from global track (not exactly what is cut on)\r
+ \r
+ \r
+ // Kinematics cuts from ESD track cuts\r
+ if( vPt < fPtMin || vPt > fPtMax) continue;\r
+ if( vEta < fEtaMin || vEta > fEtaMax) continue;\r
+ \r
+ // Extra DCA cuts (for systematic studies [!= -1])\r
+ if( fDCAxyCut != -1 && fDCAzCut != -1){\r
+ if(TMath::Sqrt((dcaXY*dcaXY)/(fDCAxyCut*fDCAxyCut)+(dcaZ*dcaZ)/(fDCAzCut*fDCAzCut)) > 1 ){\r
+ continue; // 2D cut\r
+ }\r
+ }\r
+ \r
+ // Extra TPC cuts (for systematic studies [!= -1])\r
+ if( fTPCchi2Cut != -1 && aodTrack->Chi2perNDF() > fTPCchi2Cut){\r
+ continue;\r
+ }\r
+ if( fNClustersTPCCut != -1 && aodTrack->GetTPCNcls() < fNClustersTPCCut){\r
+ continue;\r
}\r
- fCentrality = gImpactParameter;\r
- fHistEventPlane->Fill(gReactionPlane*TMath::RadToDeg(),fCentrality);\r
+ \r
+ // fill QA histograms\r
+ fHistClus->Fill(aodTrack->GetITSNcls(),aodTrack->GetTPCNcls());\r
+ fHistDCA->Fill(dcaZ,dcaXY);\r
+ fHistChi2->Fill(aodTrack->Chi2perNDF(),gCentrality);\r
+ fHistPt->Fill(vPt,gCentrality);\r
+ fHistEta->Fill(vEta,gCentrality);\r
+ fHistRapidity->Fill(vY,gCentrality);\r
+ if(vCharge > 0) fHistPhiPos->Fill(vPhi,gCentrality);\r
+ else if(vCharge < 0) fHistPhiNeg->Fill(vPhi,gCentrality);\r
+ fHistPhi->Fill(vPhi,gCentrality);\r
+ \r
+ //=======================================correction\r
+ Double_t correction = GetTrackbyTrackCorrectionMatrix(vEta, vPhi, vPt, vCharge, gCentrality); \r
+ \r
+ // add the track to the TObjArray\r
+ tracksAccepted->Add(new AliBFBasicParticle(vEta, vPhi, vPt, vCharge, correction)); \r
+ }//track loop\r
+ }// AOD analysis\r
\r
- // take only events inside centrality class (DIDN'T CHANGE THIS UP TO NOW)\r
- if((fImpactParameterMin > gImpactParameter) || (fImpactParameterMax < gImpactParameter))\r
- return;\r
- }\r
+\r
+ else if(gAnalysisLevel == "ESD" || gAnalysisLevel == "MCESD") { // handling of TPC only tracks different in AOD and ESD\r
+\r
+ AliESDtrack *trackTPC = NULL;\r
+ AliMCParticle *mcTrack = NULL;\r
+\r
+ AliMCEvent* mcEvent = NULL;\r
\r
- AliGenEventHeader *header = mcEvent->GenEventHeader();\r
- if(!header) return;\r
+ // for MC ESDs use also MC information\r
+ if(gAnalysisLevel == "MCESD"){\r
+ mcEvent = MCEvent(); \r
+ if (!mcEvent) {\r
+ AliError("mcEvent not available");\r
+ return tracksAccepted;\r
+ }\r
+ }\r
\r
- TArrayF gVertexArray;\r
- header->PrimaryVertex(gVertexArray);\r
- //Printf("Vertex: %lf (x) - %lf (y) - %lf (z)",\r
- //gVertexArray.At(0),\r
- //gVertexArray.At(1),\r
- //gVertexArray.At(2));\r
- fHistEventStats->Fill(3,fCentrality); //events with a proper vertex\r
- if(TMath::Abs(gVertexArray.At(0)) < fVxMax) {\r
- if(TMath::Abs(gVertexArray.At(1)) < fVyMax) {\r
- if(TMath::Abs(gVertexArray.At(2)) < fVzMax) {\r
- fHistEventStats->Fill(4,fCentrality); //analayzed events\r
- fHistVx->Fill(gVertexArray.At(0));\r
- fHistVy->Fill(gVertexArray.At(1));\r
- fHistVz->Fill(gVertexArray.At(2),fCentrality);\r
+ // Loop over tracks in event\r
+ for (Int_t iTracks = 0; iTracks < event->GetNumberOfTracks(); iTracks++) {\r
+ AliESDtrack* track = dynamic_cast<AliESDtrack *>(event->GetTrack(iTracks));\r
+ if (!track) {\r
+ AliError(Form("Could not receive track %d", iTracks));\r
+ continue;\r
+ } \r
+\r
+ // for MC ESDs use also MC information --> MC track not used further???\r
+ if(gAnalysisLevel == "MCESD"){\r
+ Int_t label = TMath::Abs(track->GetLabel());\r
+ if(label > mcEvent->GetNumberOfTracks()) continue;\r
+ if(label > mcEvent->GetNumberOfPrimaries()) continue;\r
+ \r
+ mcTrack = dynamic_cast<AliMCParticle *>(mcEvent->GetTrack(label));\r
+ if(!mcTrack) continue;\r
+ }\r
+\r
+ // take only TPC only tracks\r
+ trackTPC = new AliESDtrack();\r
+ if(!track->FillTPCOnlyTrack(*trackTPC)) continue;\r
+ \r
+ //ESD track cuts\r
+ if(fESDtrackCuts) \r
+ if(!fESDtrackCuts->AcceptTrack(trackTPC)) continue;\r
+ \r
+ // fill QA histograms\r
+ Float_t b[2];\r
+ Float_t bCov[3];\r
+ trackTPC->GetImpactParameters(b,bCov);\r
+ if (bCov[0]<=0 || bCov[2]<=0) {\r
+ AliDebug(1, "Estimated b resolution lower or equal zero!");\r
+ bCov[0]=0; bCov[2]=0;\r
+ }\r
+ \r
+ Int_t nClustersTPC = -1;\r
+ nClustersTPC = trackTPC->GetTPCNclsIter1(); // TPC standalone\r
+ //nClustersTPC = track->GetTPCclusters(0); // global track\r
+ Float_t chi2PerClusterTPC = -1;\r
+ if (nClustersTPC!=0) {\r
+ chi2PerClusterTPC = trackTPC->GetTPCchi2Iter1()/Float_t(nClustersTPC); // TPC standalone\r
+ //chi2PerClusterTPC = track->GetTPCchi2()/Float_t(nClustersTPC); // global track\r
+ }\r
+ \r
+ //===========================PID===============================// \r
+ if(fUsePID) {\r
+ Double_t prob[AliPID::kSPECIES]={0.};\r
+ Double_t probTPC[AliPID::kSPECIES]={0.};\r
+ Double_t probTOF[AliPID::kSPECIES]={0.};\r
+ Double_t probTPCTOF[AliPID::kSPECIES]={0.};\r
+ \r
+ Double_t nSigma = 0.;\r
+ UInt_t detUsedTPC = 0;\r
+ UInt_t detUsedTOF = 0;\r
+ UInt_t detUsedTPCTOF = 0;\r
+ \r
+ //Decide what detector configuration we want to use\r
+ switch(fPidDetectorConfig) {\r
+ case kTPCpid:\r
+ fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTPC);\r
+ nSigma = TMath::Abs(fPIDResponse->NumberOfSigmasTPC(track,(AliPID::EParticleType)fParticleOfInterest));\r
+ detUsedTPC = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTPC);\r
+ for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
+ prob[iSpecies] = probTPC[iSpecies];\r
+ break;\r
+ case kTOFpid:\r
+ fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTOF);\r
+ nSigma = TMath::Abs(fPIDResponse->NumberOfSigmasTOF(track,(AliPID::EParticleType)fParticleOfInterest));\r
+ detUsedTOF = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTOF);\r
+ for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
+ prob[iSpecies] = probTOF[iSpecies];\r
+ break;\r
+ case kTPCTOF:\r
+ fPIDCombined->SetDetectorMask(AliPIDResponse::kDetTOF|AliPIDResponse::kDetTPC);\r
+ detUsedTPCTOF = fPIDCombined->ComputeProbabilities(track, fPIDResponse, probTPCTOF);\r
+ for(Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++)\r
+ prob[iSpecies] = probTPCTOF[iSpecies];\r
+ break;\r
+ default:\r
+ break;\r
+ }//end switch: define detector mask\r
+ \r
+ //Filling the PID QA\r
+ Double_t tofTime = -999., length = 999., tof = -999.;\r
+ Double_t c = TMath::C()*1.E-9;// m/ns\r
+ Double_t beta = -999.;\r
+ Double_t nSigmaTOFForParticleOfInterest = -999.;\r
+ if ( (track->IsOn(AliESDtrack::kTOFin)) &&\r
+ (track->IsOn(AliESDtrack::kTIME)) ) { \r
+ tofTime = track->GetTOFsignal();//in ps\r
+ length = track->GetIntegratedLength();\r
+ tof = tofTime*1E-3; // ns \r
\r
- Printf("There are %d tracks in this event", mcEvent->GetNumberOfPrimaries());\r
- for (Int_t iTracks = 0; iTracks < mcEvent->GetNumberOfPrimaries(); iTracks++) {\r
- AliMCParticle* track = dynamic_cast<AliMCParticle *>(mcEvent->GetTrack(iTracks));\r
- if (!track) {\r
- Printf("ERROR: Could not receive particle %d", iTracks);\r
- continue;\r
- }\r
- \r
- //exclude non stable particles\r
- if(!mcEvent->IsPhysicalPrimary(iTracks)) continue;\r
+ if (tof <= 0) {\r
+ //Printf("WARNING: track with negative TOF time found! Skipping this track for PID checks\n");\r
+ continue;\r
+ }\r
+ if (length <= 0){\r
+ //printf("WARNING: track with negative length found!Skipping this track for PID checks\n");\r
+ continue;\r
+ }\r
+ \r
+ length = length*0.01; // in meters\r
+ tof = tof*c;\r
+ beta = length/tof;\r
+ \r
+ nSigmaTOFForParticleOfInterest = fPIDResponse->NumberOfSigmasTOF(track,(AliPID::EParticleType)fParticleOfInterest);\r
+ fHistBetavsPTOFbeforePID ->Fill(track->P()*track->Charge(),beta);\r
+ fHistProbTOFvsPtbeforePID ->Fill(track->Pt(),probTOF[fParticleOfInterest]);\r
+ fHistNSigmaTOFvsPtbeforePID ->Fill(track->Pt(),nSigmaTOFForParticleOfInterest);\r
+ }//TOF signal \r
+ \r
+ \r
+ Double_t nSigmaTPCForParticleOfInterest = fPIDResponse->NumberOfSigmasTPC(track,(AliPID::EParticleType)fParticleOfInterest);\r
+ fHistdEdxVsPTPCbeforePID -> Fill(track->P()*track->Charge(),track->GetTPCsignal());\r
+ fHistProbTPCvsPtbeforePID -> Fill(track->Pt(),probTPC[fParticleOfInterest]); \r
+ fHistNSigmaTPCvsPtbeforePID -> Fill(track->Pt(),nSigmaTPCForParticleOfInterest); \r
+ fHistProbTPCTOFvsPtbeforePID -> Fill(track->Pt(),probTPCTOF[fParticleOfInterest]);\r
+ //end of QA-before pid\r
+ \r
+ if ((detUsedTPC != 0)||(detUsedTOF != 0)||(detUsedTPCTOF != 0)) {\r
+ //Make the decision based on the n-sigma\r
+ if(fUsePIDnSigma) {\r
+ if(nSigma > fPIDNSigma) continue;}\r
+ \r
+ //Make the decision based on the bayesian\r
+ else if(fUsePIDPropabilities) {\r
+ if(fParticleOfInterest != TMath::LocMax(AliPID::kSPECIES,prob)) continue;\r
+ if (prob[fParticleOfInterest] < fMinAcceptedPIDProbability) continue; \r
+ }\r
+ \r
+ //Fill QA after the PID\r
+ fHistBetavsPTOFafterPID ->Fill(track->P()*track->Charge(),beta);\r
+ fHistProbTOFvsPtafterPID ->Fill(track->Pt(),probTOF[fParticleOfInterest]);\r
+ fHistNSigmaTOFvsPtafterPID ->Fill(track->Pt(),nSigmaTOFForParticleOfInterest);\r
+ \r
+ fHistdEdxVsPTPCafterPID -> Fill(track->P()*track->Charge(),track->GetTPCsignal());\r
+ fHistProbTPCvsPtafterPID -> Fill(track->Pt(),probTPC[fParticleOfInterest]); \r
+ fHistProbTPCTOFvsPtafterPID -> Fill(track->Pt(),probTPCTOF[fParticleOfInterest]);\r
+ fHistNSigmaTPCvsPtafterPID -> Fill(track->Pt(),nSigmaTPCForParticleOfInterest); \r
+ }\r
+ }\r
+ //===========================PID===============================//\r
+ vCharge = trackTPC->Charge();\r
+ vY = trackTPC->Y();\r
+ vEta = trackTPC->Eta();\r
+ vPhi = trackTPC->Phi();// * TMath::RadToDeg();\r
+ vPt = trackTPC->Pt();\r
+ fHistClus->Fill(trackTPC->GetITSclusters(0),nClustersTPC);\r
+ fHistDCA->Fill(b[1],b[0]);\r
+ fHistChi2->Fill(chi2PerClusterTPC,gCentrality);\r
+ fHistPt->Fill(vPt,gCentrality);\r
+ fHistEta->Fill(vEta,gCentrality);\r
+ fHistPhi->Fill(vPhi,gCentrality);\r
+ fHistRapidity->Fill(vY,gCentrality);\r
+ if(vCharge > 0) fHistPhiPos->Fill(vPhi,gCentrality);\r
+ else if(vCharge < 0) fHistPhiNeg->Fill(vPhi,gCentrality);\r
+ \r
+ //=======================================correction\r
+ Double_t correction = GetTrackbyTrackCorrectionMatrix(vEta, vPhi, vPt, vCharge, gCentrality); \r
\r
- v_eta = track->Eta();\r
- v_pt = track->Pt();\r
- v_y = track->Y();\r
+ // add the track to the TObjArray\r
+ tracksAccepted->Add(new AliBFBasicParticle(vEta, vPhi, vPt, vCharge, correction)); \r
\r
- if( v_pt < fPtMin || v_pt > fPtMax) \r
- continue;\r
- if (!fUsePID) {\r
- if( v_eta < fEtaMin || v_eta > fEtaMax) continue;\r
- }\r
- else if (fUsePID){\r
- if( v_y < fEtaMin || v_y > fEtaMax) continue;\r
- }\r
+ delete trackTPC;\r
+ }//track loop\r
+ }// ESD analysis\r
\r
- //analyze one set of particles\r
- if(fUseMCPdgCode) {\r
- TParticle *particle = track->Particle();\r
- if(!particle) continue;\r
- \r
- Int_t gPdgCode = particle->GetPdgCode();\r
- if(TMath::Abs(fPDGCodeToBeAnalyzed) != TMath::Abs(gPdgCode)) \r
- continue;\r
- }\r
- \r
- //Use the acceptance parameterization\r
- if(fAcceptanceParameterization) {\r
- Double_t gRandomNumber = gRandom->Rndm();\r
- if(gRandomNumber > fAcceptanceParameterization->Eval(track->Pt())) \r
- continue;\r
- }\r
+ else if(gAnalysisLevel == "MC"){\r
+ \r
+ // Loop over tracks in event\r
+ for (Int_t iTracks = 0; iTracks < dynamic_cast<AliMCEvent*>(event)->GetNumberOfPrimaries(); iTracks++) {\r
+ AliMCParticle* track = dynamic_cast<AliMCParticle *>(event->GetTrack(iTracks));\r
+ if (!track) {\r
+ AliError(Form("Could not receive particle %d", iTracks));\r
+ continue;\r
+ }\r
\r
- //Exclude resonances\r
- if(fExcludeResonancesInMC) {\r
- TParticle *particle = track->Particle();\r
- if(!particle) continue;\r
- \r
- Bool_t kExcludeParticle = kFALSE;\r
- Int_t gMotherIndex = particle->GetFirstMother();\r
- if(gMotherIndex != -1) {\r
- AliMCParticle* motherTrack = dynamic_cast<AliMCParticle *>(mcEvent->GetTrack(gMotherIndex));\r
- if(motherTrack) {\r
- TParticle *motherParticle = motherTrack->Particle();\r
- if(motherParticle) {\r
- Int_t pdgCodeOfMother = motherParticle->GetPdgCode();\r
- //if((pdgCodeOfMother == 113)||(pdgCodeOfMother == 213)||(pdgCodeOfMother == 221)||(pdgCodeOfMother == 223)||(pdgCodeOfMother == 331)||(pdgCodeOfMother == 333)) {\r
- if(pdgCodeOfMother == 113) {\r
- kExcludeParticle = kTRUE;\r
- }\r
- }\r
- }\r
- }\r
- \r
- //Exclude from the analysis decay products of rho0, rho+, eta, eta' and phi\r
- if(kExcludeParticle) continue;\r
- }\r
+ //exclude non stable particles\r
+ if(!(dynamic_cast<AliMCEvent*>(event)->IsPhysicalPrimary(iTracks))) continue;\r
\r
- v_charge = track->Charge();\r
- v_phi = track->Phi();\r
- v_E = track->E();\r
- track->PxPyPz(v_p);\r
- //Printf("phi (before): %lf",v_phi);\r
-\r
- fHistPt->Fill(v_pt,fCentrality);\r
- fHistEta->Fill(v_eta,fCentrality);\r
- fHistPhi->Fill(v_phi*TMath::RadToDeg(),fCentrality);\r
- fHistRapidity->Fill(v_y,fCentrality);\r
- if(v_charge > 0) fHistPhiPos->Fill(v_phi*TMath::RadToDeg(),fCentrality);\r
- else if(v_charge < 0) fHistPhiNeg->Fill(v_phi*TMath::RadToDeg(),fCentrality);\r
-\r
- //Flow after burner\r
- if(fUseFlowAfterBurner) {\r
- Double_t precisionPhi = 0.001;\r
- Int_t maxNumberOfIterations = 100;\r
-\r
- Double_t phi0 = v_phi;\r
- Double_t gV2 = fDifferentialV2->Eval(v_pt);\r
-\r
- for (Int_t j = 0; j < maxNumberOfIterations; j++) {\r
- Double_t phiprev = v_phi;\r
- Double_t fl = v_phi - phi0 + gV2*TMath::Sin(2.*(v_phi - gReactionPlane));\r
- Double_t fp = 1.0 + 2.0*gV2*TMath::Cos(2.*(v_phi - gReactionPlane)); \r
- v_phi -= fl/fp;\r
- if (TMath::AreEqualAbs(phiprev,v_phi,precisionPhi)) break;\r
+ vEta = track->Eta();\r
+ vPt = track->Pt();\r
+ vY = track->Y();\r
+ \r
+ if( vPt < fPtMin || vPt > fPtMax) \r
+ continue;\r
+ if (!fUsePID) {\r
+ if( vEta < fEtaMin || vEta > fEtaMax) continue;\r
+ }\r
+ else if (fUsePID){\r
+ if( vY < fEtaMin || vY > fEtaMax) continue;\r
+ }\r
+ \r
+ //analyze one set of particles\r
+ if(fUseMCPdgCode) {\r
+ TParticle *particle = track->Particle();\r
+ if(!particle) continue;\r
+ \r
+ Int_t gPdgCode = particle->GetPdgCode();\r
+ if(TMath::Abs(fPDGCodeToBeAnalyzed) != TMath::Abs(gPdgCode)) \r
+ continue;\r
+ }\r
+ \r
+ //Use the acceptance parameterization\r
+ if(fAcceptanceParameterization) {\r
+ Double_t gRandomNumber = gRandom->Rndm();\r
+ if(gRandomNumber > fAcceptanceParameterization->Eval(track->Pt())) \r
+ continue;\r
+ }\r
+ \r
+ //Exclude resonances\r
+ if(fExcludeResonancesInMC) {\r
+ TParticle *particle = track->Particle();\r
+ if(!particle) continue;\r
+ \r
+ Bool_t kExcludeParticle = kFALSE;\r
+ Int_t gMotherIndex = particle->GetFirstMother();\r
+ if(gMotherIndex != -1) {\r
+ AliMCParticle* motherTrack = dynamic_cast<AliMCParticle *>(event->GetTrack(gMotherIndex));\r
+ if(motherTrack) {\r
+ TParticle *motherParticle = motherTrack->Particle();\r
+ if(motherParticle) {\r
+ Int_t pdgCodeOfMother = motherParticle->GetPdgCode();\r
+ //if((pdgCodeOfMother == 113)||(pdgCodeOfMother == 213)||(pdgCodeOfMother == 221)||(pdgCodeOfMother == 223)||(pdgCodeOfMother == 331)||(pdgCodeOfMother == 333)) {\r
+ if(pdgCodeOfMother == 113) {\r
+ kExcludeParticle = kTRUE;\r
}\r
- //Printf("phi (after): %lf\n",v_phi);\r
- Double_t v_DeltaphiBefore = phi0 - gReactionPlane;\r
- if(v_DeltaphiBefore < 0) v_DeltaphiBefore += 2*TMath::Pi();\r
- fHistPhiBefore->Fill(v_DeltaphiBefore,fCentrality);\r
-\r
- Double_t v_DeltaphiAfter = v_phi - gReactionPlane;\r
- if(v_DeltaphiAfter < 0) v_DeltaphiAfter += 2*TMath::Pi();\r
- fHistPhiAfter->Fill(v_DeltaphiAfter,fCentrality);\r
- }\r
- \r
- v_phi *= TMath::RadToDeg();\r
-\r
- // fill charge vector\r
- chargeVector[0]->push_back(v_charge);\r
- chargeVector[1]->push_back(v_y);\r
- chargeVector[2]->push_back(v_eta);\r
- chargeVector[3]->push_back(v_phi);\r
- chargeVector[4]->push_back(v_p[0]);\r
- chargeVector[5]->push_back(v_p[1]);\r
- chargeVector[6]->push_back(v_p[2]);\r
- chargeVector[7]->push_back(v_pt);\r
- chargeVector[8]->push_back(v_E);\r
- \r
- if(fRunShuffling) {\r
- chargeVectorShuffle[0]->push_back(v_charge);\r
- chargeVectorShuffle[1]->push_back(v_y);\r
- chargeVectorShuffle[2]->push_back(v_eta);\r
- chargeVectorShuffle[3]->push_back(v_phi);\r
- chargeVectorShuffle[4]->push_back(v_p[0]);\r
- chargeVectorShuffle[5]->push_back(v_p[1]);\r
- chargeVectorShuffle[6]->push_back(v_p[2]);\r
- chargeVectorShuffle[7]->push_back(v_pt);\r
- chargeVectorShuffle[8]->push_back(v_E);\r
}\r
- gNumberOfAcceptedTracks += 1;\r
- \r
- } //track loop\r
- gReactionPlane *= TMath::RadToDeg();\r
- }//Vz cut\r
- }//Vy cut\r
- }//Vx cut\r
- }//MC analysis\r
- \r
- //multiplicity cut (used in pp)\r
- if(fUseMultiplicity) {\r
- if((gNumberOfAcceptedTracks < fNumberOfAcceptedTracksMin)||(gNumberOfAcceptedTracks > fNumberOfAcceptedTracksMax))\r
- return;\r
- }\r
- fHistNumberOfAcceptedTracks->Fill(gNumberOfAcceptedTracks,fCentrality);\r
+ }\r
+ }\r
+ \r
+ //Exclude from the analysis decay products of rho0, rho+, eta, eta' and phi\r
+ if(kExcludeParticle) continue;\r
+ }\r
+ \r
+ vCharge = track->Charge();\r
+ vPhi = track->Phi();\r
+ //Printf("phi (before): %lf",vPhi);\r
+ \r
+ fHistPt->Fill(vPt,gCentrality);\r
+ fHistEta->Fill(vEta,gCentrality);\r
+ fHistPhi->Fill(vPhi,gCentrality);\r
+ //fHistPhi->Fill(vPhi*TMath::RadToDeg(),gCentrality);\r
+ fHistRapidity->Fill(vY,gCentrality);\r
+ //if(vCharge > 0) fHistPhiPos->Fill(vPhi*TMath::RadToDeg(),gCentrality);\r
+ //else if(vCharge < 0) fHistPhiNeg->Fill(vPhi*TMath::RadToDeg(),gCentrality);\r
+ if(vCharge > 0) fHistPhiPos->Fill(vPhi,gCentrality);\r
+ else if(vCharge < 0) fHistPhiNeg->Fill(vPhi,gCentrality);\r
+ \r
+ //Flow after burner\r
+ if(fUseFlowAfterBurner) {\r
+ Double_t precisionPhi = 0.001;\r
+ Int_t maxNumberOfIterations = 100;\r
+ \r
+ Double_t phi0 = vPhi;\r
+ Double_t gV2 = fDifferentialV2->Eval(vPt);\r
+ \r
+ for (Int_t j = 0; j < maxNumberOfIterations; j++) {\r
+ Double_t phiprev = vPhi;\r
+ Double_t fl = vPhi - phi0 + gV2*TMath::Sin(2.*(vPhi - gReactionPlane*TMath::DegToRad()));\r
+ Double_t fp = 1.0 + 2.0*gV2*TMath::Cos(2.*(vPhi - gReactionPlane*TMath::DegToRad())); \r
+ vPhi -= fl/fp;\r
+ if (TMath::AreEqualAbs(phiprev,vPhi,precisionPhi)) break;\r
+ }\r
+ //Printf("phi (after): %lf\n",vPhi);\r
+ Double_t vDeltaphiBefore = phi0 - gReactionPlane*TMath::DegToRad();\r
+ if(vDeltaphiBefore < 0) vDeltaphiBefore += 2*TMath::Pi();\r
+ fHistPhiBefore->Fill(vDeltaphiBefore,gCentrality);\r
+ \r
+ Double_t vDeltaphiAfter = vPhi - gReactionPlane*TMath::DegToRad();\r
+ if(vDeltaphiAfter < 0) vDeltaphiAfter += 2*TMath::Pi();\r
+ fHistPhiAfter->Fill(vDeltaphiAfter,gCentrality);\r
+ \r
+ }\r
+ \r
+ //vPhi *= TMath::RadToDeg();\r
+\r
+ //=======================================correction\r
+ Double_t correction = GetTrackbyTrackCorrectionMatrix(vEta, vPhi, vPt, vCharge, gCentrality); \r
+ \r
+ tracksAccepted->Add(new AliBFBasicParticle(vEta, vPhi, vPt, vCharge, correction)); \r
+ \r
+ } //track loop\r
+ }//MC\r
\r
- // calculate balance function\r
- if(fUseMultiplicity) \r
- //fBalance->CalculateBalance(gNumberOfAcceptedTracks,gReactionPlane,chargeVector);\r
- fBalance->CalculateBalance(gReactionPlane,chargeVector);\r
- else \r
- fBalance->CalculateBalance(gReactionPlane,chargeVector);\r
+ return tracksAccepted; \r
+}\r
+//________________________________________________________________________\r
+TObjArray* AliAnalysisTaskBFPsi::GetShuffledTracks(TObjArray *tracks, Double_t gCentrality){\r
+ // Clones TObjArray and returns it with tracks after shuffling the charges\r
\r
- if(fRunShuffling) {\r
- // shuffle charges\r
- random_shuffle( chargeVectorShuffle[0]->begin(), chargeVectorShuffle[0]->end() );\r
- if(fUseMultiplicity) \r
- fShuffledBalance->CalculateBalance(gReactionPlane,chargeVectorShuffle);\r
- else \r
- fShuffledBalance->CalculateBalance(gReactionPlane,chargeVectorShuffle);\r
+ TObjArray* tracksShuffled = new TObjArray;\r
+ tracksShuffled->SetOwner(kTRUE);\r
+\r
+ vector<Short_t> *chargeVector = new vector<Short_t>; //original charge of accepted tracks \r
+\r
+ for (Int_t i=0; i<tracks->GetEntriesFast(); i++)\r
+ {\r
+ AliVParticle* track = (AliVParticle*) tracks->At(i);\r
+ chargeVector->push_back(track->Charge());\r
+ } \r
+ \r
+ random_shuffle(chargeVector->begin(), chargeVector->end());\r
+ \r
+ for(Int_t i = 0; i < tracks->GetEntriesFast(); i++){\r
+ AliVParticle* track = (AliVParticle*) tracks->At(i);\r
+ //==============================correction\r
+ Double_t correction = GetTrackbyTrackCorrectionMatrix(track->Eta(), track->Phi(),track->Pt(), chargeVector->at(i), gCentrality);\r
+ tracksShuffled->Add(new AliBFBasicParticle(track->Eta(), track->Phi(), track->Pt(),chargeVector->at(i), correction));\r
}\r
-} \r
+\r
+ delete chargeVector;\r
+ \r
+ return tracksShuffled;\r
+}\r
+\r
\r
//________________________________________________________________________\r
void AliAnalysisTaskBFPsi::FinishTaskOutput(){\r
//Printf("END BF");\r
\r
if (!fBalance) {\r
- Printf("ERROR: fBalance not available");\r
+ AliError("fBalance not available");\r
return;\r
} \r
if(fRunShuffling) {\r
if (!fShuffledBalance) {\r
- Printf("ERROR: fShuffledBalance not available");\r
+ AliError("fShuffledBalance not available");\r
return;\r
}\r
}\r