Jet and Particle identification tasks moved to different directories

[u/mrichter/AliRoot.git] / PWG4 / PartCorr / AliAnalysisTaskGammaConversion.h

diff --git a/PWG4/PartCorr/AliAnalysisTaskGammaConversion.h b/PWG4/PartCorr/AliAnalysisTaskGammaConversion.h
new file mode 100644 (file)
index 0000000..6f1e3ae
--- /dev/null
+++ b/PWG4/PartCorr/AliAnalysisTaskGammaConversion.h
@@ -0,0 +1,93 @@
+#ifndef ALIANALYSISTASKGAMMACONVERSION_H
+#define ALIANALYSISTASKGAMMACONVERSION_H
+ 
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+ 
+#include "AliAnalysisTaskSE.h"
+#include "AliV0Reader.h"
+//#include "TH2F.h"
+//#include "TH1F.h"
+#include "AliGammaConversionHistograms.h"
+#include "AliESDv0.h"
+#include <vector>
+//#include "TLorentzVector.h"
+#include "AliKFParticle.h"
+
+//class AliAnaGamma;
+class AliESDInputHandler;
+class AliESDEvent;
+class AliAODEvent;
+class TList;
+class AliStack;
+
+class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE
+{
+ public:
+    AliAnalysisTaskGammaConversion();
+    AliAnalysisTaskGammaConversion(const char* name);
+    virtual ~AliAnalysisTaskGammaConversion() ;// virtual destructor
+ 
+    // Implementation of interface methods
+    virtual void UserCreateOutputObjects();
+    virtual void Init();
+    virtual void LocalInit() {Init();}
+    virtual void Exec(Option_t *option);
+    virtual void Terminate(Option_t *option);
+    virtual void ConnectInputData(Option_t *);
+       
+    void ProcessMCData();
+    void ProcessV0s();
+    void ProcessGammasForNeutralMesonAnalysis();
+    void SetHistograms(AliGammaConversionHistograms *histograms){fHistograms=histograms;}
+    void SetDoMCTruth(Bool_t flag){fDoMCTruth=flag;}
+    void SetElectronMass(Double_t electronMass){fElectronMass = electronMass;}
+    void SetGammaMass(Double_t gammaMass){fGammaMass = gammaMass;}
+    void SetGammaWidth(Double_t gammaWidth){fGammaWidth = gammaWidth;}
+    void SetPi0Mass(Double_t pi0Mass){fPi0Mass = pi0Mass;}
+    void SetPi0Width(Double_t pi0Width){fPi0Width = pi0Width;}
+    void SetEtaMass(Double_t etaMass){fEtaMass = etaMass;}
+    void SetEtaWidth(Double_t etaWidth){fEtaWidth = etaWidth;}
+    void SetV0Reader(AliV0Reader* reader){fV0Reader=reader;}
+    void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}
+    void CalculateBackground();
+    Double_t GetMCOpeningAngle(TParticle* daughter0,TParticle* daughter1);
+
+ private:
+    AliAnalysisTaskGammaConversion(const AliAnalysisTaskGammaConversion&); // Not implemented
+    AliAnalysisTaskGammaConversion& operator=(const AliAnalysisTaskGammaConversion&); // Not implemented
+
+    AliV0Reader* fV0Reader;
+
+    AliStack * fStack;
+
+    TList * fOutputContainer ; // Histogram container
+
+    AliGammaConversionHistograms *fHistograms;
+
+    Bool_t fDoMCTruth;
+    
+    vector<TParticle*> fMCAllGammas;
+    vector<TParticle*> fMCPi0s;
+    vector<TParticle*> fMCEtas;
+    vector<TParticle*> fMCGammaChi_c;
+
+    vector<AliKFParticle> fKFReconstructedGammas;
+
+    //mass defines
+    Double_t fElectronMass;
+    Double_t fGammaMass;
+    Double_t fPi0Mass;
+    Double_t fEtaMass;
+
+    // width defines
+    Double_t fGammaWidth;
+    Double_t fPi0Width;
+    Double_t fEtaWidth;
+    Bool_t fCalculateBackground;
+
+
+    ClassDef(AliAnalysisTaskGammaConversion, 0); // Analysis task for gamma conversions
+};
+ 
+#endif //ALIANALYSISTASKGAMMA_H