mirror of
git://projects.qi-hardware.com/ben-wpan.git
synced 2024-11-30 06:47:30 +02:00
341 lines
6.2 KiB
C
341 lines
6.2 KiB
C
/*
|
|
* atrf-rssi/gui.c - Graphical output for atrf-rssi
|
|
*
|
|
* Written 2010-2011 by Werner Almesberger
|
|
* Copyright 2010-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 <signal.h>
|
|
#include <sys/time.h>
|
|
|
|
#include "SDL.h"
|
|
#include "SDL_gfxPrimitives.h"
|
|
|
|
#include "at86rf230.h"
|
|
#include "atrf.h"
|
|
#include "misctxrx.h"
|
|
#include "getkey.h"
|
|
|
|
#include "zgrid.h"
|
|
#include "digit.h"
|
|
#include "letter.h"
|
|
#include "gui.h"
|
|
|
|
|
|
#define XRES 320
|
|
#define YRES 240
|
|
|
|
#define N_CHAN 16
|
|
#define N_TIME 64
|
|
|
|
|
|
#define FG_RGBA 0xffffff00 /* grid foreround color */
|
|
#define BG_RGBA 0x00408080 /* grid background color */
|
|
#define CHAN_RGBA 0xff4040ff /* channel number color */
|
|
#define FREQ_RGBA 0x20ff00ff /* frequency color */
|
|
#define WCHAN_RGBA 0xffff00e0 /* WLAN channel number color */
|
|
#define WLAN_RGBA 0x8080ffff /* WLAN channel occupancy color */
|
|
#define SEL_RGBA WLAN_RGBA /* WLAN area selector */
|
|
|
|
#define X_STEP 17 /* grid x step */
|
|
#define Y_STEP 2 /* grid y step */
|
|
#define Z_STEP 3 /* z multiplier */
|
|
#define X_STEP_Y 1 /* x shift for each y step */
|
|
#define X_OFFSET 7 /* x coordinate of lower left grid corner */
|
|
#define Y_OFFSET 40 /* y coordinate of lower left grid corner */
|
|
|
|
#define X_WLAN_OFFSET 31
|
|
#define Y_WLAN_OFFSET 15
|
|
|
|
#define WLAN_XR (X_STEP*9.5/5)
|
|
#define WLAN_YH 6
|
|
|
|
#define X_SEL (XRES-20)
|
|
#define Y_SEL (YRES-50)
|
|
|
|
|
|
static enum {
|
|
area_off,
|
|
area_us,
|
|
area_eu,
|
|
area_jp,
|
|
} wlan_area = area_off;
|
|
|
|
|
|
static struct timeval t0;
|
|
|
|
|
|
static void shift_grid(int *z, int nx, int ny)
|
|
{
|
|
int *p1, *p0, *s;
|
|
int x, y;
|
|
|
|
p1 = z+(ny-1)*nx;
|
|
for (y = 1; y != ny; y++) {
|
|
p0 = s = p1-nx;
|
|
for (x = 0; x != nx; x++)
|
|
*p1++ = *p0++;
|
|
p1 = s;
|
|
}
|
|
}
|
|
|
|
|
|
static void sweep(struct atrf_dsc *dsc, int *z)
|
|
{
|
|
int chan;
|
|
|
|
for (chan = 11; chan <= 26; chan++) {
|
|
atrf_reg_write(dsc, REG_PHY_CC_CCA, chan);
|
|
/* 150 us, according to AVR2001 section 3.5 */
|
|
wait_for_interrupt(dsc, IRQ_PLL_LOCK, IRQ_PLL_LOCK, 1);
|
|
|
|
*z++ = Z_STEP*atrf_reg_read(dsc, REG_PHY_RSSI) & RSSI_MASK;
|
|
#if 0
|
|
if (chan >= 13 && chan <= 19 )
|
|
z[-1] = 3*28-(chan-16)*(chan-16)*(chan-16)*(chan-16);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
|
|
static void clear(SDL_Surface *s)
|
|
{
|
|
SDL_FillRect(s, NULL, SDL_MapRGB(s->format, 0, 0, 0));
|
|
}
|
|
|
|
|
|
#define CBIG(pos) \
|
|
x-5+(pos)*6, x-1+(pos)*6, y+8, y+4, y, CHAN_RGBA
|
|
#define CSMALL(pos) \
|
|
x-7+(pos)*4, x-5+(pos)*4, y+15, y+13, y+11, FREQ_RGBA
|
|
#define CWLAN(pos) \
|
|
x-4+(pos)*5, x-1+(pos)*5, y+6, y+3, y, WCHAN_RGBA
|
|
#define CSEL(pos) \
|
|
x-4+(pos)*5, x-1+(pos)*5, y+6, y+3, y, SEL_RGBA
|
|
|
|
|
|
static void label_channels(SDL_Surface *s, int sx, int x0, int y0)
|
|
{
|
|
int x, y, i, c, f;
|
|
|
|
x = x0;
|
|
y = s->h-y0+4;
|
|
for (i = 0; i != N_CHAN; i++) {
|
|
c = i+11;
|
|
digit(s, c/10, CBIG(0));
|
|
digit(s, c % 10, CBIG(1));
|
|
f = 2405+5*i;
|
|
if (i & 1)
|
|
y++;
|
|
digit(s, f/1000, CSMALL(0));
|
|
digit(s, (f/100) % 10, CSMALL(1));
|
|
digit(s, (f/10) % 10, CSMALL(2));
|
|
digit(s, f % 10, CSMALL(3));
|
|
if (i & 1)
|
|
y--;
|
|
x += sx;
|
|
}
|
|
}
|
|
|
|
|
|
static void print(SDL_Surface *s, int x, int y, const char *t)
|
|
{
|
|
int i = 0;
|
|
|
|
while (*t) {
|
|
letter(s, *t, CSEL(i));
|
|
t++;
|
|
i++;
|
|
}
|
|
}
|
|
|
|
|
|
static void area_selector(SDL_Surface *s, int x0, int y0)
|
|
{
|
|
int x, y, r;
|
|
|
|
print(s, x0, y0, "EU");
|
|
print(s, x0, y0+9, "JP");
|
|
print(s, x0, y0+18, "US");
|
|
switch (wlan_area) {
|
|
case area_off:
|
|
return;
|
|
case area_eu:
|
|
y = 0;
|
|
break;
|
|
case area_jp:
|
|
y = 1;
|
|
break;
|
|
case area_us:
|
|
y = 2;
|
|
break;
|
|
default:
|
|
abort();
|
|
}
|
|
y = y0+9*y+3;
|
|
x = x0-8;
|
|
r = 6;
|
|
filledTrigonColor(s, x, y, x-r, y+r/2, x-r, y-r/2, SEL_RGBA);
|
|
}
|
|
|
|
|
|
static int wlan_channels(void)
|
|
{
|
|
switch (wlan_area) {
|
|
case area_us:
|
|
return 11;
|
|
case area_eu:
|
|
return 13;
|
|
case area_jp:
|
|
return 14;
|
|
default:
|
|
abort();
|
|
}
|
|
}
|
|
|
|
|
|
static int show_wlan_channel(int c)
|
|
{
|
|
switch (wlan_area) {
|
|
case area_jp:
|
|
if (c == 14)
|
|
return 1;
|
|
/* fall through */
|
|
case area_us:
|
|
return !((c-1) % 5);
|
|
case area_eu:
|
|
return !((c-1) % 4);
|
|
default:
|
|
abort();
|
|
}
|
|
}
|
|
|
|
|
|
static void label_wlan_channels(SDL_Surface *s, int sx, int x0, int y0)
|
|
{
|
|
int x, xb, y, i, c;
|
|
|
|
xb = x0;
|
|
y = s->h-y0+4;
|
|
for (i = 0; i != wlan_channels(); i++) {
|
|
c = i+1;
|
|
/* Japan special channel: 22 MHz from channel 13 */
|
|
if (c == 14)
|
|
xb += X_STEP*12/5-X_STEP;
|
|
if (c > 9) {
|
|
x = xb;
|
|
digit(s, c/10, CWLAN(0));
|
|
} else {
|
|
x = xb-3;
|
|
}
|
|
digit(s, c % 10, CWLAN(1));
|
|
if (show_wlan_channel(c)) {
|
|
hlineColor(s, xb-WLAN_XR, xb+WLAN_XR, y-WLAN_YH,
|
|
WLAN_RGBA);
|
|
vlineColor(s, xb, y-2, y-WLAN_YH+1, WLAN_RGBA);
|
|
}
|
|
xb += sx;
|
|
}
|
|
}
|
|
|
|
|
|
void gui(struct atrf_dsc *dsc)
|
|
{
|
|
SDL_Surface *surf;
|
|
int z[N_CHAN*N_TIME];
|
|
SDL_Event event;
|
|
|
|
memset(z, 0, sizeof(z));
|
|
gettimeofday(&t0, NULL);
|
|
|
|
if (SDL_Init(SDL_INIT_VIDEO) < 0) {
|
|
fprintf(stderr, "SDL_init: %s\n", SDL_GetError());
|
|
exit(1);
|
|
}
|
|
atexit(SDL_Quit);
|
|
|
|
get_key_init();
|
|
|
|
surf = SDL_SetVideoMode(XRES, YRES, 0, SDL_SWSURFACE);
|
|
if (!surf) {
|
|
fprintf(stderr, "SDL_SetVideoMode: %s\n", SDL_GetError());
|
|
exit(1);
|
|
}
|
|
|
|
while (1) {
|
|
while (SDL_PollEvent(&event)) {
|
|
switch (event.type) {
|
|
case SDL_KEYDOWN:
|
|
switch (event.key.keysym.sym) {
|
|
case SDLK_j:
|
|
wlan_area = area_jp;
|
|
break;
|
|
case SDLK_e:
|
|
wlan_area = area_eu;
|
|
break;
|
|
case SDLK_u:
|
|
wlan_area = area_us;
|
|
break;
|
|
case SDLK_q:
|
|
return;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case SDL_QUIT:
|
|
return;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
switch (get_key()) {
|
|
case 'J':
|
|
case 'j':
|
|
wlan_area = area_jp;
|
|
break;
|
|
case 'E':
|
|
case 'e':
|
|
wlan_area = area_eu;
|
|
break;
|
|
case 'U':
|
|
case 'u':
|
|
wlan_area = area_us;
|
|
break;
|
|
case 'Q':
|
|
case 'q':
|
|
return;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
shift_grid(z, N_CHAN, N_TIME);
|
|
sweep(dsc, z);
|
|
|
|
SDL_LockSurface(surf);
|
|
|
|
clear(surf);
|
|
zgrid_draw(surf, z, N_CHAN, N_TIME,
|
|
X_STEP, Y_STEP, X_STEP_Y,
|
|
X_OFFSET, Y_OFFSET,
|
|
FG_RGBA, BG_RGBA);
|
|
label_channels(surf, X_STEP, X_OFFSET, Y_OFFSET);
|
|
area_selector(surf, X_SEL, Y_SEL);
|
|
if (wlan_area != area_off)
|
|
label_wlan_channels(surf, X_STEP, X_WLAN_OFFSET,
|
|
Y_WLAN_OFFSET);
|
|
|
|
SDL_UnlockSurface(surf);
|
|
SDL_UpdateRect(surf, 0, 0, 0, 0);
|
|
}
|
|
}
|