AliAnalysisTaskParticleRandomizer.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKPARTICLERANDOMIZER_H
#define ALIANALYSISTASKPARTICLERANDOMIZER_H
/* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

// 
class TClonesArray;
class TString;
class TRandom3;
class TH2D;

class AliAnalysisTaskParticleRandomizer : public AliAnalysisTaskEmcal {
public:
  AliAnalysisTaskParticleRandomizer();
  virtual ~AliAnalysisTaskParticleRandomizer();
  virtual void  UserCreateOutputObjects();
  virtual void  Terminate(Option_t *) {}

  // ###### Configuration setters
  void          SetRandomizeInPhi(Bool_t val)               {fRandomizeInPhi = val;}
  void          SetRandomizeInEta(Bool_t val)               {fRandomizeInEta = val;}
  void          SetRandomizeInTheta(Bool_t val)             {fRandomizeInTheta = val;}
  void          SetRandomizeInPt(Bool_t val)                {fRandomizeInPt = val;}

  void          SetPhiMin(Double_t val)                     {fMinPhi = val;}
  void          SetPhiMax(Double_t val)                     {fMaxPhi = val;}
  void          SetEtaMin(Double_t val)                     {fMinEta = val;}
  void          SetEtaMax(Double_t val)                     {fMaxEta = val;}
  void          SetPtMin(Double_t val)                      {fMinPt = val;}
  void          SetPtMax(Double_t val)                      {fMaxPt = val;}
  void          SetDistributionV2(TH2D* dist)               {fDistributionV2 = dist;}
  void          SetDistributionV3(TH2D* dist)               {fDistributionV3 = dist;}
  void          SetDistributionV4(TH2D* dist)               {fDistributionV4 = dist;}
  void          SetDistributionV5(TH2D* dist)               {fDistributionV5 = dist;}
  void          ActivateJetRemoval(const char* arrName, Double_t threshold, const char* rhoObj) {fJetRemovalArrayName = arrName; fJetRemovalPtThreshold = threshold; fJetRemovalRhoObj = rhoObj;}

  void          SetInputArrayName(const char* name)         {fInputArrayName = name;}
  void          SetOutputArrayName(const char* name)        {fOutputArrayName = name;}

protected:
  Bool_t        Run();
  void          ExecOnce();
  Double_t      AddFlow(Double_t phi, Double_t pt);

  Bool_t              fRandomizeInPhi;            
  Bool_t              fRandomizeInEta;            
  Bool_t              fRandomizeInTheta;          
  Bool_t              fRandomizeInPt;             

  Double_t            fMinPhi;                    
  Double_t            fMaxPhi;                    
  Double_t            fMinEta;                    
  Double_t            fMaxEta;                    
  Double_t            fMinPt;                     
  Double_t            fMaxPt;                     

  TH2D*               fDistributionV2;            
  TH2D*               fDistributionV3;            
  TH2D*               fDistributionV4;            
  TH2D*               fDistributionV5;            

  TString             fInputArrayName;            
  TString             fOutputArrayName;           

  TClonesArray*       fInputArray;                
  TClonesArray*       fOutputArray;               

  TString             fJetRemovalRhoObj;          
  TString             fJetRemovalArrayName;       
  TClonesArray*       fJetRemovalArray;           
  Double_t            fJetRemovalPtThreshold;     

  Double_t            fRandomPsi3;                
  Double_t            fRandomPsi4;                
  Double_t            fRandomPsi5;                
  TRandom3*           fRandom;                    

  Bool_t              IsParticleInJet(Int_t part);
  Double_t            GetExternalRho();

  ClassDef(AliAnalysisTaskParticleRandomizer, 6);
};

#endif