from ..symbols import LunaixSymbols
from ..structs.page import PageStruct
from ..pp import MyPrettyPrinter
import math

class PhysicalMemProfile:
    def __init__(self) -> None:
        super().__init__()
        self._pm_list = LunaixSymbols.debug_sym("pmm", "pm_table")

        self.max_mem_pg = int(LunaixSymbols.debug_sym("pmm", "max_pg").value())
        self.max_mem_sz = self.max_mem_pg * 4096
        self.mem_distr = []

    def rescan_pmem(self, distr_granule = 256):
        self.__mem_distr_granule = distr_granule
        self.mem_distr.clear()

        page_per_granule = self.max_mem_pg / self.__mem_distr_granule
        page_per_granule = math.ceil(page_per_granule)
        remainder = self.max_mem_pg % self.__mem_distr_granule
        bucket = 0
        non_contig = 0
        last_contig = False
        for i in range(self.max_mem_pg):
            element = PageStruct(self._pm_list[i].address)
            bucket += int(element.ref > 0)
            if last_contig:
                last_contig = element.ref > 0
                non_contig += int(not last_contig)
            else:
                last_contig = element.ref > 0

            if (i + 1) % page_per_granule == 0:
                self.mem_distr.append(bucket)
                bucket = 0
        
        if remainder > 0:
            if bucket > 0:
                bucket += page_per_granule - remainder
            self.mem_distr.append(bucket)

        self.consumed_pg = sum(self.mem_distr)
        self.utilisation = self.consumed_pg / self.max_mem_pg
        self.fragmentation = 2 * non_contig / self.max_mem_pg
        self.discontig = non_contig
        self.page_per_granule = page_per_granule