GooPdf.h

Go to the documentation of this file.

#pragma once

#include <thrust/functional.h>

#include <goofit/PDFs/MetricTaker.h>
#include <goofit/PdfBase.h>
#include <goofit/UnbinnedDataSet.h>

#ifdef ROOT_FOUND
class TH1D;
#endif

namespace GooFit {

enum class EvalFunc : size_t { Eval = 0, NLL, Prob, BinAvg, BinWithError, Chisq };

constexpr const char *evalfunc_vals[]
    = {"ptr_to_Eval", "ptr_to_NLL", "ptr_to_Prob", "ptr_to_BinAvg", "ptr_to_BinWithError", "ptr_to_Chisq"};

constexpr const char *evalfunc_to_string(EvalFunc val) { return evalfunc_vals[static_cast<size_t>(val)]; }

#ifdef SEPARABLE

extern __constant__ fptype cudaArray[maxParams];

extern __constant__ unsigned int paramIndices[maxParams];

extern __constant__ fptype functorConstants[maxParams];

extern __constant__ fptype normalisationFactors[maxParams];

extern __device__ void *device_function_table[200];
extern void *host_function_table[200];
extern unsigned int num_device_functions;
extern std::map<void *, int> functionAddressToDeviceIndexMap;
#endif

__device__ int dev_powi(int base, int exp); // Implemented in SmoothHistogramPdf.
void *getMetricPointer(std::string name);

typedef fptype (*device_function_ptr)(fptype *, fptype *, unsigned int *);

typedef fptype (*device_metric_ptr)(fptype, fptype *, unsigned int);

extern void *host_fcn_ptr;

__device__ fptype callFunction(fptype *eventAddress, unsigned int functionIdx, unsigned int paramIdx);

class GooPdf : public PdfBase {
  public:
    using PdfBase::PdfBase;

    double calculateNLL() const override;

    __host__ std::vector<fptype> evaluateAtPoints(Observable var);

    __host__ fptype normalize() const override;

    __host__ fptype normalise() const { return normalize(); }

    __host__ virtual fptype integrate(fptype lo, fptype hi) const { return 0; }
    __host__ bool hasAnalyticIntegral() const override { return false; }
    __host__ fptype getValue(EvalFunc evalfunc = EvalFunc::Eval);

    __host__ std::vector<std::vector<fptype>> getCompProbsAtDataPoints();

    __host__ UnbinnedDataSet makeGrid();

    __host__ void initialize(std::vector<unsigned int> pindices, void *dev_functionPtr = host_fcn_ptr);
    __host__ void scan(Observable var, std::vector<fptype> &values);
    __host__ void setFitControl(std::shared_ptr<FitControl> fc) override;
    __host__ virtual void setMetrics();
    __host__ void setParameterConstantness(bool constant = true);

    __host__ virtual void transformGrid(fptype *host_output);
    static __host__ int findFunctionIdx(void *dev_functionPtr);
    __host__ void setDebugMask(int mask, bool setSpecific = true) const;

#ifdef ROOT_FOUND
    __host__ TH1D *plotToROOT(Observable var, double normFactor = 1, std::string name = "");
#endif

  protected:
    __host__ virtual double sumOfNll(int numVars) const;
    std::shared_ptr<MetricTaker> logger;

  private:
};

} // namespace GooFit