[u/mrichter/AliRoot.git] / STEER / AOD / AliNanoAODTrackMapping.h

#ifndef _ALINANOAODTRACKMAPPING_H_
#define _ALINANOAODTRACKMAPPING_H_

#include "TObject.h"
#include "TFile.h"
#include "AliLog.h"
#include "TSystem.h"
#include "TTree.h"
#include "TDirectory.h"

class AliNanoAODTrackMapping : public TObject
{
public:
  AliNanoAODTrackMapping();
  AliNanoAODTrackMapping(const char * mappingString);
  virtual ~AliNanoAODTrackMapping(){;}

  void Print(const Option_t * opt = "") const;

  static AliNanoAODTrackMapping * GetInstance() {
    if(!fInstance) LoadInstance() ;
    return fInstance;
  }  
  

  const char * GetVarName(Int_t index) const;
  Int_t GetVarIndex(TString varName); // cannot be const (uses stl map)

  //TODO: implement custom variables

  // Getters
  //  Internal
  Int_t GetSize()             const { return fSize;             }  
  //  Kin vars
  Int_t GetPt()               const { return fPt;               }
  Int_t GetPhi()              const { return fPhi;              }
  Int_t GetTheta()            const { return fTheta;            }
  Int_t GetChi2PerNDF()       const { return fChi2PerNDF;       }
  Int_t GetPosX()             const { return fPosX;             }
  Int_t GetPosY()             const { return fPosY;             }
  Int_t GetPosZ()             const { return fPosZ;             }
  Int_t GetPDCAX()            const { return fPDCAX;            }
  Int_t GetPDCAY()            const { return fPDCAY;            }
  Int_t GetPDCAZ()            const { return fPDCAZ;            }
  Int_t GetPosDCAx()          const { return fPosDCAx;          }
  Int_t GetPosDCAy()          const { return fPosDCAy;          }
  Int_t GetRAtAbsorberEnd()   const { return fRAtAbsorberEnd;   }
  Int_t GetTPCncls()          const { return fTPCncls;          }
  Int_t Getid()               const { return fid;               }
  Int_t GetTPCnclsF()         const { return fTPCnclsF;         }
  Int_t GetTPCNCrossedRows()  const { return fTPCNCrossedRows;  }
  Int_t GetTrackPhiOnEMCal()  const { return fTrackPhiOnEMCal;  }
  Int_t GetTrackEtaOnEMCal()  const { return fTrackEtaOnEMCal;  }
  Int_t GetTrackPtOnEMCal()   const { return fTrackPtOnEMCal;   }
  Int_t GetITSsignal()        const { return fITSsignal;        }
  Int_t GetTPCsignal()        const { return fTPCsignal;        }
  Int_t GetTPCsignalTuned()   const { return fTPCsignalTuned;   }
  Int_t GetTPCsignalN()       const { return fTPCsignalN;       }
  Int_t GetTPCmomentum()      const { return fTPCmomentum;      }
  Int_t GetTPCTgl()           const { return fTPCTgl;           }
  Int_t GetTOFsignal()        const { return fTOFsignal;        }
  Int_t GetintegratedLenght() const { return fintegratedLenght; }
  Int_t GetTOFsignalTuned()   const { return fTOFsignalTuned;   }
  Int_t GetHMPIDsignal()      const { return fHMPIDsignal;      }
  Int_t GetHMPIDoccupancy()   const { return fHMPIDoccupancy;   }
  Int_t GetTRDsignal()        const { return fTRDsignal;        }
  Int_t GetTRDChi2()          const { return fTRDChi2;          }
  Int_t GetTRDnSlices()       const { return fTRDnSlices;       }
  Int_t Getcovmat()           const { return fcovmat;           }
  
  // TODO: implement custom variables


private:

  static void  LoadInstance() ;
  
