QGIS API Documentation  3.6.0-Noosa (5873452)
qgsattributetypeloaddialog.cpp
Go to the documentation of this file.
1 /***************************************************************************
2  qgsattributetypeloaddialog.cpp
3  -------------------
4  begin : June 2009
5  copyright : (C) 2000 by Richard Kostecky
6  email : [email protected]
7  ***************************************************************************/
8 
9 /***************************************************************************
10  * *
11  * This program is free software; you can redistribute it and/or modify *
12  * it under the terms of the GNU General Public License as published by *
13  * the Free Software Foundation; either version 2 of the License, or *
14  * (at your option) any later version. *
15  * *
16  ***************************************************************************/
17 
19 
20 #include "qgsmaplayer.h"
21 #include "qgsfeatureiterator.h"
22 #include "qgsvectordataprovider.h"
23 #include "qgslogger.h"
24 #include "qgsproject.h"
25 #include "qgsvectorlayer.h"
26 
27 #include <QTableWidgetItem>
28 #include <QLineEdit>
29 #include <QComboBox>
30 #include <QLabel>
31 #include <QFrame>
32 #include <QCompleter>
33 #include <QSpinBox>
34 #include <QPushButton>
35 #include <QHBoxLayout>
36 #include <QFileDialog>
37 
39  : mLayer( vl )
40 {
41  setupUi( this );
42 
43  connect( layerComboBox, static_cast<void ( QComboBox::* )( int )>( &QComboBox::currentIndexChanged ), this, &QgsAttributeTypeLoadDialog::fillComboBoxes );
44  connect( keyComboBox, static_cast<void ( QComboBox::* )( int )>( &QComboBox::currentIndexChanged ), this, [ = ]( int index ) { createPreview( index ); } );
45  connect( valueComboBox, static_cast<void ( QComboBox::* )( int )>( &QComboBox::currentIndexChanged ), this, [ = ]( int index ) { createPreview( index ); } );
46  connect( previewButton, &QAbstractButton::pressed, this, &QgsAttributeTypeLoadDialog::previewButtonPushed );
47 
48  fillLayerList();
49 
50  keyComboBox->setDisabled( true );
51  valueComboBox->setDisabled( true );
52 }
53 
55 {
56  mLayer = layer;
57 }
58 
59 void QgsAttributeTypeLoadDialog::previewButtonPushed()
60 {
61  createPreview( valueComboBox->currentIndex(), true );
62 }
63 
64 void QgsAttributeTypeLoadDialog::fillLayerList()
65 {
66  layerComboBox->blockSignals( true );
67  layerComboBox->clear();
68  Q_FOREACH ( QgsMapLayer *l, QgsProject::instance()->mapLayers() )
69  {
70  QgsVectorLayer *vl = qobject_cast< QgsVectorLayer * >( l );
71  if ( vl )
72  layerComboBox->addItem( vl->name(), vl->id() );
73  }
74  layerComboBox->setCurrentIndex( -1 );
75  layerComboBox->blockSignals( false );
76 }
77 
78 void QgsAttributeTypeLoadDialog::fillComboBoxes( int layerIndex )
79 {
80  keyComboBox->blockSignals( true );
81  valueComboBox->blockSignals( true );
82 
83  //clear comboboxes first
84  keyComboBox->clear();
85  valueComboBox->clear();
86 
87  QgsVectorLayer *vLayer = qobject_cast<QgsVectorLayer *>( layerIndex < 0 ? nullptr : QgsProject::instance()->mapLayer( layerComboBox->itemData( layerIndex ).toString() ) );
88  if ( vLayer )
89  {
90  QMap<QString, int> fieldMap = vLayer->dataProvider()->fieldNameMap();
91  QMap<QString, int>::iterator it = fieldMap.begin();
92  for ( ; it != fieldMap.end(); ++it )
93  {
94  keyComboBox->addItem( it.key(), it.value() );
95  valueComboBox->addItem( it.key(), it.value() );
96  }
97  }
98 
99  keyComboBox->setEnabled( nullptr != vLayer );
100  valueComboBox->setEnabled( nullptr != vLayer );
101 
102  keyComboBox->setCurrentIndex( -1 );
103  valueComboBox->setCurrentIndex( -1 );
104 
105  keyComboBox->blockSignals( false );
106  valueComboBox->blockSignals( false );
107 }
108 
109 void QgsAttributeTypeLoadDialog::createPreview( int fieldIndex, bool full )
110 {
111  previewTableWidget->clearContents();
112 
113  for ( int i = previewTableWidget->rowCount() - 1; i > 0; i-- )
114  {
115  previewTableWidget->removeRow( i );
116  }
117  if ( layerComboBox->currentIndex() < 0 || fieldIndex < 0 )
118  {
119  //when nothing is selected there is no reason for preview
120  return;
121  }
122  int idx = keyComboBox->currentData().toInt();
123  int idx2 = valueComboBox->currentData().toInt();
124  QgsMapLayer *dataLayer = QgsProject::instance()->mapLayer( layerComboBox->currentData().toString() );
125  QgsVectorLayer *vLayer = qobject_cast<QgsVectorLayer *>( dataLayer );
126  if ( !vLayer )
127  return;
128 
129  QgsAttributeList attributeList = QgsAttributeList();
130  attributeList.append( idx );
131  attributeList.append( idx2 );
132 
133  QgsFeatureIterator fit = vLayer->getFeatures( QgsFeatureRequest().setFlags( QgsFeatureRequest::NoGeometry ).setSubsetOfAttributes( attributeList ) );
134 
135  QgsFeature f;
136  QMap<QString, QVariant> valueMap;
137  while ( fit.nextFeature( f ) )
138  {
139  QVariant val1 = f.attribute( idx );
140  QVariant val2 = f.attribute( idx2 );
141  if ( val1.isValid() && !val1.isNull() && !val1.toString().isEmpty()
142  && val2.isValid() && !val2.isNull() && !val2.toString().isEmpty() )
143  {
144  valueMap.insert( val1.toString(), val2.toString() );
145  }
146  if ( !full && valueMap.size() > 8 )
147  break; //just first entries all on button
148  }
149  int row = 0;
150  for ( QMap<QString, QVariant>::iterator mit = valueMap.begin(); mit != valueMap.end(); ++mit, row++ )
151  {
152  previewTableWidget->insertRow( row );
153  previewTableWidget->setItem( row, 0, new QTableWidgetItem( mit.value().toString() ) );
154  previewTableWidget->setItem( row, 1, new QTableWidgetItem( mit.key() ) );
155  }
156 }
157 
158 QMap<QString, QVariant> &QgsAttributeTypeLoadDialog::valueMap()
159 {
160  return mValueMap;
161 }
162 
164 {
165  return nullCheckBox->isChecked();
166 }
167 
168 void QgsAttributeTypeLoadDialog::loadDataToValueMap()
169 {
170  mValueMap.clear();
171  int idx = keyComboBox->currentData().toInt();
172  int idx2 = valueComboBox->currentData().toInt();
173  QgsMapLayer *dataLayer = QgsProject::instance()->mapLayer( layerComboBox->currentData().toString() );
174  QgsVectorLayer *vLayer = qobject_cast<QgsVectorLayer *>( dataLayer );
175  if ( !vLayer )
176  return;
177 
178  QgsAttributeList attributeList = QgsAttributeList();
179  attributeList.append( idx );
180  attributeList.append( idx2 );
181 
182  QgsFeatureIterator fit = vLayer->getFeatures( QgsFeatureRequest().setFlags( QgsFeatureRequest::NoGeometry ).setSubsetOfAttributes( attributeList ) );
183 
184  QgsFeature f;
185  while ( fit.nextFeature( f ) )
186  {
187  QVariant val = f.attribute( idx );
188  if ( val.isValid() && !val.isNull() && !val.toString().isEmpty() )
189  {
190  mValueMap.insert( f.attribute( idx2 ).toString(), val );
191  }
192  }
193 }
194 
196 {
197  //store data to output variable
198  loadDataToValueMap();
199  QDialog::accept();
200 }
Wrapper for iterator of features from vector data provider or vector layer.
Base class for all map layer types.
Definition: qgsmaplayer.h:64
QMap< QString, int > fieldNameMap() const
Returns a map where the key is the name of the field and the value is its index.
void setVectorLayer(QgsVectorLayer *layer)
Sets predefined vector layer for selection of data.
The feature class encapsulates a single feature including its id, geometry and a list of field/values...
Definition: qgsfeature.h:55
void accept() override
Overloaded accept method which will write the feature field values, then delegate to QDialog::accept(...
QMap< QString, QVariant > & valueMap()
Returns the value map which is currently active.
QString id() const
Returns the layer&#39;s unique ID, which is used to access this layer from QgsProject.
QgsAttributeTypeLoadDialog(QgsVectorLayer *vl)
This class wraps a request for features to a vector layer (or directly its vector data provider)...
bool insertNull()
Returns true if the "Add NULL value" checkbox has been checked.
static QgsProject * instance()
Returns the QgsProject singleton instance.
Definition: qgsproject.cpp:430
QgsMapLayer * mapLayer(const QString &layerId) const
Retrieve a pointer to a registered layer by layer ID.
QgsFeatureIterator getFeatures(const QgsFeatureRequest &request=QgsFeatureRequest()) const FINAL
Query the layer for features specified in request.
QString name
Definition: qgsmaplayer.h:68
QgsVectorDataProvider * dataProvider() FINAL
Returns the layer&#39;s data provider, it may be null.
QList< int > QgsAttributeList
Definition: qgsfield.h:27
bool nextFeature(QgsFeature &f)
Geometry is not required. It may still be returned if e.g. required for a filter condition.
Represents a vector layer which manages a vector based data sets.
QVariant attribute(const QString &name) const
Lookup attribute value from attribute name.
Definition: qgsfeature.cpp:262