QGIS » QGIS Application

#include "../providers/ogr/qgsogrprovider.h"

    void loadOGRSublayers( QString uri, QStringList *list );

    /** This method will open a dialog so the user can select the sublayers

    * to load

    */

    void askUserForSublayers(QgsVectorLayer *layer);

/***************************************************************************

 *   Copyright (C) 2003 by Tim Sutton                                      *

 *   [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 "qgsogrsublayersdialog.h"

#include "qgscontexthelp.h"

//qt includes

//standard includes

QgsOGRSublayersDialog::QgsOGRSublayersDialog( QWidget* parent, Qt::WFlags fl )

    : QDialog( parent, fl )

{

  setupUi( this );

  QStringList labels=QStringList() << "Layer ID" << "Layer name" << "Nb of features" << "Geometry type";

	layersTable->setHeaderLabels(labels);

}

QgsOGRSublayersDialog::~QgsOGRSublayersDialog()

{

}

void QgsOGRSublayersDialog::setURI(QString uri)

{

  if ( uri.contains('&', Qt::CaseSensitive))

	{

  	// If we get here, there are some options added to the filename.

  	// A valid uri is of the form: filename&option1=value1&option2=value2,...

    // We want only the filename here, so we get the first part of the split.

  	QStringList theURIParts = uri.split("&");

	  uri = theURIParts.at( 0 );

	}

  mURI=QString(uri);

}

void QgsOGRSublayersDialog::on_buttonBox_accepted()

{

  //close the dialog

	QStringList list=QStringList();

	for (int i = 0; i < layersTable-> selectedItems().size(); i++)

	{

    QString theItem =layersTable-> selectedItems().at(i)->text(1);

  	list.append(theItem);

	}

	emit updated( mURI , &list );

  accept();

}

void QgsOGRSublayersDialog::on_buttonBox_rejected()

{

  reject();

}

void QgsOGRSublayersDialog::populateLayerTable (QStringList *theList)

{

	for (int i =0; i< theList->size(); i++){

		QString ligne = theList -> at(i);

		QStringList elements = ligne.split(":");

    QStringList item=QStringList();

    item << elements.at(0) << elements.at(1) << elements.at(2) << elements.at(3);

    layersTable -> addTopLevelItem(new QTreeWidgetItem(item));

	}	

}

  qgsogrsublayersdialog.cpp

  qgsogrsublayersdialog.h

#include "qgsogrsublayersdialog.h"

      // The next cast is possible because we know that the provider was ogr and it is 

      // necessary because the QgsMapLayer subLayers method returns an empty QStringList.

      // We have to get down to the actual provider to get the approriate list.

 	  	QgsOgrProvider* theCurrentProvider = ( QgsOgrProvider * )( layer->dataProvider());

			QStringList liste = theCurrentProvider->subLayers();

      // If the newly created layer has more than 1 layer of data available, we show the

      // sublayers selection dialog so the user can select the sublayers to actually load.

			if ( liste.count() > 1)

			{

        askUserForSublayers(layer);				

        // The first layer loaded is not usefull in that case. The user can select it in 

        // the list if he wants to load it.

        delete layer;

			}else  // there is 1 layer of data available

      {

        // Register this layer with the layers registry

        QgsMapLayerRegistry::instance()->addMapLayer( layer );

        // notify the project we've made a change

        QgsProject::instance()->dirty( true );

      }

// This method is the method that does the real job. If the layer given in

// parameter is NULL, then the method tries to act on the activeLayer. 

void QgisApp::askUserForSublayers(QgsVectorLayer *layer)

{

  if (NULL == layer)

  {

    if (mQgisInterface->activeLayer() && 

        mQgisInterface->activeLayer()->type() == QgsMapLayer::VectorLayer && 

        0 == (( QgsVectorLayer *)mQgisInterface->activeLayer())->dataProvider()->name().compare("ogr"))

    {

      layer=(QgsVectorLayer*)(mQgisInterface->activeLayer());

    }

    else return;

  }

  QgsOgrProvider* theCurrentProvider = ( QgsOgrProvider * )( layer->dataProvider());

  QStringList liste = theCurrentProvider->subLayers();

  QgsOGRSublayersDialog *chooseSublayersDialog;  

  // We initialize a selection dialog and display it.

  chooseSublayersDialog = new QgsOGRSublayersDialog(this , QgisGui::ModalDialogFlags );

  chooseSublayersDialog->setURI(layer-> source());

	chooseSublayersDialog->setAttribute( Qt::WA_DeleteOnClose );

	chooseSublayersDialog->show();

	chooseSublayersDialog->populateLayerTable( & liste );

  // We want to get the signal from this dialog when the user has done his choice

  connect( chooseSublayersDialog, SIGNAL( updated(QString, QStringList * ) ), this, SLOT( loadOGRSublayers(QString, QStringList *  ) ) );

}

// This method will load with OGR the layers  in parameter. 

// This method has been conceived to use the new URI

// format of the ogrprovider so as to give precisions about which

// sublayer to load into QGIS. It is normally triggered by the

// sublayer selection dialog.

void QgisApp::loadOGRSublayers( QString uri, QStringList *list)

{

  // The uri must contain the actual uri of the vectorLayer from which we are

  // going to load the sublayers.

  QString fileName = QFileInfo(uri).baseName();

	for (int i = 0; i < list->size(); i++)

	{

		QString composedURI=uri+"&layername="+list->at(i);		

		QgsVectorLayer *layer=addVectorLayer(composedURI,fileName+":"+list->at(i),"ogr");

	}

}

/***************************************************************************

 *   Copyright (C) 2003 by Tim Sutton                                      *

 *   [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.                                   *

 ***************************************************************************/

