-
Notifications
You must be signed in to change notification settings - Fork 0
/
Rotor.cpp
110 lines (97 loc) · 2.65 KB
/
Rotor.cpp
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
//
// Created by Andi Koh on 19/10/2016.
//
#include <fstream>
#include "Rotor.hpp"
#include "utilities.hpp"
Rotor::Rotor(string filename) {
this->filename = filename;
this->values = new int[NUMLETTERS];
this->inverse = new int[NUMLETTERS];
this->numRotations = 0;
generateIntCharMap();
readfile();
generateInverse();
}
void Rotor::readfile() {
ifstream rotorFile;
rotorFile.open(this->filename);
string contents;
if (rotorFile.is_open()) {
getline(rotorFile, contents);
rotorFile.close();
}
//separate values from rotor file based on " "
//save into values array values[0..25]
string token;
string delim = " ";
size_t pos = 0;
for (int i = 0; i < NUMLETTERS; i++) {
pos = contents.find(delim);
if (pos != string::npos) {
token = contents.substr(0, pos);
this->values[i] = atoi(token.c_str());
contents.erase(0, pos + delim.length());
} else {
break;
}
}
//need to manually insert last value
this->values[INPUT_Z] = atoi((contents).c_str());
}
void Rotor::generateInverse() {
//create inverse mapping for when value goes back through rotor
for (int i = 0; i < NUMLETTERS; i++) {
this->inverse[values[i]] = i;
}
}
void Rotor::rotate() {
//save mapping of A
int save = values[INPUT_A];
for (int i = 0; i < INPUT_Z; i++) {
//for A - Y, new mapping is mapping of the next value - 1
int r = values[i + 1]; //calculate new mapping
if (r - 1 < INPUT_A) { //if new mapping is negative, must wrap around
values[i] = INPUT_Z; //-1 mapping equivalent to mapping to Z (25)
} else {
values[i] = r - 1;
}
}
//mapping of Z (25) is mapping of A - 1
//if negative then wrap around to Z
if (save - 1 < INPUT_A) {
values[INPUT_Z] = INPUT_Z;
} else {
values[INPUT_Z] = save - 1;
}
//after every rotation, generate new inverse mapping
generateInverse();
this->numRotations++;
}
char Rotor::encodeOneLR(char in) {
//for first pass through rotors
char out;
int lookup = this->charToInt(in);
int result = this->values[lookup];
out = this->intToChar(result);
return out;
}
char Rotor::encodeOneRL(char in) {
//for return through rotors
char out;
int lookup = this->charToInt(in);
int result = inverse[lookup];
out = this->intToChar(result);
return out;
}
void Rotor::resetRotations() {
this->numRotations = 0;
}
int Rotor::getRotations() {
return this->numRotations;
}
Rotor::Rotor() {
}
char Rotor::encodeOne(char in) {
return '\0';
}