-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathclasses.h
81 lines (71 loc) · 1.62 KB
/
classes.h
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
#pragma once
#include <memory>
#include <string>
#include <vector>
using namespace std;
class Alphabet {
private:
public:
string* letters;
int size;
Alphabet();
Alphabet(int size);
~Alphabet();
void init(/*int size, */string in, int iterator);
int toIndex(string s);
string toAlpha(int i);
void print();
};
class Rotor {
private:
int index;
int size;
int nTurns;
int indexIncrement;
int* forward;
int* turns;
int* cables;
int* tableIncrements;
//vector<int> tableIncrements;
public:
void init(Alphabet& a);
Rotor();
~Rotor();
void setIndex(string s, Alphabet& a);
void setIndex(int ind);
int forwardSub(int i);
int reverseSub(int i);
bool onTurnover();
void increment();
void print(Alphabet& a);
void rotateCables(Alphabet& a);
};
class Reflector {
private:
int size;
int* reflections;
public:
void init(/*string in, */Alphabet& a);
Reflector();
~Reflector();
int reflect(int i);
void print(Alphabet& a);
};
class Enigma {
private:
int numRotors;
Alphabet alphabet;
Reflector reflector;
string enc;
public:
Rotor* rotors;
Enigma();
~Enigma();
void init(int numRotors, Alphabet& a);
void addRotor(Rotor& r, int i);
void setReflector(Reflector& r);
void setString(string enc);
void print();
string encrypt();
string encryptLetter(int letter);
};