]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG1/TRD/macros/AddTRDinfoGen.C
374ca6418a928ba1a487a7a5873595e86671580b
[u/mrichter/AliRoot.git] / PWG1 / TRD / macros / AddTRDinfoGen.C
1 #if ! defined (__CINT__) || defined (__MAKECINT__)
2 #include "AliLog.h"
3 #include "AliAnalysisManager.h"
4 #include "AliAnalysisDataContainer.h"
5 #include "PWG1/TRD/AliTRDinfoGen.h"
6 #include "PWG1/TRD/info/AliTRDeventInfo.h"
7 #include "PWG1/TRD/macros/AliTRDperformanceTrain.h"
8 #endif
9
10 #include "PWG1/TRD/macros/helper.C"
11 void AddTRDinfoGen(AliAnalysisManager *mgr, Char_t *trd, AliAnalysisDataContainer **/*ci*/, AliAnalysisDataContainer **co)
12 {
13   Int_t map = ParseOptions(trd);
14   if(!(TSTBIT(map, kInfoGen))) return;
15   
16   AliTRDinfoGen *info(NULL);
17   mgr->AddTask(info = new AliTRDinfoGen((char*)"genInfo"));
18   info->SetDebugLevel(0);
19   //AliLog::SetClassDebugLevel("AliTRDinfoGen", 5);
20   info->SetMCdata(mgr->GetMCtruthEventHandler());
21
22   // settings for collisions
23   info->SetCollision(kFALSE);
24   if(info->IsCollision()){
25     info->SetTrigger(
26       "CINT1B-ABCE-NOPF-ALL"
27       " CSCO1-ABCE-NOPF-CENT" // cosmic SPD trigger
28     );
29     info->SetLocalEvSelection();
30     info->SetLocalTrkSelection();
31   }
32   
33   // Connect IO slots
34   mgr->ConnectInput (info, 0, mgr->GetCommonInputContainer());
35   co[kEventInfo] = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer);
36   co[kTracksBarrel] = mgr->CreateContainer("tracksBarrel", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
37   co[kTracksSA] = mgr->CreateContainer("tracksSA", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
38   co[kTracksKink] = mgr->CreateContainer("tracksKink", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
39   co[kV0List] = mgr->CreateContainer("v0List", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
40   for(Int_t ios(1);ios<kNOutSlots;ios++) mgr->ConnectOutput(info, ios, co[ios]);
41 }
42