-Move cent, z axis into sparse

[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnaConvCorrBase.h

/* This file is property of and copyright                                 *\r
 * ALICE Experiment at CERN, All rights reserved.                         *\r
 * See cxx source for full Copyright notice                               */\r
\r
/// @file   AliAnaConvCorrPhoton.h\r
/// @author Svein Lindal\r
/// @brief  Base class for analysis of correlations between conversion particles and charged tracks\r
\r
#ifndef ALIANACONVCORRBASE_CXX\r
#define ALIANACONVCORRBASE_CXX\r
\r
#include "Rtypes.h"\r
#include "TMath.h"\r
#include "TList.h"\r
#include "TH1.h"\r
#include <THnSparse.h>\r
\r
class AliAODConversionParticle;\r
class TClonesArray;\r
class TString;\r
\r
class AliAnaConvCorrBase : public TNamed {\r
\r
public:\r
\r
  \r
\r
  //Constructor / desctructor\r
  AliAnaConvCorrBase(TString name, TString title); \r
  virtual ~AliAnaConvCorrBase();\r
  \r
  //CreateHistograms\r
  void CreateBaseHistograms();\r
  //To be overrriden by children. Should always call CreateBaseHistograms()\r
  virtual void CreateHistograms();\r
  \r
  //Get list of histograms\r
  TList * GetHistograms() const { return fHistograms;}\r
\r
  //Add histogram to list\r
  void AddHistogram(TH1 * histogram) { fHistograms->Add(dynamic_cast<TObject*>(histogram));}\r
\r
  void AddAxis(TAxis * axis) { fAxesList.Add(axis); }\r
\r
  ///Get the distance in phi between trigger particle and correlated particle\r
  Float_t GetDPhi(Float_t dPhi) { \r
    if ( dPhi < 3*TMath::PiOver2() && dPhi > - TMath::PiOver2() ) return dPhi;\r
    else return ( (dPhi>0)? dPhi - TMath::TwoPi() : dPhi + TMath::TwoPi() ); \r
  }\r
\r
  void CorrelateWithTracks(AliAODConversionParticle * particle, TObjArray * tracks, const Int_t tIDs[4], Float_t cent, Float_t vtxz);\r
  void FillCounters(TObjArray * particles, TObjArray * tracks, Float_t cent, Float_t vtxz);\r
\r
  TAxis& GetAxistPt()       { return fAxistPt;   }\r
  TAxis& GetAxiscPt()       { return fAxiscPt;   }\r
  TAxis& GetAxisdEta()      { return fAxisdEta;  }\r
  TAxis& GetAxisdPhi()      { return fAxisdPhi;  }\r
  TAxis& GetAxisIso()       { return fAxisIso;   }\r
  TAxis& GetAxisCent()      { return fAxisCent;  }\r
  TAxis& GetAxisZ()         { return fAxisZ;     }\r
  TAxis& GetAxisMEEta()     { return fAxisMEEta; }\r
  TAxis& GetAxisMEPhi()     { return fAxisMEPhi; }\r
\r
  TList& GetAxisList()      { return fAxesList;  }\r
  TList& GetTrackAxisList() { return fTrackAxisList; }\r
  TList& GetTrigAxisList()  { return fTrigAxisList; }\r
\r
\r
protected:\r
\r
  THnSparseF * CreateSparse(TString name, TString title, TList * axes);\r
\r
private:\r
\r
  void SetUpDefaultBins();\r
  \r
  //TString fName; //name of analysis\r
  TList * fHistograms; //List of histograms\r
  TList fAxesList;  //List over axes to be used in sparse\r
  TList fTrigAxisList; //list\r
  TList fTrackAxisList; //list\r
\r
  TAxis fAxistPt; //Pt axis\r
  TAxis fAxiscPt; //correlated particle pt axis\r
  TAxis fAxisdEta; //delta eta axis\r
  TAxis fAxisdPhi; //delta phi axis\r
  TAxis fAxisIso; //Isolated particle axis\r
  TAxis fAxisCent; //Centrality\r
  TAxis fAxisZ; //vtx\r
  \r
  TAxis fAxisMEEta ; //Eta axis for ME\r
  TAxis fAxisMEPhi ; //Phi axis for ME\r
\r
  THnSparseF * fCorrSparse; // Sparse for corr\r
  THnSparseF * fTrigSparse; // ME Sparse\r
  THnSparseF * fTrackSparse; //Track Sparse\r
\r
  //Default constructor prohibited\r
  AliAnaConvCorrBase(); //not implemented\r
  AliAnaConvCorrBase(const AliAnaConvCorrBase&); // not implemented\r
  AliAnaConvCorrBase& operator=(const AliAnaConvCorrBase&); // not implemented\r
\r
  ClassDef(AliAnaConvCorrBase, 6); // example of analysis\r
\r
};\r
\r
#endif\r