Analysis code updated

[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskDiJets.h

diff --git a/JETAN/AliAnalysisTaskDiJets.h b/JETAN/AliAnalysisTaskDiJets.h
index 200df5aa6f6caa7d1e035b6ab537d0a21c4b4c18..cfcf5e33a8e376bd154f70a3a8f21d50963cc9ba 100644 (file)
--- a/JETAN/AliAnalysisTaskDiJets.h
+++ b/JETAN/AliAnalysisTaskDiJets.h
@@ -22,6 +22,11 @@ class AliAnalysisTaskDiJets : public AliAnalysisTaskSE
     virtual void LocalInit() {Init();}
     virtual void UserExec(Option_t *option);
     virtual void Terminate(Option_t *option);
+
+    void SetAODInput(Bool_t b)   {fUseAODInput = b;}
+    void SetFillAOD(Bool_t fill) { fFillAOD=fill; }
+    void SetJetBranch(const Char_t *br){ fJetBranch=br; }
+
  private:
   AliAnalysisTaskDiJets(const AliAnalysisTaskDiJets &det);
   AliAnalysisTaskDiJets &operator=(const AliAnalysisTaskDiJets &det);
@@ -30,6 +35,11 @@ class AliAnalysisTaskDiJets : public AliAnalysisTaskSE
   TClonesArray* fDiJets;    // Array of dijets
   TClonesArray* fDiJetsIn;  // Array of dijets
 
+  Bool_t        fUseAODInput; // read jets from input AOD
+  Bool_t        fFillAOD;   // option to fill AOD branch
+  TString       fJetBranch; // jet branch to read
+
+  AliAODEvent  *fAOD;       // AOD event
   TList        *fHistList;  // Output list
 
   TH1F         *fH1DeltaPt;  // Pt difference