#ifndef ALIHBTANALYSIS_H
#define ALIHBTANALYSIS_H
+//_________________________________________________________
+////////////////////////////////////////////////////////////////////////////
+//
+// class AliHBTAnalysis
+//
+// Central Object Of HBTAnalyser:
+// This class performs main looping within HBT Analysis
+// User must plug a reader of Type AliReader
+// User plugs in coorelation and monitor functions
+// as well as monitor functions
+//
+// HBT Analysis Tool, which is integral part of AliRoot,
+// ALICE Off-Line framework:
+//
+// Piotr.Skowronski@cern.ch
+// more info: http://aliweb.cern.ch/people/skowron/analyzer/index.html
+//
+////////////////////////////////////////////////////////////////////////////
+//_________________________________________________________
+
+#include <AliAnalysis.h>
-#include <TObject.h>
-
-class AliHBTParticleCut;
class AliHBTCut;
-class AliHBTPairCut;
-class AliHBTPair;
+//class AliHBTPair;
class AliHBTRun;
-class AliHBTEvent;
-class AliHBTReader;
+class AliAOD;
+class AliReader;
+class AliEventBuffer;
+class AliEventCut;
+
class AliHBTOnePairFctn;
class AliHBTTwoPairFctn;
class TList;
-//____________________
-//Central Object of HBT Analyser
-//Performs looping over events/particles
-
-class AliHBTAnalysis: public TObject
+class AliHBTAnalysis: public AliAnalysis
{
public:
AliHBTAnalysis();
AliHBTAnalysis(const AliHBTAnalysis& in);
- const AliHBTAnalysis& operator=(const AliHBTAnalysis& right);
+ AliHBTAnalysis& operator=(const AliHBTAnalysis& /*right*/);
virtual ~AliHBTAnalysis();
- virtual void Process(Option_t* option = "TracksAndParticles");
+ Int_t Init();
+ Int_t ProcessEvent(AliAOD* aodrec, AliAOD* aodsim = 0x0);
+ Int_t Finish();
- void SetGlobalPairCut(AliHBTPairCut* cut);
+ enum EProcessOption{kReconstructed,kSimulated,kSimulatedAndReconstructed};
+ void SetProcessOption(EProcessOption option){fProcessOption = option;}//Init must be called to make effect
- void AddTrackFunction(AliHBTOnePairFctn* f);
- void AddParticleFunction(AliHBTOnePairFctn* f);
- void AddParticleAndTrackFunction(AliHBTTwoPairFctn* f);
+ void Process(Option_t* option = "TracksAndParticles");//Stand alone HBT analysis
- void AddParticleMonitorFunction(AliHBTMonOneParticleFctn* f); //z.ch.
- void AddTrackMonitorFunction(AliHBTMonOneParticleFctn* f); //z.ch.
- void AddParticleAndTrackMonitorFunction(AliHBTMonTwoParticleFctn* f);//z.ch.
-
- void AddResolutionFunction(AliHBTTwoPairFctn* f){AddParticleAndTrackFunction(f);}
+ void SetGlobalPairCut(AliAODPairCut* cut);
- void SetReader(AliHBTReader* r){fReader = r;}
+ void AddTrackFunction(AliHBTOnePairFctn* f);
+ void AddParticleFunction(AliHBTOnePairFctn* f);
+ void AddParticleAndTrackFunction(AliHBTTwoPairFctn* f);
- void WriteFunctions();
+ void AddParticleMonitorFunction(AliHBTMonOneParticleFctn* f); //z.ch.
+ void AddTrackMonitorFunction(AliHBTMonOneParticleFctn* f); //z.ch.
+ void AddParticleAndTrackMonitorFunction(AliHBTMonTwoParticleFctn* f);//z.ch.
+
+ void AddResolutionFunction(AliHBTTwoPairFctn* f){AddParticleAndTrackFunction(f);}
- void SetBufferSize(Int_t buffsize){fBufferSize=buffsize;}
- void SetOwner(Bool_t owner=kTRUE){fIsOwner=owner;}
+ void SetReader(AliReader* r){fReader = r;}
+
+ void WriteFunctions();
+ void SetOutputFileName(const char* fname);
+
+ void SetBufferSize(Int_t buffsize){fBufferSize=buffsize;}
+ void SetOwner(Bool_t owner=kTRUE){fIsOwner=owner;}
Bool_t IsOwner() const {return fIsOwner;}
Bool_t IsNonIdentAnalysis();
- void Init();
void ResetFunctions();
void SetDisplayInfo(Int_t howoften){fDisplayMixingInfo = howoften;}//defines every each line info about mixing is displayed
+
+ void SetApparentVertex(Double_t x, Double_t y, Double_t z);//Sets apparent vertex
+
+ static void PressAnyKey();//small utility function that helps to make comfortable macros
+
+
protected:
- Bool_t RunCoherencyCheck();
+ /**********************************************/
+ /* E V E N T P R O C E S S I N G */
+ /**********************************************/
+ // NEW AOD schema
+ Int_t (AliHBTAnalysis::*fProcEvent)(AliAOD* aodrec, AliAOD* aodsim);//Pointer to the processing method
+
+ virtual Int_t ProcessSim(AliAOD* /*aodrec*/, AliAOD* aodsim);
+ virtual Int_t ProcessRec(AliAOD* aodrec, AliAOD* /*aodsim*/);
+ virtual Int_t ProcessRecAndSim(AliAOD* aodrec, AliAOD* aodsim);
- void FilterOut(AliHBTEvent* outpart1, AliHBTEvent* outpart2, AliHBTEvent* inpart,
- AliHBTEvent* outtrack1, AliHBTEvent* outtrack2, AliHBTEvent* intrack);
- void FilterOut(AliHBTEvent* out1, AliHBTEvent* out2, AliHBTEvent* in);
- void DeleteFunctions();
+ virtual Int_t ProcessRecAndSimNonId(AliAOD* aodrec, AliAOD* aodsim);
+ virtual Int_t ProcessSimNonId(AliAOD* /*aodrec*/, AliAOD* aodsim);
+ virtual Int_t ProcessRecNonId(AliAOD* aodrec, AliAOD* /*aodsim*/);
- virtual void ProcessTracks();
- virtual void ProcessParticles();
- virtual void ProcessTracksAndParticles();
+
+ // OLD legacy schema
+
+ void ProcessTracks();
+ void ProcessParticles();
+ void ProcessTracksAndParticles();
- virtual void ProcessTracksAndParticlesNonIdentAnal();
- virtual void ProcessParticlesNonIdentAnal();
- virtual void ProcessTracksNonIdentAnal();
+ void ProcessTracksAndParticlesNonIdentAnal();
+ void ProcessParticlesNonIdentAnal();
+ void ProcessTracksNonIdentAnal();
- AliHBTReader* fReader;//! Pointer to reader
+ Bool_t RunCoherencyCheck();
+
+ void FilterOut(AliAOD* outpart1, AliAOD* outpart2, AliAOD* inpart,
+ AliAOD* outtrack1, AliAOD* outtrack2, AliAOD* intrack)const;
+ void FilterOut(AliAOD* out1, AliAOD* out2, AliAOD* in)const;
+ void DeleteFunctions();
+
+
+ AliReader* fReader;//! Pointer to reader
UInt_t fNTrackFunctions; //! Number of Tracks functions
UInt_t fNParticleFunctions; //! Number of particles functions
AliHBTMonOneParticleFctn** fTrackMonitorFunctions; //! which are used for single particle analysis,
AliHBTMonTwoParticleFctn** fParticleAndTrackMonitorFunctions; //! cut monitoring, etc.
+ AliEventCut* fBkgEventCut;// We can narrow class of events used in
+
+ AliEventBuffer* fPartBuffer;//Sim Particles event buffer
+ AliEventBuffer* fTrackBuffer;//Rec Tracks event buffer
/**********************************************/
- /* Control parameters */
+ /* Control parameters */
/**********************************************/
-
- AliHBTPairCut *fPairCut;//! Pair cut applied for all mixed particles
- Int_t fBufferSize; //!defines the size of buffer for mixed events; -1==MIX All
- Int_t fDisplayMixingInfo;//!defines every which particle mixing info is displayed
- Bool_t fIsOwner;//!defines of all functions are supposed to be deleted while by the way of analysis defaulr false
-
+ Int_t fBufferSize; //defines the size of buffer for mixed events; -1==MIX All
+ Int_t fDisplayMixingInfo;//!defines every which particle mixing info is displayed
+ Bool_t fIsOwner;//!defines of all functions are supposed to be deleted while by the way of analysis defaulr false
+
+ EProcessOption fProcessOption;//Option that says waht analysis to do (Rec, Sim or SimAndRec)
+ Bool_t fNoCorrfctns;//Internal flag indicating that no cfs are set by the user (only monitor ones)
+ TString* fOutputFileName;//Fiele name where to dump results, if not specified reults are written to gDirectory
+
+ Double_t fVertexX;//X position of apparent vertex
+ Double_t fVertexY;//Y position of apparent vertex
+ Double_t fVertexZ;//Z position of apparent vertex
+
private:
+
static const UInt_t fgkFctnArraySize;//!
static const UInt_t fgkDefaultMixingInfo;//!
static const Int_t fgkDefaultBufferSize;//!
- public:
+
ClassDef(AliHBTAnalysis,0)
};
-
-
-
-
#endif