NanoVNA/main.c

#include "ch.h"
#include "hal.h"
#include "usbcfg.h"
#include "si5351.h"
#include "nanovna.h"

#include <chprintf.h>
#include <shell.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

RTCDateTime timespec;


static const I2CConfig i2ccfg = {
  0x00300506, //voodoo magic 400kHz @ HSI 8MHz
  //0x00902025, //voodoo magic
  //0x00420F13,  // 100kHz @ 72MHz
  0,
  0
};

void I2CWrite(int addr, uint8_t d0, uint8_t d1)
{
    uint8_t buf[] = { d0, d1 };
    i2cAcquireBus(&I2CD1);
    (void)i2cMasterTransmitTimeout(&I2CD1, addr, buf, 2, NULL, 0, 1000);
    i2cReleaseBus(&I2CD1);
}

int I2CRead(int addr, uint8_t d0)
{
    uint8_t buf[] = { d0 };
    i2cAcquireBus(&I2CD1);
    i2cMasterTransmitTimeout(&I2CD1, addr, buf, 1, buf, 1, 1000);
    i2cReleaseBus(&I2CD1);
    return buf[0];
}

void scan_lcd(void);

static MUTEX_DECL(mutex);



static THD_WORKING_AREA(waThread1, 512);
static THD_FUNCTION(Thread1, arg)
{
    (void)arg;

    chRegSetThreadName("blink");

    palSetPadMode(GPIOC, 13, PAL_MODE_OUTPUT_PUSHPULL);
    while (1)
    {
#if 0
      systime_t time = 500;
      if (serusbcfg.usbp->state != USB_ACTIVE)
        palClearPad(GPIOC, 13);
      chThdSleepMilliseconds(time);
      palSetPad(GPIOC, 13);
      chThdSleepMilliseconds(time);
#else
      chMtxLock(&mutex);
      scan_lcd();
      chMtxUnlock(&mutex);
#endif
    }
}

void
pause_sweep(void)
{
  chMtxLock(&mutex);
}

void
resume_sweep(void)
{
  chMtxUnlockAll();
}

static void cmd_pause(BaseSequentialStream *chp, int argc, char *argv[])
{
    (void)chp;
    (void)argc;
    (void)argv;
    pause_sweep();
}

static void cmd_resume(BaseSequentialStream *chp, int argc, char *argv[])
{
    (void)chp;
    (void)argc;
    (void)argv;
    resume_sweep();
}

static void cmd_reset(BaseSequentialStream *chp, int argc, char *argv[])
{
    (void)argc;
    (void)argv;

    chprintf(chp, "Performing reset\r\n");

    rccEnableWWDG(FALSE);

    WWDG->CFR = 0x60;
    WWDG->CR = 0xff;

    while (1)
	;
}


int32_t frequency_offset = 5000;
int32_t frequency = 10000000;
uint8_t drive_strength = SI5351_CLK_DRIVE_STRENGTH_2MA;

int set_frequency(int freq)
{
#if 0
    si5351_set_frequency(0, freq + frequency_offset);
    si5351_set_frequency(1, freq);
    frequency = freq;
#else
    int delay;
    delay = si5351_set_frequency_with_offset(freq, frequency_offset, drive_strength);
    frequency = freq;
    return delay;
#endif
}

static void cmd_offset(BaseSequentialStream *chp, int argc, char *argv[])
{
    if (argc != 1) {
        chprintf(chp, "usage: offset {frequency offset(Hz)}\r\n");
        return;
    }
    frequency_offset = atoi(argv[0]);
    set_frequency(frequency);
}


static void cmd_freq(BaseSequentialStream *chp, int argc, char *argv[])
{
    int freq;
    pause_sweep();
    if (argc != 1) {
        chprintf(chp, "usage: freq {frequency(Hz)}\r\n");
        return;
    }
    freq = atoi(argv[0]);
    set_frequency(freq);
}

static void cmd_power(BaseSequentialStream *chp, int argc, char *argv[])
{
    if (argc != 1) {
        chprintf(chp, "usage: power {0-3}\r\n");
        return;
    }
    drive_strength = atoi(argv[0]);
    set_frequency(frequency);
}



static void cmd_time(BaseSequentialStream *chp, int argc, char *argv[])
{
    (void)argc;
    (void)argv;
    rtcGetTime(&RTCD1, &timespec);
    chprintf(chp, "%d/%d/%d %d\r\n", timespec.year+1980, timespec.month, timespec.day, timespec.millisecond);
}


static const DACConfig dac1cfg1 = {
  //init:         2047U,
  init:         1922U,
  datamode:     DAC_DHRM_12BIT_RIGHT
};

