bugfix: memory leak corrected (Stefan Kirsch)
[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 Theodor Rascanu
21     @date   
22     @brief  Handles high level (object based) and low level (pod) input data (clusters). For debug purposes only
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 const char* AliHLTTRDOfflineTrackerV1Component::GetComponentID()
90 {
91   // Return the component ID const char *
92   return "TRDOfflineTrackerV1"; // The ID of this component
93 }
94
95 int AliHLTTRDOfflineTrackerV1Component::SetParams()
96 {
97   int iResult = AliHLTTRDTrackerV1Component::SetParams();
98   fRecoParam->SetStreamLevel(AliTRDrecoParam::kTracker, 1); // in order to have the friends written
99   return iResult;
100 }
101
102 int AliHLTTRDOfflineTrackerV1Component::DoEvent(const AliHLTComponent_EventData& evtData, const AliHLTComponent_BlockData* blocks, 
103                                                   AliHLTComponent_TriggerData& trigData, AliHLTUInt8_t* outputPtr, 
104                                                   AliHLTUInt32_t& size, vector<AliHLTComponent_BlockData>& outputBlocks )
105 {
106   if(!IsDataEvent())return 0;
107   
108   if(!GetFirstInputBlock(AliHLTTRDDefinitions::fgkHiLvlClusterDataType))
109     return AliHLTTRDTrackerV1Component::DoEvent(evtData, blocks, trigData, outputPtr, size, outputBlocks );
110
111   for(const TObject *iter = GetFirstInputObject(AliHLTTRDDefinitions::fgkHiLvlClusterDataType); iter; iter = GetNextInputObject()) 
112     {
113       TClonesArray* clusterArray = dynamic_cast<TClonesArray*>(const_cast<TObject*>(iter));
114       if(!clusterArray)continue;
115       TObjString* strg = dynamic_cast<TObjString*>(const_cast<TObject*>(GetNextInputObject()));
116       if(!strg)continue;
117       
118       fNtimeBins = strg->String().Atoi();
119       fESD->Reset();
120       AliTRDtrackerV1::SetNTimeBins(fNtimeBins);
121       HLTDebug("TClonesArray of clusters: nbEntries = %i", clusterArray->GetEntriesFast());
122       fTracker->LoadClusters(clusterArray);
123       fTracker->Clusters2Tracks(fESD);
124       Int_t nTracks = fESD->GetNumberOfTracks();
125       HLTInfo("Number of tracks  == %d ==", nTracks);  
126       TClonesArray* trdTracks = fTracker->GetListOfTracks();
127
128       if(fEmulateHLTTracks && trdTracks){
129         trdTracks = new TClonesArray(*trdTracks);
130         AliHLTTRDUtils::EmulateHLTTracks(trdTracks);
131       }
132
133       AliHLTUInt32_t spec = GetSpecification(iter);
134       if(trdTracks)
135         PushBack(trdTracks, AliHLTTRDDefinitions::fgkHiLvlTracksDataType, spec);
136       else{
137         TClonesArray temp("AliTRDtrackV1");
138         PushBack(&temp, AliHLTTRDDefinitions::fgkHiLvlTracksDataType, spec);
139       }
140       PushBack(strg, AliHLTTRDDefinitions::fgkHiLvlTracksDataType, spec);
141       fTracker->UnloadClusters();
142       AliTRDReconstructor::SetClusters(0x0);
143
144       if(fEmulateHLTTracks && trdTracks){
145         trdTracks->Delete();
146         delete trdTracks;
147       }
148     }
149   return 0;
150 }