main.c

/**
 * Copyright (c) 2022 Nicolai Electronics
 *
 * SPDX-License-Identifier: MIT
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "bsp/board.h"
#include "hardware.h"
#include "hardware/adc.h"
#include "hardware/irq.h"
#include "hardware/pwm.h"
#include "hardware/structs/watchdog.h"
#include "hardware/uart.h"
#include "hardware/watchdog.h"
#include "i2c_peripheral.h"
#include "lcd.h"
#include "nec_transmit.h"
#include "pico/bootrom.h"
#include "pico/stdlib.h"
#include "tusb.h"
#include "uart_task.h"
#include "usb_descriptors.h"
#include "webusb_task.h"
#include "ws2812.h"

#ifdef PICO_PANIC_FUNCTION
#define CRASH_INDICATION_MAGIC 0xFA174200

void __attribute__((noreturn)) __printflike(1, 0) custom_panic(const char *fmt, ...) {
    hw_clear_bits(&watchdog_hw->ctrl, WATCHDOG_CTRL_ENABLE_BITS);
    watchdog_hw->scratch[5] = CRASH_INDICATION_MAGIC;
    watchdog_hw->scratch[6] = ~CRASH_INDICATION_MAGIC;
    watchdog_reboot(0, 0, 0);
    while (1) {
        tight_loop_contents();
        asm("");
    }
}

void check_crashed() {
    bool crashed = (watchdog_hw->scratch[5] == CRASH_INDICATION_MAGIC) && (watchdog_hw->scratch[6] == ~CRASH_INDICATION_MAGIC);
    i2c_set_crash_debug_state(crashed, false);
}
#else  // Debug firmware has the default panic handler to allow for debugging
void check_crashed() { i2c_set_crash_debug_state(false, true); }
#endif

int main(void) {
    board_init();
    tusb_init();
    setup_uart();

    gpio_init(BATT_CHRG_PIN);
    gpio_set_dir(BATT_CHRG_PIN, false);

    adc_init();
    adc_gpio_init(ANALOG_VBAT_PIN);
    adc_gpio_init(ANALOG_VUSB_PIN);

    int ir_statemachine = nec_tx_init(IR_PIO, IR_PIN);
    if (ir_statemachine < 0) panic("Failed to init IR");

    lcd_init();

    setup_i2c_registers(ir_statemachine);
    check_crashed();  // Populate the crash & debug state register
    setup_i2c_peripheral(I2C_SYSTEM, I2C_SYSTEM_SDA_PIN, I2C_SYSTEM_SCL_PIN, 0x17, 400000, i2c_slave_handler);
    esp32_reset(false);  // Reset ESP32 to normal mode

    ws2812_setup();

    while (1) {
        tud_task();
        uart_task();
        i2c_task();
        webusb_task();
    }

    return 0;
}

// Invoked when device is mounted
void tud_mount_cb(void) { i2c_usb_set_mounted(true); }

// Invoked when device is unmounted
void tud_umount_cb(void) { i2c_usb_set_mounted(false); }

// Invoked when usb bus is suspended
void tud_suspend_cb(bool remote_wakeup_en) { i2c_usb_set_suspended(true, remote_wakeup_en); }

// Invoked when usb bus is resumed
void tud_resume_cb(void) { i2c_usb_set_suspended(false, false); }