ConvGamma preburner update

[u/mrichter/AliRoot.git] / PWG4 / GammaConv / AliAnalysisTaskGCPartToPWG4Part.h

diff --git a/PWG4/GammaConv/AliAnalysisTaskGCPartToPWG4Part.h b/PWG4/GammaConv/AliAnalysisTaskGCPartToPWG4Part.h
index fe30e153b21a4afe999e9828ed60db7cd04bd066..011262a3cd29d9f70b33c26383a5b1054ef35212 100644 (file)
--- a/PWG4/GammaConv/AliAnalysisTaskGCPartToPWG4Part.h
+++ b/PWG4/GammaConv/AliAnalysisTaskGCPartToPWG4Part.h
@@ -34,6 +34,10 @@ public:
 \r
   void SetDebugLevel(Int_t debugLevel) { fDebugLevel = debugLevel; }\r
   Int_t GetDebugLevel() const { return fDebugLevel; }\r
+\r
+  void SetGammaCutId(TString cut) { fGammaCutString = Form("GammaConv_%s", cut.Data());}\r
+  void SetPionCutId(TString cut) { fPionCutString = Form("GammaConv_%s", cut.Data());}\r
+\r
   \r
  private:\r
 \r
@@ -43,19 +47,28 @@ public:
   //Get the AOD event from whereever it might be accessible\r
   AliAODEvent * GetAODEvent();\r
 \r
+  Bool_t BothTracksPresent(const AliAODConversionParticle * const photon, const TClonesArray * const tracks) const;\r
+  Bool_t BothGammaPresent(const AliAODConversionParticle * const pion, const TClonesArray * const photons, const TClonesArray * const tracks) const;\r
+\r
   //Get Conversion gammas branch\r
-  TClonesArray * GetConversionGammas(const AliAODEvent * aodEvent);\r
+  TClonesArray * GetConversionGammas(const AliAODEvent * aodEvent) const;\r
+  TClonesArray * GetPions(const AliAODEvent * aodEvent) const;\r
+  TClonesArray * GetAODBranch(const AliAODEvent * aodEvent, TString branchName) const;\r
 \r
   //Fill AOD tree with PWG4 particles\r
   AliAODPWG4ParticleCorrelation * AddToAOD(AliGammaConversionAODObject * aodO, TClonesArray * branch, TString detector);\r
   AliAODPWG4ParticleCorrelation * AddToAOD(AliAODConversionParticle * aodO, TClonesArray * branch, TString detector);\r
-  \r
+  AliAODPWG4ParticleCorrelation * AddPionToAOD(AliAODConversionParticle * pion, TClonesArray * branch, TString detector, TClonesArray * photons);  \r
   //Process conv gamma\r
   void ProcessConvGamma( const AliAODEvent * const aodEvent );\r
 \r
   TString     fDeltaAODFileName;//! File where Gamma Conv AOD is located, if not in default AOD\r
+  TString     fGammaCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
+  TString     fPionCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
   TString     fAODBranchName;\r
-  TClonesArray * fAODPWG4Particles;\r
+  TClonesArray * fAODPWG4Photons;\r
+  TClonesArray * fAODPWG4Pi0;\r
+\r
 \r
   Int_t fDebugLevel;\r
 \r