💿🐜 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.

197 lines
5.6 KiB

/*
* 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