[u/mrichter/AliRoot.git] / HLT / trigger / AliHLTTRDTriggerComponent.h

//-*- Mode: C++ -*-
// $Id$
#ifndef ALIHLTTRDTRIGGERCOMPONENT_H
#define ALIHLTTRDTRIGGERCOMPONENT_H

//* This file is property of and copyright by the ALICE HLT Project        *
//* ALICE Experiment at CERN, All rights reserved.                         *
//* See cxx source for full Copyright notice                               *

/// @file   AliHLTTRDTriggerComponent.h
/// @author Felix Rettig, Stefan Kirsch
/// @date   2012-08-16
/// @brief

#include "TString.h"
#include "AliHLTTrigger.h"
#include "AliESDEvent.h"
#include "AliESDTrdTrack.h"
#include "AliExternalTrackParam.h"
#include "AliTRDgeometry.h"
#ifdef __TRDHLTDEBUG
  #include "AliTRDtrackingEventDisplay.h"
#endif

#define trd_det_lsi(det) ((det) / 6)               // convert TRD detector/chamber 0-539 index to linear stack index 0-89
#define trd_det_lyr(det) ((det) % 6)               // convert detector (=chamber) number 0-539 to local layer 0-5
#define trd_det_si(det) (((det) % 30) / 6)         // convert detector (=chamber) number 0-539 to local stack index 0-4

class TObjArray;
class TH1I;
class TH2I;
class AliHLTComponentBenchmark;
class AliESDtrack;
class AliHLTGlobalBarrelTrack;
class AliESDtrackCuts;
class AliHLTESDTrackCuts;
class AliTRDonlineTrackingDataContainer;

/**
 * @class  AliHLTTRDTriggerComponent
 */
class AliHLTTRDTriggerComponent : public AliHLTTrigger
{
 public:

  AliHLTTRDTriggerComponent();
  virtual ~AliHLTTRDTriggerComponent();

  virtual const char* GetTriggerName() const;
  virtual AliHLTComponent* Spawn();

 protected:
  int DoInit(int argc, const char** argv);
  int DoDeinit();
  int Reconfigure(const char* cdbEntry, const char* chainId);
  int ReadPreprocessorValues(const char* modules);
  int ConfigureFromCDBObject(TString cdbPath);
  int ScanConfigurationArgument(int argc, const char** argv);

 private:
  AliHLTTRDTriggerComponent (const AliHLTTRDTriggerComponent&);
  AliHLTTRDTriggerComponent& operator=(const AliHLTTRDTriggerComponent&);
  virtual int DoTrigger();

  Bool_t CheckRefTrackCuts(AliESDtrack* track);

  // TRD-trigger specific
  void ScanTriggerClasses(const char* firedTriggerClasses);
  int PrepareESDData();
  int PrepareHLTData();
  int PrepareTRDData();
  int MatchTRDTracks();
  int MatchTRDTracksESD();
  int MatchTRDTracksHLT();
  Bool_t TRDElectronTrigger(const char *ident, const Double_t minPt, const UShort_t minPID);

  Bool_t TrackPlaneIntersect(AliExternalTrackParam *trk, Double_t pnt[3], Double_t norm[3], Double_t mag);

  Int_t EstimateTrackDistance(AliExternalTrackParam *refParam,
				 const UShort_t stack,
				 const UShort_t layerMask,
				 const Float_t trklLocalY[6], const Int_t trklBinZ[6],
				 Double_t mag, Double_t *ydist, Double_t *zdist);

  Int_t EstimateTrackDistance(AliESDtrack *esd_track,
				 const UShort_t stack,
				 const UShort_t layerMask,
				 const Float_t trklLocalY[6], const Int_t trklBinZ[6],
				 Double_t mag, Double_t *ydist, Double_t *zdist);

  Double_t RateTrackMatch(Double_t distY, Double_t distZ, Double_t rpt, Double_t gpt);

