[u/mrichter/AliRoot.git] / PWG1 / macros / AddTrainPerformanceTRD.C

// Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).\r
// \r
// Usage:\r
//   AddTrainPerformanceTRD.C(MC, friends, tasks)\r
//   tasks : "ALL" or one/more of the following:\r
//     "EFF"  : TRD Tracking Efficiency \r
//     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations\r
//     "MULT"  : TRD single track selection\r
//     "RES"  : TRD tracking Resolution\r
//     "CLRES": clusters Resolution\r
//     "CAL"  : TRD calibration\r
//     "ALGN" : TRD alignment\r
//     "PID"  : TRD PID - pion efficiency \r
//     "PIDR" : TRD PID - reference data\r
//     "DET"  : Basic TRD Detector checks\r
//     "NOFR" : Data set does not have AliESDfriends.root \r
//     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely\r
//              on MC information are switched off\r
//\r
// In compiled mode : \r
// Don't forget to load first the libraries\r
// gSystem->Load("libMemStat.so")\r
// gSystem->Load("libMemStatGui.so")\r
// gSystem->Load("libANALYSIS.so")\r
// gSystem->Load("libANALYSISalice.so")\r
// gSystem->Load("libPWG1.so");\r
//\r
// Authors:\r
//   Alex Bercuci (A.Bercuci@gsi.de) \r
//   Markus Fasel (m.Fasel@gsi.de) \r
\r
#if ! defined (__CINT__) || defined (__MAKECINT__)\r
//#ifndef __CINT__\r
#include <Riostream.h>\r
\r
#include "TStopwatch.h"\r
#include "TMemStat.h"\r
#include "TMemStatViewerGUI.h"\r
\r
#include "TROOT.h"\r
#include "TClass.h"\r
#include "TSystem.h"\r
#include "TString.h"\r
#include "TError.h"\r
#include "TChain.h"\r
#include "TGrid.h"\r
#include "TAlienCollection.h"\r
#include "TGridCollection.h"\r
#include "TGridResult.h"\r
#include "TGeoGlobalMagField.h"\r
\r
#include "AliMagF.h"\r
#include "AliTracker.h"\r
#include "AliLog.h"\r
#include "AliCDBManager.h"\r
#include "AliGRPManager.h"\r
#include "AliGeomManager.h"\r
#include "AliAnalysisManager.h"\r
#include "AliAnalysisDataContainer.h"\r
#include "AliMCEventHandler.h"\r
#include "AliESDInputHandler.h"\r
\r
#include "TRD/AliTRDtrackerV1.h"\r
#include "TRD/AliTRDcalibDB.h"\r
\r
#include "PWG1/TRD/macros/AddTRDcheckESD.C"\r
#include "PWG1/TRD/macros/AddTRDinfoGen.C"\r
#include "PWG1/TRD/macros/AddTRDcheckDET.C"\r
#include "PWG1/TRD/macros/AddTRDefficiency.C"\r
#include "PWG1/TRD/macros/AddTRDresolution.C"\r
#include "PWG1/TRD/macros/AddTRDcheckPID.C"\r
#include "PWG1/TRD/macros/AddTRDcheckTRK.C"\r
#include "PWG1/TRD/macros/AddTRDv0Monitor.C"\r
#endif\r
\r
TString opt("");\r
const Char_t* Translate(Bool_t doCheckESD=kTRUE, Bool_t doCheckDET=kTRUE, Bool_t doEffic=kTRUE, Bool_t doResolution=kTRUE, Bool_t doCheckPID=kTRUE, Bool_t doV0Monitor=kTRUE);\r
Bool_t AddTrainPerformanceTRD(Char_t *trd="ALL", const Char_t *addMacroPath = "$ALICE_ROOT/PWG1/TRD/macros")\r
{\r
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
  if(!mgr) { \r
    Error("AddTrainPerformanceTRD", "AliAnalysisManager not set!");\r
    return kFALSE;\r
  }\r
\r
  // TRD data containers\r
  AliAnalysisDataContainer *ci[AliTRDpwg1Helper::kNOutSlots];\r
  AliAnalysisDataContainer *ce[5];\r
\r
  Info("AddTrainPerformanceTRD", Form("Add Macros taken from %s", addMacroPath));\r
  Info("AddTrainPerformanceTRD", Form("TRD wagons \"%s\"", trd));\r
  Int_t bitmap = AliTRDpwg1Helper::ParseOptions(trd);\r
  for(Int_t it=0; it<AliTRDpwg1Helper::kNTRDQATASKS; it++){\r
    if(gROOT->LoadMacro(Form("%s/Add%s.C+", addMacroPath, TString(AliTRDpwg1Helper::TaskClassName(it))(3,20).Data()))) {\r
      Error("AddTrainPerformanceTRD()", Form("Error loading %s task.", AliTRDpwg1Helper::TaskClassName(it)));\r
      return kFALSE;\r
    } \r
    if(!AliTRDpwg1Helper::DoTask(it, bitmap)) continue;\r
\r
    switch(it){\r
    case AliTRDpwg1Helper::kCheckESD:\r
      AddTRDcheckESD(mgr); break;\r
    case AliTRDpwg1Helper::kInfoGen:\r
      AddTRDinfoGen(mgr, 0, NULL, ci); break;\r
    case AliTRDpwg1Helper::kCheckDET:\r
      // map slots\r
      ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
      ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
      ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
      ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
      AddTRDcheckDET(mgr, bitmap, ce);\r
      break;\r
    case AliTRDpwg1Helper::kEfficiency:\r
      // map slots\r
      ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
      ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
      ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
      ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
      AddTRDefficiency(mgr, bitmap, ce);\r
      break;\r
    case AliTRDpwg1Helper::kResolution:\r
      // map slots\r
      ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
      ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
      ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
      ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
      AddTRDresolution(mgr, bitmap, ce); \r
      break;\r
    case AliTRDpwg1Helper::kCheckPID:\r
      // map slots\r
      ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
      ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
      ce[2]=ci[AliTRDpwg1Helper::kV0List];\r
      AddTRDcheckPID(mgr, bitmap, ce, &ce[3]);\r
      break;\r
    case AliTRDpwg1Helper::kCheckTRK:\r
      // map slots\r
      ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
      ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
      AddTRDcheckTRK(mgr, 0, ce);\r
      break;\r
    case AliTRDpwg1Helper::kV0Monitor:\r
      // slots already mapped by checkPID\r
      AddTRDv0Monitor(mgr, 0, ce);\r
      break;\r
    default:\r
      Warning("AddTrainPerformanceTRD()", Form("No performance task registered at slot %d.", it)); \r
    }\r
  }\r
  return kTRUE;\r
}\r
\r
const Char_t* Translate(Bool_t doCheckESD, Bool_t doCheckDET, Bool_t doEffic, Bool_t doResolution, Bool_t doCheckPID, Bool_t doCheckV0)\r
{\r
  opt.Clear();\r
  if( doCheckESD==kTRUE &&\r
      doCheckDET==kTRUE &&\r
      doEffic==kTRUE &&\r
      doResolution==kTRUE &&\r
      doCheckPID==kTRUE &&\r
      doCheckV0==kTRUE\r
  ){\r
    opt="ALL";\r
  } else {\r
    Bool_t kINDENT(kFALSE);\r
    if(doCheckESD){ \r
      opt.Append("ESD");\r
      kINDENT=kTRUE;\r
    }\r
    if(doCheckDET){ \r
      if(kINDENT) opt.Append(" ");\r
      opt.Append("DET"); \r
      kINDENT = kTRUE;\r
    }\r
    if(doEffic){ \r
      if(kINDENT) opt.Append(" ");\r
      opt.Append("EFF");\r
      kINDENT=kTRUE;\r
    }\r
    if(doResolution){ \r
      if(kINDENT) opt.Append(" ");\r
      opt.Append("RES");\r
      kINDENT=kTRUE;\r
    }\r
    if(doCheckPID){ \r
      if(kINDENT) opt.Append(" ");\r
      opt.Append("PID");\r
      kINDENT=kTRUE;\r
    }\r
    if(doCheckV0){ \r
      if(kINDENT) opt.Append(" ");\r
      opt.Append("V0");\r
      kINDENT=kTRUE;\r
    }\r
  }\r
\r
  return (const Char_t*)opt.Data();\r
}\r
\r
\r
Commit	Line	Data
94b94be0	1	// Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).\r
	2	// \r
	3	// Usage:\r
	4	// AddTrainPerformanceTRD.C(MC, friends, tasks)\r
	5	// tasks : "ALL" or one/more of the following:\r
	6	// "EFF" : TRD Tracking Efficiency \r
	7	// "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations\r
	8	// "MULT" : TRD single track selection\r
	9	// "RES" : TRD tracking Resolution\r
	10	// "CLRES": clusters Resolution\r
	11	// "CAL" : TRD calibration\r
	12	// "ALGN" : TRD alignment\r
	13	// "PID" : TRD PID - pion efficiency \r
	14	// "PIDR" : TRD PID - reference data\r
	15	// "DET" : Basic TRD Detector checks\r
	16	// "NOFR" : Data set does not have AliESDfriends.root \r
	17	// "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely\r
	18	// on MC information are switched off\r
	19	//\r
	20	// In compiled mode : \r
	21	// Don't forget to load first the libraries\r
	22	// gSystem->Load("libMemStat.so")\r
	23	// gSystem->Load("libMemStatGui.so")\r
	24	// gSystem->Load("libANALYSIS.so")\r
	25	// gSystem->Load("libANALYSISalice.so")\r
	26	// gSystem->Load("libPWG1.so");\r
	27	//\r
	28	// Authors:\r
	29	// Alex Bercuci (A.Bercuci@gsi.de) \r
	30	// Markus Fasel (m.Fasel@gsi.de) \r
	31	\r
	32	#if ! defined (__CINT__) \|\| defined (__MAKECINT__)\r
	33	//#ifndef __CINT__\r
	34	#include <Riostream.h>\r
	35	\r
	36	#include "TStopwatch.h"\r
	37	#include "TMemStat.h"\r
	38	#include "TMemStatViewerGUI.h"\r
	39	\r
	40	#include "TROOT.h"\r
	41	#include "TClass.h"\r
	42	#include "TSystem.h"\r
	43	#include "TString.h"\r
	44	#include "TError.h"\r
	45	#include "TChain.h"\r
	46	#include "TGrid.h"\r
	47	#include "TAlienCollection.h"\r
	48	#include "TGridCollection.h"\r
	49	#include "TGridResult.h"\r
	50	#include "TGeoGlobalMagField.h"\r
	51	\r
	52	#include "AliMagF.h"\r
	53	#include "AliTracker.h"\r
	54	#include "AliLog.h"\r
	55	#include "AliCDBManager.h"\r
	56	#include "AliGRPManager.h"\r
	57	#include "AliGeomManager.h"\r
	58	#include "AliAnalysisManager.h"\r
	59	#include "AliAnalysisDataContainer.h"\r
	60	#include "AliMCEventHandler.h"\r
	61	#include "AliESDInputHandler.h"\r
	62	\r
	63	#include "TRD/AliTRDtrackerV1.h"\r
	64	#include "TRD/AliTRDcalibDB.h"\r
