Anti-Splitting Cut implemented

[u/mrichter/AliRoot.git] / HBTAN / AliHBTAnalysis.h

diff --git a/HBTAN/AliHBTAnalysis.h b/HBTAN/AliHBTAnalysis.h
index ffda7dd70201355eb7953d3bc02b5e22111805c5..fd0a93d8a867c25387cdbb7900c1f232d34eec5d 100644 (file)
--- a/HBTAN/AliHBTAnalysis.h
+++ b/HBTAN/AliHBTAnalysis.h
@@ -21,10 +21,11 @@
 //_________________________________________________________
 
 #include <TObject.h>
+#include "AliHBTParticle.h"
+#include "AliHBTPairCut.h"
+#include "AliHBTParticleCut.h"
 
-class AliHBTParticleCut;
 class AliHBTCut;
-class AliHBTPairCut;
 class AliHBTPair;
 
 class AliHBTRun;
@@ -71,6 +72,12 @@ class AliHBTAnalysis: public TObject
      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
+     
+     static void PressAnyKey();//small utility function that helps to make comfortable macros
    protected:
      
      Bool_t RunCoherencyCheck();
@@ -112,17 +119,53 @@ class AliHBTAnalysis: public TObject
      /* Control parameters  */
      /**********************************************/
 
-     AliHBTPairCut *fPairCut;//! Pair cut applied for all mixed particles
+     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
 
    private:
+     Bool_t (AliHBTAnalysis::*fPass)(AliHBTPair* partpair, AliHBTPair* trackpair) const;//Pointer to function that performes pair cut
+     Bool_t (AliHBTAnalysis::*fPass1)(AliHBTParticle* partpair, AliHBTParticle* trackpair) const;//Pointer to function that performes cut on first particle
+     Bool_t (AliHBTAnalysis::*fPass2)(AliHBTParticle* partpair, AliHBTParticle* trackpair) const;//Pointer to function that performes cut on second particle
+     Bool_t (AliHBTAnalysis::*fPassPairProp)(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;//!
      static const Int_t  fgkDefaultBufferSize;//!
 
      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