]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - HLT/PHOS/AliHLTPHOSClusterizerComponent.cxx
- added argument to select the task location
[u/mrichter/AliRoot.git] / HLT / PHOS / AliHLTPHOSClusterizerComponent.cxx
index 3d5c55c962e1205756f239f323c9589d22fd9e1e..d4254652f837ba588e29027646369cc79a8cd4eb 100644 (file)
@@ -1,7 +1,9 @@
+// $Id$
+
 /**************************************************************************
  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  *                                                                        *
- * Authors: Ã\98ystein Djuvsland <oysteind@ift.uib.no>                       *
+ * Authors: Oystein Djuvsland <oysteind@ift.uib.no>                       *
  *                                                                        *
  * Permission to use, copy, modify and distribute this software and its   *
  * documentation strictly for non-commercial purposes is hereby granted   *
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
+#include <iostream>
 
+#include "AliHLTPHOSClusterizerComponent.h"
+#include "AliHLTCaloRecPointDataStruct.h"
+#include "AliHLTCaloRecPointHeaderStruct.h"
+#include "AliHLTPHOSGeometry.h"
+#include "AliHLTCaloClusterAnalyser.h"
 
 
-#include "AliHLTPHOSClusterizerComponent.h"
-#include "AliHLTPHOSClusterizer.h"
-//#include "AliHLTPHOSPhysicsDefinitions.h"
-//#include "AliHLTPHOSDefinitions.h"
-#include "AliHLTPHOSRecPointDataStruct.h"
-#include "AliHLTPHOSClusterDataStruct.h"
-#include "AliHLTPHOSRecPointListDataStruct.h"
 
-using namespace std;
 
+/** @file   AliHLTPHOSClusterizerComponent.cxx
+    @author Oystein Djuvsland
+    @date   
+    @brief  A clusterizer component for PHOS HLT
+*/
 
-const AliHLTComponentDataType AliHLTPHOSClusterizerComponent::fgkInputDataTypes[]={kAliHLTVoidDataType,{0,"",""}};
+// see header file for class documentation
+// or
+// refer to README to build package
+// or
+// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
+#include "AliHLTCaloDefinitions.h"
+#include "AliHLTPHOSGeometry.h"
+#include "AliHLTPHOSRecoParamHandler.h"
+#include "AliHLTCaloClusterizer.h"
 
 AliHLTPHOSClusterizerComponent gAliHLTPHOSClusterizerComponent;
 
-//AliHLTPHOSClusterizerComponent::AliHLTPHOSClusterizerComponent(): AliHLTPHOSBase(), AliHLTProcessor(), fClusterizerPtr(0), fOutPtr(0), 
-//                                                              fRecPointStructArrayPtr(0), fRecPointListPtr(0)
-AliHLTPHOSClusterizerComponent::AliHLTPHOSClusterizerComponent(): AliHLTPHOSProcessor(), fClusterizerPtr(0), fOutPtr(0), 
-                                                                 fRecPointStructArrayPtr(0), fRecPointListPtr(0)
+AliHLTPHOSClusterizerComponent::AliHLTPHOSClusterizerComponent(): 
+  AliHLTCaloClusterizerComponent("PHOS")
 {
-  //Constructor
-}
+  //See headerfile for documentation
 
-AliHLTPHOSClusterizerComponent::~AliHLTPHOSClusterizerComponent()
-{
-  //Destructor
+  fDataOrigin = const_cast<char*>(kAliHLTDataOriginPHOS);
 
-  if(fClusterizerPtr)
-    {
-      delete fClusterizerPtr;
-      fClusterizerPtr = 0;
-    }
+  //AliHLTPHOSGeometry *geom = new AliHLTPHOSGeometry;
   
-  if(fRecPointListPtr)
-    {
-      delete fRecPointListPtr;
-      fRecPointListPtr = 0;
-    }
-
-  if(fRecPointStructArrayPtr)
-    {
-      for(int i = 0; i < 1000; i++) 
-       {
-         fRecPointStructArrayPtr[i].Del();
-       }
-      delete fRecPointStructArrayPtr;
-      fRecPointStructArrayPtr = 0;
-    }
-  
-}
-
-/*
-int 
-AliHLTPHOSClusterizerComponent::AliHLTPHOSClusterizerComponent::Deinit()
-{
-  ////////// PTH WARNING you should Define a class AliHLTPHOSModuleProcessor
-}
-*/
-
-// PTH AliHLTPHOSClusterizerComponent::AliHLTPHOSClusterizerComponent(const AliHLTPHOSClusterizerComponent &):AliHLTProcessor(), 
-//                                                                                                    fClusterizerPtr(0), 
-//                                                                                                    fOutPtr(0),
-//                                                                                                    fRecPointStructArrayPtr(0),
-//                                                                                                    fRecPointListPtr(0)
-//{
-  //Copy constructor, not implemented
-//}
-
-int
-AliHLTPHOSClusterizerComponent::Deinit()
-{
-  //Deinitialization
-
-  if(fClusterizerPtr)
-    {
-      delete fClusterizerPtr;
-      fClusterizerPtr = 0;
-    }
-  
-  if(fRecPointListPtr)
-    {
-      delete fRecPointListPtr;
-      fRecPointListPtr = 0;
-    }
-  
-  for(int i = 0; i < 1000; i++) 
-    {
-      fRecPointStructArrayPtr[i].Del();
-    }
-  
-  if(fRecPointStructArrayPtr)
-    {
-      for(int i = 0; i < 1000; i++) 
-       {
-         fRecPointStructArrayPtr[i].Del();
-       }
-      delete fRecPointStructArrayPtr;
-      fRecPointStructArrayPtr = 0;
-    }
-
-
-
-
-
-  return 0;
 }
 
