from amgen import AmaranthGenerator

from ufm_reader.page_buffer import PageBuffer
from amaranth_stdio import serial


class PageBufferGenerator(AmaranthGenerator):
    output_file = "page_buffer.v"
    module_name = "page_buffer"

    def __init__(self):
        super().__init__()

    # Generate a core to be included in another project.
    def create_module(self):
        m = PageBuffer()
        ios = [m.rand.data, m.rand.addr, m.rand.read_en,
               m.rand.flush, m.rand.valid, m.seq.data,
               m.seq.addr, m.seq.ready, m.seq.valid]

        return (m, ios)


if __name__ == "__main__":
    PageBufferGenerator().generate()