Added fit macro from M. Putis

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

#ifndef ALIANALYSISTASKJETCORRECTIONS_H\r
#define ALIANALYSISTASKJETCORRECTIONS_H\r
\r
#include "AliAnalysisTaskSE.h"\r
 \r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
// ***\r
// corrections to jet energy by sona.pochybova@cern.ch \r
// ***\r
\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 AliAnalysisTaskJetCorrections : public AliAnalysisTaskSE\r
{\r
 public:\r
  AliAnalysisTaskJetCorrections();\r
  AliAnalysisTaskJetCorrections(const char * name);\r
  virtual ~AliAnalysisTaskJetCorrections() {;}\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
\r
  virtual void SetBranchGen(const char* c){fBranchGen = c;}\r
  virtual void SetBranchRec(const char* c){fBranchRec = c;}\r
\r
  virtual Double_t SetR(Double_t b){fR = b; return fR;} \r
\r
 private:\r
  AliAnalysisTaskJetCorrections(const AliAnalysisTaskJetCorrections&);\r
  AliAnalysisTaskJetCorrections& operator = (const AliAnalysisTaskJetCorrections&);\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
  \r
  TString       fBranchRec;  // AOD branch name for reconstructed\r
  TString       fBranchGen;  // AOD brnach for genereated\r
  \r
  Bool_t        fUseAODInput; // use aod input not output\r
  Double_t fR; // radius\r
  TList * fList; // output list\r
 \r
  Int_t fGlobVar; // switch enabling checking out just one event\r
  Double_t fXsection; // cross-section \r
\r
\r
  TH1F * fhEGen; // generated energy histogram\r
  TH1F * fhERec; // reconstructed energy histogram\r
  TH1F * fhEGenRest; // generated energy in the rest frame of three-jet events\r
  TH1F * fhERecRest; // generated energy in the rest frame of three-jet events\r
  TH1F * fhEsumGenRest; //generated summed energy in the rest frame of three-jet events\r
  TH1F * fhEsumRecRest; // reconstructed summed energy in the rest frame of three-jet events\r
\r
  TH2F * fhE2vsE1Gen; // leading vs next-to leading jet energy, generated\r
  TH2F * fhE2vsE1Rec; // leading vs next-to leading jet energy, reconstruted\r
  TH2F * fhE2E1vsEsumGen; // E2,E1 diference vs. summed energy, generated\r
  TH2F * fhE2E1vsEsumRec; // E2, E1 difference vs summed jet energy, reconstructed\r
  TH2F * fhE2E1vsE1Gen; // E2, E1 difference vs E1, generated\r
  TH2F * fhE2E1vsE1Rec; // E2, E1 diference vs E1, reeconstructed\r
  TH2F * fhE2E1vsdPhiGen; // E2, E1 difference vs dPhi, generated\r
  TH2F * fhE2E1vsdPhiRec; // E2, E1 difference vs dPhi, reconstrted\r
\r
  TH2F * fhTrackBalance2; // track balance in 2-jet events\r
  TH2F * fhTrackBalance3; // track balance in 3-jet events\r
\r
  TH2F * fhEt1Et22; // Et1,2 in back-to-back cones in 2-jet events\r
  TH2F * fhEt1Et23; // Et1,2 in back-to-back cones in 3-jet events\r
\r
  TProfile * fhECorrJet10[3]; // corr. factor for jets matched within dR = 1.\r
  TProfile * fhECorrJet05[3]; // corr. factor for jets matched within dR = .5\r
  TProfile * fhECorrJet01[3]; // corr. factor for jets matched within dR = .1\r
  TProfile * fhECorrJet001[3]; // corr. factor for jets matched within dR = .01\r
  \r
  TProfile * fhdEvsErec10[3]; // energy difference vs rec. energy, dR = 1.\r
  TProfile * fhdEvsErec05[3]; // energy difference vs rec. energy, dR = .5\r
  TProfile * fhdEvsErec01[3]; // energy difference vs rec. energy, dR = .1\r
  TProfile * fhdEvsErec001[3]; // energy difference vs rec. energy, dR = .01\r
\r
  TH2F * fhdPhidEta10[3]; // dPhi vs dEta of particles, dR = 1.\r
  TH2F * fhdPhidEta05[3]; // dPhi vs dEta of particles, dR = .5\r
  TH2F * fhdPhidEta01[3]; // dPhi vs dEta of particles, dR = .1\r
  TH2F * fhdPhidEta001[3]; // dPhi vs dEta of particles, dR = .01\r
\r
  TH2F * fhdPhidEtaPt10[3]; // Pt weighted dPhi vs dEta of particles, dR = 1.\r
  TH2F * fhdPhidEtaPt05[3]; // Pt weighted dPhi vs dEta of particles, dR = .5\r
  TH2F * fhdPhidEtaPt01[3]; // Pt weighted dPhi vs dEta of particles, dR = .1\r
  TH2F * fhdPhidEtaPt001[3]; // Pt weighted dPhi vs dEta of particles, dR = .01\r
  ClassDef(AliAnalysisTaskJetCorrections, 1)\r
};\r
\r
#endif\r