void Strobe::circle() {
for (int i = 0; i < numLedsProBar; i++) {
pliste[i].set(false, color, colors[color].a, colors[color].b,
colors[color].c);
}
if (strobecounter % (speed) == 0)
switchcounter++;
if (strobecounter % (speed) < 3) {
//for (int i = 0; i < numBars; i += 2) {
for (int j = 0; j < numLedsProBar; j++) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder), targetStrip,
(switchcounter % numBars) * numLedsProBar + j,
colors[pliste[j].color].a, colors[pliste[j].color].b,
colors[pliste[j].color].c);
}
//}
}
}
void Balls::raining() {
if (raincounter % 311 == raintime) {
raintime = rand() % 311;
raindrops[0] = true;
}
if (raincounter % 20 == 0) {
bool old, act;
old = raindrops[0];
raindrops[0] = false;
for (int i = 1; i < numBars * numLedsProBar - 1; i++) {
act = raindrops[i];
raindrops[i] = old;
if (old) {
//ledscape_set_color(frame, color_channel_order_from_string(ColorOrder), targetStrip, i-1, 0.05*colors[color].a, 0.05*colors[color].b, 0.05*colors[color].c );
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i, 0.2 * colors[color].a,
0.2 * colors[color].b, 0.2 * colors[color].c);
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i + 1, 1 * colors[color].a,
1 * colors[color].b, 1 * colors[color].c);
}
old = act;
}
} else {
for (int i = 2; i < numBars * numLedsProBar - 1; i++) {
if (raindrops[i]) {
//ledscape_set_color(frame, color_channel_order_from_string(ColorOrder), targetStrip, i-2, 0.05*colors[color].a, 0.05*colors[color].b, 0.05*colors[color].c );
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i - 1, 0.05 * colors[color].a,
0.05 * colors[color].b, 0.05 * colors[color].c);
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i, 0.1 * colors[color].a,
0.1 * colors[color].b, 0.1 * colors[color].c);
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i + 1, 1 * colors[color].a,
1 * colors[color].b, 1 * colors[color].c);
}
}
}
}
void Strobe::tripleShot() {
if (counter % speed <= 5 && n < 3) {
for (int i = 0; i < numLeds; i++) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder), targetStrip, i,
colors[color].a, colors[color].b, colors[color].c);
}
if (counter % speed == 0)
n++;
} else {
for (int i = 0; i < numLeds; i++) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder), targetStrip, i,
0, 0, 0);
}
}
}
int main (int argc, char *argv[])
{
ledscape_t * const leds = ledscape_init_with_programs(LEDSCAPE_NUM_LEDS,"pru/bin/ws281x-valhalla-pru0.bin","pru/bin/ws281x-valhalla-pru1.bin");
uint8_t rgb[3];
if ( argc > 4 ) {
rgb[0] = atoi(argv[1]);
rgb[1] = atoi(argv[2]);
rgb[2] = atoi(argv[3]);
} else {
printf("Provide at least 3 colors!\n");
exit(1);
}
printf("here\n");
ledscape_frame_t * const frame = ledscape_frame(leds, 0);
if ( argc > 4 ) {
unsigned strip_input = atoi(argv[4]);
printf("Using strip %i.\n", strip_input);
for (unsigned p = 0 ; p < LEDSCAPE_NUM_LEDS; p++)
{
ledscape_set_color(frame, COLOR_ORDER_RGB, strip_input, p, rgb[0] , rgb[1], rgb[2]);
}
} else {
printf("Using all strips.\n");
for (unsigned strip = 0 ; strip < LEDSCAPE_NUM_STRIPS ; strip++)
{
for (unsigned p = 0 ; p < LEDSCAPE_NUM_LEDS; p++)
{
ledscape_set_color(frame, COLOR_ORDER_RGB, strip, p, rgb[0] , rgb[1], rgb[2]);
}
}
}
ledscape_wait(leds);
ledscape_draw(leds, 0);
//ledscape_close(leds);
return EXIT_SUCCESS;
}
void Strobe::switchedBLUEWHITE() {
for (int i = 0; i < numLedsProBar; i++) {
pliste[i].set(false, color, colors[color].a, colors[color].b,
colors[color].c);
}
if (strobecounter % (speed) == 0)
switchcounter++;
if (strobecounter % (speed) < 3) {
if (switchcounter % 2 == 0) {
for (int i = 0; i < numBars; i += 2) {
for (int j = 0; j < numLedsProBar; j++) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i * numLedsProBar + j,
colors[pliste[j].color].a,
colors[pliste[j].color].b,
colors[pliste[j].color].c);
}
}
} else {
for (int i = 1; i < numBars; i += 2) {
for (int j = 0; j < numLedsProBar; j++) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),
targetStrip, i * numLedsProBar + j, 255, 255, 255);
}
}
}
}
}
void Strobe::random() {
for (int i = 0; i < numLedsProBar; i++) {
pliste[i].set(false, color, colors[color].a, colors[color].b,
colors[color].c);
}
if (strobecounter % (speed) == 0)
{
pos[0] = ((double)rand() / RAND_MAX) * (numBars*numLedsProBar -1);
pos[1] = ((double)rand() / RAND_MAX) * (numBars*numLedsProBar -1);
pos[2] = ((double)rand() / RAND_MAX) * (numBars*numLedsProBar -1);
//cout << pos[0] << " " << pos[1] << " " << pos[2] << endl;
}
if (strobecounter % (speed) < 3) {
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),targetStrip,
pos[0],
colors[color].a,
colors[color].b,
colors[color].c);
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),targetStrip,
pos[1],
colors[color].a,
colors[color].b,
colors[color].c);
ledscape_set_color(frame,
color_channel_order_from_string(ColorOrder),targetStrip,
pos[2],
colors[color].a,
colors[color].b,
colors[color].c);
drawEqual();
}
}
int
main(
int argc,
char ** argv
)
{
uint16_t port = 9999;
uint16_t num_pixels = 256;
uint16_t num_strips = LEDSCAPE_NUM_STRIPS;
extern char *optarg;
int opt;
while ((opt = getopt(argc, argv, "p:c:d:")) != -1)
{
switch (opt)
{
case 'p':
port = atoi(optarg);
break;
case 'c':
num_pixels = atoi(optarg);
break;
case 'd': {
int width=0, height=0;
if (sscanf(optarg,"%dx%d", &width, &height) == 2) {
num_pixels = width * height;
} else {
printf("Invalid argument for -d; expected NxN; actual: %s", optarg);
exit(EXIT_FAILURE);
}
}
break;
default:
fprintf(stderr, "Usage: %s [-p <port>] [-c <led_count> | -d <width>x<height>]\n", argv[0]);
exit(EXIT_FAILURE);
}
}
const int sock = socket(AF_INET, SOCK_DGRAM, 0);
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_addr.s_addr = INADDR_ANY,
.sin_port = htons(port),
};
if (sock < 0)
die("socket failed: %s\n", strerror(errno));
if (bind(sock, (const struct sockaddr*) &addr, sizeof(addr)) < 0)
die("bind port %d failed: %s\n", port, strerror(errno));
ledscape_t * const leds = ledscape_init(num_pixels);
fprintf(stderr, "Started LEDscape UDP receiver on port %d for %d pixels\n", port, num_pixels);
fprintf(stderr, "WARNING: udp-rx is deprecated! opc-rx supports more options and should be used instead.\n\n");
unsigned frame_num = 0;
uint8_t buf[num_pixels * num_strips * 4];
time_t last_time = time(NULL);
int fps_counter=0;
while (1)
{
const ssize_t rc = recv(sock, buf, sizeof(buf), 0);
if (rc < 0) {
printf("recv failed: %s\n", strerror(errno));
continue;
}
ledscape_frame_t * const frame
= ledscape_frame(leds, frame_num);
for(uint16_t x=0 ; x < num_pixels ; x++)
{
for(uint16_t strip = 0 ; strip < num_strips ; strip++)
{
const uint8_t r = buf[strip*num_pixels*3 + x*3 + 0];
const uint8_t g = buf[strip*num_pixels*3 + x*3 + 1];
const uint8_t b = buf[strip*num_pixels*3 + x*3 + 2];
ledscape_set_color(frame, strip, x, r, g, b);
}
}
ledscape_wait(leds);
ledscape_draw(leds, frame_num);
time_t now = time(NULL);
if (now != last_time)
{
printf("%d fps\n", fps_counter);
last_time = now;
fps_counter = 0;
}
fps_counter++;
frame_num = (frame_num+1) % 2;
}
ledscape_close(leds);
return 0;
}
