AliAnalysisTaskJetJTJT.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKJETJTJT_H
#define ALIANALYSISTASKJETJTJT_H

class TH1;
class TH2;
class TH3;
class TGraphErrors;
class TProfile;
class AliJetContainer;
class AliParticleContainer;
class AliClusterContainer;
class AliAnalysisUtils;
class JTJTEfficiency;


#include "AliAnalysisTaskEmcalJet.h"

using std::cout;
using std::endl;

class AliAnalysisTaskJetJTJT : public AliAnalysisTaskEmcalJet {
 public:

  AliAnalysisTaskJetJTJT();
  AliAnalysisTaskJetJTJT(const char *name);
  virtual ~AliAnalysisTaskJetJTJT();

  void                        UserCreateOutputObjects();
  void                        Terminate(Option_t *option);

  void SetTrackArrayName( char *c ) { fTrackArrayName = c; }

  void setCentBinBorders( int n, Double_t *c);
  void setTriggPtBorders( int n, Double_t *c);
  void setAssocPtBorders( int n, Double_t *c);
  void setEffMode(int m) {fEffMode = m; };
  void setDebug(int n) {debug = n; }
  void setRunPeriod(const char *period) {runPeriod = period; cout << "RunPeriod set to " << runPeriod << endl;}
  //AliJetContainer            *AddJetContainer(const char *n, TString defaultCutType = "", Float_t jetRadius = 0.4);


 protected:
  void                        ExecOnce();
  Bool_t                      FillHistograms()   ;
  Bool_t                      Run()              ;
  void                        CheckClusTrackMatching();
  Double_t              getJt(AliVTrack *track, AliEmcalJet *jet, int reverse);
  Double_t              getJt(AliVParticle *track, AliEmcalJet *jet, int reverse);
  Double_t                    getDiffR(double phi1, double phi2, double eta1, double eta2);


  // General histograms
  TH1                       **fHistTracksPt;            
  TH1                       **fHistTracksJt;            
  TH1         **fHistTracksEta;           
  TH1                       **fHistClustersPt;          
  TH1                       **fHistLeadingJetPt;        
  TH1                       ***fHistJetsPt;           
  TH1                       ***fHistJetsCorrPt;         
  TProfile                  **fHistJetsCorrPtVsNonCorr; 
  TH1         ***fHistBackgroundDone; 

  //Jt histograms
  TH1                       ****fHistJTPta;     
  TH1                       ****fHistLogJTPta;      
  TH1                       ****fHistJTPta_all;     
  TH1                       ****fHistJTBg;      
  TH1                       ****fHistLogJTBg;     

  TProfile        ***fHistPtaVsJt;             
  TProfile        ***fHistBgPtaVsJt;           

  //non-invariant
  TH1                       ****fHistJTPtaNonInv;     
  TH1                       ****fHistLogJTPtaNonInv;      
  TH1                       ****fHistJTPta_allNonInv;     
  TH1                       ****fHistJTBgNonInv;      
  TH1                       ****fHistLogJTBgNonInv;     


  //Background statistics

  TH1         ***fHistBgMulti;    
  TH1         ***fHistBgPt;   

  //Jet statistics
  TH1         ***fHistJetEta;   
  TH1         ***fHistJetMulti;   
  TH1         ***fHistJetTracksPt;    
  TProfile        **fhTrackingEfficiency;   

  Int_t fNpttBins;
  Int_t fNptaBins;
  Int_t fEffMode;

  AliJetContainer            *fJetsCont;                   
  //AliJetContainer            **fJetsConts;              //!Jets
  //Int_t          nJetsConts;
  AliParticleContainer       *fTracksCont;                 
  AliClusterContainer        *fCaloClustersCont;           

  TH1F     *fhVertexZ;  
  TH1I     *fHistEvtSelection; 

 private:
  //TVector *EtaGapThresholds;
  //TVector *RGapThresholds;
  //TVector *KlongBorders;
  //TVector *XeBorders;


  const AliVVertex*   fPrimaryVertex;         


  TClonesArray *fTracks;  
  TString    fTrackArrayName; // track constituents array name
  TString    runPeriod; // run period name
  JTJTEfficiency *fEfficiency; 
  AliAnalysisUtils* fVertexHelper;  


  //TVector *CentBinBorders;
  //TVector *TriggPtBorders;
  //TVector *AssocPtBorders;
  Double_t CentBinBorders[10];
  Double_t TriggPtBorders[10];
  Double_t AssocPtBorders[10];
  AliAnalysisTaskJetJTJT(const AliAnalysisTaskJetJTJT&);            // not implemented
  AliAnalysisTaskJetJTJT &operator=(const AliAnalysisTaskJetJTJT&); // not implemented

  Int_t debug;

  ClassDef(AliAnalysisTaskJetJTJT, 3) // jet sample analysis task



};

class JTJTEfficiency {  // this part can occurr anywhere inside the outer accolades
#define JUNUSED(expr) do { (void)(expr); } while (0)
  public:
    enum Mode { kNotUse, kPeriod, kRunNumber, kAuto };
    enum { kJTPCOnly, kJRaa, kJGlobalTightDCA, kJGlobalDCA, kJGlobalSDD , kJHybrid, kJNTrackCuts };



    JTJTEfficiency();
    JTJTEfficiency(const JTJTEfficiency& obj);
    JTJTEfficiency& operator=(const JTJTEfficiency& obj);
    void SetMode( int i ){ fMode = i; }
    void SetDataPath(TString s ){ fDataPath=s; }
    void SetPeriodName(TString s ){ fPeriodStr=s; cout << "Eff: Run Period is set to " << fPeriodStr << endl; }
    TString GetName() const { return fName; }
    double GetCorrection( double pt, int icut, double cent ) const ;
    void SetRunNumber( Long64_t runnum ){ fRunNumber=runnum; }

    TString GetEffName() ;
    TString GetEffFullName() ;
    bool   Load();
    void   PrintOut() const {
      cout<<fInputRootName<<endl;
    }
    //void Write();

  private:
    int      fMode;             // Mode. see enum Mode
    int      fPeriod;           // Data Period index
    //AliJTrackCut fTrackCut;     // Track Cut Object. TODO:why not pointer?
    //AliJRunTable fRunTable;     // run Table. TODO:why not pointer?

    TString fDataPath;          // locaction of eff files
    TString fName;              // name of efficiency. usually empty
    TString fPeriodStr;         // DATA period
    TString fMCPeriodStr;       // MC period
    Long64_t fRunNumber;        // Runnumber
    TString fTag;               // Tags to distinguish special eff file
    TString fInputRootName;     // name of input

    TFile * fInputRoot;         // input file  
    TDirectory * fEffDir[3];    // root directory of efficiency. only second item of fEffDir with "Efficiency" is being used.
    TGraphErrors * fCorrection[20][20][20]; // Storage of Correction factor 
    TAxis * fCentBinAxis;     // Bin of Centrality. replace with AliJBin?



};

#endif