static void cmd_dac(BaseSequentialStream *chp, int argc, char *argv[])
{
    int value;
    if (argc != 1) {
        chprintf(chp, "usage: dac {value(0-4095)}\r\n");
        return;
    }
    value = atoi(argv[0]);
    dacPutChannelX(&DACD2, 0, value);
}




static struct {
  int16_t rms[2];
  int16_t ave[2];
  int callback_count;

  int32_t last_counter_value;
  int32_t interval_cycles;
  int32_t busy_cycles;
} stat;

int16_t rx_buffer[AUDIO_BUFFER_LEN * 2];

int16_t dump_buffer[AUDIO_BUFFER_LEN];
volatile int16_t wait_count = 0;
int16_t dump_selection = 0;

int16_t dsp_disabled = FALSE;
float measured[101][4];
uint32_t frequencies[101];


void i2s_end_callback(I2SDriver *i2sp, size_t offset, size_t n)
{
#if PORT_SUPPORTS_RT
  int32_t cnt_s = port_rt_get_counter_value();
  int32_t cnt_e;
#endif
  int16_t *p = &rx_buffer[offset];
  (void)i2sp;
  (void)n;
  //palClearPad(GPIOC, GPIOC_LED);

  if (!dsp_disabled)
    dsp_process(p, n);

  if (wait_count > 0) {
    if (dump_selection == 1)
      p = samp_buf;
    else if (dump_selection == 2)
      p = ref_buf;
    else if (dump_selection == 3)
      p = refiq_buf;
    if (wait_count == 1)
      memcpy(dump_buffer, p, sizeof dump_buffer);
    --wait_count;
  }

#if PORT_SUPPORTS_RT
  cnt_e = port_rt_get_counter_value();
  stat.interval_cycles = cnt_s - stat.last_counter_value;
  stat.busy_cycles = cnt_e - cnt_s;
  stat.last_counter_value = cnt_s;
#endif
  stat.callback_count++;
  //palSetPad(GPIOC, GPIOC_LED);
}

static const I2SConfig i2sconfig = {
  NULL, // TX Buffer
  rx_buffer, // RX Buffer
  AUDIO_BUFFER_LEN * 2,
  NULL, // tx callback
  i2s_end_callback, // rx callback
  0, // i2scfgr
  2 // i2spr
};

static void cmd_data(BaseSequentialStream *chp, int argc, char *argv[])
{
  int i, j;
  int len;

  pause_sweep();
  if (argc == 1)
    dump_selection = atoi(argv[0]);

  wait_count = 3;
  //palClearPad(GPIOC, GPIOC_LED);
  while (wait_count)
    ;
  len = AUDIO_BUFFER_LEN;
  if (dump_selection == 1 || dump_selection == 2)
    len /= 2;
  for (i = 0; i < len; ) {
    for (j = 0; j < 16; j++, i++) {
      chprintf(chp, "%04x ", 0xffff & (int)dump_buffer[i]);
    }
    chprintf(chp, "\r\n");
  }
  //palSetPad(GPIOC, GPIOC_LED);
}

static void cmd_gamma(BaseSequentialStream *chp, int argc, char *argv[])
{
  float gamma[2];
  (void)argc;
  (void)argv;
  
  pause_sweep();
  wait_count = 4;
  while (wait_count)
    ;
  dsp_disabled = TRUE;
  calclate_gamma(gamma);
  dsp_disabled = FALSE;

  chprintf(chp, "%d %d\r\n", gamma[0], gamma[1]);
}


int32_t freq_start = 1000000;
int32_t freq_stop = 300000000;
int16_t sweep_points = 101;

static void cmd_scan(BaseSequentialStream *chp, int argc, char *argv[])
{
  float gamma[2];
  int i;
  int32_t freq, step;
  int delay;
  (void)argc;
  (void)argv;

  pause_sweep();
  freq = freq_start;
  step = (freq_stop - freq_start) / (sweep_points-1);
  delay = set_frequency(freq);
  delay += 2;
  for (i = 0; i < sweep_points; i++) {
    freq = freq + step;
    wait_count = delay;
    while (wait_count)
      ;
    //dsp_disabled = TRUE;
    __disable_irq();
    delay = set_frequency(freq);
    palClearPad(GPIOC, GPIOC_LED);
    calclate_gamma(gamma);
    palSetPad(GPIOC, GPIOC_LED);
    //dsp_disabled = FALSE;
    __enable_irq();
    chprintf(chp, "%d %d\r\n", gamma[0], gamma[1]);
  }
}

