#![no_std]
#![no_main]

extern crate panic_halt;

use cortex_m_rt::entry;

use cortex_m::peripheral::Peripherals;

use stm32f4xx_hal::{
    delay::Delay,
    prelude::*,
    spi::{self, Spi},
    stm32,
};

use display_interface_spi::SPIInterfaceNoCS;
use ssd1320::Ssd1320z2;

#[entry]
fn main() -> ! {
    let p = stm32::Peripherals::take().unwrap();

    let cp = Peripherals::take().unwrap();

    let gpioa = p.GPIOA.split();
    let gpiob = p.GPIOB.split();
    let rcc = p.RCC.constrain();

    let clocks = rcc.cfgr.freeze();

    let mut led = gpioa.pa5.into_push_pull_output();
    led.set_low().ok();

    let sck = gpiob.pb3.into_alternate_af5();
    let miso = spi::NoMiso;
    let mosi = gpiob.pb5.into_alternate_af5();

    let dc = gpiob.pb4.into_push_pull_output();
    let mut res = gpiob.pb10.into_push_pull_output();
    let cs1 = gpiob.pb13.into_push_pull_output();
    let cs2 = gpiob.pb14.into_push_pull_output();

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

    let mode = spi::Mode {
        polarity: spi::Polarity::IdleLow,
        phase: spi::Phase::CaptureOnFirstTransition,
    };

    let spi = Spi::spi1(p.SPI1, (sck, miso, mosi), mode, 8_000_000.hz(), clocks);

    let iface = SPIInterfaceNoCS::new(spi, dc);

    let mut display = Ssd1320z2::new(iface, cs1, cs2);

    display.reset(&mut res, &mut delay);

    display.init();

    display.set_draw_area((190, 78), (190 + 83, 78 + 23));
    for _ in 0..(54 * 44) {
        display.draw(&[0x00]);
    }
    display.set_draw_area((192, 80), (192 + 79, 80 + 19));
    for i in 0..(50 * 40) {
        display.draw(&[(i % 4) as u8 | (((i >> 2) % 4) << 2) as u8]);
    }

    let mut select_figure = 0;
    loop {
        delay.delay_ms(500_u16);
        let (begin, end) = match select_figure {
            0 => {
                select_figure = 1;
                ((10, 10), (89, 89))
            }

            1 => {
                select_figure = 2;
                ((100, 10), (100 + 79, 89))
            }
            _ => {
                select_figure = 0;
                ((180 + 10, 10), (180 + 41, 41))
            }
        };
        display.set_draw_area(begin, end);
        for _ in 0..((end.1 - begin.1 + 1) / 16) {
            for _ in 0..((end.0 - begin.0 + 1) / 2) {
                display
                    .draw(&[0x77, 0x77, 0x77, 0x77, 0xff, 0xff, 0xff, 0xff])
                    .ok();
            }
            for _ in 0..((end.0 - begin.0 + 1) / 2) {
                display
                    .draw(&[0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00])
                    .ok();
            }
        }
        led.toggle().ok();
    }
}