FastSimulation/Utilities/interface/GammaNumericalGenerator.h - master - githubmirror/cmssw

FastSimulation/Utilities/interface/GammaNumericalGenerator.h

50 строк1 KB

Cms Build

Clang-Format

24 май 2019, 00:14

24 май 2019, 00:14740c44a

#ifndef GammaNumericalGenerator_H
#define GammaNumericalGenerator_H
#include "FastSimulation/Utilities/interface/BaseNumericalRandomGenerator.h"

#include <cmath>
/** Numerical Random Generator for Gamma distribution.
 *  Copy of LandauFluctuations
 */

class RandomEngineAndDistribution;

class GammaNumericalGenerator : public BaseNumericalRandomGenerator {
public:
  /// Constructor : initialization of the Random Generator
  GammaNumericalGenerator(double a = 0, double b = 0, double x1 = 0, double x2 = 0)
      : BaseNumericalRandomGenerator(x1, x2, 1000), a_(a), b_(b), valid(false) {
    if (a > 0 && b > 0) {
      valid = true;
      initialize();
    }
  }

  /// Default destructor
  ~GammaNumericalGenerator() override {}

  /// Random generator
  double gamma(RandomEngineAndDistribution const* random) const { return generate(random); }

  double gamma_exp(RandomEngineAndDistribution const* random) const { return generateExp(random); }

  double gamma_lin(RandomEngineAndDistribution const* random) const { return generateLin(random); }

  /// The probability density function implementation
  double function(double x) override { return ersatzt(x); }

  inline bool isValid() const { return valid; }

private:
  /// Gamma Function
  double ersatzt(double x) {
    double bt = b_ * x;
    return b_ * pow(bt, a_ - 1) * exp(-bt);
  }

  // gamma distribution parameters
  double a_, b_;
  bool valid;
};

#endif