IrregularSpline2D3D.h

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.
 
 
#ifndef ALICEO2_GPUCOMMON_TPCFASTTRANSFORMATION_IRREGULARSPLINE2D3D_H
#define ALICEO2_GPUCOMMON_TPCFASTTRANSFORMATION_IRREGULARSPLINE2D3D_H
 
#include "IrregularSpline1D.h"
#include "FlatObject.h"
#include "GPUCommonDef.h"
 
#if !defined(__ROOTCLING__) && !defined(GPUCA_GPUCODE) && !defined(GPUCA_NO_VC)
#include <Vc/Vc>
#include <Vc/SimdArray>
#endif
 
namespace o2
{
namespace gpu
{
 
class IrregularSpline2D3D : public FlatObject
{
 public:
 
  IrregularSpline2D3D();
 
  IrregularSpline2D3D(const IrregularSpline2D3D&) = delete;
 
  IrregularSpline2D3D& operator=(const IrregularSpline2D3D&) = delete;
 
  ~IrregularSpline2D3D() = default;
 
 
 
  using FlatObject::getBufferAlignmentBytes;
  using FlatObject::getClassAlignmentBytes;
 
 
  void cloneFromObject(const IrregularSpline2D3D& obj, char* newFlatBufferPtr);
  void destroy();
 
 
  using FlatObject::releaseInternalBuffer;
  void moveBufferTo(char* newBufferPtr);
 
 
  void setActualBufferAddress(char* actualFlatBufferPtr);
  void setFutureBufferAddress(char* futureFlatBufferPtr);
 
 
  void construct(int32_t numberOfKnotsU, const float knotsU[], int32_t numberOfAxisBinsU, int32_t numberOfKnotsV, const float knotsV[], int32_t numberOfAxisBinsV);
 
  void constructRegular(int32_t numberOfKnotsU, int32_t numberOfKnotsV);
 
 
  template <typename T>
  GPUd() void correctEdges(T* data) const;
 
  template <typename T>
  GPUd() void getSpline(GPUgeneric() const T* correctedData, float u, float v, GPUgeneric() T& x, GPUgeneric() T& y, GPUgeneric() T& z) const;
 
  GPUd() void getSplineVec(const float* correctedData, float u, float v, float& x, float& y, float& z) const;
 
  GPUd() int32_t getNumberOfKnots() const { return mGridU.getNumberOfKnots() * mGridV.getNumberOfKnots(); }
 
  GPUd() const IrregularSpline1D& getGridU() const { return mGridU; }
 
  GPUd() const IrregularSpline1D& getGridV() const { return mGridV; }
 
  GPUd() const IrregularSpline1D& getGrid(int32_t uv) const { return (uv == 0) ? mGridU : mGridV; }
 
  GPUd() void getKnotUV(int32_t iKnot, float& u, float& v) const;
 
  size_t getFlatBufferSize() const { return mFlatBufferSize; }
 
  const char* getFlatBufferPtr() const { return mFlatBufferPtr; }
 
  static constexpr size_t getClassAlignmentBytes() { return 8; }
 
  static constexpr size_t getBufferAlignmentBytes() { return 8; }
 
  static constexpr size_t getDataAlignmentBytes() { return 8; }
 
 
  size_t getGridUOffset() const { return mGridU.getFlatBufferPtr() - mFlatBufferPtr; }
 
  size_t getGridVOffset() const { return mGridV.getFlatBufferPtr() - mFlatBufferPtr; }
 
  void print() const;
 
 private:
 
  IrregularSpline1D mGridU; 
  IrregularSpline1D mGridV; 
 
