add SPD to plots

[u/mrichter/AliRoot.git] / HLT / global / AliHLTGlobalTrackMatcherComponent.cxx

diff --git a/HLT/global/AliHLTGlobalTrackMatcherComponent.cxx b/HLT/global/AliHLTGlobalTrackMatcherComponent.cxx
index 3c940395a2634d5d103a83faf6524d7a0d37c0a9..8b1efe9baa87f63e113845090f51338c735e19ce 100644 (file)
--- a/HLT/global/AliHLTGlobalTrackMatcherComponent.cxx
+++ b/HLT/global/AliHLTGlobalTrackMatcherComponent.cxx
@@ -43,7 +43,8 @@ AliHLTGlobalTrackMatcherComponent::AliHLTGlobalTrackMatcherComponent() :
   fTrackMatcher(NULL),
   fNEvents(0),
   fBz(-9999999),
-  fClusterReader(NULL)
+  fClusterReader(NULL),
+  fTrackArray(NULL)
 {
   // see header file for class documentation
   // or
@@ -94,7 +95,7 @@ void AliHLTGlobalTrackMatcherComponent::GetOutputDataSize( unsigned long& constB
   // see header file for class documentation
   // XXX TODO: Find more realistic values.
   constBase = 80000;
-  inputMultiplier = 0;
+  inputMultiplier = 1;
 }
 
 AliHLTComponent* AliHLTGlobalTrackMatcherComponent::Spawn()
@@ -114,17 +115,24 @@ int AliHLTGlobalTrackMatcherComponent::DoInit( int argc, const char** argv )
     HLTWarning("Ignoring all configuration args, starting with: argv %s", argv[0]);
   }
 
-  fClusterReader = new AliHLTCaloClusterReader();
+  if(!fClusterReader)
+    fClusterReader = new AliHLTCaloClusterReader();
 
   fBz = GetBz();
   if(fBz == -999999) {
     HLTError("Magnetic field not properly set, current value: %d", fBz);
   }
 
-  fTrackMatcher = new AliHLTGlobalTrackMatcher();
+  if(!fTrackMatcher)
+    fTrackMatcher = new AliHLTGlobalTrackMatcher();
 
   fNEvents = 0;
 
+  if(!fTrackArray){
+    fTrackArray = new TObjArray();
+    fTrackArray->SetOwner(kFALSE);
+  }
+
   return iResult; 
 }
   
@@ -158,60 +166,74 @@ int AliHLTGlobalTrackMatcherComponent::DoEvent(const AliHLTComponentEventData& /
 
 
   fNEvents++;
-  
-  //BALLE TODO check that the tracks in the TObjArray are fine over several blocks
-  TObjArray * trackArray = new TObjArray();
-  trackArray->SetOwner(kFALSE);
-  vector<AliHLTGlobalBarrelTrack> tracks;
-  
-  for (const AliHLTComponentBlockData* pBlock = GetFirstInputBlock(kAliHLTDataTypeTrack|kAliHLTDataOriginTPC); pBlock!=NULL; pBlock=GetNextInputBlock()) {
-    
-    if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>=0) {
-
-      for(UInt_t it = 0; it < tracks.size(); it++) {
-       AliHLTGlobalBarrelTrack track = tracks.at(it);
-       HLTInfo("track ID %d", track.TrackID());
-       trackArray->AddLast(dynamic_cast<TObject*>(&(tracks.at(it))));
-      }
-    } else {
-      HLTWarning("Converting tracks to vector failed");
-    }
-    
-    //Push the TPC block on, without any changes
-    PushBack(pBlock->fPtr, pBlock->fSize, pBlock->fDataType, pBlock->fSpecification);
 
-  }
+  //Loop over TPC blocks
+  //BALLE TODO check that the tracks in the TObjArray are fine over several blocks
 
