Subversion Repositories Code-Repo

Rev

Blame | Last modification | View Log | RSS feed

import socket, struct, time

# Set the address of the Cerebot board
dst_addr = '00183E00D7EB'.decode('hex')
# Open a socket on the eth0
sock = socket.socket(socket.AF_PACKET, socket.SOCK_RAW)
sock.bind(("eth0", 0x1))
# Acquire MAC address of local machine
if_name, if_proto, pkt_type, hw_type, hw_addr = sock.getsockname()

# Create and initialize the frame buffer
frame_buffer_size = 1536
frame_buffer = [0] * frame_buffer_size

def cube_init():
    '''Sets the cube into ethernet mode.'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 1)
    txOpcode = "01".decode('hex')
    sock.send(txFrame + txOpcode)
    # Wait a few seconds for the cube to reset itself
    time.sleep(6)

def cube_reset():
    '''Resets the cube into idle mode.'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 1)
    txOpcode = "02".decode('hex')
    sock.send(txFrame + txOpcode)

def cube_clear():
    '''Clear the cube's internal buffer.'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 1)
    txOpcode = "0A".decode('hex')
    sock.send(txFrame + txOpcode)

def cube_brightness(value):
    '''Sets the global brightness value from 0 to 255.'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 2)
    txOpcode = "0B".decode('hex')
    txData = format(value, '02x').decode('hex')
    sock.send(txFrame + txOpcode + txData)

def cube_rotate(direction):
    '''Rotate the entire cube (0 = clockwise, 1 = counterclockwise).'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 2)
    txOpcode = "0C".decode('hex')
    txData = format(direction, '02x').decode('hex')
    sock.send(txFrame + txOpcode + txData)

def cube_rotate_shell(direction, shell):
    '''Rotate a specific layer (0 = clockwise, 0 = outermost layer).'''
    # Generate and send the frame
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 3)
    txOpcode = "0D".decode('hex')
    txData = ''.join("%02x%02x" % (direction, shell))
    sock.send(txFrame + txOpcode + txData.decode('hex'))

# def cube_update():
#     '''Update the cube with the current frame buffer.
#         Note: this requires jumbo frames (1536 bytes).'''
#     # Generate the header, opcode, and format the frame buffer
#     txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 0x0601)
#     txOpcode = "10".decode('hex')
#     txData = ''.join("%02x" % (x) for x in frame_buffer)
#     sock.send(txFrame + txOpcode + txData.decode('hex'))

def cube_update_pixel(x, y, z, r, g, b):
    '''Set a specific pixel on the cube.'''
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 7)
    txOpcode = "11".decode('hex')
    frame = [x, y, z, r, g, b]
    txData = ''.join("%02x" % (x) for x in frame)
    sock.send(txFrame + txOpcode + txData.decode('hex'))

def cube_update():
    '''Updates the cube with the current frame buffer.
        The buffer is sent in three frames, one for each color channel.'''
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 514)
    txOpcode = "12".decode('hex')
    for c in range(3):
        txColorCh = "%02x" % c
        txData = ''.join( ["%02x" % (x) for x in frame_buffer[c::3]])
        payload = txFrame + txOpcode + txColorCh.decode('hex') + txData.decode('hex')
        sock.send(payload)

def cube_update_text(string, r, g, b, update_rate):
    '''Sets the scrolling text on the cube.'''
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, len(string) + 6)
    txOpcode = "20".decode('hex')
    txData = ''.join("%02x%02x%02x%02x%02x" % (len(string), r, g, b, update_rate))
    txString = string.encode('hex')
    sock.send(txFrame + txOpcode + txData.decode('hex') + txString.decode('hex'))

def cube_update_waterfall(c0, c1, c2, c3, c4, c5, c6, c7):
    '''Fills in one row and shifts rows by one.'''
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 9)
    txOpcode = "30".decode('hex')
    txData = ''.join("%02x%02x%02x%02x%02x%02x%02x%02x" % (c0, c1, c2, c3, c4, c5, c6, c7))
    sock.send(txFrame + txOpcode + txData.decode('hex'))

def cube_update_sphere(layer, r, g, b):
    '''Sets the sphere layer to the specified color.'''
    txFrame = struct.pack("!6s6sH", dst_addr, hw_addr, 5)
    txOpcode = "31".decode('hex')
    txData = ''.join("%02x%02x%02x%02x" % (layer, r, g, b))
    sock.send(txFrame + txOpcode + txData.decode('hex'))

def fb_clear():
    '''Clears the frame buffer/'''
    for row in range(8):
        for column in range(8):
            for layer in range(8):
                  fb_set_pixel(row, column, layer, 0x00, 0x00, 0x00)

def fb_set_pixel(row, column, layer, R, G, B):
    '''Sets a pixel to the given color in the frame buffer/'''
    frame_buffer[layer * 192 + column * 24 + row * 3 + 0] = R
    frame_buffer[layer * 192 + column * 24 + row * 3 + 1] = G
    frame_buffer[layer * 192 + column * 24 + row * 3 + 2] = B

def fb_set_layer(layer, R, G, B):
    '''Sets an entire layer to the given color in the frame buffer'''
    for row in range(8):
        for column in range(8):
            fb_set_pixel(row, column, layer, R, G, B)
    
def fb_set_all(R, G, B):
    '''Sets all the pixels in the frame buffer to the given color'''
    for row in range(8):
        for column in range(8):
            for layer in range(8):
                fb_set_pixel(row, column, layer, R, G, B)