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

Added handling for OperationCanceledException

Browse Source
This commit is contained in:
Thorsten Sommer 2020-09-26 13:33:04 +02:00
parent 2b0e92a91d
commit 29a913843a
1 changed files with 76 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
FastRng/Double/MultiThreadedRng.cs
View File

@ -20,7 +20,7 @@ namespace FastRng.Double
private const int CAPACITY_RANDOM_NUMBERS_4_SOURCE = 16_000_000; private const int CAPACITY_RANDOM_NUMBERS_4_SOURCE = 16_000_000;
#endif #endif
private readonly CancellationTokenSource producerToken = new CancellationTokenSource(); private readonly CancellationTokenSource producerTokenSource = new CancellationTokenSource();
private readonly object syncUintGenerators = new object(); private readonly object syncUintGenerators = new object();
private readonly object syncUniformDistributedDoubleGenerators = new object(); private readonly object syncUniformDistributedDoubleGenerators = new object();
private readonly Thread[] producerRandomUint = new Thread[2]; private readonly Thread[] producerRandomUint = new Thread[2];
@ -74,13 +74,13 @@ namespace FastRng.Double
private void StartProducerThreads() private void StartProducerThreads()
{ {
this.producerRandomUint[0] = new Thread(() => this.RandomProducerUint(this.channelRandomUint.Writer, this.producerToken.Token)) {IsBackground = true}; this.producerRandomUint[0] = new Thread(() => this.RandomProducerUint(this.channelRandomUint.Writer, this.producerTokenSource.Token)) {IsBackground = true};
this.producerRandomUint[1] = new Thread(() => this.RandomProducerUint(this.channelRandomUint.Writer, this.producerToken.Token)) {IsBackground = true}; this.producerRandomUint[1] = new Thread(() => this.RandomProducerUint(this.channelRandomUint.Writer, this.producerTokenSource.Token)) {IsBackground = true};
this.producerRandomUint[0].Start(); this.producerRandomUint[0].Start();
this.producerRandomUint[1].Start(); this.producerRandomUint[1].Start();
this.producerRandomUniformDistributedDouble[0] = new Thread(() => this.RandomProducerUniformDistributedDouble(this.channelRandomUint.Reader, channelRandomUniformDistributedDouble.Writer, this.producerToken.Token)) {IsBackground = true}; this.producerRandomUniformDistributedDouble[0] = new Thread(() => this.RandomProducerUniformDistributedDouble(this.channelRandomUint.Reader, channelRandomUniformDistributedDouble.Writer, this.producerTokenSource.Token)) {IsBackground = true};
this.producerRandomUniformDistributedDouble[1] = new Thread(() => this.RandomProducerUniformDistributedDouble(this.channelRandomUint.Reader, channelRandomUniformDistributedDouble.Writer, this.producerToken.Token)) {IsBackground = true}; this.producerRandomUniformDistributedDouble[1] = new Thread(() => this.RandomProducerUniformDistributedDouble(this.channelRandomUint.Reader, channelRandomUniformDistributedDouble.Writer, this.producerTokenSource.Token)) {IsBackground = true};
this.producerRandomUniformDistributedDouble[0].Start(); this.producerRandomUniformDistributedDouble[0].Start();
this.producerRandomUniformDistributedDouble[1].Start(); this.producerRandomUniformDistributedDouble[1].Start();
} }
@ -92,40 +92,52 @@ namespace FastRng.Double
[ExcludeFromCodeCoverage] [ExcludeFromCodeCoverage]
private async void RandomProducerUint(ChannelWriter<uint> channelWriter, CancellationToken cancellationToken) private async void RandomProducerUint(ChannelWriter<uint> channelWriter, CancellationToken cancellationToken)
{ {
var buffer = new uint[CAPACITY_RANDOM_NUMBERS_4_SOURCE]; try
while (!cancellationToken.IsCancellationRequested)
{ {
lock (syncUintGenerators) var buffer = new uint[CAPACITY_RANDOM_NUMBERS_4_SOURCE];
while (!cancellationToken.IsCancellationRequested)
{ {
for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++) lock (syncUintGenerators)
{ {
this.mZ = 36_969 * (this.mZ & 65_535) + (this.mZ >> 16); for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++)
this.mW = 18_000 * (this.mW & 65_535) + (this.mW >> 16); {
buffer[n] = (this.mZ << 16) + this.mW; this.mZ = 36_969 * (this.mZ & 65_535) + (this.mZ >> 16);
this.mW = 18_000 * (this.mW & 65_535) + (this.mW >> 16);
buffer[n] = (this.mZ << 16) + this.mW;
}
} }
}
for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++) for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++)
await channelWriter.WriteAsync(buffer[n], cancellationToken); await channelWriter.WriteAsync(buffer[n], cancellationToken);
}
}
catch (OperationCanceledException)
{
} }
} }
[ExcludeFromCodeCoverage] [ExcludeFromCodeCoverage]
private async void RandomProducerUniformDistributedDouble(ChannelReader<uint> channelReaderUint, ChannelWriter<double> channelWriter, CancellationToken cancellationToken) private async void RandomProducerUniformDistributedDouble(ChannelReader<uint> channelReaderUint, ChannelWriter<double> channelWriter, CancellationToken cancellationToken)
{ {
var buffer = new double[CAPACITY_RANDOM_NUMBERS_4_SOURCE]; try
var randomUint = new uint[CAPACITY_RANDOM_NUMBERS_4_SOURCE];
while (!cancellationToken.IsCancellationRequested)
{ {
for (var n = 0; n < randomUint.Length; n++) var buffer = new double[CAPACITY_RANDOM_NUMBERS_4_SOURCE];
randomUint[n] = await channelReaderUint.ReadAsync(cancellationToken); var randomUint = new uint[CAPACITY_RANDOM_NUMBERS_4_SOURCE];
while (!cancellationToken.IsCancellationRequested)
{
for (var n = 0; n < randomUint.Length; n++)
randomUint[n] = await channelReaderUint.ReadAsync(cancellationToken);
lock (syncUniformDistributedDoubleGenerators) lock (syncUniformDistributedDoubleGenerators)
for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++) for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++)
buffer[n] = (randomUint[n] + 1.0) * 2.328306435454494e-10; // 2.328 => 1/(2^32 + 2) buffer[n] = (randomUint[n] + 1.0) * 2.328306435454494e-10; // 2.328 => 1/(2^32 + 2)
for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++) for (var n = 0; n < buffer.Length && !cancellationToken.IsCancellationRequested; n++)
await channelWriter.WriteAsync(buffer[n], cancellationToken); await channelWriter.WriteAsync(buffer[n], cancellationToken);
}
}
catch (OperationCanceledException)
{
} }
} }
@ -133,7 +145,17 @@ namespace FastRng.Double
#region Implementing interface #region Implementing interface
public async ValueTask<double> GetUniform(CancellationToken cancel = default) => await this.channelRandomUniformDistributedDouble.Reader.ReadAsync(cancel); public async ValueTask<double> GetUniform(CancellationToken cancel = default)
{
try
{
return await this.channelRandomUniformDistributedDouble.Reader.ReadAsync(cancel);
}
catch (OperationCanceledException)
{
return double.NaN;
}
}
public async ValueTask<uint> NextNumber(uint rangeStart, uint rangeEnd, IDistribution distribution, CancellationToken cancel = default) public async ValueTask<uint> NextNumber(uint rangeStart, uint rangeEnd, IDistribution distribution, CancellationToken cancel = default)
{ {
@ -147,8 +169,15 @@ namespace FastRng.Double
var range = rangeEnd - rangeStart; var range = rangeEnd - rangeStart;
distribution.Random = this; distribution.Random = this;
var distributedValue = await distribution.GetDistributedValue(cancel); try
return (uint) ((distributedValue * range) + rangeStart); {
var distributedValue = await distribution.GetDistributedValue(cancel);
return (uint) ((distributedValue * range) + rangeStart);
}
catch (OperationCanceledException)
{
return 0;
}
} }
public async ValueTask<ulong> NextNumber(ulong rangeStart, ulong rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken)) public async ValueTask<ulong> NextNumber(ulong rangeStart, ulong rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken))
@ -163,8 +192,15 @@ namespace FastRng.Double
var range = rangeEnd - rangeStart; var range = rangeEnd - rangeStart;
distribution.Random = this; distribution.Random = this;
var distributedValue = await distribution.GetDistributedValue(cancel); try
return (ulong) ((distributedValue * range) + rangeStart); {
var distributedValue = await distribution.GetDistributedValue(cancel);
return (ulong) ((distributedValue * range) + rangeStart);
}
catch (OperationCanceledException)
{
return 0;
}
} }
public async ValueTask<double> NextNumber(double rangeStart, double rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken)) public async ValueTask<double> NextNumber(double rangeStart, double rangeEnd, IDistribution distribution, CancellationToken cancel = default(CancellationToken))
@ -179,13 +215,20 @@ namespace FastRng.Double
var range = rangeEnd - rangeStart; var range = rangeEnd - rangeStart;
distribution.Random = this; distribution.Random = this;
var distributedValue = await distribution.GetDistributedValue(cancel); try
return (distributedValue * range) + rangeStart; {
var distributedValue = await distribution.GetDistributedValue(cancel);
return (distributedValue * range) + rangeStart;
}
catch (OperationCanceledException)
{
return double.NaN;
}
} }
public async ValueTask<double> NextNumber(IDistribution distribution, CancellationToken cancel = default) => await this.NextNumber(0.0, 1.0, distribution, cancel); 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(); public void StopProducer() => this.producerTokenSource.Cancel();
#endregion #endregion
} }