zaklad poctiveho jakobsenova algoritmu

[krypto.git] / jakobsen.py

diff --git a/jakobsen.py b/jakobsen.py
index 40a1d1c..34eac4b 100644 (file)
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -1,12 +1,12 @@
-#!/usr/bin/env python
+"""Modul pro praci s Jakobsenovym algoritmem."""
 
 from itertools import combinations
 import pickle
 from ocesavac import ocesat
-
-MABECEDA = ' ABCDEFGHIJKLMNOPQRSTUVWXYZ'
+from spolecne import MABECEDA
 
 def nova_tabulka(zprava, abc=MABECEDA):
+    """Vraci tabulku cetnosti bigramu ve zprave nad danou abecedou."""
     tabulka = dict()
     for i in abc:
         tabulka[i] = dict()
@@ -21,6 +21,8 @@ def nova_tabulka(zprava, abc=MABECEDA):
     return tabulka
 
 def vzdalenost(tab1, tab2, abc=MABECEDA):
+    """Vraci soucet strednich kvadratickych odchylek pro dve tabulky cetnosti
+    bigramu nad danou abecedou."""
     rozdil = 0
     for i in abc:
         for j in abc:
@@ -28,15 +30,18 @@ def vzdalenost(tab1, tab2, abc=MABECEDA):
     return rozdil
 
 def substituce(zprava, slovnik, abc=MABECEDA):
+    """Vrati zpravu, ve ktere jsou znaky dane abecedy nahrazenypodle daneho
+    slovniku."""
     pole = []
-    for c in zprava:
-        if c in abc:
-            pole.append(slovnik[c])
+    for char in zprava:
+        if char in abc:
+            pole.append(slovnik[char])
         else:
-            pole.append(c)
+            pole.append(char)
     return pole
 
 def poradi_dle_frekvence(zprava, abc=MABECEDA):
+    """Vrati znaky dane abecedy v poradi podle frekvence v dane zprave."""
     freq = dict()
     for char in abc:
         freq[char] = 0
@@ -48,33 +53,46 @@ 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):
-    slovnik = dict(zip(poradi_dle_frekvence(zprava, ref.abeceda), ref.poradi))
-    tabulka = nova_tabulka(substituce(zprava, slovnik, ref.abeceda), ref.abeceda)
+    # 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)
 
     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]
-            v1 = vzdalenost(nova_tabulka(substituce(zprava, slovnik, ref.abeceda), ref.abeceda),
-                            ref.tabulka, ref.abeceda)
-            if  v1 < vzdal:
-                vzdal = v1
-            else:
+            prohod(x, y, tabulka)
+            nova_vzdalenost = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
+            if  nova_vzdalenost < vzdal:
+                vzdal = nova_vzdalenost
                 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:
-        f = open('ref')
+        soubor = open('ref')
     else:
-        f = open('bref')
-    ref = pickle.load(f)
-    f.close()
+        soubor = open('bref')
+    ref = pickle.load(soubor)
+    soubor.close()
     slovnik = jakobsen(ocesat(zprava, mezery), ref)
     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