Blame | Last modification | View Log | RSS feed
import serialfrom CubeRawCommands import *# ConstantsCUBE_LAYER_COUNT = 8CUBE_ROW_COUNT = 8CUBE_COLUMN_COUNT = 8CUBE_ROTATIONS = 7GCS_REG_SIZE = 36GCS_LAYER_SIZE = GCS_REG_SIZE * CUBE_ROW_COUNT# 3 bytes per LED, 1536 total for the cubecube_buffer = bytearray(CUBE_LAYER_COUNT*GCS_LAYER_SIZE)rotation_counter = 0# Specify the serial port to connect toserial_port = serial.Serial()'''Opens the serial port for sending commands through.'''def Cube_Init(port, baudrate):serial_port.port = portserial_port.baudrate = baudrateserial_port.open()'''Clears the local buffer for LED values.'''def Cube_Clear():for i in range(len(cube_buffer)):cube_buffer[i] = 0'''Sets a specific pixel in the local buffer.'''def Cube_Set_Pixel(layer, row, column, R, G, B):# Set the specified pixel to the given colorR &= 0x0FFFG &= 0x0FFFB &= 0x0FFFvar = row * GCS_REG_SIZE + (column // 2 * 9)offset = (layer * GCS_LAYER_SIZE) + varif column % 2 == 0:cube_buffer[offset+0] = R & 0xFFcube_buffer[offset+1] = ((G << 4) | (R >> 8)) & 0xFFcube_buffer[offset+2] = G >> 4cube_buffer[offset+3] = B & 0xFFcube_buffer[offset+4] = ((cube_buffer[offset+4] & 0xF0) | (B >> 8)) & 0xFFelif column % 2 == 1:cube_buffer[offset+4] = ((cube_buffer[offset+4] & 0x0F) | (R << 4)) & 0xFFcube_buffer[offset+5] = R >> 4cube_buffer[offset+6] = G & 0xFFcube_buffer[offset+7] = ((B << 4) | (G >> 8)) & 0xFFcube_buffer[offset+8] = B >> 4'''Retreives a specific pixel in the local buffer.'''def Cube_Get_Pixel(layer, row, column):# Get and return the color for the specified pixelR = G = B = 0var = row * GCS_REG_SIZE + (column // 2 * 9)offset = (layer * GCS_LAYER_SIZE) + varif column % 2 == 0:R = cube_buffer[offset+0] | ((cube_buffer[offset+1] & 0x0F) << 8);G = (cube_buffer[offset+1] >> 4) | (cube_buffer[offset+2] << 4);B = cube_buffer[offset+3] | ((cube_buffer[offset+4] & 0x0F) << 8);elif column % 2 == 1:R = (cube_buffer[offset+4] >> 4) | (cube_buffer[offset+5] << 4);G = cube_buffer[offset+6] | ((cube_buffer[offset+7] & 0x0F) << 8);B = (cube_buffer[offset+7] >> 4) | (cube_buffer[offset+8] << 4);return [R,G,B]'''Moves a value from one pixel to another in the local buffer.'''def Cube_Move_Pixel(layer_1, row_1, column_1, layer_2, row_2, column_2):old = Cube_Get_Pixel(layer1, row_1, column_1)Cube_Set_Pixel(layer_2, row_2, column_2, old[0], old[1], old[2])'''Rotates the specified shell in the local buffer.'''def Cube_Rotate_Shell(shell, direction):for layer in range(CUBE_LAYER_COUNT):if direction == 1:if shell == 0:# Rotate outermost layerold = Cube_Get_Pixel(layer, 0, 0);Cube_Move_Pixel(layer, 0, 1, layer, 0, 0);Cube_Move_Pixel(layer, 0, 2, layer, 0, 1);Cube_Move_Pixel(layer, 0, 3, layer, 0, 2);Cube_Move_Pixel(layer, 0, 4, layer, 0, 3);Cube_Move_Pixel(layer, 0, 5, layer, 0, 4);Cube_Move_Pixel(layer, 0, 6, layer, 0, 5);Cube_Move_Pixel(layer, 0, 7, layer, 0, 6);Cube_Move_Pixel(layer, 1, 7, layer, 0, 7);Cube_Move_Pixel(layer, 2, 7, layer, 1, 7);Cube_Move_Pixel(layer, 3, 7, layer, 2, 7);Cube_Move_Pixel(layer, 4, 7, layer, 3, 7);Cube_Move_Pixel(layer, 5, 7, layer, 4, 7);Cube_Move_Pixel(layer, 6, 7, layer, 5, 7);Cube_Move_Pixel(layer, 7, 7, layer, 6, 7);Cube_Move_Pixel(layer, 7, 6, layer, 7, 7);Cube_Move_Pixel(layer, 7, 5, layer, 7, 6);Cube_Move_Pixel(layer, 7, 4, layer, 7, 5);Cube_Move_Pixel(layer, 7, 3, layer, 7, 4);Cube_Move_Pixel(layer, 7, 2, layer, 7, 3);Cube_Move_Pixel(layer, 7, 1, layer, 7, 2);Cube_Move_Pixel(layer, 7, 0, layer, 7, 1);Cube_Move_Pixel(layer, 6, 0, layer, 7, 0);Cube_Move_Pixel(layer, 5, 0, layer, 6, 0);Cube_Move_Pixel(layer, 4, 0, layer, 5, 0);Cube_Move_Pixel(layer, 3, 0, layer, 4, 0);Cube_Move_Pixel(layer, 2, 0, layer, 3, 0);Cube_Move_Pixel(layer, 1, 0, layer, 2, 0);Cube_Set_Pixel(layer, 1, 0, old[0], old[1], old[2]);elif shell == 1:# Rotate second to outermost layerold = Cube_Get_Pixel(layer, 1, 1);Cube_Move_Pixel(layer, 1, 2, layer, 1, 1);Cube_Move_Pixel(layer, 1, 3, layer, 1, 2);Cube_Move_Pixel(layer, 1, 4, layer, 1, 3);Cube_Move_Pixel(layer, 1, 5, layer, 1, 4);Cube_Move_Pixel(layer, 1, 6, layer, 1, 5);Cube_Move_Pixel(layer, 2, 6, layer, 1, 6);Cube_Move_Pixel(layer, 3, 6, layer, 2, 6);Cube_Move_Pixel(layer, 4, 6, layer, 3, 6);Cube_Move_Pixel(layer, 5, 6, layer, 4, 6);Cube_Move_Pixel(layer, 6, 6, layer, 5, 6);Cube_Move_Pixel(layer, 6, 5, layer, 6, 6);Cube_Move_Pixel(layer, 6, 4, layer, 6, 5);Cube_Move_Pixel(layer, 6, 3, layer, 6, 4);Cube_Move_Pixel(layer, 6, 2, layer, 6, 3);Cube_Move_Pixel(layer, 6, 1, layer, 6, 2);Cube_Move_Pixel(layer, 5, 1, layer, 6, 1);Cube_Move_Pixel(layer, 4, 1, layer, 5, 1);Cube_Move_Pixel(layer, 3, 1, layer, 4, 1);Cube_Move_Pixel(layer, 2, 1, layer, 3, 1);Cube_Set_Pixel(layer, 2, 1, old[0], old[1], old[2]);elif shell == 2:# Rotate second to innermost layerold = Cube_Get_Pixel(layer, 2, 2);Cube_Move_Pixel(layer, 2, 3, layer, 2, 2);Cube_Move_Pixel(layer, 2, 4, layer, 2, 3);Cube_Move_Pixel(layer, 2, 5, layer, 2, 4);Cube_Move_Pixel(layer, 3, 5, layer, 2, 5);Cube_Move_Pixel(layer, 4, 5, layer, 3, 5);Cube_Move_Pixel(layer, 5, 5, layer, 4, 5);Cube_Move_Pixel(layer, 5, 4, layer, 5, 5);Cube_Move_Pixel(layer, 5, 3, layer, 5, 4);Cube_Move_Pixel(layer, 5, 2, layer, 5, 3);Cube_Move_Pixel(layer, 4, 2, layer, 5, 2);Cube_Move_Pixel(layer, 3, 2, layer, 4, 2);Cube_Set_Pixel(layer, 3, 2, old[0], old[1], old[2]);elif shell == 3:# Rotate innermost layerold = Cube_Get_Pixel(layer, 3, 3);Cube_Move_Pixel(layer, 3, 4, layer, 3, 3);Cube_Move_Pixel(layer, 4, 4, layer, 3, 4);Cube_Move_Pixel(layer, 4, 3, layer, 4, 4);Cube_Set_Pixel(layer, 4, 3, old[0], old[1], old[2]);else:if shell == 0:# Rotate outermost layerold = Cube_Get_Pixel(layer, 0, 0);Cube_Move_Pixel(layer, 1, 0, layer, 0, 0);Cube_Move_Pixel(layer, 2, 0, layer, 1, 0);Cube_Move_Pixel(layer, 3, 0, layer, 2, 0);Cube_Move_Pixel(layer, 4, 0, layer, 3, 0);Cube_Move_Pixel(layer, 5, 0, layer, 4, 0);Cube_Move_Pixel(layer, 6, 0, layer, 5, 0);Cube_Move_Pixel(layer, 7, 0, layer, 6, 0);Cube_Move_Pixel(layer, 7, 1, layer, 7, 0);Cube_Move_Pixel(layer, 7, 2, layer, 7, 1);Cube_Move_Pixel(layer, 7, 3, layer, 7, 2);Cube_Move_Pixel(layer, 7, 4, layer, 7, 3);Cube_Move_Pixel(layer, 7, 5, layer, 7, 4);Cube_Move_Pixel(layer, 7, 6, layer, 7, 5);Cube_Move_Pixel(layer, 7, 7, layer, 7, 6);Cube_Move_Pixel(layer, 6, 7, layer, 7, 7);Cube_Move_Pixel(layer, 5, 7, layer, 6, 7);Cube_Move_Pixel(layer, 4, 7, layer, 5, 7);Cube_Move_Pixel(layer, 3, 7, layer, 4, 7);Cube_Move_Pixel(layer, 2, 7, layer, 3, 7);Cube_Move_Pixel(layer, 1, 7, layer, 2, 7);Cube_Move_Pixel(layer, 0, 7, layer, 1, 7);Cube_Move_Pixel(layer, 0, 6, layer, 0, 7);Cube_Move_Pixel(layer, 0, 5, layer, 0, 6);Cube_Move_Pixel(layer, 0, 4, layer, 0, 5);Cube_Move_Pixel(layer, 0, 3, layer, 0, 4);Cube_Move_Pixel(layer, 0, 2, layer, 0, 3);Cube_Move_Pixel(layer, 0, 1, layer, 0, 2);Cube_Set_Pixel(layer, 0, 1, old[0], old[1], old[2]);if shell == 1:# Rotate second to outermost layerold = Cube_Get_Pixel(layer, 1, 1);Cube_Move_Pixel(layer, 2, 1, layer, 1, 1);Cube_Move_Pixel(layer, 3, 1, layer, 2, 1);Cube_Move_Pixel(layer, 4, 1, layer, 3, 1);Cube_Move_Pixel(layer, 5, 1, layer, 4, 1);Cube_Move_Pixel(layer, 6, 1, layer, 5, 1);Cube_Move_Pixel(layer, 6, 2, layer, 6, 1);Cube_Move_Pixel(layer, 6, 3, layer, 6, 2);Cube_Move_Pixel(layer, 6, 4, layer, 6, 3);Cube_Move_Pixel(layer, 6, 5, layer, 6, 4);Cube_Move_Pixel(layer, 6, 6, layer, 6, 5);Cube_Move_Pixel(layer, 5, 6, layer, 6, 6);Cube_Move_Pixel(layer, 4, 6, layer, 5, 6);Cube_Move_Pixel(layer, 3, 6, layer, 4, 6);Cube_Move_Pixel(layer, 2, 6, layer, 3, 6);Cube_Move_Pixel(layer, 1, 6, layer, 2, 6);Cube_Move_Pixel(layer, 1, 5, layer, 1, 6);Cube_Move_Pixel(layer, 1, 4, layer, 1, 5);Cube_Move_Pixel(layer, 1, 3, layer, 1, 4);Cube_Move_Pixel(layer, 1, 2, layer, 1, 3);Cube_Set_Pixel(layer, 1, 2, old[0], old[1], old[2]);if shell == 2:# Rotate second to innermost layerold = Cube_Get_Pixel(layer, 2, 2);Cube_Move_Pixel(layer, 3, 2, layer, 2, 2);Cube_Move_Pixel(layer, 4, 2, layer, 3, 2);Cube_Move_Pixel(layer, 5, 2, layer, 4, 2);Cube_Move_Pixel(layer, 5, 3, layer, 5, 2);Cube_Move_Pixel(layer, 5, 4, layer, 5, 3);Cube_Move_Pixel(layer, 5, 5, layer, 5, 4);Cube_Move_Pixel(layer, 4, 5, layer, 5, 5);Cube_Move_Pixel(layer, 3, 5, layer, 4, 5);Cube_Move_Pixel(layer, 2, 5, layer, 3, 5);Cube_Move_Pixel(layer, 2, 4, layer, 2, 5);Cube_Move_Pixel(layer, 2, 3, layer, 2, 4);Cube_Set_Pixel(layer, 2, 3, old[0], old[1], old[2]);if shell == 3:# Rotate innermost layerold = Cube_Get_Pixel(layer, 3, 3);Cube_Move_Pixel(layer, 4, 3, layer, 3, 3);Cube_Move_Pixel(layer, 4, 4, layer, 4, 3);Cube_Move_Pixel(layer, 3, 4, layer, 4, 4);Cube_Set_Pixel(layer, 3, 4, old[0], old[1], old[2]);'''Rotates the entire cube in the local buffer.'''def Cube_Rotate(direction):# Rotate outermost layerCube_Rotate_Shell(0, direction);# Rotate second to outermost layerif ((rotation_counter != 1) and (rotation_counter != 5)):Cube_Rotate_Shell(1, direction);# Rotate second to innermost layerif ((rotation_counter != 0) and (rotation_counter != 2) and (rotation_counter != 4) and (rotation_counter != 6)):Cube_Rotate_Shell(2, direction);# Rotate innermost layerif ((rotation_counter == 3) or (rotation_counter == 7)):Cube_Rotate_Shell(3, direction);if (direction == 0):if rotation_counter == CUBE_ROTATIONS - 1:rotation_counter = 0else:rotation_counter = rotation_counter + 1else:if rotation_counter == 0:rotation_counter = CUBE_ROTATIONS - 1else:rotation_counter = rotation_counter - 1'''Write the local buffer to the cube.'''def Cube_Update():serial_port.write(CMD_Set_All(cube_buffer))