  Int_t fSize; // Number of variables actually allocated
  void  SetSize (Int_t var) { fSize = var;}
  // FIXME: should this be static?
  Int_t fPt;      	  // Mapping variable
  Int_t fPhi;		  // Mapping variable
  Int_t fTheta;		  // Mapping variable
  Int_t fChi2PerNDF;	  // Mapping variable
  Int_t fPosX;		  // Mapping variable
  Int_t fPosY;		  // Mapping variable
  Int_t fPosZ;		  // Mapping variable
  Int_t fPDCAX;		  // Mapping variable
  Int_t fPDCAY;		  // Mapping variable
  Int_t fPDCAZ;		  // Mapping variable
  Int_t fPosDCAx;	  // Mapping variable
  Int_t fPosDCAy;	  // Mapping variable
  Int_t fRAtAbsorberEnd;  // Mapping variable
  Int_t fTPCncls;	  // Mapping variable
  Int_t fid;		  // Mapping variable
  Int_t fTPCnclsF;	  // Mapping variable
  Int_t fTPCNCrossedRows; // Mapping variable
  Int_t fTrackPhiOnEMCal; // Mapping variable
  Int_t fTrackEtaOnEMCal; // Mapping variable
  Int_t fTrackPtOnEMCal; // Mapping variable
  Int_t fITSsignal;	  // Mapping variable
  Int_t fTPCsignal;	  // Mapping variable
  Int_t fTPCsignalTuned;  // Mapping variable
  Int_t fTPCsignalN;	  // Mapping variable
  Int_t fTPCmomentum;	  // Mapping variable
  Int_t fTPCTgl;	  // Mapping variable
  Int_t fTOFsignal;	  // Mapping variable
  Int_t fintegratedLenght;// Mapping variable
  Int_t fTOFsignalTuned;  // Mapping variable
  Int_t fHMPIDsignal;	  // Mapping variable
  Int_t fHMPIDoccupancy;  // Mapping variable
  Int_t fTRDsignal;	  // Mapping variable
  Int_t fTRDChi2;	  // Mapping variable
  Int_t fTRDnSlices;	  // Mapping variable
  Int_t fcovmat;          // Mapping variable
  
  // Setters are private because we don't want the mapping to change once the class has been instantiated 
  void  SetPt               (Int_t var) { fPt = var;               }
  void  SetPhi              (Int_t var) { fPhi = var;              }  
  void  SetTheta            (Int_t var) { fTheta = var;            }
  void  SetChi2PerNDF       (Int_t var) { fChi2PerNDF = var;       }  
  void  SetPosX             (Int_t var) { fPosX = var;             }
  void  SetPosY             (Int_t var) { fPosY = var;             }
  void  SetPosZ             (Int_t var) { fPosZ = var;             }
  void  SetPDCAX            (Int_t var) { fPDCAX = var;            }
  void  SetPDCAY            (Int_t var) { fPDCAY = var;            }
  void  SetPDCAZ            (Int_t var) { fPDCAZ = var;            }
  void  SetPosDCAx          (Int_t var) { fPosDCAx = var;          }
  void  SetPosDCAy          (Int_t var) { fPosDCAy = var;          }
  void  SetRAtAbsorberEnd   (Int_t var) { fRAtAbsorberEnd = var;   }
  void  SetTPCncls          (Int_t var) { fTPCncls = var;          }
  void  Setid               (Int_t var) { fid = var;               }
  void  SetTPCnclsF         (Int_t var) { fTPCnclsF = var;         }
  void  SetTPCNCrossedRows  (Int_t var) { fTPCNCrossedRows = var;  }
  void  SetTrackPhiOnEMCal  (Int_t var) { fTrackPhiOnEMCal = var;  }
  void  SetTrackEtaOnEMCal  (Int_t var) { fTrackEtaOnEMCal = var;  }
  void  SetTrackPtOnEMCal   (Int_t var) { fTrackPtOnEMCal = var;   }
  void  SetITSsignal        (Int_t var) { fITSsignal = var;        }
  void  SetTPCsignal        (Int_t var) { fTPCsignal = var;        }
  void  SetTPCsignalTuned   (Int_t var) { fTPCsignalTuned = var;   }
  void  SetTPCsignalN       (Int_t var) { fTPCsignalN = var;       }
  void  SetTPCmomentum      (Int_t var) { fTPCmomentum = var;      }
  void  SetTPCTgl           (Int_t var) { fTPCTgl = var;           }
  void  SetTOFsignal        (Int_t var) { fTOFsignal = var;        }
  void  SetintegratedLenght (Int_t var) { fintegratedLenght = var; }
  void  SetTOFsignalTuned   (Int_t var) { fTOFsignalTuned = var;   }
  void  SetHMPIDsignal      (Int_t var) { fHMPIDsignal = var;      }
  void  SetHMPIDoccupancy   (Int_t var) { fHMPIDoccupancy = var;   }
  void  SetTRDsignal        (Int_t var) { fTRDsignal = var;        }
  void  SetTRDChi2          (Int_t var) { fTRDChi2 = var;          }
  void  SetTRDnSlices       (Int_t var) { fTRDnSlices = var;       }
  void  Setcovmat           (Int_t var) { fcovmat = var;           }

  static AliNanoAODTrackMapping * fInstance; //instance, needed for the singleton implementation
  static TString fMappingString; // the string which this class was initialized with
  std::map<TString,int> fMapCstVar;// Map of indexes of custom variables: CACHE THIS TO CONST INTs IN YOUR TASK TO AVOID CONTINUOUS STRING COMPARISONS
  ClassDef(AliNanoAODTrackMapping, 1)
  
};

