qgsmaptoolcapture.h

Go to the documentation of this file.

/***************************************************************************
    qgsmaptoolcapture.h  -  map tool for capturing points, lines, polygons
    ---------------------
    begin                : January 2006
    copyright            : (C) 2006 by Martin Dobias
    email                : wonder.sk at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#ifndef QGSMAPTOOLCAPTURE_H
#define QGSMAPTOOLCAPTURE_H


#include "qgsmaptooladvanceddigitizing.h"
#include "qgspointlocator.h"
#include "qgscompoundcurve.h"
#include "qgsgeometry.h"

#include <QPoint>
#include <QList>
#include "qgis_gui.h"

class QgsRubberBand;
class QgsSnapIndicator;
class QgsVertexMarker;
class QgsMapLayer;
class QgsGeometryValidator;

class GUI_EXPORT QgsMapToolCapture : public QgsMapToolAdvancedDigitizing
{
    Q_OBJECT

  public:

    enum CaptureMode
    {
      CaptureNone,    
      CapturePoint,   
      CaptureLine,    
      CapturePolygon  
    };

    QgsMapToolCapture( QgsMapCanvas *canvas, QgsAdvancedDigitizingDockWidget *cadDockWidget, CaptureMode mode );

    ~QgsMapToolCapture() override;

    void activate() override;
    void deactivate() override;

    CaptureMode mode() const { return mCaptureMode; }

    int addCurve( QgsCurve *c );

    void clearCurve( );

    const QgsCompoundCurve *captureCurve() const { return &mCaptureCurve; }

    QList<QgsPointLocator::Match> snappingMatches() const;

    void cadCanvasMoveEvent( QgsMapMouseEvent *e ) override;

    void keyPressEvent( QKeyEvent *e ) override;

    void deleteTempRubberBand();

    void clean() override;

  private slots:
    void addError( const QgsGeometry::Error &error );
    void currentLayerChanged( QgsMapLayer *layer );

  protected:

    // TODO QGIS 3.0 returns an enum instead of a magic constant
    int nextPoint( const QgsPoint &mapPoint, QgsPoint &layerPoint );

    // TODO QGIS 3.0 returns an enum instead of a magic constant
    int nextPoint( QPoint p, QgsPoint &layerPoint, QgsPoint &mapPoint );

    // TODO QGIS 3.0 returns an enum instead of a magic constant
    int fetchLayerPoint( const QgsPointLocator::Match &match, QgsPoint &layerPoint );

    QgsPoint mapPoint( const QgsMapMouseEvent &e ) const;

    QgsPoint mapPoint( const QgsPointXY &point ) const;

    // TODO QGIS 3.0 returns an enum instead of a magic constant
    int addVertex( const QgsPointXY &point );

    int addVertex( const QgsPointXY &mapPoint, const QgsPointLocator::Match &match );

    void undo();

    void startCapturing();

    bool isCapturing() const;

    int size();

    QVector<QgsPointXY> points() const;

    void setPoints( const QVector<QgsPointXY> &pointList );

    void closePolygon();

  protected slots:

    void stopCapturing();

  private:
    bool tracingEnabled();
    QgsPointXY tracingStartPoint();
    bool tracingMouseMove( QgsMapMouseEvent *e );
    bool tracingAddVertex( const QgsPointXY &point );

  private:
    CaptureMode mCaptureMode;

    bool mCapturing = false;

    QgsRubberBand *mRubberBand = nullptr;

    QgsRubberBand *mTempRubberBand = nullptr;

    QgsCompoundCurve mCaptureCurve;

    QList<QgsPointLocator::Match> mSnappingMatches;

    void validateGeometry();
    QgsGeometryValidator *mValidator = nullptr;
    QList< QgsGeometry::Error > mGeomErrors;
    QList< QgsVertexMarker * > mGeomErrorMarkers;

    bool mCaptureModeFromLayer = false;

    std::unique_ptr<QgsSnapIndicator> mSnapIndicator;

    QgsPointXY mTracingStartPoint;

    friend class TestQgsMapToolReshape;

};

#endif