  void DumpTrackingData();
  void AssignTrackInfo(TString* infoStr, const UInt_t stack, const UInt_t trackIndex, const char* flagStr = "");
#ifdef __TRDHLTDEBUG
  void RenderEvent(const Bool_t showGtuTracks = kTRUE, const Bool_t showTracklets = kTRUE, const Bool_t showRefTracks = kTRUE);
#endif
  void DbgLog(const char* prefix, ...);

  TString  fName;                           //! trigger name
  Double_t fRefTrackSelectionEtaLimit;      //! ref track preselection maximum eta
  Double_t fRefTrackSelectionVertexXYLimit; //! ref track preselection maximum distance to ip in XY plane
  Double_t fRefTrackSelectionVertexZLimit;  //! ref track preselection maximum distance to ip in Z
  Double_t fRefTrackSelectionPtThreshold;   //! pt threshold for ref track preselection in GeV/c
  Double_t fMatchRatingThreshold;           //! track match rating threshold
  Double_t fElectronTriggerPtThresholdHSE;  //! pt threshold for HSE electron trigger
  UShort_t fElectronTriggerPIDThresholdHSE; //! PID threshold for HSE electron trigger
  Double_t fElectronTriggerPtThresholdHQU;  //! pt threshold for HQU electron trigger
  UShort_t fElectronTriggerPIDThresholdHQU; //! PID threshold for HQU electron trigger
  Bool_t fApplyRefTrackCuts;                //! switch on/off ref track cuts for matching
  Bool_t fElectronTriggerOnL1TrgOnly;       //! run electron trigger only for events with L1 electron trigger
  UShort_t fHistoMode;                      //! histogramming mode, 0: single event, 1: accumulative (debugging)
  UShort_t fDebugLevel;                     //! set debug checks/output level, 0: debug off
  Bool_t fExtendedHistos;                   //! switch on/off additional histograms
  Bool_t fEventRendering;                   //! switch on/off event rendering
  Bool_t fPushHistos;                       //! switch on/off pushing of histograms event by event
  Bool_t fWriteHistos;                      //! switch on/off histogram writing on deinit

  static const char* fgkDefaultOCDBEntry;                    //! default OCDB entry
  static const char* fgkTriggerDecisionElectronHSE;          //! electron trigger flag string
  static const char* fgkTriggerDecisionElectronHQU;          //! electron trigger flag string

  static const AliHLTEventID_t fgkInvalidEventId = 0xffffffffffffffffllu;
  static const unsigned int fkTRDLayers = 6;                 //! number of layers per stack in TRD
  static const unsigned int fkTRDStacks = 90;                //! number of stacks in TRD
  static const unsigned int fkTRDSectors = 18;               //! number of sectors in TRD
  static const unsigned int fkMaxRefTracksPerStack = 25000;  //! maximum number of ref tracks per stack
  static const unsigned int fkMaxRefTracks = 25000;          //! maximum number of ref tracks

  static const unsigned int fkElectronTriggerHSE = 0x1;      //! HSE electron trigger flag
  static const unsigned int fkElectronTriggerHQU = 0x2;      //! HSE electron trigger flag

  AliHLTEventID_t fEventId;                                  //! hlt internal event id
  Int_t fRunNumber;                                          //! run number
  TString* fChunkId;                                         //! chunk identifier
  UInt_t fSectorsWithData;                                   //! data present flags for each sector
  Bool_t fIsMinBiasEvent;                                    //! indicates a minimum bias event
  Bool_t fIsTRDElectronEvent;                                //! indicates a TRD L1 electron triggered event
  Bool_t fESDtracksPresent;                                  //! indicates that ESD tracks are present
  Bool_t fHLTtracksPresent;                                  //! indicates that HLT raw tracks are present

  AliTRDgeometry* fTRDGeometry;                              //! instance of TRD geometry
  AliESDEvent* fEsdEvent;                                    //! current ESD event
  AliTRDonlineTrackingDataContainer* fTrackingData;          //! container for TRD tracking data
  vector<AliHLTGlobalBarrelTrack>* fHLTTracks;               //! HLT raw tracks
  AliESDtrackCuts* fRefTrackCuts;                            //! reference track cuts

#ifdef __TRDHLTDEBUG
  AliTRDtrackingEventDisplay* fEventDisplay;                 //! event rendering
  AliHLTComponentBenchmark* fBenchmark;                      //! benchmark instance
#endif

