]>
Commit | Line | Data |
---|---|---|
873458ab | 1 | #if ! defined (__CINT__) || defined (__MAKECINT__) |
2 | #include "TTree.h" | |
72f5f463 | 3 | #include "AliLog.h" |
873458ab | 4 | #include "AliAnalysisManager.h" |
5 | #include "AliAnalysisDataContainer.h" | |
0b790af7 | 6 | #include "TRD/qaRec/macros/AliTRDperformanceTrain.h" |
7 | #include "TRD/qaRec/AliTRDcheckPID.h" | |
72f5f463 | 8 | #include "TRD/qaRec/AliTRDpidRefMakerNN.h" |
9 | #include "TRD/qaRec/AliTRDpidRefMakerLQ.h" | |
873458ab | 10 | #endif |
11 | ||
c82c70b6 | 12 | #include "TRD/qaRec/macros/helper.C" |
0b790af7 | 13 | void AddTRDcheckPID(AliAnalysisManager *mgr, Char_t *trd, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/) |
873458ab | 14 | { |
0b790af7 | 15 | Int_t map = ParseOptions(trd); |
ea4502f6 | 16 | if(TSTBIT(map, kCheckPID)){ |
17 | AliTRDcheckPID *pid = 0x0; | |
18 | mgr->AddTask(pid = new AliTRDcheckPID()); | |
19 | pid->SetDebugLevel(0); | |
20 | pid->SetMCdata(mgr->GetMCtruthEventHandler()); | |
21 | mgr->ConnectInput(pid, 0, ci[0]); | |
22 | mgr->ConnectOutput(pid, 0, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.Performance.root")); | |
23 | } | |
0b790af7 | 24 | |
ea4502f6 | 25 | if(TSTBIT(map, kPIDRefMakerNN)){ |
26 | // TRD NN pid reference maker | |
27 | AliTRDpidRefMakerNN *ref = new AliTRDpidRefMakerNN(); | |
28 | //AliLog::SetClassDebugLevel("AliTRDpidRefMakerNN", 4); | |
29 | // Neural network PID | |
30 | mgr->AddTask(ref); | |
31 | ref->SetDebugLevel(3); | |
32 | AliLog::SetClassDebugLevel("AliTRDpidRefMakerNN", 3); | |
33 | ref->SetMCdata(mgr->GetMCtruthEventHandler()); | |
34 | mgr->ConnectInput( ref, 0, ci[0]); | |
35 | mgr->ConnectInput( ref, 1, ci[2]); | |
36 | mgr->ConnectOutput(ref, 0, mgr->CreateContainer(Form("Moni%s", ref->GetName()), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName()))); | |
37 | mgr->ConnectOutput(ref, 1, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName()))); | |
38 | } | |
72f5f463 | 39 | |
40 | // Multidimensional Likelihood PID | |
ea4502f6 | 41 | if(TSTBIT(map, kPIDRefMakerLQ)){ |
42 | AliTRDpidRefMakerLQ *reflq = new AliTRDpidRefMakerLQ(); | |
43 | mgr->AddTask(reflq); | |
44 | reflq->SetDebugLevel(3); | |
45 | //AliLog::SetClassDebugLevel("AliTRDpidRefMakerLQ", 3); | |
46 | reflq->SetMCdata(mgr->GetMCtruthEventHandler()); | |
47 | mgr->ConnectInput( reflq, 0, ci[0]); | |
48 | mgr->ConnectInput( reflq, 1, ci[2]); | |
49 | mgr->ConnectOutput(reflq, 0, mgr->CreateContainer(Form("Moni%s", reflq->GetName()), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", reflq->GetName()))); | |
50 | mgr->ConnectOutput(reflq, 1, mgr->CreateContainer(reflq->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", reflq->GetName()))); | |
51 | } | |
873458ab | 52 | } |
53 |