jakobsen priprava na mereni

[krypto.git] / jakobsen.py

diff --git a/jakobsen.py b/jakobsen.py
index aa3cc60..bf8a0f2 100644 (file)
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -76,7 +76,7 @@ def jakobsen(zprava, ref):
     vzdal_old = vzdal + 1
     while vzdal_old > vzdal:
         vzdal_old = vzdal
-        for (x, y) in kombinace(ref.poradi, 2):
+        for (x, y) in kombinace(ref.poradi):
             slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
             nova_vzdalenost = vzdalenost(
                      nova_tabulka(
@@ -89,7 +89,7 @@ def jakobsen(zprava, ref):
                 break
             else:
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
-    return slovnik
+    return slovnik, vzdal
     
 
 def desifruj(zprava, mezery=True):
@@ -100,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])])