[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpid.h

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/
#ifndef __ALIHFEPID_H__
#define __ALIHFEPID_H__

#ifndef ROOT_TObject
#include <TObject.h>
#endif

class AliHFEpidBase;
class AliESDtrack;
class AliVParticle;
class AliMCEvent;

class TList;

class AliHFEpid : public TObject{
  enum{
    kIsQAOn = BIT(14),
    kHasMCData = BIT(15)
  };
  enum{
    kMCpid = 0,
    kESDpid = 1,
    kTPCpid = 2,
    kTRDpid = 3,
    kTOFpid = 4,
    kNdetectorPID = 5
  };
  public:
    AliHFEpid();
    AliHFEpid(const AliHFEpid &c);
    AliHFEpid &operator=(const AliHFEpid &c);
    ~AliHFEpid();
    
    Bool_t InitializePID(TString detectors);
    Bool_t IsSelected(AliVParticle *track);
    void SetMCEvent(AliMCEvent *mc);

    Bool_t IsQAOn() const { return TestBit(kIsQAOn); };
    Bool_t HasMCData() const { return TestBit(kHasMCData); };
    void SetQAOn();
    void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
    TList *GetQAhistograms() const { return fQAlist; };

  protected:
    Bool_t MakePID_TPC_TOF(AliESDtrack *track);
  private:
    AliHFEpidBase *fDetectorPID[kNdetectorPID];    //! Detector PID classes
    UInt_t fEnabledDetectors;             // Enabled Detectors
    TList *fQAlist;                       //! QA histograms

  ClassDef(AliHFEpid, 1)      // Steering class for Electron ID
};

#endif
Commit	Line	Data
809a4336	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	#ifndef __ALIHFEPID_H__
	16	#define __ALIHFEPID_H__
	17
	18	#ifndef ROOT_TObject
	19	#include <TObject.h>
	20	#endif
	21
	22	class AliHFEpidBase;
	23	class AliESDtrack;
	24	class AliVParticle;
	25	class AliMCEvent;
	26
	27	class TList;
	28
	29	class AliHFEpid : public TObject{
	30	enum{
	31	kIsQAOn = BIT(14),
	32	kHasMCData = BIT(15)
	33	};
	34	enum{
	35	kMCpid = 0,
	36	kESDpid = 1,
	37	kTPCpid = 2,
	38	kTRDpid = 3,
	39	kTOFpid = 4,
	40	kNdetectorPID = 5
	41	};
	42	public:
	43	AliHFEpid();
	44	AliHFEpid(const AliHFEpid &c);
	45	AliHFEpid &operator=(const AliHFEpid &c);
	46	~AliHFEpid();
	47
	48	Bool_t InitializePID(TString detectors);
	49	Bool_t IsSelected(AliVParticle *track);
	50	void SetMCEvent(AliMCEvent *mc);
	51
	52	Bool_t IsQAOn() const { return TestBit(kIsQAOn); };
	53	Bool_t HasMCData() const { return TestBit(kHasMCData); };
	54	void SetQAOn();
	55	void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
	56	TList *GetQAhistograms() const { return fQAlist; };
	57
	58	protected:
	59	Bool_t MakePID_TPC_TOF(AliESDtrack *track);
	60	private:
	61	AliHFEpidBase *fDetectorPID[kNdetectorPID]; //! Detector PID classes
	62	UInt_t fEnabledDetectors; // Enabled Detectors
	63	TList *fQAlist; //! QA histograms
	64
65	ClassDef(AliHFEpid, 1) // Steering class for Electron ID
66	};
67
68	#endif