Add direction accessor for IOModel and export the right api

Author: robtaylor

chipflow_lib/platforms/__init__.py

@@ -10,6 +10,11 @@
 from .sim import *
 from .utils import *
 
-__all__ = ['IO_ANNOTATION_SCHEMA', 'IOSignature',
+__all__ = ['IO_ANNOTATION_SCHEMA', 'IOSignature', 'IOModel',
            'OutputIOSignature', 'InputIOSignature', 'BidirIOSignature',
-           'load_pinlock', "PACKAGE_DEFINITIONS", 'top_components']
+           'load_pinlock', "PACKAGE_DEFINITIONS", 'top_components', 'LockFile',
+           'Package', 'PortMap', 'Port', 'Process',
+           'GAPackageDef', 'QuadPackageDef', 'BareDiePackageDef', 'BasePackageDef',
+           'BringupPins', 'JTAGPins', 'PowerPins',
+           'SiliconPlatformPort', 'SiliconPlatform',
+           'SimPlatform']

chipflow_lib/platforms/utils.py

@@ -26,7 +26,7 @@
 from amaranth.lib.wiring import In, Out
 from pydantic import (
         ConfigDict, TypeAdapter, PlainSerializer,
-        WithJsonSchema, with_config
+        WithJsonSchema
         )
 
 
@@ -35,11 +35,12 @@
 if TYPE_CHECKING:
     from ..config_models import Config
 
-__all__ = ['IO_ANNOTATION_SCHEMA', 'IOSignature',
+__all__ = ['IO_ANNOTATION_SCHEMA', 'IOSignature', 'IOModel',
            'OutputIOSignature', 'InputIOSignature', 'BidirIOSignature',
            'load_pinlock', "PACKAGE_DEFINITIONS", 'top_components', 'LockFile',
-           'Package', 'PortMap', 'Port',
-           'GAPackageDef', 'QuadPackageDef', 'BareDiePackageDef', 'BasePackageDef']
+           'Package', 'PortMap', 'Port', 'Process',
+           'GAPackageDef', 'QuadPackageDef', 'BareDiePackageDef', 'BasePackageDef',
+           'BringupPins', 'JTAGPins', 'PowerPins']
 
 
 logger = logging.getLogger(__name__)
@@ -73,7 +74,7 @@ class VoltageRange:
 })
 
 
-@with_config(ConfigDict(arbitrary_types_allowed=True))  # type: ignore[reportCallIssue]
+@pydantic.with_config(ConfigDict(arbitrary_types_allowed=True))  # type: ignore[reportCallIssue]
 class _IOModelOptions(TypedDict):
     all_have_oe: NotRequired[bool]
     allocate_power: NotRequired[bool]
@@ -83,7 +84,7 @@ class _IOModelOptions(TypedDict):
     init: NotRequired[Annotated[Const, ConstSerializer, ConstSchema]]
 
 
-@with_config(ConfigDict(arbitrary_types_allowed=True))  # type: ignore[reportCallIssue]
+@pydantic.with_config(ConfigDict(arbitrary_types_allowed=True))  # type: ignore[reportCallIssue]
 class IOModel(_IOModelOptions):
     """
     Options for IO Ports
@@ -305,6 +306,13 @@ def width(self):
         assert len(self.pins) == self.iomodel['width']
         return self.iomodel['width']
 
+    @property
+    def direction(self):
+        assert self.pins and 'direction' in self.iomodel
+        assert len(self.pins) == self.iomodel['direction']
+        return self.iomodel['direction']
+
+
 
 def _group_consecutive_items(ordering: PinList, lst: PinList) -> OrderedDict[int, List[PinList]]:
     if not lst: