[u/mrichter/AliRoot.git] / HLT / TRD / AliHLTTRDClusterizerComponent.cxx

// $Id$

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Authors: Matthias Richter <Matthias.Richter@ift.uib.no>                *
 *          Timm Steinbeck <timm@kip.uni-heidelberg.de>                   *
 *          for The ALICE Off-line Project.                               *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/** @file   AliHLTTRDClusterizerComponent.cxx
    @author Timm Steinbeck, Matthias Richter
    @date   
    @brief  A TRDClusterizer processing component for the HLT. */

// see header file for class documentation                                   //
// or                                                                        //
// refer to README to build package                                          //
// or                                                                        //
// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt                          //

#if __GNUC__ >= 3
using namespace std;
#endif

#include "TTree.h"
#include "TFile.h"
#include "TBranch.h"

#include "AliHLTTRDClusterizerComponent.h"
#include "AliHLTTRDDefinitions.h"

#include "AliGeomManager.h"
#include "AliTRDReconstructor.h"
#include "AliCDBManager.h"
#include "AliTRDclusterizerHLT.h"
#include "AliTRDrecoParam.h"
#include "AliTRDrawStreamBase.h"

#include "AliRawReaderMemory.h"

#include <cstdlib>
#include <cerrno>
#include <string>

// this is a global object used for automatic component registration, do not use this
AliHLTTRDClusterizerComponent gAliHLTTRDClusterizerComponent;

ClassImp(AliHLTTRDClusterizerComponent);
   
AliHLTTRDClusterizerComponent::AliHLTTRDClusterizerComponent()
  : AliHLTProcessor()
  , fOutputPercentage(100) // By default we copy to the output exactly what we got as input  
  , fStrorageDBpath("local://$ALICE_ROOT")
  , fClusterizer(NULL)
  , fRecoParam(NULL)
  , fCDB(NULL)
  , fMemReader(NULL)
  , fGeometryFileName("")
  , fGeometryFile(NULL)
  , fReconstructor(NULL)
{
  // Default constructor

  fGeometryFileName = getenv("ALICE_ROOT");
  fGeometryFileName += "/HLT/TRD/geometry.root";
}

AliHLTTRDClusterizerComponent::~AliHLTTRDClusterizerComponent()
{
  // Destructor
  // Work is Done in DoDeInit()
}
const char* AliHLTTRDClusterizerComponent::GetComponentID()
{
  // Return the component ID const char *
  return "TRDClusterizer"; // The ID of this component
}

void AliHLTTRDClusterizerComponent::GetInputDataTypes( vector<AliHLTComponent_DataType>& list)
{
  // Get the list of input data
  list.clear(); // We do not have any requirements for our input data type(s).
  list.push_back( AliHLTTRDDefinitions::fgkDDLRawDataType );
}

AliHLTComponent_DataType AliHLTTRDClusterizerComponent::GetOutputDataType()
{
  // Get the output data type
  return AliHLTTRDDefinitions::fgkClusterDataType;
}

void AliHLTTRDClusterizerComponent::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier )
{
  // Get the output data size
  constBase = 0;
  inputMultiplier = ((double)fOutputPercentage)/100.0;
}

AliHLTComponent* AliHLTTRDClusterizerComponent::Spawn()
{
  // Spawn function, return new instance of this class
  return new AliHLTTRDClusterizerComponent;
};

