expand particle map when needed

[u/mrichter/AliRoot.git] / PWG / muon / AliAnalysisTaskSEMuonsHF.h

#ifndef ALIANALYSISTASKSEMUONSHF_H
#define ALIANALYSISTASKSEMUONSHF_H

/* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */ 

//*************************************************************************
// Class AliAnalysisTaskSEMuonsHF
// AliAnalysisTaskSE for the single muon and dimuon from HF analysis
// Author: X-M. Zhang, zhang@clermont.in2p3.fr
//                     zhangxm@iopp.ccnu.edu.cn
//*************************************************************************

#include "AliAnalysisTaskSE.h"

class TString;
class TList;
class TClonesArray;
class AliMuonsHFHeader;
class AliMuonTrackCuts;
class AliMuonPairCuts;

class AliAnalysisTaskSEMuonsHF : public AliAnalysisTaskSE {
 public:

  AliAnalysisTaskSEMuonsHF();
  AliAnalysisTaskSEMuonsHF(const char *name, const AliMuonTrackCuts& cutsMuon, const AliMuonPairCuts& cutsDimu);
  virtual ~AliAnalysisTaskSEMuonsHF();

  virtual void Init();
  virtual void LocalInit() { Init(); }
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *opt);
  virtual void Terminate(Option_t *opt);
  virtual void NotifyRun();

  void SetAnaMode(Int_t mode)      { fAnaMode      = ((mode>=0 && mode<3) ? mode : 0); }
  void SetIsOutputTree(Bool_t ist) { fIsOutputTree = ist;                              }
  void SetUseMC(Bool_t isMC)       { fIsMC         = isMC;                             }

  void SetEvsHCuts(Double_t cuts[5])  const { AliMuonsHFHeader::SetSelectionCuts(cuts);   }

 private:

  AliAnalysisTaskSEMuonsHF(const AliAnalysisTaskSEMuonsHF&);
  AliAnalysisTaskSEMuonsHF& operator=(const AliAnalysisTaskSEMuonsHF&);

  Int_t fAnaMode;        // = 0, ana both single muon and dimuon
                         // = 1, ana single muon
                         // = 2, ana dimuon
  Bool_t fIsOutputTree;  // flag used to switch on/off tree output
  Bool_t fIsMC;          // flag of whether the input is MC

  AliMuonTrackCuts *fCutsMuon;  // single muon selection cuts
  AliMuonPairCuts  *fCutsDimu;  // dimuon selection cuts

  AliMuonsHFHeader *fHeader;  // output for info at ev level
  TClonesArray  *fMuonClArr;  // output clones array for single mu
  TClonesArray  *fDimuClArr;  // output clones array for dimu
  TList *fListOutput;         // output list of histos

  ClassDef(AliAnalysisTaskSEMuonsHF, 7);
};

#endif