world_editor.hpp

Go to the documentation of this file.

#pragma once
 
#include "components/ui/imgui/imgui.h"
 
#include <limits.h>
 
#include "components/core/world/world.h"
#include "components/3rdparty/imgui/misc/imgui_stdlib.h"
 
class BlamImGuiWindow_WorldEditor : public BlamImGuiWindow
{
private:
    BlamWorldObjectType selected_primitive_type = BlamWorldObjectType::PrimitiveCuboid; 
    bool show_add_primitive_dialog = false;     
    std::string primitive_type_name = "cuboid"; 
    int index_to_remove = -1; 
 
public:
 
    void ShowAddPrimitivePopup()
    {
        ImGui::OpenPopup("Add Object");
        if (ImGui::BeginPopupModal("Add Object", &show_add_primitive_dialog))
        {
            if (ImGui::BeginCombo("object type", primitive_type_name.c_str()))
            {
                if (ImGui::Selectable("cuboid"))
                {
                    selected_primitive_type = BlamWorldObjectType::PrimitiveCuboid;
                    primitive_type_name = "cuboid";
                }
 
                if (ImGui::Selectable("cylinder"))
                {
                    selected_primitive_type = BlamWorldObjectType::PrimitiveCylinder;
                    primitive_type_name = "cylinder";
                }
 
                if (ImGui::Selectable("sphere"))
                {
                    selected_primitive_type = BlamWorldObjectType::PrimitiveSphere;
                    primitive_type_name = "sphere";
                }
 
                if (ImGui::Selectable("line"))
                {
                    selected_primitive_type = BlamWorldObjectType::PrimitiveLine;
                    primitive_type_name = "line";
                }
 
                if (ImGui::Selectable("grid"))
                {
                    selected_primitive_type = BlamWorldObjectType::GizmoGrid;
                    primitive_type_name = "grid";
                }
 
                if (ImGui::Selectable("text"))
                {
                    selected_primitive_type = BlamWorldObjectType::Text3D;
                    primitive_type_name = "text";
                }
 
                ImGui::EndCombo();
            }
 
            if (ImGui::Button("OK"))
            {
                if (Blam::World::GetWorldState())
                {
                    BlamWorldObject* new_primitive = nullptr;
 
                    if (selected_primitive_type == BlamWorldObjectType::PrimitiveCuboid)
                    {
                        new_primitive = new BlamWorldObject_PrimitiveCuboid(Blam::World::GetWorldState());
                    }
                    else if (selected_primitive_type == BlamWorldObjectType::PrimitiveCylinder)
                    {
                        new_primitive = new BlamWorldObject_PrimitiveCylinder(Blam::World::GetWorldState());
                    }
                    else if (selected_primitive_type == BlamWorldObjectType::PrimitiveSphere)
                    {
                        new_primitive = new BlamWorldObject_PrimitiveSphere(Blam::World::GetWorldState());
                    }
                    else if (selected_primitive_type == BlamWorldObjectType::PrimitiveLine)
                    {
                        new_primitive = new BlamWorldObject_PrimitiveLine(Blam::World::GetWorldState());
                    }
                    else if (selected_primitive_type == BlamWorldObjectType::GizmoGrid)
                    {
                        new_primitive = new BlamWorldObject_GizmoGrid(Blam::World::GetWorldState());
                    }
                    else if (selected_primitive_type == BlamWorldObjectType::Text3D)
                    {
                        new_primitive = new BlamWorldObject_Text3D(Blam::World::GetWorldState());
                    }
 
                    if (new_primitive)
                    {
                        Blam::World::GetWorldState()->AddObject(new_primitive);
                    }
                }
 
                show_add_primitive_dialog = false;
            }
 
            ImGui::SameLine();
 
            if (ImGui::Button("Cancel"))
            {
                show_add_primitive_dialog = false;
            }
 
            ImGui::EndPopup();
        }
    }
 
    void Draw()
    {
        if (!show)
        {
            return;
        }
 
        if (ImGui::Begin("World State Editor", &show))
        {
            BlamWorldState* world_state = Blam::World::GetWorldState();
 
            if (ImGui::Button("initialize world state"))
            {
                Blam::World::InitializeWorldState();
            }
 
            ImGui::SameLine();
 
            if (ImGui::Button("destroy world state"))
            {
                Blam::World::DestroyWorldState();
                world_state = nullptr;
            }
 
            if (world_state)
            {
                if (ImGui::Button("add primitive"))
                {
                    show_add_primitive_dialog = true;
                }
 
                ImGui::SameLine();
 
                if (ImGui::Button("delete @ "))
                {
                    if (index_to_remove > -1 && index_to_remove < world_state->GetObjectCount())
                    {
                        BlamWorldObject* object = world_state->GetObjectAtIndex(index_to_remove);
                        world_state->RemoveObject(object->id);
                        object = nullptr;
                    }
                }
 
                ImGui::SameLine();
 
                ImGui::InputInt("", &index_to_remove);
 
                for (int i = 0; i < world_state->GetObjectCount(); i++)
                {
                    BlamWorldObject* object = world_state->GetObjectAtIndex(i);
 
                    if (!object)
                    {
                        continue;
                    }
 
                    std::string entry_name = object->GetName() + " (object " + std::to_string(i) + ")";
 
                    if (ImGui::CollapsingHeader(entry_name.c_str()))
                    {
                        ImGui::PushID(object->id);
                        object->ShowImGuiEditControls();
                        ImGui::PopID();
                    }
                }
            }
            else
            {
                ImGui::Text("world state not loaded");
            }
        }
        ImGui::End();
 
        if (show_add_primitive_dialog)
        {
            ShowAddPrimitivePopup();
        }
    }
};