AliJetModelBaseTask.h

Go to the documentation of this file.

#ifndef ALIJETMODELBASETASK_H
#define ALIJETMODELBASETASK_H

class TClonesArray;
class AliEMCALGeometry;
class AliVCluster;
class AliPicoTrack;
class AliVCaloCells;
class AliAODMCParticle;
class AliNamedArrayI;
class TF2;
class AliEmcalPythiaInfo;

#include <TH1F.h>
#include <TF1.h>

#include "AliAnalysisTaskSE.h"

class AliJetModelBaseTask : public AliAnalysisTaskSE {
 public:
  AliJetModelBaseTask();
  AliJetModelBaseTask(const char *name, Bool_t drawqa=kFALSE); 
  virtual ~AliJetModelBaseTask();

  void                   SetEtaRange(Float_t min, Float_t max) { fEtaMin       = min;  fEtaMax = max; }
  void                   SetPhiRange(Float_t min, Float_t max) { fPhiMin       = min;  fPhiMax = max; }
  void                   SetPtRange(Float_t min, Float_t max)  { fPtMin        = min;  fPtMax  = max; }
  void                   SetGenType(Int_t gentype)            {fGenType      = gentype;}
  void                   SetPtSpectrum(TH1F *f)                { fPtSpectrum   = f;    }
  void                   SetPtSpectrum(TF1 *f)                 { fPtSpectrum   = new TH1F("ptSpectrum","ptSpectrum",1000,f->GetXmin(),f->GetXmax()); 
                                                                 fPtSpectrum->Add(f); }
  void                   SetPtPhiEvPlDistribution(TF2 *f)      { fPtPhiEvPlDistribution   = f;    }
  void                   SetDensitySpectrum(TH1F *f)           { fDensitySpectrum = f;    }
  void                   SetDensitySpectrum(TF1 *f)            { fDensitySpectrum = new TH1F("densitypectrum","densitypectrum",1000,f->GetXmin(),f->GetXmax()); 
                                                                 fDensitySpectrum->Add(f); }
  void                   SetMassDistribution(TH1F *hM);
  void                   SetMassDistributionFromFile(TString filename, TString histoname);
  void            SetpTDistributionFromFile(TString filename, TString histoname);
  void                   SetMassVsPtDistributionFromFile(TString filename, TString histoname);
  void                  SetMassAndPtDistributionFromFile(TString filenameM, TString filenamepT, TString histonameM, TString histonamepT);
  void                   SetMassVsPtDistribution(TH2F *hmasspt);
  void                   SetDistributionFromFile(TString filename, TString histoname, Int_t type);
  
  void                   SetDifferentialV2(TF1* f)             { fDifferentialV2 = f;  }
  void                   SetAddV2(Bool_t b)                    { fAddV2 = b;           }
  void                   SetAddFlowFluctuations(Bool_t b)      { fFlowFluctuations = b;}
  void                   SetMC(Bool_t a)                       { fIsMC         = a   ; }
  void                   SetCopyArray(Bool_t copy)             { fCopyArray    = copy; }
  void                   SetTracksName(const char *n)          { fTracksName   = n;    }
  void                   SetClusName(const char *n)            { fCaloName     = n;    }
  void                   SetCellsName(const char *n)           { fCellsName    = n;    }
  void                   SetMCParticlesName(const char *n)     { fMCParticlesName = n; }
    void                 SetPythiaInfoName(const char *n)      { fPythiaInfoName = n; }
  void                   SetSuffix(const char *s)              { fSuffix       = s;    }
  void                   SetGeometryName(const char *n)        { fGeomName     = n;    }
  void                   SetMarkMC(Int_t m)                    { fMarkMC       = m;    }
  virtual void           SetNClusters(Int_t n)                 { fNClusters    = n;    }
  virtual void           SetNCells(Int_t n)                    { fNCells       = n;    }
  virtual void           SetNTracks(Int_t n)                   { fNTracks      = n;    }
  
