up from Chiara

[u/mrichter/AliRoot.git] / PWGJE / FlavourJetTasks / AliAnalysisTaskFlavourJetCorrelations.h

#ifndef ALIANALYSISTASKFLAVOURJETCORRELATIONS_H\r
#define ALIANALYSISTASKFLAVOURJETCORRELATIONS_H\r
/**************************************************************************\r
 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *\r
 *                                                                        *\r
 * Author: The ALICE Off-line Project.                                    *\r
 * Contributors are mentioned in the code where appropriate.              *\r
 *                                                                        *\r
 * Permission to use, copy, modify and distribute this software and its   *\r
 * documentation strictly for non-commercial purposes is hereby granted   *\r
 * without fee, provided that the above copyright notice appears in all   *\r
 * copies and that both the copyright notice and this permission notice   *\r
 * appear in the supporting documentation. The authors make no claims     *\r
 * about the suitability of this software for any purpose. It is          *\r
 * provided "as is" without express or implied warranty.                  *\r
 **************************************************************************/\r
\r
//-----------------------------------------------------------------------\r
// Author : A. Grelli,  Utrecht University\r
//          C. Bianchin, Utrecht University\r
//          X. Zhang, LBNL\r
//-----------------------------------------------------------------------\r
\r
\r
#include <TH2F.h>\r
#include "AliAODEvent.h"\r
#include "AliPicoTrack.h"\r
#include "AliAnalysisTaskEmcalJet.h"\r
\r
class TH3F;\r
class TParticle ;\r
class TClonesArray ;\r
class AliMCParticle;\r
class AliAODMCParticle;\r
class AliRDHFCuts;\r
class AliEmcalJet;\r
class AliAODRecoDecayHF;\r
class AliAODRecoCascadeHF;\r
class AliAODEvent;\r
\r
class AliAnalysisTaskFlavourJetCorrelations : public AliAnalysisTaskEmcalJet \r
{\r
\r
 public:\r
\r
  enum ECandidateType{ kD0toKpi, kDstartoKpipi };\r
\r
  AliAnalysisTaskFlavourJetCorrelations();\r
  AliAnalysisTaskFlavourJetCorrelations(const Char_t* name,AliRDHFCuts* cuts, ECandidateType candtype);\r
  virtual ~AliAnalysisTaskFlavourJetCorrelations();\r
\r
  virtual void     UserCreateOutputObjects();\r
  virtual void     UserExec(Option_t *option);\r
  virtual void     Terminate(Option_t *);\r
  virtual void     Init();\r
  virtual void     LocalInit() {Init();}\r
\r
  // inizializations\r
  Bool_t DefineHistoForAnalysis();  \r
\r
  // set MC usage\r
  void   SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}\r
  Bool_t GetMC() const {return fUseMCInfo;}\r
  // set usage of reconstructed tracks\r
  void   SetUseReco(Bool_t reco) {fUseReco=reco;}\r
  Bool_t GetUseReco() {return fUseReco;}\r
  \r
  \r
  void SetMassLimits(Double_t range, Int_t pdg);\r
  void SetMassLimits(Double_t lowlimit, Double_t uplimit);\r
\r
  //jet reconstruction algorithm\r
  void SetJetArrayName(TString jetArrName) {fJetArrName=jetArrName;};\r
  TString GetJetArrayName() const {return fJetArrName;};\r
\r
  // trigger on jet events\r
  void SetTriggerOnLeadingJet(Bool_t triggerOnLeadingJet) {fLeadingJetOnly=triggerOnLeadingJet;};\r
  Bool_t GetTriggerOnLeadingJet() const {return fLeadingJetOnly;}\r
\r
\r
  // Array of D0 width for the Dstar\r
  Bool_t SetD0WidthForDStar(Int_t nptbins,Float_t* width);\r
\r
  //Bool_t   FillMCDJetInfo(AliPicoTrack *jetTrk,AliEmcalJet* jet, TClonesArray *mcArray,Double_t ptjet);\r
  void FillHistogramsRecoJetCorr(AliVParticle* candidate, AliEmcalJet *jet);\r
  void FillHistogramsD0JetCorr(AliAODRecoDecayHF* candidate, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj, Double_t deltaR, AliAODEvent* aodEvent);\r
\r
  void FillHistogramsDstarJetCorr(AliAODRecoCascadeHF* dstar, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj,Double_t deltaR);\r
  void FillHistogramsMCGenDJetCorr(Double_t dPhi, Double_t z,Double_t ptD,Double_t ptjet,Double_t deltaR);\r
  void SideBandBackground(AliAODRecoCascadeHF *candDstar, AliEmcalJet *jet);\r
  void MCBackground(AliAODRecoDecayHF *candbg, AliEmcalJet *jet);\r
  void FillMassHistograms(Double_t mass,Double_t ptD, Double_t deltaR);\r
  void FlagFlavour(AliVParticle* charm, AliEmcalJet* jet);\r
  \r
 private:\r
  \r
  AliAnalysisTaskFlavourJetCorrelations(const AliAnalysisTaskFlavourJetCorrelations &source);\r
  AliAnalysisTaskFlavourJetCorrelations& operator=(const AliAnalysisTaskFlavourJetCorrelations& source); \r
\r
  Double_t Z(AliVParticle* part,AliEmcalJet* jet) const;\r
  Float_t DeltaR(AliVParticle *p1, AliVParticle *p2) const;\r
\r
\r
  Bool_t fUseMCInfo;             //  Use MC info\r
  Bool_t fUseReco;               // use reconstructed tracks when running on MC\r
  Int_t  fCandidateType;         // Dstar or D0\r
  Int_t  fPDGmother;             // PDG code of D meson\r
  Int_t  fNProngs;               // number of prong of the decay channel  \r
  Int_t  fPDGdaughters[4];       // PDG codes of daughters\r
  Float_t fSigmaD0[30];          //\r
  TString fBranchName;           // AOD branch name\r
  TList *fmyOutput;                //! user output\r
  AliRDHFCuts *fCuts;            // Cuts \r
\r
  Double_t fMinMass;             // mass lower limit histogram\r
  Double_t fMaxMass;             // mass upper limit histogram\r
\r
  TString  fJetArrName;          // name of the jet array, taken from the task running the jet finder\r
  TString fCandArrName;          // string which correspond to the candidate type\r
  Bool_t fLeadingJetOnly;        // use only the leading jet in the event to make the correlations\r
  Double_t fJetRadius;           // jet radius (filled from the JetContainer)\r
\r
  ClassDef(AliAnalysisTaskFlavourJetCorrelations,2); // class for charm-jet correlations\r
};\r
\r
#endif\r