Distribution.h

Go to the documentation of this file.

// Class Distribution
//   This class can be used for creating a distribution object
//   with custom pdf, cdf, and estimate function that is used for
//   the sampling method.
//
 
#ifndef IPPL_DISTRIBUTION_H
#define IPPL_DISTRIBUTION_H
 
#include "Types/ViewTypes.h"
 
#include "Random/Utility.h"
 
namespace ippl {
    namespace random {


        template <typename T, unsigned Dim, unsigned DimP, typename DistributionFunctions>
        class Distribution {
        public:
            T par_m[DimP];
            typename DistributionFunctions::PDF pdf_m;
            typename DistributionFunctions::CDF cdf_m;
            typename DistributionFunctions::Estimate estimate_m;
            KOKKOS_INLINE_FUNCTION Distribution(const T* par_p) {
                for (unsigned int i = 0; i < DimP; i++) {
                    par_m[i] = par_p[i];
                }
            }

            KOKKOS_INLINE_FUNCTION ~Distribution() {}

            KOKKOS_INLINE_FUNCTION T getPdf(T x, unsigned int d) const {
                return pdf_m(x, d, par_m);
            }

            KOKKOS_INLINE_FUNCTION T getCdf(T x, unsigned int d) const {
                return cdf_m(x, d, par_m);
            }

            KOKKOS_INLINE_FUNCTION T getEstimate(T x, unsigned int d) const {
                return estimate_m(x, d, par_m);
            }

            KOKKOS_INLINE_FUNCTION T getObjFunc(T x, unsigned int d, T u) const {
                return getCdf(x, d) - u;
            }

            KOKKOS_INLINE_FUNCTION T getDerObjFunc(T x, unsigned int d) const {
                return getPdf(x, d);
            }

            KOKKOS_INLINE_FUNCTION T getFullPdf(ippl::Vector<T, Dim> x) const {
                T totalPdf = 1.0;
                for (unsigned int d = 0; d < Dim; d++) {
                    totalPdf *= getPdf(x[d], d);
                }
                return totalPdf;
            }
        };
    }  // namespace random
}  // namespace ippl
 
#endif