  ClassDefNV(IrregularSpline2D3D, 1);
};
 
 
GPUdi() void IrregularSpline2D3D::getKnotUV(int32_t iKnot, float& u, float& v) const
{
  const IrregularSpline1D& gridU = getGridU();
  const IrregularSpline1D& gridV = getGridV();
  int32_t nu = gridU.getNumberOfKnots();
  int32_t iv = iKnot / nu;
  int32_t iu = iKnot % nu;
  u = gridU.getKnot(iu).u;
  v = gridV.getKnot(iv).u;
}
 
template <typename T>
GPUd() void IrregularSpline2D3D::correctEdges(T* data) const
{
  const IrregularSpline1D& gridU = getGridU();
  const IrregularSpline1D& gridV = getGridV();
  int32_t nu = gridU.getNumberOfKnots();
  int32_t nv = gridV.getNumberOfKnots();
 
  { // left edge of U
    int32_t iu = 0;
    const IrregularSpline1D::Knot* s = gridU.getKnots() + iu;
    double c0, c1, c2, c3;
    gridU.getEdgeCorrectionCoefficients(s[0].u, s[1].u, s[2].u, s[3].u, c0, c1, c2, c3);
    for (int32_t iv = 0; iv < nv; iv++) {
      T* f0 = data + (nu * (iv) + iu) * 3;
      T* f1 = f0 + 3;
      T* f2 = f0 + 6;
      T* f3 = f0 + 9;
      for (int32_t idim = 0; idim < 3; idim++) {
        f0[idim] = (T)(c0 * f0[idim] + c1 * f1[idim] + c2 * f2[idim] + c3 * f3[idim]);
      }
    }
  }
 
  { // right edge of U
    int32_t iu = nu - 4;
    const IrregularSpline1D::Knot* s = gridU.getKnots() + iu;
    double c0, c1, c2, c3;
    gridU.getEdgeCorrectionCoefficients(s[3].u, s[2].u, s[1].u, s[0].u, c3, c2, c1, c0);
    for (int32_t iv = 0; iv < nv; iv++) {
      T* f0 = data + (nu * (iv) + iu) * 3;
      T* f1 = f0 + 3;
      T* f2 = f0 + 6;
      T* f3 = f0 + 9;
      for (int32_t idim = 0; idim < 3; idim++) {
        f3[idim] = (T)(c0 * f0[idim] + c1 * f1[idim] + c2 * f2[idim] + c3 * f3[idim]);
      }
    }
  }
 
  { // low edge of V
    int32_t iv = 0;
    const IrregularSpline1D::Knot* s = gridV.getKnots() + iv;
    double c0, c1, c2, c3;
    gridV.getEdgeCorrectionCoefficients(s[0].u, s[1].u, s[2].u, s[3].u, c0, c1, c2, c3);
    for (int32_t iu = 0; iu < nu; iu++) {
      T* f0 = data + (nu * iv + iu) * 3;
      T* f1 = f0 + nu * 3;
      T* f2 = f0 + nu * 6;
      T* f3 = f0 + nu * 9;
      for (int32_t idim = 0; idim < 3; idim++) {
        f0[idim] = (T)(c0 * f0[idim] + c1 * f1[idim] + c2 * f2[idim] + c3 * f3[idim]);
      }
    }
  }
 
  { // high edge of V
    int32_t iv = nv - 4;
    const IrregularSpline1D::Knot* s = gridV.getKnots() + iv;
    double c0, c1, c2, c3;
    gridV.getEdgeCorrectionCoefficients(s[3].u, s[2].u, s[1].u, s[0].u, c3, c2, c1, c0);
    for (int32_t iu = 0; iu < nu; iu++) {
      T* f0 = data + (nu * iv + iu) * 3;
      T* f1 = f0 + nu * 3;
      T* f2 = f0 + nu * 6;
      T* f3 = f0 + nu * 9;
      for (int32_t idim = 0; idim < 3; idim++) {
        f3[idim] = (T)(c0 * f0[idim] + c1 * f1[idim] + c2 * f2[idim] + c3 * f3[idim]);
      }
    }
  }
}
 
template <typename T>
GPUdi() void IrregularSpline2D3D::getSpline(GPUgeneric() const T* correctedData, float u, float v, GPUgeneric() T& x, GPUgeneric() T& y, GPUgeneric() T& z) const
{
  // Get interpolated value for f(u,v) using data array correctedData[getNumberOfKnots()] with corrected edges
 
  const IrregularSpline1D& gridU = getGridU();
  const IrregularSpline1D& gridV = getGridV();
  int32_t nu = gridU.getNumberOfKnots();
  int32_t iu = gridU.getKnotIndex(u);
  int32_t iv = gridV.getKnotIndex(v);
 
  const IrregularSpline1D::Knot& knotU = gridU.getKnot(iu);
  const IrregularSpline1D::Knot& knotV = gridV.getKnot(iv);
 
  const T* dataV0 = correctedData + (nu * (iv - 1) + iu - 1) * 3;
  const T* dataV1 = dataV0 + 3 * nu;
  const T* dataV2 = dataV0 + 6 * nu;
  const T* dataV3 = dataV0 + 9 * nu;
 
  T dataV[12];
  for (int32_t i = 0; i < 12; i++) {
    dataV[i] = gridV.getSpline(knotV, dataV0[i], dataV1[i], dataV2[i], dataV3[i], v);
  }
 
  T* dataU0 = dataV + 0;
  T* dataU1 = dataV + 3;
  T* dataU2 = dataV + 6;
  T* dataU3 = dataV + 9;
 
  T res[3];
  for (int32_t i = 0; i < 3; i++) {
    res[i] = gridU.getSpline(knotU, dataU0[i], dataU1[i], dataU2[i], dataU3[i], u);
  }
  x = res[0];
  y = res[1];
  z = res[2];
}
 
GPUdi() void IrregularSpline2D3D::getSplineVec(const float* correctedData, float u, float v, float& x, float& y, float& z) const
{
  // Same as getSpline, but using vectorized calculation.
  // \param correctedData should be at least 128-bit aligned
 
#if !defined(__ROOTCLING__) && !defined(GPUCA_GPUCODE) && !defined(GPUCA_NO_VC)
  const IrregularSpline1D& gridU = getGridU();
  const IrregularSpline1D& gridV = getGridV();
  int32_t nu = gridU.getNumberOfKnots();
  int32_t iu = gridU.getKnotIndex(u);
  int32_t iv = gridV.getKnotIndex(v);
  const IrregularSpline1D::Knot& knotU = gridU.getKnot(iu);
  const IrregularSpline1D::Knot& knotV = gridV.getKnot(iv);
 
  const float* dataV0 = correctedData + (nu * (iv - 1) + iu - 1) * 3;
  const float* dataV1 = dataV0 + 3 * nu;
  const float* dataV2 = dataV0 + 6 * nu;
  const float* dataV3 = dataV0 + 9 * nu;
 
  // calculate F values at V==v and U == Ui of knots
 
  Vc::SimdArray<float, 12> dataV0vec(dataV0), dataV1vec(dataV1), dataV2vec(dataV2), dataV3vec(dataV3), dataVvec = gridV.getSpline(knotV, dataV0vec, dataV1vec, dataV2vec, dataV3vec, v);
 
  using V = std::conditional_t<(Vc::float_v::size() >= 4),
                               Vc::float_v,
                               Vc::SimdArray<float, 3>>;
 
  float dataV[9 + V::size()];
  // dataVvec.scatter( dataV, Vc::SimdArray<float,12>::IndexType::IndexesFromZero() );
  // dataVvec.scatter(dataV, Vc::SimdArray<uint32_t, 12>(Vc::IndexesFromZero));
  dataVvec.store(dataV, Vc::Unaligned);
 
  for (uint32_t i = 12; i < 9 + V::size(); i++) { // fill not used part of the vector with 0
    dataV[i] = 0.f;
  }
  // calculate F values at V==v and U == u
  V dataU0vec(dataV), dataU1vec(dataV + 3), dataU2vec(dataV + 6), dataU3vec(dataV + 9);
 
  V dataUVvec = gridU.getSpline(knotU, dataU0vec, dataU1vec, dataU2vec, dataU3vec, u);
 
  x = dataUVvec[0];
  y = dataUVvec[1];
  z = dataUVvec[2];
#else
  getSpline(correctedData, u, v, x, y, z);
#endif
}
} // namespace gpu
} // namespace o2
 
#endif