#![deny(unsafe_code)]
#![no_std]
#![no_main]

use arrayvec::ArrayString;
use core::fmt::Write;
use cortex_m::asm;
use cortex_m_rt::{entry, exception};
use embedded_graphics::{
    drawable::Drawable, fonts::Text, image::Image, pixelcolor::BinaryColor, pixelcolor::Rgb565,
    prelude::*, primitives::rectangle::Rectangle, style::PrimitiveStyleBuilder,
    style::TextStyleBuilder,
};
use embedded_hal::digital::v2::{InputPin, OutputPin};
use profont::{ProFont12Point, ProFont9Point};
use rtt_target::{rprintln, rtt_init_print};
use st7735_lcd::Orientation;
use stm32f1xx_hal::{
    delay::Delay,
    pac,
    prelude::*,
    qei::QeiOptions,
    spi::{Mode, Phase, Polarity, Spi},
    timer::Timer,
};
use tinybmp::Bmp;

mod max6675;

#[entry]
fn main() -> ! {
    rtt_init_print!();

    // Get access to the core peripherals from the cortex-m crate
    let cp = cortex_m::Peripherals::take().unwrap();
    // Get access to the device specific peripherals from the peripheral access crate
    let dp = pac::Peripherals::take().unwrap();

    // Take ownership over the raw flash and rcc devices and convert them into the corresponding
    // HAL structs
    let mut flash = dp.FLASH.constrain();

    let mut rcc = dp.RCC.constrain();

    // Freeze the configuration of all the clocks in the system and store the frozen frequencies in
    // `clocks`
    let clocks = rcc
        .cfgr
        .use_hse(8.mhz())
        .sysclk(72.mhz())
        .pclk1(36.mhz())
        .freeze(&mut flash.acr);

    // Acquire the GPIOC peripheral
    let mut gpioc = dp.GPIOC.split(&mut rcc.apb2);
    let mut gpioa = dp.GPIOA.split(&mut rcc.apb2);

    // Configure gpio C pin 13 as a push-pull output. The `crh` register is passed to the function
    // in order to configure the port. For pins 0-7, crl should be passed instead.
    let mut led = gpioc.pc13.into_push_pull_output(&mut gpioc.crh);

    let mut delay = Delay::new(cp.SYST, clocks);

    let gpiob = dp.GPIOB.split(&mut rcc.apb2);
    let mut afio = dp.AFIO.constrain(&mut rcc.apb2);

    let (_, _, pb4) = afio.mapr.disable_jtag(gpioa.pa15, gpiob.pb3, gpiob.pb4);

    let qei = Timer::tim3(dp.TIM3, &clocks, &mut rcc.apb1).qei(
        (pb4, gpiob.pb5),
        &mut afio.mapr,
        QeiOptions::default(),
    );

    // SPI1
    let sck = gpioa.pa5.into_alternate_push_pull(&mut gpioa.crl);
    let miso = gpioa.pa6;
    let mosi = gpioa.pa7.into_alternate_push_pull(&mut gpioa.crl);

    let mut disp_cs = gpioa.pa0.into_push_pull_output(&mut gpioa.crl);
    let mut max_cs = gpioa.pa9.into_push_pull_output(&mut gpioa.crh);

    let mut rst = gpioa.pa1.into_push_pull_output(&mut gpioa.crl);
    let mut dc = gpioa.pa4.into_push_pull_output(&mut gpioa.crl);
    let mut disp_led = gpioa.pa8.into_push_pull_output(&mut gpioa.crh);

    disp_led.set_high().unwrap();

    let mut spi = Spi::spi1(
        dp.SPI1,
        (sck, miso, mosi),
        &mut afio.mapr,
        Mode {
            polarity: Polarity::IdleLow,
            phase: Phase::CaptureOnFirstTransition,
        },
        16.mhz(),
        clocks,
        &mut rcc.apb2,
    );

    let mut disp = st7735_lcd::ST7735::new(spi, dc, rst, true, false, 160, 128);
    disp_cs.set_low().unwrap();
    disp.init(&mut delay).unwrap();
    disp.set_orientation(&Orientation::LandscapeSwapped)
        .unwrap();

    let style_black = PrimitiveStyleBuilder::new()
        .fill_color(Rgb565::BLACK)
        .build();
    Rectangle::new(Point::new(0, 0), Point::new(160, 128))
        .into_styled(style_black)
        .draw(&mut disp)
        .unwrap();

    let bmp = Bmp::from_slice(include_bytes!("logo.bmp")).unwrap();
    let image = Image::new(&bmp, Point::new(16, 0));
    image.draw(&mut disp).unwrap();

    let (_spi, _dc, _rst) = disp.release();
    spi = _spi;
    dc = _dc;
    rst = _rst;
    disp_cs.set_high().unwrap();

    let text_lager = TextStyleBuilder::new(ProFont12Point)
        .text_color(Rgb565::WHITE)
        .build();

    loop {
        let temp = max6675::read(&mut spi, &mut max_cs).unwrap();
        rprintln!("T: {}", temp);

        // Create a fixed buffer of length 12
        let mut buf = ArrayString::<[_; 10]>::new();
        // Output `Value: 12.35`
        write!(&mut buf, "T: {}", qei.count() / 4).expect("Failed to write to buffer");

        disp_cs.set_low().unwrap();
        let mut disp = st7735_lcd::ST7735::new(spi, dc, rst, true, false, 160, 128);

        Rectangle::new(Point::new(0, 100), Point::new(160, 128))
            .into_styled(style_black)
            .draw(&mut disp)
            .unwrap();

        Text::new(&buf, Point::new(60, 100))
            .into_styled(text_lager)
            .draw(&mut disp)
            .unwrap();

        let (_spi, _dc, _rst) = disp.release();
        spi = _spi;
        dc = _dc;
        rst = _rst;
        disp_cs.set_high().unwrap();

        delay.delay_ms(250u16);
    }
}

#[exception]
fn PendSV() {
    rprintln!("PendSV");
    panic!()
}

#[panic_handler]
fn panic(info: &core::panic::PanicInfo) -> ! {
    rprintln!("{}", info);
    exit()
}

fn exit() -> ! {
    loop {
        asm::bkpt() // halt = exit probe-run
    }
}