//! Toshiba/Kioxia SPI NAND Flash Chips
//!
//! Toshiba Corporation (now Kioxia) - Manufacturer ID: 0x98

use crate::domain::chip::*;
use crate::domain::types::*;

/// Toshiba/Kioxia Manufacturer ID
pub const MANUFACTURER_ID: u8 = 0x98;
pub const MANUFACTURER_NAME: &str = "Toshiba/Kioxia";

pub fn get_chips() -> Vec<ChipSpec> {
    vec![
        // =========================================================================
        // TC58CVG Series - SPI NAND Flash
        // =========================================================================
        // 1Gbit (128MB)
        nand_chip("TC58CVG0S3H", [0x98, 0xC2, 0x00], 1, 2048, 64, 128),
        // 2Gbit (256MB)
        nand_chip("TC58CVG1S3H", [0x98, 0xCB, 0x00], 2, 2048, 64, 128),
        // 4Gbit (512MB)
        nand_chip("TC58CVG2S0H", [0x98, 0xCD, 0x00], 4, 4096, 128, 256),
        // 4Gbit Kioxia variant
        nand_chip("TC58CVG2S0HRAIJ", [0x98, 0xED, 0x00], 4, 4096, 128, 256),
    ]
}

/// Helper function to create a Toshiba/Kioxia NAND chip spec
fn nand_chip(
    name: &str,
    jedec_id: [u8; 3],
    capacity_gbit: u32,
    page_size: u32,
    oob_size: u32,
    block_size_kb: u32,
) -> ChipSpec {
    ChipSpec {
        name: name.to_string(),
        manufacturer: MANUFACTURER_NAME.to_string(),
        jedec_id: JedecId::new(jedec_id),
        flash_type: FlashType::Nand,
        capacity: Capacity::gigabits(capacity_gbit),
        layout: ChipLayout {
            page_size,
            block_size: block_size_kb * 1024,
            oob_size: Some(oob_size),
        },
        capabilities: ChipCapabilities {
            supports_ecc_control: true,
            supports_dual_spi: true,
            ..Default::default()
        },
    }
}