int AliHLTTRDClusterizerComponent::DoInit( int argc, const char** argv )
{
  // perform initialization. We check whether our relative output size is specified in the arguments.
  fOutputPercentage = 100;
  Int_t iRawDataVersion = 2;
  int i = 0;
  char* cpErr;
  Bool_t bWriteClusters = kTRUE;
  

  Int_t iRecoParamType = -1; // default will be the low flux

  // the data type will become obsolete as soon as the formats are established
  Int_t iRecoDataType = -1; // default will be simulation
  
  while ( i < argc )
    {
      HLTDebug("argv[%d] == %s", i, argv[i] );
      if ( !strcmp( argv[i], "output_percentage" ) )
	{
	  if ( i+1>=argc )
	    {
	      HLTError("Missing output_percentage parameter");
	      return ENOTSUP;
	    }
	  HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
	  fOutputPercentage = strtoul( argv[i+1], &cpErr, 0 );
	  if ( *cpErr )
	    {
	      HLTError("Cannot convert output_percentage parameter '%s'", argv[i+1] );
	      return EINVAL;
	    }
	  HLTInfo("Output percentage set to %lu %%", fOutputPercentage );
	  i += 2;
	}
      else if ( strcmp( argv[i], "-cdb" ) == 0)
	{
	  if ( i+1 >= argc )
	    {
	      HLTError("Missing -cdb argument");
	      return ENOTSUP;	      
	    }
	  fStrorageDBpath = argv[i+1];
	  HLTInfo("DB storage is %s", fStrorageDBpath.c_str() );	  
	  i += 2;
	  
	}      

      else if ( strcmp( argv[i], "-lowflux" ) == 0)
	{
	  iRecoParamType = 0;	  
	  HLTDebug("Low flux reco selected.");
	  i++;
	  
	}      

      else if ( strcmp( argv[i], "-highflux" ) == 0)
	{
	  iRecoParamType = 1;	  
	  HLTDebug("High flux reco selected.");
	  i++;
	  
	}      

      else if ( strcmp( argv[i], "-cosmics" ) == 0)
	{
	  iRecoParamType = 2;	  
	  HLTDebug("Cosmic test reco selected.");
	  i++;
	  
	}      

      // raw data type - sim or experiment
      else if ( strcmp( argv[i], "-simulation" ) == 0)
	{
	  iRecoDataType = 0;
	  i++;
	  
	}

      else if ( strcmp( argv[i], "-experiment" ) == 0)
	{
	  iRecoDataType = 1;
	  i++;
	  
	}

      else if ( strcmp( argv[i], "-rawver" ) == 0)
	{
	  if ( i+1 >= argc )
	    {
	      HLTError("Missing -rawver argument");
	      return ENOTSUP;	      
	    }
	  iRawDataVersion = atoi( argv[i+1] );
	  HLTInfo("Raw data version is %d", iRawDataVersion );	  
	  i += 2;
	  
	}      

      else if ( strcmp( argv[i], "-geometry" ) == 0)
	{
	  if ( i+1 >= argc )
	    {
	      HLTError("Missing -geometry argument");
	      return ENOTSUP;	      
	    }
	  fGeometryFileName = argv[i+1];
	  HLTInfo("GeomFile storage is %s", fGeometryFileName.c_str() );	  
	  i += 2;
	  
	}      
      else{
	HLTError("Unknown option '%s'", argv[i] );
	return EINVAL;
      }
      
    }

  // THE "REAL" INIT COMES HERE

  if (iRecoParamType < 0 || iRecoParamType > 2)
    {
      HLTWarning("No reco param selected. Use -lowflux or -highflux flag. Defaulting to low flux.");
      iRecoParamType = 0;
    }

  if (iRecoParamType == 0)
    {
      fRecoParam = AliTRDrecoParam::GetLowFluxParam();
      HLTDebug("Low flux params init.");
    }

  if (iRecoParamType == 1)
    {
      fRecoParam = AliTRDrecoParam::GetHighFluxParam();
      HLTDebug("High flux params init.");
    }

  if (iRecoParamType == 2)
    {
      fRecoParam = AliTRDrecoParam::GetCosmicTestParam();
      HLTDebug("Cosmic Test params init.");
    }

  if (fRecoParam == 0)
    {
      HLTError("No reco params initialized. Sniffing big trouble!");
      return -1;
    }

  fReconstructor = new AliTRDReconstructor();
  fReconstructor->SetRecoParam(fRecoParam);
  //fReconstructor->SetStreamLevel(0, AliTRDReconstructor::kClusterizer); // default value
  fReconstructor->SetOption("!cw,sl_cf_0");

  // init the raw data type to be used...
  // the switch here will become obsolete as soon as the data structures is fixed 
  // both: in sim and reality
  if (iRecoDataType < 0 || iRecoDataType > 1)
    {
      HLTWarning("No data type selected. Use -simulation or -experiment flag. Defaulting to simulation.");
      iRecoDataType = 0;
    }

  if (iRecoDataType == 0)
    {
      AliTRDrawStreamBase::SetRawStreamVersion(AliTRDrawStreamBase::kTRDsimStream);
      HLTDebug("Data type expected is SIMULATION!");
    }

  if (iRecoDataType == 1)
    {
      AliTRDrawStreamBase::SetRawStreamVersion(AliTRDrawStreamBase::kTRDrealStream);
      HLTDebug("Data type expected is EXPERIMENT!");
    }

  // the DATA BASE STUFF
  fCDB = AliCDBManager::Instance();
  if (!fCDB)
    {
      HLTError("Could not get CDB instance", "fCDB 0x%x", fCDB);
    }
  else
    {
      fCDB->SetRun(0); // THIS HAS TO BE RETRIEVED !!!
      fCDB->SetDefaultStorage(fStrorageDBpath.c_str());
      HLTDebug("CDB instance; fCDB 0x%x", fCDB);
    }

  fGeometryFile = TFile::Open(fGeometryFileName.c_str());
  if (fGeometryFile)
    {
      AliGeomManager::LoadGeometry(fGeometryFileName.c_str());
    }
  else
    {
      HLTError("Unable to open file. FATAL!");
      return -1;
    }

  fMemReader = new AliRawReaderMemory;

  fClusterizer = new AliTRDclusterizerHLT("TRDCclusterizer", "TRDCclusterizer");
  fClusterizer->SetReconstructor(fReconstructor);

  fClusterizer->SetRawVersion(iRawDataVersion);
  fClusterizer->InitClusterTree();
  return 0;
}

