QGIS API Documentation 3.37.0-Master (fdefdf9c27f)
qgsgeopdflayertreemodel.cpp
Go to the documentation of this file.
1/***************************************************************************
2 qgsgeopdflayertreemodel.cpp
3 ---------------------
4 begin : August 2019
5 copyright : (C) 2019 by Nyall Dawson
6 email : nyall dot dawson at gmail dot com
7 ***************************************************************************
8 * *
9 * This program is free software; you can redistribute it and/or modify *
10 * it under the terms of the GNU General Public License as published by *
11 * the Free Software Foundation; either version 2 of the License, or *
12 * (at your option) any later version. *
13 * *
14 ***************************************************************************/
15
16#include <QComboBox>
17#include <QDoubleSpinBox>
18
20#include "qgslayertree.h"
21#include "qgsproject.h"
22#include "qgsvectorlayer.h"
23#include "qgsapplication.h"
24
25QgsGeoPdfLayerTreeModel::QgsGeoPdfLayerTreeModel( const QList<QgsMapLayer *> &layers, QObject *parent )
26 : QgsMapLayerModel( layers, parent )
27{
29}
30
31int QgsGeoPdfLayerTreeModel::columnCount( const QModelIndex &parent ) const
32{
33 Q_UNUSED( parent )
34 return 4;
35}
36
37Qt::ItemFlags QgsGeoPdfLayerTreeModel::flags( const QModelIndex &idx ) const
38{
39 if ( !idx.isValid() )
40 return Qt::ItemIsDropEnabled;
41
42 if ( idx.column() == IncludeVectorAttributes )
43 {
44 if ( vectorLayer( idx ) )
45 return QgsMapLayerModel::flags( idx ) | Qt::ItemIsUserCheckable;
46 else
47 return QgsMapLayerModel::flags( idx );
48 }
49
50 if ( idx.column() == InitiallyVisible )
51 {
52 return QgsMapLayerModel::flags( idx ) | Qt::ItemIsUserCheckable;
53 }
54
55 if ( !mapLayer( idx ) )
56 return Qt::ItemFlags();
57
58 return Qt::ItemIsEnabled | Qt::ItemIsEditable | Qt::ItemIsDragEnabled;
59}
60
61QgsMapLayer *QgsGeoPdfLayerTreeModel::mapLayer( const QModelIndex &idx ) const
62{
63 return layerFromIndex( index( idx.row(), LayerColumn, idx.parent() ) );
64}
65
66QgsVectorLayer *QgsGeoPdfLayerTreeModel::vectorLayer( const QModelIndex &idx ) const
67{
68 return qobject_cast<QgsVectorLayer *>( mapLayer( idx ) );
69}
70
71QVariant QgsGeoPdfLayerTreeModel::headerData( int section, Qt::Orientation orientation, int role ) const
72{
73 if ( orientation == Qt::Horizontal )
74 {
75 if ( role == Qt::DisplayRole )
76 {
77 switch ( section )
78 {
79 case LayerColumn:
80 return tr( "Layer" );
81 case GroupColumn:
82 return tr( "PDF Group" );
84 return tr( "Initially Visible" );
86 return tr( "Include Attributes" );
87 default:
88 return QVariant();
89 }
90 }
91 }
92 return QgsMapLayerModel::headerData( section, orientation, role );
93}
94
95QVariant QgsGeoPdfLayerTreeModel::data( const QModelIndex &idx, int role ) const
96{
97 switch ( idx.column() )
98 {
99 case LayerColumn:
100 if ( role == Qt::CheckStateRole )
101 return QVariant();
102
103 return QgsMapLayerModel::data( idx, role );
104
105 case GroupColumn:
106 {
107 switch ( role )
108 {
109 case Qt::DisplayRole:
110 case Qt::EditRole:
111 {
112 if ( QgsMapLayer *ml = mapLayer( idx ) )
113 {
114 return ml->customProperty( QStringLiteral( "geopdf/groupName" ) ).toString();
115 }
116 break;
117 }
118 }
119
120 return QVariant();
121 }
122
123 case InitiallyVisible:
124 {
125 if ( role == Qt::CheckStateRole )
126 {
127 if ( QgsMapLayer *ml = mapLayer( idx ) )
128 {
129 const QVariant v = ml->customProperty( QStringLiteral( "geopdf/initiallyVisible" ) );
130 if ( v.isValid() )
131 {
132 return v.toBool() ? Qt::Checked : Qt::Unchecked;
133 }
134 else
135 {
136 // otherwise, we default to showing by default
137 return Qt::Checked;
138 }
139 }
140 return QVariant();
141 }
142 return QVariant();
143 }
144
146 {
147 if ( role == Qt::CheckStateRole )
148 {
149 if ( QgsVectorLayer *vl = vectorLayer( idx ) )
150 {
151 const QVariant v = vl->customProperty( QStringLiteral( "geopdf/includeFeatures" ) );
152 if ( v.isValid() )
153 {
154 return v.toBool() ? Qt::Checked : Qt::Unchecked;
155 }
156 else
157 {
158 // otherwise, we default to true
159 return Qt::Checked;
160 }
161 }
162 return QVariant();
163 }
164 }
165 }
166
167 return QVariant();
168}
169
170bool QgsGeoPdfLayerTreeModel::setData( const QModelIndex &index, const QVariant &value, int role )
171{
172 switch ( index.column() )
173 {
175 {
176 if ( role == Qt::CheckStateRole )
177 {
178 if ( QgsVectorLayer *vl = vectorLayer( index ) )
179 {
180 vl->setCustomProperty( QStringLiteral( "geopdf/includeFeatures" ), value.toInt() == Qt::Checked );
181 emit dataChanged( index, index );
182 return true;
183 }
184 }
185 break;
186 }
187
188 case GroupColumn:
189 {
190 if ( role == Qt::EditRole )
191 {
192 if ( QgsMapLayer *ml = mapLayer( index ) )
193 {
194 ml->setCustomProperty( QStringLiteral( "geopdf/groupName" ), value.toString() );
195 emit dataChanged( index, index );
196 return true;
197 }
198 }
199 break;
200 }
201
202 case InitiallyVisible:
203 {
204 if ( role == Qt::CheckStateRole )
205 {
206 if ( QgsMapLayer *ml = mapLayer( index ) )
207 {
208 ml->setCustomProperty( QStringLiteral( "geopdf/initiallyVisible" ), value.toInt() == Qt::Checked );
209 emit dataChanged( index, index );
210 return true;
211 }
212 }
213 break;
214 }
215
216 case LayerColumn:
217 return QgsMapLayerModel::setData( index, value, role );
218 }
219 return false;
220}
221
222void QgsGeoPdfLayerTreeModel::checkAll( bool checked, const QModelIndex &parent, int column )
223{
224 for ( int row = 0; row < rowCount( parent ); ++row )
225 {
226 const QModelIndex childIndex = index( row, column, parent );
227 setData( childIndex, checked ? Qt::Checked : Qt::Unchecked, Qt::CheckStateRole );
228 checkAll( checked, childIndex );
229 }
230}
231
232
234QgsGeoPdfLayerFilteredTreeModel::QgsGeoPdfLayerFilteredTreeModel( QgsGeoPdfLayerTreeModel *sourceModel, QObject *parent )
235 : QSortFilterProxyModel( parent )
236 , mLayerTreeModel( sourceModel )
237{
238 setSourceModel( sourceModel );
239}
240
241bool QgsGeoPdfLayerFilteredTreeModel::filterAcceptsRow( int source_row, const QModelIndex &source_parent ) const
242{
243 if ( QgsMapLayer *layer = mLayerTreeModel->layerFromIndex( sourceModel()->index( source_row, 0, source_parent ) ) )
244 {
245 // filter out non-spatial layers
246 if ( !layer->isSpatial() )
247 return false;
248
249 }
250 return true;
251}
252
Layer tree model for Geo-PDF layers.
@ InitiallyVisible
Initial visibility state.
@ IncludeVectorAttributes
Vector attribute.
bool setData(const QModelIndex &index, const QVariant &value, int role) override
QVariant headerData(int section, Qt::Orientation orientation, int role) const override
QVariant data(const QModelIndex &index, int role) const override
void checkAll(bool checked, const QModelIndex &parent=QModelIndex(), int column=IncludeVectorAttributes)
Checks (or unchecks) all rows and children from the specified parent index.
Qt::ItemFlags flags(const QModelIndex &idx) const override
QgsGeoPdfLayerTreeModel(const QList< QgsMapLayer * > &layers, QObject *parent=nullptr)
constructor
int columnCount(const QModelIndex &parent) const override
The QgsMapLayerModel class is a model to display layers in widgets.
void setItemsCanBeReordered(bool allow)
Sets whether items in the model can be reordered via drag and drop.
Qt::ItemFlags flags(const QModelIndex &index) const override
int rowCount(const QModelIndex &parent=QModelIndex()) const override
QVariant data(const QModelIndex &index, int role=Qt::DisplayRole) const override
QModelIndex parent(const QModelIndex &child) const override
QModelIndex index(int row, int column, const QModelIndex &parent=QModelIndex()) const override
bool setData(const QModelIndex &index, const QVariant &value, int role=Qt::EditRole) override
QgsMapLayer * layerFromIndex(const QModelIndex &index) const
Returns the map layer corresponding to the specified index.
Base class for all map layer types.
Definition: qgsmaplayer.h:75
Represents a vector layer which manages a vector based data sets.