1/*
2 * Copyright 2018 Google, Inc.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met: redistributions of source code must retain the above copyright
7 * notice, this list of conditions and the following disclaimer;
8 * redistributions in binary form must reproduce the above copyright
9 * notice, this list of conditions and the following disclaimer in the
10 * documentation and/or other materials provided with the distribution;
11 * neither the name of the copyright holders nor the names of its
12 * contributors may be used to endorse or promote products derived from
13 * this software without specific prior written permission.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 *
27 * Authors: Gabe Black
28 */
29
30#ifndef __SYSTEMC_CORE_PROCESS_TYPES_HH__
31#define __SYSTEMC_CORE_PROCESS_TYPES_HH__
32
33#include "systemc/core/process.hh"
34#include "systemc/core/scheduler.hh"
35
36namespace sc_gem5
37{
38
39class Method : public Process
40{
41  public:
42    Method(const char *name, ProcessFuncWrapper *func, bool internal=false) :
43        Process(name, func, internal)
44    {}
45
46    const char *kind() const override { return "sc_method_process"; }
47
48    sc_core::sc_curr_proc_kind
49    procKind() const override
50    {
51        return sc_core::SC_METHOD_PROC_;
52    }
53};
54
55class Thread : public Process
56{
57  public:
58    Thread(const char *name, ProcessFuncWrapper *func, bool internal=false) :
59        Process(name, func, internal), ctx(nullptr)
60    {}
61
62    ~Thread() { delete ctx; }
63
64    const char *kind() const override { return "sc_thread_process"; }
65
66    sc_core::sc_curr_proc_kind
67    procKind() const override
68    {
69        return sc_core::SC_THREAD_PROC_;
70    }
71
72    Fiber *
73    fiber() override
74    {
75        if (!ctx)
76            ctx = new Context(this, stackSize);
77        return ctx;
78    }
79
80  private:
81    class Context : public Fiber
82    {
83      public:
84        Context(Thread *thread, size_t size) : Fiber(size), thread(thread) {}
85
86      private:
87        Thread *thread;
88
89        void
90        main() override
91        {
92            thread->_needsStart = false;
93            try {
94                thread->run();
95            } catch (...) {
96                thread->terminate();
97                scheduler.throwUp();
98                return;
99            }
100            thread->terminate();
101            scheduler.yield();
102        }
103    };
104    friend class Context;
105
106    Context *ctx;
107};
108
109class CThread : public Thread
110{
111  public:
112    CThread(const char *name, ProcessFuncWrapper *func, bool internal=false) :
113        Thread(name, func, internal)
114    {
115        // We'll be in the initialization list now, but we shouldn't be.
116        popListNode();
117    }
118
119    const char *kind() const override { return "sc_cthread_process"; }
120
121    sc_core::sc_curr_proc_kind
122    procKind() const override
123    {
124        return sc_core::SC_CTHREAD_PROC_;
125    }
126};
127
128} // namespace sc_gem5
129
130#endif  //__SYSTEMC_CORE_PROCESS_TYPES_HH__
131