IfcOpenShell/src/qtviewer/IfcViewerWidget.cpp

#include "IfcViewerWidget.h"
#include <OpenGL/OpenGL.h>
#include <osg/ShapeDrawable>
#include <osg/Material>
#include <osg/MatrixTransform>
#include <osg/ComputeBoundsVisitor>
#include <QMouseEvent>
#include <QWheelEvent>
#include <string>
#include <vector>
#include "MainWindow.h"
#include "MessageLogger.h"
#include "osg/Geode"
#include "osg/Geometry"
#include "osg/StateSet"
#include "osg/Vec3"
#include "osg/ref_ptr"
#include "osgGA/TrackballManipulator"

IfcViewerWidget::IfcViewerWidget(qreal dpiScale, QWidget *parent) : 
    QOpenGLWidget(parent),
    m_dpiScale(dpiScale),
    m_graphicsWindow(new osgViewer::GraphicsWindowEmbedded(
                         this->x(), this->y(),
                         this->width(), this->height()
                     )),
    m_viewer(new osgViewer::Viewer),
    root(new osg::Group)
{ 
    m_clearColor = osg::Vec4(0.8, 0.8, 0.8, 1.0);

    //debug - override
    //std::string filePath = "/Users/onizudb/Downloads/IFC/wallAtOrigin.ifc";
    //std::string filePath = "/Users/onizudb/Downloads/IFC/FromAC_IFC4ref.ifc";
    //this->loadFile(filePath);
}

void IfcViewerWidget::initializeGL()
{
    MessageLogger::log("Press `Cmd/Ctrl + o` or use the File menu to open an IFC file");

    // Set up the rendering context, load shaders and other resources, etc.:

    // Set the root node as the scene data for the viewer
    m_viewer->setSceneData(root);

    osg::StateSet* state = root->getOrCreateStateSet();
    state->setMode(GL_DEPTH_TEST, osg::StateAttribute::ON);

    m_viewer->getCamera()->setViewport(0, 0, this->width(), this->height());
    m_viewer->getCamera()->setClearColor(m_clearColor);
    m_viewer->getCamera()->setGraphicsContext(m_graphicsWindow);

    // set camera manipulator, etc.
    setupViewController();

    m_viewer->setThreadingModel(osgViewer::Viewer::SingleThreaded);
    m_viewer->realize();
}

void IfcViewerWidget::resizeGL(int w, int h)
{
    this->getEventQueue()->windowResize(this->x() * m_dpiScale, this->y() * m_dpiScale, w * m_dpiScale, h * m_dpiScale);
    m_graphicsWindow->resized(this->x() * m_dpiScale, this->y() * m_dpiScale, w * m_dpiScale, h * m_dpiScale);

    m_viewer->getCamera()->setViewport(0, 0, this->width() * m_dpiScale, this->height() * m_dpiScale);
    
    const float aspectRatio = static_cast<float>(w) / static_cast<float>(h);
    m_viewer->getCamera()->setProjectionMatrixAsPerspective(30.0f, aspectRatio, 1.0f, 1000.0f);
}

void IfcViewerWidget::paintGL()
{
    // Render OSG scene
    m_viewer->frame();
}

void IfcViewerWidget::mouseMoveEvent(QMouseEvent *event)
{
    this->getEventQueue()->mouseMotion(event->position().x() * m_dpiScale, event->position().y() * m_dpiScale);
}

void IfcViewerWidget::mousePressEvent(QMouseEvent *event)
{
    unsigned int button = this->getMouseButtonNum(event);
    this->getEventQueue()->mouseButtonPress(event->position().x() * m_dpiScale, event->position().y() * m_dpiScale, button);
}

void IfcViewerWidget::mouseReleaseEvent(QMouseEvent *event)
{
    unsigned int button = this->getMouseButtonNum(event);
    this->getEventQueue()->mouseButtonRelease(event->position().x() * m_dpiScale, event->position().y() * m_dpiScale, button);
}

void IfcViewerWidget::wheelEvent(QWheelEvent *event)
{
    int delta = event->angleDelta().y();
    osgGA::GUIEventAdapter::ScrollingMotion motion = delta > 0 ?
            osgGA::GUIEventAdapter::SCROLL_UP : osgGA::GUIEventAdapter::SCROLL_DOWN;
    this->getEventQueue()->mouseScroll(motion);
}

bool IfcViewerWidget::event(QEvent* event)
{
    bool handled = QOpenGLWidget::event(event);
    this->update();
    return handled;
}

osgGA::EventQueue* IfcViewerWidget::getEventQueue() const 
{
    osgGA::EventQueue* eventQueue = m_graphicsWindow->getEventQueue();
    return eventQueue;
}

unsigned int IfcViewerWidget::getMouseButtonNum(QMouseEvent* event)
{
    unsigned int button = 0;
    switch (event->button()){
    case Qt::LeftButton:
        button = 1;
        break;
    case Qt::MiddleButton:
        button = 2;
        break;
    case Qt::RightButton:
        button = 3;
        break;
    default:
        break;
    }
    return button;
}

void IfcViewerWidget::setupViewController()
{
    setMouseTracking(false);
    m_mouseHandler = new osgGA::TrackballManipulator;
    m_mouseHandler->setAllowThrow(false);
    m_viewer->setCameraManipulator(m_mouseHandler);
}

void IfcViewerWidget::loadFile(const std::string& filePath)
{
    std::vector<osg::ref_ptr<osg::Geometry>> geometries;

    if(!m_parser.Parse(filePath, geometries)) {
        MessageLogger::log("Failed to prepare IFC geometry");
        return;
    }

    MessageLogger::log("\nIFC geometry prepared");

    osg::ref_ptr<osg::Geode> geode = new osg::Geode;
    for (auto geometry : geometries) {
        geode->addDrawable(geometry);
    }

    // Add the model to the root node
    root->addChild(geode);

    // set initial camera position
    orientScene(geode);
}

void IfcViewerWidget::orientScene(osg::ref_ptr<osg::Geode> geode)
{
    // Get the bounding-box
    osg::ComputeBoundsVisitor cbv;
    geode->accept(cbv);
    //root.get()->accept(cbv);
    const osg::BoundingBox& bb = cbv.getBoundingBox();

    // Center (average of opposite corners of bb)
    osg::Vec3d center = (bb.corner(0) + bb.corner(7)) * 0.5;

    // Radius (distance between the center and the corner)
    // assuming corner(7) has maximum x,y,z coords
    float radius = (bb.corner(7) - center).length();

    // Calc. eye position
    osg::Vec3d eye = center + osg::Vec3d(radius * 3, -radius * 3, 0.0);

    osg::Vec3d up = osg::Vec3d(0.0, 0.0, 1.0);

    // Set the camera's view matrix
    // In case a cameraManipulator is set, lookAt position setting does not work on the camera
    // as it is overridden.
    // So set it on the manipulator and reapply it to the viewer
    m_mouseHandler->setHomePosition(eye, center, up);
    m_viewer->setCameraManipulator(m_mouseHandler);

    auto log = [center, radius, eye]() {
        MessageLogger::log("center: " + std::to_string(center.length()));
        MessageLogger::log("radius: " + std::to_string(radius));
        MessageLogger::log("center: " + std::to_string(center.x()) + ", " + std::to_string(center.y()) + ", " + std::to_string(center.z()));
        MessageLogger::log("eye: " + std::to_string(eye.x()) + ", " + std::to_string(eye.y()) + ", " + std::to_string(eye.z()));
    };
    //log();
}