Entity.cpp

#include "Entity.h"
#include "Component.h"
 
#include "../Reflection/Field.h"
#include "../Reflection/TypeDatabase.h"
 
#include <cassert>
 
void Cogs::ComponentModel::Entity::registerType()
{
  Reflection::Field fields[] = {
    Reflection::Field(Reflection::Name("id"), &Entity::id),
  };
 
  Reflection::TypeDatabase::createAbstractType<Entity>().setFields(fields);
}
 
void Cogs::ComponentModel::Entity::addComponent(ComponentHandle handle)
{
  if (auto * component = handle.resolve(); component) {
    this->components.add(handle);
    component->setContainer(this);
  }
  else {
    assert(false);
  }
}
 
void Cogs::ComponentModel::Entity::removeComponent(ComponentHandle handle)
{
  if(auto * component = handle.resolve(); component) {
    component->setContainer(nullptr);
  }
 
  this->components.remove(handle);
}
 
Cogs::ComponentModel::Component * Cogs::ComponentModel::Entity::getComponentPtr(const Reflection::Type & type) const
{
  auto handle = getComponentHandle(type);
 
  return handle.resolve();
}
 
Cogs::ComponentModel::ComponentHandle Cogs::ComponentModel::Entity::getComponentHandle(const Reflection::Type & type) const
{
  return getComponentHandle(type.getTypeId());
}
 
Cogs::ComponentModel::ComponentHandle Cogs::ComponentModel::Entity::getComponentHandle(const Reflection::TypeId & id) const
{
  for (const auto & c : components) {
    if (c.typeId == id) return c;
  }
 
  for (const auto & c : components) {
    auto & componentType = Reflection::TypeDatabase::getType(c.typeId);
 
    if (componentType.canCastTo(id)) {
      return c;
    }
  }
 
  return ComponentHandle::Empty();
}
 
Cogs::ComponentModel::Component * Cogs::ComponentModel::Entity::getComponent(const Reflection::Name & componentName) const
{
  return getComponentPtr(Reflection::TypeDatabase::getType(componentName));
}
 
Cogs::ComponentModel::Component * Cogs::ComponentModel::Entity::getComponent(const Reflection::TypeId & id) const
{
  auto handle = getComponentHandle(id);
 
  return handle.resolve();
}
 
void Cogs::ComponentModel::Entity::getComponents(const Reflection::Name & componentName, ComponentCollectionBase& collection) const
{
  collection.clear();
 
  auto & componentType = Reflection::TypeDatabase::getType(componentName);
  
  for (const auto & c : components) {
    auto& type = Reflection::TypeDatabase::getType(c.typeId);
    if (type.canCastTo(componentType)) {
      collection.add(c);
    }
  }
}
 
Cogs::ComponentModel::ComponentHandle Cogs::ComponentModel::Entity::getComponentHandle(const Reflection::Name & componentName) const
{
  return getComponentHandle(Reflection::TypeDatabase::getType(componentName));
}
 
void Cogs::ComponentModel::Entity::setName(const StringView& name)
{
  this->name = std::string(name);
}