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;

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

58 Throttle(NodeID node, Cycles link_latency, int link_bandwidth_multiplier,
59 int endpoint_bandwidth, ClockedObject *em);
60 ~Throttle() {}
61
62 std::string name()
63 { return csprintf("Throttle-%i", m_sID); }
64
65 void addLinks(const std::vector<MessageBuffer*>& in_vec,
66 const std::vector<MessageBuffer*>& out_vec);
67 void wakeup();
68
69 // The average utilization (a fraction) since last clearStats()
70 const Stats::Scalar & getUtilization() const
71 { return m_link_utilization; }
72 const Stats::Vector & getMsgCount(unsigned int type) const
73 { return m_msg_counts[type]; }
74
75 int getLinkBandwidth() const
76 { return m_endpoint_bandwidth * m_link_bandwidth_multiplier; }
77
78 Cycles getLatency() const { return m_link_latency; }
79
80 void clearStats();
81 void collateStats();
82 void regStats(std::string name);
83 void print(std::ostream& out) const;
84
85 private:
86 void init(NodeID node, Cycles link_latency, int link_bandwidth_multiplier,
87 int endpoint_bandwidth);
88 void addVirtualNetwork(MessageBuffer* in_ptr, MessageBuffer* out_ptr);
89
90 // Private copy constructor and assignment operator
91 Throttle(const Throttle& obj);
92 Throttle& operator=(const Throttle& obj);
93
94 std::vector<MessageBuffer*> m_in;
95 std::vector<MessageBuffer*> m_out;
96 unsigned int m_vnets;
97 std::vector<int> m_units_remaining;
98 int m_sID;
99 NodeID m_node;
100 int m_link_bandwidth_multiplier;
101 Cycles m_link_latency;
102 int m_wakeups_wo_switch;
103 int m_endpoint_bandwidth;
104
105 // Statistical variables
106 Stats::Scalar m_link_utilization;
107 std::vector<Stats::Vector> m_msg_counts;
108 std::vector<Stats::Formula> m_msg_bytes;
109
110 double m_link_utilization_proxy;
111};
112
113inline std::ostream&
114operator<<(std::ostream& out, const Throttle& obj)
115{
116 obj.print(out);
117 out << std::flush;
118 return out;
119}
120
121#endif // __MEM_RUBY_NETWORK_SIMPLE_THROTTLE_HH__