diff --git a/util/tlm/sim_control.cc b/util/tlm/sim_control.cc index f6a3e4ffe..a96daf2c8 100644 --- a/util/tlm/sim_control.cc +++ b/util/tlm/sim_control.cc @@ -152,6 +152,9 @@ Gem5SimControl::end_of_elaboration() void Gem5SimControl::run() { + // notify callback + beforeSimulate(); + GlobalSimLoopExitEvent *exit_event = NULL; if (simulationEnd == 0) { @@ -165,6 +168,9 @@ Gem5SimControl::run() getEventQueue(0)->dump(); + // notify callback + afterSimulate(); + #if TRY_CLEAN_DELETE config_manager->deleteObjects(); #endif diff --git a/util/tlm/sim_control.hh b/util/tlm/sim_control.hh index b411022de..399372c7f 100644 --- a/util/tlm/sim_control.hh +++ b/util/tlm/sim_control.hh @@ -77,6 +77,22 @@ class Gem5SimControl : public Module, public Gem5SimControlInterface /// Pointer to a previously created instance. static Gem5SimControl* instance; + /** A callback that is called from the run thread before gem5 simulation is + * started. + * + * A derived class may use this to perform any additional initializations + * prior simulation. + */ + virtual void beforeSimulate() {} + + /** A callback that is called from the run thread after gem5 simulation + * completed. + * + * A derived class may use this to perform any additional tasks after gem5 + * exits. For instance, a derived class could use this to call sc_stop(). + */ + virtual void afterSimulate() {} + public: SC_HAS_PROCESS(Gem5SimControl);