0a14d9761d4bbb5d6816da4c7ccec57b939fcbe4
[u/mrichter/AliRoot.git] / HLT / TRD / AliHLTTRDOfflineTrackerV1Component.cxx
1 // $Id$
2
3 //**************************************************************************
4 //* This file is property of and copyright by the ALICE HLT Project        * 
5 //* ALICE Experiment at CERN, All rights reserved.                         *
6 //*                                                                        *
7 //* Primary Authors:                                                       *
8 //*                  for The ALICE HLT Project.                            *
9 //*                                                                        *
10 //* Permission to use, copy, modify and distribute this software and its   *
11 //* documentation strictly for non-commercial purposes is hereby granted   *
12 //* without fee, provided that the above copyright notice appears in all   *
13 //* copies and that both the copyright notice and this permission notice   *
14 //* appear in the supporting documentation. The authors make no claims     *
15 //* about the suitability of this software for any purpose. It is          *
16 //* provided "as is" without express or implied warranty.                  *
17 //**************************************************************************
18
19 /** @file   AliHLTTRDOfflineTrackerV1Component.cxx
20     @author 
21     @date   2009-08-31
22     @brief  A TRDClusterizer processing component for the HLT. 
23 */
24
25 #include "AliHLTTRDOfflineTrackerV1Component.h"
26 #include "AliHLTTRDDefinitions.h"
27 #include "AliHLTTRDUtils.h"
28 #include "AliTRDrecoParam.h"
29 #include "AliTRDtrackerV1.h"
30 #include "AliTRDReconstructor.h"
31 #include "AliCDBManager.h"
32 #include "AliESDEvent.h"
33 #include "TClonesArray.h"
34 #include "TObjString.h"
35
36 #include "AliTRDtrackV1.h"
37 #include "AliTRDseedV1.h"
38 #include "AliTRDcluster.h"
39
40 ClassImp(AliHLTTRDOfflineTrackerV1Component)
41     
42 AliHLTTRDOfflineTrackerV1Component::AliHLTTRDOfflineTrackerV1Component()
43   :AliHLTTRDTrackerV1Component()
44 {
45   // Default constructor
46 }
47
48 AliHLTTRDOfflineTrackerV1Component::~AliHLTTRDOfflineTrackerV1Component()
49 {
50   // Destructor
51   // Work is Done in DoDeInit()
52 }
53
54 void AliHLTTRDOfflineTrackerV1Component::GetInputDataTypes( vector<AliHLTComponent_DataType>& list)
55 {
56   // Get the list of input data
57   list.clear();
58   AliHLTTRDTrackerV1Component::GetInputDataTypes(list);
59   list.push_back(AliHLTTRDDefinitions::fgkHiLvlClusterDataType);
60 }
61
62 AliHLTComponentDataType AliHLTTRDOfflineTrackerV1Component::GetOutputDataType()
63 {
64   // Get the output data type
65   return kAliHLTMultipleDataType;
66 }
67
68 int AliHLTTRDOfflineTrackerV1Component::GetOutputDataTypes(AliHLTComponentDataTypeList& tgtList)
69 {
70   // Get the output data types
71   tgtList.clear();
72   tgtList.push_back(AliHLTTRDDefinitions::fgkHiLvlTracksDataType);
73   return tgtList.size();
74 }
75
76 void AliHLTTRDOfflineTrackerV1Component::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier )
77 {
78   // Get the output data size
79   AliHLTTRDTrackerV1Component::GetOutputDataSize(constBase, inputMultiplier);
80   constBase += 500;
81 }
82
83 AliHLTComponent* AliHLTTRDOfflineTrackerV1Component::Spawn()
84 {
85   // Spawn function, return new instance of this class
86   return new AliHLTTRDOfflineTrackerV1Component;
87 };
88
89 int AliHLTTRDOfflineTrackerV1Component::DoInit( int argc, const char** argv )
90 {
91   int iResult = 0;
92   iResult=AliHLTTRDTrackerV1Component::DoInit(argc, argv);
93   fRecoParam->SetStreamLevel(AliTRDrecoParam::kTracker, 1); // in order to have the friends written
94   return iResult;
95 }
96
97 const char* AliHLTTRDOfflineTrackerV1Component::GetComponentID()
98 {
99   // Return the component ID const char *
100   return "TRDOfflineTrackerV1"; // The ID of this component
101 }
102
103 int AliHLTTRDOfflineTrackerV1Component::DoEvent(const AliHLTComponent_EventData& evtData, const AliHLTComponent_BlockData* blocks, 
104                                                   AliHLTComponent_TriggerData& trigData, AliHLTUInt8_t* outputPtr, 
105                                                   AliHLTUInt32_t& size, vector<AliHLTComponent_BlockData>& outputBlocks )
106 {
107   if ( GetFirstInputBlock( kAliHLTDataTypeSOR ) || GetFirstInputBlock( kAliHLTDataTypeEOR ) )
108     return 0;
109   
110   if(!GetFirstInputBlock(AliHLTTRDDefinitions::fgkHiLvlClusterDataType))
111     return AliHLTTRDTrackerV1Component::DoEvent(evtData, blocks, trigData, outputPtr, size, outputBlocks );
112
113   for(const TObject *iter = GetFirstInputObject(AliHLTTRDDefinitions::fgkHiLvlClusterDataType); iter; iter = GetNextInputObject()) 
114     {
115       TClonesArray* clusterArray = dynamic_cast<TClonesArray*>(const_cast<TObject*>(iter));
116       if(!clusterArray)continue;
117       TObjString* strg = dynamic_cast<TObjString*>(const_cast<TObject*>(GetNextInputObject()));
118       if(!strg)continue;
119       
120       fNtimeBins = strg->String().Atoi();
121       fESD->Reset();
122       AliTRDtrackerV1::SetNTimeBins(fNtimeBins);
123       HLTDebug("TClonesArray of clusters: nbEntries = %i", clusterArray->GetEntriesFast());
124       fTracker->LoadClusters(clusterArray);
125       fTracker->Clusters2Tracks(fESD);
126       Int_t nTracks = fESD->GetNumberOfTracks();
127       HLTInfo("Number of tracks  == %d ==", nTracks);  
128       TClonesArray* trdTracks = fTracker->GetListOfTracks();
129
130       if(fEmulateHLTTracks && trdTracks){
131         trdTracks = new TClonesArray(*trdTracks);
132         AliHLTTRDUtils::EmulateHLTTracks(trdTracks);
133       }
134
135       PushBack(trdTracks, AliHLTTRDDefinitions::fgkHiLvlTracksDataType, 0);
136       PushBack(strg, AliHLTTRDDefinitions::fgkHiLvlTracksDataType, 0);
137       fTracker->UnloadClusters();
138       AliTRDReconstructor::SetClusters(0x0);
139
140       if(fEmulateHLTTracks && trdTracks){
141         trdTracks->Delete();
142         delete trdTracks;
143       }
144     }
145   return 0;
146 }