AliAnalysisTaskEmcalJetMassStructure.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKEMCALJETMASSSTRUCTURE_H
#define ALIANALYSISTASKEMCALJETMASSSTRUCTURE_H

class TH1;
class TH2;
class TH3;
class TProfile;
class THnSparse;
class TClonesArray;
class TArrayI;
class TRandom3;
class TList;

class AliAnalysisManager;
class AliJetContainer;
class AliEmcalJet;
class AliEmcalJetByJetCorrection;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalJetMassStructure : public AliAnalysisTaskEmcalJet {
 public:
  enum JetMassType {
    kRaw   = 0,  //mass form anti-kt 4-vector
    kDeriv = 1   //area based subtracted jet mass
  };

  enum JetByJetCorrType {
    kNoCorr  = 0,
    kAnnulus = 1, //reproduce existing particles (over-corrects)
    kMeanPtR = 2
  };

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

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

  //Setters
  void SetJetContainerBase(Int_t c)                             { fContainerBase     = c   ; }
  void SetMinFractionShared(Double_t f)                         { fMinFractionShared = f   ; }
  void SetJetMassType(JetMassType t)                            { fJetMassType       = t   ; }
  void SetFixedTrackEfficiency(Double_t eff)                    { fEfficiencyFixed   = eff ; }
  void SetJetByJetCorrType(JetByJetCorrType t)                  { fCorrType          = t   ; }
  void SetJetByJetCorrObject(AliEmcalJetByJetCorrection *a)     { fEJetByJetCorr     = a   ; }
  void SetParticleArray(TString particles)                { fPartArrayN         = particles;}
  Int_t CalculateNMissingTracks(AliEmcalJet *jet1, AliEmcalJet *jPart);
  //Getters
  AliEmcalJetByJetCorrection *GetJetByJetCorrObject() const     { return fEJetByJetCorr    ; }

 protected:
  Bool_t                              RetrieveEventObjects();
  Bool_t                              Run();
  Bool_t                              FillHistograms();

  Bool_t                              GetSortedArray(Int_t indexes[], Float_t dr[], AliEmcalJet *jet) const;
  Double_t                            GetJetMass(AliEmcalJet *jet);
  Double_t                            GetEfficiency(Double_t pt);
 
  Int_t                               fContainerBase;              
  Double_t                            fMinFractionShared;          
  JetMassType                         fJetMassType;                

  TH1F                               *fhtmppTRec;      
  TH1F                               *fhtmppTGen;                  

  TRandom3                           *fRandom;                     
  Double_t                            fEfficiencyFixed;            
                                                                     
  JetByJetCorrType                    fCorrType;                   
  AliEmcalJetByJetCorrection         *fEJetByJetCorr;              
 
  TH3F                              **fh3PtDRMass;                 
  TH3F                              **fh3PtDRRho;                  
  TH3F                              **fh3PtDRMassCorr;             
  TH3F                              **fh3PtDRRhoCorr;              
  TH2F                              **fh2PtMass;                   
  TH2F                              **fh2PtMassCorr;               
  THnSparse                          *fhnMassResponse;             
  THnSparse                          *fhnMassResponseCorr;         
  TH3F                              **fh3JetPtDRTrackPt;           
  THnSparse                          *fhnDeltaMass;                
  THnSparse                          *fhnDeltaMassCorr;            
  TH2F                               *fhAllpTRec;
  TH2F                               *fhAllpTGen;
  TH2F                               *fhAllpTCor;
  THnSparse                          *fhConstRecGen;           
  TList                              *fListOfOutputFromClass;      
  Bool_t                             fSwitchResolutionhn;          
  TString                       fPartArrayN;                 
  private:
  AliAnalysisTaskEmcalJetMassStructure(const AliAnalysisTaskEmcalJetMassStructure&);            // not implemented
  AliAnalysisTaskEmcalJetMassStructure &operator=(const AliAnalysisTaskEmcalJetMassStructure&); // not implemented

  ClassDef(AliAnalysisTaskEmcalJetMassStructure, 6)
};
#endif