forked from brownan/Rubiks-Cube-Solver
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cube_convert.py
executable file
·123 lines (102 loc) · 3.85 KB
/
cube_convert.py
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
111
112
113
114
115
116
117
118
119
120
121
122
#!/usr/bin/env python
import sys
"""A quick script that asks the user via the console to input a cube,
and prints out the portable 120 character cube string
"""
template = \
"""
%2s %2s %2s
%2s R %2s
%2s %2s %2s
%2s %2s %2s %2s %2s %2s %2s %2s %2s %2s %2s %2s
%2s B %2s %2s W %2s %2s G %2s %2s Y %2s
%2s %2s %2s %2s %2s %2s %2s %2s %2s %2s %2s %2s
%2s %2s %2s
%2s O %2s
%2s %2s %2s
"""
print template % tuple(range(1,49))
print "Enter the color of each square, all on one line."
print "R - Red, B - BLUE, W - WHITE"
print "G - Green, Y - Yellow, O - Orange"
cc = raw_input(">")
# Prepending a character because I used the wrong indexes below and
# I'm not correcting it
cc = "@" + cc.strip().lower()
if len(cc) != 49:
print "Bad input"
sys.exit(1)
# Now for some painful processing
cubies = []
# the cubie in position 1
cubies.append("nn%s%s%sn" % (cc[29],cc[40],cc[46]))
# 2
cubies.append("nnn%s%sn" % (cc[39],cc[47]))
# 3
cubies.append("nnn%s%s%s" % (cc[38],cc[48],cc[37]))
# 4
cubies.append("nn%s%snn" % (cc[21],cc[28]))
# 5
cubies.append("nnn%sn%s" % (cc[27],cc[26]))
# 6
cubies.append("n%s%s%snn" % (cc[1],cc[9],cc[20]))
# 7
cubies.append("n%sn%snn" % (cc[2],cc[19]))
# 8
cubies.append("n%sn%sn%s" % (cc[3],cc[18],cc[17]))
# 9
cubies.append("nn%sn%sn" % (cc[30],cc[44]))
# 10
cubies.append("nnnn%s%s" % (cc[45],cc[36]))
# 11
cubies.append("n%s%snnn" % (cc[4],cc[10]))
# 12
cubies.append("n%snnn%s" % (cc[5],cc[16]))
# 13
cubies.append("%sn%sn%sn" % (cc[32],cc[31],cc[41]))
# 14
cubies.append("%snnn%sn" % (cc[33],cc[42]))
# 15
cubies.append("%snnn%s%s" % (cc[34],cc[43],cc[35]))
# 16
cubies.append("%sn%snnn" % (cc[23],cc[22]))
# 17
cubies.append("%snnnn%s" % (cc[24],cc[25]))
# 18
cubies.append("%s%s%snnn" % (cc[12],cc[6],cc[11]))
# 19
cubies.append("%s%snnnn" % (cc[13],cc[7]))
# 20
cubies.append("%s%snnn%s" % (cc[14],cc[8],cc[15]))
# Now that we have all the cubies, put them in order
fc = ""
try:
# Pos 1
fc += filter(lambda a: 'y' in a and 'b' in a and 'o' in a, cubies)[0]
fc += filter(lambda a: 'y' in a and 'o' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'y' in a and 'o' in a and 'g' in a, cubies)[0]
fc += filter(lambda a: 'y' in a and 'b' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'y' in a and 'g' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'y' in a and 'b' in a and 'r' in a, cubies)[0]
fc += filter(lambda a: 'y' in a and 'r' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'y' in a and 'r' in a and 'g' in a, cubies)[0]
# Pos 9
fc += filter(lambda a: 'o' in a and 'b' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'o' in a and 'g' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'b' in a and 'r' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'g' in a and 'r' in a and a.count('n')==4, cubies)[0]
# Pos 13
fc += filter(lambda a: 'w' in a and 'b' in a and 'o' in a, cubies)[0]
fc += filter(lambda a: 'w' in a and 'o' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'w' in a and 'o' in a and 'g' in a, cubies)[0]
fc += filter(lambda a: 'w' in a and 'b' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'w' in a and 'g' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'w' in a and 'b' in a and 'r' in a, cubies)[0]
fc += filter(lambda a: 'w' in a and 'r' in a and a.count('n')==4, cubies)[0]
fc += filter(lambda a: 'w' in a and 'r' in a and 'g' in a, cubies)[0]
except IndexError:
sys.exit( "Invalid cube entered: not all cubies accounted for")
if len(fc) != 120:
sys.exit( "An odd error has occured, final code for cube is not correct length: %r" % fc)
print "\nYour cube string:"
print fc