CalibdEdxTrackTopologySpline.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.
 
 
#if !defined(GPUCA_STANDALONE)
#include "TFile.h"
#endif
#include "CalibdEdxTrackTopologySpline.h"
 
using namespace o2::gpu;
using namespace o2::tpc;
 
#if !defined(GPUCA_STANDALONE)
CalibdEdxTrackTopologySpline::CalibdEdxTrackTopologySpline(const char* dEdxSplinesFile, const char* name)
  : FlatObject()
{
  TFile dEdxFile(dEdxSplinesFile);
  setFromFile(dEdxFile, name);
}
#endif
 
CalibdEdxTrackTopologySpline::CalibdEdxTrackTopologySpline(const CalibdEdxTrackTopologySpline& obj)
  : FlatObject()
{
  this->cloneFromObject(obj, nullptr);
}
 
CalibdEdxTrackTopologySpline& CalibdEdxTrackTopologySpline::operator=(const CalibdEdxTrackTopologySpline& obj)
{
  this->cloneFromObject(obj, nullptr);
  return *this;
}
 
void CalibdEdxTrackTopologySpline::recreate(const int32_t nKnots[])
{
  FlatObject::startConstruction();
 
  int32_t buffSize = 0;
  int32_t offsets1[FSplines];
  int32_t offsets2[FSplines];
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqMax[i].recreate(nKnots);
    buffSize = alignSize(buffSize, mCalibSplinesqMax[i].getBufferAlignmentBytes());
    offsets1[i] = buffSize;
    buffSize += mCalibSplinesqMax[i].getFlatBufferSize();
  }
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqTot[i].recreate(nKnots);
    buffSize = alignSize(buffSize, mCalibSplinesqTot[i].getBufferAlignmentBytes());
    offsets2[i] = buffSize;
    buffSize += mCalibSplinesqTot[i].getFlatBufferSize();
  }
 
  FlatObject::finishConstruction(buffSize);
 
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqMax[i].moveBufferTo(mFlatBufferPtr + offsets1[i]);
  }
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqTot[i].moveBufferTo(mFlatBufferPtr + offsets2[i]);
  }
}
 
void CalibdEdxTrackTopologySpline::cloneFromObject(const CalibdEdxTrackTopologySpline& obj, char* newFlatBufferPtr)
{
 
  const char* oldFlatBufferPtr = obj.mFlatBufferPtr;
  FlatObject::cloneFromObject(obj, newFlatBufferPtr);
 
  for (uint32_t i = 0; i < FSplines; i++) {
    char* buffer = FlatObject::relocatePointer(oldFlatBufferPtr, mFlatBufferPtr, obj.mCalibSplinesqMax[i].getFlatBufferPtr());
    mCalibSplinesqMax[i].cloneFromObject(obj.mCalibSplinesqMax[i], buffer);
  }
 
  for (uint32_t i = 0; i < FSplines; i++) {
    char* buffer = FlatObject::relocatePointer(oldFlatBufferPtr, mFlatBufferPtr, obj.mCalibSplinesqTot[i].getFlatBufferPtr());
    mCalibSplinesqTot[i].cloneFromObject(obj.mCalibSplinesqTot[i], buffer);
  }
  mMaxTanTheta = obj.mMaxTanTheta;
  mMaxSinPhi = obj.mMaxSinPhi;
 
  for (uint32_t i = 0; i < FSplines; ++i) {
    mScalingFactorsqTot[i] = obj.mScalingFactorsqTot[i];
    mScalingFactorsqMax[i] = obj.mScalingFactorsqMax[i];
  }
}
 
void CalibdEdxTrackTopologySpline::moveBufferTo(char* newFlatBufferPtr)
{
  char* oldFlatBufferPtr = mFlatBufferPtr;
  FlatObject::moveBufferTo(newFlatBufferPtr);
  char* currFlatBufferPtr = mFlatBufferPtr;
  mFlatBufferPtr = oldFlatBufferPtr;
  setActualBufferAddress(currFlatBufferPtr);
}
 
void CalibdEdxTrackTopologySpline::destroy()
{
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqMax[i].destroy();
    mCalibSplinesqTot[i].destroy();
  }
  FlatObject::destroy();
}
 
void CalibdEdxTrackTopologySpline::setActualBufferAddress(char* actualFlatBufferPtr)
{
 
  FlatObject::setActualBufferAddress(actualFlatBufferPtr);
  int32_t offset = 0;
  for (uint32_t i = 0; i < FSplines; i++) {
    offset = alignSize(offset, mCalibSplinesqMax[i].getBufferAlignmentBytes());
    mCalibSplinesqMax[i].setActualBufferAddress(mFlatBufferPtr + offset);
    offset += mCalibSplinesqMax[i].getFlatBufferSize();
  }
  for (uint32_t i = 0; i < FSplines; i++) {
    offset = alignSize(offset, mCalibSplinesqTot[i].getBufferAlignmentBytes());
    mCalibSplinesqTot[i].setActualBufferAddress(mFlatBufferPtr + offset);
    offset += mCalibSplinesqTot[i].getFlatBufferSize();
  }
}
 
