New more general analysis implemention for particle identification and correlation...

[u/mrichter/AliRoot.git] / PWG4 / AliAnaMaker.h

#ifndef ALIANAMAKER_H
#define ALIANAMAKER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */
/* $Id:  $ */

/* History of cvs commits:
 *
 * $Log$
 *
 */

//_________________________________________________________________________
// Base class for gamma and particle correlation analysis
// It is called by the task class AliAnalysisGammaTask and it connects the input (ESD/AOD/MonteCarlo)
// got with AliCaloTrackReader (produces TClonesArrays of TParticles), with the analysis classes 
// that derive from AliAnaBaseClass
//
//*-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
/* #include <TParticle.h>  */
#include <TList.h> 
#include <TClonesArray.h> 
/* #include <TH2F.h> */
#include<TObject.h>
/* #include <TTree.h> */
#include <TString.h>

// --- AliRoot system ---
#include <AliLog.h>
#include "AliAnaBaseClass.h"
/* #include "AliAODCaloCluster.h" */
/* #include "AliAODTrack.h" */
/* #include "AliAODPhoton.h" */

class AliCaloTrackReader ;

// --- AliRoot
class AliAnaMaker : public TObject {

public: 
  
  AliAnaMaker() ; // default ctor
  AliAnaMaker(const AliAnaMaker & g) ; // cpy ctor
  AliAnaMaker & operator = (const AliAnaMaker & g) ;//cpy assignment
  virtual ~AliAnaMaker() ; //virtual dtor

  //Setter and getters
  TList * GetOutputContainer()      const {return fOutputContainer ; }

  Int_t GetAnaDebug() const  { return fAnaDebug ; }
  void SetAnaDebug(Int_t d)   { fAnaDebug = d ; }

  Bool_t AreHistogramsMade() const  { return fMakeHisto ; }
  void SwitchOnHistogramsMaker()   { fMakeHisto = kTRUE ; }
  void SwitchOffHistogramsMaker()   { fMakeHisto = kFALSE ; }
 
  Bool_t AreAODsMade() const  { return fMakeAOD ; }
  void SwitchOnAODsMaker()   { fMakeAOD = kTRUE ; }
  void SwitchOffAODsMaker()   { fMakeAOD = kFALSE ; }

  void SetAODBranch(TClonesArray* aodbranch) {fAODBranch = aodbranch;}
  
  void AddAnalysis(TObject* ana, Int_t n) {if ( fAnalysisContainer) fAnalysisContainer->AddAt(ana,n); 
    else AliFatal("AnalysisContainer not initialized");}

  TString GetAODBranchName() const {return  fAODBranchName;}
  void SetAODBranchName(TString name)  {fAODBranchName = name ;}

  AliCaloTrackReader * GetReader() const {return fReader ; }
  void SetReader(AliCaloTrackReader * reader) { fReader = reader ; }

  //Others
  void Init();
  void InitParameters();

  void Print(const Option_t * opt) const;

  Bool_t ProcessEvent(Int_t iEntry) ;

 private:
  
  //General Data members

  TList * fOutputContainer ; // output histograms container
  TList * fAnalysisContainer ; // List with analysis pointers
  Bool_t  fMakeHisto ; // If true makes final analysis with histograms as output
  Bool_t  fMakeAOD ; // If true makes analysis generating AODs
  Int_t   fAnaDebug; // Debugging info.
 
  AliCaloTrackReader *  fReader ; // Pointer to reader 
  TClonesArray* fAODBranch ;        //! selected particles branch
  TString fAODBranchName; // New AOD branch name

  ClassDef(AliAnaMaker,1)
} ;
 

#endif //ALIANAMAKER_H