qgscomposition.h

Go to the documentation of this file.

/***************************************************************************
                              qgscomposition.h
                             -------------------
    begin                : January 2005
    copyright            : (C) 2005 by Radim Blazek
    email                : [email protected]
 ***************************************************************************/
/***************************************************************************
 *                                                                         *
 *   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 QGSCOMPOSITION_H
#define QGSCOMPOSITION_H

#include "qgscomposeritem.h"
#include <memory>

#include <QDomDocument>
#include <QGraphicsScene>
#include <QLinkedList>
#include <QList>
#include <QPair>
#include <QSet>
#include <QUndoStack>
#include <QPrinter>
#include <QPainter>

#include "qgsaddremoveitemcommand.h"
#include "qgscomposeritemcommand.h"
#include "qgsatlascomposition.h"

class QgisApp;
class QgsComposerFrame;
class QgsComposerMap;
class QgsPaperItem;
class QGraphicsRectItem;
class QgsMapRenderer;
class QDomElement;
class QgsComposerArrow;
class QgsComposerHtml;
class QgsComposerItem;
class QgsComposerLabel;
class QgsComposerLegend;
class QgsComposerMap;
class QgsComposerPicture;
class QgsComposerScaleBar;
class QgsComposerShape;
class QgsComposerAttributeTable;
class QgsComposerMultiFrame;
class QgsComposerMultiFrameCommand;
class QgsVectorLayer;
class QgsComposer;

class CORE_EXPORT QgsComposition : public QGraphicsScene
{
    Q_OBJECT
  public:

    enum PlotStyle
    {
      Preview = 0, // Use cache etc
      Print,       // Render well
      Postscript   // Fonts need different scaling!
    };

    enum GridStyle
    {
      Solid,
      Dots,
      Crosses
    };

    QgsComposition( QgsMapRenderer* mapRenderer );
    ~QgsComposition();

    void setPaperSize( double width, double height );

    double paperHeight() const;

    double paperWidth() const;

    double spaceBetweenPages() const { return mSpaceBetweenPages; }

    void setNumPages( int pages );
    int numPages() const;

    void setSnapToGridEnabled( bool b );
    bool snapToGridEnabled() const {return mSnapToGrid;}

    void setSnapGridResolution( double r );
    double snapGridResolution() const {return mSnapGridResolution;}

    void setSnapGridOffsetX( double offset );
    double snapGridOffsetX() const {return mSnapGridOffsetX;}

    void setSnapGridOffsetY( double offset );
    double snapGridOffsetY() const {return mSnapGridOffsetY;}

    void setGridPen( const QPen& p );
    const QPen& gridPen() const {return mGridPen;}

    void setGridStyle( GridStyle s );
    GridStyle gridStyle() const {return mGridStyle;}

    void setAlignmentSnap( bool s ) { mAlignmentSnap = s; }
    bool alignmentSnap() const { return mAlignmentSnap; }

    void setAlignmentSnapTolerance( double t ) { mAlignmentSnapTolerance = t; }
    double alignmentSnapTolerance() const { return mAlignmentSnapTolerance; }

    QUndoStack* undoStack() { return &mUndoStack; }

    QgsComposerItem* composerItemAt( const QPointF & position );

    int pageNumberAt( const QPointF& position ) const;

    int itemPageNumber( const QgsComposerItem* ) const;

    QList<QgsComposerItem*> selectedComposerItems();

    QList<const QgsComposerMap*> composerMapItems() const;

    template<class T> void composerItems( QList<T*>& itemList );

    const QgsComposerMap* getComposerMapById( int id ) const;

    /*Returns the composer html with specified id (a string as named in the
      composer user interface item properties).
      @note Added in QGIS 2.0
      @param id - A QString representing the id of the item.
      @return QgsComposerHtml pointer or 0 pointer if no such item exists.
    */
    const QgsComposerHtml* getComposerHtmlByItem( QgsComposerItem *item ) const;

    const QgsComposerItem* getComposerItemById( QString theId ) const;

    const QgsComposerItem* getComposerItemByUuid( QString theUuid ) const;

    int printResolution() const {return mPrintResolution;}
    void setPrintResolution( int dpi ) {mPrintResolution = dpi;}

    bool printAsRaster() const {return mPrintAsRaster;}
    void setPrintAsRaster( bool enabled ) { mPrintAsRaster = enabled; }

    bool useAdvancedEffects() const {return mUseAdvancedEffects;}
    void setUseAdvancedEffects( bool effectsEnabled );

    double selectionTolerance() const { return mSelectionTolerance; }
    void setSelectionTolerance( double tol );

    QgsMapRenderer* mapRenderer() {return mMapRenderer;}

    QgsComposition::PlotStyle plotStyle() const {return mPlotStyle;}
    void setPlotStyle( QgsComposition::PlotStyle style ) {mPlotStyle = style;}

    int pixelFontSize( double pointSize ) const;

    double pointFontSize( int pixelSize ) const;

    bool writeXML( QDomElement& composerElem, QDomDocument& doc );

    bool readXML( const QDomElement& compositionElem, const QDomDocument& doc );

    bool loadFromTemplate( const QDomDocument& doc, QMap<QString, QString>* substitutionMap = 0, bool addUndoCommands = false );

    void addItemsFromXML( const QDomElement& elem, const QDomDocument& doc, QMap< QgsComposerMap*, int >* mapsToRestore = 0,
                          bool addUndoCommands = false, QPointF* pos = 0, bool pasteInPlace = false );

    void addItemToZList( QgsComposerItem* item );
    void removeItemFromZList( QgsComposerItem* item );

    //functions to move selected items in hierarchy
    void raiseSelectedItems();
    void raiseItem( QgsComposerItem* item );
    void lowerSelectedItems();
    void lowerItem( QgsComposerItem* item );
    void moveSelectedItemsToTop();
    void moveItemToTop( QgsComposerItem* item );
    void moveSelectedItemsToBottom();
    void moveItemToBottom( QgsComposerItem* item );

    //functions to align selected items
    void alignSelectedItemsLeft();
    void alignSelectedItemsHCenter();
    void alignSelectedItemsRight();
    void alignSelectedItemsTop();
    void alignSelectedItemsVCenter();
    void alignSelectedItemsBottom();

    void sortZList();

    QPointF snapPointToGrid( const QPointF& scenePoint ) const;

    QPointF alignItem( const QgsComposerItem* item, double& alignX, double& alignY, double dx = 0, double dy = 0 );

    QPointF alignPos( const QPointF& pos, const QgsComposerItem* excludeItem, double& alignX, double& alignY );

    QGraphicsLineItem* addSnapLine();
    void removeSnapLine( QGraphicsLineItem* line );
    QGraphicsLineItem* nearestSnapLine( bool horizontal, double x, double y, double tolerance, QList< QPair< QgsComposerItem*, QgsComposerItem::ItemPositionMode > >& snappedItems );
    void setSnapLinesVisible( bool visible );

    void beginCommand( QgsComposerItem* item, const QString& commandText, QgsComposerMergeCommand::Context c = QgsComposerMergeCommand::Unknown );

    void endCommand();
    void cancelCommand();

    void beginMultiFrameCommand( QgsComposerMultiFrame* multiFrame, const QString& text );
    void endMultiFrameCommand();

    void addMultiFrame( QgsComposerMultiFrame* multiFrame );
    void removeMultiFrame( QgsComposerMultiFrame* multiFrame );
    void addComposerArrow( QgsComposerArrow* arrow );
    void addComposerLabel( QgsComposerLabel* label );
    void addComposerMap( QgsComposerMap* map, bool setDefaultPreviewStyle = true );
    void addComposerScaleBar( QgsComposerScaleBar* scaleBar );
    void addComposerLegend( QgsComposerLegend* legend );
    void addComposerPicture( QgsComposerPicture* picture );
    void addComposerShape( QgsComposerShape* shape );
    void addComposerTable( QgsComposerAttributeTable* table );
    void addComposerHtmlFrame( QgsComposerHtml* html, QgsComposerFrame* frame );

    void removeComposerItem( QgsComposerItem* item, bool createCommand = true );

    void pushAddRemoveCommand( QgsComposerItem* item, const QString& text, QgsAddRemoveItemCommand::State state = QgsAddRemoveItemCommand::Added );


    //printing

    void beginPrint( QPrinter& printer );
    void beginPrintAsPDF( QPrinter& printer, const QString& file );
    void doPrint( QPrinter& printer, QPainter& painter );

    void print( QPrinter &printer );

    void exportAsPDF( const QString& file );

    QImage printPageAsRaster( int page );

    void renderPage( QPainter* p, int page );

    QgsAtlasComposition& atlasComposition() { return mAtlasComposition; }

  public slots:
    void sendItemAddedSignal( QgsComposerItem* item );

  private:
    QgsMapRenderer* mMapRenderer;
    QgsComposition::PlotStyle mPlotStyle;
    double mPageWidth;
    double mPageHeight;
    QList< QgsPaperItem* > mPages;
    double mSpaceBetweenPages; //space in preview between pages

    QLinkedList<QgsComposerItem*> mItemZList;

    QSet<QgsComposerMultiFrame*> mMultiFrames;

    int mPrintResolution;

    bool mPrintAsRaster;

    bool mUseAdvancedEffects;

    double mSelectionTolerance;

    bool mSnapToGrid;
    double mSnapGridResolution;
    double mSnapGridOffsetX;
    double mSnapGridOffsetY;
    QPen mGridPen;
    GridStyle mGridStyle;

    bool mAlignmentSnap;
    double mAlignmentSnapTolerance;

    QList< QGraphicsLineItem* > mSnapLines;

    QUndoStack mUndoStack;

    QgsComposerItemCommand* mActiveItemCommand;
    QgsComposerMultiFrameCommand* mActiveMultiFrameCommand;

    QgsAtlasComposition mAtlasComposition;

    QgsComposition(); //default constructor is forbidden

    void updateZValues();

    int boundingRectOfSelectedItems( QRectF& bRect );

    void loadSettings();
    void saveSettings();

    void connectAddRemoveCommandSignals( QgsAddRemoveItemCommand* c );

    void updatePaperItems();
    void addPaperItem();
    void removePaperItems();
    void deleteAndRemoveMultiFrames();

    static QString encodeStringForXML( const QString& str );

    //helper functions for item align
    void collectAlignCoordinates( QMap< double, const QgsComposerItem* >& alignCoordsX,
                                  QMap< double, const QgsComposerItem* >& alignCoordsY, const QgsComposerItem* excludeItem );

    void checkNearestItem( double checkCoord, const QMap< double, const QgsComposerItem* >& alignCoords, double& smallestDiff,
                           double itemCoordOffset, double& itemCoord, double& alignCoord ) const;

    static bool nearestItem( const QMap< double, const QgsComposerItem* >& coords, double value, double& nearestValue );

  signals:
    void paperSizeChanged();
    void nPagesChanged();

    void selectedItemChanged( QgsComposerItem* selected );
    void composerArrowAdded( QgsComposerArrow* arrow );
    void composerHtmlFrameAdded( QgsComposerHtml* html, QgsComposerFrame* frame );
    void composerLabelAdded( QgsComposerLabel* label );
    void composerMapAdded( QgsComposerMap* map );
    void composerScaleBarAdded( QgsComposerScaleBar* scalebar );
    void composerLegendAdded( QgsComposerLegend* legend );
    void composerPictureAdded( QgsComposerPicture* picture );
    void composerShapeAdded( QgsComposerShape* shape );
    void composerTableAdded( QgsComposerAttributeTable* table );
    void itemRemoved( QgsComposerItem* );
};

template<class T> void QgsComposition::composerItems( QList<T*>& itemList )
{
  itemList.clear();
  QList<QGraphicsItem *> graphicsItemList = items();
  QList<QGraphicsItem *>::iterator itemIt = graphicsItemList.begin();
  for ( ; itemIt != graphicsItemList.end(); ++itemIt )
  {
    T* item = dynamic_cast<T*>( *itemIt );
    if ( item )
    {
      itemList.push_back( item );
    }
  }
}

#endif