From 6414f34a55cc9a78df39d1cae8a5c5c5ffe7736f Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Mon, 17 Jun 2019 14:12:53 +0200
Subject: Fixed diffusion rotation

---
 main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 3882f29..54e42ad 100644
--- a/main.py
+++ b/main.py
@@ -115,9 +115,13 @@ def diffusion(merged_matrix):
     :param merged_matrix: Merged matrix of key and text
     :return: New "diffused" matrix
     """
-    merged_matrix = merged_matrix.copy()
+    # Rotate matrix CCW
+    merged_matrix = [list(element) for element in list(zip(*reversed(merged_matrix)))]
+    # Shift matrix
     for i in range(matrix_size):
         merged_matrix[i] = merged_matrix[i][-i:] + merged_matrix[i][:-i]
+    # Rotate matrix back (CW)
+    merged_matrix = [list(element)[::-1] for element in list(zip(*reversed(merged_matrix)))][::-1]
     pprint(merged_matrix)
     return merged_matrix
 
-- 
cgit v1.2.3