pmat 3.15.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#![cfg_attr(coverage_nightly, coverage(off))]

use std::collections::{HashMap, HashSet};

use super::grade::{MetricCategory, PenaltyAttribution};

/// Penalty tracker.
pub struct PenaltyTracker {
    applied: HashMap<String, PenaltyAttribution>,
}

impl Default for PenaltyTracker {
    fn default() -> Self {
        Self::new()
    }
}

impl PenaltyTracker {
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Create a new instance.
    pub fn new() -> Self {
        Self {
            applied: HashMap::new(),
        }
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Apply the operation.
    pub fn apply(
        &mut self,
        issue_id: String,
        category: MetricCategory,
        amount: f32,
        issue: String,
    ) -> Option<f32> {
        if self.applied.contains_key(&issue_id) {
            return None;
        }

        self.applied.insert(
            issue_id,
            PenaltyAttribution {
                source_metric: category,
                amount,
                applied_to: HashSet::from([category]),
                issue,
            },
        );

        Some(amount)
    }

    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Get attributions.
    pub fn get_attributions(&self) -> Vec<PenaltyAttribution> {
        self.applied.values().cloned().collect()
    }
}