| 0,0 → 1,136 |
|---|
| 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) |