#ifndef QGSOGRSUBLAYERSDIALOG_H

#define QGSOGRSUBLAYERSDIALOG_H

#include <QDialog>

#include <QTableWidgetItem>

#include <ui_qgsogrsublayersdialogbase.h>

#include <qgslogger.h>

/**

@author Tim Sutton

*/

class QgsOGRSublayersDialog : public QDialog, private Ui::QgsOGRSublayersDialogBase

{

    Q_OBJECT

  public:

    QgsOGRSublayersDialog( QWidget* parent = 0, Qt::WFlags fl = 0 );

    ~QgsOGRSublayersDialog();

		void populateLayerTable (QStringList *theList);

		QStringList * getSelection ();

    void setURI(QString uri);

  private:

    static const int context_id = 0;

    QString mURI;

  private slots:

		void on_buttonBox_accepted();

    void on_buttonBox_rejected();

	signals:

		void updated(QString uri, QStringList*list);

};

#endif

	   /**

     * Sub-layers handled by this provider, in order from bottom to top

     *

     * Sub-layers are used when the provider's source can combine layers

     * it knows about in some way before it hands them off to the provider.

     */

    virtual QStringList subLayers();

  protected:

    /** loads fields from input file to member attributeFields */

    void loadFields();

    /**Get an attribute associated with a feature*/

    void getFeatureAttribute( OGRFeatureH ogrFet, QgsFeature & f, int attindex );

  // This part of the code parses the uri transmitted to the ogr provider to 

  // get the options the client wants us to apply

  QString mFilePath;

  QString theLayerName;

  int theLayerIndex=0;

  // If there is no & in the uri, then the uri is just the filename. The loaded

  // layer will be layer 0.

  if ( ! uri.contains('&', Qt::CaseSensitive))

  {

    mFilePath = uri;

  }

  else

  {

  // If we get here, there are some options added to the filename. We must parse

  // the different parts separated by &, and among each option, the name and the

  // value around the =.

  // A valid uri is of the form: filename&option1=value1&option2=value2,...

 	  QStringList theURIParts = uri.split("&");

    mFilePath = theURIParts.at( 0 );

    for (int i = 1 ; i < theURIParts.size(); i++ )

    {

      QStringList theInstruction = theURIParts.at( i ).split( "=" );

      if ( 0 == theInstruction.at( 0 ).compare( QString( "layerid" )))

      {

        bool ok;

        theLayerIndex = theInstruction.at( 1 ).toInt( &ok );

        if ( ! ok )

        {

          theLayerIndex = 0;

        }

      }

      if ( 0 == theInstruction.at( 0 ).compare( QString( "layername" )))

      {

        theLayerName = theInstruction.at( 1 );

      }				

    }

  }

  QgsLogger::debug("[OGR] mFilePath: " + mFilePath);

  QgsLogger::debug("[OGR] theLayerIndex: "+theLayerIndex);

  QgsLogger::debug("[OGR] theLayerName: "+theLayerName);

  ogrDataSource = OGROpen( QFile::encodeName( mFilePath ).constData(), TRUE, &ogrDriver );

    ogrDataSource = OGROpen( QFile::encodeName( mFilePath ).constData(), FALSE, &ogrDriver );

    // We get the layer which was requested by the uri. The layername

    // has precedence over the layerid if both are given.

    if ( NULL == theLayerName )

    {

      ogrLayer = OGR_DS_GetLayer( ogrDataSource, theLayerIndex );

    }else

    {

      ogrLayer = OGR_DS_GetLayerByName( ogrDataSource, (char*)(theLayerName.toLocal8Bit().data()) );

    }

QStringList QgsOgrProvider::subLayers()