#endif /* _ALINANOAODTRACKMAPPING_H_ */
Commit	Line	Data
778636d4	1	#ifndef _ALINANOAODTRACKMAPPING_H_
	2	#define _ALINANOAODTRACKMAPPING_H_
	3
	4	#include "TObject.h"
	5	#include "TFile.h"
	6	#include "AliLog.h"
	7	#include "TSystem.h"
	8	#include "TTree.h"
	9	#include "TDirectory.h"
	10
	11	class AliNanoAODTrackMapping : public TObject
	12	{
	13	public:
	14	AliNanoAODTrackMapping();
	15	AliNanoAODTrackMapping(const char * mappingString);
	16	virtual ~AliNanoAODTrackMapping(){;}
	17
	18	void Print(const Option_t * opt = "") const;
	19
	20	static AliNanoAODTrackMapping * GetInstance() {
	21	if(!fInstance) LoadInstance() ;
	22	return fInstance;
	23	}
	24
	25
	26	const char * GetVarName(Int_t index) const;
	27	Int_t GetVarIndex(TString varName); // cannot be const (uses stl map)
	28
	29	//TODO: implement custom variables
	30
	31	// Getters
	32	// Internal
	33	Int_t GetSize() const { return fSize; }
	34	// Kin vars
	35	Int_t GetPt() const { return fPt; }
	36	Int_t GetPhi() const { return fPhi; }
	37	Int_t GetTheta() const { return fTheta; }
	38	Int_t GetChi2PerNDF() const { return fChi2PerNDF; }
	39	Int_t GetPosX() const { return fPosX; }
	40	Int_t GetPosY() const { return fPosY; }
	41	Int_t GetPosZ() const { return fPosZ; }
	42	Int_t GetPDCAX() const { return fPDCAX; }
	43	Int_t GetPDCAY() const { return fPDCAY; }
	44	Int_t GetPDCAZ() const { return fPDCAZ; }
	45	Int_t GetPosDCAx() const { return fPosDCAx; }
	46	Int_t GetPosDCAy() const { return fPosDCAy; }
	47	Int_t GetRAtAbsorberEnd() const { return fRAtAbsorberEnd; }
	48	Int_t GetTPCncls() const { return fTPCncls; }
	49	Int_t Getid() const { return fid; }
	50	Int_t GetTPCnclsF() const { return fTPCnclsF; }
	51	Int_t GetTPCNCrossedRows() const { return fTPCNCrossedRows; }
	52	Int_t GetTrackPhiOnEMCal() const { return fTrackPhiOnEMCal; }
	53	Int_t GetTrackEtaOnEMCal() const { return fTrackEtaOnEMCal; }
	54	Int_t GetTrackPtOnEMCal() const { return fTrackPtOnEMCal; }
	55	Int_t GetITSsignal() const { return fITSsignal; }
	56	Int_t GetTPCsignal() const { return fTPCsignal; }
	57	Int_t GetTPCsignalTuned() const { return fTPCsignalTuned; }
	58	Int_t GetTPCsignalN() const { return fTPCsignalN; }
	59	Int_t GetTPCmomentum() const { return fTPCmomentum; }
	60	Int_t GetTPCTgl() const { return fTPCTgl; }
	61	Int_t GetTOFsignal() const { return fTOFsignal; }
	62	Int_t GetintegratedLenght() const { return fintegratedLenght; }
	63	Int_t GetTOFsignalTuned() const { return fTOFsignalTuned; }
	64	Int_t GetHMPIDsignal() const { return fHMPIDsignal; }
65	Int_t GetHMPIDoccupancy() const { return fHMPIDoccupancy; }
66	Int_t GetTRDsignal() const { return fTRDsignal; }
67	Int_t GetTRDChi2() const { return fTRDChi2; }
68	Int_t GetTRDnSlices() const { return fTRDnSlices; }
69	Int_t Getcovmat() const { return fcovmat; }
70
71	// TODO: implement custom variables
72
73
74	private:
75
76	static void LoadInstance() ;
77
78	Int_t fSize; // Number of variables actually allocated
79	void SetSize (Int_t var) { fSize = var;}
80	// FIXME: should this be static?
81	Int_t fPt; // Mapping variable
82	Int_t fPhi; // Mapping variable
83	Int_t fTheta; // Mapping variable
84	Int_t fChi2PerNDF; // Mapping variable
85	Int_t fPosX; // Mapping variable
86	Int_t fPosY; // Mapping variable
87	Int_t fPosZ; // Mapping variable
88	Int_t fPDCAX; // Mapping variable
89	Int_t fPDCAY; // Mapping variable
90	Int_t fPDCAZ; // Mapping variable
91	Int_t fPosDCAx; // Mapping variable
92	Int_t fPosDCAy; // Mapping variable
93	Int_t fRAtAbsorberEnd; // Mapping variable
94	Int_t fTPCncls; // Mapping variable
95	Int_t fid; // Mapping variable
96	Int_t fTPCnclsF; // Mapping variable
97	Int_t fTPCNCrossedRows; // Mapping variable
98	Int_t fTrackPhiOnEMCal; // Mapping variable
99	Int_t fTrackEtaOnEMCal; // Mapping variable
100	Int_t fTrackPtOnEMCal; // Mapping variable
101	Int_t fITSsignal; // Mapping variable
102	Int_t fTPCsignal; // Mapping variable
103	Int_t fTPCsignalTuned; // Mapping variable
104	Int_t fTPCsignalN; // Mapping variable
105	Int_t fTPCmomentum; // Mapping variable
106	Int_t fTPCTgl; // Mapping variable
107	Int_t fTOFsignal; // Mapping variable
108	Int_t fintegratedLenght;// Mapping variable
109	Int_t fTOFsignalTuned; // Mapping variable
110	Int_t fHMPIDsignal; // Mapping variable
111	Int_t fHMPIDoccupancy; // Mapping variable
112	Int_t fTRDsignal; // Mapping variable
113	Int_t fTRDChi2; // Mapping variable
114	Int_t fTRDnSlices; // Mapping variable
115	Int_t fcovmat; // Mapping variable
116
117	// Setters are private because we don't want the mapping to change once the class has been instantiated
118	void SetPt (Int_t var) { fPt = var; }
119	void SetPhi (Int_t var) { fPhi = var; }
120	void SetTheta (Int_t var) { fTheta = var; }
121	void SetChi2PerNDF (Int_t var) { fChi2PerNDF = var; }
122	void SetPosX (Int_t var) { fPosX = var; }
123	void SetPosY (Int_t var) { fPosY = var; }
124	void SetPosZ (Int_t var) { fPosZ = var; }
125	void SetPDCAX (Int_t var) { fPDCAX = var; }
126	void SetPDCAY (Int_t var) { fPDCAY = var; }
127	void SetPDCAZ (Int_t var) { fPDCAZ = var; }
128	void SetPosDCAx (Int_t var) { fPosDCAx = var; }
129	void SetPosDCAy (Int_t var) { fPosDCAy = var; }
130	void SetRAtAbsorberEnd (Int_t var) { fRAtAbsorberEnd = var; }
131	void SetTPCncls (Int_t var) { fTPCncls = var; }
132	void Setid (Int_t var) { fid = var; }
133	void SetTPCnclsF (Int_t var) { fTPCnclsF = var; }
134	void SetTPCNCrossedRows (Int_t var) { fTPCNCrossedRows = var; }
135	void SetTrackPhiOnEMCal (Int_t var) { fTrackPhiOnEMCal = var; }
136	void SetTrackEtaOnEMCal (Int_t var) { fTrackEtaOnEMCal = var; }
137	void SetTrackPtOnEMCal (Int_t var) { fTrackPtOnEMCal = var; }
138	void SetITSsignal (Int_t var) { fITSsignal = var; }
139	void SetTPCsignal (Int_t var) { fTPCsignal = var; }
140	void SetTPCsignalTuned (Int_t var) { fTPCsignalTuned = var; }
141	void SetTPCsignalN (Int_t var) { fTPCsignalN = var; }
142	void SetTPCmomentum (Int_t var) { fTPCmomentum = var; }
143	void SetTPCTgl (Int_t var) { fTPCTgl = var; }
144	void SetTOFsignal (Int_t var) { fTOFsignal = var; }
145	void SetintegratedLenght (Int_t var) { fintegratedLenght = var; }
146	void SetTOFsignalTuned (Int_t var) { fTOFsignalTuned = var; }
147	void SetHMPIDsignal (Int_t var) { fHMPIDsignal = var; }
148	void SetHMPIDoccupancy (Int_t var) { fHMPIDoccupancy = var; }
149	void SetTRDsignal (Int_t var) { fTRDsignal = var; }
150	void SetTRDChi2 (Int_t var) { fTRDChi2 = var; }
151	void SetTRDnSlices (Int_t var) { fTRDnSlices = var; }
152	void Setcovmat (Int_t var) { fcovmat = var; }
153
154	static AliNanoAODTrackMapping * fInstance; //instance, needed for the singleton implementation
155	static TString fMappingString; // the string which this class was initialized with
cfd24ec2	156	std::map<TString,int> fMapCstVar;// Map of indexes of custom variables: CACHE THIS TO CONST INTs IN YOUR TASK TO AVOID CONTINUOUS STRING COMPARISONS
778636d4	157	ClassDef(AliNanoAODTrackMapping, 1)
	158
	159	};
	160
	161	#endif /* _ALINANOAODTRACKMAPPING_H_ */