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 QgsComposerMouseHandles;
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 QgsFillSymbolV2;

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
    };

    enum ZValueDirection
    {
      ZValueBelow,
      ZValueAbove
    };

    enum AtlasMode
    {
      AtlasOff,     // Composition is not being controlled by an atlas
      PreviewAtlas, // An atlas composition is being previewed in the app
      ExportAtlas   // The composition is being exported as an atlas
    };

    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 setPageStyleSymbol( QgsFillSymbolV2* symbol );
    QgsFillSymbolV2* pageStyleSymbol() { return mPageStyleSymbol; }

    QPointF positionOnPage( const QPointF & position ) const;

    int pageNumberForPoint( const QPointF & position ) const;

    void setStatusMessage( const QString & message );

    void updateSettings();

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

    void setGridVisible( bool b );
    bool gridVisible() const {return mGridVisible;}

    void setSnapLinesVisible( bool visible );
    bool snapLinesVisible() const {return mGuidesVisible;}

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

    void setSmartGuidesEnabled( bool b ) { mSmartGuides = b; }
    bool smartGuidesEnabled() const {return mSmartGuides;}

    void clearSnapLines();

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

    void setSnapGridTolerance( double tolerance );
    double snapGridTolerance() const {return mSnapGridTolerance;}

    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 setAlignmentSnapTolerance( double t ) { mAlignmentSnapTolerance = t; }
    double alignmentSnapTolerance() const { return mAlignmentSnapTolerance; }

    QUndoStack* undoStack() { return &mUndoStack; }

    QgsComposerItem* composerItemAt( const QPointF & position );

    QgsComposerItem* composerItemAt( const QPointF & position, const QgsComposerItem* belowItem );

    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 generateWorldFile() const { return mGenerateWorldFile; }
    void setGenerateWorldFile( bool enabled ) { mGenerateWorldFile = enabled; }

    QgsComposerMap* worldFileMap() const { return mWorldFileMap; }
    void setWorldFileMap( QgsComposerMap* map ) { mWorldFileMap = map; }

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

    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 find items by their position in the z list
    void selectNextByZOrder( ZValueDirection direction );
    QgsComposerItem* getComposerItemBelow( QgsComposerItem* item );
    QgsComposerItem* getComposerItemAbove( QgsComposerItem* item );

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

    //functions to lock and unlock items
    void lockSelectedItems();
    void unlockAllItems();

    void sortZList();

    void refreshZList();

    QPointF snapPointToGrid( const QPointF& scenePoint ) const;

    QList< QGraphicsLineItem* >* snapLines() {return &mSnapLines;}

    QgsComposerMouseHandles* selectionHandles() {return mSelectionHandles;}

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

    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 );

    void setPreventCursorChange( bool preventChange ) { mPreventCursorChange = preventChange; }
    bool preventCursorChange() { return mPreventCursorChange; }

    //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 );

    void computeWorldFileParameters( double& a, double& b, double& c, double& d, double& e, double& f ) const;

    QgsAtlasComposition& atlasComposition() { return mAtlasComposition; }

    static void relativeResizeRect( QRectF& rectToResize, const QRectF& boundsBefore, const QRectF& boundsAfter );
    static double relativePosition( double position, double beforeMin, double beforeMax, double afterMin, double afterMax );

    QgsComposition::AtlasMode atlasMode() const { return mAtlasMode; }
    bool setAtlasMode( QgsComposition::AtlasMode mode );

  public slots:
    void sendItemAddedSignal( QgsComposerItem* item );

    void updateBounds();

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

    QgsFillSymbolV2* mPageStyleSymbol;
    void createDefaultPageStyleSymbol();

    QLinkedList<QgsComposerItem*> mItemZList;

    QSet<QgsComposerMultiFrame*> mMultiFrames;

    int mPrintResolution;

    bool mPrintAsRaster;

    bool mGenerateWorldFile;
    QgsComposerMap* mWorldFileMap;

    bool mUseAdvancedEffects;

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

    bool mAlignmentSnap;
    bool mGuidesVisible;
    bool mSmartGuides;
    double mAlignmentSnapTolerance;

    QList< QGraphicsLineItem* > mSnapLines;

    QgsComposerMouseHandles* mSelectionHandles;

    QUndoStack mUndoStack;

    QgsComposerItemCommand* mActiveItemCommand;
    QgsComposerMultiFrameCommand* mActiveMultiFrameCommand;

    QgsAtlasComposition mAtlasComposition;

    QgsComposition::AtlasMode mAtlasMode;

    QgsComposition(); //default constructor is forbidden

    QRectF compositionBounds() const;

    void updateZValues( bool addUndoCommands = true );

    int boundingRectOfSelectedItems( QRectF& bRect );

    void loadDefaults();

    void loadSettings();

    QPointF minPointFromXml( const QDomElement& elem ) const;

    void connectAddRemoveCommandSignals( QgsAddRemoveItemCommand* c );

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

    static QString encodeStringForXML( const QString& str );

    bool mPreventCursorChange;

  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* );

    void statusMsgChanged( QString message );
};

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