[u/mrichter/AliRoot.git] / CORRFW / AliCFTrackCutPid.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.                  *
 **************************************************************************/

//
// author : A. Mastroserio
//

#ifndef ALICFTRACKCUTPID_H
#define ALICFTRACKCUTPID_H

#include "AliCFCutBase.h"
#include "AliPID.h"
#include "AliESDtrack.h"
#include "AliAODTrack.h"
#include <TString.h>
#include <TObject.h>
#include <TH1F.h>
#include <TF1.h>
//__________________________________________________________________________________
// CUT ON TRACK PID
//__________________________________________________________________________________
class AliESDtrack;

class AliCFTrackCutPid : public AliCFCutBase
{
  public :
    AliCFTrackCutPid() ;
  AliCFTrackCutPid(const Char_t* name, const Char_t* title) ;
  AliCFTrackCutPid(const AliCFTrackCutPid& c) ;
  AliCFTrackCutPid& operator=(const AliCFTrackCutPid& c) ;
  
  virtual ~AliCFTrackCutPid();
  
  enum EDetType {kITS = 0, kTPC, kTRD,  kTOF, kHMPID=4, kNoDet=-11};
  enum EDetNum {kNdets=5};
  enum ENoId {kCheckProb = -10, kCheckResp = -11, kDetRestr = -12};
  
  // Setters 
  
  void SetDetectors(TString dets);                            
  void SetPriors(Double_t r[AliPID::kSPECIES]);                    
  void SetProbabilityCut(Double_t cut) {fCut=cut;}                  
  void SetParticleType(Int_t iType, Bool_t tocombine) {fgParticleType=iType; fgIsComb=tocombine;} 
  void SetMinDiffResp(Bool_t check, Double_t mindiff) {fCheckResponse=check; fMinDiffResponse=mindiff;}  
  void SetMinDiffProb(Bool_t check, Double_t mindiff) {fCheckSelection=check; fMinDiffProbability=mindiff;} 
  void SetPriorFunctions(TF1 *func[AliPID::kSPECIES]);
  void SetANDstatus(TString dets);
  void SetDetectorProbabilityRestriction(TString det, Int_t iPart, Double_t upperprob); 
  void SetHistogramAxis(Int_t nbins, Double_t xmin, Double_t xmax) {fNbins=nbins; fXmin = xmin; fXmax = xmax;}
  void SetAODmode(Bool_t isaod = kFALSE) {fgIsAOD=isaod;}  
 
