vigenere

[krypto.git] / referencni.py

diff --git a/referencni.py b/referencni.py
index d135b3d..06bfeec 100644 (file)
--- a/referencni.py
+++ b/referencni.py
@@ -4,9 +4,8 @@ from itertools import combinations
 import pickle
 from ocesavac import ocesat
 import codecs
-
-ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
-MABECEDA = ' ABCDEFGHIJKLMNOPQRSTUVWXYZ'
+from koincidence import index_koincidence
+from spolecne import ABECEDA, MABECEDA
 
 def nova_tabulka(tabulka, abc=MABECEDA):
     celkem = sum(sum(tab2.values()) for tab2 in tabulka.values())
@@ -61,3 +60,12 @@ def uloz(co, kam):
 
 def nacti(odkud):
     return pickle.load(open(odkud))
+
+def prumerny_index(tref):
+    suma = 0
+    for soubor in ref.soubory:
+        f = codecs.open(soubor, encoding='UTF-8')
+        text = ocesat(f.read(), tref.mezery)
+        f.close()
+        suma += index_koincidence2(text[:len(text)/2], text[len(text)/2:])
+    return float(suma)/len(tref.soubory)