Move WMS data items to separate file

qgis · Oct 14, 2011 · e045d34 · e045d34
1 parent 30a6766
commit e045d34
Show file tree

Hide file tree

Showing 5 changed files with 246 additions and 230 deletions.
diff --git a/src/providers/wms/CMakeLists.txt b/src/providers/wms/CMakeLists.txt
@@ -3,11 +3,13 @@ SET (WMS_SRCS
   qgswmsprovider.cpp 
   qgswmssourceselect.cpp 
   qgswmsconnection.cpp 
+  qgswmsdataitems.cpp
 )
 SET (WMS_MOC_HDRS  
   qgswmsprovider.h 
   qgswmssourceselect.h 
   qgswmsconnection.h 
+  qgswmsdataitems.h
 )
 
 QT4_WRAP_CPP (WMS_MOC_SRCS ${WMS_MOC_HDRS})

diff --git a/src/providers/wms/qgswmsdataitems.cpp b/src/providers/wms/qgswmsdataitems.cpp
@@ -0,0 +1,192 @@
+#include "qgswmsdataitems.h"
+
+#include "qgslogger.h"
+
+#include "qgswmsconnection.h"
+#include "qgswmssourceselect.h"
+
+// ---------------------------------------------------------------------------
+QgsWMSConnectionItem::QgsWMSConnectionItem( QgsDataItem* parent, QString name, QString path )
+    : QgsDataCollectionItem( parent, name, path )
+{
+}
+
+QgsWMSConnectionItem::~QgsWMSConnectionItem()
+{
+}
+
+QVector<QgsDataItem*> QgsWMSConnectionItem::createChildren()
+{
+  QgsDebugMsg( "Entered" );
+  QVector<QgsDataItem*> children;
+  QgsWMSConnection connection( mName );
+  QgsWmsProvider *wmsProvider = connection.provider( );
+  if ( !wmsProvider )
+    return children;
+
+  QString mConnInfo = connection.connectionInfo();
+  QgsDebugMsg( "mConnInfo = " + mConnInfo );
+
+  // Attention: supportedLayers() gives tree leafes, not top level
+  if ( !wmsProvider->supportedLayers( mLayerProperties ) )
+    return children;
+
+  QgsWmsCapabilitiesProperty mCapabilitiesProperty = wmsProvider->capabilitiesProperty();
+  QgsWmsCapabilityProperty capabilityProperty = mCapabilitiesProperty.capability;
+
+  // Top level layer is present max once
+  // <element name="Capability">
+  //    <element ref="wms:Layer" minOccurs="0"/>  - default maxOccurs=1
+  QgsWmsLayerProperty topLayerProperty = capabilityProperty.layer;
+  foreach( QgsWmsLayerProperty layerProperty, topLayerProperty.layer )
+  {
+    // Attention, the name may be empty
+    QgsDebugMsg( QString::number( layerProperty.orderId ) + " " + layerProperty.name + " " + layerProperty.title );
+    QString pathName = layerProperty.name.isEmpty() ? QString::number( layerProperty.orderId ) : layerProperty.name;
+
+    QgsWMSLayerItem * layer = new QgsWMSLayerItem( this, layerProperty.title, mPath + "/" + pathName, mCapabilitiesProperty, mConnInfo, layerProperty );
+
+    children.append( layer );
+  }
+  return children;
+}
+
+bool QgsWMSConnectionItem::equal( const QgsDataItem *other )
+{
+  if ( type() != other->type() )
+  {
+    return false;
+  }
+  const QgsWMSConnectionItem *o = dynamic_cast<const QgsWMSConnectionItem *>( other );
+  return ( mPath == o->mPath && mName == o->mName && mConnInfo == o->mConnInfo );
+}
+// ---------------------------------------------------------------------------
+QgsWMSLayerItem::QgsWMSLayerItem( QgsDataItem* parent, QString name, QString path, QgsWmsCapabilitiesProperty capabilitiesProperty, QString connInfo, QgsWmsLayerProperty layerProperty )
+    : QgsLayerItem( parent, name, path, QString(), QgsLayerItem::Raster, "wms" ),
+    mCapabilitiesProperty( capabilitiesProperty ),
+    mConnInfo( connInfo ),
+    mLayerProperty( layerProperty )
+    //mProviderKey ("wms"),
+    //mLayerType ( QgsLayerItem::Raster )
+{
+  mUri = createUri();
+  // Populate everything, it costs nothing, all info about layers is collected
+  foreach( QgsWmsLayerProperty layerProperty, mLayerProperty.layer )
+  {
+    // Attention, the name may be empty
+    QgsDebugMsg( QString::number( layerProperty.orderId ) + " " + layerProperty.name + " " + layerProperty.title );
+    QString pathName = layerProperty.name.isEmpty() ? QString::number( layerProperty.orderId ) : layerProperty.name;
+    QgsWMSLayerItem * layer = new QgsWMSLayerItem( this, layerProperty.title, mPath + "/" + pathName, mCapabilitiesProperty, mConnInfo, layerProperty );
+    mChildren.append( layer );
+  }
+
+  if ( mChildren.size() == 0 )
+  {
+    mIcon = QIcon( getThemePixmap( "mIconWmsLayer.png" ) );
+  }
+  mPopulated = true;
+}
+
+QgsWMSLayerItem::~QgsWMSLayerItem()
+{
+}
+
+QString QgsWMSLayerItem::createUri()
+{
+  QString uri;
+  if ( mLayerProperty.name.isEmpty() )
+    return uri; // layer collection
+
+  QString rasterLayerPath = mConnInfo;
+  QString baseName = mLayerProperty.name;
+
+  // Number of styles must match number of layers
+  QStringList layers;
+  layers << mLayerProperty.name;
+  QStringList styles;
+  if ( mLayerProperty.style.size() > 0 )
+  {
+    styles.append( mLayerProperty.style[0].name );
+  }
+  else
+  {
+    styles << ""; // TODO: use loadDefaultStyleFlag
+  }
+
+  QString format;
+  // get first supporte by qt and server
+  QVector<QgsWmsSupportedFormat> formats = QgsWmsProvider::supportedFormats();
+  foreach( QgsWmsSupportedFormat f, formats )
+  {
+    if ( mCapabilitiesProperty.capability.request.getMap.format.indexOf( f.format ) >= 0 )
+    {
+      format = f.format;
+      break;
+    }
+  }
+  QString crs;
+  // get first known if possible
+  QgsCoordinateReferenceSystem testCrs;
+  foreach( QString c, mLayerProperty.crs )
+  {
+    testCrs.createFromOgcWmsCrs( c );
+    if ( testCrs.isValid() )
+    {
+      crs = c;
+      break;
+    }
+  }
+  if ( crs.isEmpty() && mLayerProperty.crs.size() > 0 )
+  {
+    crs = mLayerProperty.crs[0];
+  }
+  uri = rasterLayerPath + "|layers=" + layers.join( "," ) + "|styles=" + styles.join( "," ) + "|format=" + format + "|crs=" + crs;
+
+  return uri;
+}
+
+// ---------------------------------------------------------------------------
+QgsWMSRootItem::QgsWMSRootItem( QgsDataItem* parent, QString name, QString path )
+    : QgsDataCollectionItem( parent, name, path )
+{
+  mIcon = QIcon( getThemePixmap( "mIconWms.png" ) );
+
+  populate();
+}
+
+QgsWMSRootItem::~QgsWMSRootItem()
+{
+}
+
+QVector<QgsDataItem*>QgsWMSRootItem::createChildren()
+{
+  QVector<QgsDataItem*> connections;
+
+  foreach( QString connName, QgsWMSConnection::connectionList() )
+  {
+    QgsDataItem * conn = new QgsWMSConnectionItem( this, connName, mPath + "/" + connName );
+    connections.append( conn );
+  }
+  return connections;
+}
+
+QWidget * QgsWMSRootItem::paramWidget()
+{
+  QgsWMSSourceSelect *select = new QgsWMSSourceSelect( 0, 0, true, true );
+  connect( select, SIGNAL( connectionsChanged() ), this, SLOT( connectionsChanged() ) );
+  return select;
+}
+void QgsWMSRootItem::connectionsChanged()
+{
+  refresh();
+}
+
+// ---------------------------------------------------------------------------
+
+QGISEXTERN QgsDataItem * dataItem( QString thePath, QgsDataItem* parentItem )
+{
+  Q_UNUSED( thePath );
+
+  return new QgsWMSRootItem( parentItem, "WMS", "wms:" );
+}
+
diff --git a/src/providers/wms/qgswmsdataitems.h b/src/providers/wms/qgswmsdataitems.h
@@ -0,0 +1,52 @@
+#ifndef QGSWMSDATAITEMS_H
+#define QGSWMSDATAITEMS_H
+
+#include "qgswmsprovider.h"
+
+class QgsWMSConnectionItem : public QgsDataCollectionItem
+{
+  public:
+    QgsWMSConnectionItem( QgsDataItem* parent, QString name, QString path );
+    ~QgsWMSConnectionItem();
+
+    QVector<QgsDataItem*> createChildren();
+    virtual bool equal( const QgsDataItem *other );
+
+    QgsWmsCapabilitiesProperty mCapabilitiesProperty;
+    QString mConnInfo;
+    QVector<QgsWmsLayerProperty> mLayerProperties;
+};
+
+// WMS Layers may be nested, so that they may be both QgsDataCollectionItem and QgsLayerItem
+// We have to use QgsDataCollectionItem and support layer methods if necessary
+class QgsWMSLayerItem : public QgsLayerItem
+{
+    Q_OBJECT
+  public:
+    QgsWMSLayerItem( QgsDataItem* parent, QString name, QString path,
+                     QgsWmsCapabilitiesProperty capabilitiesProperty, QString connInfo, QgsWmsLayerProperty layerProperties );
+    ~QgsWMSLayerItem();
+
+    QString createUri();
+
+    QgsWmsCapabilitiesProperty mCapabilitiesProperty;
+    QString mConnInfo;
+    QgsWmsLayerProperty mLayerProperty;
+};
+
+class QgsWMSRootItem : public QgsDataCollectionItem
+{
+    Q_OBJECT
+  public:
+    QgsWMSRootItem( QgsDataItem* parent, QString name, QString path );
+    ~QgsWMSRootItem();
+
+    QVector<QgsDataItem*> createChildren();
+
+    virtual QWidget * paramWidget();
+
+  public slots:
+    void connectionsChanged();
+};
+
+#endif // QGSWMSDATAITEMS_H