-
Notifications
You must be signed in to change notification settings - Fork 0
/
Rotor.py
65 lines (38 loc) · 1.36 KB
/
Rotor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from Translator import Translator
class Rotor(Translator):
alphabet_size = 26
dir = "FW"
def __init__(self, num, permutation, turnOverN):
self.num=num
self.permutation = permutation
self.turnOverN = Translator.letterToindex(turnOverN)
self.offset = 0
self.settings = 0
self.reversepermutation = [0]*self.alphabet_size
self.reversetranslation()
def getSettings(self):
return self.settings
def setSettings(self,settings):
self.settings = settings
def setOffset(self, offset):
self.offset= offset
def step(self):
if self.offset == self.alphabet_size:
self.offset = 0
self.offset = self.offset + 1
def turnOver(self):
if self.offset != self.turnOverN:
return False
return True
def changeDir(self):
if self.dir == "FW":
self.dir = "REV"
else:
self.dir = "FW"
def translation(self,letter):
message = self.circularshift(self.offset-self.settings,letter)
if self.dir == "FW":
newmessage = self.permutation[Translator.letterToindex(message)]
else:
newmessage = self.reversepermutation[Translator.letterToindex(message)]
return self.circularshift((self.settings - 1) - (self.offset -1) , newmessage)