wip

Author: robtaylor

chipflow_lib/platforms/_signatures.py

@@ -6,6 +6,7 @@
 from amaranth.lib import wiring
 from amaranth.lib.wiring import Out
 
+from .. import ChipFlowError
 from ._utils import InputIOSignature, OutputIOSignature, BidirIOSignature, IOModelOptions, _chipflow_schema_uri, amaranth_annotate
 
 SIM_ANNOTATION_SCHEMA = str(_chipflow_schema_uri("sim-interface", 0))
@@ -101,3 +102,17 @@ def __chipflow_parameters__(self):
 
     def __repr__(self) -> str:
         return f"GPIOSignature(pin_count={self._pin_count}, {dict(self.members.items())})"
+
+
+class SimulationCanLoadData:
+    """
+    Inherit from this in your object's Signature if you want a simulation model
+    to be able to load data from your object
+    """
+    @classmethod
+    def __init_submodule__(cls, /, *args, **kwargs):
+        if wiring.Signature not in cls.mro():
+            raise ChipFlowError("SimulationCanLoadData can only be used with ``wiring.Signature`` classes")
+        original_annotations = getattr(cls, 'annotations')
+        #def annotations(self, obj, /):
+        #cls.annotate

chipflow_lib/platforms/_utils.py

@@ -156,12 +156,13 @@ def origin(self):  # type: ignore
         def as_json(self):  # type: ignore
             return PydanticModel.dump_python(self.__chipflow_annotation__)
 
-    def annotations(self, *args):  # type: ignore
-        annotations = wiring.Signature.annotations(self, *args)  # type: ignore
-        annotation = Annotation(self.__chipflow_annotation__)
-        return annotations + (annotation,)  # type: ignore
-
     def decorator(klass):
+        def annotations(self, *args):  # type: ignore
+            annotations = super(klass, self).annotations(*args)  # type: ignore
+            annotation = Annotation(self.__chipflow_annotation__)
+            return annotations + (annotation,)  # type: ignore
+
+
         klass.annotations = annotations
         return klass
     return decorator