jakobsen

Jakasi podivna neprilis funkcni verze.

diff --git a/jakobsen.py b/jakobsen.py
index 34eac4b..799559b 100644 (file)
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -29,6 +29,15 @@ def vzdalenost(tab1, tab2, abc=MABECEDA):
             rozdil += abs(tab1[i][j] - tab2[i][j])
     return rozdil
 
+def vzdalenost2(tab1, tab2):
+    """Vraci soucet strednich kvadratickych odchylek pro dve tabulky cetnosti
+    bigramu nad danou abecedou."""
+    rozdil = 0
+    for i in range(len(tab1)):
+        for j in range(len(tab1)):
+            rozdil += abs(tab1[i][j] - tab2[i][j])
+    return rozdil
+
 def substituce(zprava, slovnik, abc=MABECEDA):
     """Vrati zpravu, ve ktere jsou znaky dane abecedy nahrazenypodle daneho
     slovniku."""
@@ -60,26 +69,42 @@ def prohod(sl_a, sl_b, tabulka):
         tabulka[c][sl_a], tabulka[c][sl_b] = tabulka[c][sl_b], tabulka[c][sl_a]
     return
 
+def prohod2(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 range(len(tabulka)):
+        tabulka[c][sl_a], tabulka[c][sl_b] = tabulka[c][sl_b], tabulka[c][sl_a]
+    return
+
+def dicty_na_listy(tabulka, abeceda):
+    listy = []
+    for a in abeceda:
+        listy.append([tabulka[a][b] for b in abeceda])
+    return listy
+
 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(ref.abeceda, ref.abeceda))
-    tabulka = nova_tabulka(zprava, ref.abeceda)
-    vzdal = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
-
+    slovnik = poradi_dle_frekvence(zprava, ref.abeceda)
+    tab = nova_tabulka(zprava, ref.abeceda)
+    
+    tabulka = dicty_na_listy(tab, slovnik)
+    reftab = dicty_na_listy(ref.tabulka, ref.poradi)    
+ 
+    vzdal = vzdalenost2(tabulka, reftab)
+    
     vzdal_old = vzdal + 1
     while vzdal_old > vzdal:
         vzdal_old = vzdal
-        for (x, y) in combinations(ref.abeceda, 2):
-            prohod(x, y, tabulka)
-            nova_vzdalenost = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
-            if  nova_vzdalenost < vzdal:
+        for (x, y) in combinations(range(len(ref.abeceda)), 2):
+            prohod2(x, y, tabulka)
+            nova_vzdalenost = vzdalenost2(tabulka, reftab)
+            if nova_vzdalenost < vzdal:
                 vzdal = nova_vzdalenost
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
             else:
-                prohod(x, y, tabulka)
-    return slovnik
+                prohod2(x, y, tabulka)
+    return dict(zip(slovnik, ref.poradi))
     
 def desifruj(zprava, mezery=True):
     """Vrati vysledek pokusu dekodovat zpravu Jakobsenovym algoritmem."""
@@ -95,4 +120,7 @@ def desifruj(zprava, mezery=True):
                       ' '.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
+    print desifruj('Mxbhfxn cbfbhinpr, wnx fv f gvzuyr cbenqv.'.upper())
+
+def __test2():
+    print desifruj('Sel pes do lesa a potkal dlazebni kostku. Chtelo by to jeste o neco delsi test, to tedy jo.'.upper())