qgsorganizetablecolumnsdialog.cpp

Go to the documentation of this file.

/***************************************************************************
  qgsorganizetablecolumnsdialog.cpp
  -------------------
         date                 : Feb 2016
         copyright            : Stéphane Brunner
         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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include <QMessageBox>
 
#include "qgsorganizetablecolumnsdialog.h"
#include "moc_qgsorganizetablecolumnsdialog.cpp"
#include "qgsattributetableview.h"
 
#include "qgsapplication.h"
#include "qgsvectorlayer.h"
#include "qgsexpression.h"
 
#include "qgssearchquerybuilder.h"
#include "qgsmapcanvas.h"
#include "qgsproject.h"
#include "qgsmessagebar.h"
#include "qgsrubberband.h"
#include "qgsfields.h"
#include "qgseditorwidgetregistry.h"
 
#include "qgsgui.h"
 
 
QgsOrganizeTableColumnsDialog::QgsOrganizeTableColumnsDialog( const QgsVectorLayer *vl, const QgsAttributeTableConfig &config, QWidget *parent, Qt::WindowFlags flags )
  : QDialog( parent, flags )
{
  setupUi( this );
  QgsGui::enableAutoGeometryRestore( this );
 
  connect( mShowAllButton, &QAbstractButton::clicked, this, &QgsOrganizeTableColumnsDialog::showAll );
  connect( mHideAllButton, &QAbstractButton::clicked, this, &QgsOrganizeTableColumnsDialog::hideAll );
  connect( mToggleSelectionButton, &QAbstractButton::clicked, this, &QgsOrganizeTableColumnsDialog::toggleSelection );
 
  if ( vl )
  {
    mConfig = config;
    const QgsFields fields = vl->fields();
    mConfig.update( fields );
 
    mFieldsList->clear();
 
    const auto constColumns = mConfig.columns();
    for ( const QgsAttributeTableConfig::ColumnConfig &columnConfig : constColumns )
    {
      QListWidgetItem *item = nullptr;
      if ( columnConfig.type == QgsAttributeTableConfig::Action )
      {
        item = new QListWidgetItem( tr( "[Action Widget]" ), mFieldsList );
        item->setIcon( QgsApplication::getThemeIcon( QStringLiteral( "/propertyicons/action.svg" ) ) );
      }
      else
      {
        const int idx = fields.lookupField( columnConfig.name );
        item = new QListWidgetItem( vl->attributeDisplayName( idx ), mFieldsList );
        item->setIcon( fields.iconForField( idx, true ) );
      }
 
      item->setCheckState( columnConfig.hidden ? Qt::Unchecked : Qt::Checked );
      item->setData( Qt::UserRole, QVariant::fromValue( columnConfig ) );
    }
  }
 
  if ( !vl || mConfig.columns().count() < 5 )
  {
    mShowAllButton->hide();
    mHideAllButton->hide();
    mToggleSelectionButton->hide();
  }
}
 
QgsOrganizeTableColumnsDialog::QgsOrganizeTableColumnsDialog( const QgsVectorLayer *vl, QWidget *parent, Qt::WindowFlags flags )
  : QgsOrganizeTableColumnsDialog( vl, vl->attributeTableConfig(), parent, flags )
{
}
 
QgsAttributeTableConfig QgsOrganizeTableColumnsDialog::config() const
{
  QVector<QgsAttributeTableConfig::ColumnConfig> columns;
  columns.reserve( mFieldsList->count() );
 
  for ( int i = 0; i < mFieldsList->count(); i++ )
  {
    const QListWidgetItem *item = mFieldsList->item( i );
    QgsAttributeTableConfig::ColumnConfig columnConfig = item->data( Qt::UserRole ).value<QgsAttributeTableConfig::ColumnConfig>();
 
    columnConfig.hidden = item->checkState() == Qt::Unchecked;
 
    columns.append( columnConfig );
  }
 
  QgsAttributeTableConfig config = mConfig;
  config.setColumns( columns );
  return config;
}
 
void QgsOrganizeTableColumnsDialog::showAll()
{
  for ( int i = 0; i < mFieldsList->count(); i++ )
  {
    mFieldsList->item( i )->setCheckState( Qt::Checked );
  }
}
 
void QgsOrganizeTableColumnsDialog::hideAll()
{
  for ( int i = 0; i < mFieldsList->count(); i++ )
  {
    mFieldsList->item( i )->setCheckState( Qt::Unchecked );
  }
}
 
void QgsOrganizeTableColumnsDialog::toggleSelection()
{
  for ( QListWidgetItem *item : mFieldsList->selectedItems() )
  {
    item->setCheckState( item->checkState() == Qt::Checked ? Qt::Unchecked : Qt::Checked );
  }
}