1 files changed, 118 insertions, 0 deletions
diff --git a/src/systemc/core/process_types.hh b/src/systemc/core/process_types.hh
new file mode 100644
index 000000000..6d13592b6
--- /dev/null
+++ b/src/systemc/core/process_types.hh
@@ -0,0 +1,118 @@
+/*
+ * Copyright 2018 Google, Inc.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met: redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer;
+ * redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution;
+ * neither the name of the copyright holders nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * Authors: Gabe Black
+ */
+
+#ifndef __SYSTEMC_CORE_PROCESS_TYPES_HH__
+#define __SYSTEMC_CORE_PROCESS_TYPES_HH__
+
+#include "systemc/core/process.hh"
+#include "systemc/core/scheduler.hh"
+
+namespace sc_gem5
+{
+
+class Method : public Process
+{
+  public:
+    Method(const char *name, ProcessFuncWrapper *func, bool _dynamic=false) :
+        Process(name, func, _dynamic)
+    {}
+
+    const char *kind() const override { return "sc_method_process"; }
+
+    sc_core::sc_curr_proc_kind
+    procKind() const override
+    {
+        return sc_core::SC_METHOD_PROC_;
+    }
+};
+
+class Thread : public Process
+{
+  public:
+    Thread(const char *name, ProcessFuncWrapper *func, bool _dynamic=false) :
+        Process(name, func, _dynamic), ctx(nullptr)
+    {}
+
+    ~Thread() { delete ctx; }
+
+    const char *kind() const override { return "sc_thread_process"; }
+
+    void throw_it(ExceptionWrapperBase &exc, bool inc_kids) override;
+
+    sc_core::sc_curr_proc_kind
+    procKind() const override
+    {
+        return sc_core::SC_THREAD_PROC_;
+    }
+
+    Fiber *
+    fiber() override
+    {
+        if (!ctx)
+            ctx = new Context(this, stackSize);
+        return ctx;
+    }
+
+  private:
+    class Context : public Fiber
+    {
+      public:
+        Context(Thread *thread, size_t size) : Fiber(size), thread(thread) {}
+
+      private:
+        Thread *thread;
+
+        void main() override { thread->run(); }
+    };
+
+    Context *ctx;
+};
+
+class CThread : public Thread
+{
+  public:
+    CThread(const char *name, ProcessFuncWrapper *func, bool _dynamic=false) :
+        Thread(name, func, _dynamic)
+    {
+        // We'll be in the initialization list now, but we shouldn't be.
+        popListNode();
+    }
+
+    const char *kind() const override { return "sc_cthread_process"; }
+
+    sc_core::sc_curr_proc_kind
+    procKind() const override
+    {
+        return sc_core::SC_CTHREAD_PROC_;
+    }
+};
+
+} // namespace sc_gem5
+
+#endif  //__SYSTEMC_CORE_PROCESS_TYPES_HH__