visitor.cpp

Go to the documentation of this file.

 
#include "entity/visitor.h"
 
#include "block/mblock_mul.h"
#include "mode/mode.h"
#include "proto/main.pb.h"
 
namespace entities {
 
 
pb_OverloadStatus_Element& OverloadVisitor::create(){
  auto& elem = m_elements.emplace_back();
  copy_entity(elem.entity);
  return elem;
}
 
void OverloadVisitor::to(pb_OverloadStatus& status){
  status.global_overload = mode::is_global_overload_active();
  status.elements_count = m_elements.size();
  status.elements = new pb_OverloadStatus_Element[status.elements_count];
  std::copy(m_elements.begin(), m_elements.end(), status.elements);
}
 
void OverloadVisitor::handle_visit(Entity* entity) {
  if (entity->is_entity_class(blocks::MMulBlock::CLASS_)) {
    auto *mblock = static_cast<blocks::MBlock *>(entity);
    mblock->collect_overloads(*this);
  }
}
 
class Setup;
 
pb_Item& ExtractVisitor::create(int config_type) const{
  auto& config = m_setup.add_config();
  config.which_kind = config_type;
  config.has_entity = true;
  copy_entity(config.entity);
  return config;
}
 
void Setup::reset() {
  m_items.clear();
}
 
std::vector<pb_Item>& Setup::configs() {
  return m_items;
}
 
void Setup::extract(Entity* entity, ExtractSettings config) {
  ExtractVisitor collector(*this, config);
  collector.visit(entity, config.recursive);
}
 
void Setup::set(const pb_Module &setup) {
  reset();
  extend(setup.items, setup.items + setup.items_count);
}
 
void Setup::get(pb_Module &setup) const {
  setup.items_count = m_items.size();
  setup.items = new pb_Item[setup.items_count];
  std::copy(m_items.begin(), m_items.end(), setup.items);
}
 
 
void Setup::extend(const pb_Item* begin, const pb_Item* end) {
  m_items.reserve(m_items.size() + std::distance(begin, end));
  for (auto it = begin ; it != end ; ++it) {
    m_items.push_back(*it);
  }
}
 
UnitResult Setup::apply(Entity *entity, bool write_hardware) const {
  for (auto& config : m_items) {
    Path path;
    path.push(config.entity.path);
 
    auto resolved_entity = entity->resolve_child_entity(path);
 
    if (resolved_entity == nullptr) continue;
    if (!resolved_entity) {
      return UnitResult::err("Could not resolve entity to apply config to");
    }
 
    //empty config
    if (config.which_kind == 0)
      continue;
 
    if (!TRY(resolved_entity->config(config))) {
      LOG4("unhandled config in ", resolved_entity->get_entity_id().c_str(), " with config id ", config.which_kind);
      //config is not beeing applied to anything
      //TODO: has to be handled properly
    }
  }
 
  if (!write_hardware) return UnitResult::ok();
  return entity->write_to_hardware();
}
 
pb_Item& Setup::add_config(){
  return m_items.emplace_back();
}
 
}