#![no_std]

#[cfg(feature = "rt")]
pub use cortex_m_rt::entry;

pub use atsamd_hal as hal;
pub use hal::ehal;
pub use hal::pac;

use hal::clock::GenericClockController;
use hal::prelude::*;
use hal::sercom::v2::spi;
use hal::sercom::v2::uart::{self, BaudMode, Oversampling};
use hal::sercom::v2::{Sercom0, Sercom3, Sercom4};
use hal::sercom::I2CMaster5;
use hal::time::{Hertz, MegaHertz};

#[cfg(feature = "usb")]
use hal::usb::{usb_device::bus::UsbBusAllocator, UsbBus};

/// Definitions related to pins and pin aliases
pub mod pins {
    use super::hal;

    hal::bsp_pins!(
        PB09 {
            /// Pin 0, rx. Also analog input (A6)
            name: a6,
            aliases: {
                AlternateD: UartRx
            }
        },
        PB08 {
            /// Pin 1, tx. Also analog input (A7)
            name: a7,
            aliases: {
                AlternateD: UartTx
            }
        },
        PA28 {
            /// Pin 4, button A.
            name: d4,
        },
        PA14 {
            /// Pin 5, button B.
            name: d5,
        },
        PA15 {
            /// Pin 7, slide switch.
            name: d7,
        },
        PA30 {
            /// Pin 11, speaker enable.
            name: d11,
        },
        PA17 {
            /// Digital pin number 13, which is also attached to the red LED. PWM capable.
            name: d13,
            aliases: {
                PushPullOutput: RedLed
            }
        },
        PB02 {
            /// The I2C SDA. Also D2 and A5.
            name: a5,
            aliases: {
                AlternateD: Sda
            }
        },
        PB03 {
            /// The I2C SCL. Also D3 and A4
            name: a4,
            aliases: {
                AlternateD: Scl
            }
        },
        PB23 {
            /// The data line attached to the neopixel. Also D8.
            name: d8,
            aliases: {
                PushPullOutput: NeoPixel
            }
        },
        PA02 {
            /// The line attached to the speaker. Also D12 and A0.
            name: a0,
            aliases: {
                PushPullOutput: Speaker
            }
        },
        PA05 {
            /// The SPI SCK. Also D6 and A1
            name: a1,
            aliases: {
                AlternateD: Sck
            }
        },
        PA07 {
            /// The SPI MOSI. Also D10 and A3
            name: a3,
            aliases: {
                AlternateD: Mosi
            }
        },
        PA06 {
            /// The SPI MISO. Also D9 and A2
            name: a2,
            aliases: {
                AlternateD: Miso
            }
        },
        PA21 {
            /// The SCK pin attached to the on-board SPI flash
            name: flash_sck,
            aliases: {
                AlternateD: FlashSck
            }
        },
        PA20 {
            /// The MOSI pin attached to the on-board SPI flash
            name: flash_mosi,
            aliases: {
                AlternateD: FlashMosi
            }
        },
        PA16 {
            /// The MISO pin attached to the on-board SPI flash
            name: flash_miso,
            aliases: {
                AlternateD: FlashMiso
            }
        },
        PB22 {
            /// The CS pin attached to the on-board SPI flash
            name: flash_cs,
            aliases: {
                PushPullOutput: FlashCs
            }
        },
        PA00 {
            name: accel_sda,
        },
        PA01 {
            name: accel_scl,
        },
        PA24 {
            /// The USB D- pad
            name: usb_dm
            aliases: {
                AlternateG: UsbDm
            }
        }
        PA25 {
            /// The USB D+ pad
            name: usb_dp
            aliases: {
                AlternateG: UsbDp
            }
        }
    );
}
pub use pins::*;

/// SPI pads for the labelled SPI peripheral
///
/// You can use these pads with other, user-defined [`spi::Config`]urations.
pub type SpiPads = spi::Pads<Sercom0, Miso, Mosi, Sck>;

/// SPI master for the labelled SPI peripheral
///
/// This type implements [`FullDuplex<u8>`](ehal::spi::FullDuplex).
pub type Spi = spi::Spi<spi::Config<SpiPads>, spi::Duplex>;

