AliHFsubtractBFDcuts.h

Go to the documentation of this file.

#ifndef ALIHFSUBTRACTBFDCUTS_H
#define ALIHFSUBTRACTBFDCUTS_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.                  *
 **************************************************************************/

#include <vector>

#include "TNamed.h"
#include "THnSparse.h"

#include "AliAODRecoDecayHF2Prong.h"
#include "AliAODMCParticle.h"

class TList;
class TClonesArray;
class AliAODMCHeader;
class AliAODEvent;
class AliAODVertex;
class AliNeutralTrackParam;

class AliHFsubtractBFDcuts : public TNamed{

public:
  AliHFsubtractBFDcuts();
  AliHFsubtractBFDcuts(const char* name,const char* title);
  ~AliHFsubtractBFDcuts();

  void InitHistos();

  void FillGenStep(AliAODMCParticle* dzeroMC,Double_t pt=-1,Double_t weight=1.,TClonesArray* mcArray=0x0, AliAODMCHeader* mcHeader=0x0);
  THnSparseF* GetSparseData() const {return fTHnData;}
  THnSparseF* GetSparseMC()   const {return fTHnMC;}
  THnSparseF* GetSparseMCgen()     const {return fTHnGenStep;}

  void SetHistoPtMCgen(THnSparseF* h){if(fTHnGenStep)delete fTHnGenStep;fTHnGenStep=(THnSparseF*)h->Clone();return;}
  void SetSparseData(THnSparseF* h){if(fTHnData)delete fTHnData;fTHnData=(THnSparseF*)h->Clone();return;}
  void SetSparseMC(THnSparseF* h){if(fTHnMC)delete fTHnMC;fTHnMC=(THnSparseF*)h->Clone();return;}

  void SetFillMC (Bool_t fillMC = kTRUE) {fIsMC = fillMC;}

  void FillSparses(AliAODRecoDecayHF2Prong* dzeroPart,Int_t isSelected,Double_t pt=-1,Double_t massD0=-1,Double_t massD0bar=-1,Double_t weight=1.,TClonesArray* mcArray=0x0, AliAODEvent* aodEvent=0x0, AliAODMCHeader* mcHeader=0x0);
  TList* GetDecayStrings() { return fDecayStrList; }
  TList* GetQAhists() { return fQAhists; }

private:
  AliHFsubtractBFDcuts(const AliHFsubtractBFDcuts& c);
  AliHFsubtractBFDcuts operator=(const AliHFsubtractBFDcuts& c);

  Bool_t GetCandidateLabel();
  Bool_t AnalyseDecay(Bool_t generateString, Bool_t mcOnly);  
  void   CountProngs(Int_t labCurrMother, Int_t labCurrExcl, Bool_t generateString, Bool_t mConly);
  Bool_t IsStable(Int_t labProng);             
  Bool_t IsInAcceptance(Int_t labProng) const; 
  AliAODVertex* RecBvtx(TObjArray* tracks) const; 
  Bool_t CheckBhypothesis(Int_t iAODtrack, Bool_t Bprong); 

  Bool_t      fIsMC;              
  Bool_t      fCheckAcceptance;   
  Bool_t      fResolveResonances; 
  THnSparseF* fTHnGenStep;        
  THnSparseF* fTHnData;           
  THnSparseF* fTHnMC;             
  TList*      fQAhists;           

  TClonesArray*             fMCarray;      
  TClonesArray*             fAODtracks;    
  AliAODVertex*             fPriVtx;       
  Double_t                  fBkG;          
  AliAODRecoDecayHF2Prong*  fD0Cand;       
  AliNeutralTrackParam*     fD0CandParam;  
  Int_t                     fLabCand;      
  Double_t                  fPtCand;       
  Int_t                     fLabMother;    
  UInt_t                    fNprongs;      
  UInt_t                    fNprongsInAcc; 
  Bool_t                    fFoundElectron;
  Bool_t                    fDecayChain;   
  Double_t                  fMotherPt;     
  Bool_t             fGenerateDecayList;   
  std::vector<Int_t> fDecayProngs;         
  TList*             fDecayStrList;        

  ClassDef(AliHFsubtractBFDcuts,10);
};

#endif