Source code for elasticai.creator_plugins.skeleton

import importlib.resources as res
from string import Template

import elasticai.creator.ir.ir_v2 as ir
from elasticai.creator.ir2vhdl import (
    Code,
    DataGraph,
    EntityTemplateDirector,
    Ir2Vhdl,
    PluginSymbol,
)

from ._skeleton import skeleton as skeleton


class _BufferedNetworkWrapper(PluginSymbol):
    def __init__(self, template: Template):
        self._template = template

    def __call__(self, args: DataGraph, registry: ir.Registry) -> Code:
        return args.name, self._template.substitute(
            dict(entity=args.name) | args.attributes["generic_map"]
        )

    @classmethod
    def load_into(cls, receiver: Ir2Vhdl) -> None:
        def load_file():
            return (
                res.files("elasticai.creator_plugins.skeleton")
                .joinpath("vhdl/buffered_network_wrapper.vhd")
                .read_text()
            )

        template = (
            EntityTemplateDirector()
            .set_prototype(load_file())
            .add_generic("KERNEL_SIZE")
            .add_generic("STRIDE")
        ).build()
        receiver.register("buffered_network_wrapper", cls(template))


buffered_network_wrapper = _BufferedNetworkWrapper