/// Convenience for setting up the SPI bus on A1, A2, A3.
/// This powers up SERCOM0 and configures it for use as an
/// SPI Master in SPI Mode 0.
/// Unlike the `flash_spi_master` function, this
/// one does not accept a CS pin; configuring a pin for CS
/// is the responsibility of the caller, because it could be
/// any OutputPin, or even a pulled up line on the slave.
pub fn spi_master(
    clocks: &mut GenericClockController,
    baud: impl Into<Hertz>,
    sercom0: pac::SERCOM0,
    pm: &mut pac::PM,
    sck: impl Into<Sck>,
    mosi: impl Into<Mosi>,
    miso: impl Into<Miso>,
) -> Spi {
    let gclk0 = clocks.gclk0();
    let clock = clocks.sercom0_core(&gclk0).unwrap();
    let freq = clock.freq();
    let (miso, mosi, sck) = (miso.into(), mosi.into(), sck.into());
    let pads = spi::Pads::default().data_in(miso).data_out(mosi).sclk(sck);
    spi::Config::new(pm, sercom0, pads, freq)
        .baud(baud)
        .spi_mode(spi::MODE_0)
        .enable()
}

/// SPI pads for the flash chip
pub type FlashPads = spi::Pads<Sercom3, FlashMiso, FlashMosi, FlashSck>;

/// SPI master for the onboard SPI flash chip.
///
/// This type implements [`FullDuplex<u8>`](ehal::spi::FullDuplex).
pub type FlashSpi = spi::Spi<spi::Config<FlashPads>, spi::Duplex>;

/// Convenience for accessing the on-board SPI Flash device.
/// This powers up SERCOM3 and configures it for use as an
/// SPI Master.
pub fn flash_spi_master(
    clocks: &mut GenericClockController,
    sercom3: pac::SERCOM3,
    pm: &mut pac::PM,
    sck: impl Into<FlashSck>,
    mosi: impl Into<FlashMosi>,
    miso: impl Into<FlashMiso>,
    cs: impl Into<FlashCs>,
) -> (FlashSpi, FlashCs) {
    let gclk0 = clocks.gclk0();
    let clock = clocks.sercom3_core(&gclk0).unwrap();
    let freq = clock.freq();
    let (sck, mosi, miso, mut cs) = (sck.into(), mosi.into(), miso.into(), cs.into());
    let pads = spi::Pads::default().data_in(miso).data_out(mosi).sclk(sck);
    let spi = spi::Config::new(pm, sercom3, pads, freq)
        .baud(MegaHertz(48))
        .spi_mode(spi::MODE_0)
        .enable();

    cs.set_high().unwrap();

    (spi, cs)
}

/// I2C master for the labelled SDA & SCL pins
pub type I2C = I2CMaster5<Sda, Scl>;

/// Convenience for setting up the labelled SDA, SCL pins to
/// operate as an I2C master running at the specified frequency.
pub fn i2c_master(
    clocks: &mut GenericClockController,
    baud: impl Into<Hertz>,
    sercom5: pac::SERCOM5,
    pm: &mut pac::PM,
    sda: impl Into<Sda>,
    scl: impl Into<Scl>,
) -> I2C {
    let gclk0 = clocks.gclk0();
    let clock = &clocks.sercom5_core(&gclk0).unwrap();
    let baud = baud.into();
    let sda = sda.into();
    let scl = scl.into();
    I2CMaster5::new(clock, baud, sercom5, pm, sda, scl)
}

/// UART pads for the labelled RX & TX pins
pub type UartPads = uart::Pads<Sercom4, UartRx, UartTx>;

/// UART device for the labelled RX & TX pins
pub type Uart = uart::Uart<uart::Config<UartPads>, uart::Duplex>;

/// Convenience for setting up the labelled RX, TX pins to
/// operate as a UART device running at the specified baud.
pub fn uart(
    clocks: &mut GenericClockController,
    baud: impl Into<Hertz>,
    sercom4: pac::SERCOM4,
    pm: &mut pac::PM,
    uart_rx: impl Into<UartRx>,
    uart_tx: impl Into<UartTx>,
) -> Uart {
    let gclk0 = clocks.gclk0();
    let clock = &clocks.sercom4_core(&gclk0).unwrap();
    let baud = baud.into();
    let pads = uart::Pads::default().rx(uart_rx.into()).tx(uart_tx.into());
    uart::Config::new(pm, sercom4, pads, clock.freq())
        .baud(baud, BaudMode::Fractional(Oversampling::Bits16))
        .enable()
}

#[cfg(feature = "usb")]
/// Convenience function for setting up USB
pub fn usb_allocator(
    usb: pac::USB,
    clocks: &mut GenericClockController,
    pm: &mut pac::PM,
    dm: impl Into<UsbDm>,
    dp: impl Into<UsbDp>,
) -> UsbBusAllocator<UsbBus> {
    let gclk0 = clocks.gclk0();
    let clock = &clocks.usb(&gclk0).unwrap();
    let (dm, dp) = (dm.into(), dp.into());
    UsbBusAllocator::new(UsbBus::new(clock, pm, dm, dp, usb))
}