bugfix
[krypto.git] / koincidence.py
1 from ocesavac import ocesat
2 from collections import deque
3
4 def index_koincidence(text):
5     t = ocesat(text, False) #zatim bez mezer
6     deq = deque(t)
7     shod = 0
8     for i in range(1, len(t)):
9         deq.rotate(1)
10         shod += sum (x == y for x, y in zip(deq, t))
11     return float(shod)/(len(t)*(len(t) - 1))
12
13 def index_koincidence2(text1, text2):
14     #predpokladam, ze v textech jsou jenom znaky se kterymi pocitam
15     shod = sum (x == y for x, y in zip(text1, text2))
16     return float(shod)/min(len(text1), len(text2))
17     
18 def index_rozkladu(text, deleni):
19     if (deleni == 1):
20         return index_koincidence(text)
21     t = ocesat(text, False) #zatim bez mezer
22     texty = rozklad(t, deleni)
23     return sum([index_koincidence(text) for text in texty])/len(texty)
24
25 def tabulka_indexu(text):
26     return '\n'.join(['{1:>2} {0:.3%}'.format(
27         index_rozkladu(text, d), d) for d in range(1, 11)])
28
29 def rozklad(text, deleni):
30     texty = []
31     for i in range(deleni):
32         texty.append(''.join(
33             [text[j] for j in range(len(text)) if j % deleni == i]))
34     return texty