Implementing to handling for ESD members which are not TClonesArrays

[u/mrichter/AliRoot.git] / HLT / TPCLib / tracking-ca / AliHLTTPCCASliceOutput.cxx

// @(#) $Id$
// **************************************************************************
// This file is property of and copyright by the ALICE HLT Project          *
// ALICE Experiment at CERN, All rights reserved.                           *
//                                                                          *
// Primary Authors: Sergey Gorbunov <sergey.gorbunov@kip.uni-heidelberg.de> *
//                  Ivan Kisel <kisel@kip.uni-heidelberg.de>                *
//                  for The ALICE HLT 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.                    *
//                                                                          *
//***************************************************************************

#include "AliHLTTPCCASliceOutput.h"
#include "MemoryAssignmentHelpers.h"


GPUhd() int AliHLTTPCCASliceOutput::EstimateSize( int nOfTracks, int nOfTrackClusters )
{
  // calculate the amount of memory [bytes] needed for the event

  const int kClusterDataSize = sizeof(  int ) + sizeof( unsigned short ) + sizeof( float2 ) + sizeof( float ) + sizeof( UChar_t ) + sizeof( UChar_t );

  return sizeof( AliHLTTPCCASliceOutput ) + sizeof( AliHLTTPCCASliceTrack )*nOfTracks + kClusterDataSize*nOfTrackClusters;
}

#ifndef HLTCA_GPUCODE

void AliHLTTPCCASliceOutput::Clear()
{
	//Clear Slice Output and free Memory
	if (fMemory) delete[] fMemory;
	fMemory = NULL;
	fNOutTracks = 0;
	fNOutTrackHits = 0;
	fNTracks = 0;
	fNTrackClusters = 0;
}

template<typename T> inline void AssignNoAlignment( T *&dst, char *&mem, int count )
{
  // assign memory to the pointer dst
  dst = ( T* ) mem;
  mem = ( char * )( dst + count );
}

void AliHLTTPCCASliceOutput::SetPointers()
{
  // set all pointers

  char *mem = fMemory;
  AssignNoAlignment( fTracks,            mem, fNTracks );
  AssignNoAlignment( fClusterUnpackedYZ, mem, fNTrackClusters );
  AssignNoAlignment( fClusterUnpackedX,  mem, fNTrackClusters );
  AssignNoAlignment( fClusterId,         mem, fNTrackClusters );
  AssignNoAlignment( fClusterPackedYZ,   mem, fNTrackClusters );
  AssignNoAlignment( fClusterRow,        mem, fNTrackClusters );
  AssignNoAlignment( fClusterPackedAmp,  mem, fNTrackClusters );

  // memory for output tracks

  AssignMemory( fOutTracks, mem, fNTracks );

  // arrays for track hits

  AssignMemory( fOutTrackHits, mem, fNTrackClusters );


  fMemorySize = (mem - fMemory);
}

void AliHLTTPCCASliceOutput::Allocate()
{
	//Allocate All memory needed for slice output
  if (fMemory) delete[] fMemory;
  SetPointers(); // to calculate the size
  fMemory = reinterpret_cast<char*> ( new uint4 [ fMemorySize/sizeof( uint4 ) + 100] );
  SetPointers(); // set pointers
}
#endif