X-Git-Url: http://git.tomasm.cz/krypto.git/blobdiff_plain/ec94fc01a90b006e61d37d0f152bd60e38d05cd4..1c2f16f2d055b5175052787f27b7d1ff2c21d2a9:/koincidence.py?ds=inline

diff --git a/koincidence.py b/koincidence.py
index 07231b7..efb903c 100644
--- a/koincidence.py
+++ b/koincidence.py
@@ -1,8 +1,10 @@
 from ocesavac import ocesat
 from collections import deque
 
-def index_koincidence(text):
-    t = ocesat(text, False) #zatim bez mezer
+def index_koincidence(text, mezery=False):
+    t = ocesat(text, mezery)
+    if mezery:
+        t = t[1:-1]
     deq = deque(t)
     shod = 0
     for i in range(1, len(t)):
@@ -11,15 +13,15 @@ def index_koincidence(text):
     return float(shod)/(len(t)*(len(t) - 1))
 
 def index_koincidence2(text1, text2):
-    #predpokladam, ze v textech jsou jenom znaky se kterymi pocitam
+    """Pro dva texty vraci index koincidence. Texty nijak neupravuje,
+    ocekava ze se skladaji pouze ze znaku pracovni abecedy."""
     shod = sum (x == y for x, y in zip(text1, text2))
     return float(shod)/min(len(text1), len(text2))
     
 def index_rozkladu(text, deleni):
     if (deleni == 1):
         return index_koincidence(text)
-    t = ocesat(text, False) #zatim bez mezer
-    texty = rozklad(t, deleni)
+    texty = rozklad(ocesat(text, False), deleni)
     return sum([index_koincidence(text) for text in texty])/len(texty)
 
 def tabulka_indexu(text):