AliJetResponseMaker.h

Go to the documentation of this file.

#ifndef ALIJETRESPONSEMAKER_H
#define ALIJETRESPONSEMAKER_H
/**************************************************************************
* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

//-----------------------------------------------------------------------
// Author : Salvatore Aiola, Yale University, salvatore.aiola@cern.ch
//-----------------------------------------------------------------------

class TClonesArray;
class TH2;
class THnSparse;
class AliNamedArrayI;

#include "AliEmcalJet.h"
#include "AliAnalysisTaskEmcalJet.h"
#include "AliEmcalEmbeddingQA.h"

class AliJetResponseMaker : public AliAnalysisTaskEmcalJet {
 public:
  AliJetResponseMaker();
  AliJetResponseMaker(const char *name);
  virtual ~AliJetResponseMaker();

  enum MatchingType{
    kNoMatching = 0,
    kGeometrical = 1,
    kMCLabel = 2,
    kSameCollections = 3
  };

  void                        UserCreateOutputObjects();

  void                        SetMatching(MatchingType t, Double_t p1=1, Double_t p2=1)       { fMatching = t; fMatchingPar1 = p1; fMatchingPar2 = p2; }
  void                        SetPtHardBin(Int_t b)                                           { fSelectPtHardBin   = b         ; }
  void                        SetUseCellsToMatch(Bool_t i)                                    { fUseCellsToMatch   = i         ; }
  void                        SetMinJetMCPt(Float_t pt)                                       { fMinJetMCPt        = pt        ; }
  void                        SetHistoType(Int_t b)                                           { fHistoType         = b         ; }
  void                        SetDeltaPtAxis(Int_t b)                                         { fDeltaPtAxis       = b         ; }
  void                        SetDeltaEtaDeltaPhiAxis(Int_t b)                                { fDeltaEtaDeltaPhiAxis= b       ; }
  void                        SetNEFAxis(Int_t b)                                             { fNEFAxis           = b         ; }
  void                        SetZAxis(Int_t b)                                               { fZAxis             = b         ; }
  void                        SetFlavourZAxis(Int_t b)                                        { fFlavourZAxis      = b         ; }
  void                        SetFlavourPtAxis(Int_t b)                                       { fFlavourPtAxis     = b         ; }
  void                        SetZgAxis(Int_t b)                                              { fZgAxis            = b         ; }
  void                        SetdRAxis(Int_t b)                                              { fdRAxis            = b         ; }
  void                        SetPtgAxis(Int_t b)                                             { fPtgAxis           = b         ; }
  void                        SetDBCAxis(Int_t b)                                             { fDBCAxis           = b         ; }
  void                        SetJetRelativeEPAngleAxis(Int_t b)                              { fJetRelativeEPAngle = b        ; }

  static AliJetResponseMaker * AddTaskJetResponseMaker(
      const char *ntracks1           = "Tracks",
      const char *nclusters1         = "CaloClusters",
      const char *njets1             = "Jets",
      const char *nrho1              = "Rho",
      const Double_t    jetradius1         = 0.2,
      const char *ntracks2           = "MCParticles",
      const char *nclusters2         = "",
      const char *njets2             = "MCJets",
      const char *nrho2              = "",
      const Double_t    jetradius2         = 0.2,
      const Double_t    jetptcut           = 1,
      const Double_t    jetareacut         = 0.557,
      const Double_t    jetBias            = 5,
      const Int_t       biasType           = 0,   //  0 = charged, 1 = neutral, 2 = both
      const AliJetResponseMaker::MatchingType matching = AliJetResponseMaker::kGeometrical,
      const Double_t    maxDistance1       = 0.25,
      const Double_t    maxDistance2       = 0.25,
      const char *cutType            = "TPC",
      const Int_t       ptHardBin          = -999,
      const Double_t    minCent            = -999,
      const Double_t    maxCent            = -999,
      const char *taskname           = "AliJetResponseMaker",
      const Bool_t      biggerMatrix       = kFALSE,
      AliJetResponseMaker* address   = 0,
      const Double_t    nefmincut          = -10,
      const Double_t    nefmaxcut          = 10,
      const Int_t       jetTagging         = 0,
      const Double_t    maxTrackPt         = 100);

 protected:
  void                        ExecOnce();
  void                        DoJetLoop();
  Bool_t                      FillHistograms();
  Bool_t                      Run();
  Bool_t                      DoJetMatching();
  void                        SetMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, MatchingType matching);
  void                        GetGeometricalMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d) const;
  void                        GetMCLabelMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d1, Double_t &d2) const;
  void                        GetSameCollectionsMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d1, Double_t &d2) const;
  void                        FillMatchingHistos(AliEmcalJet* jet1, AliEmcalJet* jet2, Double_t d, Double_t CE1, Double_t CE2);
  void                        FillJetHisto(AliEmcalJet* jet, Int_t Set);
  void                        AllocateTH2();
  void                        AllocateTHnSparse();
  Double_t                    GetRelativeEPAngle(Double_t jetAngle, Double_t epAngle) const;

  MatchingType                fMatching;                               // matching type
  Double_t                    fMatchingPar1;                           // matching parameter for jet1-jet2 matching
  Double_t                    fMatchingPar2;                           // matching parameter for jet2-jet1 matching
  Bool_t                      fUseCellsToMatch;                        // use cells instead of clusters to match jets (slower but sometimes needed)
  Double_t                    fMinJetMCPt;                             // minimum jet MC pt
  AliEmcalEmbeddingQA         fEmbeddingQA;                            
  Int_t                       fHistoType;                              // histogram type (0=TH2, 1=THnSparse)
  Int_t                       fDeltaPtAxis;                            // add delta pt axis in THnSparse (default=0)
  Int_t                       fDeltaEtaDeltaPhiAxis;                   // add delta eta and delta phi axes in THnSparse (default=0)
  Int_t                       fNEFAxis;                                // add NEF axis in matching THnSparse (default=0)
  Int_t                       fZAxis;                                  // add Z axis in matching THnSparse (default=0)
  Int_t                       fFlavourZAxis;                           // add flavour Z axis in matching THnSparse (default=0)
  Int_t                       fFlavourPtAxis;                          // add flavour pt axis in matching THnSparse (default=0)
  Int_t                       fZgAxis;                                 // add Zg axis in matching THnSparse (default=0)
  Int_t                       fdRAxis;                                 // add dR axis in matching THnSparse (default=0)
  Int_t                       fPtgAxis;                                // add Ptg axis in matching THnSparse (default=0)
  Int_t                       fDBCAxis;                                // add DBC (number of soft dropped branches) axis in matching THnSparse (default=0)
  Int_t                       fJetRelativeEPAngle;                     

  Bool_t                      fIsJet1Rho;                              
  Bool_t                      fIsJet2Rho;                              

  TH2                        *fHistRejectionReason1;                   
  TH2                        *fHistRejectionReason2;                   

  // THnSparse
  THnSparse                  *fHistJets1;                              
  THnSparse                  *fHistJets2;                              
  THnSparse                  *fHistMatching;                           

  // Jets 1
  TH2                        *fHistJets1PhiEta;                        
  TH2                        *fHistJets1PtArea;                        
  TH2                        *fHistJets1CorrPtArea;                    
  TH2                        *fHistJets1NEFvsPt;                       
  TH2                        *fHistJets1CEFvsCEFPt;                    
  TH2                        *fHistJets1ZvsPt;                         

  // Jets 2
  TH2                        *fHistJets2PhiEta;                        
  TH2                        *fHistJets2PtArea;                        
  TH2                        *fHistJets2CorrPtArea;                    
  TH2                        *fHistJets2NEFvsPt;                       
  TH2                        *fHistJets2CEFvsCEFPt;                    
  TH2                        *fHistJets2ZvsPt;                         

  // Jet1-Jet2 matching
  TH2                        *fHistCommonEnergy1vsJet1Pt;              
  TH2                        *fHistCommonEnergy2vsJet2Pt;              
  TH2                        *fHistDistancevsJet1Pt;                   
  TH2                        *fHistDistancevsJet2Pt;                   
  TH2                        *fHistDistancevsCommonEnergy1;            
  TH2                        *fHistDistancevsCommonEnergy2;            
  TH2                        *fHistCommonEnergy1vsCommonEnergy2;       
  TH2                        *fHistDeltaEtaDeltaPhi;                   

  TH2                        *fHistJet2PtOverJet1PtvsJet2Pt;           
  TH2                        *fHistJet1PtOverJet2PtvsJet1Pt;           
  TH2                        *fHistDeltaPtvsJet1Pt;                    
  TH2                        *fHistDeltaPtvsJet2Pt;                    
  TH2                        *fHistDeltaPtOverJet1PtvsJet1Pt;          
  TH2                        *fHistDeltaPtOverJet2PtvsJet2Pt;          
  TH2                        *fHistDeltaPtvsDistance;                  
  TH2                        *fHistDeltaPtvsCommonEnergy1;             
  TH2                        *fHistDeltaPtvsCommonEnergy2;             
  TH2                        *fHistDeltaPtvsArea1;                     
  TH2                        *fHistDeltaPtvsArea2;                     
  TH2                        *fHistDeltaPtvsDeltaArea;                 
  TH2                        *fHistJet1PtvsJet2Pt;                     

  TH2                        *fHistDeltaCorrPtOverJet1CorrPtvsJet1CorrPt;
  TH2                        *fHistDeltaCorrPtOverJet2CorrPtvsJet2CorrPt;
  TH2                        *fHistDeltaCorrPtvsJet1CorrPt;            
  TH2                        *fHistDeltaCorrPtvsJet2CorrPt;            
  TH2                        *fHistDeltaCorrPtvsDistance;              
  TH2                        *fHistDeltaCorrPtvsCommonEnergy1;         
  TH2                        *fHistDeltaCorrPtvsCommonEnergy2;         
  TH2                        *fHistDeltaCorrPtvsArea1;                 
  TH2                        *fHistDeltaCorrPtvsArea2;                 
  TH2                        *fHistDeltaCorrPtvsDeltaArea;             
  TH2                        *fHistJet1CorrPtvsJet2CorrPt;             

  TH2                        *fHistDeltaMCPtOverJet1MCPtvsJet1MCPt;    
  TH2                        *fHistDeltaMCPtOverJet2PtvsJet2Pt;        
  TH2                        *fHistDeltaMCPtvsJet1MCPt;                
  TH2                        *fHistDeltaMCPtvsJet2Pt;                  
  TH2                        *fHistDeltaMCPtvsDistance;                
  TH2                        *fHistDeltaMCPtvsCommonEnergy1;           
  TH2                        *fHistDeltaMCPtvsCommonEnergy2;           
  TH2                        *fHistDeltaMCPtvsArea1;                   
  TH2                        *fHistDeltaMCPtvsArea2;                   
  TH2                        *fHistDeltaMCPtvsDeltaArea;               
  TH2                        *fHistJet1MCPtvsJet2Pt;                   

 private:
  AliJetResponseMaker(const AliJetResponseMaker&);            // not implemented
  AliJetResponseMaker &operator=(const AliJetResponseMaker&); // not implemented

  ClassDef(AliJetResponseMaker, 29) // Jet response matrix producing task
};
#endif