GPUTPCSectorOutput.cxx

Go to the documentation of this file.

// 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.
 
 
#include "GPUOutputControl.h"
#include "GPUTPCSectorOutput.h"
#include "GPUCommonMath.h"
#include <atomic>
 
using namespace o2::gpu;
 
uint32_t GPUTPCSectorOutput::EstimateSize(uint32_t nOfTracks, uint32_t nOfTrackClusters)
{
  // calculate the amount of memory [bytes] needed for the event
  return sizeof(GPUTPCSectorOutput) + sizeof(GPUTPCTrack) * nOfTracks + sizeof(GPUTPCSectorOutCluster) * nOfTrackClusters;
}
 
#ifndef GPUCA_GPUCODE
void GPUTPCSectorOutput::Allocate(GPUTPCSectorOutput*& ptrOutput, int32_t nTracks, int32_t nTrackHits, GPUOutputControl* outputControl, void*& internalMemory)
{
  // Allocate All memory needed for sector output
  const size_t memsize = EstimateSize(nTracks, nTrackHits);
 
  if (outputControl && outputControl->useExternal()) {
    static std::atomic_flag lock = ATOMIC_FLAG_INIT;
    while (lock.test_and_set(std::memory_order_acquire)) {
    }
    outputControl->checkCurrent();
    if (outputControl->size - ((char*)outputControl->ptrCurrent - (char*)outputControl->ptrBase) < memsize) {
      outputControl->size = 1;
      ptrOutput = nullptr;
      lock.clear(std::memory_order_release);
      return;
    }
    ptrOutput = reinterpret_cast<GPUTPCSectorOutput*>(outputControl->ptrCurrent);
    outputControl->ptrCurrent = (char*)outputControl->ptrCurrent + memsize;
    lock.clear(std::memory_order_release);
  } else {
    if (internalMemory) {
      free(internalMemory);
    }
    internalMemory = malloc(memsize);
    ptrOutput = reinterpret_cast<GPUTPCSectorOutput*>(internalMemory);
  }
  ptrOutput->SetMemorySize(memsize);
}
#endif