vram/nand_block.py

from nand_page import Page

class Block:
    def __init__(self, num_pages, num_cells_per_page, channel_length, drift_velocity, levels):
        self.pages = [Page(num_cells_per_page, channel_length, drift_velocity, levels) for _ in range(num_pages)]
        self.wear_count = 0

    def erase(self):
        for page in self.pages:
            page.erase()
        self.wear_count += 1