Main.cpp

// skeleton C++ file, you will need to edit this and other files to implement your enigma machine
#include <stdexcept>
#include <iostream>
#include "Plugboard.hpp"
#include "Reflector.hpp"
#include "Rotor.hpp"
#include "Axle.hpp"

using namespace std;

int main(int argc, char **argv) {
    int numRotors = argc - 2;
    //plugboard file is always last argument
    char* pbFilename = argv[numRotors + 1];
    Reflector* reflector = new Reflector();
    Plugboard* plugboard = new Plugboard(pbFilename);
    Rotor** rotors = new Rotor*[numRotors];

    //make array of rotors from rotor files
    for (int i = 0; i < numRotors; i++) {
         rotors[i] = new Rotor (argv[i+1]);
    }
    Axle* axle = new Axle(numRotors, rotors);

    char in = '\0';
    while (cin >> ws >> in) {
        char pb1 = plugboard->encodeOne(in);
        char a1 = axle->encodeOneLR(pb1);
        char r = reflector->encodeOne(a1);
        char a2 = axle->encodeOneRL(r);
        char pb2 = plugboard->encodeOne(a2);
        cout << pb2;
    }

    cout << endl;

    delete reflector;
    delete plugboard;
    delete axle;
    for (int i = numRotors - 1; i >= 0; i--) {
        delete rotors[i];
    }

    return 0;
}