qgsgeometryrubberband.cpp

Go to the documentation of this file.

/***************************************************************************
                         qgsgeometryrubberband.cpp
                         -------------------------
    begin                : December 2014
    copyright            : (C) 2014 by Marco Hugentobler
    email                : marco at sourcepole dot 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 "qgsgeometryrubberband.h"
#include "qgsabstractgeometry.h"
#include "qgsmapcanvas.h"
#include "qgsrendercontext.h"
#include "qgspoint.h"
#include <QPainter>
 
QgsGeometryRubberBand::QgsGeometryRubberBand( QgsMapCanvas *mapCanvas, Qgis::GeometryType geomType )
  : QgsMapCanvasItem( mapCanvas ), mIconSize( 5 ), mIconType( ICON_BOX ), mGeometryType( geomType )
{
  mPen = QPen( QColor( 255, 0, 0 ) );
  mBrush = QBrush( QColor( 255, 0, 0 ) );
}
 
QgsGeometryRubberBand::~QgsGeometryRubberBand()
{
}
 
void QgsGeometryRubberBand::paint( QPainter *painter )
{
  if ( !mGeometry || !painter )
  {
    return;
  }
 
  const QgsScopedQPainterState painterState( painter );
  painter->translate( -pos() );
 
  if ( mGeometryType == Qgis::GeometryType::Polygon )
  {
    painter->setBrush( mBrush );
  }
  else
  {
    painter->setBrush( Qt::NoBrush );
  }
  painter->setPen( mPen );
 
 
  std::unique_ptr<QgsAbstractGeometry> paintGeom( mGeometry->clone() );
 
  paintGeom->transform( mMapCanvas->getCoordinateTransform()->transform() );
  paintGeom->draw( *painter );
 
  if ( !mDrawVertices )
    return;
 
  //draw vertices
  QgsVertexId vertexId;
  QgsPoint vertex;
  while ( paintGeom->nextVertex( vertexId, vertex ) )
  {
    drawVertex( painter, vertex.x(), vertex.y() );
  }
}
 
Qgis::GeometryType QgsGeometryRubberBand::geometryType() const
{
  return mGeometryType;
}
 
void QgsGeometryRubberBand::setGeometryType( Qgis::GeometryType geometryType )
{
  mGeometryType = geometryType;
}
 
void QgsGeometryRubberBand::drawVertex( QPainter *p, double x, double y )
{
  const qreal s = ( mIconSize - 1 ) / 2.0;
 
  switch ( mIconType )
  {
    case ICON_NONE:
      break;
 
    case ICON_CROSS:
      p->drawLine( QLineF( x - s, y, x + s, y ) );
      p->drawLine( QLineF( x, y - s, x, y + s ) );
      break;
 
    case ICON_X:
      p->drawLine( QLineF( x - s, y - s, x + s, y + s ) );
      p->drawLine( QLineF( x - s, y + s, x + s, y - s ) );
      break;
 
    case ICON_BOX:
      p->drawLine( QLineF( x - s, y - s, x + s, y - s ) );
      p->drawLine( QLineF( x + s, y - s, x + s, y + s ) );
      p->drawLine( QLineF( x + s, y + s, x - s, y + s ) );
      p->drawLine( QLineF( x - s, y + s, x - s, y - s ) );
      break;
 
    case ICON_FULL_BOX:
      p->drawRect( x - s, y - s, mIconSize, mIconSize );
      break;
 
    case ICON_CIRCLE:
      p->drawEllipse( x - s, y - s, mIconSize, mIconSize );
      break;
  }
}
 
void QgsGeometryRubberBand::setGeometry( QgsAbstractGeometry *geom )
{
  mGeometry.reset( geom );
 
  if ( mGeometry )
  {
    setRect( rubberBandRectangle() );
  }
}
 
void QgsGeometryRubberBand::moveVertex( QgsVertexId id, const QgsPoint &newPos )
{
  if ( mGeometry )
  {
    mGeometry->moveVertex( id, newPos );
    setRect( rubberBandRectangle() );
  }
}
 
void QgsGeometryRubberBand::setFillColor( const QColor &c )
{
  mBrush.setColor( c );
}
 
void QgsGeometryRubberBand::setStrokeColor( const QColor &c )
{
  mPen.setColor( c );
}
 
void QgsGeometryRubberBand::setStrokeWidth( int width )
{
  mPen.setWidth( width );
}
 
void QgsGeometryRubberBand::setLineStyle( Qt::PenStyle penStyle )
{
  mPen.setStyle( penStyle );
}
 
void QgsGeometryRubberBand::setBrushStyle( Qt::BrushStyle brushStyle )
{
  mBrush.setStyle( brushStyle );
}
 
void QgsGeometryRubberBand::setVertexDrawingEnabled( bool isVerticesDrawn )
{
  mDrawVertices = isVerticesDrawn;
}
 
QgsRectangle QgsGeometryRubberBand::rubberBandRectangle() const
{
  if ( !mGeometry || mGeometry->isEmpty() )
  {
    return QgsRectangle();
  }
  const QgsMapToPixel &m2p = *( mMapCanvas->getCoordinateTransform() );
 
  qreal w = ( ( mIconSize - 1 ) / 2 + mPen.width() ); // in canvas units
 
  QgsRectangle r;                                  // in canvas units
  QgsRectangle rectMap = mGeometry->boundingBox(); // in map units
  QList<QgsPointXY> pl;
  pl << QgsPointXY( rectMap.xMinimum(), rectMap.yMinimum() )
     << QgsPointXY( rectMap.xMinimum(), rectMap.yMaximum() )
     << QgsPointXY( rectMap.xMaximum(), rectMap.yMaximum() )
     << QgsPointXY( rectMap.xMaximum(), rectMap.yMinimum() );
 
  for ( QgsPointXY &p : pl )
  {
    p = toCanvasCoordinates( p );
    // no need to normalize the rectangle -- we know it is already normal
    QgsRectangle rect( p.x() - w, p.y() - w, p.x() + w, p.y() + w, false );
    r.combineExtentWith( rect );
  }
 
  // This is an hack to pass QgsMapCanvasItem::setRect what it
  // expects (encoding of position and size of the item)
  qreal res = m2p.mapUnitsPerPixel();
  QgsPointXY topLeft = m2p.toMapCoordinates( r.xMinimum(), r.yMinimum() );
  QgsRectangle rect( topLeft.x(), topLeft.y(), topLeft.x() + r.width() * res, topLeft.y() - r.height() * res );
 
  return rect;
}
 
void QgsGeometryRubberBand::updatePosition()
{
  // re-compute rectangle
  // See https://github.com/qgis/QGIS/issues/20566
  // NOTE: could be optimized by saving map-extent
  //       of rubberband and simply re-projecting
  //       that to device-rectangle on "updatePosition"
  setRect( rubberBandRectangle() );
}