ogre_scene_editor.hpp

Go to the documentation of this file.

#pragma once
 
#include "components/ui/imgui/imgui.h"
 
#include <limits.h>
#include <OGRE/OgreSceneManager.h>
 
#include "components/3rdparty/imgui/misc/imgui_stdlib.h"
#include "components/rendering/rendering.h"
 
class BlamImGuiWindow_OgreSceneEditor : public BlamImGuiWindow
{
private:
    void ShowSceneNodeControls(Ogre::Node* node, int iterator_index)
    {
        if (!node)
        {
            return;
        }
 
        std::string entry_name = node->getName() + "##" + node->getName() + std::to_string(iterator_index);
 
        if (ImGui::CollapsingHeader(entry_name.c_str()))
        {
            Ogre::Vector3 position = node->getPosition();
            Ogre::Quaternion orientation = node->getOrientation();
            Ogre::Vector3 scale = node->getScale();
 
            std::string position_name = "position##" + node->getName() + std::to_string(iterator_index);
            std::string orientation_name = "orientation##" + node->getName() + std::to_string(iterator_index);
            std::string rotation_name = "rotation##" + node->getName() + std::to_string(iterator_index);
            std::string scale_name = "scale##" + node->getName() + std::to_string(iterator_index);
 
            if (ImGui::DragFloat3(position_name.c_str(), (float*)&position))
            {
                node->setPosition(position);
            }
            
            BlamVector3 rotation = BlamVector3(orientation.getPitch().valueDegrees(), orientation.getYaw().valueDegrees(),
                orientation.getRoll().valueDegrees());
            
            if (ImGui::DragFloat3(rotation_name.c_str(), (float*)&rotation))
            {
                BlamVector4 quaternion = BlamStrings::Utils::Math::GetQuaternionFromRotation(rotation);
                node->setOrientation(quaternion.w, quaternion.x, quaternion.y, quaternion.z);
            }
 
            if (ImGui::DragFloat4(orientation_name.c_str(), (float*)&orientation, 0.001f))
            {
                node->setOrientation(orientation);
            }
 
            if (ImGui::DragFloat3(scale_name.c_str(), (float*)&scale))
            {
                node->setScale(scale);
            }
 
            Ogre::SceneNode* sn = (Ogre::SceneNode*)node;
            for (int i = 0; i < sn->numAttachedObjects(); i++)
            {
                Ogre::MovableObject* object = sn->getAttachedObject(i);
                std::string header_name = "attach" + std::to_string(i) + ": " + object->getName() + "##" + std::to_string(iterator_index) + "_" + std::to_string(i);
 
                if (ImGui::CollapsingHeader(header_name.c_str()))
                {
                    std::string flags_name = "query flags##" + node->getName() + std::to_string(iterator_index) + "_" + std::to_string(i);
 
                    std::string flags = std::to_string(sn->getAttachedObject(0)->getQueryFlags());
                    ImGui::InputText(flags_name.c_str(), &flags, ImGuiInputTextFlags_ReadOnly);
                }
            }
 
            if (node->numChildren() > 0)
            {
                std::string child_count_message = "children: " + std::to_string(node->numChildren());
                ImGui::Text(child_count_message.c_str());
 
                Ogre::Node::NodeVecIterator it = node->getChildIterator();
                ImGui::Indent(20.0f);
                int index = 0;
                while (it.current() != it.end())
                {
                    Ogre::Node* node = it.getNext();
                    ShowSceneNodeControls(node, index);
                    index++;
                }
                ImGui::Unindent(20.0f);
            }
        }
    }
 
public:
    void Draw()
    {
        if (!show)
        {
            return;
        }
 
        if (ImGui::Begin("Ogre Scene Editor", &show))
        {
            Demo::GraphicsSystem* ogre_graphics_system = Blam::Rendering::GetGraphicsSystem();
 
            if (ogre_graphics_system)
            {
                Ogre::SceneNode* root = ogre_graphics_system->getSceneManager()->getRootSceneNode();
                Ogre::SceneNode* root_2d = Blam::Rendering::Get2DSceneManager()->getRootSceneNode();
 
                if (ImGui::CollapsingHeader("world"))
                {
                    ShowSceneNodeControls(root, 0);
                }
 
                if (ImGui::CollapsingHeader("2d ortho"))
                {
                    ShowSceneNodeControls(root_2d, 0);
                }
            }
            else
            {
                ImGui::Text("ogre graphics system was nullptr");
            }
        }
        ImGui::End();
    }
};