mirror of
git://projects.qi-hardware.com/ben-wpan.git
synced 2024-12-24 00:54:15 +02:00
6ca63fc6e7
- gui.c (old_term, restore_term, raw, get_key): raw and non-blocking console input - gui,c (gui): accept console input in addition to keypressed in the SDL window
421 lines
7.2 KiB
C
421 lines
7.2 KiB
C
/*
|
|
* atrf-path/gui.c - Graphical output for atrf-path
|
|
*
|
|
* Written 2011 by Werner Almesberger
|
|
* Copyright 2011 Werner Almesberger
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*/
|
|
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <termios.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#include "SDL.h"
|
|
#include "SDL_gfxPrimitives.h"
|
|
|
|
#include "at86rf230.h"
|
|
#include "atrf.h"
|
|
#include "misctxrx.h"
|
|
|
|
#include "sweep.h"
|
|
#include "gui.h"
|
|
|
|
|
|
#define XRES 320
|
|
#define YRES 240
|
|
|
|
#define FG_RGBA 0xffffffff /* measurement color */
|
|
#define OK_RGBA 0x00ff00ff
|
|
#define OVER_RGBA 0xffff00ff
|
|
#define UNDER_RGBA 0xff0000ff
|
|
#define LIMIT_RGBA 0xff0000ff
|
|
#define DUMP_RGBA 0x4040ffff
|
|
|
|
#define CHAN_STEP 20 /* 4 pixels/MHz */
|
|
#define SIDE_STEP 2
|
|
#define CHAN_X_OFFSET 10
|
|
#define Y_MIN -94
|
|
#define Y_MAX -10
|
|
|
|
#define STATUS_X (XRES-15)
|
|
#define STATUS_Y 15
|
|
#define STATUS_R 8
|
|
|
|
|
|
static struct sample **dumps = NULL;
|
|
static int n_dumps = 0;
|
|
|
|
|
|
static int avg2y(double avg)
|
|
{
|
|
return YRES-(avg-Y_MIN)/(Y_MAX-Y_MIN)*YRES-1;
|
|
}
|
|
|
|
|
|
static void segment(SDL_Surface *s, int *last_x, int *last_y, int x,
|
|
const struct sample *res, int first, uint32_t color)
|
|
{
|
|
int y = avg2y(res->avg);
|
|
|
|
if (!first) {
|
|
aalineColor(s, *last_x, *last_y, x, y, color);
|
|
}
|
|
*last_x = x;
|
|
*last_y = y;
|
|
}
|
|
|
|
|
|
static void draw(SDL_Surface *s, const struct sample *res, int cont_tx,
|
|
uint32_t color)
|
|
{
|
|
int last_x, last_y;
|
|
int first, x, i;
|
|
|
|
x = CHAN_X_OFFSET;
|
|
first = 1;
|
|
for (i = 0; i != N_CHAN; i++) {
|
|
if (cont_tx != CONT_TX_P500K) {
|
|
segment(s, &last_x, &last_y, x, res, first, color);
|
|
first = 0;
|
|
}
|
|
res++;
|
|
x += 2*SIDE_STEP;
|
|
|
|
if (cont_tx != CONT_TX_M500K) {
|
|
segment(s, &last_x, &last_y, x, res, first, color);
|
|
first = 0;
|
|
}
|
|
res++;
|
|
x += CHAN_STEP-2*SIDE_STEP;
|
|
}
|
|
}
|
|
|
|
|
|
static void draw_limit(SDL_Surface *s, const double *v)
|
|
{
|
|
int x, y, i, last = 0;
|
|
|
|
x = CHAN_X_OFFSET;
|
|
for (i = 0; i != N_CHAN; i++) {
|
|
y = avg2y(*v);
|
|
if (i)
|
|
vlineColor(s, x-CHAN_STEP/2, last, y, LIMIT_RGBA);
|
|
hlineColor(s, x-CHAN_STEP/2, x+CHAN_STEP/2, y, LIMIT_RGBA);
|
|
last = y;
|
|
x += CHAN_STEP;
|
|
v++;
|
|
}
|
|
}
|
|
|
|
|
|
static void disc(SDL_Surface *s, uint32_t color)
|
|
{
|
|
filledCircleColor(s, STATUS_X, STATUS_Y, STATUS_R, color);
|
|
aacircleColor(s, STATUS_X, STATUS_Y, STATUS_R, color);
|
|
}
|
|
|
|
|
|
static void triangle(SDL_Surface *s, int cx, int cy, int r, uint32_t color)
|
|
{
|
|
filledTrigonColor(s, cx, cy-r, cx+r*1.2, cy+r, cx-r*1.2, cy+r, color);
|
|
// aatrigonColor(s, cx, cy-r, cx+r*1.2, cy+r, cx-r*1.2, cy+r, color);
|
|
}
|
|
|
|
|
|
static void up(SDL_Surface *s, uint32_t color)
|
|
{
|
|
triangle(s, STATUS_X, STATUS_Y, STATUS_R, color);
|
|
}
|
|
|
|
|
|
static void down(SDL_Surface *s, uint32_t color)
|
|
{
|
|
triangle(s, STATUS_X, STATUS_Y, -STATUS_R, color);
|
|
}
|
|
|
|
|
|
static void indicate(SDL_Surface *s, int fail)
|
|
{
|
|
static uint32_t last = 0;
|
|
uint32_t color;
|
|
|
|
switch (fail) {
|
|
case 0:
|
|
color = OK_RGBA;
|
|
break;
|
|
case 1:
|
|
color = OVER_RGBA;
|
|
break;
|
|
case -1:
|
|
color = UNDER_RGBA;
|
|
break;
|
|
default:
|
|
abort();
|
|
}
|
|
if (color == last)
|
|
color = 0;
|
|
last = color;
|
|
switch (color) {
|
|
case OK_RGBA:
|
|
disc(s, color);
|
|
break;
|
|
case OVER_RGBA:
|
|
up(s, color);
|
|
break;
|
|
case UNDER_RGBA:
|
|
down(s, color);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
static void clear(SDL_Surface *s)
|
|
{
|
|
SDL_FillRect(s, NULL, SDL_MapRGB(s->format, 0, 0, 0));
|
|
}
|
|
|
|
|
|
static void dump(const struct sweep *sweep, const struct sample *res)
|
|
{
|
|
const size_t size = sizeof(struct sample)*N_CHAN*2;
|
|
|
|
print_sweep(sweep, res);
|
|
|
|
dumps = realloc(dumps, (n_dumps+1)*sizeof(struct sample *));
|
|
if (!dumps) {
|
|
perror("realloc");
|
|
exit(1);
|
|
}
|
|
dumps[n_dumps] = malloc(size);
|
|
if (!dumps[n_dumps]) {
|
|
perror("malloc");
|
|
exit(1);
|
|
}
|
|
memcpy(dumps[n_dumps], res, size);
|
|
n_dumps++;
|
|
}
|
|
|
|
|
|
static void draw_dumps(SDL_Surface *s, int cont_tx)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i != n_dumps; i++)
|
|
draw(s, dumps[i], cont_tx, DUMP_RGBA);
|
|
}
|
|
|
|
|
|
/* ----- Console input ----------------------------------------------------- */
|
|
|
|
|
|
static struct termios old_term;
|
|
|
|
|
|
static void restore_term(void)
|
|
{
|
|
if (tcsetattr(0, TCSAFLUSH, &old_term) < 0)
|
|
perror("tcsetattr");
|
|
}
|
|
|
|
|
|
static void raw(void)
|
|
{
|
|
struct termios term;
|
|
|
|
if (tcgetattr(0, &old_term) < 0) {
|
|
perror("tcgetattr");
|
|
exit(1);
|
|
}
|
|
term = old_term;
|
|
cfmakeraw(&term);
|
|
if (tcsetattr(0, TCSAFLUSH, &term) < 0) {
|
|
perror("tcsetattr");
|
|
exit(1);
|
|
}
|
|
atexit(restore_term);
|
|
if (fcntl(0, F_SETFL, O_NONBLOCK) < 0) {
|
|
perror("fcntl");
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
|
|
static char get_key(void)
|
|
{
|
|
ssize_t got;
|
|
char ch;
|
|
|
|
got = read(0, &ch, 1);
|
|
if (got == 1)
|
|
return ch;
|
|
if (got >= 0) {
|
|
fprintf(stderr, "unexpected read() return value %d\n",
|
|
(int) got);
|
|
exit(1);
|
|
}
|
|
if (errno == EAGAIN)
|
|
return 0;
|
|
perror("read");
|
|
exit(1);
|
|
}
|
|
|
|
|
|
/* --- temporarily, for optimizing --- */
|
|
|
|
#if 0
|
|
|
|
#include <sys/time.h>
|
|
|
|
|
|
static double t0;
|
|
|
|
|
|
static double t(void)
|
|
{
|
|
struct timeval tv;
|
|
|
|
gettimeofday(&tv, NULL);
|
|
return tv.tv_sec+tv.tv_usec/1000000.0;
|
|
}
|
|
|
|
|
|
static void tstart(void)
|
|
{
|
|
t0 = t();
|
|
}
|
|
|
|
|
|
static void tstop(void)
|
|
{
|
|
fprintf(stderr, "%.3f\n", t()-t0);
|
|
}
|
|
|
|
#else
|
|
|
|
static void tstart(void) {}
|
|
static void tstop(void) {}
|
|
|
|
#endif
|
|
|
|
|
|
int gui(const struct sweep *sweep, int sweeps)
|
|
{
|
|
SDL_Surface *surf;
|
|
SDL_Event event;
|
|
int cycle;
|
|
int fail = 0;
|
|
|
|
if (SDL_Init(SDL_INIT_VIDEO) < 0) {
|
|
fprintf(stderr, "SDL_init: %s\n", SDL_GetError());
|
|
exit(1);
|
|
}
|
|
atexit(SDL_Quit);
|
|
|
|
raw();
|
|
|
|
surf = SDL_SetVideoMode(XRES, YRES, 0, SDL_SWSURFACE);
|
|
if (!surf) {
|
|
fprintf(stderr, "SDL_SetVideoMode: %s\n", SDL_GetError());
|
|
exit(1);
|
|
}
|
|
|
|
for (cycle = 0; cycle != sweeps || !sweeps; cycle++) {
|
|
struct sample res[N_CHAN*2];
|
|
|
|
/*
|
|
* Pass/fail logic:
|
|
*
|
|
* Quit exit at any time, without making a pass/fail decision
|
|
* Pass exit if the current result is "pass"
|
|
* ignored if the current result is "over"/"under"
|
|
* Fail exit if the current result is "under"
|
|
* ignored if the current result is "pass"
|
|
* ignored if the current result is "over", because this
|
|
* indicates an invalid measurement, not a defective
|
|
* device
|
|
*/
|
|
|
|
while (SDL_PollEvent(&event))
|
|
switch (event.type) {
|
|
case SDL_KEYDOWN:
|
|
switch (event.key.keysym.sym) {
|
|
case SDLK_f:
|
|
if (cycle && fail < 0)
|
|
return -1;
|
|
break;
|
|
case SDLK_p:
|
|
if (cycle && !fail)
|
|
return 1;
|
|
break;
|
|
case SDLK_d:
|
|
if (cycle)
|
|
dump(sweep, res);
|
|
break;
|
|
case SDLK_q:
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case SDL_QUIT:
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
switch (get_key()) {
|
|
case 'F':
|
|
case 'f':
|
|
if (cycle && fail < 0)
|
|
return -1;
|
|
break;
|
|
case 'P':
|
|
case 'p':
|
|
if (cycle && !fail)
|
|
return 1;
|
|
break;
|
|
case 'D':
|
|
case 'd':
|
|
if (cycle)
|
|
dump(sweep, res);
|
|
break;
|
|
case 'Q':
|
|
case 'q':
|
|
case 3: /* Ctrl-C */
|
|
return 0;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
tstart();
|
|
fail = do_sweep(sweep, res);
|
|
tstop();
|
|
|
|
SDL_LockSurface(surf);
|
|
|
|
clear(surf);
|
|
|
|
draw_dumps(surf, sweep->cont_tx);
|
|
draw_limit(surf, sweep->min);
|
|
draw_limit(surf, sweep->max);
|
|
indicate(surf, fail);
|
|
draw(surf, res, sweep->cont_tx, FG_RGBA);
|
|
|
|
SDL_UnlockSurface(surf);
|
|
SDL_UpdateRect(surf, 0, 0, 0, 0);
|
|
}
|
|
|
|
return 0;
|
|
}
|