From 872cd6c4525fbe9fd190b0bd8a4cc8e5ab148728 Mon Sep 17 00:00:00 2001
From: incredibleLeitman <michael.leithner@gmail.com>
Date: Mon, 30 Nov 2020 18:25:18 +0100
Subject: [PATCH] added rng mode for diagonals

---
 NumberGenerator.h | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/NumberGenerator.h b/NumberGenerator.h
index dca6093..9de5e1f 100644
--- a/NumberGenerator.h
+++ b/NumberGenerator.h
@@ -2,6 +2,7 @@
 #include <cstdlib>  // for srand, rand
 #include <fstream> // ifstream
 #include <ctime> // time
+#include <algorithm>
 
 class NumberGenerator
 {
@@ -20,6 +21,10 @@ public:
         {
             return NumberGenerator::get_circle_numbers(count);
         }
+        else if (mode == 4)
+        {
+            return NumberGenerator::get_diagonal_numbers(count);
+        }
         else throw "unspecified rng Mode"; // TODO: could handle in parse args
     }
 
@@ -114,4 +119,19 @@ public:
         }
         return points;
     }
+
+    static std::list<Point> get_diagonal_numbers(const int valCount)
+    {
+        std::list<Point> points;
+
+        float x = 0;
+        float y = 0;
+        for (int i = 0; i < valCount; ++i)
+        {
+            x = OFFSET + static_cast <float> (rand()) / (static_cast <float> (RAND_MAX / (std::min(WIDTH, HEIGHT) - 2 * OFFSET)));
+            points.push_back(Point(x, HEIGHT - x)); // /
+            //points.push_back(Point(x, HEIGHT - x)); // \
+        }
+        return points;
+    }
 };
\ No newline at end of file