  TString                GetOutTrackName() const;
 protected:
  void                   UserExec(Option_t* /*option*/);
  void                   UserCreateOutputObjects();
  Int_t                  SetNumberOfOutCells(Int_t n);                                          
  Int_t                  AddCell(Double_t e = -1, Double_t eta = -999, Double_t phi = -1);      
  Int_t                  AddCell(Double_t e, Int_t absId, Double_t time = 0, Int_t label=0);    
  AliVCluster           *AddCluster(Double_t e = -1, Double_t eta = -999, Double_t phi = -1, Int_t label=0); 
  AliVCluster           *AddCluster(Double_t e, Int_t absId, Int_t label=0);                    
  AliVCluster           *AddCluster(AliVCluster *oc);                                           
  AliPicoTrack          *AddTrack(Double_t pt = -999, Double_t eta = -999, Double_t phi = -999, Byte_t type=0, 
          Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=kFALSE, 
          Int_t label=0, Short_t charge=1, Double_t mass = 0.1396);          
  AliAODMCParticle      *AddMCParticle(AliAODMCParticle *part, Int_t origIndex);                // add a MC particle
  void                   AddV2(Double_t &phi, Double_t &pt) const;
  void                   CopyCells();
  void                   CopyClusters();
  void                   CopyTracks();
  void                   CopyMCParticles();
  void                   GetRandomCell(Double_t &eta, Double_t &phi, Int_t &absId);             
  Double_t               GetRandomEta(Bool_t emcal=kFALSE);                                     
  Double_t               GetRandomPhi(Bool_t emcal=kFALSE);                                     
  Double_t               GetRandomPt();                                                         
  Double_t               GetRandomM();                                                         
  void                   GetRandomParticle(Double_t &pt, Double_t &eta, Double_t &phi, Bool_t emcal=kFALSE);  
  void                   GetRandomMassiveParticle(Double_t &pt, Double_t &eta, Double_t &phi, Bool_t emcal, Double_t& m);   
  void                   GetRandomMvsPt(Double_t &m, Double_t &pt);                             
  void                   GetRandomMvsPtParticle(Double_t &pt, Double_t &m, Double_t &eta, Double_t &phi, Bool_t emcal = kFALSE);      
  
  virtual Bool_t         ExecOnce();                                                            
  virtual void           Run();                                                                 
  void                   FillHistograms();  
  
  
  
  TString                fGeomName;               
  TString                fTracksName;             
  TString                fOutTracksName;          
  TString                fCaloName;               
  TString                fOutCaloName;            
  TString                fCellsName;              
  TString                fOutCellsName;           
  TString                fMCParticlesName;        
  TString                fOutMCParticlesName;     
  TString                fPythiaInfoName;         
  Bool_t                 fIsMC;                   
  TString                fSuffix;                 
  Float_t                fEtaMin;                 
  Float_t                fEtaMax;                 
  Float_t                fPhiMin;                 
  Float_t                fPhiMax;                 
  Float_t                fPtMin;                  
  Float_t                fPtMax;                  
  Int_t                  fGenType;               
  Bool_t                 fCopyArray;               
  Int_t                  fNClusters;              
  Int_t                  fNCells;                 
  Int_t                  fNTracks;                
  Int_t                  fMarkMC;                 
  TH1F                  *fPtSpectrum;             
  TF2                   *fPtPhiEvPlDistribution;  
  TH1F                  *fDensitySpectrum;        
  TF1                   *fDifferentialV2;         
  Bool_t                 fAddV2;                  
  Bool_t                 fFlowFluctuations;       
  Bool_t                 fQAhistos;               
  Double_t               fPsi;                    
  Bool_t                 fIsInit;                 
  AliEMCALGeometry      *fGeom;                   
  Double_t               fVertex[3];              
  TClonesArray          *fClusters;               
  TClonesArray          *fOutClusters;            
  TClonesArray          *fTracks;                 
  TClonesArray          *fOutTracks;              
  AliVCaloCells         *fCaloCells;              
  AliVCaloCells         *fOutCaloCells;           
  Int_t                  fAddedCells;             
  TClonesArray          *fMCParticles;            
  AliNamedArrayI        *fMCParticlesMap;         
  TClonesArray          *fOutMCParticles;         
  AliNamedArrayI        *fOutMCParticlesMap;      
  Int_t                  fMCLabelShift;           
  Bool_t                 fEsdMode;                
  TList                 *fOutput;                 
  AliEmcalPythiaInfo    *fPythiaInfo;             
  TH1F                  *fhpTEmb  ;               
  TH1F                  *fhMEmb   ;               
  TH1F                  *fhEtaEmb ;               
  TH1F                  *fhPhiEmb ;               
  TH1I                  *fhEvents ;               
  Bool_t                 fMassFromDistr;          
  TH1F*                  fHMassDistrib;           
  TH2F*                  fHMassPtDistrib;         
 private:
  AliJetModelBaseTask(const AliJetModelBaseTask&);            // not implemented
  AliJetModelBaseTask &operator=(const AliJetModelBaseTask&); // not implemented

  ClassDef(AliJetModelBaseTask, 13) // Jet modeling task
};
#endif