-int
-AliHLTPHOSClusterizerComponent::DoDeinit()
-{
-  //Do deinitialization
-  Logging(kHLTLogInfo, "HLT", "PHOS", ",AliHLTPHOSClusterizerComponent DoDeinit");
-
-  return 0;
-}
-
-
-const Char_t* 
-AliHLTPHOSClusterizerComponent::GetComponentID()
+AliHLTPHOSClusterizerComponent::~AliHLTPHOSClusterizerComponent()
 {
-  return "AliHltPhosClusterizer";
+  //See headerfile for documentation
 }
 
 void
 AliHLTPHOSClusterizerComponent::GetInputDataTypes( vector<AliHLTComponentDataType>& list)
 {
-  //Get datatypes for input
-  const AliHLTComponentDataType* pType=fgkInputDataTypes;
-  while (pType->fID!=0) {
-    list.push_back(*pType);
-    pType++;
-  }
+  //See headerfile for documentation
+  list.clear();
+  list.push_back(AliHLTCaloDefinitions::fgkDigitDataType|kAliHLTDataOriginPHOS);
 }
 
-AliHLTComponentDataType 
+AliHLTComponentDataType
 AliHLTPHOSClusterizerComponent::GetOutputDataType()
 {
-  //  return AliHLTPHOSPhysicsDefinitions::fgkAliHLTClusterDataType;
-  return AliHLTPHOSDefinitions::fgkAliHLTClusterDataType;
+  //See headerfile for documentation
+  return kAliHLTDataTypeCaloCluster|kAliHLTDataOriginPHOS;
 }
 
 void
 AliHLTPHOSClusterizerComponent::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier )
 
 {
-  constBase = 30;
-  inputMultiplier = 0.2;
+  //See headerfile for documentation
+  constBase = sizeof(AliHLTCaloRecPointHeaderStruct) + sizeof(AliHLTCaloRecPointDataStruct) + (sizeof(AliHLTCaloDigitDataStruct) << 7); //Reasonable estimate... ;
+  inputMultiplier = 2.0;
 }
 
-int 
-AliHLTPHOSClusterizerComponent::DoEvent(const AliHLTComponentEventData& evtData, const AliHLTComponentBlockData* blocks,
-                                       AliHLTComponentTriggerData& /*trigData*/, AliHLTUInt8_t* outputPtr, AliHLTUInt32_t& size,
-                                       std::vector<AliHLTComponentBlockData>& outputBlocks)
+const Char_t*
+AliHLTPHOSClusterizerComponent::GetComponentID()
 {
-  //Do event
-   
-  UInt_t tSize            = 0;
-  UInt_t offset           = 0; 
-  UInt_t mysize           = 0;
-  Int_t nRecPoints        = 0;
-  Int_t index             = 0;
-
-  AliHLTUInt8_t* outBPtr;
-  outBPtr = outputPtr;
-  const AliHLTComponentBlockData* iter = 0; 
-  unsigned long ndx; 
-
-  for( ndx = 0; ndx < evtData.fBlockCnt; ndx++ )
-    {
-      iter = blocks+ndx;
-      
-      if(iter->fDataType != AliHLTPHOSDefinitions::fgkCellEnergyDataType)
-       {
-         cout << "Warning: data type is not fgkCellEnergyDataType " << endl;
-         continue;
-       }
-      index = fClusterizerPtr->BuildCellEnergyArray( reinterpret_cast<AliHLTPHOSRcuCellEnergyDataStruct*>(iter->fPtr),
-                                                    fRecPointListPtr);
-      
-    }
-  nRecPoints = fClusterizerPtr->CreateRecPointStructArray(fRecPointStructArrayPtr, fRecPointListPtr, index);
-  
-  cout << "Number of clusters found: " << nRecPoints << endl;
-  
-  for(Int_t i = 0; i < nRecPoints; i++)
-    {
-      mysize = 0;
-      offset = tSize;
-      
-      fOutPtr =  (AliHLTPHOSClusterDataStruct*)outBPtr;
-      fClusterizerPtr->CalculateCenterOfGravity(&fRecPointStructArrayPtr[i]);
-      //      fClusterizerPtr->CalculateMoments(&fRecPointStructArrayPtr[i], 0);
-      //     fClusterizerPtr->ClusterizeStruct(&fRecPointStructArrayPtr[i], fOutPtr);
-
-      mysize += sizeof(AliHLTPHOSClusterDataStruct);
-      
-      AliHLTComponentBlockData bd;
-      FillBlockData( bd );
-      bd.fOffset = offset;
-      bd.fSize = mysize;
-      // PTH      bd.fDataType = AliHLTPHOSPhysicsDefinitions::fgkAliHLTClusterDataType;
-      bd.fDataType = AliHLTPHOSDefinitions::fgkAliHLTClusterDataType;
-      bd.fSpecification = 0xFFFFFFFF;
-      outputBlocks.push_back( bd );
-       
-      tSize += mysize;
-      outBPtr += mysize;
-      
-      if( tSize > size )
-       {
-         Logging( kHLTLogFatal, "HLT::AliHLTPHOSClusterizerComponent::DoEvent", "Too much data",
-                  "Data written over allowed buffer. Amount written: %lu, allowed amount: %lu."
-                  , tSize, size );
-         return EMSGSIZE;
-       }
-    }
+  //See headerfile for documentation
+  return "PhosClusterizer";
+}
 