-  //Vector to contain the phos clusters
-  vector<AliHLTCaloClusterDataStruct *> phosClustersVector;
-  AliHLTCaloClusterDataStruct * caloClusterStructPtr;
-  
-  for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeCaloCluster | kAliHLTDataOriginAny); pBlock!=NULL; pBlock=GetNextInputBlock()) {
-    AliHLTCaloClusterHeaderStruct *caloClusterHeaderPtr = reinterpret_cast<AliHLTCaloClusterHeaderStruct*>(pBlock->fPtr);
-    fClusterReader->SetMemory(caloClusterHeaderPtr);
-    
-    //BALLE, TODO, make it able to do EMCAL as well!!!
-    phosClustersVector.resize((Int_t) (caloClusterHeaderPtr->fNClusters));
-    
-    Int_t nClusters = 0;
-    while((caloClusterStructPtr = fClusterReader->NextCluster()) != 0) {
-      //BALLE stil just phos
-      phosClustersVector[nClusters++] = caloClusterStructPtr;  
-    }
+   fTrackArray->Clear();
+   vector<AliHLTGlobalBarrelTrack> tracks;
+   
+   for (const AliHLTComponentBlockData* pBlock = GetFirstInputBlock(kAliHLTDataTypeTrack|kAliHLTDataOriginTPC); pBlock!=NULL; pBlock=GetNextInputBlock()) {
+
+     if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>=0) {
+       for(UInt_t it = 0; it < tracks.size(); it++) {
+        AliHLTGlobalBarrelTrack track = tracks.at(it);
+       fTrackArray->AddLast(dynamic_cast<TObject*>(&(tracks.at(it))));
+              }
+     } else {
+             HLTWarning("Converting tracks to vector failed");
+     }
     
-    iResult = fTrackMatcher->Match(trackArray, phosClustersVector, fBz);
-    if(iResult <0) {
-      HLTWarning("Error in track matcher");
-    }
-
-    PushBack(pBlock->fPtr, pBlock->fSize, pBlock->fDataType, pBlock->fSpecification);
-  }
-
-  //BALLE TODO phos only !!!!
-  
-  delete trackArray;
-  return iResult;
-
+     //     //Push the TPC block on, without any changes
+     //PushBack(pBlock->fPtr, pBlock->fSize, pBlock->fDataType, pBlock->fSpecification);
+
+   }
+
+   AliHLTCaloClusterDataStruct * caloClusterStruct;
+   //Get the PHOS Clusters
+   vector<AliHLTCaloClusterDataStruct*> phosClustersVector;
+   
+   for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeCaloCluster | kAliHLTDataOriginPHOS); pBlock!=NULL; pBlock=GetNextInputBlock()) {
+     AliHLTCaloClusterHeaderStruct *caloClusterHeader = reinterpret_cast<AliHLTCaloClusterHeaderStruct*>(pBlock->fPtr);
+     fClusterReader->SetMemory(caloClusterHeader);
+     if ( (caloClusterHeader->fNClusters) < 0) {
+       HLTWarning("Event has negative number of clusters: %d! Very bad for vector resizing", (Int_t) (caloClusterHeader->fNClusters));
+       continue;
+     } else {    
+       phosClustersVector.reserve( (int) (caloClusterHeader->fNClusters) + phosClustersVector.size() ); 
+       while( (caloClusterStruct = fClusterReader->NextCluster()) != 0) {
+        phosClustersVector.push_back(caloClusterStruct);  
+       }
+     }
+   }
+   
+
+   //Get the EMCAL Clusters
+   vector<AliHLTCaloClusterDataStruct*> emcalClustersVector;
+   for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeCaloCluster | kAliHLTDataOriginEMCAL); pBlock!=NULL; pBlock=GetNextInputBlock()) {
+     AliHLTCaloClusterHeaderStruct *caloClusterHeader = reinterpret_cast<AliHLTCaloClusterHeaderStruct*>(pBlock->fPtr);
+     fClusterReader->SetMemory(caloClusterHeader);
+     if ( (caloClusterHeader->fNClusters) < 0) {
+       HLTWarning("Event has negative number of clusters: %d! Very bad for vector resizing", (Int_t) (caloClusterHeader->fNClusters));
+       continue;
+     } else {    
+       emcalClustersVector.reserve( (int) (caloClusterHeader->fNClusters) + emcalClustersVector.size() ); 
+       while( (caloClusterStruct = fClusterReader->NextCluster()) != 0) {
+        emcalClustersVector.push_back(caloClusterStruct);  
+       }
+     }
+   }
+   
+   iResult = fTrackMatcher->Match(fTrackArray, phosClustersVector, emcalClustersVector, fBz);
+
+   //Push the blocks on
+   for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeCaloCluster | kAliHLTDataOriginAny); pBlock!=NULL; pBlock=GetNextInputBlock()) {
+     PushBack(pBlock->fPtr, pBlock->fSize, pBlock->fDataType, pBlock->fSpecification);
+   }
+
+   fTrackArray->Clear();
+   
+   return iResult;
 }
 
 // int AliHLTGlobalTrackMatcherComponent::Configure(const char* arguments)