First implementation of EMCAL trigger QA from Nicola Arbor

[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskThreeJets.h

#ifndef ALIANALYSISTASKTHREEJETS_H\r
#define ALIANALYSISTASKTHREEJETS_H\r
\r
#include "AliAnalysisTaskSE.h"\r
\r
// \r
// Task for three jet ana by sona \r
// \r
\r
//class AliJetHistos;\r
//class AliJetCorrHistos;\r
class AliJetFinder;\r
class AliESDEvent;\r
class AliAODEvent;\r
class AliAODJet;\r
class AliGenPythiaEventHeader;\r
class AliAODPid;\r
\r
class TList;\r
class TArrayD;\r
class TChain;\r
class TH1;\r
class TH2;\r
class TH1F;\r
class TH2F;\r
class TH2I;\r
class TH3D;\r
class TTree;\r
class TProfile;\r
class TLorentzVector;\r
class TVector3;\r
class TParticle;\r
\r
class AliAnalysisTaskThreeJets : public AliAnalysisTaskSE\r
{\r
 public:\r
  AliAnalysisTaskThreeJets();\r
  AliAnalysisTaskThreeJets(const char * name);\r
  virtual ~AliAnalysisTaskThreeJets() {;}\r
  \r
  //Implementation of interface methods\r
  virtual Bool_t Notify(); \r
  virtual void UserCreateOutputObjects();\r
  virtual void Init();\r
  virtual void LocalInit() { Init(); };\r
  virtual void UserExec(Option_t * option);\r
  virtual void Terminate(Option_t * option);\r
\r
  virtual void SetAODInput(Bool_t b){fUseAODInput = b;}\r
  virtual void SetUseMC(Bool_t b){fUseMC = b;}\r
\r
  virtual void SetBranchGen(const char* c){fBranchGen = c;}\r
  virtual void SetBranchRec(const char* c){fBranchRec = c;}\r
\r
  virtual void FillTopology(TH2F * Dalitz, TH1F * fhMu34, TH1F * fhMu45, TH1F * fhMu35, Double_t * x, TVector3 * pRest, Double_t xsection);\r
\r
  virtual Double_t SetR(Double_t b){fR = b; return fR;} \r
  virtual Bool_t IsPrimChar(TParticle* aParticle, Int_t aTotalPrimaries, Bool_t adebug);\r
\r
  Double_t Exponent(Double_t x,const Double_t * const par) const;\r
  Double_t Exponent2(Double_t x,const Double_t * const par) const;\r
  Double_t Gauss(Double_t x,const Double_t * const par) const;\r
  Double_t Total(Double_t x,const Double_t * const par) const;\r
\r
 private:\r
  AliAnalysisTaskThreeJets(const AliAnalysisTaskThreeJets&);\r
  AliAnalysisTaskThreeJets& operator = (const AliAnalysisTaskThreeJets&);\r
\r
  enum {kMaxJets = 6};\r
  enum {kMaxEvents = 10};\r
  enum {kJets = 3};\r
  enum {kTracks = 1000};\r
\r
  AliAODEvent  *fAOD; // where we take the jets from can be input or output AOD\r
  Bool_t       fUseMC; //switch for using MC info  \r
\r
  TString       fBranchRec;  // AOD branch name for reconstructed\r
  TString       fBranchGen;  // AOD brnach for genereated\r
  \r
  Bool_t        fUseAODInput; // read the AOD from the input no from the output\r
\r
  Double_t fR; // radius\r
  TList * fList; // output list\r
\r
  TH1I * fhStopHisto; //Histogram to check were the code returns
  Int_t fGlobVar; // globvar\r
  Double_t fXsection; // xsectio\r
\r
TH2F * fhX3X4Rec; // Dalitz variables, reconstructed\r
  TH2F * fhX3X4Gen; // Dalitz variables, generated\r
  \r
  TH1F * fhMu35Rec; // scaled masses, 35, reconstructed\r
  TH1F * fhMu34Rec; // scaled masses, 34, reconstructed\r
  TH1F * fhMu45Rec; // scaled masses, 45, reconstructed\r
  \r
  TH1F * fhMu35Gen; // scaled masses, 35, generated\r
  TH1F * fhMu34Gen; // scaled masses, 34, generated\r
  TH1F * fhMu45Gen; // scaled masses, 45, generated\r
\r
  TH1I * fhInOut; // number of gen. vs number of rec.\r
  TH1F * fhThrustRec2; // thrust for reco 2-jet events\r
  TH1F * fhThrustRec3; // thrust for reco 3-jet events\r
\r
  TH1F * fhThrustGen2; // thrust for gen 2-jet events\r
  TH1F * fhThrustGen3; // thrust for gen 3-jet events\r
\r
  TH1F * fhCGen2; // C-variable for gen 2-jets\r
  TH1F * fhCGen3; // C-variable for gen 3-jets\r
\r
  TH1F * fhSGen2; // Sphericity for gen 2-jets\r
  TH1F * fhSGen3; // Sphericity for gen 3-jets\r
\r
  TH1F * fhAGen2; // A-variable for gen 2-jets\r
TH1F * fhAGen3; // A-variable for gen 3-jets\r
\r
  TH1F * fhCRec2; // C-variable for reco 2-jets\r
  TH1F * fhCRec3; // C-variable for reco 3-jets\r
\r
  TH1F * fhSRec2; // Sphericity for reco 2-jets\r
  TH1F * fhSRec3; // Sphericity for reco 3-jets\r
\r
  TH1F * fhARec2; // A-variable for reco 2-jets\r
  TH1F * fhARec3; // A-variable for reco 3-jets\r
\r
  TH2F * fhX3; // dX3 vs X3 rec\r
  TH2F * fhX4; // dX4 vs X4 rec\r
  TH2F * fhX5; // dX5 vs X5 rec\r
\r
  TProfile * fhXSec; // cross-section vs PtHard\r
  TH2F * fhX3X4Rec60; // Dalitz plane for Esum < 60, reco\r
  TH2F * fhX3X4Rec60100; // Dalitz plane for 60 < Esum < 100, reco\r
  TH2F * fhX3X4Rec100; // Dalitz plane for Esum > 100, reco^M \r
  TH2F * fhX3X4Gen60; // Dalitz plane for Esum < 60, gen\r
  TH2F * fhX3X4Gen60100; // Dalitz plane for 60 < Esum < 100, gen\r
  TH2F * fhX3X4Gen100; // Dalitz plane for Esum > 100, gen\r
\r
  TH2F * fhdPhiThrustGen; // energy distribution with rspct to thrust axis, gen, 2-jets\r
  TH2F * fhdPhiThrustGenALL; // energy distribution with rspct to thrust axis, gen 3-jets\r
  TH2F * fhdPhiThrustRec; // energy distribution with rspct to thrust axis, reco, 2-jets\r
  TH2F * fhdPhiThrustRecALL; // energy distribution with rspct to thrust axis, reco, 3-jets\r
\r
  ClassDef(AliAnalysisTaskThreeJets, 1)\r
};\r
\r
#endif\r