1/*
2 * Copyright (c) 1999-2008 Mark D. Hill and David A. Wood
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are
7 * met: redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer;

--- 24 unchanged lines hidden (view full) ---

33 */
34
35#ifndef __MEM_RUBY_COMMON_CONSUMER_HH__
36#define __MEM_RUBY_COMMON_CONSUMER_HH__
37
38#include <iostream>
39#include <set>
40
41#include "mem/ruby/common/Global.hh"
42#include "mem/ruby/eventqueue/RubyEventQueue.hh"
43
44class MessageBuffer;
45
46class Consumer
47{
48 public:
49 Consumer()
50 : m_last_scheduled_wakeup(0), m_last_wakeup(0)
51 {
52 }
53
54 virtual
55 ~Consumer()
56 { }
57
58 void
59 triggerWakeup(RubyEventQueue *eventQueue)
60 {
61 Time time = eventQueue->getTime();
62 if (m_last_wakeup != time) {
63 wakeup();
64 m_last_wakeup = time;
65 }
66 }
67
68 virtual void wakeup() = 0;
69 virtual void print(std::ostream& out) const = 0;
70 virtual void storeEventInfo(int info) {}
71
72 const Time&
73 getLastScheduledWakeup() const
74 {
75 return m_last_scheduled_wakeup;
76 }
77
78 void
79 setLastScheduledWakeup(const Time& time)
80 {
81 m_last_scheduled_wakeup = time;
82 }
83
84 bool
85 alreadyScheduled(Time time)
86 {
87 return m_scheduled_wakeups.find(time) != m_scheduled_wakeups.end();
88 }
89
90 void
91 insertScheduledWakeupTime(Time time)
92 {
93 m_scheduled_wakeups.insert(time);
94 }
95
96 void
97 removeScheduledWakeupTime(Time time)
98 {
99 assert(alreadyScheduled(time));
100 m_scheduled_wakeups.erase(time);
101 }
102
103 private:
104 Time m_last_scheduled_wakeup;
105 std::set<Time> m_scheduled_wakeups;
106 Time m_last_wakeup;
107};
108
109inline std::ostream&
110operator<<(std::ostream& out, const Consumer& obj)
111{
112 obj.print(out);
113 out << std::flush;
114 return out;
115}
116
117#endif // __MEM_RUBY_COMMON_CONSUMER_HH__