Consumer.hh (7973:e5550966464a) Consumer.hh (9171:ae88ecf37145)
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
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"
41#include "mem/ruby/common/TypeDefines.hh"
42#include "sim/eventq.hh"
43
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
44class Consumer
45{
46 public:
47 Consumer()
48 : m_last_scheduled_wakeup(0), m_last_wakeup(0)
49 {
50 }
51
52 virtual
53 ~Consumer()
54 { }
55
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
56 virtual void wakeup() = 0;
57 virtual void print(std::ostream& out) const = 0;
58 virtual void storeEventInfo(int info) {}
59
72 const Time&
60 const Tick&
73 getLastScheduledWakeup() const
74 {
75 return m_last_scheduled_wakeup;
76 }
77
78 void
61 getLastScheduledWakeup() const
62 {
63 return m_last_scheduled_wakeup;
64 }
65
66 void
79 setLastScheduledWakeup(const Time& time)
67 setLastScheduledWakeup(const Tick& time)
80 {
81 m_last_scheduled_wakeup = time;
82 }
83
84 bool
68 {
69 m_last_scheduled_wakeup = time;
70 }
71
72 bool
85 alreadyScheduled(Time time)
73 alreadyScheduled(Tick time)
86 {
87 return m_scheduled_wakeups.find(time) != m_scheduled_wakeups.end();
88 }
89
90 void
74 {
75 return m_scheduled_wakeups.find(time) != m_scheduled_wakeups.end();
76 }
77
78 void
91 insertScheduledWakeupTime(Time time)
79 insertScheduledWakeupTime(Tick time)
92 {
93 m_scheduled_wakeups.insert(time);
94 }
95
96 void
80 {
81 m_scheduled_wakeups.insert(time);
82 }
83
84 void
97 removeScheduledWakeupTime(Time time)
85 removeScheduledWakeupTime(Tick time)
98 {
99 assert(alreadyScheduled(time));
100 m_scheduled_wakeups.erase(time);
101 }
102
86 {
87 assert(alreadyScheduled(time));
88 m_scheduled_wakeups.erase(time);
89 }
90
91 void scheduleEvent(EventManager* em, Time timeDelta);
92 void scheduleEventAbsolute(EventManager* em, Time timeAbs);
93 void scheduleEvent(Time timeDelta);
94 void scheduleEventAbsolute(Time timeAbs);
95
103 private:
96 private:
104 Time m_last_scheduled_wakeup;
105 std::set<Time> m_scheduled_wakeups;
106 Time m_last_wakeup;
97 Tick m_last_scheduled_wakeup;
98 std::set<Tick> m_scheduled_wakeups;
99 Tick m_last_wakeup;
100
101 class ConsumerEvent : public Event
102 {
103 public:
104 ConsumerEvent(Consumer* _consumer)
105 : Event(Default_Pri, AutoDelete), m_consumer_ptr(_consumer)
106 {
107 }
108
109 void process()
110 {
111 m_consumer_ptr->wakeup();
112 m_consumer_ptr->removeScheduledWakeupTime(when());
113 }
114
115 private:
116 Consumer* m_consumer_ptr;
117 };
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__
118};
119
120inline std::ostream&
121operator<<(std::ostream& out, const Consumer& obj)
122{
123 obj.print(out);
124 out << std::flush;
125 return out;
126}
127
128#endif // __MEM_RUBY_COMMON_CONSUMER_HH__