1 files changed, 32 insertions, 0 deletions
diff --git a/doc/diagrams/engineModule.puml b/doc/diagrams/engineModule.puml
new file mode 100644
index 0000000..c6f3a3e
--- /dev/null
+++ b/doc/diagrams/engineModule.puml
@@ -0,0 +1,32 @@
+@startuml
+
+!include skinparams.puml
+
+package "Engine module" {
+
+    !include ImagoIO.pumlc
+    !include GameEngine.pumlc
+    !include DecisionAlgorithm.pumlc
+    !include MCTS.pumlc
+    !include MCTSNode.pumlc
+    !include Keras.pumlc
+    !include NeuralNetwork.pumlc
+    !include DenseNeuralNetwork.pumlc
+    !include ConvNeuralNetwork.pumlc
+
+    ImagoIO ..> GameEngine
+    GameEngine ..> DecisionAlgorithm
+
+    DecisionAlgorithm <|.. MCTS
+    MCTSNode <. MCTS
+    MCTSNode -> MCTSNode
+    MCTSNode o--> MCTSNode
+
+    DecisionAlgorithm <|.. Keras
+    Keras ..> NeuralNetwork
+    NeuralNetwork <|-- DenseNeuralNetwork
+    NeuralNetwork <|-- ConvNeuralNetwork
+
+}
+
+@enduml