  TObjArray* fHistArray;
  TH1I* fHistMatchRating;                                    //! histo
  TH2I* fHistMatchRatingByPt;                                //! histo
  TH2I* fHistMatchRatingByPid;                               //! histo
  TH1I* fHistTrackPt;                                        //! histo
  TH1I* fHistTrackPtMatched;                                 //! histo
  TH2I* fHistTrackPtCorr;                                    //! histo
  TH1I* fHistTrackPid;                                       //! histo
  TH1I* fHistTrackPidMatched;                                //! histo
  TH1I* fHistElectronCandidatePt;                            //! histo
  TH1I* fHistElectronCandidateMatchedPt;                     //! histo
  TH1I* fHistElectronCandidatePid;                           //! histo
  TH1I* fHistElectronCandidateMatchedPid;                    //! histo
  TH2I* fHistRefTrackPid;                                    //! histo
  TH2I* fHistMatchedRefTrackPid;                             //! histo
  TH2I* fHistPIDvsTruncPID;                                  //! histo
  TH2I* fHistElectronFalsePIDvsTruncPID;                     //! histo
  TH2I* fHistElectronConfirmedPIDvsTruncPID;                 //! histo
  TH2I* fHistTrackMatchingCombinations;                      //! histo
  TH1I* fHistElectronTriggerBaseMinBias;                     //! histo
  TH1I* fHistElectronTriggerBaseTrdL1;                       //! histo

  ClassDef(AliHLTTRDTriggerComponent, 0)
};

