""" batons.py
"""

from permutations import Permutation,LOWER

def rotor(strng):
    return Permutation(tuple(strng.lower()))

# The rotors
I     = rotor("EKMFLGDQVZNTOWYHXUSPAIBRCJ")
II    = rotor("AJDKSIRUXBLHWTMCQGZNPYFVOE")
III   = rotor("BDFHJLCPRTXVZNYEIWGAKMUSQO")
IV    = rotor("ESOVPZJAYQUIRHXLNFTGKDCMWB")
V     = rotor("VZBRGITYUPSDNHLXAWMJQOFECK")
VI    = rotor("JPGVOUMFYQBENHZRDKASXLICTW")
VII   = rotor("NZJHGRCXMYSWBOUFAIVLPEKQDT")
VIII  = rotor("FKQHTLXOCBJSPDZRAMEWNIUYGV")

beta  = rotor("LEYJVCNIXWPBQMDRTAKZGFUHOS")
gamma = rotor("FSOKANUERHMBTIYCWLQPZXVGJD")

# The reflectors
B     = rotor("YRUHQSLDPXNGOKMIEBFZCWVJAT")
C     = rotor("FVPJIAOYEDRZXWGCTKUQSBNMHL")
thinB = rotor("ENKQAUYWJICOPBLMDXZVFTHRGS")
thinC = rotor("RDOBJNTKVEHMLFCWZAXGYIPSUQ")

# The cyclic permutation
rho   = rotor("BCDEFGHIJKLMNOPQRSTUVWXYZA")

# The example
P     = rotor("LWFTBAXJDSCKPRZQYOEHUGMIVN")