PWG2/RESONANCES/AliRsnPIDWeightsMgr.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  * See cxx source for full Copyright notice                               *
   4  **************************************************************************/
   5
   6 /* $Id: AliRsnPIDWeightsMgr.h,v 1.5 2007/02/21 14:33:25 pulvir Exp $ */
   7
   8 //-------------------------------------------------------------------------
   9 //                      Class AliRsnPIDWeightsMgr
  10 //  Simple collection of reconstructed tracks, selected from an ESD event
  11 //
  12 // author: A. Pulvirenti             (email: alberto.pulvirenti@ct.infn.it)
  13 //-------------------------------------------------------------------------
  14
  15 #ifndef AliRsnPIDWeightsMgr_H
  16 #define AliRsnPIDWeightsMgr_H
  17
  18 #include "AliRsnPID.h"
  19
  20 class AliRsnPIDWeightsMgr : public TObject
  21 {
  22 public:
  23
  24     // detectors for customizing PID weights
  25     enum EDetector {
  26         kITS,
  27         kTPC,
  28         kTRD,
  29         kTOF,
  30         kHMPID,
  31         kDetectors
  32     };
  33
  34     AliRsnPIDWeightsMgr();
  35     virtual ~AliRsnPIDWeightsMgr() {}
  36
  37     void      UseDetector(EDetector det, Bool_t use) {if (CheckBounds(det)) fUseDet[det] = use;}
  38     void      SetDetectorWeights(EDetector det, Double_t *weights);
  39     void      SetAcceptanceRange(EDetector det, Double_t ptmin, Double_t ptmax);
  40     Double_t  GetWeight(AliRsnPID::EType type, Double_t pt);
  41     Double_t* GetWeightArray(EDetector det) {if (CheckBounds(det)) return fWeights[det]; else return 0x0;}
  42
  43 private:
  44
  45     Bool_t   CheckBounds(EDetector det);
  46
  47     Bool_t   fUseDet[kDetectors];                        // flag to switch off info from a detector
  48     Double_t fDetPtMin[kDetectors];                      // min value for detector weight acceptance
  49     Double_t fDetPtMax[kDetectors];                      // max value for detector weight acceptance
  50     Double_t fWeights[kDetectors][AliRsnPID::kSpecies];  // PID weights of a single detector
  51
  52     ClassDef(AliRsnPIDWeightsMgr,1);
  53 };
  54
  55 #endif