Small changes and fixes (Carmelo)

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEDStar.h

#ifndef ALIANALYSISTASKSEDSTAR_H\r
#define ALIANALYSISTASKSEDSTAR_H\r
\r
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
//*************************************************************************\r
// Class AliAnalysisTaskSEDStar\r
// AliAnalysisTaskSE for D* candidates invariant mass histogram\r
// and comparison to MC truth (kinematics stored in the AOD)\r
// Authors: Y.wang, yifei@physi.uni-heidelberg.de\r
//*************************************************************************\r
\r
#include <TROOT.h>\r
#include <TSystem.h>\r
#include <TH1F.h>\r
\r
#include "AliAnalysisTaskSE.h"\r
#include "AliAnalysisVertexingHF.h"\r
\r
class AliAnalysisTaskSEDStar : public AliAnalysisTaskSE\r
{\r
 public:\r
\r
  AliAnalysisTaskSEDStar();\r
  AliAnalysisTaskSEDStar(const char *name);\r
  virtual ~AliAnalysisTaskSEDStar();\r
\r
\r
  // Implementation of interface methods\r
  virtual void UserCreateOutputObjects();\r
  virtual void Init();\r
  virtual void LocalInit() {Init();}\r
  virtual void UserExec(Option_t *option);\r
  virtual void Terminate(Option_t *option);\r
\r
  void	       SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}\r
  void         PIDon() {fPID=kTRUE;}\r
  void         PIDoff() {fPID=kFALSE;}\r
  Bool_t       GetPIDStatus() const {return fPID;}\r
  void         SetNSigmaTPC(Double_t nsigma) {fNSigma=nsigma;}\r
  Double_t     GetNSigmaTPC() const {return fNSigma;}\r
 private:\r
\r
  AliAnalysisTaskSEDStar(const AliAnalysisTaskSEDStar &source);\r
  AliAnalysisTaskSEDStar& operator=(const AliAnalysisTaskSEDStar& source); \r
  TList   *fOutput; //! list send on output slot 0\r
\r
  TH1F    *fhistMass;  //! output total invariant mass histogram - no MC truth\r
  TH1F    *fhistSgn;  //! output signal invariant mass histogram - use cuts\r
  TH1F    *fhistBkg;  //! output background invariant mass histogram - use cuts\r
  Bool_t   fReadMC;  // flag for MC array: kTRUE = read it, kFALSE = do not read it\r
  Bool_t   fPID;  //flag of TPCpid selection\r
  Double_t fNSigma;   //nsigma selection for TPCpid\r
\r
  AliAnalysisVertexingHF *fVHF;  // Vertexer heavy flavour (used to pass the cuts)\r
  Bool_t SelectTPCPID(AliAODTrack *trk, Int_t pid, Double_t nsig);\r
\r
  ClassDef(AliAnalysisTaskSEDStar,1); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates\r
};\r
\r
#endif\r
\r