NanoVNA/ui.c

/*
 * Copyright (c) 2014-2015, TAKAHASHI Tomohiro (TTRFTECH) edy555@gmail.com
 * All rights reserved.
 *
 * This 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 3, or (at your option)
 * any later version.
 *
 * The software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ch.h"
#include "hal.h"
#include "nanovna.h"
#include <stdlib.h>

struct {
  int digit; /* 0~5 */
  int current_trace; /* 0..3 */
} uistat;


#define NO_EVENT					0
#define EVT_BUTTON_SINGLE_CLICK		0x01
#define EVT_BUTTON_DOUBLE_CLICK		0x02
#define EVT_BUTTON_DOWN_LONG		0x04
#define EVT_UP					0x10
#define EVT_DOWN				0x20
#define EVT_REPEAT				0x40

#define BUTTON_DOWN_LONG_TICKS		10000  /* 1sec */
#define BUTTON_DOUBLE_TICKS			5000   /* 500ms */
#define BUTTON_DEBOUNCE_TICKS		10

/* lever switch assignment */
#define BIT_UP1 	3
#define BIT_PUSH	2
#define BIT_DOWN1	1

#define READ_PORT() palReadPort(GPIOA)
#define BUTTON_MASK 0b1111

static uint16_t last_button = 0b0000;
static uint32_t last_button_down_ticks;
static uint8_t inhibit_until_release = FALSE;
static uint8_t last_button_event;
uint8_t operation_requested = FALSE;

int ui_status = FALSE;
int selection = 1;


static int btn_check(void)
{
    int cur_button = READ_PORT() & BUTTON_MASK;
	int changed = last_button ^ cur_button;
	int status = 0;
    uint32_t ticks = chVTGetSystemTime();
	if (changed & (1<<BIT_PUSH)) {
		if (ticks >= last_button_down_ticks + BUTTON_DEBOUNCE_TICKS) {
            if (cur_button & (1<<BIT_PUSH)) {
				// button pushed
              status |= EVT_BUTTON_SINGLE_CLICK;
              last_button_down_ticks = ticks;
			} else {
              // button released
              if (inhibit_until_release) {
                status = 0;
                inhibit_until_release = FALSE;
              }
			}
		}
	} else {
		// button unchanged
		if (cur_button & (1<<BIT_PUSH)
			&& ticks >= last_button_down_ticks + BUTTON_DOWN_LONG_TICKS) {
			status |= EVT_BUTTON_DOWN_LONG;
            inhibit_until_release = TRUE;
		}
	}

	if (cur_button & (1<<BIT_UP1)) {
		if (ticks >= last_button_down_ticks + BUTTON_DEBOUNCE_TICKS) {
          if (changed & (1<<BIT_UP1)) {
            status |= EVT_UP;
            last_button_down_ticks = ticks;
          }
          if (ticks >= last_button_down_ticks + BUTTON_DOWN_LONG_TICKS) {
            status |= EVT_UP;
          }
        }
    }
	if (cur_button & (1<<BIT_DOWN1)) {
		if (ticks >= last_button_down_ticks + BUTTON_DEBOUNCE_TICKS) {
          if (changed & (1<<BIT_DOWN1)) {
            status |= EVT_DOWN;
            last_button_down_ticks = ticks;
          }
          if (ticks >= last_button_down_ticks + BUTTON_DOWN_LONG_TICKS) {
            status |= EVT_DOWN;
          }
        }
    }
    last_button = cur_button;

	return status;
}

static int btn_wait_release(void)
{
  while (TRUE) {
    int cur_button = READ_PORT() & BUTTON_MASK;
    int changed = last_button ^ cur_button;
    uint32_t ticks = chVTGetSystemTime();
    int status = 0;
    if (changed) {
      // finished
      last_button = cur_button;
      return 0;
    }

    if (ticks - last_button_down_ticks >= BUTTON_DOWN_LONG_TICKS) {
      if (cur_button & (1<<BIT_DOWN1)) {
        status |= EVT_DOWN | EVT_REPEAT;
      }
      if (cur_button & (1<<BIT_UP1)) {
        status |= EVT_UP | EVT_REPEAT;
      }
      return status;
    }
  }
}

