1 /*************************************************************************** 2 qgsmeshstaticdatasetwidget.cpp 3 ------------------------------------- 4 begin : March 2020 5 copyright : (C) 2020 by Vincent Cloarec 6 email : vcloarec at gmail dot com 7 *************************************************************************** 8 * * 9 * This program is free software; you can redistribute it and/or modify * 10 * it under the terms of the GNU General Public License as published by * 11 * the Free Software Foundation; either version 2 of the License, or * 12 * (at your option) any later version. * 13 * * 14 ***************************************************************************/ 15 16 #include "qgsmeshstaticdatasetwidget.h" 17 18 #include "qgsmeshlayer.h" 19 QgsMeshStaticDatasetWidget(QWidget * parent)20QgsMeshStaticDatasetWidget::QgsMeshStaticDatasetWidget( QWidget *parent ): QWidget( parent ) 21 { 22 setupUi( this ); 23 24 mDatasetScalarModel = new QgsMeshDatasetListModel( this ); 25 mScalarDatasetComboBox->setModel( mDatasetScalarModel ); 26 mDatasetVectorModel = new QgsMeshDatasetListModel( this ); 27 mVectorDatasetComboBox->setModel( mDatasetVectorModel ); 28 } 29 setLayer(QgsMeshLayer * layer)30void QgsMeshStaticDatasetWidget::setLayer( QgsMeshLayer *layer ) 31 { 32 mLayer = layer; 33 } 34 syncToLayer()35void QgsMeshStaticDatasetWidget::syncToLayer() 36 { 37 if ( !mLayer ) 38 return; 39 40 mDatasetScalarModel->setMeshLayer( mLayer ); 41 mDatasetVectorModel->setMeshLayer( mLayer ); 42 setScalarDatasetGroup( mLayer->rendererSettings().activeScalarDatasetGroup() ); 43 setVectorDatasetGroup( mLayer->rendererSettings().activeVectorDatasetGroup() ); 44 } 45 apply()46void QgsMeshStaticDatasetWidget::apply() 47 { 48 if ( !mLayer ) 49 return; 50 51 int scalarIndex; 52 // if only one item, there is no active dataset group. 53 // Set to 0 instead of -1 to avoid none dataset (item 0) when the group is reactivate 54 if ( mScalarDatasetComboBox->count() == 1 ) 55 scalarIndex = 0; 56 else 57 scalarIndex = mScalarDatasetComboBox->currentIndex() - 1; 58 int vectorIndex; 59 // Same as scalar 60 if ( mVectorDatasetComboBox->count() == 1 ) 61 vectorIndex = 0; 62 else 63 vectorIndex = mVectorDatasetComboBox->currentIndex() - 1; 64 65 mLayer->setStaticScalarDatasetIndex( QgsMeshDatasetIndex( mScalarDatasetGroup, scalarIndex ) ); 66 mLayer->setStaticVectorDatasetIndex( QgsMeshDatasetIndex( mVectorDatasetGroup, vectorIndex ) ); 67 } 68 setScalarDatasetGroup(int index)69void QgsMeshStaticDatasetWidget::setScalarDatasetGroup( int index ) 70 { 71 mScalarDatasetGroup = index; 72 mDatasetScalarModel->setDatasetGroup( index ); 73 mScalarDatasetComboBox->setEnabled( mScalarDatasetGroup >= 0 ); 74 if ( mLayer ) 75 { 76 mScalarName->setText( mLayer->datasetGroupMetadata( index ).name() ); 77 setScalarDatasetIndex( mLayer->staticScalarDatasetIndex().dataset() ); 78 } 79 } 80 setVectorDatasetGroup(int index)81void QgsMeshStaticDatasetWidget::setVectorDatasetGroup( int index ) 82 { 83 mVectorDatasetGroup = index; 84 mDatasetVectorModel->setDatasetGroup( index ); 85 mVectorDatasetComboBox->setEnabled( mVectorDatasetGroup >= 0 ); 86 if ( mLayer ) 87 { 88 mVectorName->setText( mLayer->datasetGroupMetadata( index ).name() ); 89 setVectorDatasetIndex( mLayer->staticVectorDatasetIndex().dataset() ); 90 } 91 } 92 setScalarDatasetIndex(int index)93void QgsMeshStaticDatasetWidget::setScalarDatasetIndex( int index ) 94 { 95 if ( index < mLayer->datasetCount( mScalarDatasetGroup ) ) 96 mScalarDatasetComboBox->setCurrentIndex( index + 1 ); 97 else 98 mScalarDatasetComboBox->setCurrentIndex( 0 ); 99 } 100 setVectorDatasetIndex(int index)101void QgsMeshStaticDatasetWidget::setVectorDatasetIndex( int index ) 102 { 103 if ( index < mLayer->datasetCount( mVectorDatasetGroup ) ) 104 mVectorDatasetComboBox->setCurrentIndex( index + 1 ); 105 else 106 mVectorDatasetComboBox->setCurrentIndex( 0 ); 107 } 108 QgsMeshDatasetListModel(QObject * parent)109QgsMeshDatasetListModel::QgsMeshDatasetListModel( QObject *parent ): QAbstractListModel( parent ) 110 {} 111 setMeshLayer(QgsMeshLayer * layer)112void QgsMeshDatasetListModel::setMeshLayer( QgsMeshLayer *layer ) 113 { 114 beginResetModel(); 115 mLayer = layer; 116 endResetModel(); 117 } 118 setDatasetGroup(int group)119void QgsMeshDatasetListModel::setDatasetGroup( int group ) 120 { 121 beginResetModel(); 122 mDatasetGroup = group; 123 endResetModel(); 124 } 125 rowCount(const QModelIndex & parent) const126int QgsMeshDatasetListModel::rowCount( const QModelIndex &parent ) const 127 { 128 Q_UNUSED( parent ) 129 130 if ( mLayer ) 131 return mLayer->datasetCount( mDatasetGroup ) + 1; 132 else 133 return 0; 134 } 135 data(const QModelIndex & index,int role) const136QVariant QgsMeshDatasetListModel::data( const QModelIndex &index, int role ) const 137 { 138 if ( !index.isValid() ) 139 return QVariant(); 140 141 if ( role == Qt::DisplayRole ) 142 { 143 if ( !mLayer || mDatasetGroup < 0 || index.row() == 0 ) 144 return tr( "none" ); 145 146 else if ( index.row() == 1 && mLayer->datasetCount( mDatasetGroup ) == 1 ) 147 { 148 return tr( "Display dataset" ); 149 } 150 else 151 { 152 QgsInterval time = mLayer->datasetRelativeTime( QgsMeshDatasetIndex( mDatasetGroup, index.row() - 1 ) ); 153 return mLayer->formatTime( time.hours() ); 154 } 155 } 156 157 return QVariant(); 158 } 159