posuny.py

   1 import referencni
   2 from referencni import reference
   3 from ocesavac import ocesat
   4
   5 ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
   6
   7 def posun(char):
   8     if 65 <= ord(char) < 90:
   9         return chr(ord(char) + 1)
  10     if ord(char) == 90:
  11         return 'A'
  12     else:
  13         return char
  14
  15 def vsechny(zprava):
  16     posunuty = zprava
  17     analyza = []
  18     for i in range(26):
  19         posunuty = map(posun, posunuty)
  20         analyza.append(''.join(posunuty))
  21     return '\n'.join(analyza)
  22
  23 def eu_vzdalenost(v1, v2):
  24     return sum([(x - y)**2 for (x, y) in zip(v1, v2)])
  25
  26 def nejlepsi_posun(text):
  27     ref = referencni.nacti('bref')
  28     text = ocesat(text, False)
  29     freq = dict()
  30     for char in text:
  31         freq[char] = freq.get(char, 0) + 1
  32     celkem = sum(freq.values())
  33     frekv = [float(freq.get(char, 0))/celkem for char in ABECEDA]
  34     vzdalenost = eu_vzdalenost(frekv, ref.frekvence)
  35     posun = 0
  36     for i in range(1, 26):
  37         frekv = frekv[1:] + [frekv[0]]
  38         v2 = eu_vzdalenost(frekv, ref.frekvence)
  39         if v2 < vzdalenost:
  40             vzdalenost = v2
  41             posun = i
  42     return posun