From 7e0f6789a8187ca8c691e4211b91eb16e5588203 Mon Sep 17 00:00:00 2001
From: Camil Staps
Date: Thu, 23 Apr 2015 13:23:43 +0200
Subject: Performance

---
 pypride.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'pypride.py')

diff --git a/pypride.py b/pypride.py
index b30b565..c011f74 100644
--- a/pypride.py
+++ b/pypride.py
@@ -239,7 +239,7 @@ def matrixMultiply(matrix, input):
             input as Int
     Output: Int"""
 
-    mult = [ mapXor([c * ((input >> (15 - c_i)) & 0x1) for c_i, c in reversed(list(enumerate(r)))]) for r in matrix ]
+    mult = [ mapXor([1 for c_i, c in enumerate(r) if c == 1 and input & (0x1 << (15 - c_i))]) for r in matrix ]
     return sum([(1 << (15-i)) * v for i,v in enumerate(mult)])
 
 def roundKey(key, i):
-- 
cgit v1.2.3