int AliHLTTRDClusterizerComponent::DoDeinit()
{
  // Deinitialization of the component
  delete fMemReader;
  fMemReader = 0;
  delete fClusterizer;
  fClusterizer = 0;
  
  fReconstructor->SetClusters(0x0);
  delete fReconstructor;
  fReconstructor = 0x0;
  return 0;

  if (fGeometryFile)
    {
      fGeometryFile->Close();
      delete fGeometryFile;
      fGeometryFile = 0;
    }

  if (fCDB)
    {
      HLTDebug("destroy fCDB");
      fCDB->Destroy();
      fCDB = 0;
    }

  if (fRecoParam)
    {
      HLTDebug("Deleting fRecoParam");
      delete fRecoParam;
      fRecoParam = 0;
    }
}

int AliHLTTRDClusterizerComponent::DoEvent( const AliHLTComponentEventData& evtData, 
					    const AliHLTComponentBlockData* blocks, 
					    AliHLTComponent_TriggerData& /*trigData*/, 
					    AliHLTUInt8_t* /*outputPtr*/, 
					    AliHLTUInt32_t& size, 
					    vector<AliHLTComponent_BlockData>& /*outputBlocks*/ )
{
  // Process an event
  HLTDebug( "NofBlocks %lu", evtData.fBlockCnt );
  // Process an event

  //implement a usage of the following
  //   AliHLTUInt32_t triggerDataStructSize = trigData.fStructSize;
  //   AliHLTUInt32_t triggerDataSize = trigData.fDataSize;
  //   void *triggerData = trigData.fData;
  //HLTDebug( "Trigger data received. Struct size %d Data size %d Data location 0x%x", trigData.fStructSize, trigData.fDataSize, (UInt_t*)trigData.fData);

  // Loop over all input blocks in the event
  AliHLTComponentDataType expectedDataType = (kAliHLTDataTypeDDLRaw | kAliHLTDataOriginTRD);
  for ( unsigned long i = 0; i < evtData.fBlockCnt; i++ )
    {
      // lets not use the internal TRD data types here : AliHLTTRDDefinitions::fgkDDLRawDataType
      // which is depreciated - we use HLT global defs instead
      //      if ( blocks[i].fDataType != (kAliHLTDataTypeDDLRaw | kAliHLTDataOriginTRD) )
      AliHLTComponentDataType inputDataType = blocks[i].fDataType;
      if ( inputDataType != expectedDataType)
	{
	  HLTDebug( "Block # %i/%i; Event 0x%08LX (%Lu) received datatype: %s - required datatype: %s; Skipping",
		    i, evtData.fBlockCnt,
		    evtData.fEventID, evtData.fEventID, 
		    DataType2Text(inputDataType).c_str(), 
		    DataType2Text(expectedDataType).c_str());
	  continue;
	}
      else 
	{
	  HLTDebug("We get the right data type!");
	}
      
      //      fMemReader->Reset();
      fMemReader->SetMemory((UChar_t*) blocks[i].fPtr, blocks[i].fSize);

      AliHLTUInt32_t spec = blocks[i].fSpecification;
      
      Int_t id = 1024;
      
      for ( Int_t ii = 0; ii < 18 ; ii++ ) {
	if ( spec & 0x00000001 ) {
	  id += ii;
	  break;
	}
	spec = spec >> 1 ;
      }
      
      fMemReader->SetEquipmentID( id ); 

      fClusterizer->ResetTree();  
      Bool_t iclustered = fClusterizer->Raw2ClustersChamber(fMemReader);
      if (iclustered == kTRUE)
	{
	  HLTDebug( "Clustered successfully");
	}
      else
	{
	  HLTError("Clustering ERROR");
	  return -1;
	}

      // put the tree into output blocks
      //fcTree->Print();
      TClonesArray *clusterArray = fClusterizer->RecPoints();
      fClusterizer->SetClustersOwner(kFALSE);
	
      if (clusterArray){
	HLTDebug("clusterArray: Entries - %i; Size - %i",clusterArray->GetEntriesFast(),sizeof(clusterArray));
	PushBack(clusterArray, AliHLTTRDDefinitions::fgkClusterDataType, blocks[i].fSpecification);
      }
      else 
	HLTWarning("Array of clusters is empty!");
	
      if (clusterArray){
	clusterArray->Delete();
	delete clusterArray;
      }
      

    }//  for ( unsigned long i = 0; i < evtData.fBlockCnt; i++ )

  
  
  
  size=0; // this function did not write data to the buffer directly
  return 0;
}
Commit	Line	Data
0af7cb2e	1	// $Id$
	2
	3	/**************************************************************************
	4	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* *
	6	* Authors: Matthias Richter <Matthias.Richter@ift.uib.no> *
	7	* Timm Steinbeck <timm@kip.uni-heidelberg.de> *
	8	* for The ALICE Off-line 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 AliHLTTRDClusterizerComponent.cxx
	20	@author Timm Steinbeck, Matthias Richter
	21	@date
	22	@brief A TRDClusterizer processing component for the HLT. */
	23