void CalibdEdxTrackTopologySpline::setFutureBufferAddress(char* futureFlatBufferPtr)
{
 
  for (uint32_t i = 0; i < FSplines; i++) {
    char* buffer = relocatePointer(mFlatBufferPtr, futureFlatBufferPtr, mCalibSplinesqMax[i].getFlatBufferPtr());
    mCalibSplinesqMax[i].setFutureBufferAddress(buffer);
  }
  for (uint32_t i = 0; i < FSplines; i++) {
    char* buffer = relocatePointer(mFlatBufferPtr, futureFlatBufferPtr, mCalibSplinesqTot[i].getFlatBufferPtr());
    mCalibSplinesqTot[i].setFutureBufferAddress(buffer);
  }
  FlatObject::setFutureBufferAddress(futureFlatBufferPtr);
}
 
#if !defined(GPUCA_STANDALONE)
 
CalibdEdxTrackTopologySpline* CalibdEdxTrackTopologySpline::readFromFile(
  TFile& inpf, const char* name)
{
  return FlatObject::readFromFile<CalibdEdxTrackTopologySpline>(inpf, name);
}
 
void CalibdEdxTrackTopologySpline::setFromFile(TFile& inpf, const char* name)
{
  LOGP(info, "Warnings when reading from file can be ignored");
  o2::tpc::CalibdEdxTrackTopologySpline* cont = readFromFile(inpf, name);
  *this = *cont;
  delete cont;
  LOGP(info, "CalibdEdxTrackTopologySpline sucessfully loaded from file");
}
 
int32_t CalibdEdxTrackTopologySpline::writeToFile(TFile& outf, const char* name)
{
  LOGP(info, "Warnings when writting to file can be ignored");
  return FlatObject::writeToFile(*this, outf, name);
}
 
void CalibdEdxTrackTopologySpline::setDefaultSplines()
{
  FlatObject::startConstruction();
 
  int32_t buffSize = 0;
  int32_t offsets1[FSplines];
  int32_t offsets2[FSplines];
 
  auto defaultF = [&](const double x[], double f[]) {
    f[0] = 1.f;
  };
  double xMin[FDimX]{};
  double xMax[FDimX]{};
 
  for (int32_t iDimX = 0; iDimX < FDimX; ++iDimX) {
    xMin[iDimX] = 0;
    xMax[iDimX] = 1;
  }
 
  for (uint32_t ireg = 0; ireg < FSplines; ++ireg) {
    SplineType splineTmpqMax;
    splineTmpqMax.approximateFunction(xMin, xMax, defaultF);
    mCalibSplinesqMax[ireg] = splineTmpqMax;
    buffSize = alignSize(buffSize, mCalibSplinesqMax[ireg].getBufferAlignmentBytes());
    offsets1[ireg] = buffSize;
    buffSize += mCalibSplinesqMax[ireg].getFlatBufferSize();
  }
 
  for (uint32_t ireg = 0; ireg < FSplines; ++ireg) {
    SplineType splineTmpqTot;
    splineTmpqTot.approximateFunction(xMin, xMax, defaultF);
    mCalibSplinesqTot[ireg] = splineTmpqTot;
    buffSize = alignSize(buffSize, mCalibSplinesqTot[ireg].getBufferAlignmentBytes());
    offsets2[ireg] = buffSize;
    buffSize += mCalibSplinesqTot[ireg].getFlatBufferSize();
  }
 
  FlatObject::finishConstruction(buffSize);
 
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqMax[i].moveBufferTo(mFlatBufferPtr + offsets1[i]);
  }
  for (uint32_t i = 0; i < FSplines; i++) {
    mCalibSplinesqTot[i].moveBufferTo(mFlatBufferPtr + offsets2[i]);
  }
}
 
inline void CalibdEdxTrackTopologySpline::setRangesFromFile(TFile& inpf)
{
  std::vector<float>* tanThetaMax = nullptr;
  std::vector<float>* sinPhiMax = nullptr;
  inpf.GetObject("tanThetaMax", tanThetaMax);
  inpf.GetObject("sinPhiMax", sinPhiMax);
  if (tanThetaMax) {
    mMaxTanTheta = (*tanThetaMax).front();
    delete tanThetaMax;
  }
  if (sinPhiMax) {
    mMaxSinPhi = (*sinPhiMax).front();
    delete sinPhiMax;
  }
}
 
std::string CalibdEdxTrackTopologySpline::getSplineName(const int32_t region, const ChargeType charge)
{
  const std::string typeName[2] = {"qMax", "qTot"};
  const std::string polname = fmt::format("spline_{}_region{}", typeName[charge], region).data();
  return polname;
}
 
#endif