/// <summary>
/// Filters a generator's values based on a predicate. Does not alter the structure of the stream. Instead,
/// it replaces the filtered value with a token, which enables discard counting whilst running the generator.
/// </summary>
/// <param name="gen">The generator to apply the predicate to.</param>
/// <param name="pred">A predicate function that tests each value.</param>
/// <returns>A new generator with the filter applied.</returns>
public static IGen <T> Where <T>(this IGen <T> gen, Func <T, bool> pred)
{
GenInstanceTransformation <T, T> applyPredicateToInstance = (instance) =>
{
var filteredExampleSpace = instance.ExampleSpace.Filter(pred);
if (filteredExampleSpace == null)
{
return(GenIterationFactory.Discard <T>(
instance.ReplayParameters,
instance.NextParameters,
instance.ExampleSpace));
}
return(GenIterationFactory.Instance(
instance.ReplayParameters,
instance.NextParameters,
filteredExampleSpace,
instance.ExampleSpaceHistory));
};
GenStreamTransformation <T, T> resizeAndTerminateAfterConsecutiveDiscards = (stream) =>
{
const int MaxConsecutiveDiscards = 10;
return(stream
.WithConsecutiveDiscardCount(_ => true, iteration => iteration.IsDiscard())
.Select((x) =>
{
var(iteration, consecutiveDiscardCount) = x;
if (consecutiveDiscardCount >= MaxConsecutiveDiscards)
{
var resizedIteration = GenIterationFactory.Discard <T>(
iteration.ReplayParameters,
iteration.NextParameters.With(size: iteration.NextParameters.Size.BigIncrement()),
iteration.Data.Discard !.ExampleSpace);
return (iteration: resizedIteration, consecutiveDiscardCount);
}
public static IGen <T> Cast <T>(this IGen gen) => new FunctionalGen <T>(parameters =>
gen.Advanced
.Run(parameters)
.Select(iteration => iteration.Data.Match(
onInstance: instance => GenIterationFactory.Instance <T>(
iteration.ReplayParameters,
iteration.NextParameters,
instance.ExampleSpace.Cast <T>(),
instance.ExampleSpaceHistory),
onError: error => GenIterationFactory.Error <T>(
iteration.ReplayParameters,
iteration.NextParameters,
error.GenName,
error.Message),
onDiscard: discard => GenIterationFactory.Discard <T>(
iteration.ReplayParameters,
iteration.NextParameters,
discard.ExampleSpace))));
