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

Added NextNumber for (0, 1)

Browse Source
This commit is contained in:
Thorsten Sommer 2020-09-26 12:00:46 +02:00
parent 14ea7dac4a
commit 0432dd066e
3 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
FastRng/Double/Distributions/Gamma.cs
View File

@ -6,6 +6,8 @@ namespace FastRng.Double.Distributions
{
public sealed class Gamma : IDistribution
{
private static readonly IDistribution NORMAL_DISTRIBUTION = new Normal();
private double shape = 1.0;
public IRandom Random { get; set; }
@ -35,7 +37,6 @@ namespace FastRng.Double.Distributions
if (shape >= 1.0)
{
var distNormal = new Normal();
var d = shape - 1.0 / 3.0;
var c = 1.0 / Math.Sqrt(9.0 * d);
while(true)
@ -43,7 +44,7 @@ namespace FastRng.Double.Distributions
double x, v;
do
{
x = await this.Random.NextNumber(0, 1, distNormal, token);
x = await this.Random.NextNumber(NORMAL_DISTRIBUTION, token);
v = 1.0 + c * x;
}
while (v <= 0.0);
@ -59,8 +60,7 @@ namespace FastRng.Double.Distributions
else
{
var dist = new Gamma{ Scale = 1, Shape = 1 + this.Shape};
var g = await this.Random.NextNumber(0.0f, 1.0f, dist, token); // TODO: Use double
var g = await this.Random.NextNumber(dist, token);
var w = await this.Random.GetUniform(token);
return this.Scale * g * Math.Pow(w, 1.0 / this.Shape);
}

4
FastRng/Double/IRandom.cs
View File

@ -12,7 +12,9 @@ namespace FastRng.Double
public ValueTask<ulong> NextNumber(ulong rangeStart, ulong rangeEnd, IDistribution distribution, CancellationToken cancel = default);
public ValueTask<float> NextNumber(float rangeStart, float rangeEnd, IDistribution distribution, CancellationToken cancel = default);
public ValueTask<double> NextNumber(double rangeStart, double rangeEnd, IDistribution distribution, CancellationToken cancel = default);
public ValueTask<double> NextNumber(IDistribution distribution, CancellationToken cancel = default);
public void StopProducer();
}

6
FastRng/Double/MultiThreadedRng.cs
View File

@ -167,7 +167,7 @@ namespace FastRng.Double
return (ulong) ((distributedValue * range) + rangeStart);
}
public async ValueTask<float> NextNumber(float rangeStart, float rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken))
public async ValueTask<double> NextNumber(double rangeStart, double rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken))
{
if (rangeStart > rangeEnd)
{
@ -180,9 +180,11 @@ namespace FastRng.Double
distribution.Random = this;
var distributedValue = await distribution.GetDistributedValue(cancel);
return (float) ((distributedValue * range) + rangeStart);
return (distributedValue * range) + rangeStart;
}
public async ValueTask<double> NextNumber(IDistribution distribution, CancellationToken cancel = default) => await this.NextNumber(0.0, 1.0, distribution, cancel);
public void StopProducer() => this.producerToken.Cancel();
#endregion