Use SpecialOutput for the CF container (Chiara Z)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODRecoCascadeHF.h

diff --git a/PWG3/vertexingHF/AliAODRecoCascadeHF.h b/PWG3/vertexingHF/AliAODRecoCascadeHF.h
index c9e583195161a24a8a65fd282871623af0682945..508b6d3bddc0a0200a3303c2b9600a7d1970f24c 100644 (file)
--- a/PWG3/vertexingHF/AliAODRecoCascadeHF.h
+++ b/PWG3/vertexingHF/AliAODRecoCascadeHF.h
@@ -9,11 +9,12 @@
 // (D*->D0pi, ...)
 // The convention is: prong 0 is the bachelor, prong 1 is the "V0"
 //
-// Author: X-M. Zhang, zhangxm@ccnu.iop.edu.cn
+// Author: X-M. Zhang, zhangxm@iopp.ccnu.edu.cn
 //***********************************************************
 
 #include <TRef.h>
 #include <TRefArray.h>
+#include <TClonesArray.h>
 #include "AliAODVertex.h"
 #include "AliAODRecoDecayHF2Prong.h"
 
@@ -33,12 +34,7 @@ class AliAODRecoCascadeHF : public AliAODRecoDecayHF2Prong {
   AliAODRecoCascadeHF& operator=(const AliAODRecoCascadeHF& source);
 
   // 2prong (D0 for Dstar)
-  void Set2Prong(TObject *the2Prong) { f2Prong = the2Prong; }
-  void Check2Prong() const { if(!f2Prong.GetObject()) printf("f2Prong not set\n"); return; }
-  // GetDaughter(1) doesn't work because the V0 and the bachelor have different 
-  // process id and they can't go to the same TRefArray
-  AliAODRecoDecayHF2Prong* Get2Prong() const { if(GetDaughter(1)) { return (AliAODRecoDecayHF2Prong*)GetDaughter(1); } else { return (AliAODRecoDecayHF2Prong*)f2Prong.GetObject(); } }
-  void Unset2Prong () { f2Prong=0; return; }
+  AliAODRecoDecayHF2Prong* Get2Prong() const {return (AliAODRecoDecayHF2Prong*)GetDaughter(1);}
 
   // Bachelor (soft pion for Dstar)
   AliAODTrack* GetBachelor() const {return (AliAODTrack*)GetDaughter(0);}
@@ -51,12 +47,11 @@ class AliAODRecoCascadeHF : public AliAODRecoDecayHF2Prong {
   Double_t InvMassDstarKpipi() const;
   Double_t DeltaInvMass() const {return (InvMassDstarKpipi()-InvMassD0());}
 
- protected:
-
+  Int_t MatchToMC(Int_t pdgabs,Int_t pdgabs2prong,TClonesArray *mcArray) const;
 
-  TRef f2Prong; // the "V0" (e.g. D0->Kpi)
+ protected:
 
-  ClassDef(AliAODRecoCascadeHF, 1); // heavy-flavour cascade class
+  ClassDef(AliAODRecoCascadeHF, 2); // heavy-flavour cascade class
 };
 
 #endif