100 lines
2.3 KiB
C++
100 lines
2.3 KiB
C++
#include "machinebase.h"
|
|
#include "devices/hwcomponent.h"
|
|
#include <memory>
|
|
#include <string>
|
|
#include <thirdparty/loguru/loguru.hpp>
|
|
|
|
std::unique_ptr<MachineBase> gMachineObj = 0;
|
|
|
|
|
|
MachineBase::MachineBase(std::string name) {
|
|
this->name = name;
|
|
|
|
/* initialize internal maps */
|
|
this->comp_map.clear();
|
|
this->subdev_map.clear();
|
|
this->aliases.clear();
|
|
}
|
|
|
|
MachineBase::~MachineBase() {
|
|
for (auto it = this->comp_map.begin(); it != this->comp_map.end(); it++) {
|
|
delete it->second;
|
|
}
|
|
this->comp_map.clear();
|
|
this->aliases.clear();
|
|
this->subdev_map.clear();
|
|
}
|
|
|
|
bool MachineBase::add_component(std::string name, HWComponent* dev_obj) {
|
|
if (this->comp_map.count(name)) {
|
|
LOG_F(ERROR, "Component %s already exists!", name.c_str());
|
|
return false;
|
|
}
|
|
|
|
this->comp_map[name] = dev_obj;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool MachineBase::add_subdevice(std::string name, HWComponent* dev_obj) {
|
|
if (this->subdev_map.count(name)) {
|
|
LOG_F(ERROR, "Subdevice %s already exists!", name.c_str());
|
|
return false;
|
|
}
|
|
|
|
this->subdev_map[name] = dev_obj;
|
|
|
|
return true;
|
|
}
|
|
|
|
void MachineBase::add_alias(std::string name, std::string alias) {
|
|
this->aliases[alias] = name;
|
|
}
|
|
|
|
HWComponent* MachineBase::get_comp_by_name(std::string name) {
|
|
if (this->aliases.count(name)) {
|
|
name = this->aliases[name];
|
|
}
|
|
|
|
if (this->comp_map.count(name)) {
|
|
return this->comp_map[name];
|
|
}
|
|
|
|
if (this->subdev_map.count(name)) {
|
|
return this->subdev_map[name];
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
HWComponent* MachineBase::get_comp_by_type(HWCompType type) {
|
|
std::string comp_name;
|
|
bool found = false;
|
|
|
|
for (auto it = this->comp_map.begin(); it != this->comp_map.end(); it++) {
|
|
if (it->second->supports_type(type)) {
|
|
comp_name = it->first;
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (found) {
|
|
return this->get_comp_by_name(comp_name);
|
|
}
|
|
|
|
for (auto it = this->subdev_map.begin(); it != this->subdev_map.end(); it++) {
|
|
if (it->second->supports_type(type)) {
|
|
comp_name = it->first;
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (found) {
|
|
return this->get_comp_by_name(comp_name);
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|