GPUITSFitter.cxx

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.

/// \file GPUITSFitter.cxx
/// \author David Rohr, Maximiliano Puccio

#include "GPUITSFitter.h"

#include "ITStracking/Road.h"
#include "ITStracking/Cluster.h"
#include "GPUITSTrack.h"
#include "GPUReconstruction.h"

using namespace o2::gpu;

#ifndef GPUCA_GPUCODE
void GPUITSFitter::InitializeProcessor()
{
}

void* GPUITSFitter::SetPointersInput(void* mem)
{
  computePointerWithAlignment(mem, mRoads, mNumberOfRoads);
  for (int32_t i = 0; i < 7; i++) {
    computePointerWithAlignment(mem, mTF[i], mNTF[i]);
  }
  return mem;
}

void* GPUITSFitter::SetPointersTracks(void* mem)
{
  computePointerWithAlignment(mem, mTracks, mNMaxTracks);
  return mem;
}

void* GPUITSFitter::SetPointersMemory(void* mem)
{
  computePointerWithAlignment(mem, mMemory, 1);
  return mem;
}

void GPUITSFitter::RegisterMemoryAllocation()
{
  AllocateAndInitializeLate();
  mMemoryResInput = mRec->RegisterMemoryAllocation(this, &GPUITSFitter::SetPointersInput, GPUMemoryResource::MEMORY_INPUT, "ITSInput");
  mMemoryResTracks = mRec->RegisterMemoryAllocation(this, &GPUITSFitter::SetPointersTracks, GPUMemoryResource::MEMORY_OUTPUT, "ITSTracks");
  mMemoryResMemory = mRec->RegisterMemoryAllocation(this, &GPUITSFitter::SetPointersMemory, GPUMemoryResource::MEMORY_PERMANENT, "ITSMemory");
}

void GPUITSFitter::SetMaxData(const GPUTrackingInOutPointers& io) { mNMaxTracks = mNumberOfRoads; }
#endif

void GPUITSFitter::clearMemory()
{
  new (mMemory) Memory;
}