65	\r
66	#include "PWG1/TRD/macros/AddTRDcheckESD.C"\r
67	#include "PWG1/TRD/macros/AddTRDinfoGen.C"\r
68	#include "PWG1/TRD/macros/AddTRDcheckDET.C"\r
69	#include "PWG1/TRD/macros/AddTRDefficiency.C"\r
70	#include "PWG1/TRD/macros/AddTRDresolution.C"\r
71	#include "PWG1/TRD/macros/AddTRDcheckPID.C"\r
3ceb45ae	72	#include "PWG1/TRD/macros/AddTRDcheckTRK.C"\r
94b94be0	73	#include "PWG1/TRD/macros/AddTRDv0Monitor.C"\r
	74	#endif\r
	75	\r
	76	TString opt("");\r
	77	const Char_t* Translate(Bool_t doCheckESD=kTRUE, Bool_t doCheckDET=kTRUE, Bool_t doEffic=kTRUE, Bool_t doResolution=kTRUE, Bool_t doCheckPID=kTRUE, Bool_t doV0Monitor=kTRUE);\r
	78	Bool_t AddTrainPerformanceTRD(Char_t trd="ALL", const Char_t addMacroPath = "$ALICE_ROOT/PWG1/TRD/macros")\r
	79	{\r
	80	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();\r
	81	if(!mgr) { \r
	82	Error("AddTrainPerformanceTRD", "AliAnalysisManager not set!");\r
	83	return kFALSE;\r
	84	}\r
	85	\r
	86	// TRD data containers\r
	87	AliAnalysisDataContainer *ci[AliTRDpwg1Helper::kNOutSlots];\r
	88	AliAnalysisDataContainer *ce[5];\r
	89	\r
	90	Info("AddTrainPerformanceTRD", Form("Add Macros taken from %s", addMacroPath));\r
	91	Info("AddTrainPerformanceTRD", Form("TRD wagons \"%s\"", trd));\r
	92	Int_t bitmap = AliTRDpwg1Helper::ParseOptions(trd);\r
	93	for(Int_t it=0; it<AliTRDpwg1Helper::kNTRDQATASKS; it++){\r
	94	if(gROOT->LoadMacro(Form("%s/Add%s.C+", addMacroPath, TString(AliTRDpwg1Helper::TaskClassName(it))(3,20).Data()))) {\r
	95	Error("AddTrainPerformanceTRD()", Form("Error loading %s task.", AliTRDpwg1Helper::TaskClassName(it)));\r
	96	return kFALSE;\r
	97	} \r
	98	if(!AliTRDpwg1Helper::DoTask(it, bitmap)) continue;\r
	99	\r
	100	switch(it){\r
	101	case AliTRDpwg1Helper::kCheckESD:\r
	102	AddTRDcheckESD(mgr); break;\r
	103	case AliTRDpwg1Helper::kInfoGen:\r
	104	AddTRDinfoGen(mgr, 0, NULL, ci); break;\r
	105	case AliTRDpwg1Helper::kCheckDET:\r
	106	// map slots\r
	107	ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
	108	ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
	109	ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
	110	ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
	111	AddTRDcheckDET(mgr, bitmap, ce);\r
	112	break;\r
	113	case AliTRDpwg1Helper::kEfficiency:\r
	114	// map slots\r
	115	ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
	116	ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
	117	ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
	118	ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
	119	AddTRDefficiency(mgr, bitmap, ce);\r
	120	break;\r
	121	case AliTRDpwg1Helper::kResolution:\r
	122	// map slots\r
	123	ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
	124	ce[1]=ci[AliTRDpwg1Helper::kTracksSA];\r
	125	ce[2]=ci[AliTRDpwg1Helper::kTracksKink];\r
	126	ce[3]=ci[AliTRDpwg1Helper::kEventInfo];\r
	127	AddTRDresolution(mgr, bitmap, ce); \r
	128	break;\r
	129	case AliTRDpwg1Helper::kCheckPID:\r
	130	// map slots\r
	131	ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
	132	ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
	133	ce[2]=ci[AliTRDpwg1Helper::kV0List];\r
	134	AddTRDcheckPID(mgr, bitmap, ce, &ce[3]);\r
	135	break;\r
3ceb45ae	136	case AliTRDpwg1Helper::kCheckTRK:\r
	137	// map slots\r
	138	ce[0]=ci[AliTRDpwg1Helper::kTracksBarrel];\r
	139	ce[1]=ci[AliTRDpwg1Helper::kEventInfo];\r
	140	AddTRDcheckTRK(mgr, 0, ce);\r
	141	break;\r
94b94be0	142	case AliTRDpwg1Helper::kV0Monitor:\r
	143	// slots already mapped by checkPID\r
	144	AddTRDv0Monitor(mgr, 0, ce);\r
	145	break;\r
	146	default:\r
	147	Warning("AddTrainPerformanceTRD()", Form("No performance task registered at slot %d.", it)); \r
	148	}\r
	149	}\r
	150	return kTRUE;\r
	151	}\r
	152	\r
	153	const Char_t* Translate(Bool_t doCheckESD, Bool_t doCheckDET, Bool_t doEffic, Bool_t doResolution, Bool_t doCheckPID, Bool_t doCheckV0)\r
	154	{\r
	155	opt.Clear();\r
	156	if( doCheckESD==kTRUE &&\r
	157	doCheckDET==kTRUE &&\r
	158	doEffic==kTRUE &&\r
	159	doResolution==kTRUE &&\r
	160	doCheckPID==kTRUE &&\r
	161	doCheckV0==kTRUE\r
	162	){\r
	163	opt="ALL";\r
	164	} else {\r
	165	Bool_t kINDENT(kFALSE);\r
	166	if(doCheckESD){ \r
	167	opt.Append("ESD");\r
	168	kINDENT=kTRUE;\r
	169	}\r
	170	if(doCheckDET){ \r
	171	if(kINDENT) opt.Append(" ");\r
	172	opt.Append("DET"); \r
	173	kINDENT = kTRUE;\r
	174	}\r
	175	if(doEffic){ \r
	176	if(kINDENT) opt.Append(" ");\r
	177	opt.Append("EFF");\r
	178	kINDENT=kTRUE;\r
	179	}\r
	180	if(doResolution){ \r
	181	if(kINDENT) opt.Append(" ");\r
	182	opt.Append("RES");\r
	183	kINDENT=kTRUE;\r
	184	}\r
	185	if(doCheckPID){ \r
	186	if(kINDENT) opt.Append(" ");\r
	187	opt.Append("PID");\r
	188	kINDENT=kTRUE;\r
	189	}\r
	190	if(doCheckV0){ \r
	191	if(kINDENT) opt.Append(" ");\r
	192	opt.Append("V0");\r
	193	kINDENT=kTRUE;\r
	194	}\r
	195	}\r
	196	\r
	197	return (const Char_t*)opt.Data();\r
	198	}\r
	199	\r
	200	\r