+/**************************************************************************\r
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
+ * *\r
+ * Author: The ALICE Off-line Project. *\r
+ * Contributors are mentioned in the code where appropriate. *\r
+ * *\r
+ * Permission to use, copy, modify and distribute this software and its *\r
+ * documentation strictly for non-commercial purposes is hereby granted *\r
+ * without fee, provided that the above copyright notice appears in all *\r
+ * copies and that both the copyright notice and this permission notice *\r
+ * appear in the supporting documentation. The authors make no claims *\r
+ * about the suitability of this software for any purpose. It is *\r
+ * provided "as is" without express or implied warranty. *\r
+ **************************************************************************/\r
+//------------------------------------------------------------------------------\r
+// AliPtResolAnalysisPbPb class. \r
+// \r
+// a. functionality:\r
+// - fills analysis control histograms\r
+//\r
+// b. data members:\r
+// - control histograms\r
+//\r
+// Author: J.Otwinowski 04/11/2008 \r
+//------------------------------------------------------------------------------\r
+\r
+#include "TH1.h"\r
+#include "TH2.h"\r
+#include "TCanvas.h"\r
+#include "THnSparse.h"\r
+\r
+#include "AliHeader.h" \r
+#include "AliInputEventHandler.h" \r
+#include "AliAnalysisManager.h" \r
+#include "AliGenEventHeader.h" \r
+#include "AliStack.h" \r
+#include "AliESDEvent.h" \r
+#include "AliMCEvent.h" \r
+#include "AliESDtrackCuts.h" \r
+#include "AliLog.h" \r
+#include "AliMultiplicity.h"\r
+#include "AliTracker.h"\r
+#include "AliCentrality.h"\r
+\r
+#include "AlidNdPtEventCuts.h"\r
+#include "AlidNdPtAcceptanceCuts.h"\r
+#include "AliPhysicsSelection.h"\r
+#include "AliTriggerAnalysis.h"\r
+\r
+#include "AliPWG0Helper.h"\r
+#include "AlidNdPtHelper.h"\r
+#include "AliPtResolAnalysisPbPb.h"\r
+\r
+\r
+using namespace std;\r
+\r
+ClassImp(AliPtResolAnalysisPbPb)\r
+\r
+//_____________________________________________________________________________\r
+ AliPtResolAnalysisPbPb::AliPtResolAnalysisPbPb(): AlidNdPt(),\r
+ fAnalysisFolder(0),\r
+ fTrackParamHist(0),\r
+ fCentralityEstimator(0)\r
+{\r
+ // default constructor\r
+ Init();\r
+}\r
+\r
+//_____________________________________________________________________________\r
+AliPtResolAnalysisPbPb::AliPtResolAnalysisPbPb(Char_t* name, Char_t* title): AlidNdPt(name,title),\r
+ fAnalysisFolder(0),\r
+ fTrackParamHist(0),\r
+ fCentralityEstimator(0)\r
+{\r
+ Init();\r
+}\r
+\r
+//_____________________________________________________________________________\r
+AliPtResolAnalysisPbPb::~AliPtResolAnalysisPbPb() {\r
+ //\r
+ // destructor\r
+ //\r
+ if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;\r
+ if(fTrackParamHist) delete fTrackParamHist; fTrackParamHist=0;\r
+}\r
+\r
+//_____________________________________________________________________________\r
+void AliPtResolAnalysisPbPb::Init(){\r
+ //\r
+ // Generic histograms to be corrected\r
+ //\r
+ //1/pT:#sigma(1/pT):centrality\r
+ Int_t binsTrackParamHist[3]={400,300,11};\r
+ Double_t minTrackParamHist[3]={0,0,0}; \r
+ Double_t maxTrackParamHist[3]={1,0.015,100};\r
+\r
+ Double_t centrBins[12] = {0.0,5.,10.,20.,30.,40.,50.,60.,70.,80.,90.,100}; \r
+\r
+\r
+ fTrackParamHist = new THnSparseF("fTrackParamHist","1/pT:#sigma(1/pT):centrality",3,binsTrackParamHist,minTrackParamHist,maxTrackParamHist);\r
+ fTrackParamHist->SetBinEdges(2,centrBins);\r
+ fTrackParamHist->GetAxis(0)->SetTitle("1/pT (GeV/c)^{-1}");\r
+ fTrackParamHist->GetAxis(1)->SetTitle("#sigma(1/pT)" );\r
+ fTrackParamHist->GetAxis(2)->SetTitle("centrality" );\r
+ fTrackParamHist->Sumw2();\r
+ \r
+ // init folder\r
+ fAnalysisFolder = CreateFolder("folderdNdPt","Analysis dNdPt Folder");\r
+ \r
+}\r
+\r
+//_____________________________________________________________________________\r
+void AliPtResolAnalysisPbPb::Process(AliESDEvent *const esdEvent, AliMCEvent *const mcEvent)\r
+{\r
+ //\r
+ // Process real and/or simulated events\r
+ //\r
+ if(!esdEvent) {\r
+ AliDebug(AliLog::kError, "esdEvent not available");\r
+ return;\r
+ }\r
+\r
+ // get selection cuts\r
+ AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
+ AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
+ AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
+\r
+ if(!evtCuts || !accCuts || !esdTrackCuts) {\r
+ AliDebug(AliLog::kError, "cuts not available");\r
+ return;\r
+ }\r
+\r
+ // trigger selection\r
+ Bool_t isEventTriggered = kTRUE;\r
+ AliPhysicsSelection *physicsSelection = NULL;\r
+ AliTriggerAnalysis* triggerAnalysis = NULL;\r
+\r
+ // \r
+ AliInputEventHandler* inputHandler = (AliInputEventHandler*) AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler();\r
+ if (!inputHandler)\r
+ {\r
+ Printf("ERROR: Could not receive input handler");\r
+ return;\r
+ }\r
+\r
+ if(evtCuts->IsTriggerRequired()) \r
+ {\r
+ // always MB\r
+ isEventTriggered = inputHandler->IsEventSelected() & AliVEvent::kMB;\r
+\r
+ physicsSelection = static_cast<AliPhysicsSelection*> (inputHandler->GetEventSelection());\r
+ if(!physicsSelection) return;\r
+ //SetPhysicsTriggerSelection(physicsSelection);\r
+\r
+ if (isEventTriggered && (GetTrigger() == AliTriggerAnalysis::kV0AND)) {\r
+ // set trigger (V0AND)\r
+ triggerAnalysis = physicsSelection->GetTriggerAnalysis();\r
+ if(!triggerAnalysis) return;\r
+ isEventTriggered = triggerAnalysis->IsOfflineTriggerFired(esdEvent, GetTrigger());\r
+ }\r
+\r
+\r
+ // centrality determination\r
+ Float_t centralityF = -1.;\r
+ AliCentrality *esdCentrality = esdEvent->GetCentrality();\r
+ centralityF = esdCentrality->GetCentralityPercentile(fCentralityEstimator.Data());\r
+\r
+ // get reconstructed vertex \r
+ const AliESDVertex* vtxESD = 0; \r
+ Bool_t isRecVertex = kFALSE;\r
+ if(evtCuts->IsRecVertexRequired()) \r
+ {\r
+ Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r
+ Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r
+ vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r
+ isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, esdEvent->GetPrimaryVertexSPD(), GetAnalysisMode(), kFALSE);\r
+ }\r
+\r
+ Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex; \r
+ //printf("isEventOK %d, isEventTriggered %d \n",isEventOK, isEventTriggered);\r
+ //printf("GetAnalysisMode() %d \n",GetAnalysisMode());\r
+\r
+ //\r
+ TObjArray *allChargedTracks=0;\r
+ // check event cuts\r
+ if(isEventOK && isEventTriggered)\r
+ {\r
+ // get all charged tracks\r
+ allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r
+ if(!allChargedTracks) return;\r
+\r
+ Int_t entries = allChargedTracks->GetEntries();\r
+ //printf("entries %d \n",entries);\r
+\r
+ // fill histograms\r
+ for(Int_t i=0; i<entries;++i) \r
+ {\r
+ AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r
+ if(!track) continue;\r
+ if(track->Charge()==0) continue;\r
+\r
+ // only postive charged \r
+ if(GetParticleMode() == AlidNdPtHelper::kPlus && track->Charge() < 0) \r
+ continue;\r
+ \r
+ // only negative charged \r
+ if(GetParticleMode() == AlidNdPtHelper::kMinus && track->Charge() > 0) \r
+ continue;\r
+\r
+ if(esdTrackCuts->AcceptTrack(track)) \r
+ {\r
+ if(accCuts->AcceptTrack(track)) \r
+ {\r
+ //Double_t x, par[5], cov[15];\r
+ //track->GetExternalParameters(x, p);\r
+ //track->GetExternalCovariance(cov);\r
+\r
+ Double_t v[3] = {track->OneOverPt(),TMath::Sqrt(track->GetSigma1Pt2()),centralityF};\r
+ fTrackParamHist->Fill(v);\r
+ }\r
+ } \r
+ }\r
+ }\r
+ }\r
+}\r
+\r
+//_____________________________________________________________________________\r
+Long64_t AliPtResolAnalysisPbPb::Merge(TCollection* const list) \r
+{\r
+ // Merge list of objects (needed by PROOF)\r
+\r
+ if (!list)\r
+ return 0;\r
+\r
+ if (list->IsEmpty())\r
+ return 1;\r
+\r
+ TIterator* iter = list->MakeIterator();\r
+ TObject* obj = 0;\r
+\r
+ //\r
+ //TList *collPhysSelection = new TList;\r
+\r
+ // collection of generated histograms\r
+\r
+ Int_t count=0;\r
+ while((obj = iter->Next()) != 0) {\r
+ AliPtResolAnalysisPbPb* entry = dynamic_cast<AliPtResolAnalysisPbPb*>(obj);\r
+ if (entry == 0) continue; \r
+ \r
+ //\r
+ fTrackParamHist->Add(entry->fTrackParamHist);\r
+ }\r
+\r
+return count;\r
+}\r
+\r
+//_____________________________________________________________________________\r
+void AliPtResolAnalysisPbPb::Analyse() \r
+{\r
+ // Analyse histograms\r
+ //\r
+ TH1::AddDirectory(kFALSE);\r
+ TObjArray *aFolderObj = new TObjArray;\r
+ \r
+ //\r
+ // Reconstructed event vertex\r
+ //\r
+ \r
+ // export objects to analysis folder\r
+ fAnalysisFolder = ExportToFolder(aFolderObj);\r
+\r
+ // delete only TObjArray\r
+ if(aFolderObj) delete aFolderObj;\r
+}\r
+\r
+//_____________________________________________________________________________\r
+TFolder* AliPtResolAnalysisPbPb::ExportToFolder(TObjArray * const array) \r
+{\r
+ // recreate folder avery time and export objects to new one\r
+ //\r
+ AliPtResolAnalysisPbPb * comp=this;\r
+ TFolder *folder = comp->GetAnalysisFolder();\r
+\r
+ TString name, title;\r
+ TFolder *newFolder = 0;\r
+ Int_t i = 0;\r
+ Int_t size = array->GetSize();\r
+\r
+ if(folder) { \r
+ // get name and title from old folder\r
+ name = folder->GetName(); \r
+ title = folder->GetTitle(); \r
+\r
+ // delete old one\r
+ delete folder;\r
+\r
+ // create new one\r
+ newFolder = CreateFolder(name.Data(),title.Data());\r
+ newFolder->SetOwner();\r
+\r
+ // add objects to folder\r
+ while(i < size) {\r
+ newFolder->Add(array->At(i));\r
+ i++;\r
+ }\r
+ }\r
+\r
+return newFolder;\r
+}\r
+\r
+//_____________________________________________________________________________\r
+TFolder* AliPtResolAnalysisPbPb::CreateFolder(TString name,TString title) { \r
+// create folder for analysed histograms\r
+//\r
+TFolder *folder = 0;\r
+ folder = new TFolder(name.Data(),title.Data());\r
+\r
+ return folder;\r
+}\r