From 700ad655c2013e1a06fa0352d47d3680a6b02b84 Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Sun, 19 Jul 2015 12:41:42 -0700
Subject: [PATCH] Allow turning on/off the rotor advance

---
 Enigma/Machine.swift | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/Enigma/Machine.swift b/Enigma/Machine.swift
index b769246..c6e721a 100644
--- a/Enigma/Machine.swift
+++ b/Enigma/Machine.swift
@@ -14,6 +14,8 @@ class Machine {
     let reflector: Reflector
     let plugboard: Plugboard
 
+    var rotorAdvanceEnabled: Bool = true
+
     init(rotors: [Rotor], reflector: Reflector, plugboard: Plugboard) {
         self.rotors = rotors
         self.reflector = reflector
@@ -21,7 +23,9 @@ class Machine {
     }
 
     func encode(c: Character) throws -> Character {
-        advanceRotors()
+        if rotorAdvanceEnabled {
+            advanceRotors()
+        }
         var output = c
         output = try plugboard.encode(output)
         for rotor in rotors.reverse() {