Line data    Source code

       1             : #ifndef ALI_DETECTOR_PID_H
       2             : #define ALI_DETECTOR_PID_H
       3             : /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
       4             :  * See cxx source for full Copyright notice                               */
       5             : 
       6             : //---------------------------------------------------------------//
       7             : //      Class to store raw probabilities and nsigmas             //
       8             : //        of all detectors                                       //
       9             : //                                                               //
      10             : //                                                               //
      11             : //   Origin: Jens Wiechula, Uni Tuebingen, jens.wiechula@cern.ch //
      12             : //---------------------------------------------------------------//
      13             : 
      14             : #include <TObject.h>
      15             : #include <TClonesArray.h>
      16             : 
      17             : #include "AliPID.h"
      18             : #include "AliPIDResponse.h"
      19             : 
      20             : class AliDetectorPID : public TObject {
      21             : public:
      22             :   AliDetectorPID();
      23             :   AliDetectorPID(const AliDetectorPID &pid);
      24             :   virtual ~AliDetectorPID();
      25             :   AliDetectorPID& operator= (const AliDetectorPID &pid);
      26             :   
      27             :   void SetRawProbability(AliPIDResponse::EDetector det, const Double_t prob[], Int_t nspecies, AliPIDResponse::EDetPidStatus status);
      28             :   void SetNumberOfSigmas(AliPIDResponse::EDetector det, const Double_t nsig[], Int_t nspecies, AliPIDResponse::EDetPidStatus status=AliPIDResponse::kDetPidOk);
      29             : 
      30             :   AliPIDResponse::EDetPidStatus GetRawProbability(AliPIDResponse::EDetector det, Double_t prob[], Int_t nspecies) const;
      31             :   AliPIDResponse::EDetPidStatus GetNumberOfSigmas(AliPIDResponse::EDetector det, Double_t nsig[], Int_t nspecies) const;
      32             :   
      33             :   Double_t GetRawProbability(AliPIDResponse::EDetector det, AliPID::EParticleType type) const;
      34             :   Double_t GetNumberOfSigmas(AliPIDResponse::EDetector det, AliPID::EParticleType type) const;
      35             : 
      36             :   AliPIDResponse::EDetPidStatus GetRawProbability(AliPIDResponse::EDetector det, AliPID::EParticleType type, Double_t &prob) const;
      37             :   AliPIDResponse::EDetPidStatus GetNumberOfSigmas(AliPIDResponse::EDetector det, AliPID::EParticleType type, Double_t &nsig) const;
      38             :   
      39           0 :   Bool_t HasRawProbability(AliPIDResponse::EDetector det) const { return fArrRawProbabilities.UncheckedAt((Int_t)det)!=0x0; }
      40           0 :   Bool_t HasNumberOfSigmas (AliPIDResponse::EDetector det) const { return fArrNsigmas.UncheckedAt((Int_t)det)!=0x0;          }
      41             : 
      42             :   AliPIDResponse::EDetPidStatus GetPIDStatus(AliPIDResponse::EDetector det) const;
      43             : private:
      44             :   TClonesArray fArrNsigmas;          // array to store nsigma values of all detectors
      45             :   TClonesArray fArrRawProbabilities; // array to strore raw probabilities of all detectors
      46             : 
      47         176 :   ClassDef(AliDetectorPID,1);        //Store raw probabilities and nsigmas for all detectors
      48             : };
      49             : 
      50             : #endif
      51             :