RealSpaceCanvas.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/View/Realspace/RealSpaceCanvas.cpp
//! @brief     Implements class RealSpaceScene
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************
 
#include "GUI/View/Realspace/RealSpaceCanvas.h"
#include "GUI/Application/ApplicationSettings.h"
#include "GUI/Model/Data/SessionData.h"
#include "GUI/Model/Project/ProjectDocument.h"
#include "GUI/Model/Sample/MaterialItem.h"
#include "GUI/Model/Sample/MultiLayerItem.h"
#include "GUI/View/Info/CautionSign.h"
#include "GUI/View/Realspace/RealSpaceBuilder.h"
#include "GUI/ba3d/model/model.h"
#include "GUI/ba3d/view/canvas.h"
#include <QApplication>
#include <QFileDialog>
#include <QMessageBox>
#include <QVBoxLayout>
#include <memory>
 
RealSpaceCanvas::RealSpaceCanvas(QWidget* parent)
    : QWidget(parent)
    , m_view(new GUI::RealSpace::Canvas)
    , m_cautionSign(new CautionSign(this))
    , m_containingMultiLayerItem(nullptr)
    , m_firstView(true)
{
    auto* layout = new QVBoxLayout;
    layout->setMargin(0);
    layout->setSpacing(0);
    layout->addWidget(m_view);
    setLayout(layout);
}
 
void RealSpaceCanvas::setCurrentItem(MultiLayerItem* containingMultiLayerItem,
                                     ItemForRealSpace item)
{
    bool isValid = false;
    std::visit([&](auto&& arg) { isValid = arg != nullptr; }, item);
 
    if (isValid) {
        m_containingMultiLayerItem = containingMultiLayerItem;
        m_currentItem = item;
        updateScene();
        defaultView(); // Enforces default view and also sets the zoomLevel to default i.e. 0
    } else
        resetScene();
}
 
void RealSpaceCanvas::changeLayerSize(double layerSizeChangeScale)
{
    // when no object is selected --> take no action
    if (!m_currentItem)
        return;
 
    m_sceneGeometry.layerSize = m_sceneGeometry.layerSize * layerSizeChangeScale;
    updateScene();
}
 
void RealSpaceCanvas::savePicture()
{
    QPixmap pixmap(this->size());
    render(&pixmap, QPoint(), childrenRegion());
    savePicture(pixmap);
}
 
void RealSpaceCanvas::showEvent(QShowEvent*)
{
    updateScene();
    if (m_firstView)
        m_view->defaultView();
    m_firstView = false;
}
 
void RealSpaceCanvas::savePicture(const QPixmap& pixmap)
{
    QString dirname = gSessionData->projectDocument.value()->userExportDir();
    QString defaultExtension = ".png";
    QString selectedFilter("*" + defaultExtension);
    QString defaultName = dirname + "/untitled";
    QString fileName = QFileDialog::getSaveFileName(
        nullptr, "Save 3D real space view", defaultName, selectedFilter, nullptr,
        appSettings->useNativeFileDialog() ? QFileDialog::Options()
                                           : QFileDialog::DontUseNativeDialog);
    QString nameToSave =
        fileName.endsWith(defaultExtension) ? fileName : fileName + defaultExtension;
 
    if (!nameToSave.isEmpty()) {
        try {
            pixmap.save(nameToSave);
        } catch (const std::exception& ex) {
            QString message = "Attempt to save file with the name '";
            message.append(nameToSave);
            message.append("' has failed with following error message\n\n");
            message.append(QString::fromStdString(ex.what()));
            QMessageBox::warning(nullptr, "Houston, we have a problem.", message);
        }
    }
}
 
void RealSpaceCanvas::updateScene()
{
    if (!isVisible()) // to improve performance
        return;
 
    if (!m_view->isValid()) // GL not initialized (widget not shown so far) -> no updating possible
        return;
 
    QApplication::setOverrideCursor(Qt::WaitCursor);
 
    m_realSpaceModel = std::make_unique<GUI::RealSpace::Model>();
 
    const auto colorForMaterialName = [&](const QString& materialName) {
        return m_containingMultiLayerItem->materialItems().materialFromName(materialName)->color();
    };
 
    RealSpaceBuilder builder3D(colorForMaterialName);
 
    try {
        m_cautionSign->clear();
        if (m_currentItem)
            builder3D.populate(m_realSpaceModel.get(), *m_currentItem, m_sceneGeometry,
                               m_view->cam()->getPos());
    } catch (const std::exception& ex) {
        m_cautionSign->setCautionMessage(ex.what());
    } catch (...) {
        // ignore other exceptions thrown
    }
    m_view->setModel(m_realSpaceModel.get());
 
    QApplication::restoreOverrideCursor();
}
 
void RealSpaceCanvas::resetScene()
{
    m_realSpaceModel.reset();
    m_view->setModel(nullptr);
    m_currentItem.reset();
    m_containingMultiLayerItem = nullptr;
}
 
void RealSpaceCanvas::defaultView()
{
    m_view->defaultView();
}
 
void RealSpaceCanvas::sideView()
{
    m_view->sideView();
}
 
void RealSpaceCanvas::topView()
{
    m_view->topView();
}