insanityBlockPuzzleSolver/insanityBlockPuzzleSolver.py at master · alectrolytes/insanityBlockPuzzleSolver · GitHub

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
# Author: Alec Huang
# Establish 4 blocks
# Red = 1, Blue = 10, Green = 100, White = 1000
# Front, Right, Back, Left, Top, Bottom

# green, red, green, blue, white, blue -- first cube
# white, green, red, white, red, blue -- second cube
# red, white, green, green, white, blue -- third cube
# green, red, red, red, white, blue -- fourth cube

blocks = []
print "[ \tFront \tRight  \tBack \tLeft \tTop \tBottom \t]"
firstCube = input("First Cube: ")
secondCube = input("Second Cube: ")
thirdCube = input("Third Cube: ")
fourthCube = input("Fourth Cube: ")

# Convert string to numeric values
def convert(cube):
  newCube = []
  for i in cube:
    if i.lower() == 'red':
      newCube.append(1)
    if i.lower() == 'blue':
      newCube.append(10)
    if i.lower() == 'green':
      newCube.append(100)
    if i.lower() == 'white':
      newCube.append(1000)
  return newCube

blocks.append(convert(firstCube))
blocks.append(convert(secondCube))
blocks.append(convert(thirdCube))
blocks.append(convert(fourthCube))

#print block

# Check correct results
def check(cubelist):
    correct_sides = 0
    for i in range(4):
        if (cubelist[0][i] + cubelist[1][i] + cubelist[2][i] + cubelist[3][i] == 1111):
            correct_sides += 1
    if correct_sides == 4:
        return True
    else:
        return False


# Turn cube
def spin(cube): # Spin cube on Y axis
    newCube = []
    newCube = [cube[1],cube[2],cube[3],cube[0],cube[4],cube[5]]
    return newCube
#print spin(blocks[0])

def rotateZ(cube): # Rotate cube on Z axis
    newCube = [cube[0],cube[4],cube[2],cube[5],cube[3],cube[1]]
    return newCube
#print rotate(blocks[0])

def rotateX(cube):
    newCube = [cube[4],cube[1],cube[5],cube[3],cube[2],cube[0]]
    return newCube
#print rotateX(blocks[0])

def printcube(cubelist):
    print "[ \tFront \tRight  \tBack \tLeft \tTop \tBottom \t]"
    for i in cubelist:
        print "[ ",
        for j in i:
          if j == 1:
            print "\tRed ",
          if j == 10:
            print "\tBlue ",
          if j == 100:
            print "\tGreen ",
          if j == 1000:
            print "\tWhite ",
        print "\t]"

# Solve puzzle
for i in range(24):
    print i
    for j in range(24):
        for k in range(24):
            for l in range(24):
                if l % 4 == 0 and l != 12:
                    blocks[3] = rotateZ(blocks[3])
                elif l % 12 == 0:
                    blocks[3] = rotateX(blocks[3])
                else:
                    blocks[3] = spin(blocks[3])
                if check(blocks):
                    print "SUCCESS!"
                    printcube(blocks)
                    break
            if check(blocks):
                break
            if k % 4 == 0 and k != 12:
                blocks[2] = rotateZ(blocks[2])
            elif k % 12 == 0:
                blocks[2] = rotateX(blocks[2])
            else:
                blocks[2] = spin(blocks[2])
        if check(blocks):
            break
        if j % 4 == 0 and j != 12:
            blocks[1] = rotateZ(blocks[1])
        elif j % 12 == 0:
            blocks[1] = rotateX(blocks[1])
        else:
            blocks[1] = spin(blocks[1])
    if check(blocks):
        break
    if i % 4 == 0 and i != 12:
        blocks[0] = rotateZ(blocks[0])
    elif i % 12 == 0:
        blocks[0] = rotateX(blocks[0])
    else:
        blocks[0] = spin(blocks[0])