ctsystemview.h

#ifndef CTL_CTSYSTEMVIEW_H
#define CTL_CTSYSTEMVIEW_H

#include <QWidget>
#include <Qt3DWindow>

// forward delcarations
namespace CTL {
class AbstractDetector;
class AbstractGantry;
class AbstractSource;
class SimpleCTSystem;
template <typename T>
class VoxelVolume;
} // namespace CTL
class QGridLayout;
namespace Qt3DCore {
class QEntity;
} // namespace Qt3DCore
namespace Qt3DExtras {
class QOrbitCameraController;
} // namespace Qt3DExtras
namespace Qt3DRender {
class QCamera;
class QMaterial;
} // namespace Qt3DRender


namespace CTL {
namespace gui {

namespace details {
class CTL3DWindow;
}

class CTSystemView : public QWidget
{
    Q_OBJECT
public:
    explicit CTSystemView(QWidget* parent = nullptr, float visualScale = 50.0f);

    void setCTSystem(const SimpleCTSystem& system);
    void setData(const SimpleCTSystem& system);

    static void plot(const SimpleCTSystem& system, float visualScale = 50.0f);

    void addSystemVisualization(const SimpleCTSystem& system);
    void addVolume(const VoxelVolume<uchar>& volume);

public Q_SLOTS:
    void clearScene();
    void resetCamera();
    void resetView();

protected:
    QGridLayout* _mainLayout;

    details::CTL3DWindow* _view;
    Qt3DCore::QEntity* _rootEntity;

    void addBoxObject(const QVector3D& dimensions,
                      const QVector3D& translation,
                      const QQuaternion& rotation,
                      Qt3DRender::QMaterial* material = nullptr);
    void addBallObject(float radius,
                       const QVector3D& translation,
                       Qt3DRender::QMaterial* material = nullptr);
    void addDetectorComponent(AbstractGantry* gantry, AbstractDetector* detector);
    void addSourceComponent(AbstractGantry* gantry, AbstractSource* source);

private:
    Qt3DRender::QCamera* _camera;
    Qt3DExtras::QOrbitCameraController* _camController;
    Qt3DRender::QMaterial* _defaultMaterial;

    float _visualScale = 50.0f;

    void initializeView();
    void addAxis(Qt::Axis axis, float lineLength = 10.0f);
    void addCoordinateSystem();
};

namespace details {

class CTL3DWindow : public Qt3DExtras::Qt3DWindow
{
    Q_OBJECT
public:
    using Qt3DExtras::Qt3DWindow::Qt3DWindow;

protected:
    void keyPressEvent(QKeyEvent* e) override;

Q_SIGNALS:
    void saveRequest();
};

} // namespace details

// free function plot()
inline void plot(const SimpleCTSystem& system, float visualScale = 50.0f)
{ CTSystemView::plot(system, visualScale); }

} // namespace gui
} // namespace CTL

#endif // CTL_CTSYSTEMVIEW_H