X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/09f076927cd6294ae6c3db9268496dc3a8fc87f4..d7c8d3913bbc5a2119789f1d82c2f3de3f3fa1f3:/referencni.py

diff --git a/referencni.py b/referencni.py
index d135b3d..e00d0b7 100644
--- 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)