  //loads the track detector responses and the track status
  void TrackInfo(const AliESDtrack *pTrk,ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;  
  
  //identifies the track
  Int_t Identify(Double_t pid[AliPID::kSPECIES]) const;
  
  //identifies the track filling the QA histograms                                                          
  Int_t IdentifyQA(const Double_t pid[AliPID::kSPECIES],Int_t idets) const;
  
  //returns the track identification number  
  Int_t GetID(ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;  
  //returns the track identification number in caso of an AliAODTrack
  Int_t GetAODID(AliAODTrack *aodtrack) const;
  
 
  //main 
  virtual Bool_t IsSelected(TObject *track); 
  Bool_t IsSelected(TList* /*list*/) {return kTRUE;}  
  //histos are added to a list
  void AddQAHistograms(TList *qalist);
  
  
 private:
  void SetPPriors(AliESDtrack *pTrk);                          
  ULong_t StatusForAND(ULong_t status[kNdets+1]) const; //
  void InitialiseHisto();
  void DefineHistograms();   //histo booking  
  Bool_t Check(const Double_t *p, Int_t iPsel, Double_t minDiff) const;
  void CombPID(ULong_t status[kNdets+1],Double_t pid[kNdets+1][AliPID::kSPECIES],Double_t *combpid) const;
  
  Double_t fCut;                                            //probability cut
  Double_t fMinDiffResponse;                                // minimum difference between detector resposes
  Double_t fMinDiffProbability;                             // minimum difference between probability values
  Int_t fgParticleType;                                     // requested particle type
  Bool_t fgIsComb;                                          // flag for the combined pid
  Bool_t fgIsAOD;                                            // flag for the AOD QA histograms
  Bool_t fCheckResponse;                                    // flag to check the minimum difference of det responsess
  Bool_t fCheckSelection;                                   // flag to check the minimum difference of probabilities
  Bool_t fIsPpriors;                                        //flag for momentum dependent priors
  Bool_t fIsDetAND;                                         //flag for AND with multiple detectors
  Double_t fXmin;                                           //x min QA histo
  Double_t fXmax;                                           //x max QA histo
  Int_t fNbins;                                             //n bins QA histo 
  Int_t fDetRestr;                                          //id of the detector for the restriction
  Int_t fiPartRestr;                                        //id of the particle for the restriction
  Double_t fDetProbRestr;                                   //probability restriction value

  Double_t fPriors[AliPID::kSPECIES];                       //a priori concentrations
  TF1 *fPriorsFunc[AliPID::kSPECIES];                       //momentum dependent priors
  Bool_t fDets[kNdets];                                     //boolean(s) corresponding to the chosen detector(s) 
  Bool_t fDetsInAnd[kNdets];                                //detector to be in AND for the combined PID
  TH1F *fhResp[kNdets][AliPID::kSPECIES];                   //QA histo
  TH1F *fhProb[kNdets][AliPID::kSPECIES];                   //QA histo
  TH1F *fhCombResp[AliPID::kSPECIES];                       //QA histo
  TH1F *fhCombProb[AliPID::kSPECIES];                       //QA histo
  
  ClassDef(AliCFTrackCutPid,1);
};
#endif
Commit	Line	Data
563113d0	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
	16	//
	17	// author : A. Mastroserio
	18	//
	19
	20	#ifndef ALICFTRACKCUTPID_H
	21	#define ALICFTRACKCUTPID_H
	22
	23	#include "AliCFCutBase.h"
	24	#include "AliPID.h"
	25	#include "AliESDtrack.h"
bc6176fb	26	#include "AliAODTrack.h"
563113d0	27	#include <TString.h>
	28	#include <TObject.h>
	29	#include <TH1F.h>
	30	#include <TF1.h>
	31	//__________________________________________________________________________________
	32	// CUT ON TRACK PID
	33	//__________________________________________________________________________________
	34	class AliESDtrack;
	35
	36	class AliCFTrackCutPid : public AliCFCutBase
	37	{
	38	public :
	39	AliCFTrackCutPid() ;
	40	AliCFTrackCutPid(const Char_t* name, const Char_t* title) ;
	41	AliCFTrackCutPid(const AliCFTrackCutPid& c) ;
	42	AliCFTrackCutPid& operator=(const AliCFTrackCutPid& c) ;
	43
	44	virtual ~AliCFTrackCutPid();
	45
	46	enum EDetType {kITS = 0, kTPC, kTRD, kTOF, kHMPID=4, kNoDet=-11};
	47	enum EDetNum {kNdets=5};
	48	enum ENoId {kCheckProb = -10, kCheckResp = -11, kDetRestr = -12};
	49
	50	// Setters
	51
	52	void SetDetectors(TString dets);
	53	void SetPriors(Double_t r[AliPID::kSPECIES]);
	54	void SetProbabilityCut(Double_t cut) {fCut=cut;}
	55	void SetParticleType(Int_t iType, Bool_t tocombine) {fgParticleType=iType; fgIsComb=tocombine;}
	56	void SetMinDiffResp(Bool_t check, Double_t mindiff) {fCheckResponse=check; fMinDiffResponse=mindiff;}
	57	void SetMinDiffProb(Bool_t check, Double_t mindiff) {fCheckSelection=check; fMinDiffProbability=mindiff;}
	58	void SetPriorFunctions(TF1 *func[AliPID::kSPECIES]);
	59	void SetANDstatus(TString dets);
	60	void SetDetectorProbabilityRestriction(TString det, Int_t iPart, Double_t upperprob);
	61	void SetHistogramAxis(Int_t nbins, Double_t xmin, Double_t xmax) {fNbins=nbins; fXmin = xmin; fXmax = xmax;}
bc6176fb	62	void SetAODmode(Bool_t isaod = kFALSE) {fgIsAOD=isaod;}
bc6176fb	63
563113d0	64	//loads the track detector responses and the track status
	65	void TrackInfo(const AliESDtrack *pTrk,ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;
	66
	67	//identifies the track
	68	Int_t Identify(Double_t pid[AliPID::kSPECIES]) const;
	69
	70	//identifies the track filling the QA histograms
	71	Int_t IdentifyQA(const Double_t pid[AliPID::kSPECIES],Int_t idets) const;
	72
	73	//returns the track identification number
	74	Int_t GetID(ULong_t status[kNdets+1], Double_t pid[kNdets+1][AliPID::kSPECIES]) const;
bc6176fb	75	//returns the track identification number in caso of an AliAODTrack
bc6176fb	76	Int_t GetAODID(AliAODTrack *aodtrack) const;
563113d0	77
bc6176fb	78
563113d0	79	//main
563113d0	80	virtual Bool_t IsSelected(TObject *track);
264ebaac	81	Bool_t IsSelected(TList* /list/) {return kTRUE;}
563113d0	82	//histos are added to a list
107a3100	83	void AddQAHistograms(TList *qalist);
563113d0	84
	85
	86	private:
	87	void SetPPriors(AliESDtrack *pTrk);
	88	ULong_t StatusForAND(ULong_t status[kNdets+1]) const; //
	89	void InitialiseHisto();
107a3100	90	void DefineHistograms(); //histo booking
563113d0	91	Bool_t Check(const Double_t *p, Int_t iPsel, Double_t minDiff) const;
	92	void CombPID(ULong_t status[kNdets+1],Double_t pid[kNdets+1][AliPID::kSPECIES],Double_t *combpid) const;
	93
	94	Double_t fCut; //probability cut
	95	Double_t fMinDiffResponse; // minimum difference between detector resposes
	96	Double_t fMinDiffProbability; // minimum difference between probability values
	97	Int_t fgParticleType; // requested particle type
	98	Bool_t fgIsComb; // flag for the combined pid
bc6176fb	99	Bool_t fgIsAOD; // flag for the AOD QA histograms
563113d0	100	Bool_t fCheckResponse; // flag to check the minimum difference of det responsess
	101	Bool_t fCheckSelection; // flag to check the minimum difference of probabilities
	102	Bool_t fIsPpriors; //flag for momentum dependent priors
	103	Bool_t fIsDetAND; //flag for AND with multiple detectors
	104	Double_t fXmin; //x min QA histo
	105	Double_t fXmax; //x max QA histo
	106	Int_t fNbins; //n bins QA histo
	107	Int_t fDetRestr; //id of the detector for the restriction
	108	Int_t fiPartRestr; //id of the particle for the restriction
	109	Double_t fDetProbRestr; //probability restriction value
	110
	111	Double_t fPriors[AliPID::kSPECIES]; //a priori concentrations
	112	TF1 *fPriorsFunc[AliPID::kSPECIES]; //momentum dependent priors
	113	Bool_t fDets[kNdets]; //boolean(s) corresponding to the chosen detector(s)
	114	Bool_t fDetsInAnd[kNdets]; //detector to be in AND for the combined PID
	115	TH1F *fhResp[kNdets][AliPID::kSPECIES]; //QA histo
	116	TH1F *fhProb[kNdets][AliPID::kSPECIES]; //QA histo
	117	TH1F *fhCombResp[AliPID::kSPECIES]; //QA histo
	118	TH1F *fhCombProb[AliPID::kSPECIES]; //QA histo
	119
	120	ClassDef(AliCFTrackCutPid,1);
	121	};
	122	#endif
	123