Dotnet/FastRng
Dotnet/FastRng
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases 5 Wiki Activity

Added optional threshold

Browse Source
This commit is contained in:
Thorsten Sommer 2020-09-29 20:10:55 +02:00
parent c04ac3b4d4
commit e6ec8aeafa
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
FastRng/Double/ShapeFitter.cs
View File

@ -10,10 +10,12 @@ namespace FastRng.Double
private readonly double[] samples; private readonly double[] samples;
private readonly IRandom rng; private readonly IRandom rng;
private readonly ushort sampleSize; private readonly ushort sampleSize;
private readonly double threshold;
public ShapeFitter(Func<double, double> shapeFunction, IRandom rng, ushort sampleSize = 100) public ShapeFitter(Func<double, double> shapeFunction, IRandom rng, ushort sampleSize = 100, double threshold = 0.99)
{ {
this.rng = rng; this.rng = rng;
this.threshold = threshold;
this.sampleSize = sampleSize; this.sampleSize = sampleSize;
this.samples = new double[sampleSize]; this.samples = new double[sampleSize];
this.probabilities = new double[sampleSize]; this.probabilities = new double[sampleSize];
@ -35,7 +37,7 @@ namespace FastRng.Double
var nextBucket = (int)Math.Floor(nextNumber * this.sampleSize); var nextBucket = (int)Math.Floor(nextNumber * this.sampleSize);
this.samples[nextBucket] += this.probabilities[nextBucket]; this.samples[nextBucket] += this.probabilities[nextBucket];
if (this.samples[nextBucket] >= 1.0) if (this.samples[nextBucket] >= this.threshold)
{ {
this.samples[nextBucket] = 0.0; this.samples[nextBucket] = 0.0;
return nextNumber; return nextNumber;