X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/79366cf11cc0d5dd88734b2fad70def327f0c511..e3f91bea0ff143bca677d5d14151990a3f08108e:/jakobsen.py

diff --git a/jakobsen.py b/jakobsen.py
index 9b7c0eb..bf8a0f2 100644
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -53,6 +53,16 @@ def poradi_dle_frekvence(zprava, abc=MABECEDA):
 class reference:
     pass
 
+def kombinace(abeceda):
+    a = 0
+    b = 1
+    while b < len(abeceda):
+        yield (abeceda[a], abeceda[a + b])
+        a += 1
+        if a + b == len(abeceda):
+            a = 0
+            b += 1
+
 def jakobsen(zprava, ref):
     """Pro danou sifrovanou zpravu vrati substitucni slovnik odvozeny s pomoci
     Jakobsenova algoritmu s danou referencni tabulkou."""
@@ -66,7 +76,7 @@ def jakobsen(zprava, ref):
     vzdal_old = vzdal + 1
     while vzdal_old > vzdal:
         vzdal_old = vzdal
-        for (x, y) in combinations(ref.abeceda, 2):
+        for (x, y) in kombinace(ref.poradi):
             slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
             nova_vzdalenost = vzdalenost(
                      nova_tabulka(
@@ -76,9 +86,10 @@ def jakobsen(zprava, ref):
                      ref.abeceda)
             if  nova_vzdalenost < vzdal:
                 vzdal = nova_vzdalenost
+                break
             else:
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
-    return slovnik
+    return slovnik, vzdal
     
 
 def desifruj(zprava, mezery=True):
@@ -89,7 +100,7 @@ def desifruj(zprava, mezery=True):
         soubor = open('bref')
     ref = pickle.load(soubor)
     soubor.close()
-    slovnik = jakobsen(ocesat(zprava, mezery), ref)
+    slovnik, _ = jakobsen(ocesat(zprava, mezery), ref)
     return '\n'.join([''.join(substituce(zprava, slovnik)),
                       ' '.join(ref.abeceda),
                       ' '.join([slovnik[c] for c in ref.abeceda])])