-  size = tSize;
-  fClusterizerPtr->ResetCellEnergyArray();
+AliHLTComponent*
+AliHLTPHOSClusterizerComponent::Spawn()
+{
+  //See headerfile for documentation
 
-  return 0;
+  return new AliHLTPHOSClusterizerComponent();
+}
 
+int AliHLTPHOSClusterizerComponent::DoInit(int argc, const char** argv)
+{
+   
+   fClusterizerPtr = new AliHLTCaloClusterizer("PHOS");
+   
+   fRecoParamsPtr = new AliHLTPHOSRecoParamHandler(); 
+    
+    return AliHLTCaloClusterizerComponent::DoInit(argc, argv);
 }
 
-int
-AliHLTPHOSClusterizerComponent::DoInit(int argc, const char** argv )
+int AliHLTPHOSClusterizerComponent::DoDeinit()
 {
-  //Do initialization
-  fClusterizerPtr = new AliHLTPHOSClusterizer();
-  for(int i = 0; i < argc; i++)
-    {
-      if(!strcmp("-threshold", argv[i]))
-       fClusterizerPtr->SetThreshold(atof(argv[i+1]));
-      if(!strcmp("-clusterthreshold", argv[i]))
-       fClusterizerPtr->SetClusterThreshold(atof(argv[i+1]));
-      if(!strcmp("-highgain", argv[i]))
-       fClusterizerPtr->SetHighGainFactor(atof(argv[i+1]));
-      if(!strcmp("-lowgain", argv[i]))
-       fClusterizerPtr->SetLowGainFactor(atof(argv[i+1]));
-      if(!strcmp("-arraysize", argv[i]))
-       fClusterizerPtr->SetArraySize(atoi(argv[i+1]));
-    }
-  fClusterizerPtr->ResetCellEnergyArray();
-  fRecPointListPtr = new AliHLTPHOSRecPointListDataStruct[N_ZROWS_MOD*N_XCOLUMNS_MOD];
-  fRecPointStructArrayPtr = new AliHLTPHOSRecPointDataStruct[1000];
-  for(int i = 0; i < 1000; i++) 
+    if(fRecoParamsPtr) 
     {
-      fRecPointStructArrayPtr[i].fMultiplicity = atoi(argv[4])* atoi(argv[4]);
-      fRecPointStructArrayPtr[i].New();
+       delete fRecoParamsPtr;
+       fRecoParamsPtr = 0;
     }
-  printf("Clusterizer component started with:\n");
-  printf(" Cell threshold:     %f\n", fClusterizerPtr->GetThreshold());
-  printf(" Cluster threshold:  %f\n", fClusterizerPtr->GetClusterThreshold());
-  printf(" High gain factor:   %f\n", fClusterizerPtr->GetHighGainFactor());
-  printf(" Low gain factor:    %f\n", fClusterizerPtr->GetLowGainFactor());
-  printf(" Cluster array size: %d\n\n", fClusterizerPtr->GetArraySize());
-
-  return 0;
+    return AliHLTCaloClusterizerComponent::DoDeinit();
 }
 
-AliHLTComponent*
-AliHLTPHOSClusterizerComponent::Spawn()
+
+Int_t AliHLTPHOSClusterizerComponent::InitialiseGeometry()
 {
-  //Spawn a new AliHLTPHOSClusterizerComponent, for HLT framework
-  return new AliHLTPHOSClusterizerComponent();
+  fAnalyserPtr->SetGeometry(new AliHLTPHOSGeometry);
+  return 0;
 }