NoiseGenerator.h

Go to the documentation of this file.

 
#ifndef TOOLS_NOISEGENERATOR_H
#define TOOLS_NOISEGENERATOR_H
 
//----------------//
//   C++ StdLib   //
//----------------//
#include <time.h>
 
#include <iostream>
#include <vector>
 
//--------------//
//  boost::math //
//--------------//
#include <boost/math/distributions/poisson.hpp>
 
//----------//
//   ROOT   //
//----------//
#include "Framework/Exception/Exception.h"
#include "Framework/Logger.h"
#include "Math/DistFunc.h"
#include "TRandom3.h"
 
namespace ldmx {
 
class NoiseGenerator {
 public:
  NoiseGenerator(double noiseValue = 0.0001, bool gauss = true);
 
  ~NoiseGenerator() = default;
 
  void seedGenerator(uint64_t seed);
 
  bool hasSeed() const { return random_.get() != nullptr; }
 
  std::vector<double> generateNoiseHits(int emptyChannels);
 
  void setNoiseThreshold(double noiseThreshold) {
    noise_threshold_ = noiseThreshold;
  }
 
  void setNoise(double noise) { noise_ = noise; };
 
  void setPedestal(double pedestal) { pedestal_ = pedestal; };
 
 private:
  std::unique_ptr<TRandom3> random_{nullptr};
 
  double noise_threshold_{4};
 
  double noise_{1};
 
  double pedestal_{0};
 
  bool use_gaussian_model_{true};
 
  std::unique_ptr<boost::math::poisson_distribution<> > poisson_dist_;
 
  enableLogging("NoiseGenerator")
};  // NoiseGenerator
 
}  // namespace ldmx
 
#endif  // TOOLS_NOISEGENERATOR_H