scene.cpp

//  gamma-viewer-3d - 3d visualization of sessions generated by gamma-analyzer
//  Copyright (C) 2017  Dag Robole
//
//  This program is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "scene.h"
#include <QVector3D>
#include <Qt3DRender/QCameraLens>
#include <Qt3DExtras/QForwardRenderer>

Scene::Scene(const QColor &clearColor, QString sessionFileName, QString doserateScriptFileName)
    :
      session(std::make_unique<Gamma::Session>(sessionFileName, doserateScriptFileName)),
      window(new Qt3DExtras::Qt3DWindow),
      root(new Qt3DCore::QEntity),
      camera(nullptr),
      cameraController(new Qt3DExtras::QOrbitCameraController(root)),
      selected(std::make_unique<SelectionEntity>(QVector3D(0.0, 0.0, 0.0), QColor(255, 0, 255), root)),
      marked(std::make_unique<SelectionEntity>(QVector3D(0.0, 0.0, 0.0), QColor(255, 255, 255), root))
{
    window->defaultFrameGraph()->setClearColor(clearColor);
    window->setIcon(QIcon(":/images/crash.ico"));
    window->setTitle(session->name());

    camera = window->camera();
    camera->lens()->setPerspectiveProjection(45.0f, 16.0f / 9.0f, 0.1f, 10000.0f);
    camera->setUpVector(QVector3D(0.0, 1.0, 0.0));
    camera->setPosition(QVector3D(0, 20, 100.0f));
    camera->setViewCenter(QVector3D(0, 0, 0));

    cameraController->setLinearSpeed(50.0f);
    cameraController->setLookSpeed(180.0f);
    cameraController->setCamera(camera);

    selected->setEnabled(false);
    marked->setEnabled(false);

    window->setRootEntity(root);
}

Scene::~Scene()
{
    for(auto *node : root->childNodes())
    {
        if(auto entity = qobject_cast<Qt3DCore::QEntity *>(node))
        {
            entity->components().clear();
            entity->deleteLater();
        }
    }

    root->components().clear();
    root->deleteLater();
    cameraController->setCamera(nullptr);
    cameraController->deleteLater();
    camera = nullptr;
    window->disconnect();
    window->destroy();
    window->deleteLater();
}

bool Scene::hasChildEntity(Qt3DCore::QEntity *entity) const
{
    QObject *e = entity;

    while(e)
    {
        if(e->parent() == root)
            return true;
        e = e->parent();
    }

    return false;
}