💿🐜 Antkeeper source code https://antkeeper.com
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

89 lines
2.8 KiB

/*
* Copyright (C) 2020 Christopher J. Howard
*
* This file is part of Antkeeper source code.
*
* Antkeeper source code is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Antkeeper source code is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Antkeeper source code. If not, see <http://www.gnu.org/licenses/>.
*/
#include "model-system.hpp"
#include "entity/components/transform-component.hpp"
using namespace ecs;
model_system::model_system(entt::registry& registry, ::scene& scene):
entity_system(registry),
scene(scene)
{
registry.on_construct<model_component>().connect<&model_system::on_model_construct>(this);
registry.on_replace<model_component>().connect<&model_system::on_model_replace>(this);
registry.on_destroy<model_component>().connect<&model_system::on_model_destroy>(this);
}
void model_system::update(double t, double dt)
{
registry.view<transform_component, model_component>().each(
[this](auto entity, auto& transform, auto& model)
{
model_instance* instance = model_instances[entity];
instance->set_transform(transform.transform);
if (transform.warp)
{
instance->get_transform_tween().update();
instance->update_tweens();
transform.warp = false;
}
});
}
void model_system::update_model_and_materials(entt::entity entity, model_component& model)
{
if (auto model_it = model_instances.find(entity); model_it != model_instances.end())
{
model_it->second->set_model(model.model);
model_it->second->set_instanced((model.instance_count > 0), model.instance_count);
for (auto material_it = model.materials.begin(); material_it != model.materials.end(); ++material_it)
{
model_it->second->set_material(material_it->first, material_it->second);
}
}
}
void model_system::on_model_construct(entt::registry& registry, entt::entity entity, model_component& model)
{
::model_instance* model_instance = new ::model_instance();
scene.add_object(model_instance);
model_instances[entity] = model_instance;
update_model_and_materials(entity, model);
}
void model_system::on_model_replace(entt::registry& registry, entt::entity entity, model_component& model)
{
update_model_and_materials(entity, model);
}
void model_system::on_model_destroy(entt::registry& registry, entt::entity entity)
{
if (auto it = model_instances.find(entity); it != model_instances.end())
{
scene.remove_object(it->second);
delete it->second;
model_instances.erase(it);
}
}