+++ /dev/null
-/**************************************************************************
- * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
- * *
- * Author: The ALICE Off-line Project. *
- * Contributors are mentioned in the code where appropriate. *
- * *
- * Permission to use, copy, modify and distribute this software and its *
- * documentation strictly for non-commercial purposes is hereby granted *
- * without fee, provided that the above copyright notice appears in all *
- * copies and that both the copyright notice and this permission notice *
- * appear in the supporting documentation. The authors make no claims *
- * about the suitability of this software for any purpose. It is *
- * provided "as is" without express or implied warranty. *
- **************************************************************************/
-
-//-----------------------------------------------------------------------
-// Author : Marta Verweij, D. Keijdener - UU
-//-----------------------------------------------------------------------
-
-#ifndef ALIPWG4HIGHPTSPECTRA_H
-#define ALIPWG4HIGHPTSPECTRA_H
-
-#include "AliAnalysisTask.h"
-#include "AliCFManager.h"
-
-class TH1I;
-class TH1F;
-class TH1D;
-class TH2F;
-class TProfile;
-class TFile;
-class TList;
-
-//class AliCFManager;
-class AliESDtrackCuts;
-class AliESDEvent;
-class AliAODEvent;
-class AliVEvent;
-class AliVVertex;
-class AliMCEvent;
-class AliStack;
-class AliGenPythiaEventHeader;
-class AliGenHijingEventHeader;
-
-class AliPWG4HighPtSpectra : public AliAnalysisTask {
- public:
-
- enum {
- kStepReconstructed = 0,
- kStepSecondaries = 1,
- kStepReconstructedMC = 2,
- kStepMCAcceptance = 3
- };
-
- AliPWG4HighPtSpectra();
- AliPWG4HighPtSpectra(const Char_t* name);
- // AliPWG4HighPtSpectra& operator= (const AliPWG4HighPtSpectra& c);
- // AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra& c);
- ~AliPWG4HighPtSpectra() {;};
-
- // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
- virtual void LocalInit();
- virtual void ConnectInputData(Option_t *);
- virtual void CreateOutputObjects();
- virtual void Exec(Option_t *option);
- virtual void Terminate(Option_t *);
- virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
-
- Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data?
- Bool_t SelectEvent(); //decides if event is used for analysis
- Int_t CalculateCentrality(AliVEvent *event);
-
- //Setters
- void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;}
- void SetCentralityClass(int cent) {fCentClass=cent;}
- void SetTriggerMask(UInt_t t) {fTriggerMask=t;}
-
- // CORRECTION FRAMEWORK RELATED FUNCTIONS
- void SetCFManagerPos(const AliCFManager* io1) {fCFManagerPos = io1;} // global correction manager
- const AliCFManager * GetCFManagerPos() const {return fCFManagerPos;} // get corr manager
- void SetCFManagerNeg(const AliCFManager* io2) {fCFManagerNeg = io2;} // global correction manager
- const AliCFManager * GetCFManagerNeg() const {return fCFManagerNeg;} // get corr manager
-
- //if fTrackType=0 (Global)
- //if fTrackType=1 (TPConly)
- //if fTrackType=2 (TPConly constrained)
- void SetTrackType(Int_t trackType) {fTrackType = trackType;}
- //AliESDtrackCuts setters
- void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;} // Needs to be specified for ESD analysis, not for AOD analysis
- void SetFilterMask(Int_t filtermask) {fFilterMask = filtermask;} // Needs to be specified for AOD analysis, not for ESD analysis
- void SetCutsReject(AliESDtrackCuts* trackCuts) {fTrackCutsReject = trackCuts;}
- void SelectHIJINGOnly(Bool_t b) {fbSelectHIJING = b;}
-
- Bool_t IsHIJINGParticle(Int_t label);
-
- void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
-
- // Data types
- Bool_t IsReadAODData() const {return fReadAODData;}
- void SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;}
-
- Bool_t IsUsingPythiaInfo() const {return fNoPythiaInfo;}
- void SetNoPythiaInfo() {fNoPythiaInfo=kTRUE;}
-
- static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent);
- static AliGenHijingEventHeader* GetHijingEventHeader(AliMCEvent *mcEvent);
- static AliGenHijingEventHeader* GetHijingEventHeaderAOD(AliAODEvent *aodEvent);
-
- static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root
-
-
-
- protected:
- Bool_t fReadAODData ; // flag for AOD/ESD input files
- Bool_t fNoPythiaInfo ; // flag to skip reading pyxsec.root and plotting output
- const AliCFManager *fCFManagerPos ; // pointer to the CF manager for positive charged particles
- const AliCFManager *fCFManagerNeg ; // pointer to the CF manager for negative charged particles
-
- AliESDEvent *fESD; //! ESD object
- AliAODEvent *fAOD; //! AOD object
- AliMCEvent *fMC; //! MC event object, only used in ESD analysis
- AliStack *fStack; //! stack object, only used in ESD analysis
- TClonesArray *fArrayMCAOD; //! TClonesArray of AliAODMCParticles, only used in AOD analysis
-
- const AliVVertex *fVtx; //! vertex object
-
- UInt_t fTriggerMask; // Trigger mask to select events
- Bool_t fIsPbPb; // kTRUE if PbPb
- Int_t fCentClass; // Select only events from predefined centrality class
-
- Int_t fTrackType; // Type of track to be used in analysis
- //AliESDtrackCuts options and FilterMask. The former is must be setted in AddTaskPWG4HighPTSpectra.C for ESD analysis, the latter must be setted in AddTaskPWG4HighPTSpectra.C for AOD analysis. The AliESDtrackCuts correspond with different steps in container.
- AliESDtrackCuts *fTrackCuts; // trackCuts applied to global tracks
- AliESDtrackCuts *fTrackCutsReject; // trackCuts to reject tracks (hybrid case)
- Int_t fFilterMask; // Filtermask specifying track cuts. See https://twiki.cern.ch/twiki/bin/view/ALICE/PWGPPAODTrackCuts for values.
-
- Bool_t fbSelectHIJING; //Select only particles from HIJING event
-
- Double_t fSigmaConstrainedMax; // max sigma on constrained fit
-
- private:
- AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra&);
- AliPWG4HighPtSpectra& operator=(const AliPWG4HighPtSpectra&);
-
- Float_t fAvgTrials; // Average number of trials
-
- // Histograms
- //Number of events
- TList *fHistList; //! List of output histograms
- TH1F *fNEventAll; //! Event counter
- TH1F *fNEventSel; //! Event counter: Selected events for analysis
- TH1F *fNEventReject; //! Book keeping of reason of rejecting events
-
- TH1F *fh1Centrality; //! Centrality
-
- TProfile* fh1Xsec; //! pythia cross section and trials
- TH1F* fh1Trials; //! trials which are added
- TH1F* fh1PtHard; //! pt hard of the event
- TH1F* fh1PtHardTrials; //! pt hard of the event
-
- TH2F *fPtRelUncertainty1PtPrim; //! Pt vs relUncertainty1Pt for primary particles
- TH2F *fPtRelUncertainty1PtSec; //! Pt vs relUncertainty1Pt for secondary particles
-
- ClassDef(AliPWG4HighPtSpectra,5);
-};
-
-#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff