model_viewer.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_MODEL_VIEWER_HH_
#define SDFORMAT_EDITOR_MODEL_VIEWER_HH_
 
#include <interfaces/model_viewer_interface.h>
 
// Ogre dependencies
#include "Ogre.h"
#include "OgreApplicationContext.h"
#include "OgreInput.h"
#include "OgreRTShaderSystem.h"
#include "OgreWindowEventUtilities.h"
#include "OgreAssimpLoader.h"
#include "OgreCameraMan.h"
 
// GLM (for vector math)
#include <glm/glm.hpp>
#include <glm/gtc/quaternion.hpp>
 
#include <string>
#include <vector>
#include <filesystem>
#include <queue>
 
class ModelViewer : public ModelViewerI
{
 
    public: ModelViewer() : ctx("SDFormat Editor Model Viewer") // Initialize ctx here
    {
    }
 
    private: void Initialize() override;
 
    private: void RenderFrame() override;
 
    private: void Quit() override;
 
    public: bool IsRunning() override;
 
    private: void Deinitialize() ;
 
    private: std::mutex model_viewer_mutex;
 
    private: bool should_quit = false;
 
    private: bool is_running = false;
 
    public: void AddModel(ModelInfo model_info) override;
 
    public: void AddModel(PresetModelInfo model_info) override;
 
    public: void ResetModels() override;
    
    private: std::mutex& GetMutex() override;
 
    private: void HandleAddModelQueue();
 
    private: void HandleResetModelsFlag();
 
    private: OgreBites::ApplicationContext ctx;
 
    private: Ogre::Root* ogreRoot;
 
    private: Ogre::SceneManager* scnMgr;
 
    private: Ogre::RTShader::ShaderGenerator *shadergen;
 
    private: Ogre::Light *sceneLight;
 
    private: Ogre::SceneNode *sceneLightNode;
 
    private: Ogre::Camera *sceneCamera;
 
    private: Ogre::SceneNode* sceneCameraNode;
 
    private: OgreBites::CameraMan* cameraController;
 
    private: std::queue<ModelInfo> add_model_queue;
 
    private:
        class ModelViewerKeyHandler : public OgreBites::InputListener
        {
            private: ModelViewer *m;
            public:void addModelViewerContext(ModelViewer *m);
            public: bool keyPressed(const OgreBites::KeyboardEvent &evt) override;
            public: bool mouseWheelRolled(const OgreBites::MouseWheelEvent &evt) override;
            public: bool mouseMoved(const OgreBites::MouseMotionEvent& evt) override;
            public: bool mousePressed(const OgreBites::MouseButtonEvent& evt) override;
            public: bool mouseReleased(const OgreBites::MouseButtonEvent& evt) override;
        };
 
    private: ModelViewerKeyHandler keyHandler;
    
    // private: ModelViewerWindowEventHandler windowEventHandler;
    
    // private:
    //     class ModelViewerWindowEventHandler : public OgreBites::WindowEventListener
    //     {
    //         private: ModelViewer *m;
    //         public: void addModelViewerContext(ModelViewer *m);
    //         public: void windowClosed(Ogre::RenderWindow* rw) override;
    //         public: void windowMoved(Ogre::RenderWindow* rw) override;
    //     };
 
    private: unsigned long long unique_naming_counter = 0;
 
    private: Ogre::ColourValue color_list[6] = {
        Ogre::ColourValue(3.3699f, 3.1256f, 0.633f),
        Ogre::ColourValue(0.549f, 3.1569f, 0.1569f),
        Ogre::ColourValue(4.1325f, 1.1339f, 0.234f),
        Ogre::ColourValue(1.0f, 0.5f, 0.0f), 
        Ogre::ColourValue(0.0f, 0.5f, 1.0f), 
        Ogre::ColourValue(0.5f, 0.0f, 0.5f)  
    };
 
    private: unsigned int color_list_index = 0;
 
    private: std::atomic<bool> reset_models_flag;
};
 
#endif