#![no_main]
#![no_std]

extern crate cortex_m;
extern crate cortex_m_rt;
extern crate embedded_hal;
extern crate panic_halt;

extern crate stm32f030_hal as hal;

extern crate ina260;
extern crate numtoa;

use hal::i2c::*;
use hal::prelude::*;
use hal::serial::*;
use hal::stm32;

use numtoa::NumToA;

use ina260::*;

use core::fmt::Write;
use cortex_m_rt::entry;

#[entry]
fn main() -> ! {
    if let Some(p) = stm32::Peripherals::take() {
        let gpioa = p.GPIOA.split();
        let mut clocks = p.RCC.constrain().cfgr.freeze();

        /* Initialise serial pins */
        let tx = gpioa.pa2.into_alternate_af1();
        let rx = gpioa.pa3.into_alternate_af1();

        /* Setup serial port */
        let serial = Serial::usart1(p.USART1, (tx, rx), 115_200.bps(), clocks);
        let (mut tx, mut _rx) = serial.split();

        /* Initialise I2C pins */
        let scl = gpioa
            .pa9
            .into_alternate_af4()
            .internal_pull_up(true)
            .set_open_drain();
        let sda = gpioa
            .pa10
            .into_alternate_af4()
            .internal_pull_up(true)
            .set_open_drain();

        /* Setup I2C1 */
        let mut i2c = I2c::i2c1(p.I2C1, (scl, sda), 1.khz());
        let mut ina260 = INA260::new(i2c, 0x40).unwrap();

        /* Endless loop */
        loop {
            let mut buffer = [0u8; 10];

            /* Read and print voltage */
            let voltage = ina260.voltage().unwrap();
            let _ = tx.write_str(unsafe {
                core::str::from_utf8_unchecked(voltage.numtoa(10, &mut buffer))
            });
            let _ = tx.write_str("uV\n\r");

            /* Read and print current */
            let current = ina260.current().unwrap();
            let _ = tx.write_str(unsafe {
                core::str::from_utf8_unchecked(current.numtoa(10, &mut buffer))
            });
            let _ = tx.write_str("uA\n\r");
        }
    }

    loop {
        continue;
    }
}