diff options
Diffstat (limited to 'source/Concept/Framework/robot.h')
| -rw-r--r-- | source/Concept/Framework/robot.h | 33 |
1 files changed, 15 insertions, 18 deletions
diff --git a/source/Concept/Framework/robot.h b/source/Concept/Framework/robot.h index 543f1ed..db7fe15 100644 --- a/source/Concept/Framework/robot.h +++ b/source/Concept/Framework/robot.h @@ -9,32 +9,29 @@ class Engine; class Robot
{
-public:
+public: + Robot();
~Robot();
private:
- typedef hash_map<uint32, IO_Module*> moduleMap;
- moduleMap modules;
+ IO_Module* modules[IO_END];
public:
bool AddModule(IO_Module* newModule);
- template <class T> T* GetModule(uint32 moduleId)
- {
- moduleMap::const_iterator itr = modules.find(moduleId);
- if(itr == modules.end())
- return NULL;
- else
- return ((T*)itr->second);
+ template <class T> T* GetModule(uint8 moduleId)
+ { + return ((T*)modules[moduleId]);
}
- bool RemoveModule(uint32 moduleId)
- {
- moduleMap::iterator itr = modules.find(moduleId);
- if(itr == modules.end())
- return false;
-
- modules.erase(itr);
+ bool RemoveModule(uint8 moduleId)
+ { + if(!modules[moduleId]) + return false; + + delete modules[moduleId]; + modules[moduleId] = NULL; +
return true;
}
@@ -43,4 +40,4 @@ public: void Update();
};
-#endif
\ No newline at end of file +#endif |