Rev 1646 | Rev 1665 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | SVN | RSS feed
| Rev 1646 | Rev 1657 | ||
|---|---|---|---|
| 1 | /* |
1 | /* |
| 2 | * |
2 | * |
| 3 | * Copyright (C) 2010 Mattias Runge |
3 | * Copyright (C) 2010 Mattias Runge |
| 4 | * |
4 | * |
| 5 | * This program is free software; you can redistribute it and/or modify |
5 | * This program is free software; you can redistribute it and/or modify |
| 6 | * it under the terms of the GNU General Public License as published by |
6 | * it under the terms of the GNU General Public License as published by |
| 7 | * the Free Software Foundation; either version 2 of the License, or |
7 | * the Free Software Foundation; either version 2 of the License, or |
| 8 | * (at your option) any later version. |
8 | * (at your option) any later version. |
| 9 | * |
9 | * |
| 10 | * This program is distributed in the hope that it will be useful, |
10 | * This program is distributed in the hope that it will be useful, |
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | * GNU General Public License for more details. |
13 | * GNU General Public License for more details. |
| 14 | * |
14 | * |
| 15 | * You should have received a copy of the GNU General Public License along |
15 | * You should have received a copy of the GNU General Public License along |
| 16 | * with this program; if not, write to the Free Software Foundation, Inc., |
16 | * with this program; if not, write to the Free Software Foundation, Inc., |
| 17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| 18 | * |
18 | * |
| 19 | */ |
19 | */ |
| 20 | 20 | ||
| 21 | #include "Manager.h" |
21 | #include "Manager.h" |
| 22 | 22 | ||
| 23 | #include <boost/lexical_cast.hpp> |
23 | #include <boost/lexical_cast.hpp> |
| 24 | 24 | ||
| 25 | #include "broker/Manager.h" |
25 | #include "broker/Manager.h" |
| 26 | #include "broker/Message.h" |
26 | #include "broker/Message.h" |
| 27 | #include "common/common.h" |
27 | #include "common/common.h" |
| 28 | #include "timer/Manager.h" |
28 | #include "timer/Manager.h" |
| 29 | #include "storage/Manager.h" |
29 | #include "storage/Manager.h" |
| 30 | 30 | ||
| 31 | #include "can/Message.h" |
31 | #include "can/Message.h" |
| 32 | 32 | ||
| 33 | namespace atom { |
33 | namespace atom { |
| 34 | namespace control { |
34 | namespace control { |
| 35 | 35 | ||
| 36 | Manager::Pointer Manager::instance_; |
36 | Manager::Pointer Manager::instance_; |
| 37 | 37 | ||
| 38 | Manager::Manager() : broker::Subscriber(false), LOG("control::Manager") |
38 | Manager::Manager() : broker::Subscriber(false), LOG("control::Manager") |
| 39 | { |
39 | { |
| 40 | Node::SetupStateMachine(); |
40 | Node::SetupStateMachine(); |
| 41 | 41 | ||
| 42 | this->active_programming_node_id_ = |
42 | this->active_programming_node_id_ = ""; |
| 43 | 43 | ||
| 44 | // Nyquist–Shannon sampling theorem state that we need to double the time, modules send every 10 seconds |
44 | // Nyquist–Shannon sampling theorem state that we need to double the time, modules send every 10 seconds |
| 45 | this->timer_id_ = timer::Manager::Instance()->SetTimer(20000, true); |
45 | this->timer_id_ = timer::Manager::Instance()->SetTimer(20000, true); |
| 46 | } |
46 | } |
| 47 | 47 | ||
| 48 | Manager::~Manager() |
48 | Manager::~Manager() |
| 49 | { |
49 | { |
| 50 | } |
50 | } |
| 51 | 51 | ||
| 52 | Manager::Pointer Manager::Instance() |
52 | Manager::Pointer Manager::Instance() |
| 53 | { |
53 | { |
| 54 | return Manager::instance_; |
54 | return Manager::instance_; |
| 55 | } |
55 | } |
| 56 | 56 | ||
| 57 | void Manager::Create() |
57 | void Manager::Create() |
| 58 | { |
58 | { |
| 59 | Manager::instance_ = Manager::Pointer(new Manager()); |
59 | Manager::instance_ = Manager::Pointer(new Manager()); |
| 60 | } |
60 | } |
| 61 | 61 | ||
| 62 | void Manager::Delete() |
62 | void Manager::Delete() |
| 63 | { |
63 | { |
| 64 | Manager::instance_.reset(); |
64 | Manager::instance_.reset(); |
| 65 | } |
65 | } |
| 66 | 66 | ||
| 67 | void Manager:: |
67 | void Manager::ConnectSlotNode(const SignalOnNodeChange::slot_type& signal_on_node_change_) |
| 68 | { |
68 | { |
| 69 | this->signal_on_node_change_.connect(signal_on_node_change_); |
69 | this->signal_on_node_change_.connect(signal_on_node_change_); |
| - | 70 | } |
|
| - | 71 | ||
| - | 72 | void Manager::ConnectSlotModule(const SignalOnModuleChange::slot_type& slot_on_module_change) |
|
| - | 73 | { |
|
| 70 | this->signal_on_module_change_.connect(slot_on_module_change); |
74 | this->signal_on_module_change_.connect(slot_on_module_change); |
| - | 75 | } |
|
| - | 76 | ||
| - | 77 | void Manager::ConnectSlotModule(const SignalOnModuleMessage::slot_type& slot_on_module_message) |
|
| - | 78 | { |
|
| 71 | this->signal_on_module_message_.connect(slot_on_module_message); |
79 | this->signal_on_module_message_.connect(slot_on_module_message); |
| 72 | } |
80 | } |
| 73 | 81 | ||
| 74 | void Manager::SlotOnTimeoutHandler(timer::TimerId timer_id, bool repeat) |
82 | void Manager::SlotOnTimeoutHandler(timer::TimerId timer_id, bool repeat) |
| 75 | { |
83 | { |
| 76 | if (timer_id != this->timer_id_) |
84 | if (timer_id != this->timer_id_) |
| 77 | { |
85 | { |
| 78 | return; |
86 | return; |
| 79 | } |
87 | } |
| 80 | 88 | ||
| 81 | for (NodeList::iterator it = this->nodes_.begin(); it != this->nodes_.end(); it++) |
89 | for (NodeList::iterator it = this->nodes_.begin(); it != this->nodes_.end(); it++) |
| 82 | { |
90 | { |
| 83 | if (!it->second->CheckTimeout()) |
91 | if (!it->second->CheckTimeout()) |
| 84 | { |
92 | { |
| 85 | LOG.Info("Node " + |
93 | LOG.Info("Node " +it->second->GetId() + " has not sent anything in a long time, setting offline."); |
| 86 | this->RemoveModules(it->second->GetId()); |
94 | this->RemoveModules(it->second->GetId()); |
| 87 | 95 | ||
| 88 | storage::Manager::Instance()->FlushStore("NodeList"); |
96 | storage::Manager::Instance()->FlushStore("NodeList"); |
| 89 | storage::Manager::Instance()->FlushStore("ModuleList"); |
97 | storage::Manager::Instance()->FlushStore("ModuleList"); |
| 90 | } |
98 | } |
| 91 | } |
99 | } |
| 92 | } |
100 | } |
| 93 | 101 | ||
| 94 | void Manager::SlotOnMessageHandler(broker::Message::Pointer message) |
102 | void Manager::SlotOnMessageHandler(broker::Message::Pointer message) |
| 95 | { |
103 | { |
| 96 | if (message->GetType() == broker::Message::CAN_MESSAGE) |
104 | if (message->GetType() == broker::Message::CAN_MESSAGE) |
| 97 | { |
105 | { |
| 98 | //LOG.Debug("Message received"); |
106 | //LOG.Debug("Message received"); |
| 99 | can::Message* payload = static_cast<can::Message*>(message->GetPayload().get()); |
107 | can::Message* payload = static_cast<can::Message*>(message->GetPayload().get()); |
| 100 | Node::Pointer node; |
108 | Node::Pointer node; |
| 101 | 109 | ||
| 102 | if (payload->GetClassName() == "nmt") |
110 | if (payload->GetClassName() == "nmt") |
| 103 | { |
111 | { |
| 104 | if (payload->GetCommandName() == "Bios_Start") |
112 | if (payload->GetCommandName() == "Bios_Start") |
| 105 | { |
113 | { |
| 106 | node = this->GetNode(boost::lexical_cast<unsigned int |
114 | node = this->GetNode(common::ToHex(boost::lexical_cast<unsigned int>(payload->GetVariables()["HardwareId"]))); |
| 107 | node->Trigger(Node::EVENT_BIOS_START, payload->GetVariables()); |
115 | node->Trigger(Node::EVENT_BIOS_START, payload->GetVariables()); |
| 108 | } |
116 | } |
| 109 | else if (payload->GetCommandName() == "App_Start") |
117 | else if (payload->GetCommandName() == "App_Start") |
| 110 | { |
118 | { |
| 111 | node = this->GetNode(boost::lexical_cast<unsigned int |
119 | node = this->GetNode(common::ToHex(boost::lexical_cast<unsigned int>(payload->GetVariables()["HardwareId"]))); |
| 112 | node->Trigger(Node::EVENT_APP_START, payload->GetVariables()); |
120 | node->Trigger(Node::EVENT_APP_START, payload->GetVariables()); |
| 113 | } |
121 | } |
| 114 | else if (payload->GetCommandName() == "Heartbeat") |
122 | else if (payload->GetCommandName() == "Heartbeat") |
| 115 | { |
123 | { |
| 116 | node = this->GetNode(boost::lexical_cast<unsigned int |
124 | node = this->GetNode(common::ToHex(boost::lexical_cast<unsigned int>(payload->GetVariables()["HardwareId"]))); |
| 117 | node->Trigger(Node::EVENT_HEARTBEAT, payload->GetVariables()); |
125 | node->Trigger(Node::EVENT_HEARTBEAT, payload->GetVariables()); |
| 118 | } |
126 | } |
| 119 | else if (payload->GetCommandName() == "Pgm_Ack") |
127 | else if (payload->GetCommandName() == "Pgm_Ack") |
| 120 | { |
128 | { |
| 121 | node = this->GetNode(this->active_programming_node_id_); |
129 | node = this->GetNode(this->active_programming_node_id_); |
| 122 | node->Trigger(Node::EVENT_PGM_ACK, payload->GetVariables()); |
130 | node->Trigger(Node::EVENT_PGM_ACK, payload->GetVariables()); |
| 123 | } |
131 | } |
| 124 | else if (payload->GetCommandName() == "Pgm_Nack") |
132 | else if (payload->GetCommandName() == "Pgm_Nack") |
| 125 | { |
133 | { |
| 126 | node = this->GetNode(this->active_programming_node_id_); |
134 | node = this->GetNode(this->active_programming_node_id_); |
| 127 | node->Trigger(Node::EVENT_PGM_NACK, payload->GetVariables()); |
135 | node->Trigger(Node::EVENT_PGM_NACK, payload->GetVariables()); |
| 128 | } |
136 | } |
| 129 | } |
137 | } |
| 130 | else if (payload->GetDirectionName() == "From_Owner") |
138 | else if (payload->GetDirectionName() == "From_Owner") |
| 131 | { |
139 | { |
| 132 | Module::Pointer module = this->GetModule(boost::lexical_cast<unsigned int>(payload->GetId()), payload->GetModuleName(), payload->GetClassName()); |
140 | Module::Pointer module = this->GetModule(boost::lexical_cast<unsigned int>(payload->GetId()), payload->GetModuleName(), payload->GetClassName()); |
| 133 | 141 | ||
| 134 | if (payload->GetCommandName() == "List") |
142 | if (payload->GetCommandName() == "List") |
| 135 | { |
143 | { |
| 136 | node = this->GetNode(boost::lexical_cast<unsigned int |
144 | node = this->GetNode(common::ToHex(boost::lexical_cast<unsigned int>(payload->GetVariables()["HardwareId"]))); |
| 137 | node->ResetTimeout(); |
145 | node->ResetTimeout(); |
| 138 | 146 | ||
| 139 | module->SetNodeId(node->GetId()); |
147 | module->SetNodeId(node->GetId()); |
| 140 | 148 | ||
| 141 | if (this->GetNumberOfModules(node->GetId()) == boost::lexical_cast<unsigned int>(payload->GetVariables()["NumberOfModules"])) |
149 | if (this->GetNumberOfModules(node->GetId()) == boost::lexical_cast<unsigned int>(payload->GetVariables()["NumberOfModules"])) |
| 142 | { |
150 | { |
| 143 | node->Trigger(Node::EVENT_LIST_DONE, payload->GetVariables()); |
151 | node->Trigger(Node::EVENT_LIST_DONE, payload->GetVariables()); |
| 144 | } |
152 | } |
| 145 | 153 | ||
| 146 | LOG.Info("Module " + module->GetFullId() + " is available on node " + |
154 | LOG.Info("Module " + module->GetFullId() + " is available on node " + node->GetId() + "."); |
| 147 | this->signal_on_module_change_(module->GetFullId(), true); |
155 | this->signal_on_module_change_(module->GetFullId(), true); |
| 148 | } |
156 | } |
| 149 | else |
157 | else if (module->GetNodeId() != "") |
| 150 | { |
158 | { |
| 151 | node = this->GetNode(module->GetNodeId()); |
159 | node = this->GetNode(module->GetNodeId()); |
| 152 | node->ResetTimeout(); |
160 | node->ResetTimeout(); |
| 153 | 161 | ||
| 154 | this->signal_on_module_message_(module->GetFullId(), payload->GetCommandName(), payload->GetVariables()); |
162 | this->signal_on_module_message_(module->GetFullId(), payload->GetCommandName(), payload->GetVariables()); |
| 155 | } |
163 | } |
| 156 | } |
164 | } |
| 157 | } |
165 | } |
| 158 | } |
166 | } |
| 159 | 167 | ||
| 160 | void Manager::SlotOnNewState(Node::Id node_id, Node::State current_state, Node::State target_state) |
168 | void Manager::SlotOnNewState(Node::Id node_id, Node::State current_state, Node::State target_state) |
| 161 | { |
169 | { |
| 162 | if (target_state != Node::STATE_NORM_INITIALIZED) |
170 | if (target_state != Node::STATE_NORM_INITIALIZED) |
| 163 | { |
171 | { |
| 164 | this->RemoveModules(node_id); |
172 | this->RemoveModules(node_id); |
| - | 173 | ||
| - | 174 | if (current_state != target_state) |
|
| - | 175 | { |
|
| - | 176 | this->signal_on_node_change_(node_id, false); |
|
| - | 177 | } |
|
| - | 178 | } |
|
| - | 179 | else |
|
| - | 180 | { |
|
| - | 181 | if (current_state != target_state) |
|
| - | 182 | { |
|
| - | 183 | this->signal_on_node_change_(node_id, true); |
|
| - | 184 | } |
|
| 165 | } |
185 | } |
| 166 | 186 | ||
| 167 | if (target_state == Node::STATE_BPGM_OFFLINE || target_state == Node::STATE_APGM_OFFLINE) |
187 | if (target_state == Node::STATE_BPGM_OFFLINE || target_state == Node::STATE_APGM_OFFLINE) |
| 168 | { |
188 | { |
| 169 | this->active_programming_node_id_ = node_id; |
189 | this->active_programming_node_id_ = node_id; |
| 170 | } |
190 | } |
| 171 | else if (target_state == Node::STATE_NORM_OFFLINE) |
191 | else if (target_state == Node::STATE_NORM_OFFLINE) |
| 172 | { |
192 | { |
| 173 | this->active_programming_node_id_ = |
193 | this->active_programming_node_id_ = ""; |
| 174 | } |
194 | } |
| 175 | } |
195 | } |
| 176 | 196 | ||
| 177 | void Manager::SendMessageHandler(std::string full_id, std::string command, common::StringMap variables) |
197 | void Manager::SendMessageHandler(std::string full_id, std::string command, common::StringMap variables) |
| 178 | { |
198 | { |
| 179 | ModuleList::iterator it = this->modules_.find(full_id); |
199 | ModuleList::iterator it = this->modules_.find(full_id); |
| 180 | 200 | ||
| 181 | if (it == this->modules_.end()) |
201 | if (it == this->modules_.end()) |
| 182 | { |
202 | { |
| 183 | LOG.Warning("Can not send message to " + full_id + ", it is not available"); |
203 | LOG.Warning("Can not send message to " + full_id + ", it is not available"); |
| 184 | return; |
204 | return; |
| 185 | } |
205 | } |
| 186 | 206 | ||
| 187 | try |
207 | try |
| 188 | { |
208 | { |
| 189 | can::Message* payload = new can::Message(it->second->GetClassName(), "To_Owner", it->second->GetName(), it->second->GetId(), command); |
209 | can::Message* payload = new can::Message(it->second->GetClassName(), "To_Owner", it->second->GetName(), it->second->GetId(), command); |
| 190 | payload->SetVariables(variables); |
210 | payload->SetVariables(variables); |
| 191 | 211 | ||
| 192 | broker::Manager::Instance()->Post(broker::Message::Pointer(new broker::Message(broker::Message::CAN_MESSAGE, broker::Message::PayloadPointer(payload), this))); |
212 | broker::Manager::Instance()->Post(broker::Message::Pointer(new broker::Message(broker::Message::CAN_MESSAGE, broker::Message::PayloadPointer(payload), this))); |
| 193 | //LOG.Debug("Command " + command + " sent to " + it->second->GetName()); |
213 | //LOG.Debug("Command " + command + " sent to " + it->second->GetName()); |
| 194 | } |
214 | } |
| 195 | catch (std::runtime_error& e) |
215 | catch (std::runtime_error& e) |
| 196 | { |
216 | { |
| 197 | LOG.Error("Failed to send message, " + std::string(e.what())); |
217 | LOG.Error("Failed to send message, " + std::string(e.what())); |
| 198 | } |
218 | } |
| 199 | } |
219 | } |
| 200 | 220 | ||
| 201 | void Manager::SendMessage(std::string full_id, std::string command, common::StringMap variables) |
221 | void Manager::SendMessage(std::string full_id, std::string command, common::StringMap variables) |
| 202 | { |
222 | { |
| 203 | this->io_service_.post(boost::bind(&Manager::SendMessageHandler, this, full_id, command, variables)); |
223 | this->io_service_.post(boost::bind(&Manager::SendMessageHandler, this, full_id, command, variables)); |
| 204 | } |
224 | } |
| 205 | 225 | ||
| 206 | common::StringList Manager::GetAvailableModules() |
226 | common::StringList Manager::GetAvailableModules() |
| 207 | { |
227 | { |
| 208 | common::StringList available_modules; |
228 | common::StringList available_modules; |
| 209 | 229 | ||
| 210 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
230 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
| 211 | { |
231 | { |
| 212 | available_modules.push_back(it->first); |
232 | available_modules.push_back(it->first); |
| 213 | } |
233 | } |
| 214 | 234 | ||
| 215 | return available_modules; |
235 | return available_modules; |
| - | 236 | } |
|
| - | 237 | ||
| - | 238 | common::StringList Manager::GetAvailableNodes() |
|
| - | 239 | { |
|
| - | 240 | common::StringList available_nodes; |
|
| - | 241 | ||
| - | 242 | for (NodeList::iterator it = this->nodes_.begin(); it != this->nodes_.end(); it++) |
|
| - | 243 | { |
|
| - | 244 | std::string node = it->first; |
|
| - | 245 | ||
| - | 246 | for (ModuleList::iterator it2 = this->modules_.begin(); it2 != this->modules_.end(); it2++) |
|
| - | 247 | { |
|
| - | 248 | if (it2->second->GetNodeId() == it->first) |
|
| - | 249 | { |
|
| - | 250 | node += "," + it2->first; |
|
| - | 251 | } |
|
| - | 252 | } |
|
| - | 253 | ||
| - | 254 | available_nodes.push_back(node); |
|
| - | 255 | } |
|
| - | 256 | ||
| - | 257 | return available_nodes; |
|
| - | 258 | } |
|
| - | 259 | ||
| - | 260 | bool Manager::ResetNode(Node::Id node_id) |
|
| - | 261 | { |
|
| - | 262 | NodeList::iterator it = this->nodes_.find(node_id); |
|
| - | 263 | ||
| - | 264 | if (it == this->nodes_.end()) |
|
| - | 265 | { |
|
| - | 266 | LOG.Error("Could not find node " + node_id); |
|
| - | 267 | return false; |
|
| - | 268 | } |
|
| - | 269 | ||
| - | 270 | it->second->Reset(); |
|
| - | 271 | return true; |
|
| 216 | } |
272 | } |
| 217 | 273 | ||
| 218 | bool Manager::ProgramNode(Node::Id node_id, bool is_bios, std::string filename) |
274 | bool Manager::ProgramNode(Node::Id node_id, bool is_bios, std::string filename) |
| 219 | { |
275 | { |
| 220 | NodeList::iterator it = this->nodes_.find(node_id); |
276 | NodeList::iterator it = this->nodes_.find(node_id); |
| 221 | 277 | ||
| 222 | if (it == this->nodes_.end()) |
278 | if (it == this->nodes_.end()) |
| 223 | { |
279 | { |
| 224 | LOG.Error("Could not find node " + |
280 | LOG.Error("Could not find node " + node_id); |
| 225 | return false; |
281 | return false; |
| 226 | } |
282 | } |
| 227 | 283 | ||
| 228 | Code::Pointer code = Code::Pointer(new Code()); |
284 | Code::Pointer code = Code::Pointer(new Code()); |
| 229 | if (!code->LoadIntelHexFile(filename)) |
285 | if (!code->LoadIntelHexFile(filename)) |
| 230 | { |
286 | { |
| 231 | LOG.Error("Failed to parse " + filename); |
287 | LOG.Error("Failed to parse " + filename); |
| 232 | return false; |
288 | return false; |
| 233 | } |
289 | } |
| 234 | 290 | ||
| 235 | if (is_bios) |
291 | if (is_bios) |
| 236 | { |
292 | { |
| 237 | LOG.Debug("is_bios true"); |
293 | LOG.Debug("is_bios true"); |
| 238 | it->second->ProgramBios(code); |
294 | it->second->ProgramBios(code); |
| 239 | } |
295 | } |
| 240 | else |
296 | else |
| 241 | { |
297 | { |
| 242 | LOG.Debug("is_bios false"); |
298 | LOG.Debug("is_bios false"); |
| 243 | it->second->ProgramApplication(code); |
299 | it->second->ProgramApplication(code); |
| 244 | } |
300 | } |
| 245 | 301 | ||
| 246 | return true; |
302 | return true; |
| 247 | } |
303 | } |
| 248 | 304 | ||
| 249 | Node::Pointer Manager::GetNode(Node::Id node_id) |
305 | Node::Pointer Manager::GetNode(Node::Id node_id) |
| 250 | { |
306 | { |
| 251 | Node::Pointer node; |
307 | Node::Pointer node; |
| 252 | NodeList::iterator it = this->nodes_.find(node_id); |
308 | NodeList::iterator it = this->nodes_.find(node_id); |
| 253 | 309 | ||
| 254 | if (it == this->nodes_.end()) |
310 | if (it == this->nodes_.end()) |
| 255 | { |
311 | { |
| 256 | node = Node::Pointer(new Node(node_id)); |
312 | node = Node::Pointer(new Node(node_id)); |
| 257 | node->ConnectSlots(Node::SignalOnNewState::slot_type(&Manager::SlotOnNewState, this, _1, _2, _3).track(this->tracker_)); |
313 | node->ConnectSlots(Node::SignalOnNewState::slot_type(&Manager::SlotOnNewState, this, _1, _2, _3).track(this->tracker_)); |
| 258 | this->nodes_[node_id] = node; |
314 | this->nodes_[node_id] = node; |
| 259 | } |
315 | } |
| 260 | else |
316 | else |
| 261 | { |
317 | { |
| 262 | node = it->second; |
318 | node = it->second; |
| 263 | } |
319 | } |
| 264 | 320 | ||
| 265 | return node; |
321 | return node; |
| 266 | } |
322 | } |
| 267 | 323 | ||
| 268 | Module::Pointer Manager::GetModule(Module::Id module_id, std::string module_name, std::string class_name) |
324 | Module::Pointer Manager::GetModule(Module::Id module_id, std::string module_name, std::string class_name) |
| 269 | { |
325 | { |
| 270 | Module::Pointer module; |
326 | Module::Pointer module; |
| 271 | ModuleList::iterator it = this->modules_.find(Module::MakeFullId(module_id, module_name)); |
327 | ModuleList::iterator it = this->modules_.find(Module::MakeFullId(module_id, module_name)); |
| 272 | 328 | ||
| 273 | if (it == this->modules_.end()) |
329 | if (it == this->modules_.end()) |
| 274 | { |
330 | { |
| 275 | module = Module::Pointer(new Module(module_id, module_name, class_name)); |
331 | module = Module::Pointer(new Module(module_id, module_name, class_name)); |
| 276 | this->modules_[module->GetFullId()] = module; |
332 | this->modules_[module->GetFullId()] = module; |
| 277 | } |
333 | } |
| 278 | else |
334 | else |
| 279 | { |
335 | { |
| 280 | module = it->second; |
336 | module = it->second; |
| 281 | } |
337 | } |
| 282 | 338 | ||
| 283 | return module; |
339 | return module; |
| 284 | } |
340 | } |
| 285 | 341 | ||
| 286 | void Manager::RemoveModules(Node::Id node_id) |
342 | void Manager::RemoveModules(Node::Id node_id) |
| 287 | { |
343 | { |
| 288 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
344 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
| 289 | { |
345 | { |
| 290 | if (it->second->GetNodeId() == node_id) |
346 | if (it->second->GetNodeId() == node_id) |
| 291 | { |
347 | { |
| 292 | LOG.Info("Module " + it->second->GetFullId() + " is unavailable."); |
348 | LOG.Info("Module " + it->second->GetFullId() + " is unavailable."); |
| 293 | this->signal_on_module_change_(it->second->GetFullId(), false); |
349 | this->signal_on_module_change_(it->second->GetFullId(), false); |
| 294 | 350 | ||
| 295 | this->modules_.erase(it); |
351 | this->modules_.erase(it); |
| 296 | } |
352 | } |
| 297 | } |
353 | } |
| 298 | } |
354 | } |
| 299 | 355 | ||
| 300 | unsigned int Manager::GetNumberOfModules(Node::Id node_id) |
356 | unsigned int Manager::GetNumberOfModules(Node::Id node_id) |
| 301 | { |
357 | { |
| 302 | unsigned int number_of_modules = 0; |
358 | unsigned int number_of_modules = 0; |
| 303 | 359 | ||
| 304 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
360 | for (ModuleList::iterator it = this->modules_.begin(); it != this->modules_.end(); it++) |
| 305 | { |
361 | { |
| 306 | if (it->second->GetNodeId() == node_id) |
362 | if (it->second->GetNodeId() == node_id) |
| 307 | { |
363 | { |
| 308 | number_of_modules++; |
364 | number_of_modules++; |
| 309 | } |
365 | } |
| 310 | } |
366 | } |
| 311 | 367 | ||
| 312 | return number_of_modules; |
368 | return number_of_modules; |
| 313 | } |
369 | } |
| 314 | 370 | ||
| 315 | }; // namespace control |
371 | }; // namespace control |
| 316 | }; // namespace atom |
372 | }; // namespace atom |
| 317 | 373 | ||