| 214 |
Kevin |
1 |
import serial
|
|
|
2 |
from CubeRawCommands import *
|
|
|
3 |
|
|
|
4 |
# Constants
|
|
|
5 |
CUBE_LAYER_COUNT = 8
|
|
|
6 |
CUBE_ROW_COUNT = 8
|
|
|
7 |
CUBE_COLUMN_COUNT = 8
|
|
|
8 |
CUBE_ROTATIONS = 7
|
|
|
9 |
GCS_REG_SIZE = 36
|
|
|
10 |
GCS_LAYER_SIZE = GCS_REG_SIZE * CUBE_ROW_COUNT
|
|
|
11 |
|
|
|
12 |
# 3 bytes per LED, 1536 total for the cube
|
|
|
13 |
cube_buffer = bytearray(CUBE_LAYER_COUNT*GCS_LAYER_SIZE)
|
|
|
14 |
rotation_counter = 0
|
|
|
15 |
|
|
|
16 |
# Specify the serial port to connect to
|
|
|
17 |
serial_port = serial.Serial()
|
|
|
18 |
|
|
|
19 |
'''Opens the serial port for sending commands through.'''
|
|
|
20 |
def Cube_Init(port, baudrate):
|
|
|
21 |
serial_port.port = port
|
|
|
22 |
serial_port.baudrate = baudrate
|
|
|
23 |
serial_port.open()
|
|
|
24 |
|
|
|
25 |
'''Clears the local buffer for LED values.'''
|
|
|
26 |
def Cube_Clear():
|
|
|
27 |
for i in range(len(cube_buffer)):
|
|
|
28 |
cube_buffer[i] = 0
|
|
|
29 |
|
|
|
30 |
'''Sets a specific pixel in the local buffer.'''
|
|
|
31 |
def Cube_Set_Pixel(layer, row, column, R, G, B):
|
|
|
32 |
# Set the specified pixel to the given color
|
|
|
33 |
R &= 0x0FFF
|
|
|
34 |
G &= 0x0FFF
|
|
|
35 |
B &= 0x0FFF
|
|
|
36 |
var = row * GCS_REG_SIZE + (column // 2 * 9)
|
|
|
37 |
offset = (layer * GCS_LAYER_SIZE) + var
|
|
|
38 |
if column % 2 == 0:
|
|
|
39 |
cube_buffer[offset+0] = R & 0xFF
|
|
|
40 |
cube_buffer[offset+1] = ((G << 4) | (R >> 8)) & 0xFF
|
|
|
41 |
cube_buffer[offset+2] = G >> 4
|
|
|
42 |
cube_buffer[offset+3] = B & 0xFF
|
|
|
43 |
cube_buffer[offset+4] = ((cube_buffer[offset+4] & 0xF0) | (B >> 8)) & 0xFF
|
|
|
44 |
elif column % 2 == 1:
|
|
|
45 |
cube_buffer[offset+4] = ((cube_buffer[offset+4] & 0x0F) | (R << 4)) & 0xFF
|
|
|
46 |
cube_buffer[offset+5] = R >> 4
|
|
|
47 |
cube_buffer[offset+6] = G & 0xFF
|
|
|
48 |
cube_buffer[offset+7] = ((B << 4) | (G >> 8)) & 0xFF
|
|
|
49 |
cube_buffer[offset+8] = B >> 4
|
|
|
50 |
|
|
|
51 |
'''Retreives a specific pixel in the local buffer.'''
|
|
|
52 |
def Cube_Get_Pixel(layer, row, column):
|
|
|
53 |
# Get and return the color for the specified pixel
|
|
|
54 |
R = G = B = 0
|
|
|
55 |
var = row * GCS_REG_SIZE + (column // 2 * 9)
|
|
|
56 |
offset = (layer * GCS_LAYER_SIZE) + var
|
|
|
57 |
if column % 2 == 0:
|
|
|
58 |
R = cube_buffer[offset+0] | ((cube_buffer[offset+1] & 0x0F) << 8);
|
|
|
59 |
G = (cube_buffer[offset+1] >> 4) | (cube_buffer[offset+2] << 4);
|
|
|
60 |
B = cube_buffer[offset+3] | ((cube_buffer[offset+4] & 0x0F) << 8);
|
|
|
61 |
elif column % 2 == 1:
|
|
|
62 |
R = (cube_buffer[offset+4] >> 4) | (cube_buffer[offset+5] << 4);
|
|
|
63 |
G = cube_buffer[offset+6] | ((cube_buffer[offset+7] & 0x0F) << 8);
|
|
|
64 |
B = (cube_buffer[offset+7] >> 4) | (cube_buffer[offset+8] << 4);
|
|
|
65 |
return [R,G,B]
|
|
|
66 |
|
|
|
67 |
'''Moves a value from one pixel to another in the local buffer.'''
|
|
|
68 |
def Cube_Move_Pixel(layer_1, row_1, column_1, layer_2, row_2, column_2):
|
|
|
69 |
old = Cube_Get_Pixel(layer1, row_1, column_1)
|
|
|
70 |
Cube_Set_Pixel(layer_2, row_2, column_2, old[0], old[1], old[2])
|
|
|
71 |
|
|
|
72 |
'''Rotates the specified shell in the local buffer.'''
|
|
|
73 |
def Cube_Rotate_Shell(shell, direction):
|
|
|
74 |
for layer in range(CUBE_LAYER_COUNT):
|
|
|
75 |
if direction == 1:
|
|
|
76 |
if shell == 0:
|
|
|
77 |
# Rotate outermost layer
|
|
|
78 |
old = Cube_Get_Pixel(layer, 0, 0);
|
|
|
79 |
Cube_Move_Pixel(layer, 0, 1, layer, 0, 0);
|
|
|
80 |
Cube_Move_Pixel(layer, 0, 2, layer, 0, 1);
|
|
|
81 |
Cube_Move_Pixel(layer, 0, 3, layer, 0, 2);
|
|
|
82 |
Cube_Move_Pixel(layer, 0, 4, layer, 0, 3);
|
|
|
83 |
Cube_Move_Pixel(layer, 0, 5, layer, 0, 4);
|
|
|
84 |
Cube_Move_Pixel(layer, 0, 6, layer, 0, 5);
|
|
|
85 |
Cube_Move_Pixel(layer, 0, 7, layer, 0, 6);
|
|
|
86 |
Cube_Move_Pixel(layer, 1, 7, layer, 0, 7);
|
|
|
87 |
Cube_Move_Pixel(layer, 2, 7, layer, 1, 7);
|
|
|
88 |
Cube_Move_Pixel(layer, 3, 7, layer, 2, 7);
|
|
|
89 |
Cube_Move_Pixel(layer, 4, 7, layer, 3, 7);
|
|
|
90 |
Cube_Move_Pixel(layer, 5, 7, layer, 4, 7);
|
|
|
91 |
Cube_Move_Pixel(layer, 6, 7, layer, 5, 7);
|
|
|
92 |
Cube_Move_Pixel(layer, 7, 7, layer, 6, 7);
|
|
|
93 |
Cube_Move_Pixel(layer, 7, 6, layer, 7, 7);
|
|
|
94 |
Cube_Move_Pixel(layer, 7, 5, layer, 7, 6);
|
|
|
95 |
Cube_Move_Pixel(layer, 7, 4, layer, 7, 5);
|
|
|
96 |
Cube_Move_Pixel(layer, 7, 3, layer, 7, 4);
|
|
|
97 |
Cube_Move_Pixel(layer, 7, 2, layer, 7, 3);
|
|
|
98 |
Cube_Move_Pixel(layer, 7, 1, layer, 7, 2);
|
|
|
99 |
Cube_Move_Pixel(layer, 7, 0, layer, 7, 1);
|
|
|
100 |
Cube_Move_Pixel(layer, 6, 0, layer, 7, 0);
|
|
|
101 |
Cube_Move_Pixel(layer, 5, 0, layer, 6, 0);
|
|
|
102 |
Cube_Move_Pixel(layer, 4, 0, layer, 5, 0);
|
|
|
103 |
Cube_Move_Pixel(layer, 3, 0, layer, 4, 0);
|
|
|
104 |
Cube_Move_Pixel(layer, 2, 0, layer, 3, 0);
|
|
|
105 |
Cube_Move_Pixel(layer, 1, 0, layer, 2, 0);
|
|
|
106 |
Cube_Set_Pixel(layer, 1, 0, old[0], old[1], old[2]);
|
|
|
107 |
elif shell == 1:
|
|
|
108 |
# Rotate second to outermost layer
|
|
|
109 |
old = Cube_Get_Pixel(layer, 1, 1);
|
|
|
110 |
Cube_Move_Pixel(layer, 1, 2, layer, 1, 1);
|
|
|
111 |
Cube_Move_Pixel(layer, 1, 3, layer, 1, 2);
|
|
|
112 |
Cube_Move_Pixel(layer, 1, 4, layer, 1, 3);
|
|
|
113 |
Cube_Move_Pixel(layer, 1, 5, layer, 1, 4);
|
|
|
114 |
Cube_Move_Pixel(layer, 1, 6, layer, 1, 5);
|
|
|
115 |
Cube_Move_Pixel(layer, 2, 6, layer, 1, 6);
|
|
|
116 |
Cube_Move_Pixel(layer, 3, 6, layer, 2, 6);
|
|
|
117 |
Cube_Move_Pixel(layer, 4, 6, layer, 3, 6);
|
|
|
118 |
Cube_Move_Pixel(layer, 5, 6, layer, 4, 6);
|
|
|
119 |
Cube_Move_Pixel(layer, 6, 6, layer, 5, 6);
|
|
|
120 |
Cube_Move_Pixel(layer, 6, 5, layer, 6, 6);
|
|
|
121 |
Cube_Move_Pixel(layer, 6, 4, layer, 6, 5);
|
|
|
122 |
Cube_Move_Pixel(layer, 6, 3, layer, 6, 4);
|
|
|
123 |
Cube_Move_Pixel(layer, 6, 2, layer, 6, 3);
|
|
|
124 |
Cube_Move_Pixel(layer, 6, 1, layer, 6, 2);
|
|
|
125 |
Cube_Move_Pixel(layer, 5, 1, layer, 6, 1);
|
|
|
126 |
Cube_Move_Pixel(layer, 4, 1, layer, 5, 1);
|
|
|
127 |
Cube_Move_Pixel(layer, 3, 1, layer, 4, 1);
|
|
|
128 |
Cube_Move_Pixel(layer, 2, 1, layer, 3, 1);
|
|
|
129 |
Cube_Set_Pixel(layer, 2, 1, old[0], old[1], old[2]);
|
|
|
130 |
elif shell == 2:
|
|
|
131 |
# Rotate second to innermost layer
|
|
|
132 |
old = Cube_Get_Pixel(layer, 2, 2);
|
|
|
133 |
Cube_Move_Pixel(layer, 2, 3, layer, 2, 2);
|
|
|
134 |
Cube_Move_Pixel(layer, 2, 4, layer, 2, 3);
|
|
|
135 |
Cube_Move_Pixel(layer, 2, 5, layer, 2, 4);
|
|
|
136 |
Cube_Move_Pixel(layer, 3, 5, layer, 2, 5);
|
|
|
137 |
Cube_Move_Pixel(layer, 4, 5, layer, 3, 5);
|
|
|
138 |
Cube_Move_Pixel(layer, 5, 5, layer, 4, 5);
|
|
|
139 |
Cube_Move_Pixel(layer, 5, 4, layer, 5, 5);
|
|
|
140 |
Cube_Move_Pixel(layer, 5, 3, layer, 5, 4);
|
|
|
141 |
Cube_Move_Pixel(layer, 5, 2, layer, 5, 3);
|
|
|
142 |
Cube_Move_Pixel(layer, 4, 2, layer, 5, 2);
|
|
|
143 |
Cube_Move_Pixel(layer, 3, 2, layer, 4, 2);
|
|
|
144 |
Cube_Set_Pixel(layer, 3, 2, old[0], old[1], old[2]);
|
|
|
145 |
elif shell == 3:
|
|
|
146 |
# Rotate innermost layer
|
|
|
147 |
old = Cube_Get_Pixel(layer, 3, 3);
|
|
|
148 |
Cube_Move_Pixel(layer, 3, 4, layer, 3, 3);
|
|
|
149 |
Cube_Move_Pixel(layer, 4, 4, layer, 3, 4);
|
|
|
150 |
Cube_Move_Pixel(layer, 4, 3, layer, 4, 4);
|
|
|
151 |
Cube_Set_Pixel(layer, 4, 3, old[0], old[1], old[2]);
|
|
|
152 |
else:
|
|
|
153 |
if shell == 0:
|
|
|
154 |
# Rotate outermost layer
|
|
|
155 |
old = Cube_Get_Pixel(layer, 0, 0);
|
|
|
156 |
Cube_Move_Pixel(layer, 1, 0, layer, 0, 0);
|
|
|
157 |
Cube_Move_Pixel(layer, 2, 0, layer, 1, 0);
|
|
|
158 |
Cube_Move_Pixel(layer, 3, 0, layer, 2, 0);
|
|
|
159 |
Cube_Move_Pixel(layer, 4, 0, layer, 3, 0);
|
|
|
160 |
Cube_Move_Pixel(layer, 5, 0, layer, 4, 0);
|
|
|
161 |
Cube_Move_Pixel(layer, 6, 0, layer, 5, 0);
|
|
|
162 |
Cube_Move_Pixel(layer, 7, 0, layer, 6, 0);
|
|
|
163 |
Cube_Move_Pixel(layer, 7, 1, layer, 7, 0);
|
|
|
164 |
Cube_Move_Pixel(layer, 7, 2, layer, 7, 1);
|
|
|
165 |
Cube_Move_Pixel(layer, 7, 3, layer, 7, 2);
|
|
|
166 |
Cube_Move_Pixel(layer, 7, 4, layer, 7, 3);
|
|
|
167 |
Cube_Move_Pixel(layer, 7, 5, layer, 7, 4);
|
|
|
168 |
Cube_Move_Pixel(layer, 7, 6, layer, 7, 5);
|
|
|
169 |
Cube_Move_Pixel(layer, 7, 7, layer, 7, 6);
|
|
|
170 |
Cube_Move_Pixel(layer, 6, 7, layer, 7, 7);
|
|
|
171 |
Cube_Move_Pixel(layer, 5, 7, layer, 6, 7);
|
|
|
172 |
Cube_Move_Pixel(layer, 4, 7, layer, 5, 7);
|
|
|
173 |
Cube_Move_Pixel(layer, 3, 7, layer, 4, 7);
|
|
|
174 |
Cube_Move_Pixel(layer, 2, 7, layer, 3, 7);
|
|
|
175 |
Cube_Move_Pixel(layer, 1, 7, layer, 2, 7);
|
|
|
176 |
Cube_Move_Pixel(layer, 0, 7, layer, 1, 7);
|
|
|
177 |
Cube_Move_Pixel(layer, 0, 6, layer, 0, 7);
|
|
|
178 |
Cube_Move_Pixel(layer, 0, 5, layer, 0, 6);
|
|
|
179 |
Cube_Move_Pixel(layer, 0, 4, layer, 0, 5);
|
|
|
180 |
Cube_Move_Pixel(layer, 0, 3, layer, 0, 4);
|
|
|
181 |
Cube_Move_Pixel(layer, 0, 2, layer, 0, 3);
|
|
|
182 |
Cube_Move_Pixel(layer, 0, 1, layer, 0, 2);
|
|
|
183 |
Cube_Set_Pixel(layer, 0, 1, old[0], old[1], old[2]);
|
|
|
184 |
if shell == 1:
|
|
|
185 |
# Rotate second to outermost layer
|
|
|
186 |
old = Cube_Get_Pixel(layer, 1, 1);
|
|
|
187 |
Cube_Move_Pixel(layer, 2, 1, layer, 1, 1);
|
|
|
188 |
Cube_Move_Pixel(layer, 3, 1, layer, 2, 1);
|
|
|
189 |
Cube_Move_Pixel(layer, 4, 1, layer, 3, 1);
|
|
|
190 |
Cube_Move_Pixel(layer, 5, 1, layer, 4, 1);
|
|
|
191 |
Cube_Move_Pixel(layer, 6, 1, layer, 5, 1);
|
|
|
192 |
Cube_Move_Pixel(layer, 6, 2, layer, 6, 1);
|
|
|
193 |
Cube_Move_Pixel(layer, 6, 3, layer, 6, 2);
|
|
|
194 |
Cube_Move_Pixel(layer, 6, 4, layer, 6, 3);
|
|
|
195 |
Cube_Move_Pixel(layer, 6, 5, layer, 6, 4);
|
|
|
196 |
Cube_Move_Pixel(layer, 6, 6, layer, 6, 5);
|
|
|
197 |
Cube_Move_Pixel(layer, 5, 6, layer, 6, 6);
|
|
|
198 |
Cube_Move_Pixel(layer, 4, 6, layer, 5, 6);
|
|
|
199 |
Cube_Move_Pixel(layer, 3, 6, layer, 4, 6);
|
|
|
200 |
Cube_Move_Pixel(layer, 2, 6, layer, 3, 6);
|
|
|
201 |
Cube_Move_Pixel(layer, 1, 6, layer, 2, 6);
|
|
|
202 |
Cube_Move_Pixel(layer, 1, 5, layer, 1, 6);
|
|
|
203 |
Cube_Move_Pixel(layer, 1, 4, layer, 1, 5);
|
|
|
204 |
Cube_Move_Pixel(layer, 1, 3, layer, 1, 4);
|
|
|
205 |
Cube_Move_Pixel(layer, 1, 2, layer, 1, 3);
|
|
|
206 |
Cube_Set_Pixel(layer, 1, 2, old[0], old[1], old[2]);
|
|
|
207 |
if shell == 2:
|
|
|
208 |
# Rotate second to innermost layer
|
|
|
209 |
old = Cube_Get_Pixel(layer, 2, 2);
|
|
|
210 |
Cube_Move_Pixel(layer, 3, 2, layer, 2, 2);
|
|
|
211 |
Cube_Move_Pixel(layer, 4, 2, layer, 3, 2);
|
|
|
212 |
Cube_Move_Pixel(layer, 5, 2, layer, 4, 2);
|
|
|
213 |
Cube_Move_Pixel(layer, 5, 3, layer, 5, 2);
|
|
|
214 |
Cube_Move_Pixel(layer, 5, 4, layer, 5, 3);
|
|
|
215 |
Cube_Move_Pixel(layer, 5, 5, layer, 5, 4);
|
|
|
216 |
Cube_Move_Pixel(layer, 4, 5, layer, 5, 5);
|
|
|
217 |
Cube_Move_Pixel(layer, 3, 5, layer, 4, 5);
|
|
|
218 |
Cube_Move_Pixel(layer, 2, 5, layer, 3, 5);
|
|
|
219 |
Cube_Move_Pixel(layer, 2, 4, layer, 2, 5);
|
|
|
220 |
Cube_Move_Pixel(layer, 2, 3, layer, 2, 4);
|
|
|
221 |
Cube_Set_Pixel(layer, 2, 3, old[0], old[1], old[2]);
|
|
|
222 |
if shell == 3:
|
|
|
223 |
# Rotate innermost layer
|
|
|
224 |
old = Cube_Get_Pixel(layer, 3, 3);
|
|
|
225 |
Cube_Move_Pixel(layer, 4, 3, layer, 3, 3);
|
|
|
226 |
Cube_Move_Pixel(layer, 4, 4, layer, 4, 3);
|
|
|
227 |
Cube_Move_Pixel(layer, 3, 4, layer, 4, 4);
|
|
|
228 |
Cube_Set_Pixel(layer, 3, 4, old[0], old[1], old[2]);
|
|
|
229 |
|
|
|
230 |
'''Rotates the entire cube in the local buffer.'''
|
|
|
231 |
def Cube_Rotate(direction):
|
|
|
232 |
# Rotate outermost layer
|
|
|
233 |
Cube_Rotate_Shell(0, direction);
|
|
|
234 |
# Rotate second to outermost layer
|
|
|
235 |
if ((rotation_counter != 1) and (rotation_counter != 5)):
|
|
|
236 |
Cube_Rotate_Shell(1, direction);
|
|
|
237 |
# Rotate second to innermost layer
|
|
|
238 |
if ((rotation_counter != 0) and (rotation_counter != 2) and (rotation_counter != 4) and (rotation_counter != 6)):
|
|
|
239 |
Cube_Rotate_Shell(2, direction);
|
|
|
240 |
# Rotate innermost layer
|
|
|
241 |
if ((rotation_counter == 3) or (rotation_counter == 7)):
|
|
|
242 |
Cube_Rotate_Shell(3, direction);
|
|
|
243 |
|
|
|
244 |
if (direction == 0):
|
|
|
245 |
if rotation_counter == CUBE_ROTATIONS - 1:
|
|
|
246 |
rotation_counter = 0
|
|
|
247 |
else:
|
|
|
248 |
rotation_counter = rotation_counter + 1
|
|
|
249 |
else:
|
|
|
250 |
if rotation_counter == 0:
|
|
|
251 |
rotation_counter = CUBE_ROTATIONS - 1
|
|
|
252 |
else:
|
|
|
253 |
rotation_counter = rotation_counter - 1
|
|
|
254 |
|
|
|
255 |
'''Write the local buffer to the cube.'''
|
|
|
256 |
def Cube_Update():
|
| 216 |
Kevin |
257 |
serial_port.write(CMD_Set_All(cube_buffer))
|