MetricTaker.h

Go to the documentation of this file.

#pragma once

#include <thrust/functional.h>

#include <goofit/PdfBase.h>

namespace GooFit {

// Notice that operators are distinguished by the order of the operands,
// and not otherwise! It's up to the user to make his tuples correctly.
class MetricTaker : public thrust::unary_function<thrust::tuple<int, fptype *, int>, fptype> {
  public:
    MetricTaker(PdfBase *dat, void *dev_functionPtr);
    MetricTaker(int fIdx, int pIdx);

    __device__ fptype operator()(thrust::tuple<int, fptype *, int> t) const;

    __device__ fptype operator()(thrust::tuple<int, int, fptype *> t) const;

  private:
    unsigned int metricIndex;

    unsigned int functionIdx;

    unsigned int parameters;
};

} // namespace GooFit