#endif //ALIHLTTRDTRIGGERCOMPONENT_H
Commit	Line	Data
c7b7f445	1	//-- Mode: C++ --
	2	// $Id$
	3	#ifndef ALIHLTTRDTRIGGERCOMPONENT_H
	4	#define ALIHLTTRDTRIGGERCOMPONENT_H
	5
	6	//* This file is property of and copyright by the ALICE HLT Project *
	7	//* ALICE Experiment at CERN, All rights reserved. *
	8	//* See cxx source for full Copyright notice *
	9
	10	/// @file AliHLTTRDTriggerComponent.h
	11	/// @author Felix Rettig, Stefan Kirsch
	12	/// @date 2012-08-16
	13	/// @brief
	14
	15	#include "TString.h"
	16	#include "AliHLTTrigger.h"
	17	#include "AliESDEvent.h"
	18	#include "AliESDTrdTrack.h"
	19	#include "AliExternalTrackParam.h"
	20	#include "AliTRDgeometry.h"
	21	#ifdef __TRDHLTDEBUG
	22	#include "AliTRDtrackingEventDisplay.h"
	23	#endif
	24
	25	#define trd_det_lsi(det) ((det) / 6) // convert TRD detector/chamber 0-539 index to linear stack index 0-89
	26	#define trd_det_lyr(det) ((det) % 6) // convert detector (=chamber) number 0-539 to local layer 0-5
	27	#define trd_det_si(det) (((det) % 30) / 6) // convert detector (=chamber) number 0-539 to local stack index 0-4
	28
	29	class TObjArray;
	30	class TH1I;
	31	class TH2I;
	32	class AliHLTComponentBenchmark;
	33	class AliESDtrack;
	34	class AliHLTGlobalBarrelTrack;
	35	class AliESDtrackCuts;
	36	class AliHLTESDTrackCuts;
	37	class AliTRDonlineTrackingDataContainer;
	38
	39	/**
	40	* @class AliHLTTRDTriggerComponent
	41	*/
	42	class AliHLTTRDTriggerComponent : public AliHLTTrigger
	43	{
	44	public:
	45
	46	AliHLTTRDTriggerComponent();
	47	virtual ~AliHLTTRDTriggerComponent();
	48
	49	virtual const char* GetTriggerName() const;
	50	virtual AliHLTComponent* Spawn();
	51
	52	protected:
	53	int DoInit(int argc, const char** argv);
	54	int DoDeinit();
	55	int Reconfigure(const char* cdbEntry, const char* chainId);
	56	int ReadPreprocessorValues(const char* modules);
	57	int ConfigureFromCDBObject(TString cdbPath);
	58	int ScanConfigurationArgument(int argc, const char** argv);
	59
	60	private:
	61	AliHLTTRDTriggerComponent (const AliHLTTRDTriggerComponent&);
	62	AliHLTTRDTriggerComponent& operator=(const AliHLTTRDTriggerComponent&);
	63	virtual int DoTrigger();
	64
65	Bool_t CheckRefTrackCuts(AliESDtrack* track);
66
67	// TRD-trigger specific
68	void ScanTriggerClasses(const char* firedTriggerClasses);
69	int PrepareESDData();
70	int PrepareHLTData();
71	int PrepareTRDData();
72	int MatchTRDTracks();
73	int MatchTRDTracksESD();
74	int MatchTRDTracksHLT();
75	Bool_t TRDElectronTrigger(const char *ident, const Double_t minPt, const UShort_t minPID);
76
77	Bool_t TrackPlaneIntersect(AliExternalTrackParam *trk, Double_t pnt[3], Double_t norm[3], Double_t mag);
78
79	Int_t EstimateTrackDistance(AliExternalTrackParam *refParam,
80	const UShort_t stack,
81	const UShort_t layerMask,
82	const Float_t trklLocalY[6], const Int_t trklBinZ[6],
83	Double_t mag, Double_t ydist, Double_t zdist);
84
85	Int_t EstimateTrackDistance(AliESDtrack *esd_track,
86	const UShort_t stack,
87	const UShort_t layerMask,
88	const Float_t trklLocalY[6], const Int_t trklBinZ[6],
89	Double_t mag, Double_t ydist, Double_t zdist);
90
91	Double_t RateTrackMatch(Double_t distY, Double_t distZ, Double_t rpt, Double_t gpt);
92
93	void DumpTrackingData();
94	void AssignTrackInfo(TString* infoStr, const UInt_t stack, const UInt_t trackIndex, const char* flagStr = "");
95	#ifdef __TRDHLTDEBUG
96	void RenderEvent(const Bool_t showGtuTracks = kTRUE, const Bool_t showTracklets = kTRUE, const Bool_t showRefTracks = kTRUE);
97	#endif
98	void DbgLog(const char* prefix, ...);
99
100	TString fName; //! trigger name
101	Double_t fRefTrackSelectionEtaLimit; //! ref track preselection maximum eta
102	Double_t fRefTrackSelectionVertexXYLimit; //! ref track preselection maximum distance to ip in XY plane
103	Double_t fRefTrackSelectionVertexZLimit; //! ref track preselection maximum distance to ip in Z
104	Double_t fRefTrackSelectionPtThreshold; //! pt threshold for ref track preselection in GeV/c
105	Double_t fMatchRatingThreshold; //! track match rating threshold
106	Double_t fElectronTriggerPtThresholdHSE; //! pt threshold for HSE electron trigger
107	UShort_t fElectronTriggerPIDThresholdHSE; //! PID threshold for HSE electron trigger
108	Double_t fElectronTriggerPtThresholdHQU; //! pt threshold for HQU electron trigger
109	UShort_t fElectronTriggerPIDThresholdHQU; //! PID threshold for HQU electron trigger
110	Bool_t fApplyRefTrackCuts; //! switch on/off ref track cuts for matching
111	Bool_t fElectronTriggerOnL1TrgOnly; //! run electron trigger only for events with L1 electron trigger
112	UShort_t fHistoMode; //! histogramming mode, 0: single event, 1: accumulative (debugging)
113	UShort_t fDebugLevel; //! set debug checks/output level, 0: debug off
114	Bool_t fExtendedHistos; //! switch on/off additional histograms
115	Bool_t fEventRendering; //! switch on/off event rendering
116	Bool_t fPushHistos; //! switch on/off pushing of histograms event by event
117	Bool_t fWriteHistos; //! switch on/off histogram writing on deinit
118
119	static const char* fgkDefaultOCDBEntry; //! default OCDB entry
120	static const char* fgkTriggerDecisionElectronHSE; //! electron trigger flag string
121	static const char* fgkTriggerDecisionElectronHQU; //! electron trigger flag string
122
123	static const AliHLTEventID_t fgkInvalidEventId = 0xffffffffffffffffllu;
124	static const unsigned int fkTRDLayers = 6; //! number of layers per stack in TRD
125	static const unsigned int fkTRDStacks = 90; //! number of stacks in TRD
126	static const unsigned int fkTRDSectors = 18; //! number of sectors in TRD
127	static const unsigned int fkMaxRefTracksPerStack = 25000; //! maximum number of ref tracks per stack
128	static const unsigned int fkMaxRefTracks = 25000; //! maximum number of ref tracks
129
130	static const unsigned int fkElectronTriggerHSE = 0x1; //! HSE electron trigger flag
131	static const unsigned int fkElectronTriggerHQU = 0x2; //! HSE electron trigger flag
132
133	AliHLTEventID_t fEventId; //! hlt internal event id
134	Int_t fRunNumber; //! run number
135	TString* fChunkId; //! chunk identifier
136	UInt_t fSectorsWithData; //! data present flags for each sector
137	Bool_t fIsMinBiasEvent; //! indicates a minimum bias event
138	Bool_t fIsTRDElectronEvent; //! indicates a TRD L1 electron triggered event
139	Bool_t fESDtracksPresent; //! indicates that ESD tracks are present
140	Bool_t fHLTtracksPresent; //! indicates that HLT raw tracks are present
141
142	AliTRDgeometry* fTRDGeometry; //! instance of TRD geometry
143	AliESDEvent* fEsdEvent; //! current ESD event
144	AliTRDonlineTrackingDataContainer* fTrackingData; //! container for TRD tracking data
145	vector<AliHLTGlobalBarrelTrack>* fHLTTracks; //! HLT raw tracks
146	AliESDtrackCuts* fRefTrackCuts; //! reference track cuts
147
148	#ifdef __TRDHLTDEBUG
149	AliTRDtrackingEventDisplay* fEventDisplay; //! event rendering
150	AliHLTComponentBenchmark* fBenchmark; //! benchmark instance
151	#endif
152
153	TObjArray* fHistArray;
154	TH1I* fHistMatchRating; //! histo
155	TH2I* fHistMatchRatingByPt; //! histo
156	TH2I* fHistMatchRatingByPid; //! histo
157	TH1I* fHistTrackPt; //! histo
158	TH1I* fHistTrackPtMatched; //! histo
159	TH2I* fHistTrackPtCorr; //! histo
160	TH1I* fHistTrackPid; //! histo
161	TH1I* fHistTrackPidMatched; //! histo
162	TH1I* fHistElectronCandidatePt; //! histo
163	TH1I* fHistElectronCandidateMatchedPt; //! histo
164	TH1I* fHistElectronCandidatePid; //! histo
165	TH1I* fHistElectronCandidateMatchedPid; //! histo
166	TH2I* fHistRefTrackPid; //! histo
167	TH2I* fHistMatchedRefTrackPid; //! histo
168	TH2I* fHistPIDvsTruncPID; //! histo
169	TH2I* fHistElectronFalsePIDvsTruncPID; //! histo
170	TH2I* fHistElectronConfirmedPIDvsTruncPID; //! histo
171	TH2I* fHistTrackMatchingCombinations; //! histo
172	TH1I* fHistElectronTriggerBaseMinBias; //! histo
173	TH1I* fHistElectronTriggerBaseTrdL1; //! histo
174
175	ClassDef(AliHLTTRDTriggerComponent, 0)
176	};
177
178	#endif //ALIHLTTRDTRIGGERCOMPONENT_H