missed to include moved querybuilder files

src/gui/qgsquerybuilder.cpp

@@ -0,0 +1,315 @@
+/***************************************************************************
+                qgsquerybuilder.cpp - Query Builder
+                     --------------------------------------
+               Date                 : 2004-11-19
+               Copyright            : (C) 2004 by Gary E.Sherman
+               Email                : sherman at mrcc.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.                                   *
+ *                                                                         *
+ ***************************************************************************/
+#include "qgsquerybuilder.h"
+#include "qgslogger.h"
+#include <QListView>
+#include <QMessageBox>
+#include <QRegExp>
+#include <QPushButton>
+#include "qgsvectorlayer.h"
+#include "qgsvectordataprovider.h"
+
+// constructor used when the query builder must make its own
+// connection to the database
+QgsQueryBuilder::QgsQueryBuilder( QgsVectorLayer *layer,
+                                  QWidget *parent, Qt::WFlags fl )
+    : QDialog( parent, fl )
+    , mPreviousFieldRow( -1 )
+    , mLayer( layer )
+{
+  setupUi( this );
+
+  QPushButton *pbn = new QPushButton( tr( "&Test" ) );
+  buttonBox->addButton( pbn, QDialogButtonBox::ActionRole );
+  connect( pbn, SIGNAL( clicked() ), this, SLOT( test() ) );
+
+  pbn = new QPushButton( tr( "&Clear" ) );
+  buttonBox->addButton( pbn, QDialogButtonBox::ActionRole );
+  connect( pbn, SIGNAL( clicked() ), this, SLOT( clear() ) );
+
+  setupGuiViews();
+
+  mOrigSubsetString = layer->subsetString();
+
+  lblDataUri->setText( layer->name() );
+  txtSQL->setText( mOrigSubsetString );
+
+  populateFields();
+}
+
+QgsQueryBuilder::~QgsQueryBuilder()
+{
+}
+
+void QgsQueryBuilder::populateFields()
+{
+  for ( QgsFieldMap::const_iterator it = mLayer->pendingFields().begin(); it != mLayer->pendingFields().end(); it++ )
+  {
+    QStandardItem *myItem = new QStandardItem( it->name() );
+    myItem->setData( it.key() );
+    myItem->setEditable( false );
+    mModelFields->insertRow( mModelFields->rowCount(), myItem );
+  }
+
+  // All fields get ... setup
+  setupLstFieldsModel();
+}
+
+void QgsQueryBuilder::setupLstFieldsModel()
+{
+  lstFields->setModel( mModelFields );
+}
+
+void QgsQueryBuilder::setupGuiViews()
+{
+  //Initialize the models
+  mModelFields = new QStandardItemModel();
+  mModelValues = new QStandardItemModel();
+  // Modes
+  lstFields->setViewMode( QListView::ListMode );
+  lstValues->setViewMode( QListView::ListMode );
+  lstFields->setSelectionBehavior( QAbstractItemView::SelectRows );
+  lstValues->setSelectionBehavior( QAbstractItemView::SelectRows );
+  // Performance tip since Qt 4.1
+  lstFields->setUniformItemSizes( true );
+  lstValues->setUniformItemSizes( true );
+  // Colored rows
+  lstFields->setAlternatingRowColors( true );
+  lstValues->setAlternatingRowColors( true );
+}
+
+void QgsQueryBuilder::fillValues( int idx, int limit )
+{
+  // clear the model
+  mModelValues->clear();
+
+  // determine the field type
+  QList<QVariant> values;
+  mLayer->uniqueValues( idx, values, limit );
+
+  for ( int i = 0; i < values.size(); i++ )
+  {
+    QStandardItem *myItem = new QStandardItem( values[i].toString() );
+    myItem->setEditable( false );
+    mModelValues->insertRow( mModelValues->rowCount(), myItem );
+  }
+}
+
+void QgsQueryBuilder::on_btnSampleValues_clicked()
+{
+  lstValues->setCursor( Qt::WaitCursor );
+
+  //delete connection mModelValues and lstValues
+  QStandardItemModel *tmp = new QStandardItemModel();
+  lstValues->setModel( tmp );
+  //Clear and fill the mModelValues
+  fillValues( mModelFields->data( lstFields->currentIndex(), Qt::UserRole + 1 ).toInt(), 25 );
+  lstValues->setModel( mModelValues );
+  lstValues->setCursor( Qt::ArrowCursor );
+  //delete the tmp
+  delete tmp;
+
+}
+
+void QgsQueryBuilder::on_btnGetAllValues_clicked()
+{
+  lstValues->setCursor( Qt::WaitCursor );
+
+  //delete connection mModelValues and lstValues
+  QStandardItemModel *tmp = new QStandardItemModel();
+  lstValues->setModel( tmp );
+  //Clear and fill the mModelValues
+  fillValues( mModelFields->data( lstFields->currentIndex(), Qt::UserRole + 1 ).toInt(), -1 );
+  lstValues->setModel( mModelValues );
+  lstValues->setCursor( Qt::ArrowCursor );
+  //delete the tmp
+  delete tmp;
+}
+
+void QgsQueryBuilder::test()
+{
+  // test the sql statement to see if it works
+  // by counting the number of records that would be
+  // returned
+
+  if ( mLayer->setSubsetString( txtSQL->toPlainText() ) )
+  {
+    QMessageBox::information( this,
+                              tr( "Query Result" ),
+                              tr( "The where clause returned %n row(s).", "returned test rows", mLayer->featureCount() ) );
+  }
+  else if ( mLayer->dataProvider()->hasErrors() )
+  {
+    QMessageBox::warning( this,
+                          tr( "Query Failed" ),
+                          tr( "An error occurred when executing the query." )
+                          + tr( "\nThe data provider said:\n%1" ).arg( mLayer->dataProvider()->errors().join( "\n" ) ) );
+    mLayer->dataProvider()->clearErrors();
+  }
+  else
+  {
+    QMessageBox::warning( this,
+                          tr( "Query Failed" ),
+                          tr( "An error occurred when executing the query." ) );
+  }
+}
+
+void QgsQueryBuilder::accept()
+{
+  // if user hits Ok and there is no query, skip the validation
+  if ( !txtSQL->toPlainText().trimmed().isEmpty() )
+  {
+    if ( !mLayer->setSubsetString( txtSQL->toPlainText() ) )
+    {
+      //error in query - show the problem
+      if ( mLayer->dataProvider()->hasErrors() )
+      {
+        QMessageBox::warning( this,
+                              tr( "Query Failed" ),
+                              tr( "An error occurred when executing the query." )
+                              + tr( "\nThe data provider said:\n%1" ).arg( mLayer->dataProvider()->errors().join( "\n" ) ) );
+        mLayer->dataProvider()->clearErrors();
+      }
+      else
+      {
+        QMessageBox::warning( this, tr( "Error in Query" ), tr( "The subset string could not be set" ) );
+      }
+      return;
+    }
+  }
+
+  QDialog::accept();
+}
+
+void QgsQueryBuilder::reject()
+{
+  if ( mLayer->subsetString() != mOrigSubsetString )
+    mLayer->setSubsetString( mOrigSubsetString );
+
+  QDialog::reject();
+}
+
+void QgsQueryBuilder::on_btnEqual_clicked()
+{
+  txtSQL->insertPlainText( " = " );
+}
+
+void QgsQueryBuilder::on_btnLessThan_clicked()
+{
+  txtSQL->insertPlainText( " < " );
+}
+
+void QgsQueryBuilder::on_btnGreaterThan_clicked()
+{
+  txtSQL->insertPlainText( " > " );
+}
+
+void QgsQueryBuilder::on_btnPct_clicked()
+{
+  txtSQL->insertPlainText( "%" );
+}
+
+void QgsQueryBuilder::on_btnIn_clicked()
+{
+  txtSQL->insertPlainText( " IN " );
+}
+
+void QgsQueryBuilder::on_btnNotIn_clicked()
+{
+  txtSQL->insertPlainText( " NOT IN " );
+}
+
+void QgsQueryBuilder::on_btnLike_clicked()
+{
+  txtSQL->insertPlainText( " LIKE " );
+}
+
+QString QgsQueryBuilder::sql()
+{
+  return txtSQL->toPlainText();
+}
+
+void QgsQueryBuilder::setSql( QString sqlStatement )
+{
+  txtSQL->setText( sqlStatement );
+}
+
+void QgsQueryBuilder::on_lstFields_clicked( const QModelIndex &index )
+{
+  if ( mPreviousFieldRow != index.row() )
+  {
+    mPreviousFieldRow = index.row();
+
+    btnSampleValues->setEnabled( true );
+    btnGetAllValues->setEnabled( true );
+
+    mModelValues->clear();
+  }
+}
+
+void QgsQueryBuilder::on_lstFields_doubleClicked( const QModelIndex &index )
+{
+  txtSQL->insertPlainText( "\"" + mLayer->pendingFields()[ mModelFields->data( index, Qt::UserRole+1 ).toInt()].name() + "\"" );
+}
+
+void QgsQueryBuilder::on_lstValues_doubleClicked( const QModelIndex &index )
+{
+  txtSQL->insertPlainText( "'" + mModelValues->data( index ).toString() + "'" );
+}
+
+void QgsQueryBuilder::on_btnLessEqual_clicked()
+{
+  txtSQL->insertPlainText( " <= " );
+}
+
+void QgsQueryBuilder::on_btnGreaterEqual_clicked()
+{
+  txtSQL->insertPlainText( " >= " );
+}
+
+void QgsQueryBuilder::on_btnNotEqual_clicked()
+{
+  txtSQL->insertPlainText( " != " );
+}
+
+void QgsQueryBuilder::on_btnAnd_clicked()
+{
+  txtSQL->insertPlainText( " AND " );
+}
+
+void QgsQueryBuilder::on_btnNot_clicked()
+{
+  txtSQL->insertPlainText( " NOT " );
+}
+
+void QgsQueryBuilder::on_btnOr_clicked()
+{
+  txtSQL->insertPlainText( " OR " );
+}
+
+void QgsQueryBuilder::clear()
+{
+  txtSQL->clear();
+  mLayer->setSubsetString( "" );
+}
+
+void QgsQueryBuilder::on_btnILike_clicked()
+{
+  txtSQL->insertPlainText( " ILIKE " );
+}
+void QgsQueryBuilder::setDatasourceDescription( QString uri )
+{
+  lblDataUri->setText( uri );
+}