c7500dae	24	// see header file for class documentation //
	25	// or //
	26	// refer to README to build package //
	27	// or //
	28	// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt //
	29
0af7cb2e	30	#if __GNUC__ >= 3
	31	using namespace std;
	32	#endif
	33
051a0e2d	34	#include "TTree.h"
	35	#include "TFile.h"
	36	#include "TBranch.h"
	37
0af7cb2e	38	#include "AliHLTTRDClusterizerComponent.h"
	39	#include "AliHLTTRDDefinitions.h"
	40
886e8d3d	41	#include "AliGeomManager.h"
886e8d3d	42	#include "AliTRDReconstructor.h"
0af7cb2e	43	#include "AliCDBManager.h"
051a0e2d	44	#include "AliTRDclusterizerHLT.h"
519f385f	45	#include "AliTRDrecoParam.h"
	46	#include "AliTRDrawStreamBase.h"
	47
0af7cb2e	48	#include "AliRawReaderMemory.h"
0af7cb2e	49
0af7cb2e	50	#include <cstdlib>
	51	#include <cerrno>
	52	#include <string>
	53
	54	// this is a global object used for automatic component registration, do not use this
	55	AliHLTTRDClusterizerComponent gAliHLTTRDClusterizerComponent;
	56
3b021c25	57	ClassImp(AliHLTTRDClusterizerComponent);
	58
	59	AliHLTTRDClusterizerComponent::AliHLTTRDClusterizerComponent()
	60	: AliHLTProcessor()
	61	, fOutputPercentage(100) // By default we copy to the output exactly what we got as input
	62	, fStrorageDBpath("local://$ALICE_ROOT")
	63	, fClusterizer(NULL)
