gui.h

/*
* sdformat-editor
* Copyright 2025 sdformat-editor
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Developer: Zaid Duraid, Ean Wheeler, Evan Vokey
*/
 
#ifndef SDFORMAT_EDITOR_GUI_HH_
#define SDFORMAT_EDITOR_GUI_HH_
 
// IMGUI dependencies
#include <stdio.h>
#include "imgui.h"
#include "imgui_impl_glfw.h"
#include "imgui_impl_opengl3.h"
#include <GLFW/glfw3.h> 
 
#include <interfaces/gui_interface.h>
#include "command_factory.h"
 
class GUI : public GUII
{
 
  public: GUI(const std::string &window_name, std::shared_ptr<SDFormatParserI> sdformat_parser, bool &success);
 
  public: ~GUI();
 
  private: void Initialize(const std::string &window_name, std::shared_ptr<SDFormatParserI> sdformat_parser, bool &success) override;
 
  private: bool ShouldClose() override;
 
  private: std::unique_ptr<CommandI> Update(std::shared_ptr<CommandFactoryI> command_factory) override;
 
  private: void SetPreventInputFlag(bool set) override;
 
  private: void OpenChoiceDialog(DialogMessage dialogMessage, std::vector<std::pair<std::string, bool>>& choices) override;
  
  private: static void GLFWErrorCallback(int error, const char *description);
 
  private: void DisplaySDFRootElement(std::unique_ptr<CommandI> &command, std::shared_ptr<SDFormatParserI> sdformat_parser, 
    std::shared_ptr<CommandFactoryI> command_factory);
 
  private: bool SetupNewFrame();
 
  private: void DrawCoreFrame(std::unique_ptr<CommandI>& command, std::shared_ptr<CommandFactoryI> command_factory);
 
  private: void CreateModifyAttributeDropdown(sdf::ParamPtr attribute, std::unique_ptr<CommandI> &command, 
    std::shared_ptr<CommandFactoryI> command_factory, int& unique_id);
 
  private: void CreateModifyElementDropdown(sdf::ElementPtr element, std::unique_ptr<CommandI> &command, 
    std::shared_ptr<CommandFactoryI> command_factory, int& unique_id);
 
  private: void CreateAppendElementDropdown(sdf::ElementPtr element, std::unique_ptr<CommandI> &command, 
    std::shared_ptr<CommandFactoryI> command_factory, int& unique_id);
 
  private: void CreateDropdown(const std::vector<std::string>& items, const std::vector<std::string>& item_descriptions, int& selected_item, int& unique_id);
 
  private: std::unique_lock<std::mutex> LockMutex() override;
 
  private: std::shared_ptr<CommandFactory> command_factory;
 
  private: std::atomic<bool> prevent_input_flag = false;
 
  private: bool model_viewer_running = false;
 
  private: bool render_collisions_in_model_viewer = false;
  
  private: GLFWwindow *window = nullptr;
 
  private: ImGuiIO *io = nullptr;
 
  private: std::shared_ptr<SDFormatParserI> sdformat_parser;
 
  private: sdf::ElementPtr element_to_append_to;
 
  private: ImVec4 background_colour;
 
  private: std::mutex gui_mutex;
 
  private: bool use_dropdown_for_editing_attribute = false;
 
  private: sdf::ParamPtr attribute_to_edit;
 
  private: sdf::ElementPtr element_to_edit;
 
  private: bool use_dropdown_for_editing_element = false;
 
};
 
#endif