GPUROOTDumpCore.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 GPUROOTDumpCore.cxx
/// \author David Rohr

#include "GPUROOTDumpCore.h"

#if (!defined(GPUCA_STANDALONE) || defined(GPUCA_BUILD_QA))
#include <atomic>
#include <memory>
#include <TFile.h>

using namespace o2::gpu;

std::weak_ptr<GPUROOTDumpCore> GPUROOTDumpCore::sInstance;
std::atomic_flag GPUROOTDumpBase::mMutex = ATOMIC_FLAG_INIT;

GPUROOTDumpCore::GPUROOTDumpCore(GPUROOTDumpCore::GPUROOTDumpCorePrivate)
{
}

GPUROOTDumpCore::~GPUROOTDumpCore()
{
  if (mFile) {
    mFile->cd();
    for (uint32_t i = 0; i < mBranches.size(); i++) {
      mBranches[i]->write();
    }
    mFile->Close();
  }
}

std::shared_ptr<GPUROOTDumpCore> GPUROOTDumpCore::getAndCreate(const char* filename)
{
  static std::atomic_flag lock = ATOMIC_FLAG_INIT;
  while (lock.test_and_set(std::memory_order_acquire)) {
  }
  std::shared_ptr<GPUROOTDumpCore> retVal = sInstance.lock();
  if (!retVal) {
    retVal = std::make_shared<GPUROOTDumpCore>(GPUROOTDumpCorePrivate());
    sInstance = retVal;
  }
  if (*filename != 0 && retVal->mFileName != "" && retVal->mFileName != filename) {
    throw std::runtime_error("GPUROOTDump reinitialized with different file name");
  }
  retVal->mFileName = filename;
  lock.clear(std::memory_order_release);
  return retVal;
}

GPUROOTDumpBase::GPUROOTDumpBase()
{
  std::shared_ptr<GPUROOTDumpCore> p = GPUROOTDumpCore::get().lock();
  if (!p) {
    throw std::runtime_error("No instance of GPUROOTDumpCore exists");
  }
  p->mBranches.emplace_back(this);
  if (!p->mFile) {
    if (p->mFileName == "") {
      throw std::runtime_error("GPUROOTDump output file name not set");
    }
    std::remove(p->mFileName.c_str());
    p->mFile.reset(new TFile(p->mFileName.c_str(), "recreate"));
  }
  p->mFile->cd();
}

#endif