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)20 QgsMeshStaticDatasetWidget::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)30 void QgsMeshStaticDatasetWidget::setLayer( QgsMeshLayer *layer )
31 {
32   mLayer = layer;
33 }
34 
syncToLayer()35 void 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()46 void 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)69 void 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)81 void 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)93 void 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)101 void 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)109 QgsMeshDatasetListModel::QgsMeshDatasetListModel( QObject *parent ): QAbstractListModel( parent )
110 {}
111 
setMeshLayer(QgsMeshLayer * layer)112 void QgsMeshDatasetListModel::setMeshLayer( QgsMeshLayer *layer )
113 {
114   beginResetModel();
115   mLayer = layer;
116   endResetModel();
117 }
118 
setDatasetGroup(int group)119 void QgsMeshDatasetListModel::setDatasetGroup( int group )
120 {
121   beginResetModel();
122   mDatasetGroup = group;
123   endResetModel();
124 }
125 
rowCount(const QModelIndex & parent) const126 int 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) const136 QVariant 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