SplineUtil.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_SPLINEUTIL_H
#define ALICEO2_GPUCOMMON_TPCFASTTRANSFORMATION_SPLINEUTIL_H
 
namespace o2
{
namespace gpu
{
 
class SplineUtil
{
 public:
  static constexpr int32_t getSpec(int32_t nXdim, int32_t nYdim)
  {
    // List of the Spline class specializations:
    //
    //  0 - a parent class for other specializations
    //  1 - nXdim>0, nYdim>0: both nXdim and nYdim are set at the compile time
    //  2 - at least one of the dimensions must be set during runtime
    //  3 - specialization where nYdim==1 (a small add-on on top of the other specs)
 
    return (nYdim == 1) ? 3 : ((nXdim > 0 && nYdim > 0) ? 1 : 2);
    /*
    if (nYdim == 1) {
      return 3;
    }
    if (nXdim > 0 && nYdim > 0) {
      return 1;
    } else {
      return 2;
    }
    */
  }
 
  static constexpr int32_t getSpec(int32_t nYdim)
  {
    return getSpec(1, nYdim);
  }
 
  static constexpr int32_t abs(int32_t v) { return (v >= 0) ? v : -v; }
 
  template <bool ConditionT, int32_t ValTrueT>
  class Switch;
 
  template <int32_t NdimT>
  GPUd() static Switch<(NdimT > 0), NdimT> getNdim(int32_t Ndim)
  {
    return Switch<(NdimT > 0), NdimT>(Ndim);
  }
 
  template <int32_t NdimT>
  GPUd() static Switch<(NdimT != 0), abs(NdimT)> getMaxNdim(int32_t Ndim)
  {
    return Switch<(NdimT != 0), abs(NdimT)>(Ndim);
  }
};
 
template <int32_t ValTrueT>
class SplineUtil::Switch<true, ValTrueT>
{
 public:
  GPUd() Switch(int32_t /*valFalse*/) {}
  GPUd() static constexpr int32_t get() { return ValTrueT; }
};
 
template <int32_t ValTrueT>
class SplineUtil::Switch<false, ValTrueT>
{
 public:
  GPUd() Switch(int32_t valFalse) : mValFalse(valFalse) {}
  GPUd() int32_t get() const { return mValFalse; }
 
 private:
  int32_t mValFalse;
};
 
} // namespace gpu
} // namespace o2
 
#endif