void
ui_digit(void)
{
    int count = 0;
    while (TRUE) {
        int status = btn_check();
        if (status & EVT_BUTTON_SINGLE_CLICK)
            break;
        if (status & EVT_UP && uistat.digit < 7)
            uistat.digit++;
        if (status & EVT_DOWN && uistat.digit > 0)
            uistat.digit--;
        if (count++ % 4 < 2) {
          //i2clcd_cmd(0x0e); // enable show-cursor flag
          //i2clcd_pos(7 - uistat.digit, 1);
        } else {
          //i2clcd_cmd(0x0c); // disable show-cursor flag
        }
        chThdSleepMilliseconds(100);
    }
}


// type of menu item 
enum {
  MT_NONE,
  MT_BLANK,
  MT_SUBMENU,
  MT_CALLBACK,
  MT_CANCEL,
  MT_CLOSE
};

typedef void (*menuaction_cb_t)(int item);


static void menu_move_back(void);


static void
menu_cal_cb(int item)
{
  switch (item) {
  case 0: // OPEN
    cal_collect(CAL_OPEN);
    break;
  case 1: // SHORT
    cal_collect(CAL_SHORT);
    break;
  case 2: // LOAD
    cal_collect(CAL_LOAD);
    break;
  case 3: // ISOLN
    cal_collect(CAL_ISOLN);
    break;
  case 4: // THRU
    cal_collect(CAL_THRU);
    break;
  }
  selection++;
  draw_cal_status();
}

static void
menu_caldone_cb(int item)
{
  cal_done();
  draw_cal_status();
  menu_move_back();
}

static void
menu_recall_cb(int item)
{
  if (item < 0 || item >= 5)
    return;
  if (caldata_recall(item) == 0) {
    ui_status = FALSE;
    ui_hide();
    set_sweep(freq_start, freq_stop);
    draw_cal_status();
  }
}

static void
menu_trace_cb(int item)
{
  if (item < 0 || item >= 4)
    return;
  uistat.current_trace = item;
  menu_move_back();
}

static void
menu_format_cb(int item)
{
  set_trace_type(uistat.current_trace, item);
  ui_status = FALSE;
  ui_hide();
}

static void
menu_format2_cb(int item)
{
  menu_format_cb(item + 5);
  ui_status = FALSE;
  ui_hide();
}

static void
menu_channel_cb(int item)
{
  if (item < 0 || item >= 2)
    return;
  set_trace_channel(uistat.current_trace, item);
  ui_status = FALSE;
  ui_hide();
}

static void 
choose_active_marker(void)
{
  int i;
  for (i = 0; i < 4; i++)
    if (markers[i].enabled) {
      active_marker = i;
      return;
    }
  active_marker = -1;
}

static void
menu_marker_cb(int item)
{
  if (item < 0 || item >= 4)
    return;

  if (active_marker == item) {
    markers[active_marker].enabled = FALSE;
    choose_active_marker();
  } else {
    active_marker = item;
    markers[active_marker].enabled = TRUE;
  }
  if (active_marker >= 0)
    redraw_marker(active_marker, TRUE);
  ui_status = FALSE;
  ui_hide();
}

typedef struct {
  uint8_t type;
  char *label;
  const void *reference;
} menuitem_t;

