qgssnaptogridcanvasitem.cpp

Go to the documentation of this file.

/***************************************************************************
    qgssnaptogridcanvasitem.cpp
    ----------------------
    begin                : August 2018
    copyright            : (C) Matthias Kuhn
    email                : matthias@opengis.ch
 ***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgssnaptogridcanvasitem.h"
 
#include "qgsmapcanvas.h"
#include "qgspainting.h"
#include "qgsrendercontext.h"
 
#include "moc_qgssnaptogridcanvasitem.cpp"
 
QgsSnapToGridCanvasItem::QgsSnapToGridCanvasItem( QgsMapCanvas *mapCanvas )
  : QgsMapCanvasItem( mapCanvas )
{
  updateMapCanvasCrs();
  connect( mMapCanvas, &QgsMapCanvas::extentsChanged, this, &QgsSnapToGridCanvasItem::updateZoomFactor );
  connect( mMapCanvas, &QgsMapCanvas::destinationCrsChanged, this, &QgsSnapToGridCanvasItem::updateMapCanvasCrs );
}
 
void QgsSnapToGridCanvasItem::paint( QPainter *painter )
{
  if ( !mEnabled || !mAvailableByZoomFactor )
    return;
 
  const QgsScopedQPainterState painterState( painter );
  const QgsRectangle mapRect = mMapCanvas->extent();
 
  painter->setRenderHints( QPainter::Antialiasing );
  painter->setCompositionMode( QPainter::CompositionMode_Difference );
 
  const double scaleFactor = painter->fontMetrics().xHeight() * .2;
 
  mGridPen.setWidth( scaleFactor );
  mCurrentPointPen.setWidth( scaleFactor * 3 );
  const int gridMarkerLength = scaleFactor * 3;
 
  try
  {
    QgsCoordinateTransform extentTransform = mTransform;
    extentTransform.setBallparkTransformsAreAppropriate( true );
    const QgsRectangle layerExtent = extentTransform.transformBoundingBox( mapRect, Qgis::TransformDirection::Reverse );
    const QgsPointXY layerPt = mTransform.transform( mPoint, Qgis::TransformDirection::Reverse );
 
    const double gridXMin = std::ceil( layerExtent.xMinimum() / mPrecision ) * mPrecision;
    const double gridXMax = std::ceil( layerExtent.xMaximum() / mPrecision ) * mPrecision;
    const double gridYMin = std::ceil( layerExtent.yMinimum() / mPrecision ) * mPrecision;
    const double gridYMax = std::ceil( layerExtent.yMaximum() / mPrecision ) * mPrecision;
 
    for ( double x = gridXMin; x < gridXMax; x += mPrecision )
    {
      for ( double y = gridYMin; y < gridYMax; y += mPrecision )
      {
        const QgsPointXY pt = mTransform.transform( x, y );
        const QPointF canvasPt = toCanvasCoordinates( pt );
 
        if ( qgsDoubleNear( layerPt.x(), x, mPrecision / 2 ) && qgsDoubleNear( layerPt.y(), y, mPrecision / 2 ) )
        {
          painter->setPen( mCurrentPointPen );
        }
        else
        {
          painter->setPen( mGridPen );
        }
        painter->drawLine( canvasPt.x() - gridMarkerLength, canvasPt.y(), canvasPt.x() + gridMarkerLength, canvasPt.y() );
        painter->drawLine( canvasPt.x(), canvasPt.y() - gridMarkerLength, canvasPt.x(), canvasPt.y() + gridMarkerLength );
      }
    }
  }
  catch ( QgsCsException &e )
  {
    Q_UNUSED( e )
    mAvailableByZoomFactor = false;
  }
}
 
QgsPointXY QgsSnapToGridCanvasItem::point() const
{
  return mPoint;
}
 
void QgsSnapToGridCanvasItem::setPoint( const QgsPointXY &point )
{
  mPoint = point;
  update();
}
 
double QgsSnapToGridCanvasItem::precision() const
{
  return mPrecision;
}
 
void QgsSnapToGridCanvasItem::setPrecision( double precision )
{
  mPrecision = precision;
  updateZoomFactor();
}
 
QgsCoordinateReferenceSystem QgsSnapToGridCanvasItem::crs() const
{
  return mTransform.sourceCrs();
}
 
void QgsSnapToGridCanvasItem::setCrs( const QgsCoordinateReferenceSystem &crs )
{
  mTransform.setSourceCrs( crs );
  updateZoomFactor();
}
 
bool QgsSnapToGridCanvasItem::enabled() const
{
  return mEnabled;
}
 
void QgsSnapToGridCanvasItem::setEnabled( bool enabled )
{
  mEnabled = enabled;
  update();
}
 
void QgsSnapToGridCanvasItem::updateMapCanvasCrs()
{
  mTransform.setContext( mMapCanvas->mapSettings().transformContext() );
  mTransform.setDestinationCrs( mMapCanvas->mapSettings().destinationCrs() );
  update();
}
 
 
void QgsSnapToGridCanvasItem::updateZoomFactor()
{
  if ( !isVisible() )
    return;
 
  try
  {
    const int threshold = 5;
 
    const QgsRectangle extent = mMapCanvas->extent();
    if ( extent != rect() )
      setRect( extent );
 
    const QgsPointXY centerPoint = mMapCanvas->extent().center();
    const QPointF canvasCenter = toCanvasCoordinates( centerPoint );
 
    const QgsPointXY pt1 = mMapCanvas->mapSettings().mapToPixel().toMapCoordinates( static_cast<int>( canvasCenter.x() - threshold ), static_cast<int>( canvasCenter.y() - threshold ) );
    const QgsPointXY pt2 = mMapCanvas->mapSettings().mapToPixel().toMapCoordinates( static_cast<int>( canvasCenter.x() + threshold ), static_cast<int>( canvasCenter.y() + threshold ) );
 
    const QgsPointXY layerPt1 = mTransform.transform( pt1, Qgis::TransformDirection::Reverse );
    const QgsPointXY layerPt2 = mTransform.transform( pt2, Qgis::TransformDirection::Reverse );
 
    const double dist = layerPt1.distance( layerPt2 );
 
    if ( dist < mPrecision )
      mAvailableByZoomFactor = true;
    else
      mAvailableByZoomFactor = false;
  }
  catch ( QgsCsException & )
  {
    // transform errors?
    // you've probably got worse problems than the grid with your digitizing operations in the current projection.
    mAvailableByZoomFactor = false;
  }
}