File radialbincomputer.h

File List > code_source > templat > include > TempLat > lattice > measuringtools > projectionhelpers > radialbincomputer.h

Go to the documentation of this file

#ifndef TEMPLAT_LATTICE_MEASUREMENTS_PROJECTIONHELPERS_RADIALBINCOMPUTER_H
#define TEMPLAT_LATTICE_MEASUREMENTS_PROJECTIONHELPERS_RADIALBINCOMPUTER_H

/* This file is part of TempLat, available at https://cosmolattice.github.io/templat .
   Copyright 2021-2026 The TempLat authors, see AUTHORS.md.
   Released under the MIT license, see LICENSE.md. */

// File info: Main contributor(s): Wessel Valkenburg, Year: 2019

#include <algorithm>

#include "TempLat/parallel/device.h"

namespace TempLat
{
  class RadialBinComputer
  {
  public:
    RadialBinComputer(double minVal, double maxVal, device::Idx nBins, double deltaKBin)
        : mMinVal(minVal), mMaxVal(maxVal), mRange(mMaxVal - mMinVal), mNBins(nBins), mHighestBin(nBins - 1),
          mDeltakBin(deltaKBin)
    {
      if (mRange <= 0) mRange = 1;
    }

    DEVICE_FUNCTION
    device::Idx operator()(double value) const
    {
      const device::Idx bin = static_cast<device::Idx>(device::floor((value - mMinVal) / mDeltakBin));
      return device::min(mHighestBin, device::max(device::Idx(0), bin));
    }

    template <typename T> void setCentralBinBounds(std::vector<T> &res)
    {
      res = std::vector<T>(mNBins);
      T steps = mDeltakBin;
      for (device::Idx i = 0; i < mNBins; ++i) {
        res[i] = mMinVal + mDeltakBin / 2. + i * steps;
      }
    }

  private:
    /* Put all member variables and private methods here. These may change arbitrarily. */
    double mMinVal;
    double mMaxVal;
    double mRange;
    device::Idx mNBins;
    device::Idx mHighestBin;
    double mDeltakBin;
  };
} // namespace TempLat

#endif