const menuitem_t menu_cal[] = {
  { MT_CALLBACK, "OPEN", menu_cal_cb },
  { MT_CALLBACK, "SHORT", menu_cal_cb },
  { MT_CALLBACK, "LOAD", menu_cal_cb },
  { MT_CALLBACK, "ISOLN", menu_cal_cb },
  { MT_CALLBACK, "THRU", menu_cal_cb },
  { MT_CALLBACK, "DONE", menu_caldone_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_trace[] = {
  { MT_CALLBACK, "0", menu_trace_cb },
  { MT_CALLBACK, "1", menu_trace_cb },
  { MT_CALLBACK, "2", menu_trace_cb },
  { MT_CALLBACK, "3", menu_trace_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_format2[] = {
  { MT_CALLBACK, "LINEAR", menu_format2_cb },
  { MT_CALLBACK, "SWR", menu_format2_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_format[] = {
  { MT_CALLBACK, "LOGMAG", menu_format_cb },
  { MT_CALLBACK, "PHASE", menu_format_cb },
  { MT_CALLBACK, "SMITH", menu_format_cb },
  { MT_CALLBACK, "ADMIT", menu_format_cb },
  { MT_CALLBACK, "POLAR", menu_format_cb },
  { MT_SUBMENU, "NEXT", menu_format2 },  
  //{ MT_CALLBACK, "LINEAR", menu_format_cb },
  //{ MT_CALLBACK, "SWR", menu_format_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_channel[] = {
  { MT_CALLBACK, "0", menu_channel_cb },
  { MT_CALLBACK, "1", menu_channel_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_display[] = {
  { MT_SUBMENU, "TRACE", menu_trace },
  { MT_SUBMENU, "FORMAT", menu_format },
  { MT_SUBMENU, "SCALE", menu_format },
  { MT_SUBMENU, "CHANNEL", menu_channel },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_marker[] = {
  { MT_CALLBACK, "1", menu_marker_cb },
  { MT_CALLBACK, "2", menu_marker_cb },
  { MT_CALLBACK, "3", menu_marker_cb },
  { MT_CALLBACK, "4", menu_marker_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_recall[] = {
  { MT_CALLBACK, "0", menu_recall_cb },
  { MT_CALLBACK, "1", menu_recall_cb },
  { MT_CALLBACK, "2", menu_recall_cb },
  { MT_CALLBACK, "3", menu_recall_cb },
  { MT_CALLBACK, "4", menu_recall_cb },
  { MT_CANCEL, "BACK", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

const menuitem_t menu_top[] = {
  { MT_SUBMENU, "CAL", menu_cal },
  { MT_SUBMENU, "DISPLAY", menu_display },
  { MT_SUBMENU, "MARKER", menu_marker },
  { MT_SUBMENU, "RECALL", menu_recall },
  { MT_CLOSE, "CLOSE", NULL },
  { MT_NONE, NULL, NULL } // sentinel
};

#define MENU_STACK_DEPTH_MAX 4
uint8_t menu_current_level = 0;
const menuitem_t *menu_stack[4] = {
  menu_top, NULL, NULL, NULL
};

static void
ensure_selection(void)
{
  const menuitem_t *menu = menu_stack[menu_current_level];
  int i;
  for (i = 0; menu[i].type != MT_NONE; i++)
    ;
  if (selection >= i)
    selection = i-1;
}

static void menu_move_back(void)
{
  if (menu_current_level == 0)
    return;
  menu_current_level--;
  ensure_selection();
  erase_buttons();
}

static void menu_move_top(void)
{
  if (menu_current_level == 0)
    return;
  menu_current_level = 0;
  ensure_selection();
  erase_buttons();
}

void menu_invoke(int selection)
{
  const menuitem_t *menu = menu_stack[menu_current_level];
  menu = &menu[selection];

  switch (menu->type) {
  case MT_NONE:
  case MT_BLANK:
  case MT_CLOSE:
    ui_status = FALSE;
    ui_hide();
    break;

  case MT_CANCEL:
    menu_move_back();
    break;

  case MT_CALLBACK: {
    menuaction_cb_t cb = (menuaction_cb_t)menu->reference;
    if (cb == NULL)
      return;
    (*cb)(selection);
    break;
  }

  case MT_SUBMENU:
    if (menu_current_level >= 3)
      break;
    menu_current_level++;
    menu_stack[menu_current_level] = (const menuitem_t*)menu->reference;
    erase_buttons();
    selection = 0;
    break;
  }
}


void
draw_menu_buttons(const menuitem_t *menu)
{
  int i = 0;
  for (i = 0; i < 7; i++) {
    if (menu[i].type == MT_NONE)
      break;
    if (menu[i].type == MT_BLANK) 
      continue;
    int y = 32*i;
    uint16_t bg = 0xffff;
    if (i == selection)
      bg = 0x7777;
    ili9341_fill(320-60, y, 60, 30, bg);
    ili9341_drawstring_5x7(menu[i].label, 320-54, y+12, 0x0000, bg);
  }
}

const char *menu_items[] = {
  "OPEN", "SHORT", "LOAD", "ISOLN", "THRU", NULL, "DONE"
};

void
draw_buttons(void)
{
#if 1
  draw_menu_buttons(menu_stack[menu_current_level]);
#else
  int i = 0;
  for (i = 0; i < 7; i++) {
    int y = 32*i;
    uint16_t bg = 0xffff;
    if (i == selection)
      bg = 0x7777;
    if (menu_items[i] != NULL) {
      ili9341_fill(320-60, y, 60, 30, bg);
      ili9341_drawstring_5x7(menu_items[i], 320-54, y+12, 0x0000, bg);
    }
  }
#endif
}

void
erase_buttons(void)
{
  uint16_t bg = 0;
  ili9341_fill(320-60, 0, 60, 32*7, bg);
}

void
ui_show(void)
{
  area_width = WIDTH - (64-14-4);
  area_height = HEIGHT;
  draw_buttons();
}

void
ui_hide(void)
{
  area_width = WIDTH;
  area_height = HEIGHT;
  erase_buttons();
  force_draw_cells();
}

void
ui_process(void)
{
  if (!operation_requested)
    return;
  
  int status = btn_check();
  int n;
  if (status != 0) {
    if (status & EVT_BUTTON_SINGLE_CLICK) {
      if (ui_status) {
        menu_invoke(selection);
      } else {
        ui_status = TRUE;
      }
      if (ui_status)
        ui_show();
    } else {
      if (ui_status) {
        if (status & EVT_UP
            && menu_stack[menu_current_level][selection+1].type != MT_NONE) {
          selection++;
          draw_buttons();
        }
        if (status & EVT_DOWN
            && selection > 0) {
          selection--;
          draw_buttons();
        }
      } else {
        do {
          if (active_marker >= 0 && markers[active_marker].enabled) {
            if ((status & EVT_DOWN) && markers[active_marker].index > 0) {
              markers[active_marker].index--;
              redraw_marker(active_marker, FALSE);
            }
            if ((status & EVT_UP) && markers[active_marker].index < 100) {
              markers[active_marker].index++;
              redraw_marker(active_marker, FALSE);
            }
          }
          status = btn_wait_release();
        } while (status != 0);
        if (active_marker >= 0)
          redraw_marker(active_marker, TRUE);
      }
    }
  }
  operation_requested = FALSE;
}

/* Triggered when the button is pressed or released. The LED4 is set to ON.*/
static void extcb1(EXTDriver *extp, expchannel_t channel) {
  (void)extp;
  (void)channel;
  operation_requested = TRUE;
  //cur_button = READ_PORT() & BUTTON_MASK;
}

static const EXTConfig extcfg = {
  {
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_RISING_EDGE | EXT_CH_MODE_AUTOSTART | EXT_MODE_GPIOA, extcb1},
    {EXT_CH_MODE_RISING_EDGE | EXT_CH_MODE_AUTOSTART | EXT_MODE_GPIOA, extcb1},
    {EXT_CH_MODE_RISING_EDGE | EXT_CH_MODE_AUTOSTART | EXT_MODE_GPIOA, extcb1},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL},
    {EXT_CH_MODE_DISABLED, NULL}
  }
};

void
ui_init()
{
  /*
   * Activates the EXT driver 1.
   */
  extStart(&EXTD1, &extcfg);
}