visualization.h

#pragma once
#include <QColor>
#include <QFont>
#include <QPointF>
#include <QRectF>
#include <tl/expected.hpp>
#include <variant>
 
#include "ierror.h"
#include "opencv2/core.hpp"
 
namespace VIS {
 
struct RectStyle {
    QColor strokeColor = Qt::green;  // Default color: Green
    int strokeWidth = 2;             // Default thickness in screen space pixels
    QColor fillColor = Qt::transparent;  // Default fill color: Transparent
};
 
struct RectOverlay {
    QRectF rectangle;  // Rectangle coordinates in [0,1] normalized space
    RectStyle style;   // Style for rendering the rectangle
};
 
struct TextStyle {
    QColor textColor = Qt::white;        // Default text color: White
    int fontSize = 12;                   // Default font size in points
    QFont font = QFont();                // Default font: System default
    qreal padding = 3.0;                 // Default padding in pixels
    QColor backgroundColor = Qt::black;  // Default background color: Black
};
 
enum class TextAnchor {
    TopLeft,
    TopCenter,
    TopRight,
    CenterLeft,
    Center,
    CenterRight,
    BottomLeft,
    BottomCenter,
    BottomRight
};
 
struct TextOverlay {
    QString text;      // Text content to display
    QPointF position;  // Position in [0,1] normalized space
    TextAnchor anchor = TextAnchor::BottomLeft;  // Default anchor position
    TextStyle style;                             // Style for rendering the text
};
 
struct ImageStyle {
    float opacity = 1.0f;  // Default opacity: fully opaque
    QRectF position = QRectF(0, 0, 1, 1);  // Default position: covers the entire image
};
 
struct ImageOverlay {
    cv::Mat image;     // Image to overlay
    ImageStyle style;  // Style for rendering the image
};
 
using OverlayItem = std::variant<RectOverlay, TextOverlay, ImageOverlay>;
 
enum class ViewportType {
    FullImage,        // show the entire image
    RegionOfInterest  // show only the region of interest
};
 
struct ViewStyle {
    QColor backgroundColor =
        Qt::transparent;  // transparent shows the original image
    ViewportType viewport =
        ViewportType::FullImage;  // whether to show only the region of interest
                                  // or the full image
    QPointF relativeSize = QPointF(
        1.0, 1.0);  // scale factor relative to viewport size (width, height)
    bool showTitle = true;        // if true, show the title of the view
};
 
struct View {
    QString title;                      // Title of the view
    ViewStyle style;                    // Style for the view
    std::vector<OverlayItem> overlays;  // List of overlay items to render
};
 
struct Visualization {
    std::vector<View> views;  // Views to visualize
};
 
using VisualizationResult = tl::expected<Visualization, PLUG::Error>;
 
} // namespace VIS