void scan_lcd(void)
{
  int i;
  int delay;
  int first = TRUE;

  delay = set_frequency(frequencies[0]);
  delay += 2;
  for (i = 0; i < sweep_points; i++) {
    wait_count = delay;
    tlv320aic3204_select_in3();
    while (wait_count)
      ;
    palClearPad(GPIOC, GPIOC_LED);
    __disable_irq();
    calclate_gamma(&measured[i][0]);
    __enable_irq();

    tlv320aic3204_select_in1();
    wait_count = 3;
    while (wait_count)
      ;
    __disable_irq();
    calclate_gamma(&measured[i][2]);
    __enable_irq();

    delay = set_frequency(frequencies[(i+1)%sweep_points]);
#if 0
    sweep_plot(frequencies[i], first, measured[i]);
    first = FALSE;
#endif
    palSetPad(GPIOC, GPIOC_LED);
  }
#if 1
  for (i = 0; i < sweep_points; i++) {
    sweep_plot(frequencies[i], first, measured[i]);
    first = FALSE;
  }  
#endif
  sweep_tail();
  polar_plot(measured);
}

static void cmd_scan_lcd(BaseSequentialStream *chp, int argc, char *argv[])
{
  (void)chp;
  (void)argc;
  (void)argv;
  pause_sweep();
  scan_lcd();
}


void
set_frequencies(void)
{
  int i;
  int32_t span = (freq_stop - freq_start)/100;
  for (i = 0; i < sweep_points; i++)
    frequencies[i] = freq_start + span * i / (sweep_points - 1) * 100;
}

static void cmd_sweep(BaseSequentialStream *chp, int argc, char *argv[])
{
  if (argc == 0) {
    chprintf(chp, "%d %d %d\r\n", freq_start, freq_stop, sweep_points);
    return;
  } else if (argc > 3) {
    chprintf(chp, "usage: sweep {start(Hz)} [stop] [points]\r\n");
    return;
  }
  if (argc >= 1) {
    int32_t x = atoi(argv[0]);
    if (x < 300000) {
      chprintf(chp, "bad parameter\r\n");
      return;
    }
    freq_start = x;
  }
  if (argc >= 2) {
    int32_t x = atoi(argv[1]);
    if (x < 300000 || x <= freq_start) {
      chprintf(chp, "bad parameter\r\n");
      return;
    }
    freq_stop = x;
  }
  if (argc >= 3) {
    int32_t x = atoi(argv[2]);
    if (x < 1 || x > 1601) {
      chprintf(chp, "bad parameter\r\n");
      return;
    }
    sweep_points = x;
  }

  set_frequencies();
  set_sweep(freq_start, freq_stop);
}

static void cmd_test(BaseSequentialStream *chp, int argc, char *argv[])
{
  int i;
  (void)chp;
  (void)argc;
  (void)argv;

  pause_sweep();
#if 0
  for (i = 0; i < 100; i++) {
    palClearPad(GPIOC, GPIOC_LED);
    set_frequency(10000000);
    palSetPad(GPIOC, GPIOC_LED);
    chThdSleepMilliseconds(50);

    palClearPad(GPIOC, GPIOC_LED);
    set_frequency(90000000);
    palSetPad(GPIOC, GPIOC_LED);
    chThdSleepMilliseconds(50);
  }
#endif

#if 1
  int mode = 0;
  if (argc >= 1)
    mode = atoi(argv[0]);

  for (i = 0; i < 20; i++) {
    palClearPad(GPIOC, GPIOC_LED);
    ili9341_test(mode);
    palSetPad(GPIOC, GPIOC_LED);
    chThdSleepMilliseconds(50);
  }
#endif
}

static void cmd_gain(BaseSequentialStream *chp, int argc, char *argv[])
{
  int rvalue;
  int lvalue = 0;
  if (argc != 1 && argc != 2) {
    chprintf(chp, "usage: gain {lgain(0-95)} [rgain(0-95)]\r\n");
    return;
  }
  rvalue = atoi(argv[0]);
  if (argc == 2) 
    lvalue = atoi(argv[1]);
  tlv320aic3204_set_gain(lvalue, rvalue);
}

static void cmd_port(BaseSequentialStream *chp, int argc, char *argv[])
{
  int port;
  if (argc != 1) {
    chprintf(chp, "usage: port {0:TX 1:RX}\r\n");
    return;
  }
  port = atoi(argv[0]);
  if (port)
    tlv320aic3204_select_in1();
  else
    tlv320aic3204_select_in3(); // default
}

