From a39813445873e41ff1227494ef790b4bb20dd8ee Mon Sep 17 00:00:00 2001
From: Antonis Geralis <b3lievem@yandex.com>
Date: Wed, 14 Sep 2022 12:16:30 +0300
Subject: [PATCH] omg it works

---
 drchaos/mutator.nim | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/drchaos/mutator.nim b/drchaos/mutator.nim
index 4f3c3d8..6c356d0 100644
--- a/drchaos/mutator.nim
+++ b/drchaos/mutator.nim
@@ -262,7 +262,7 @@ template sampleAttempt(call: untyped) =
 
 proc sample[T: distinct](x: T; s: var Sampler; r: var Rand; res: var int) =
   when compiles(mutate(x, 0, false, r)):
-    sampleAttempt(attempt(s, r, DefaultMutateWeight, res)) # distinct bool lol
+    sampleAttempt(attempt(s, r, DefaultMutateWeight, res))
   else:
     sample(x.distinctBase, s, r, res)
 
@@ -283,6 +283,9 @@ proc sample[T: SomeNumber](x: T; s: var Sampler; r: var Rand; res: var int) =
 
 proc sample[T](x: seq[T]; s: var Sampler; r: var Rand; res: var int) =
   sampleAttempt(attempt(s, r, DefaultMutateWeight, res))
+  when T is PostProcessTypes:
+    for i in 0..<x.len:
+      sample(x[i], s, r, res)
 
 proc sample(x: string; s: var Sampler; r: var Rand; res: var int) =
   sampleAttempt(attempt(s, r, DefaultMutateWeight, res))
@@ -290,6 +293,14 @@ proc sample(x: string; s: var Sampler; r: var Rand; res: var int) =
 proc sample[T: tuple|object](x: T; s: var Sampler; r: var Rand; res: var int) =
   when compiles(mutate(x, 0, false, r)):
     sampleAttempt(attempt(s, r, DefaultMutateWeight, res))
+    when compiles(for v in mitems(x): discard):
+      when typeof(for v in mitems(x): v) is PostProcessTypes:
+        for v in mitems(x):
+          sample(v, s, r, res)
+    elif compiles(for k, v in mpairs(x): discard):
+      when typeof(for k, v in mpairs(x): v) is PostProcessTypes:
+        for k, v in mpairs(x):
+          sample(v, s, r, res)
   else:
     for v in fields(x):
       sample(v, s, r, res)
@@ -303,6 +314,9 @@ proc sample[T](x: ref T; s: var Sampler; r: var Rand; res: var int) =
 proc sample[S, T](x: array[S, T]; s: var Sampler; r: var Rand; res: var int) =
   when compiles(mutate(x, 0, false, r)):
     sampleAttempt(attempt(s, r, DefaultMutateWeight, res))
+    when T is PostProcessTypes:
+      for i in low(x)..high(x):
+        sample(x[i], s, r, res)
   else:
     for i in low(x)..high(x):
       sample(x[i], s, r, res)
@@ -355,6 +369,9 @@ proc pick[T: SomeNumber](x: var T; sizeIncreaseHint: int; enforceChanges: bool;
 proc pick[T](x: var seq[T]; sizeIncreaseHint: int; enforceChanges: bool; r: var Rand;
     res: var int) =
   pickMutate(mutate(x, sizeIncreaseHint, enforceChanges, r))
+  when T is PostProcessTypes:
+    for i in 0..<x.len:
+      pick(x[i], sizeIncreaseHint, enforceChanges, r, res)
 
 proc pick(x: var string; sizeIncreaseHint: int; enforceChanges: bool; r: var Rand;
     res: var int) =
@@ -372,6 +389,14 @@ proc pick[T: object](x: var T; sizeIncreaseHint: int; enforceChanges: bool;
     r: var Rand; res: var int) =
   when compiles(mutate(x, sizeIncreaseHint, enforceChanges, r)):
     pickMutate(mutate(x, sizeIncreaseHint, enforceChanges, r))
+    when compiles(for v in mitems(x): discard):
+      when typeof(for v in mitems(x): v) is PostProcessTypes:
+        for v in mitems(x):
+          pick(v, sizeIncreaseHint, enforceChanges, r, res)
+    elif compiles(for k, v in mpairs(x): discard):
+      when typeof(for k, v in mpairs(x): v) is PostProcessTypes:
+        for k, v in mpairs(x):
+          sample(v, sizeIncreaseHint, enforceChanges, r, res)
   else:
     template pickImpl(x: untyped) =
       pick(x, sizeIncreaseHint, enforceChanges, r, res)
@@ -388,6 +413,9 @@ proc pick[S, T](x: var array[S, T]; sizeIncreaseHint: int; enforceChanges: bool;
     r: var Rand; res: var int) =
   when compiles(mutate(x, sizeIncreaseHint, enforceChanges, r)):
     pickMutate(mutate(x, sizeIncreaseHint, enforceChanges, r))
+    when T is PostProcessTypes:
+      for i in low(x)..high(x):
+        pick(x[i], sizeIncreaseHint, enforceChanges, r, res)
   else:
     for i in low(x)..high(x):
       pick(x[i], sizeIncreaseHint, enforceChanges, r, res)