Skip to content

Commit

Permalink
feat(gizmos): add global/local axis toggle
Browse files Browse the repository at this point in the history
  • Loading branch information
@DiogoMendonc-a
DiogoMendonc-a committed Mar 8, 2024
1 parent 0b60c04 commit fb0e220
Show file tree
Hide file tree
Showing 2 changed files with 107 additions and 56 deletions.
5 changes: 4 additions & 1 deletion tools/tesseratos/include/tesseratos/transform_gizmo/plugin.hpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,10 @@ namespace tesseratos
/// @defgroup tesseratos-transform-gizmo-plugin Transform gizmo
/// @ingroup tesseratos
/// @brief Add a gizmo that allows changing an entity's position.

///
/// ## Settings
/// - `"transformGizmo.useLocalAxis"` - whether to use local instead of global axis (default: `true`).
///
/// @brief Plugin entry function.
/// @param cubos @b CUBOS. main class
/// @ingroup tesseratos-transform-gizmo-plugin
Expand Down
158 changes: 103 additions & 55 deletions tools/tesseratos/src/tesseratos/transform_gizmo/plugin.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,12 @@
#include <glm/gtx/intersect.hpp>
#include <glm/gtx/vector_angle.hpp>

#include <cubos/core/reflection/external/glm.hpp>

#include <cubos/engine/gizmos/plugin.hpp>
#include <cubos/engine/input/plugin.hpp>
#include <cubos/engine/renderer/plugin.hpp>
#include <cubos/engine/settings/plugin.hpp>
#include <cubos/engine/transform/plugin.hpp>
#include <cubos/engine/window/plugin.hpp>

Expand All @@ -28,6 +31,7 @@ using cubos::engine::LocalToWorld;
using cubos::engine::Position;
using cubos::engine::Rotation;
using cubos::engine::Scale;
using cubos::engine::Settings;

using namespace tesseratos;

Expand Down Expand Up @@ -61,15 +65,30 @@ static glm::vec3 mouseWorldCoordinates(glm::ivec2 windowSize, glm::ivec2 mousePo
return intersectLinePlane(cameraPosition, dir, transformPosition, transformNormal);
}

static void checkMovement(Position& position, glm::vec3 globalPosition, glm::vec3 transformVector,
glm::vec3 transformNormal, glm::ivec2 windowSize, const Input& input, const glm::mat4& pv,
glm::vec3 cameraPosition)
static void checkAxialMovement(Position& position, glm::vec3 globalPosition, glm::vec3 transformVector,
glm::vec3 transformNormal, glm::ivec2 windowSize, const Input& input,
const glm::mat4& pv, glm::vec3 cameraPosition)
{
auto mousePosWorld =
mouseWorldCoordinates(windowSize, input.mousePosition(), globalPosition, transformNormal, cameraPosition, pv);
auto oldMousePosWorld = mouseWorldCoordinates(windowSize, input.previousMousePosition(), globalPosition,
transformNormal, cameraPosition, pv);
auto mousDisp = (mousePosWorld - oldMousePosWorld);
position.vec +=
(glm::dot(mousDisp, transformVector) / glm::dot(transformVector, transformVector)) * transformVector;
}

static void checkPlanarMovement(Position& position, glm::vec3 globalPosition, glm::vec3 transformNormal,
glm::ivec2 windowSize, const Input& input, const glm::mat4& pv,
glm::vec3 cameraPosition)
{
auto mousePosWorld =
mouseWorldCoordinates(windowSize, input.mousePosition(), globalPosition, transformNormal, cameraPosition, pv);
auto oldMousePosWorld = mouseWorldCoordinates(windowSize, input.previousMousePosition(), globalPosition,
transformNormal, cameraPosition, pv);
position.vec += (mousePosWorld - oldMousePosWorld) * transformVector;
auto mousDisp = (mousePosWorld - oldMousePosWorld);
position.vec +=
mousDisp - (glm::dot(mousDisp, transformNormal) / glm::dot(transformNormal, transformNormal)) * transformNormal;
}

