1 files changed, 34 insertions, 11 deletions

diff --git a/src/systemc/core/module.hh b/src/systemc/core/module.hh
index 56e125ee6..3734e633f 100644
--- a/src/systemc/core/module.hh
+++ b/src/systemc/core/module.hh
@@ -30,31 +30,54 @@
 #ifndef __SYSTEMC_CORE_MODULE_HH__
 #define __SYSTEMC_CORE_MODULE_HH__
 
-namespace SystemC
+#include <cassert>
+
+#include "systemc/core/object.hh"
+#include "systemc/ext/core/sc_module.hh"
+
+namespace sc_gem5
 {
 
 class Module
 {
   private:
     const char *_name;
+    sc_core::sc_module *_sc_mod;
+    Object *_obj;
 
   public:
-    Module(const char *name) : _name(name) {}
 
-    const char *name() { return _name; }
+    Module(const char *name);
+    void finish(Object *this_obj);
 
-    void push();
-    void pop();
-};
+    const char *name() const { return _name; }
 
-extern Module *topModule();
+    sc_core::sc_module *
+    sc_mod() const
+    {
+        assert(_sc_mod);
+        return _sc_mod;
+    }
 
-} // namespace SystemC
+    void
+    sc_mod(sc_core::sc_module *sc_mod)
+    {
+        assert(!_sc_mod);
+        _sc_mod = sc_mod;
+    }
 
-namespace sc_gem5
-{
+    Object *
+    obj()
+    {
+        assert(_obj);
+        return _obj;
+    }
+
+    void pop();
+};
 
-using SystemC::Module;
+Module *currentModule();
+Module *newModule();
 
 } // namespace sc_gem5