519f385f	64	, fRecoParam(NULL)
3b021c25	65	, fCDB(NULL)
3b021c25	66	, fMemReader(NULL)
051a0e2d	67	, fGeometryFileName("")
051a0e2d	68	, fGeometryFile(NULL)
886e8d3d	69	, fReconstructor(NULL)
0af7cb2e	70	{
3b021c25	71	// Default constructor
051a0e2d	72
	73	fGeometryFileName = getenv("ALICE_ROOT");
	74	fGeometryFileName += "/HLT/TRD/geometry.root";
0af7cb2e	75	}
	76
	77	AliHLTTRDClusterizerComponent::~AliHLTTRDClusterizerComponent()
	78	{
3b021c25	79	// Destructor
886e8d3d	80	// Work is Done in DoDeInit()
0af7cb2e	81	}
0af7cb2e	82	const char* AliHLTTRDClusterizerComponent::GetComponentID()
0af7cb2e	83	{
3b021c25	84	// Return the component ID const char *
0af7cb2e	85	return "TRDClusterizer"; // The ID of this component
	86	}
	87
	88	void AliHLTTRDClusterizerComponent::GetInputDataTypes( vector<AliHLTComponent_DataType>& list)
	89	{
3b021c25	90	// Get the list of input data
0af7cb2e	91	list.clear(); // We do not have any requirements for our input data type(s).
efbd3157	92	list.push_back( AliHLTTRDDefinitions::fgkDDLRawDataType );
0af7cb2e	93	}
	94
	95	AliHLTComponent_DataType AliHLTTRDClusterizerComponent::GetOutputDataType()
	96	{
3b021c25	97	// Get the output data type
efbd3157	98	return AliHLTTRDDefinitions::fgkClusterDataType;
0af7cb2e	99	}
	100
	101	void AliHLTTRDClusterizerComponent::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier )
	102	{
3b021c25	103	// Get the output data size
0af7cb2e	104	constBase = 0;
	105	inputMultiplier = ((double)fOutputPercentage)/100.0;
	106	}
	107
