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