From df5ccb6dd968b5eb5589ffda3777d80cb0997153 Mon Sep 17 00:00:00 2001
From: Jean THOMAS <git0@pub.jeanthomas.me>
Date: Thu, 11 Jun 2020 11:54:07 +0200
Subject: [PATCH] Restrict exposed classes

---
 gram/core/__init__.py    | 3 +--
 gram/core/bankmachine.py | 3 ++-
 gram/core/controller.py  | 2 +-
 gram/core/multiplexer.py | 3 +--
 4 files changed, 5 insertions(+), 6 deletions(-)

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
-- 
2.30.2