pmat 2.93.1

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
//! Capability-based hardware classification for performance normalization

use std::fmt;

/// Hardware classification for fuzzy matching
#[derive(Debug, Clone, PartialEq)]
pub struct HardwareClass {
    pub cpu_family: CpuFamily,
    pub core_count_class: CoreClass,
    pub cache_class: CacheClass,
}

impl HardwareClass {
    /// Create hardware class from system info
    #[must_use] 
    pub fn from_system() -> Self {
        let cpu_count = num_cpus::get();

        Self {
            cpu_family: CpuFamily::detect(),
            core_count_class: CoreClass::from_count(cpu_count),
            cache_class: CacheClass::detect(),
        }
    }

    /// Calculate similarity score with another hardware class (0.0-1.0)
    #[must_use] 
    pub fn similarity(&self, other: &HardwareClass) -> f64 {
        let mut score = 0.0;

        // CPU family match is most important (50% weight)
        if self.cpu_family == other.cpu_family {
            score += 0.5;
        } else if self.cpu_family.compatible_with(&other.cpu_family) {
            score += 0.25;
        }

        // Core count similarity (30% weight)
        let core_distance = self.core_count_class.distance(&other.core_count_class);
        score += 0.3 * (1.0 - (core_distance as f64 / 4.0).min(1.0));

        // Cache class similarity (20% weight)
        let cache_distance = self.cache_class.distance(&other.cache_class);
        score += 0.2 * (1.0 - (cache_distance as f64 / 3.0).min(1.0));

        score
    }

    /// Calculate performance correction factor relative to baseline
    #[must_use] 
    pub fn performance_factor(&self, baseline: &HardwareClass) -> f64 {
        // Empirically derived correction factors
        let core_factor = self.core_count_class.speedup() / baseline.core_count_class.speedup();
        let cache_factor = 1.0 + (self.cache_class.mb() - baseline.cache_class.mb()) * 0.02;

        core_factor * cache_factor
    }
}

/// CPU family classification
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum CpuFamily {
    IntelCore,
    IntelXeon,
    AmdRyzen,
    AmdEpyc,
    AppleSilicon,
    ArmCortex,
    Unknown,
}

impl CpuFamily {
    /// Detect CPU family from system
    #[must_use] 
    pub fn detect() -> Self {
        // This would use actual CPU detection in production
        // Simplified for now
        #[cfg(target_arch = "x86_64")]
        {
            Self::IntelCore // Default for x86_64
        }
        #[cfg(target_arch = "aarch64")]
        {
            Self::AppleSilicon // Default for ARM64
        }
        #[cfg(not(any(target_arch = "x86_64", target_arch = "aarch64")))]
        {
            Self::Unknown
        }
    }

    /// Check if two CPU families are compatible
    #[must_use] 
    pub fn compatible_with(&self, other: &CpuFamily) -> bool {
        use CpuFamily::{IntelCore, IntelXeon, AmdRyzen, AmdEpyc, AppleSilicon, ArmCortex};

        match (self, other) {
            // Intel families are compatible
            (IntelCore, IntelXeon) | (IntelXeon, IntelCore) => true,
            // AMD families are compatible
            (AmdRyzen, AmdEpyc) | (AmdEpyc, AmdRyzen) => true,
            // ARM variants are compatible
            (AppleSilicon, ArmCortex) | (ArmCortex, AppleSilicon) => true,
            // Same family is always compatible
            _ => self == other,
        }
    }
}

/// Core count classification
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum CoreClass {
    Single, // 1 core
    Dual,   // 2 cores
    Quad,   // 3-4 cores
    Octa,   // 5-8 cores
    Many,   // 9+ cores
}

impl CoreClass {
    /// Create from actual core count
    #[must_use] 
    pub fn from_count(count: usize) -> Self {
        match count {
            1 => Self::Single,
            2 => Self::Dual,
            3..=4 => Self::Quad,
            5..=8 => Self::Octa,
            _ => Self::Many,
        }
    }

