qgscomposerattributetablemodel.h

Go to the documentation of this file.

/***************************************************************************
                      qgscomposerattributetablemodel.h
                         --------------------
    begin                : April 2014
    copyright            : (C) 2014 by Nyall Dawson
    email                : nyall dot dawson at gmail dot 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 QGSCOMPOSERATTRIBUTETABLEMODEL_H
#define QGSCOMPOSERATTRIBUTETABLEMODEL_H

#include <QAbstractTableModel>
#include <QSortFilterProxyModel>

class QgsComposerAttributeTable;
class QgsComposerTableColumn;

//QgsComposerAttributeTableColumnModel

class CORE_EXPORT QgsComposerAttributeTableColumnModel: public QAbstractTableModel
{
    Q_OBJECT

  public:

    enum ShiftDirection
    {
      ShiftUp, 
      ShiftDown 
    };

    QgsComposerAttributeTableColumnModel( QgsComposerAttributeTable *composerTable, QObject *parent = 0 );
    virtual ~QgsComposerAttributeTableColumnModel();

    virtual int rowCount( const QModelIndex &parent = QModelIndex() ) const;
    int columnCount( const QModelIndex &parent = QModelIndex() ) const;
    virtual QVariant data( const QModelIndex &index, int role ) const;
    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const;
    virtual bool setData( const QModelIndex &index, const QVariant &value, int role = Qt::EditRole );
    Qt::ItemFlags flags( const QModelIndex &index ) const;
    bool removeRows( int row, int count, const QModelIndex &parent = QModelIndex() );
    bool insertRows( int row, int count, const QModelIndex &parent = QModelIndex() );
    QModelIndex index( int row, int column, const QModelIndex &parent ) const;
    QModelIndex parent( const QModelIndex &child ) const;

    bool moveRow( int row, ShiftDirection direction );

    void resetToLayer();

    QgsComposerTableColumn* columnFromIndex( const QModelIndex & index ) const;

    QModelIndex indexFromColumn( QgsComposerTableColumn *column );

    void setColumnAsSorted( QgsComposerTableColumn *column, Qt::SortOrder order );

    void setColumnAsUnsorted( QgsComposerTableColumn * column );

    bool moveColumnInSortRank( QgsComposerTableColumn * column, ShiftDirection direction );

  private:
    QgsComposerAttributeTable * mComposerTable;

};


//QgsComposerTableSortColumnsProxyModel

class CORE_EXPORT QgsComposerTableSortColumnsProxyModel: public QSortFilterProxyModel
{
    Q_OBJECT

  public:

    enum ColumnFilterType
    {
      ShowSortedColumns, 
      ShowUnsortedColumns
    };

    QgsComposerTableSortColumnsProxyModel( QgsComposerAttributeTable *composerTable, ColumnFilterType filterType, QObject *parent = 0 );

    virtual ~QgsComposerTableSortColumnsProxyModel();

    bool lessThan( const QModelIndex &left, const QModelIndex &right ) const;
    int columnCount( const QModelIndex &parent = QModelIndex() ) const;
    virtual QVariant data( const QModelIndex &index, int role ) const;
    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const;
    Qt::ItemFlags flags( const QModelIndex &index ) const;
    virtual bool setData( const QModelIndex &index, const QVariant &value, int role = Qt::EditRole );

    QgsComposerTableColumn* columnFromRow( int row );

    QgsComposerTableColumn* columnFromIndex( const QModelIndex & index ) const;


    QgsComposerTableColumn* columnFromSourceIndex( const QModelIndex& sourceIndex ) const;

    void resetFilter();

  protected:
    bool filterAcceptsRow( int source_row, const QModelIndex & source_parent ) const;

  private:
    QgsComposerAttributeTable * mComposerTable;
    ColumnFilterType mFilterType;

    QList<QgsComposerTableColumn*> columnsWithoutSortRank() const;

};
#endif // QGSCOMPOSERATTRIBUTETABLEMODEL_H