From: Jean THOMAS <git0@pub.jeanthomas.me>
Date: Thu, 11 Jun 2020 09:54:07 +0000 (+0200)
Subject: Restrict exposed classes
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=df5ccb6dd968b5eb5589ffda3777d80cb0997153;p=gram.git

Restrict exposed classes
---

diff --git a/gram/core/__init__.py b/gram/core/__init__.py
index 16b022e..f45a89a 100644
--- a/gram/core/__init__.py
+++ b/gram/core/__init__.py
@@ -7,8 +7,7 @@ from gram.compat import CSRPrefixProxy
 from gram.core.controller import ControllerSettings, gramController
 from gram.core.crossbar import gramCrossbar
 
-# Core ---------------------------------------------------------------------------------------------
-
+__ALL__ = ["gramCore"]
 
 class gramCore(Peripheral, Elaboratable):
     def __init__(self, phy, geom_settings, timing_settings, clk_freq, **kwargs):
diff --git a/gram/core/bankmachine.py b/gram/core/bankmachine.py
index 8cb74c5..787bf83 100644
--- a/gram/core/bankmachine.py
+++ b/gram/core/bankmachine.py
@@ -14,6 +14,8 @@ from gram.core.multiplexer import *
 from gram.compat import delayed_enter
 import gram.stream as stream
 
+__ALL__ = ["BankMachine"]
+
 # AddressSlicer ------------------------------------------------------------------------------------
 
 
@@ -115,7 +117,6 @@ class BankMachine(Elaboratable):
         cmd_buffer = stream.Buffer(cmd_buffer_layout)
         m.submodules += cmd_buffer_lookahead, cmd_buffer
         m.d.comb += [
-            #self.req.connect(cmd_buffer_lookahead.sink, include={"valid", "ready", "payload.we", "payload.addr"}),
             cmd_buffer_lookahead.sink.valid.eq(self.req.valid),
             self.req.ready.eq(cmd_buffer_lookahead.sink.ready),
             cmd_buffer_lookahead.sink.payload.we.eq(self.req.we),
diff --git a/gram/core/controller.py b/gram/core/controller.py
index 612afcd..a7e376e 100644
--- a/gram/core/controller.py
+++ b/gram/core/controller.py
@@ -15,7 +15,7 @@ from gram.core.bankmachine import BankMachine
 from gram.core.multiplexer import Multiplexer
 
 # Settings -----------------------------------------------------------------------------------------
-
+__ALL__ = ["gramController"]
 
 class ControllerSettings(Settings):
     def __init__(self,
diff --git a/gram/core/multiplexer.py b/gram/core/multiplexer.py
index 6e3584a..d53177a 100644
--- a/gram/core/multiplexer.py
+++ b/gram/core/multiplexer.py
@@ -16,8 +16,7 @@ from gram.common import *
 import gram.stream as stream
 from gram.compat import RoundRobin, delayed_enter
 
-# _CommandChooser ----------------------------------------------------------------------------------
-
+__ALL__ = ["Multiplexer"]
 
 class _CommandChooser(Elaboratable):
     """Arbitrates between requests, filtering them based on their type