maptoolpick2dcells.h

Go to the documentation of this file.

 
#ifndef OPENSWMMVIS_MAP_TOOLS_MAPTOOLPICK2DCELLS_H
#define OPENSWMMVIS_MAP_TOOLS_MAPTOOLPICK2DCELLS_H
 
#include "map/tools/maptool.h"
#include "plot/plotattribute.h"
 
#include <QPoint>
#include <QPolygon>
#include <QPolygonF>
#include <QPointer>
#include <QRectF>
#include <QVector>
 
class SWMM2DResultsLayer;
class SWMM2DMeshLayer;
class SelectionManager;
 
class MapToolPick2DCells : public OpenSWMMVisMapTool
{
    Q_OBJECT
public:
    enum class Mode { Box, Lasso };
 
    explicit MapToolPick2DCells(MapCanvas *canvas,
                                SelectionManager *selection,
                                QObject *parent = nullptr);
    ~MapToolPick2DCells() override = default;
 
    void setMode(Mode m);
    [[nodiscard]] Mode mode() const noexcept { return m_mode; }
 
    void setTargetLayer(SWMM2DResultsLayer *layer);
 
    void activate() override;
    void deactivate() override;
 
    void mousePressEvent(QMouseEvent *event) override;
    void mouseMoveEvent(QMouseEvent *event) override;
    void mouseReleaseEvent(QMouseEvent *event) override;
    void mouseDoubleClickEvent(QMouseEvent *event) override;
    void keyPressEvent(QKeyEvent *event) override;
 
    void paint(QPainter *painter,
               const MapExtent &canvasExtent,
               const SpatialReferenceSystem *canvasSRS) override;
 
signals:
    void cellsPicked(SWMM2DResultsLayer *layer, const QVector<int> &triIdxList,
                     const QVector<openswmmvis::plot::PlotAttribute> &attrs);
 
private:
    SWMM2DResultsLayer *findResultsLayer_() const;
 
    SWMM2DMeshLayer *findMeshLayer_() const;
 
    void applySelection_(const QVector<int> &hits, Qt::KeyboardModifiers mods);
 
    [[nodiscard]] QVector<int> selectedCells_() const;
 
    void requestPlotAt_(const QPoint &pixel, const QPoint &globalPos);
 
    QPointF pixelToScene_(int px, int py) const;
 
    int pickCellAtScene_(const QPointF &scenePt);
 
    QVector<int> pickCellsInRectScene_(const QRectF &sceneRect);
 
    QVector<int> pickCellsInPolygonScene_(const QPolygonF &scenePoly);
 
    Mode                            m_mode = Mode::Box;
 
    // Box state.
    bool                            m_dragging   = false;
    QPoint                          m_startPixel;
    QPoint                          m_currentPixel;
    static constexpr int            kDragThreshPx = 8;
 
    // Lasso state.
    bool                            m_drawing    = false;
    QVector<QPointF>                m_lassoMapPts;     // map coords
    QPoint                          m_cursorPixel;
 
    // Right-click plot is armed on press but fired on RELEASE: the handler
    // exec()s the context menu, and starting a modal session while the
    // physical button is still down makes AppKit discard the pending
    // mouse-up for the (now modally blocked) canvas window — Qt's per-view
    // button state then latches RightButton forever and its implicit grab
    // redirects every later click app-wide (whole-UI input freeze).
    bool                            m_pendingRightPlot = false;
 
    QPointer<SWMM2DResultsLayer>    m_targetLayer;
    QPointer<SWMM2DMeshLayer>       m_meshLayer;
    QPointer<SelectionManager>      m_selection;
};
 
#endif // OPENSWMMVIS_MAP_TOOLS_MAPTOOLPICK2DCELLS_H