/* * vim: sw=2 ts=2 expandtab * * "THE PIZZA-WARE LICENSE" (derived from "THE BEER-WARE LICENCE"): * Thomas Kolb wrote this file. As long as you retain this * notice you can do whatever you want with this stuff. If we meet some day, * and you think this stuff is worth it, you can buy me a pizza in return. * - Thomas Kolb */ #include #include #include #include #include #include #include "ws2801.h" #define SET_COLOR 0 #define FADE_COLOR 1 #define ADD_COLOR 2 #define SET_FADESTEP 3 struct WS2801Packet { uint8_t action; uint8_t module; uint8_t data[3]; }; int ws2801_socket = -1; struct WS2801Packet packetQueue[1024]; int queueIndex = 0; // creates the socket needed for steering the LED strip int ws2801_init(const char *host, unsigned short port) { struct addrinfo hints; struct addrinfo *result; char portstr[6]; memset(&hints, 0, sizeof(struct addrinfo)); hints.ai_family = AF_UNSPEC; hints.ai_socktype = SOCK_DGRAM; hints.ai_flags = 0; hints.ai_protocol = 0; sprintf(portstr, "%u", port); if(getaddrinfo(host, portstr, &hints, &result) != 0) { perror("getaddrinfo() failed"); return 1; } ws2801_socket = socket(result->ai_family, result->ai_socktype, result->ai_protocol); if (ws2801_socket == -1) { perror("socket() failed"); freeaddrinfo(result); return 2; } if (connect(ws2801_socket, result->ai_addr, result->ai_addrlen) == -1) { perror("connect() failed"); freeaddrinfo(result); return 3; } freeaddrinfo(result); return 0; } void ws2801_set_color(uint8_t module, uint8_t r, uint8_t g, uint8_t b) { packetQueue[queueIndex].action = SET_COLOR; packetQueue[queueIndex].module = module; packetQueue[queueIndex].data[0] = r; packetQueue[queueIndex].data[1] = g; packetQueue[queueIndex].data[2] = b; queueIndex++; } void ws2801_fade_color(uint8_t module, uint8_t r, uint8_t g, uint8_t b) { packetQueue[queueIndex].action = FADE_COLOR; packetQueue[queueIndex].module = module; packetQueue[queueIndex].data[0] = r; packetQueue[queueIndex].data[1] = g; packetQueue[queueIndex].data[2] = b; queueIndex++; } void ws2801_add_color(uint8_t module, uint8_t r, uint8_t g, uint8_t b) { packetQueue[queueIndex].action = ADD_COLOR; packetQueue[queueIndex].module = module; packetQueue[queueIndex].data[0] = r; packetQueue[queueIndex].data[1] = g; packetQueue[queueIndex].data[2] = b; queueIndex++; } void ws2801_set_fadestep(uint8_t fadestep) { packetQueue[queueIndex].action = SET_FADESTEP; packetQueue[queueIndex].data[0] = fadestep; queueIndex++; } int ws2801_commit(void) { if(send(ws2801_socket, packetQueue, queueIndex * sizeof(struct WS2801Packet), 0) == -1) { return 1; } queueIndex = 0; return 0; } void ws2801_shutdown() { close(ws2801_socket); }