ProjektMaus/src/main.rs

#![no_std]
#![no_main]

extern crate alloc;

mod custom_hid;
mod mouse_hid;
mod mouse_sensor;
mod serial;

use alloc::boxed::Box;
use custom_hid::CustomHID;
use mouse_hid::MouseHID;

// Ensure we halt the program on panic (if we don't mention this crate it won't
// be linked)
use panic_halt as _;

use rp_pico as bsp;

use bsp::{
    entry,
    hal::{self, pac, prelude::*},
    XOSC_CRYSTAL_FREQ,
};

use crate::mouse_sensor::MouseSensor;

const SERIAL: bool = true;

pub trait MouseInfoSender {
    fn send(&mut self, x: i8, y: i8, buttons: u8, wheel: i8, pan: i8);
}

#[entry]
fn main() -> ! {
    // Grab our singleton objects
    let mut pac = pac::Peripherals::take().unwrap();
    let core = pac::CorePeripherals::take().unwrap();

    // Set up the watchdog driver - needed by the clock setup code
    let mut watchdog = hal::Watchdog::new(pac.WATCHDOG);

    // Configure the clocks
    let clocks = hal::clocks::init_clocks_and_plls(
        XOSC_CRYSTAL_FREQ,
        pac.XOSC,
        pac.CLOCKS,
        pac.PLL_SYS,
        pac.PLL_USB,
        &mut pac.RESETS,
        &mut watchdog,
    )
    .ok()
    .unwrap();

    let peripheral_clock = clocks.peripheral_clock.freq();

    let mut hid: Box<dyn MouseInfoSender> = if SERIAL {
        Box::new(CustomHID::new(
            pac.USBCTRL_REGS,
            pac.USBCTRL_DPRAM,
            &mut pac.RESETS,
            clocks,
            10,
        ))
    } else {
        Box::new(MouseHID::new(
            pac.USBCTRL_REGS,
            pac.USBCTRL_DPRAM,
            &mut pac.RESETS,
            core,
            clocks,
        ))
    };

    // The single-cycle I/O block controls our GPIO pins
    let sio = hal::Sio::new(pac.SIO);

    // Set the pins up according to their function on this particular board
    let pins = rp_pico::Pins::new(
        pac.IO_BANK0,
        pac.PADS_BANK0,
        sio.gpio_bank0,
        &mut pac.RESETS,
    );

    let mut mouse_sensor = MouseSensor::new(pins, &mut pac.RESETS, pac.SPI0, peripheral_clock);

    loop {
        if let Some((x, y)) = mouse_sensor.read_movement() {
            hid.send(x, y, 0, 0, 0);
        }
    }
}