static void cmd_stat(BaseSequentialStream *chp, int argc, char *argv[])
{
  int16_t *p = &rx_buffer[0];
  int32_t acc0, acc1;
  int32_t ave0, ave1;
  int32_t count = AUDIO_BUFFER_LEN;
  int i;
  (void)argc;
  (void)argv;
  acc0 = acc1 = 0;
  for (i = 0; i < AUDIO_BUFFER_LEN*2; i += 2) {
    acc0 += p[i];
    acc1 += p[i+1];
  }
  ave0 = acc0 / count;
  ave1 = acc1 / count;
  acc0 = acc1 = 0;
  for (i = 0; i < AUDIO_BUFFER_LEN*2; i += 2) {
    acc0 += (p[i] - ave0)*(p[i] - ave0);
    acc1 += (p[i+1] - ave1)*(p[i+1] - ave1);
  }
  stat.rms[0] = sqrtf(acc0 / count);
  stat.rms[1] = sqrtf(acc1 / count);
  stat.ave[0] = ave0;
  stat.ave[1] = ave1;

  chprintf(chp, "average: %d %d\r\n", stat.ave[0], stat.ave[1]);
  chprintf(chp, "rms: %d %d\r\n", stat.rms[0], stat.rms[1]);
  chprintf(chp, "callback count: %d\r\n", stat.callback_count);
  chprintf(chp, "interval cycle: %d\r\n", stat.interval_cycles);
  chprintf(chp, "busy cycle: %d\r\n", stat.busy_cycles);
  chprintf(chp, "load: %d\r\n", stat.busy_cycles * 100 / stat.interval_cycles);
}





#define SHELL_WA_SIZE THD_WORKING_AREA_SIZE(2048)

static const ShellCommand commands[] =
{
    { "reset", cmd_reset },
    { "freq", cmd_freq },
    { "offset", cmd_offset },
    { "time", cmd_time },
    { "dac", cmd_dac },
    { "data", cmd_data },
    { "port", cmd_port },
    { "stat", cmd_stat },
    { "gain", cmd_gain },
    { "power", cmd_power },
    { "gamma", cmd_gamma },
    { "scan", cmd_scan },
    { "sweep", cmd_sweep },
    { "test", cmd_test },
    { "plot", cmd_scan_lcd },
    { "pause", cmd_pause },
    { "resume", cmd_resume },
    { NULL, NULL }
};

static const ShellConfig shell_cfg1 =
{
    (BaseSequentialStream *)&SDU1,
    commands
};

int main(void)
{
    halInit();
    chSysInit();

    chMtxObjectInit(&mutex);

    /*
     * Starting DAC1 driver, setting up the output pin as analog as suggested
     * by the Reference Manual.
     */
    //palSetPadMode(GPIOA, 5, PAL_MODE_INPUT_ANALOG);
    //palSetPadMode(GPIOA, 5, PAL_MODE_OUTPUT_PUSHPULL);
    //palSetPadMode(GPIOA, 5, PAL_MODE_INPUT);
    dacStart(&DACD2, &dac1cfg1);

    //palSetPadMode(GPIOB, 8, PAL_MODE_ALTERNATE(1) | PAL_STM32_OTYPE_OPENDRAIN);
    //palSetPadMode(GPIOB, 9, PAL_MODE_ALTERNATE(1) | PAL_STM32_OTYPE_OPENDRAIN);
    i2cStart(&I2CD1, &i2ccfg);
    si5351_init();

    // MCO on PA8
    //palSetPadMode(GPIOA, 8, PAL_MODE_ALTERNATE(0));
  /*
   * Initializes a serial-over-USB CDC driver.
   */
    sduObjectInit(&SDU1);
    sduStart(&SDU1, &serusbcfg);

  /*
   * Activates the USB driver and then the USB bus pull-up on D+.
   * Note, a delay is inserted in order to not have to disconnect the cable
   * after a reset.
   */
    usbDisconnectBus(serusbcfg.usbp);
    chThdSleepMilliseconds(100);
    usbStart(serusbcfg.usbp, &usbcfg);
    usbConnectBus(serusbcfg.usbp);

  /*
   * SPI LCD Initialize
   */
  ili9341_init();
  set_sweep(freq_start, freq_stop);
  redraw();

  /*
   */
  set_frequencies();

  /*
   * I2S Initialize
   */
  tlv320aic3204_init();
  i2sInit();
  i2sObjectInit(&I2SD2);
  i2sStart(&I2SD2, &i2sconfig);
  i2sStartExchange(&I2SD2);

  /*
   * Shell manager initialization.
   */
    shellInit();

    chThdCreateStatic(waThread1, sizeof(waThread1), NORMALPRIO, Thread1, NULL);

    //set_frequency(10000000);

    while (1)
    {
    if (SDU1.config->usbp->state == USB_ACTIVE) {
      //palSetPad(GPIOC, GPIOC_LED);
      thread_t *shelltp = chThdCreateFromHeap(NULL, SHELL_WA_SIZE,
                                              "shell", NORMALPRIO + 1,
                                              shellThread, (void *)&shell_cfg1);
      chThdWait(shelltp);               /* Waiting termination.             */
      //palClearPad(GPIOC, GPIOC_LED);
    }
	chThdSleepMilliseconds(1000);
    }
}