{

  QStringList theList = QStringList();

  if (! valid )

  {

    return theList;

  }

  for ( int i = 0; i < layerCount() ; i++ )

  {  

    QString theLayerName = QString(OGR_FD_GetName(OGR_L_GetLayerDefn(OGR_DS_GetLayer( ogrDataSource, i ))));

    QGis::WkbType layerGeomType = (QGis::WkbType)(OGR_FD_GetGeomType(OGR_L_GetLayerDefn(OGR_DS_GetLayer( ogrDataSource, i ))));

    int theLayerFeatureCount=OGR_L_GetFeatureCount(OGR_DS_GetLayer( ogrDataSource, i ),1) ;

    QString geom;

    switch (layerGeomType)

    {

      case QGis::WKBPoint:

        geom="WKBPoint";

        break;

      case QGis::WKBLineString:

        geom="WKBLineString";

        break;

      case QGis::WKBPolygon:

        geom="WKBPolygon";

        break;

      case QGis::WKBMultiPoint:

        geom="WKBMultiPoint";

        break;

      case QGis::WKBMultiLineString:

        geom="WKBMultiLineString";

        break;

      case QGis::WKBUnknown:

        geom="WKBUnknown";

        break;

      case QGis::WKBPoint25D:

        geom="WKBPoint25D";

        break;

      case QGis::WKBLineString25D:

        geom="WKBLineString25D";

        break;

      case QGis::WKBPolygon25D:

        geom="WKBPolygon25D";

        break;

      case QGis::WKBMultiPoint25D:

        geom="WKBMultiPoint25D";

        break;

      case QGis::WKBMultiLineString25D:

        geom="WKBMultiLineString25D";

        break;

      case QGis::WKBMultiPolygon25D:

        geom="WKBMultiPolygon25D";

        break;

      default:

        geom="Unknown OGR geometry ID->" + QString::number(layerGeomType);

    }

    theList.append(QString::number(i)+":"+ theLayerName+":"+QString::number(theLayerFeatureCount)+":"+geom);

  }

  return theList;

}

<ui version="4.0" >

 <class>QgsOGRSublayersDialogBase</class>

 <widget class="QDialog" name="QgsOGRSublayersDialogBase" >

  <property name="geometry" >

   <rect>

    <x>0</x>

    <y>0</y>

    <width>584</width>

    <height>535</height>

   </rect>

  </property>

  <property name="windowTitle" >

   <string>Select OGR layers to load</string>

  </property>

  <property name="windowIcon" >

   <iconset>../../../qgis_1.0.0/src/plugins/ogrsublayers</iconset>

  </property>

  <layout class="QGridLayout" >

   <property name="leftMargin" >

    <number>9</number>

   </property>

   <property name="topMargin" >

    <number>9</number>

   </property>

   <property name="rightMargin" >

    <number>9</number>

   </property>

   <property name="bottomMargin" >

    <number>9</number>

   </property>

   <property name="horizontalSpacing" >

    <number>6</number>

   </property>

   <property name="verticalSpacing" >

    <number>6</number>

   </property>

   <item row="0" column="0" >

    <widget class="QLabel" name="txtHeading" >

     <property name="sizePolicy" >

      <sizepolicy vsizetype="Minimum" hsizetype="Preferred" >

       <horstretch>0</horstretch>

       <verstretch>0</verstretch>

      </sizepolicy>

     </property>

     <property name="font" >

      <font>

       <family>Sans Serif</family>

       <pointsize>24</pointsize>

       <weight>75</weight>

       <italic>false</italic>

       <bold>true</bold>

       <underline>false</underline>

       <strikeout>false</strikeout>

      </font>

     </property>

     <property name="text" >

      <string>Sub layers list</string>

     </property>

     <property name="alignment" >

      <set>Qt::AlignCenter</set>

     </property>

    </widget>

   </item>

   <item row="8" column="0" >

    <widget class="QDialogButtonBox" name="buttonBox" >

     <property name="orientation" >

      <enum>Qt::Horizontal</enum>

     </property>

     <property name="standardButtons" >

      <set>QDialogButtonBox::Cancel|QDialogButtonBox::NoButton|QDialogButtonBox::Ok</set>

     </property>

    </widget>

   </item>

   <item row="1" column="0" >

    <widget class="QTextEdit" name="textEdit" >

     <property name="html" >

      <string>&lt;html>&lt;head>&lt;meta name="qrichtext" content="1" />&lt;style type="text/css">

p, li { white-space: pre-wrap; }

</style></head><body style=" font-family:'Sans Serif'; font-size:9pt; font-weight:400; font-style:normal;">

<p style=" margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;">This is the list of all layers available in the datasource of the active layer. You can select the layers to load. The layers will be loaded when you press "OK".</p>

<p style="-qt-paragraph-type:empty; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;"></p>

<p style=" margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;">The layer name is format dependant. Consult the OGR documentation or the documentation of your data format to determine the nature of the included information.</p>

<p style="-qt-paragraph-type:empty; margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;"></p>

&lt;p style=" margin-top:0px; margin-bottom:0px; margin-left:0px; margin-right:0px; -qt-block-indent:0; text-indent:0px;">&lt;span style=" font-weight:600;">Be advised: &lt;/span>selecting an already opened layer will not generate an error message and the layer will end up loaded twice!&lt;/p>&lt;/body>&lt;/html></string>

     </property>

    </widget>

   </item>

   <item row="2" column="0" >

    <widget class="QTreeWidget" name="layersTable" >

     <property name="windowModality" >

      <enum>Qt::NonModal</enum>

     </property>

     <property name="selectionMode" >

      <enum>QAbstractItemView::ExtendedSelection</enum>

     </property>

     <property name="selectionBehavior" >

      <enum>QAbstractItemView::SelectRows</enum>

     </property>

     <column>

      <property name="text" >

       <string>1</string>

      </property>

     </column>

    </widget>

   </item>

  </layout>

 </widget>

 <layoutdefault spacing="6" margin="11" />

 <resources/>

 <connections/>

</ui>