0af7cb2e	108	AliHLTComponent* AliHLTTRDClusterizerComponent::Spawn()
0af7cb2e	109	{
3b021c25	110	// Spawn function, return new instance of this class
0af7cb2e	111	return new AliHLTTRDClusterizerComponent;
	112	};
	113
	114	int AliHLTTRDClusterizerComponent::DoInit( int argc, const char** argv )
	115	{
	116	// perform initialization. We check whether our relative output size is specified in the arguments.
	117	fOutputPercentage = 100;
d137c50b	118	Int_t iRawDataVersion = 2;
0af7cb2e	119	int i = 0;
0af7cb2e	120	char* cpErr;
9aea5deb	121	Bool_t bWriteClusters = kTRUE;
9aea5deb	122
519f385f	123
	124	Int_t iRecoParamType = -1; // default will be the low flux
	125
	126	// the data type will become obsolete as soon as the formats are established
	127	Int_t iRecoDataType = -1; // default will be simulation
	128
0af7cb2e	129	while ( i < argc )
0af7cb2e	130	{
886e8d3d	131	HLTDebug("argv[%d] == %s", i, argv[i] );
0af7cb2e	132	if ( !strcmp( argv[i], "output_percentage" ) )
	133	{
	134	if ( i+1>=argc )
	135	{
886e8d3d	136	HLTError("Missing output_percentage parameter");
0af7cb2e	137	return ENOTSUP;
0af7cb2e	138	}
886e8d3d	139	HLTDebug("argv[%d+1] == %s", i, argv[i+1] );
0af7cb2e	140	fOutputPercentage = strtoul( argv[i+1], &cpErr, 0 );
	141	if ( *cpErr )
	142	{
886e8d3d	143	HLTError("Cannot convert output_percentage parameter '%s'", argv[i+1] );
0af7cb2e	144	return EINVAL;
0af7cb2e	145	}
886e8d3d	146	HLTInfo("Output percentage set to %lu %%", fOutputPercentage );
0af7cb2e	147	i += 2;
0af7cb2e	148	}
9aea5deb	149	else if ( strcmp( argv[i], "-cdb" ) == 0)
0af7cb2e	150	{
	151	if ( i+1 >= argc )
	152	{
886e8d3d	153	HLTError("Missing -cdb argument");
0af7cb2e	154	return ENOTSUP;
	155	}
	156	fStrorageDBpath = argv[i+1];
886e8d3d	157	HLTInfo("DB storage is %s", fStrorageDBpath.c_str() );
0af7cb2e	158	i += 2;
9aea5deb	159
0af7cb2e	160	}
0af7cb2e	161
9aea5deb	162	else if ( strcmp( argv[i], "-lowflux" ) == 0)
519f385f	163	{
	164	iRecoParamType = 0;
	165	HLTDebug("Low flux reco selected.");
	166	i++;
9aea5deb	167
519f385f	168	}
519f385f	169
9aea5deb	170	else if ( strcmp( argv[i], "-highflux" ) == 0)
519f385f	171	{
519f385f	172	iRecoParamType = 1;
886e8d3d	173	HLTDebug("High flux reco selected.");
886e8d3d	174	i++;
9aea5deb	175
886e8d3d	176	}
886e8d3d	177
9aea5deb	178	else if ( strcmp( argv[i], "-cosmics" ) == 0)
886e8d3d	179	{
	180	iRecoParamType = 2;
	181	HLTDebug("Cosmic test reco selected.");
519f385f	182	i++;
9aea5deb	183
519f385f	184	}
	185
	186	// raw data type - sim or experiment
9aea5deb	187	else if ( strcmp( argv[i], "-simulation" ) == 0)
519f385f	188	{
	189	iRecoDataType = 0;
	190	i++;
9aea5deb	191
519f385f	192	}
519f385f	193
9aea5deb	194	else if ( strcmp( argv[i], "-experiment" ) == 0)
519f385f	195	{
	196	iRecoDataType = 1;
	197	i++;
9aea5deb	198
519f385f	199	}
519f385f	200
9aea5deb	201	else if ( strcmp( argv[i], "-rawver" ) == 0)
0af7cb2e	202	{
	203	if ( i+1 >= argc )
	204	{
886e8d3d	205	HLTError("Missing -rawver argument");
0af7cb2e	206	return ENOTSUP;
0af7cb2e	207	}
d137c50b	208	iRawDataVersion = atoi( argv[i+1] );
886e8d3d	209	HLTInfo("Raw data version is %d", iRawDataVersion );
0af7cb2e	210	i += 2;
9aea5deb	211
0af7cb2e	212	}
0af7cb2e	213
9aea5deb	214	else if ( strcmp( argv[i], "-geometry" ) == 0)
051a0e2d	215	{
	216	if ( i+1 >= argc )
	217	{
886e8d3d	218	HLTError("Missing -geometry argument");
051a0e2d	219	return ENOTSUP;
	220	}
	221	fGeometryFileName = argv[i+1];
886e8d3d	222	HLTInfo("GeomFile storage is %s", fGeometryFileName.c_str() );
051a0e2d	223	i += 2;
9aea5deb	224
051a0e2d	225	}
9aea5deb	226	else{
	227	HLTError("Unknown option '%s'", argv[i] );
	228	return EINVAL;
	229	}
	230
0af7cb2e	231	}
0af7cb2e	232
519f385f	233	// THE "REAL" INIT COMES HERE
519f385f	234
886e8d3d	235	if (iRecoParamType < 0 \|\| iRecoParamType > 2)
519f385f	236	{
	237	HLTWarning("No reco param selected. Use -lowflux or -highflux flag. Defaulting to low flux.");
	238	iRecoParamType = 0;
	239	}
	240
	241	if (iRecoParamType == 0)
	242	{
	243	fRecoParam = AliTRDrecoParam::GetLowFluxParam();
	244	HLTDebug("Low flux params init.");
	245	}
	246
	247	if (iRecoParamType == 1)
	248	{
	249	fRecoParam = AliTRDrecoParam::GetHighFluxParam();
	250	HLTDebug("High flux params init.");
	251	}
	252
886e8d3d	253	if (iRecoParamType == 2)
	254	{
	255	fRecoParam = AliTRDrecoParam::GetCosmicTestParam();
	256	HLTDebug("Cosmic Test params init.");
	257	}
	258
519f385f	259	if (fRecoParam == 0)
	260	{
	261	HLTError("No reco params initialized. Sniffing big trouble!");
	262	return -1;
	263	}
	264
0d66dbf5	265	fReconstructor = new AliTRDReconstructor();
0d66dbf5	266	fReconstructor->SetRecoParam(fRecoParam);
9aea5deb	267	//fReconstructor->SetStreamLevel(0, AliTRDReconstructor::kClusterizer); // default value
9aea5deb	268	fReconstructor->SetOption("!cw,sl_cf_0");
519f385f	269
	270	// init the raw data type to be used...
	271	// the switch here will become obsolete as soon as the data structures is fixed
	272	// both: in sim and reality
	273	if (iRecoDataType < 0 \|\| iRecoDataType > 1)
	274	{
	275	HLTWarning("No data type selected. Use -simulation or -experiment flag. Defaulting to simulation.");
	276	iRecoDataType = 0;
	277	}
	278
	279	if (iRecoDataType == 0)
	280	{
	281	AliTRDrawStreamBase::SetRawStreamVersion(AliTRDrawStreamBase::kTRDsimStream);
	282	HLTDebug("Data type expected is SIMULATION!");
	283	}
	284
	285	if (iRecoDataType == 1)
	286	{
	287	AliTRDrawStreamBase::SetRawStreamVersion(AliTRDrawStreamBase::kTRDrealStream);
	288	HLTDebug("Data type expected is EXPERIMENT!");
	289	}
	290
	291	// the DATA BASE STUFF
3b021c25	292	fCDB = AliCDBManager::Instance();
3b021c25	293	if (!fCDB)
0af7cb2e	294	{
9aea5deb	295	HLTError("Could not get CDB instance", "fCDB 0x%x", fCDB);
0af7cb2e	296	}
	297	else
	298	{
3b021c25	299	fCDB->SetRun(0); // THIS HAS TO BE RETRIEVED !!!
3b021c25	300	fCDB->SetDefaultStorage(fStrorageDBpath.c_str());
886e8d3d	301	HLTDebug("CDB instance; fCDB 0x%x", fCDB);
0af7cb2e	302	}
0af7cb2e	303
051a0e2d	304	fGeometryFile = TFile::Open(fGeometryFileName.c_str());
	305	if (fGeometryFile)
	306	{
886e8d3d	307	AliGeomManager::LoadGeometry(fGeometryFileName.c_str());
051a0e2d	308	}
	309	else
	310	{
886e8d3d	311	HLTError("Unable to open file. FATAL!");
051a0e2d	312	return -1;
	313	}
	314
3b021c25	315	fMemReader = new AliRawReaderMemory;
051a0e2d	316
9aea5deb	317	fClusterizer = new AliTRDclusterizerHLT("TRDCclusterizer", "TRDCclusterizer");
	318	fClusterizer->SetReconstructor(fReconstructor);
	319
d137c50b	320	fClusterizer->SetRawVersion(iRawDataVersion);
3b021c25	321	fClusterizer->InitClusterTree();
0af7cb2e	322	return 0;
	323	}
	324
	325	int AliHLTTRDClusterizerComponent::DoDeinit()
	326	{
3b021c25	327	// Deinitialization of the component
	328	delete fMemReader;
	329	fMemReader = 0;
	330	delete fClusterizer;
	331	fClusterizer = 0;
9aea5deb	332
9aea5deb	333	fReconstructor->SetClusters(0x0);
0d66dbf5	334	delete fReconstructor;
0d66dbf5	335	fReconstructor = 0x0;
0af7cb2e	336	return 0;
0af7cb2e	337
051a0e2d	338	if (fGeometryFile)
	339	{
	340	fGeometryFile->Close();
	341	delete fGeometryFile;
	342	fGeometryFile = 0;
	343	}
	344
3b021c25	345	if (fCDB)
0af7cb2e	346	{
886e8d3d	347	HLTDebug("destroy fCDB");
3b021c25	348	fCDB->Destroy();
3b021c25	349	fCDB = 0;
0af7cb2e	350	}
519f385f	351
	352	if (fRecoParam)
	353	{
	354	HLTDebug("Deleting fRecoParam");
	355	delete fRecoParam;
	356	fRecoParam = 0;
	357	}
0af7cb2e	358	}
0af7cb2e	359
886e8d3d	360	int AliHLTTRDClusterizerComponent::DoEvent( const AliHLTComponentEventData& evtData,
	361	const AliHLTComponentBlockData* blocks,
	362	AliHLTComponent_TriggerData& /trigData/,
b39f18ce	363	AliHLTUInt8_t* /outputPtr/,
a28bfa0d	364	AliHLTUInt32_t& size,
b39f18ce	365	vector<AliHLTComponent_BlockData>& /outputBlocks/ )
0af7cb2e	366	{
3b021c25	367	// Process an event
886e8d3d	368	HLTDebug( "NofBlocks %lu", evtData.fBlockCnt );
0af7cb2e	369	// Process an event
0af7cb2e	370
051a0e2d	371	//implement a usage of the following
519f385f	372	// AliHLTUInt32_t triggerDataStructSize = trigData.fStructSize;
	373	// AliHLTUInt32_t triggerDataSize = trigData.fDataSize;
	374	// void *triggerData = trigData.fData;
886e8d3d	375	//HLTDebug( "Trigger data received. Struct size %d Data size %d Data location 0x%x", trigData.fStructSize, trigData.fDataSize, (UInt_t*)trigData.fData);
051a0e2d	376
0af7cb2e	377	// Loop over all input blocks in the event
886e8d3d	378	AliHLTComponentDataType expectedDataType = (kAliHLTDataTypeDDLRaw \| kAliHLTDataOriginTRD);
0af7cb2e	379	for ( unsigned long i = 0; i < evtData.fBlockCnt; i++ )
0af7cb2e	380	{
519f385f	381	// lets not use the internal TRD data types here : AliHLTTRDDefinitions::fgkDDLRawDataType
519f385f	382	// which is depreciated - we use HLT global defs instead
886e8d3d	383	// if ( blocks[i].fDataType != (kAliHLTDataTypeDDLRaw \| kAliHLTDataOriginTRD) )
	384	AliHLTComponentDataType inputDataType = blocks[i].fDataType;
	385	if ( inputDataType != expectedDataType)
0af7cb2e	386	{
9aea5deb	387	HLTDebug( "Block # %i/%i; Event 0x%08LX (%Lu) received datatype: %s - required datatype: %s; Skipping",
	388	i, evtData.fBlockCnt,
	389	evtData.fEventID, evtData.fEventID,
	390	DataType2Text(inputDataType).c_str(),
	391	DataType2Text(expectedDataType).c_str());
0af7cb2e	392	continue;
0af7cb2e	393	}
9aea5deb	394	else
	395	{
	396	HLTDebug("We get the right data type!");
	397	}
	398
886e8d3d	399	// fMemReader->Reset();
	400	fMemReader->SetMemory((UChar_t*) blocks[i].fPtr, blocks[i].fSize);
	401
	402	AliHLTUInt32_t spec = blocks[i].fSpecification;
	403
	404	Int_t id = 1024;
	405
	406	for ( Int_t ii = 0; ii < 18 ; ii++ ) {
	407	if ( spec & 0x00000001 ) {
	408	id += ii;
	409	break;
	410	}
	411	spec = spec >> 1 ;
	412	}
	413
	414	fMemReader->SetEquipmentID( id );
	415
	416	fClusterizer->ResetTree();
	417	Bool_t iclustered = fClusterizer->Raw2ClustersChamber(fMemReader);
	418	if (iclustered == kTRUE)
0af7cb2e	419	{
886e8d3d	420	HLTDebug( "Clustered successfully");
	421	}
	422	else
	423	{
	424	HLTError("Clustering ERROR");
0af7cb2e	425	return -1;
0af7cb2e	426	}
519f385f	427
886e8d3d	428	// put the tree into output blocks
886e8d3d	429	//fcTree->Print();
9aea5deb	430	TClonesArray *clusterArray = fClusterizer->RecPoints();
	431	fClusterizer->SetClustersOwner(kFALSE);
	432
	433	if (clusterArray){
	434	HLTDebug("clusterArray: Entries - %i; Size - %i",clusterArray->GetEntriesFast(),sizeof(clusterArray));
	435	PushBack(clusterArray, AliHLTTRDDefinitions::fgkClusterDataType, blocks[i].fSpecification);
	436	}
	437	else
	438	HLTWarning("Array of clusters is empty!");
	439
	440	if (clusterArray){
	441	clusterArray->Delete();
	442	delete clusterArray;
	443	}
886e8d3d	444
519f385f	445
9aea5deb	446	}// for ( unsigned long i = 0; i < evtData.fBlockCnt; i++ )
b39f18ce	447
9aea5deb	448
	449
	450
b39f18ce	451	size=0; // this function did not write data to the buffer directly
0af7cb2e	452	return 0;
0af7cb2e	453	}