]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - HBTAN/AliHBTAnalysis.h
changes for proper protection against failed retrieval of CDB Reco object (moved...
[u/mrichter/AliRoot.git] / HBTAN / AliHBTAnalysis.h
index 7cfa4d2ca55ed4f6bd39efd73d42fe84e59d5648..7f0782738e58ca2890cb019a69cd440c6d0b7747 100644 (file)
@@ -7,7 +7,7 @@
 //
 // Central Object Of HBTAnalyser: 
 // This class performs main looping within HBT Analysis
-// User must plug a reader of Type AliHBTReader
+// User must plug a reader of Type AliReader
 // User plugs in coorelation and monitor functions
 // as well as monitor functions
 //
 // ALICE Off-Line framework:
 //
 // Piotr.Skowronski@cern.ch
-// more info: http://alisoft.cern.ch/people/skowron/analyzer/index.html
+// more info: http://aliweb.cern.ch/people/skowron/analyzer/index.html
 //
 ////////////////////////////////////////////////////////////////////////////
 //_________________________________________________________
 
-#include <TObject.h>
-#include "AliHBTPairCut.h"
-#include "AliHBTParticleCut.h"
+#include <AliAnalysis.h>
 
 class AliHBTCut;
-class AliHBTPair;
+//class AliHBTPair;
 
 class AliHBTRun;
-class AliHBTEvent;
-class AliHBTReader;
+class AliAOD;
+class AliReader;
+class AliEventBuffer;
+class AliEventCut;
+
 class AliHBTOnePairFctn;      
 class AliHBTTwoPairFctn;
 
@@ -38,7 +39,7 @@ class AliHBTMonTwoParticleFctn;
 
 class TList;
 
-class AliHBTAnalysis: public TObject
+class AliHBTAnalysis: public AliAnalysis
  {
    public:
      AliHBTAnalysis();
@@ -46,55 +47,80 @@ class AliHBTAnalysis: public TObject
      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   AddTrackFunction(AliHBTOnePairFctn* f);
+     void   AddParticleFunction(AliHBTOnePairFctn* f);
+     void   AddParticleAndTrackFunction(AliHBTTwoPairFctn* f);
      
-     void SetReader(AliHBTReader* r){fReader = r;}
+     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 WriteFunctions();
+     void   SetReader(AliReader* r){fReader = r;}
      
-     void SetBufferSize(Int_t buffsize){fBufferSize=buffsize;}
-     void SetOwner(Bool_t owner=kTRUE){fIsOwner=owner;}
+     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   SetCutsOnParticles(); // -- aplies only to Process Tracks And Particles
-     void   SetCutsOnTracks();// -- aplies only to Process Tracks And Particles
-     void   SetCutsOnTracksAndParticles();// Default // -- aplies only to Process Tracks And Particles
+     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);
+     
+     virtual Int_t ProcessRecAndSimNonId(AliAOD* aodrec, AliAOD* aodsim);
+     virtual Int_t ProcessSimNonId(AliAOD* /*aodrec*/, AliAOD* aodsim);
+     virtual Int_t ProcessRecNonId(AliAOD* aodrec, AliAOD* /*aodsim*/);
      
-     void FilterOut(AliHBTEvent* outpart1, AliHBTEvent* outpart2, AliHBTEvent* inpart,
-                    AliHBTEvent* outtrack1, AliHBTEvent* outtrack2, AliHBTEvent* intrack)const;
-     void FilterOut(AliHBTEvent* out1, AliHBTEvent* out2, AliHBTEvent* in)const;
-     void DeleteFunctions();
+
+     // OLD legacy schema
+
+     void   ProcessTracks();
+     void   ProcessParticles();
+     void   ProcessTracksAndParticles();
+     
+     void   ProcessTracksAndParticlesNonIdentAnal();
+     void   ProcessParticlesNonIdentAnal();
+     void   ProcessTracksNonIdentAnal();
+
+     Bool_t RunCoherencyCheck();
      
-     virtual void ProcessTracks();
-     virtual void ProcessParticles();
-     virtual void ProcessTracksAndParticles();
+     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();
      
-     virtual void ProcessTracksAndParticlesNonIdentAnal();
-     virtual void ProcessParticlesNonIdentAnal();
-     virtual void ProcessTracksNonIdentAnal();
 
-     AliHBTReader* fReader;//! Pointer to reader
+     AliReader* fReader;//! Pointer to reader
      
      UInt_t fNTrackFunctions; //! Number of Tracks functions 
      UInt_t fNParticleFunctions; //! Number of particles functions
@@ -113,37 +139,28 @@ class AliHBTAnalysis: public TObject
      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:
-     Bool_t (AliHBTAnalysis::*fkPass)(AliHBTPair* partpair, AliHBTPair* trackpair) const;//Pointer to function that performes pair cut
-     Bool_t (AliHBTAnalysis::*fkPass1)(AliHBTParticle* partpair, AliHBTParticle* trackpair) const;//Pointer to function that performes cut on first particle
-     Bool_t (AliHBTAnalysis::*fkPass2)(AliHBTParticle* partpair, AliHBTParticle* trackpair) const;//Pointer to function that performes cut on second particle
-     Bool_t (AliHBTAnalysis::*fkPassPairProp)(AliHBTPair* partpair, AliHBTPair* trackpair) const;//Pointer to function that performes pair cut
-     
-     Bool_t PassPartAndTrack (AliHBTPair* partpair, AliHBTPair* trackpair) const {return (fPairCut->Pass(partpair))?kTRUE:fPairCut->Pass(trackpair);}
-     Bool_t PassPartAndTrack1(AliHBTParticle* part, AliHBTParticle* track) const;
-     Bool_t PassPartAndTrack2(AliHBTParticle* part, AliHBTParticle* track) const;
-     Bool_t PassPairPropPartAndTrack (AliHBTPair* partpair, AliHBTPair* trackpair) const {return (fPairCut->PassPairProp(partpair))?kTRUE:fPairCut->PassPairProp(trackpair);}
-     
-     Bool_t PassPart (AliHBTPair* partpair, AliHBTPair* /*trackpair*/) const{return fPairCut->Pass(partpair);}
-     Bool_t PassPart1(AliHBTParticle* part, AliHBTParticle* /*track*/) const{return fPairCut->GetFirstPartCut()->Pass(part);}
-     Bool_t PassPart2(AliHBTParticle* part, AliHBTParticle* /*track*/) const{return fPairCut->GetSecondPartCut()->Pass(part);}
-     Bool_t PassPairPropPart (AliHBTPair* partpair, AliHBTPair* /*trackpair*/) const{return fPairCut->PassPairProp(partpair);}
-     
-     Bool_t PassTrack (AliHBTPair* /*partpair*/, AliHBTPair* trackpair) const{return fPairCut->Pass(trackpair);}
-     Bool_t PassTrack1(AliHBTParticle* /*part*/, AliHBTParticle* track) const{return fPairCut->GetFirstPartCut()->Pass(track);}
-     Bool_t PassTrack2(AliHBTParticle* /*part*/, AliHBTParticle* track) const{return fPairCut->GetSecondPartCut()->Pass(track);}
-     Bool_t PassPairPropTrack (AliHBTPair* /*partpair*/, AliHBTPair* trackpair) const{return fPairCut->PassPairProp(trackpair);}
 
      static const UInt_t fgkFctnArraySize;//!
      static const UInt_t fgkDefaultMixingInfo;//!
@@ -151,20 +168,4 @@ class AliHBTAnalysis: public TObject
 
      ClassDef(AliHBTAnalysis,0)
  };
-inline Bool_t AliHBTAnalysis::PassPartAndTrack1(AliHBTParticle* part,AliHBTParticle* track) const
-{
-//Checks first particle from both, particle and track pairs
-  AliHBTParticleCut* pc = fPairCut->GetFirstPartCut();
-  return (pc->Pass(part))?kTRUE:pc->Pass(track);
-}
-/*************************************************************************************/ 
-inline Bool_t AliHBTAnalysis::PassPartAndTrack2(AliHBTParticle* part,AliHBTParticle* track) const
-{
-//Checks second particle from both, particle and track pairs
-  AliHBTParticleCut* pc = fPairCut->GetSecondPartCut();
-  return (pc->Pass(part))?kTRUE:pc->Pass(track);
-}
-/*************************************************************************************/ 
 #endif