src/gui/qgsquerybuilder.h

@@ -0,0 +1,139 @@
+/***************************************************************************
+    qgsquerybuilder.h - query builder
+     --------------------------------------
+    Date                 : 2004-11-19
+    Copyright            : (C) 2004 by Gary E.Sherman
+    Email                : sherman at mrcc.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 QGSQUERYBUILDER_H
+#define QGSQUERYBUILDER_H
+#include <map>
+#include <vector>
+#include <QStandardItemModel>
+#include <QStandardItem>
+#include <QModelIndex>
+#include "ui_qgsquerybuilderbase.h"
+#include "qgisgui.h"
+#include "qgsfield.h"
+#include "qgscontexthelp.h"
+
+class QgsVectorLayer;
+
+/*!
+ * \class QgsQueryBuilder
+ * \brief Query Builder for layers.
+ *
+ * The query builder allows interactive creation of a SQL for limiting the
+ * features displayed in a database layer.  The fields in the table are
+ * displayed and sample values (or all values) can be viewed to aid in
+ * constructing the query. A test function returns the number of features that
+ * will be returned.
+ *
+ */
+class GUI_EXPORT QgsQueryBuilder : public QDialog, private Ui::QgsQueryBuilderBase
+{
+    Q_OBJECT
+  public:
+    /*! Constructor which takes the table name.
+    * This constructor is used when adding layers to the map since
+    * the query builder connection as the layer selection dialog.
+    * @param tableName Name of the table being queried
+    * @param parent Parent widget
+    * @param fl dialog flags
+    */
+    QgsQueryBuilder( QString tableName, QWidget *parent = 0,
+                     Qt::WFlags fl = QgisGui::ModalDialogFlags );
+
+    /*! This constructor is used when the query builder is called from the
+     * vector layer properties dialog
+     * @param layer existing vector layer
+     * @param parent Parent widget
+     * @param fl dialog flags
+     */
+    QgsQueryBuilder( QgsVectorLayer *layer, QWidget *parent = 0,
+                     Qt::WFlags fl = QgisGui::ModalDialogFlags );
+
+    ~QgsQueryBuilder();
+
+  public slots:
+    void accept();
+    void reject();
+    void clear();
+    void on_btnEqual_clicked();
+    void on_btnLessThan_clicked();
+    void on_btnGreaterThan_clicked();
+    void on_btnPct_clicked();
+    void on_btnIn_clicked();
+    void on_btnNotIn_clicked();
+    void on_btnLike_clicked();
+    void on_btnILike_clicked();
+    QString sql();
+    void setSql( QString sqlStatement );
+    void on_lstFields_clicked( const QModelIndex &index );
+    void on_lstFields_doubleClicked( const QModelIndex &index );
+    void on_lstValues_doubleClicked( const QModelIndex &index );
+    void on_btnLessEqual_clicked();
+    void on_btnGreaterEqual_clicked();
+    void on_btnNotEqual_clicked();
+    void on_btnAnd_clicked();
+    void on_btnNot_clicked();
+    void on_btnOr_clicked();
+
+    void on_buttonBox_helpRequested() { QgsContextHelp::run( metaObject()->className() ); }
+
+
+    /*! Test the constructed sql statement to see if the database likes it.
+     * The number of rows that would be returned is displayed in a message box.
+     * The test uses a "select count(*) from ..." query to test the SQL
+     * statement.
+     * @param showResults If true, the results are displayed in a QMessageBox
+     */
+    void test();
+    /*!
+     * Get all distinct values for the field. Values are inserted
+     * into the value list box
+     */
+    void on_btnGetAllValues_clicked();
+    /*!
+     * Get sample distinct values for the selected field. The sample size is
+     * limited to an arbitrary value (currently set to 25). The values
+     * are inserted into the values list box.
+     */
+    void on_btnSampleValues_clicked();
+    void setDatasourceDescription( QString uri );
+
+  private:
+    /*!
+     * Populate the field list for the selected table
+     */
+    void populateFields();
+
+    /*!
+     * Setup models for listviews
+     */
+    void setupGuiViews();
+    void setupLstFieldsModel();
+    void fillValues( int idx, int limit );
+
+    // private members
+    //! Model for fields ListView
+    QStandardItemModel *mModelFields;
+    //! Model for values ListView
+    QStandardItemModel *mModelValues;
+    //! Previous field row to delete model
+    int mPreviousFieldRow;
+
+    //! vector layer
+    QgsVectorLayer *mLayer;
+
+    //! original subset string
+    QString mOrigSubsetString;
+};
+#endif //QGSQUERYBUILDER_H