1 #if ! defined (__CINT__) || defined (__MAKECINT__)
5 #include "AliAnalysisManager.h"
6 #include "AliAnalysisDataContainer.h"
7 #include "PWGPP/TRD/AliTRDpwgppHelper.h"
8 #include "PWGPP/TRD/AliTRDcheckPID.h"
9 #include "PWGPP/TRD/AliTRDpidRefMaker.h"
10 #include "PWGPP/TRD/AliTRDpidRefMakerNN.h"
11 #include "PWGPP/TRD/AliTRDpidRefMakerLQ.h"
14 void AddTRDcheckPID(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer **ci, AliAnalysisDataContainer **co)
16 Info("AddTRDcheckPID", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName()));
18 AliTRDcheckPID *pid(NULL);
19 mgr->AddTask(pid = new AliTRDcheckPID((char*)"TRDcheckPID"));
20 //AliLog::SetClassDebugLevel("AliTRDcheckPID", 5);
21 pid->SetDebugLevel(0);
22 pid->SetMCdata(mgr->GetMCtruthEventHandler());
24 // define PID exchange container
25 co[0] = mgr->CreateContainer("InfoPID", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
26 mgr->ConnectInput (pid, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
27 mgr->ConnectInput (pid, 1, ci[0]); // connect barrel tracks container
28 mgr->ConnectInput (pid, 2, ci[1]); // connect event info container
29 mgr->ConnectInput (pid, 3, ci[2]); // connect V0s container
30 mgr->ConnectOutput(pid, 1, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance",mgr->GetCommonFileName())));
31 mgr->ConnectOutput(pid, 2, co[0]);
33 if(TESTBIT(map, AliTRDpwgppHelper::kPIDRefMaker)){
35 //AliLog::SetClassDebugLevel("AliTRDpidRefMaker", 3);
36 //AliLog::SetClassDebugLevel("AliTRDpidRefMakerNN", 3);
37 //AliLog::SetClassDebugLevel("AliTRDpidRefMakerLQ", 3);
39 // TRD pid reference maker NN
40 AliTRDpidRefMaker *ref(NULL);
41 mgr->AddTask(ref = new AliTRDpidRefMakerNN((char*)"TRDrefMakerNN"));
42 ref->SetDebugLevel(3);
43 ref->SetMCdata(mgr->GetMCtruthEventHandler());
44 ref->SetFriends(kTRUE);
45 mgr->ConnectInput( ref, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
46 mgr->ConnectInput( ref, 1, ci[0]); // connect barrel tracks container
47 mgr->ConnectInput( ref, 2, ci[1]); // connect event info container
48 mgr->ConnectInput( ref, 3, ci[2]); // connect V0s container
49 mgr->ConnectInput( ref, 4, co[0]); // connect pid Info container
50 mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorNN", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration",mgr->GetCommonFileName())));
51 mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
53 // TRD pid reference maker LQ
54 mgr->AddTask(ref = new AliTRDpidRefMakerLQ((char*)"TRDrefMakerLQ"));
55 ref->SetDebugLevel(3);
56 ref->SetMCdata(mgr->GetMCtruthEventHandler());
57 ref->SetFriends(kTRUE);
58 mgr->ConnectInput(ref, 0, mgr->GetCommonInputContainer()); // connect main (ESD) container
59 mgr->ConnectInput(ref, 1, ci[0]); // connect barrel tracks container
60 mgr->ConnectInput(ref, 2, ci[1]); // connect event info container
61 mgr->ConnectInput(ref, 3, ci[2]); // connect V0s container
62 mgr->ConnectInput(ref, 4, co[0]); // connect pid Info container
63 mgr->ConnectOutput(ref, 1, mgr->CreateContainer("MonitorLQ", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
64 mgr->ConnectOutput(ref, 2, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
65 mgr->ConnectOutput(ref, 3, mgr->CreateContainer("PDF", TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));