adding cuts for muons

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

#ifndef ALIMUONSHFHEADER_H
#define ALIMUONSHFHEADER_H

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

/* $Id$ */ 

//***********************************************************
// Class AliMuonsHFHeader
// class used to extract and store info at event level
// Author: X-M. Zhang, zhang@clermont.in2p3.fr
//                     zhangxm@iopp.ccnu.edu.cn
//***********************************************************

#include <TNamed.h>
#include <TString.h>

class TList;
class AliMCEvent;
class AliInputEventHandler;

class AliMuonInfoStoreRD;
class AliDimuInfoStoreRD;
class AliMuonInfoStoreMC;
class AliDimuInfoStoreMC;

class AliMuonsHFHeader : public TNamed {
 public :

  AliMuonsHFHeader();
  AliMuonsHFHeader(const AliMuonsHFHeader &src);
  AliMuonsHFHeader& operator=(const AliMuonsHFHeader &src);
  ~AliMuonsHFHeader();

  void GetVMC(Double_t *vtx)  const { for (Int_t i=3; i--;) vtx[i]=fVMC[i]; }
  void GetXYZ(Double_t *vtx)  const { for (Int_t i=3; i--;) vtx[i]=fVtx[i]; }
  Double_t Vx()               const { return fVtx[0]; }
  Double_t Vy()               const { return fVtx[1]; }
  Double_t Vz()               const { return fVtx[2]; }
  Double_t Vt()               const { return TMath::Sqrt(fVtx[0]*fVtx[0] + fVtx[1]*fVtx[1]); }
  Int_t VtxContrsN()          const { return fVtxContrsN; }
  TString FiredTriggerClass() const { return fFiredTriggerClass; }
  UInt_t SelectionMask()      const { return fSelMask; }
  Bool_t IsMB()               const { return fIsMB; }
  Bool_t IsMU()               const { return fIsMU; }
  Bool_t IsPileupSPD()        const { return fIsPileupSPD; }
  Float_t    Centrality()     const { return fCentrality; }
  Int_t      CentQA()         const { return fCentQA;}
  Double32_t EventPlane()     const { return fEventPlane; }
  Bool_t IsSelected();

  void SetEventInfo(AliInputEventHandler* const handler, AliMCEvent* const eventMC);

  void CreateHistograms(TList *list);
  void FillHistosEvnH(TList *list);
  void FillHistosMuon(TList *list, AliMuonInfoStoreRD* infoStore, Int_t src=-1);
  void FillHistosDimu(TList *list, AliDimuInfoStoreRD* infoStore, Int_t src=-1);

  static const char* StdBranchName()             { return fgkStdBranchName.Data();          }
  static void SetAnaMode(Int_t anaMode=0)        { fgAnaMode=anaMode;                       }
  static void SetIsMC(Int_t isMC=kFALSE)         { fgIsMC   =isMC;                          }
  static void SetSelectionCuts(Double_t cuts[5]) { for (Int_t i=5; i--;) fgCuts[i]=cuts[i]; }

 private :

  void CreateHistosEvnH(TList *list, TString sName="");
  void CreateHistosMuon(TList *list, TString sName="");
  void CreateHistosDimu(TList *list, TString sName="");

  static const TString fgkStdBranchName;  // Standard branch name
  static Int_t  fgAnaMode;                // analysis mode
  static Bool_t fgIsMC;                   // flag to use MC
  static Double_t fgCuts[5];  // 0, low limit of num. of vtx contributors
                              // 1, up limit of vz
                              // 2, up limit of vt
                              // 3, centrality max
                              // 4, centrality min

  UInt_t fSelMask;     // mask of physics selection
  Bool_t fIsMB;        // is min. bias triggered event (for real data)
  Bool_t fIsMU;        // is MUON triggered event (for real data)
  Bool_t fIsPileupSPD; // is pileup from SPD
  Double_t fVtx[3];    // position of vtx
  Double_t fVMC[3];    // position of vtx in MC
  Int_t fVtxContrsN;   // num. of contributors of vtx rec

  TString fFiredTriggerClass; // trigger class

  Float_t fCentrality;  // event centrality class
  Int_t   fCentQA;      // quality of centrality determination
  Double32_t fEventPlane; // event plane angle

  ClassDef(AliMuonsHFHeader, 7)
};

#endif