entity_manager.h

Go to the documentation of this file.

#pragma once
 
#include <cassert>
#include <unordered_map>
 
#include <block/cblock.h>
#include <block/ctrlblock.h>
#include <block/iblock.h>
#include <block/mblock.h>
#include <block/shblock.h>
#include <block/ublock.h>
#include <entity/base.h>
#include <entity/entity.h>
 
namespace entities {
 
class EntityManager_ {
  static EntityClassifier unknown() { return {EntityClass::UNKNOWN, 0, 0, 0, 0, 0}; }
 
public:
  EntityClassifier read_classifier(bus::addr_t address) {
    // First look for classifiers
    auto aiter = classifiers.find(address);
    if(aiter != classifiers.end()) {
      return aiter->second;
    }
 
    // Then look for entities
    auto iter = entities.find(address);
    if (iter == entities.end()) {
      return unknown();
    }
    auto entity = iter->second;
    return entity->get_entity_classifier();
  }
 
  void addEntity(bus::addr_t address, EntityClassifier classifier) {
 
    blocks::FunctionBlock *new_entity_block;
    switch (classifier.class_enum) {
    case EntityClass::SH_BLOCK:
      new_entity_block = blocks::SHBlock::from_entity_classifier(classifier, address);
      break;
    case EntityClass::CTRL_BLOCK:
      new_entity_block = blocks::CTRLBlock::from_entity_classifier(classifier, address);
      break;
    case EntityClass::M_BLOCK:
      new_entity_block = blocks::MBlock::from_entity_classifier(classifier, address);
      break;
    case EntityClass::U_BLOCK:
      new_entity_block = blocks::UBlock::from_entity_classifier(classifier, address);
      break;
    case EntityClass::C_BLOCK:
      new_entity_block = blocks::CBlock::from_entity_classifier(classifier, address);
      break;
    case EntityClass::I_BLOCK:
      new_entity_block = blocks::IBlock::from_entity_classifier(classifier, address);
      break;
    default:
      throw std::runtime_error("runtime error");
    }
 
    assert(new_entity_block != nullptr);
    entities[address] = new_entity_block;
  }
 
  void addClassifier(bus::addr_t address, EntityClassifier classifier) {
    classifiers[address] = classifier;
  }
 
private:
  std::unordered_map<bus::addr_t, Entity *> entities;
  std::unordered_map<bus::addr_t, EntityClassifier> classifiers;
};
 
extern EntityManager_ EntityManager;
 
} // namespace entities