From 80bc5ef3c8276a33d0c3b1cb962939e7c252ae7c Mon Sep 17 00:00:00 2001
From: Renaud G <renaud@rolisteam.org>
Date: Wed, 10 Jul 2019 11:55:14 +0200
Subject: Change behaviour of occurence count node. Now it can generate a dice
 result.

---
 node/occurencecountnode.h | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'node/occurencecountnode.h')

diff --git a/node/occurencecountnode.h b/node/occurencecountnode.h
index 125f340..492b295 100644
--- a/node/occurencecountnode.h
+++ b/node/occurencecountnode.h
@@ -24,6 +24,7 @@
 #include "validator.h"
 
 class StringResult;
+class DiceResult;
 class OccurenceCountNode : public ExecutionNode
 {
 public:
@@ -44,11 +45,16 @@ public:
     Validator* getValidator() const;
     void setValidator(Validator* validator);
 
+private:
+    void runForStringResult(const std::map<qint64, qint64>& mapOccurence, QVector<qint64>& vec);
+    void runForDiceResult(const std::map<qint64, qint64>& mapOccurence);
+
 private:
     qint64 m_width= 1;
     qint64 m_height= 0;
     Validator* m_validator= nullptr;
     StringResult* m_stringResult= nullptr;
+    DiceResult* m_diceResult= nullptr;
 };
 
 #endif // OCCURENCECOUNTNODE_H
-- 
cgit v1.2.3-70-g09d2