Cleaning up files, added documentation. Added ntuple for v0s.
[u/mrichter/AliRoot.git] / PWG4 / GammaConv / AliAnalysisTaskGammaConversion.h
index 9e48cdf..bcff03f 100644 (file)
@@ -13,6 +13,7 @@
 #include "AliAnalysisTaskSE.h"\r
 #include <vector>\r
 #include "AliV0Reader.h"\r
+#include "TNtuple.h"\r
 \r
 class AliGammaConversionHistograms;\r
 class AliESDv0;\r
@@ -53,6 +54,8 @@ class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE
     void SetV0Reader(AliV0Reader* reader){fV0Reader=reader;}\r
     void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}\r
     void CalculateBackground();\r
+    void SetWriteNtuple(Bool_t writeNtuple){fWriteNtuple = writeNtuple;}\r
+    void FillNtuple();\r
     Double_t GetMCOpeningAngle(TParticle* daughter0, TParticle* daughter1) const;\r
 \r
  private:\r
@@ -87,9 +90,13 @@ class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE
     Double_t fPi0Width;\r
     Double_t fEtaWidth;\r
     Bool_t fCalculateBackground;\r
+    Bool_t fWriteNtuple;\r
+    TNtuple *fGammaNtuple;\r
+    TNtuple *fNeutralMesonNtuple;\r
 \r
+    Int_t fTotalNumberOfAddedNtupleEntries;\r
 \r
-    ClassDef(AliAnalysisTaskGammaConversion, 0); // Analysis task for gamma conversions\r
+    ClassDef(AliAnalysisTaskGammaConversion, 1); // Analysis task for gamma conversions\r
 };\r
  \r
 #endif //ALIANALYSISTASKGAMMA_H\r