]>
Commit | Line | Data |
---|---|---|
808618f5 | 1 | /************************************************************************** |
2 | * This file is property of and copyright by the ALICE HLT Project * | |
3 | * ALICE Experiment at CERN, All rights reserved. * | |
4 | * * | |
5 | * Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> * | |
6 | * for The ALICE HLT Project. * | |
7 | * * | |
8 | * Permission to use, copy, modify and distribute this software and its * | |
9 | * documentation strictly for non-commercial purposes is hereby granted * | |
10 | * without fee, provided that the above copyright notice appears in all * | |
11 | * copies and that both the copyright notice and this permission notice * | |
12 | * appear in the supporting documentation. The authors make no claims * | |
13 | * about the suitability of this software for any purpose. It is * | |
14 | * provided "as is" without express or implied warranty. * | |
15 | **************************************************************************/ | |
16 | ||
17 | /** @file AliHLTTRDEsdWriterComponent.cxx | |
18 | @author Mateusz Ploskon | |
19 | @date | |
20 | @brief Writer component to store tracks of the HLT TRD | |
21 | ||
22 | */ | |
23 | #include "AliHLTTRDEsdWriterComponent.h" | |
24 | #include "AliESDEvent.h" | |
25 | #include "AliESDtrack.h" | |
26 | #include "TTree.h" | |
27 | #include "AliHLTTRDDefinitions.h" | |
28 | ||
29 | /** global instance for component registration */ | |
30 | AliHLTTRDEsdWriterComponent gTRDEsdWriter; | |
31 | ||
32 | /** ROOT macro for the implementation of ROOT specific class methods */ | |
33 | ClassImp(AliHLTTRDEsdWriterComponent) | |
34 | ||
35 | AliHLTTRDEsdWriterComponent::AliHLTTRDEsdWriterComponent() | |
36 | : | |
37 | fTree(NULL), | |
38 | fESD(NULL) | |
39 | { | |
40 | // see header file for class documentation | |
41 | // or | |
42 | // refer to README to build package | |
43 | // or | |
44 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
45 | } | |
46 | ||
47 | AliHLTTRDEsdWriterComponent::AliHLTTRDEsdWriterComponent(const AliHLTTRDEsdWriterComponent&) | |
48 | : | |
49 | fTree(NULL), | |
50 | fESD(NULL) | |
51 | { | |
52 | } | |
53 | ||
54 | AliHLTTRDEsdWriterComponent& AliHLTTRDEsdWriterComponent::operator=(const AliHLTTRDEsdWriterComponent&) | |
55 | { | |
56 | return *this; | |
57 | } | |
58 | ||
59 | void AliHLTTRDEsdWriterComponent::GetInputDataTypes( vector<AliHLTComponent_DataType>& list) | |
60 | { | |
61 | // Get the list of input data | |
62 | list.clear(); // We do not have any requirements for our input data type(s). | |
63 | list.push_back( AliHLTTRDDefinitions::fgkTRDSATracksDataType ); | |
64 | } | |
65 | ||
66 | AliHLTTRDEsdWriterComponent::~AliHLTTRDEsdWriterComponent() | |
67 | { | |
68 | // see header file for class documentation | |
69 | } | |
70 | ||
71 | int AliHLTTRDEsdWriterComponent::InitWriter() | |
72 | { | |
73 | // see header file for class documentation | |
74 | int iResult=0; | |
75 | fESD = new AliESDEvent; | |
76 | if (fESD) { | |
77 | fESD->CreateStdContent(); | |
78 | fTree = new TTree("esdTree", "Tree with HLT ESD objects"); | |
79 | if (fTree) { | |
80 | fESD->WriteToTree(fTree); | |
81 | } | |
82 | } | |
83 | if (fTree==NULL) { | |
84 | iResult=-ENOMEM; | |
85 | } | |
86 | return iResult; | |
87 | } | |
88 | ||
89 | int AliHLTTRDEsdWriterComponent::CloseWriter() | |
90 | { | |
91 | // see header file for class documentation | |
92 | int iResult=0; | |
93 | if (fTree) { | |
94 | WriteObject(kAliHLTVoidEventID, fTree); | |
95 | TTree* pTree=fTree; | |
96 | fTree=NULL; | |
97 | delete pTree; | |
98 | } else { | |
99 | HLTWarning("not initialized"); | |
100 | } | |
101 | iResult=AliHLTRootFileWriterComponent::CloseWriter(); | |
102 | return iResult; | |
103 | } | |
104 | ||
105 | int AliHLTTRDEsdWriterComponent::DumpEvent( const AliHLTComponentEventData& evtData, | |
106 | const AliHLTComponentBlockData* blocks, | |
107 | AliHLTComponentTriggerData& trigData ) | |
108 | { | |
109 | // see header file for class documentation | |
110 | int iResult=0; | |
111 | TTree* pTree=fTree; | |
112 | ||
113 | AliHLTUInt32_t fDblock_Specification = 0; | |
114 | ||
115 | //implement a usage of the following | |
116 | // AliHLTUInt32_t triggerDataStructSize = trigData.fStructSize; | |
117 | // AliHLTUInt32_t triggerDataSize = trigData.fDataSize; | |
118 | // void *triggerData = trigData.fData; | |
119 | Logging( kHLTLogDebug, "HLT::TRDEsdWriter::DumpEvent", "Trigger data received", | |
120 | "Struct size %d Data size %d Data location 0x%x", trigData.fStructSize, trigData.fDataSize, (UInt_t*)trigData.fData); | |
121 | ||
122 | //AliHLTComponentBlockData *dblock = (AliHLTComponentBlockData *)GetFirstInputBlock( AliHLTTRDDefinitions::fgkTRDSAEsdDataType ); | |
123 | AliHLTComponentBlockData *dblock = (AliHLTComponentBlockData *)GetFirstInputBlock( AliHLTTRDDefinitions::fgkTRDSATracksDataType ); | |
124 | if (dblock != 0) | |
125 | { | |
126 | fDblock_Specification = dblock->fSpecification; | |
127 | } | |
128 | else | |
129 | { | |
130 | Logging( kHLTLogWarning, "HLT::TRDEsdWriter::DumpEvent", "DATAIN", "First Input Block not found! 0x%x", dblock); | |
131 | return -1; | |
132 | } | |
133 | ||
134 | int ibForce = 1; | |
135 | TObject *tobjin = (TObject *)GetFirstInputObject( AliHLTTRDDefinitions::fgkTRDSATracksDataType, "AliESDtrack", ibForce); | |
136 | Logging( kHLTLogInfo, "HLT::TRDEsdWriter::DumpEvent", "1stBLOCK", "Pointer = 0x%x", tobjin); | |
137 | ||
138 | AliESDtrack* track = (AliESDtrack *)tobjin; | |
139 | if (!track) | |
140 | { | |
141 | Logging( kHLTLogWarning, "HLT::TRDEsdWriter::DumpEvent", "DATAIN", "First Input Block not a ESDtrack! 0x%x", tobjin); | |
142 | return -1; | |
143 | } | |
144 | ||
145 | Int_t nTracks = 0; | |
146 | while (tobjin != 0) | |
147 | { | |
148 | if (track != 0) | |
149 | { | |
150 | //Logging( kHLTLogInfo, "HLT::TRDEsdWriter::DumpEvent", "Track found", "0x%x", track); | |
151 | Logging( kHLTLogInfo, "HLT::TRDEsdWriter::DumpEvent", "DONE", "Track %d 0x%x Pt %1.2f", nTracks, track, track->Pt()); | |
152 | fESD->AddTrack(track); | |
153 | nTracks++; | |
154 | } | |
155 | ||
156 | track = 0; | |
157 | tobjin = 0; | |
158 | tobjin = (TObject *)GetNextInputObject( ibForce ); | |
159 | //Logging( kHLTLogInfo, "HLT::TRDEsdWriter::DumpEvent", "nextBLOCK", "Pointer = 0x%x", tobjin); | |
160 | track = (AliESDtrack *)tobjin; | |
161 | } | |
162 | ||
163 | Logging( kHLTLogInfo, "HLT::TRDEsdWriter::DumpEvent", "Fill", "Ntracks: %d", nTracks); | |
164 | pTree->Fill(); | |
165 | fESD->Reset(); | |
166 | ||
167 | return iResult; | |
168 | } | |
169 | ||
170 | int AliHLTTRDEsdWriterComponent::ScanArgument(int argc, const char** argv) | |
171 | { | |
172 | // see header file for class documentation | |
173 | int iResult=AliHLTRootFileWriterComponent::ScanArgument(argc, argv); | |
174 | return iResult; | |
175 | } |