|
|
- /*
- * Copyright (C) 2021 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 "entity/commands.hpp"
- #include "entity/components/model.hpp"
- #include "entity/components/transform.hpp"
- #include "entity/components/copy-transform.hpp"
- #include "entity/components/snap.hpp"
- #include "entity/components/parent.hpp"
- #include "entity/components/celestial-body.hpp"
- #include "entity/components/terrain.hpp"
- #include "entity/components/name.hpp"
- #include <limits>
-
- namespace entity {
- namespace command {
-
- void translate(entity::registry& registry, entity::id eid, const float3& translation)
- {
- if (registry.has<component::transform>(eid))
- {
- component::transform& transform = registry.get<component::transform>(eid);
- transform.local.translation += translation;
- }
- }
-
- void move_to(entity::registry& registry, entity::id eid, const float3& position)
- {
- if (registry.has<component::transform>(eid))
- {
- component::transform& transform = registry.get<component::transform>(eid);
- transform.local.translation = position;
- }
- }
-
- void warp_to(entity::registry& registry, entity::id eid, const float3& position)
- {
- if (registry.has<component::transform>(eid))
- {
- component::transform& transform = registry.get<component::transform>(eid);
- transform.local.translation = position;
- transform.warp = true;
- }
- }
-
- void set_scale(entity::registry& registry, entity::id eid, const float3& scale)
- {
- if (registry.has<component::transform>(eid))
- {
- component::transform& transform = registry.get<component::transform>(eid);
- transform.local.scale = scale;
- }
- }
-
- void set_transform(entity::registry& registry, entity::id eid, const math::transform<float>& transform, bool warp)
- {
- if (registry.has<component::transform>(eid))
- {
- component::transform& component = registry.get<component::transform>(eid);
- component.local = transform;
- component.warp = warp;
- }
- }
-
- void place(entity::registry& registry, entity::id eid, entity::id celestial_body_id, double altitude, double latitude, double longitude)
- {
- if (registry.has<component::transform>(eid))
- {
- double x = 0.0;
- double y = altitude;
- double z = 0.0;
-
- if (registry.has<component::celestial_body>(celestial_body_id))
- {
- const component::celestial_body& celestial_body = registry.get<component::celestial_body>(celestial_body_id);
-
- x = longitude * math::two_pi<double> * celestial_body.radius;
- z = -latitude * math::two_pi<double> * celestial_body.radius;
-
- if (registry.has<component::terrain>(celestial_body_id))
- {
- const component::terrain& terrain = registry.get<component::terrain>(celestial_body_id);
-
- if (terrain.elevation != nullptr)
- {
- y += terrain.elevation(latitude, longitude);
- }
- }
- }
-
- component::transform& transform = registry.get<component::transform>(eid);
- transform.local.translation = math::type_cast<float>(double3{x, y, z});
- transform.warp = true;
- }
- }
-
- void assign_render_layers(entity::registry& registry, entity::id eid, unsigned int layers)
- {
- if (registry.has<component::model>(eid))
- {
- component::model model = registry.get<component::model>(eid);
- model.layers = layers;
- registry.replace<component::model>(eid, model);
-
- // Apply to child layers
- registry.view<component::parent>().each(
- [&](entity::id eid, auto& component)
- {
- if (component.parent == eid)
- assign_render_layers(registry, eid, layers);
- });
- }
- }
-
- void bind_transform(entity::registry& registry, entity::id source, entity::id target)
- {
- component::copy_transform copy_transform = {target};
- registry.assign_or_replace<component::copy_transform>(source, copy_transform);
- }
-
- math::transform<float> get_local_transform(entity::registry& registry, entity::id eid)
- {
- if (registry.has<component::transform>(eid))
- {
- const component::transform& component = registry.get<component::transform>(eid);
- return component.local;
- }
-
- return math::identity_transform<float>;
- }
-
- math::transform<float> get_world_transform(entity::registry& registry, entity::id eid)
- {
- if (registry.has<component::transform>(eid))
- {
- const component::transform& component = registry.get<component::transform>(eid);
- return component.world;
- }
-
- return math::identity_transform<float>;
- }
-
- void parent(entity::registry& registry, entity::id child, entity::id parent)
- {
- component::parent component;
- component.parent = parent;
- registry.assign_or_replace<component::parent>(child, component);
- }
-
- void rename(entity::registry& registry, entity::id eid, const std::string& name)
- {
- registry.assign_or_replace<component::name>(eid, name);
- }
-
- entity::id find(entity::registry& registry, const std::string& name)
- {
- auto view = registry.view<component::name>();
-
- for (auto eid: view)
- {
- if (view.get(eid).id == name)
- return eid;
- }
-
- return entt::null;
- }
-
- entity::id create(entity::registry& registry)
- {
- return registry.create();
- }
-
- entity::id create(entity::registry& registry, const std::string& name)
- {
- auto eid = registry.create();
- rename(registry, eid, name);
- return eid;
- }
-
- } // namespace command
- } // namespace entity
|