You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
// -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
|
|
// A program that reads frames form STDIN as RGB24, much like
|
|
// https://github.com/polyfloyd/ledcat does.
|
|
//
|
|
// This code is public domain
|
|
// (but note, that the led-matrix library this depends on is GPL v2)
|
|
|
|
#include "led-matrix.h"
|
|
|
|
#include <math.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
|
|
#define FPS 60
|
|
|
|
using rgb_matrix::RGBMatrix;
|
|
using rgb_matrix::Canvas;
|
|
|
|
volatile bool interrupt_received = false;
|
|
static void InterruptHandler(int signo) {
|
|
interrupt_received = true;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
RGBMatrix::Options defaults;
|
|
defaults.hardware_mapping = "regular"; // or e.g. "adafruit-hat"
|
|
defaults.rows = 32;
|
|
defaults.chain_length = 1;
|
|
defaults.parallel = 1;
|
|
Canvas *canvas = RGBMatrix::CreateFromFlags(&argc, &argv, &defaults);
|
|
if (canvas == NULL) {
|
|
return 1;
|
|
}
|
|
|
|
// It is always good to set up a signal handler to cleanly exit when we
|
|
// receive a CTRL-C for instance. The DrawOnCanvas() routine is looking
|
|
// for that.
|
|
signal(SIGTERM, InterruptHandler);
|
|
signal(SIGINT, InterruptHandler);
|
|
|
|
ssize_t frame_size = canvas->width() * canvas->height() * 3;
|
|
uint8_t buf[frame_size];
|
|
|
|
while (1) {
|
|
struct timespec start;
|
|
timespec_get(&start, TIME_UTC);
|
|
|
|
ssize_t nread;
|
|
ssize_t total_nread = 0;
|
|
while ((nread = read(STDIN_FILENO, &buf[total_nread], frame_size - total_nread)) > 0) {
|
|
if (interrupt_received) {
|
|
return 1;
|
|
}
|
|
total_nread += nread;
|
|
}
|
|
if (total_nread < frame_size){
|
|
break;
|
|
}
|
|
|
|
for (int y = 0; y < canvas->height(); y++) {
|
|
for (int x = 0; x < canvas->width(); x++) {
|
|
uint8_t *p = &buf[(y * canvas->width() + x) * 3];
|
|
uint8_t r = *(p+0), g = *(p+1), b = *(p+2);
|
|
canvas->SetPixel(x, y, r, g, b);
|
|
}
|
|
}
|
|
|
|
struct timespec end;
|
|
timespec_get(&end, TIME_UTC);
|
|
long tudiff = (end.tv_nsec / 1000 + end.tv_sec * 1000000) - (start.tv_nsec / 1000 + start.tv_sec * 1000000);
|
|
if (tudiff < 1000000l / FPS) {
|
|
usleep(1000000l / FPS - tudiff);
|
|
}
|
|
}
|
|
|
|
// Animation finished. Shut down the RGB matrix.
|
|
canvas->Clear();
|
|
delete canvas;
|
|
return 0;
|
|
}
|