AtSample.h

Go to the documentation of this file.

#ifndef ATHITSAMPLER_H
#define ATHITSAMPLER_H
 
#include <Math/Point3Dfwd.h> // for XYZPoint
 
#include <algorithm>
#include <memory>
#include <vector>
 
class AtHit;
 
namespace RandomSample {
enum class SampleMethod;
 
class AtSample {
protected:
   using HitPtr = std::unique_ptr<AtHit>;
   const std::vector<const AtHit *> *fHits; //< Hits to sample from
   std::vector<double> fCDF;                //< Cummulative distribution function for hits
   bool fWithReplacement{false};            //< If we should sample with replacement
 
public:
   virtual ~AtSample() = default;
 
   virtual std::vector<AtHit> SampleHits(int N);
   std::vector<ROOT::Math::XYZPoint> SamplePoints(int N);
 
   virtual void SetHitsToSample(const std::vector<const AtHit *> &hits) = 0;
   [[deprecated]] void SetHitsToSample(const std::vector<HitPtr> &hits);
   [[deprecated]] void SetHitsToSample(const std::vector<AtHit> &hits);
 
   void SetSampleWithReplacement(bool val) { fWithReplacement = val; }
 
protected:
   virtual std::vector<double> PDF(const AtHit &hit) = 0;
   void FillCDF();
 
   std::vector<int> sampleIndicesFromCDF(int N, std::vector<int> vetoed = {});
   int getIndexFromCDF(double r, double rmCFD, std::vector<int> vetoed);
 
   double getPDFfromCDF(int index);
 
   template <typename T>
   static inline bool isInVector(T val, std::vector<T> vec)
   {
      if (vec.size() == 0)
         return false;
      return std::find(vec.begin(), vec.end(), val) != vec.end();
   }
};
} // namespace RandomSample
 
#endif //#ifndef ATHITSAMPLER_H