bugfix

[krypto.git] / referencni.py

diff --git a/referencni.py b/referencni.py
index d135b3d..e00d0b7 100644 (file)
--- a/referencni.py
+++ b/referencni.py
@@ -4,6 +4,7 @@ from itertools import combinations
 import pickle
 from ocesavac import ocesat
 import codecs
+from koincidence import index_koincidence
 
 ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
 MABECEDA = ' ABCDEFGHIJKLMNOPQRSTUVWXYZ'
@@ -61,3 +62,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)