static void checkRotation(Rotation& rotation, glm::vec3 globalPosition, glm::vec3 axisVector, glm::ivec2 windowSize,
Expand All @@ -93,7 +112,7 @@ static void checkRotation(Rotation& rotation, glm::vec3 globalPosition, glm::vec

static void drawPositionGizmo(Query<const Camera&, LocalToWorld&> cameraQuery,
Query<Position&, LocalToWorld&> positionQuery, Gizmos& gizmos, const Input& input,
const Window& window, Entity cameraEntity, Entity selectedEntity)
const Window& window, Entity cameraEntity, Entity selectedEntity, bool useLocalAxis)
{
auto [camera, cameraLtw] = *cameraQuery.at(cameraEntity);

Expand All @@ -117,74 +136,89 @@ static void drawPositionGizmo(Query<const Camera&, LocalToWorld&> cameraQuery,
glm::vec3 xzColor = {1, 0, 1};
glm::vec3 yzColor = {0, 1, 1};

glm::vec3 rightVector = {1, 0, 0};
glm::vec3 upVector = {0, 1, 0};
glm::vec3 forwardVector = {0, 0, 1};

if (useLocalAxis)
{

rightVector = glm::normalize(localToWorld.worldRotation() * rightVector);
upVector = glm::normalize(localToWorld.worldRotation() * upVector);
forwardVector = glm::normalize(localToWorld.worldRotation() * forwardVector);
}

// Axis movement
if (gizmos.locked("transform_gizmo.position.x"))
{
checkMovement(position, localToWorld.worldPosition(), {1, 0, 0}, {0, 1, 0}, window->size(), input, pv,
cameraPosition);
checkAxialMovement(position, localToWorld.worldPosition(), rightVector, upVector, window->size(), input, pv,
cameraPosition);
xColor = {1, 1, 0};
}
if (gizmos.locked("transform_gizmo.position.y"))
{
checkMovement(position, localToWorld.worldPosition(), {0, 1, 0}, {1, 0, 0}, window->size(), input, pv,
cameraPosition);
checkAxialMovement(position, localToWorld.worldPosition(), upVector, rightVector, window->size(), input, pv,
cameraPosition);
yColor = {1, 1, 0};
}
if (gizmos.locked("transform_gizmo.position.z"))
{
checkMovement(position, localToWorld.worldPosition(), {0, 0, 1}, {0, 1, 0}, window->size(), input, pv,
cameraPosition);
checkAxialMovement(position, localToWorld.worldPosition(), forwardVector, upVector, window->size(), input, pv,
cameraPosition);
zColor = {1, 1, 0};
}

// Planar movement
if (gizmos.locked("transform_gizmo.position.xy"))
{
checkMovement(position, localToWorld.worldPosition(), {1, 1, 0}, {0, 0, 1}, window->size(), input, pv,
cameraPosition);
xyColor = {1, 1, 1};
}
if (gizmos.locked("transform_gizmo.position.xz"))
if (!useLocalAxis)
{
checkMovement(position, localToWorld.worldPosition(), {1, 0, 1}, {0, 1, 0}, window->size(), input, pv,
cameraPosition);
xzColor = {1, 1, 1};
// Planar movement
if (gizmos.locked("transform_gizmo.position.xy"))
{
checkPlanarMovement(position, localToWorld.worldPosition(), forwardVector, window->size(), input, pv,
cameraPosition);
xyColor = {1, 1, 1};
}
if (gizmos.locked("transform_gizmo.position.xz"))
{
checkPlanarMovement(position, localToWorld.worldPosition(), upVector, window->size(), input, pv,
cameraPosition);
xzColor = {1, 1, 1};
}
if (gizmos.locked("transform_gizmo.position.yz"))
{
checkPlanarMovement(position, localToWorld.worldPosition(), rightVector, window->size(), input, pv,
cameraPosition);
yzColor = {1, 1, 1};
}

gizmos.color(xyColor);
gizmos.drawBox("transform_gizmo.position.xy", localToWorld.worldPosition() - glm::vec3{0, 0, 0.015F},
localToWorld.worldPosition() + glm::vec3{0.25F, 0.25F, 0.015F});
gizmos.color(xzColor);
gizmos.drawBox("transform_gizmo.position.xz", localToWorld.worldPosition() - glm::vec3{0, 0.015F, 0},
localToWorld.worldPosition() + glm::vec3{0.25F, 0.015F, 0.25F});
gizmos.color(yzColor);
gizmos.drawBox("transform_gizmo.position.yz", localToWorld.worldPosition() - glm::vec3{0.015F, 0, 0},
localToWorld.worldPosition() + glm::vec3{0.015F, 0.25F, 0.25F});
}
if (gizmos.locked("transform_gizmo.position.yz"))
{
checkMovement(position, localToWorld.worldPosition(), {0, 1, 1}, {1, 0, 0}, window->size(), input, pv,
cameraPosition);
yzColor = {1, 1, 1};
}

gizmos.color(xyColor);
gizmos.drawBox("transform_gizmo.position.xy", localToWorld.worldPosition() - glm::vec3{0, 0, 0.015F},
localToWorld.worldPosition() + glm::vec3{0.25F, 0.25F, 0.015F});
gizmos.color(xzColor);
gizmos.drawBox("transform_gizmo.position.xz", localToWorld.worldPosition() - glm::vec3{0, 0.015F, 0},
localToWorld.worldPosition() + glm::vec3{0.25F, 0.015F, 0.25F});
gizmos.color(yzColor);
gizmos.drawBox("transform_gizmo.position.yz", localToWorld.worldPosition() - glm::vec3{0.015F, 0, 0},
localToWorld.worldPosition() + glm::vec3{0.015F, 0.25F, 0.25F});

gizmos.color({0.7F, 0.7F, 0.7F});
gizmos.drawBox("", localToWorld.worldPosition() + glm::vec3{0.03F, 0.03F, 0.03F},
localToWorld.worldPosition() - glm::vec3{0.03F, 0.03F, 0.03F});

gizmos.color(xColor);
gizmos.drawArrow("transform_gizmo.position.x", localToWorld.worldPosition() + glm::vec3{0.03F, 0, 0}, {1, 0, 0},
gizmos.drawArrow("transform_gizmo.position.x", localToWorld.worldPosition() + rightVector * 0.03F, rightVector,
0.03F, 0.07F, 0.7F);
gizmos.color(yColor);
gizmos.drawArrow("transform_gizmo.position.y", localToWorld.worldPosition() + glm::vec3{0, 0.03F, 0}, {0, 1, 0},
0.03F, 0.07F, 0.7F);
gizmos.drawArrow("transform_gizmo.position.y", localToWorld.worldPosition() + upVector * 0.03F, upVector, 0.03F,
0.07F, 0.7F);
gizmos.color(zColor);
gizmos.drawArrow("transform_gizmo.position.z", localToWorld.worldPosition() + glm::vec3{0, 0, 0.03F}, {0, 0, 1},
gizmos.drawArrow("transform_gizmo.position.z", localToWorld.worldPosition() + forwardVector * 0.03F, forwardVector,
0.03F, 0.07F, 0.7F);
}

static void drawRotationGizmo(Query<const Camera&, LocalToWorld&> cameraQuery,
Query<Rotation&, LocalToWorld&> positionQuery, Gizmos& gizmos, const Input& input,
const Window& window, Entity cameraEntity, Entity selectedEntity)
const Window& window, Entity cameraEntity, Entity selectedEntity, bool useLocalAxis)
{
auto [camera, cameraLtw] = *cameraQuery.at(cameraEntity);

Expand All @@ -204,31 +238,42 @@ static void drawRotationGizmo(Query<const Camera&, LocalToWorld&> cameraQuery,
glm::vec3 yColor = {0, 1, 0};
glm::vec3 zColor = {0, 0, 1};

glm::vec3 rightVector = {1, 0, 0};
glm::vec3 upVector = {0, 1, 0};
glm::vec3 forwardVector = {0, 0, 1};

if (useLocalAxis)
{
rightVector = glm::normalize(localToWorld.worldRotation() * rightVector);
upVector = glm::normalize(localToWorld.worldRotation() * upVector);
forwardVector = glm::normalize(localToWorld.worldRotation() * forwardVector);
}

if (gizmos.locked("transform_gizmo.rotation.x"))
{
checkRotation(rotation, localToWorld.worldPosition(), {1, 0, 0}, window->size(), input, pv, cameraPosition);
checkRotation(rotation, localToWorld.worldPosition(), rightVector, window->size(), input, pv, cameraPosition);
xColor = {1, 1, 0};
}
if (gizmos.locked("transform_gizmo.rotation.y"))
{
checkRotation(rotation, localToWorld.worldPosition(), {0, 1, 0}, window->size(), input, pv, cameraPosition);
checkRotation(rotation, localToWorld.worldPosition(), upVector, window->size(), input, pv, cameraPosition);
yColor = {1, 1, 0};
}
if (gizmos.locked("transform_gizmo.rotation.z"))
{
checkRotation(rotation, localToWorld.worldPosition(), {0, 0, 1}, window->size(), input, pv, cameraPosition);
checkRotation(rotation, localToWorld.worldPosition(), forwardVector, window->size(), input, pv, cameraPosition);
zColor = {1, 1, 0};
}

gizmos.color(xColor);
gizmos.drawRing("transform_gizmo.rotation.x", localToWorld.worldPosition() + glm::vec3{0.01F, 0, 0},
localToWorld.worldPosition() - glm::vec3{0.01F, 0, 0}, 1.4F, 1.25F);
gizmos.drawRing("transform_gizmo.rotation.x", localToWorld.worldPosition() + (rightVector * 0.1F),
localToWorld.worldPosition() - (rightVector * 0.01F), 1.4F, 1.25F);
gizmos.color(yColor);
gizmos.drawRing("transform_gizmo.rotation.y", localToWorld.worldPosition() + glm::vec3{0, 0.01F, 0},
localToWorld.worldPosition() - glm::vec3{0, 0.01F, 0}, 1.4F, 1.25F);
gizmos.drawRing("transform_gizmo.rotation.y", localToWorld.worldPosition() + (upVector * 0.1F),
localToWorld.worldPosition() - (upVector * 0.01F), 1.4F, 1.25F);
gizmos.color(zColor);
gizmos.drawRing("transform_gizmo.rotation.z", localToWorld.worldPosition() + glm::vec3{0, 0, 0.01F},
localToWorld.worldPosition() - glm::vec3{0, 0, 0.01F}, 1.4F, 1.25F);
gizmos.drawRing("transform_gizmo.rotation.z", localToWorld.worldPosition() + (forwardVector * 0.1F),
localToWorld.worldPosition() - (forwardVector * 0.01F), 1.4F, 1.25F);
}

void tesseratos::transformGizmoPlugin(Cubos& cubos)
Expand All @@ -237,12 +282,13 @@ void tesseratos::transformGizmoPlugin(Cubos& cubos)
cubos.addPlugin(cubos::engine::transformPlugin);
cubos.addPlugin(cubos::engine::gizmosPlugin);
cubos.addPlugin(cubos::engine::inputPlugin);
cubos.addPlugin(cubos::engine::settingsPlugin);

cubos.system("draw Transform Gizmo")
.after("cubos.transform.update")
.tagged("cubos.imgui")
.call([](const EntitySelector& entitySelector, const ActiveCameras& activeCameras, const Window& window,
const Input& input, Gizmos& gizmos, Query<Position&, LocalToWorld&> positionQuery,
const Input& input, Settings& settings, Gizmos& gizmos, Query<Position&, LocalToWorld&> positionQuery,
Query<Rotation&, LocalToWorld&> rotationQuery, Query<const Camera&, LocalToWorld&> cameraQuery) {
if (entitySelector.selection.isNull())
{
Expand All @@ -261,9 +307,11 @@ void tesseratos::transformGizmoPlugin(Cubos& cubos)
return;
}

bool useLocal = settings.getBool("transformGizmo.useLocalAxis", true);

drawPositionGizmo(cameraQuery, positionQuery, gizmos, input, window, activeCameras.entities[0],
entitySelector.selection);
entitySelector.selection, useLocal);
drawRotationGizmo(cameraQuery, rotationQuery, gizmos, input, window, activeCameras.entities[0],
entitySelector.selection);
entitySelector.selection, useLocal);
});
}

0 comments on commit fb0e220

Please sign in to comment.