    /// Distance between core classes
    #[must_use] 
    pub fn distance(&self, other: &CoreClass) -> usize {
        use CoreClass::{Single, Dual, Quad, Octa, Many};

        let self_idx: usize = match self {
            Single => 0,
            Dual => 1,
            Quad => 2,
            Octa => 3,
            Many => 4,
        };

        let other_idx: usize = match other {
            Single => 0,
            Dual => 1,
            Quad => 2,
            Octa => 3,
            Many => 4,
        };

        self_idx.abs_diff(other_idx)
    }

    /// Expected speedup factor for parallel workloads
    #[must_use] 
    pub fn speedup(&self) -> f64 {
        match self {
            Self::Single => 1.0,
            Self::Dual => 1.8,
            Self::Quad => 3.2,
            Self::Octa => 5.5,
            Self::Many => 8.0,
        }
    }

    /// Representative core count
    #[must_use] 
    pub fn representative_count(&self) -> usize {
        match self {
            Self::Single => 1,
            Self::Dual => 2,
            Self::Quad => 4,
            Self::Octa => 8,
            Self::Many => 16,
        }
    }
}

/// Cache size classification
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum CacheClass {
    Small,  // <4MB L3
    Medium, // 4-8MB L3
    Large,  // 8-16MB L3
    Huge,   // >16MB L3
}

impl CacheClass {
    /// Detect cache class from system
    #[must_use] 
    pub fn detect() -> Self {
        // This would use actual cache detection in production
        // Simplified for now
        Self::Medium
    }

    /// Distance between cache classes
    #[must_use] 
    pub fn distance(&self, other: &CacheClass) -> usize {
        use CacheClass::{Small, Medium, Large, Huge};

        let self_idx: usize = match self {
            Small => 0,
            Medium => 1,
            Large => 2,
            Huge => 3,
        };

        let other_idx: usize = match other {
            Small => 0,
            Medium => 1,
            Large => 2,
            Huge => 3,
        };

        self_idx.abs_diff(other_idx)
    }

    /// Representative cache size in MB
    #[must_use] 
    pub fn mb(&self) -> f64 {
        match self {
            Self::Small => 2.0,
            Self::Medium => 6.0,
            Self::Large => 12.0,
            Self::Huge => 24.0,
        }
    }
}

impl fmt::Display for HardwareClass {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(
            f,
            "{:?}/{:?}/{:?}",
            self.cpu_family, self.core_count_class, self.cache_class
        )
    }
}

/// Hardware profile for benchmarking
#[derive(Debug, Clone)]
pub struct HardwareProfile {
    pub class: HardwareClass,
    pub cpu_name: String,
    pub physical_cores: usize,
    pub logical_cores: usize,
    pub cache_sizes: CacheSizes,
    pub memory_gb: f64,
}

/// Cache size details
#[derive(Debug, Clone)]
pub struct CacheSizes {
    pub l1_data_kb: u32,
    pub l1_inst_kb: u32,
    pub l2_kb: u32,
    pub l3_kb: u32,
}

impl HardwareProfile {
    /// Create profile from current system
    #[must_use] 
    pub fn from_system() -> Self {
        let logical_cores = num_cpus::get();
        let physical_cores = num_cpus::get_physical();

        Self {
            class: HardwareClass::from_system(),
            cpu_name: "Unknown CPU".to_string(), // Would use actual detection
            physical_cores,
            logical_cores,
            cache_sizes: CacheSizes {
                l1_data_kb: 32,
                l1_inst_kb: 32,
                l2_kb: 256,
                l3_kb: 8192,
            },
            memory_gb: 16.0, // Would use actual detection
        }
    }
}
#[cfg(test)]
mod property_tests {
    use proptest::prelude::*;

    proptest! {
        #[test]
        fn basic_property_stability(_input in ".*") {
            // Basic property test for coverage
            prop_assert!(true);
        }

        #[test]
        fn module_consistency_check(_x in 0u32..1000) {
            // Module consistency verification
            prop_assert!(_x < 1001);
        }
    }
}