X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/d4d908e66a7d5208a37e9a91c403e092b0bada22..716701f58ac91d6806c43e77f4070cba571a133c:/jakobsen.py?ds=sidebyside

diff --git a/jakobsen.py b/jakobsen.py
index 1913a27..34eac4b 100644
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -53,34 +53,34 @@ def poradi_dle_frekvence(zprava, abc=MABECEDA):
 class reference:
     pass
 
+def prohod(sl_a, sl_b, tabulka):
+    """Vrati tabulku s prohozenou dvojici sloupcu a radku."""
+    tabulka[sl_a], tabulka[sl_b] = tabulka[sl_b], tabulka[sl_a]
+    for c in tabulka.keys():
+        tabulka[c][sl_a], tabulka[c][sl_b] = tabulka[c][sl_b], tabulka[c][sl_a]
+    return
+
 def jakobsen(zprava, ref):
+    # TODO?zacit se substituci podle poradi frekvenci
     """Pro danou sifrovanou zpravu vrati substitucni slovnik odvozeny s pomoci
     Jakobsenova algoritmu s danou referencni tabulkou."""
-    slovnik = dict(zip(poradi_dle_frekvence(zprava, ref.abeceda), ref.poradi))
-    tabulka = nova_tabulka(
-                  substituce(zprava, slovnik, ref.abeceda),
-                  ref.abeceda
-                  )
+    slovnik = dict(zip(ref.abeceda, ref.abeceda))
+    tabulka = nova_tabulka(zprava, ref.abeceda)
     vzdal = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
 
     vzdal_old = vzdal + 1
     while vzdal_old > vzdal:
         vzdal_old = vzdal
         for (x, y) in combinations(ref.abeceda, 2):
-            slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
-            nova_vzdalenost = vzdalenost(
-                     nova_tabulka(
-                         substituce(zprava, slovnik, ref.abeceda),
-                         ref.abeceda),
-                     ref.tabulka,
-                     ref.abeceda)
+            prohod(x, y, tabulka)
+            nova_vzdalenost = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
             if  nova_vzdalenost < vzdal:
                 vzdal = nova_vzdalenost
-            else:
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
+            else:
+                prohod(x, y, tabulka)
     return slovnik
     
-
 def desifruj(zprava, mezery=True):
     """Vrati vysledek pokusu dekodovat zpravu Jakobsenovym algoritmem."""
     if mezery:
@@ -93,3 +93,6 @@ def desifruj(zprava, mezery=True):
     return '\n'.join([''.join(substituce(zprava, slovnik, ref.abeceda)),
                       ' '.join(ref.abeceda),
                       ' '.join([slovnik[c] for c in ref.abeceda])])
+
+def __test():
+    print desifruj('Sel pes do lesa a potkal dlazebni kostku.'.upper())
\ No newline at end of file