//! Nova GPU Driver

mod gpu;
mod bios;
mod fwsec;

use crate::gpu::CardType;
use alloc::boxed::Box;
use core::{
    format_args,
    pin::Pin,
};
use kernel::{
    bindings,
    c_str,
    driver,
    error::code::*,
    firmware::{Firmware},
    io_mem::IoMem,
    pci, pci::define_pci_id_table,
    prelude::*,
};

use gpu::Gpu;
struct NovaDevice;

/// BAR Size
pub(crate) const BAR_SIZE: usize = 16777216;

impl pci::Driver for NovaDevice {

    define_pci_id_table! {
        (),
        [ (pci::DeviceId::new(bindings::PCI_VENDOR_ID_NVIDIA, bindings::PCI_ANY_ID as u32), None) ]
    }

    fn probe(dev: &mut pci::Device, _id_info: Option<&Self::IdInfo>) -> Result {
        pr_info!("probe()\n");

        dev.enable_device_mem()?;
        dev.set_master();

        let _bars = dev.select_bars(bindings::IORESOURCE_MEM.into());

        let res = dev.take_resource(0).ok_or(ENXIO)?;
        let bar = unsafe { IoMem::<BAR_SIZE>::try_new(res) }?;

        let boot0 = u64::from_le(bar.readq(0));

        if boot0 & 0x1f000000 == 0 {
            return Err(ENODEV);
        }
        pr_info!("nvidia hw rev 0x{:#x}", boot0);
        let chipset = (boot0 & 0x1ff00000) >> 20;
        let _chiprev = boot0 & 0xff;

        let card_type = match chipset & 0x1f0 {
            0x160 => { pr_info!("TU100"); CardType::TU100 },
            0x170 => { pr_info!("GA100"); CardType::GA100 },
            0x190 => { pr_info!("AD100"); CardType::AD100 },
            _ => return Err(ENODEV)
        };

        let mut gpu = Gpu::new(card_type, bar);
        let _ = gpu.init(dev)?;

        Ok(())
    }

    fn remove(_data: &Self::Data) {
        pr_info!("remove()\n");
    }
}

struct NovaModule {
    _registration: Pin<Box<driver::Registration<pci::Adapter<NovaDevice>>>>,
}

impl kernel::Module for NovaModule {
    fn init(_name: &'static CStr, module: &'static ThisModule) -> Result<Self> {
        pr_info!("Module loaded!\n");

        let registration = driver::Registration::new_pinned(c_str!("nova"), module)?;
        pr_info!("Registerd PCI driver.\n");

        Ok(Self {
            _registration: registration,
        })
    }
}

module! {
    type: NovaModule,
    name: "Nova",
    author: "Danilo Krummrich",
    description: "Nova GPU driver",
    license: "GPL v2",
    params: {
    },
}