src/core/qgsvectorlayerundocommand.h

Go to the documentation of this file.

/***************************************************************************
    qgsvectorlayerundocommand.h
    ---------------------
    begin                : June 2009
    copyright            : (C) 2009 by Martin Dobias
    email                : wonder.sk at gmail.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 QGSVECTORLAYERUNDOCOMMAND_H
#define QGSVECTORLAYERUNDOCOMMAND_H

#include <QUndoCommand>

#include <QVariant>
#include <QSet>
#include <QList>

#include "qgsfield.h"
#include "qgsfeature.h"

class QgsGeometry;
class QgsVectorLayer;


// TODO: copied from qgsvectorlayer.h
typedef QList<int> QgsAttributeList;
typedef QSet<int> QgsAttributeIds;


class QgsUndoCommand : public QUndoCommand
{
  public:

    class AttributeChangeEntry
    {
      public:
        bool isFirstChange;
        QVariant original;
        QVariant target;
    };

    typedef QMap<int, AttributeChangeEntry> AttributeChanges;

    class GeometryChangeEntry
    {
      public:
        GeometryChangeEntry();
        ~GeometryChangeEntry();

        void setOriginalGeometry( QgsGeometry& orig );
        void setTargetGeometry( QgsGeometry& target );

        QgsGeometry* original;
        QgsGeometry* target;
    };


    QgsUndoCommand( QgsVectorLayer* layer, QString text );

    void undo();

    void redo();

    void storeGeometryChange( QgsFeatureId featureId, QgsGeometry& original, QgsGeometry& target );

    void storeAttributeChange( QgsFeatureId featureId, int field, QVariant original, QVariant target, bool isFirstChange );

    void storeFeatureDelete( QgsFeatureId featureId );

    void storeFeatureAdd( QgsFeature& feature );

    void storeAttributeAdd( int index, const QgsField & value );

    void storeAttributeDelete( int index, const QgsField & orig );

  private:
    bool mFirstRun;

    QgsVectorLayer* mLayer;

    QMap<QgsFeatureId, GeometryChangeEntry> mGeometryChange;

    QMap<QgsFeatureId, AttributeChanges> mAttributeChange;

    QgsFeatureIds mDeletedFeatureIdChange;

    QgsFieldMap mAddedAttributes;

    QgsFieldMap mDeletedAttributes;

    QgsFeatureList mAddedFeatures;

    friend class QgsVectorLayer;
};

#endif