1 // Copyright (C) 2010-2014 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // Author : Anthony Geay
21 #include "pqMEDReaderPanel.h"
22 #include "ui_MEDReaderPanel.h"
23 #include "VectBoolWidget.h"
25 #include "vtkProcessModule.h"
26 #include "vtkMultiBlockDataSet.h"
27 #include "vtkInformation.h"
28 #include "vtkIntArray.h"
29 #include "vtkSMDoubleVectorProperty.h"
30 #include "vtkSMIntVectorProperty.h"
31 #include "vtkSMStringVectorProperty.h"
32 #include "vtkSMProxy.h"
33 #include "vtkEventQtSlotConnect.h"
34 #include "vtkPVSILInformation.h"
36 #include "vtkMutableDirectedGraph.h"
37 #include "vtkAdjacentVertexIterator.h"
38 #include "vtkSMPropertyHelper.h"
39 #include "vtkStringArray.h"
40 #include "vtkDataSetAttributes.h"
41 #include "vtkMEDReader.h"
43 #include "pqTreeWidgetItemObject.h"
44 #include "pqSMAdaptor.h"
46 #include "pqPropertyManager.h"
47 #include "pqSILModel.h"
48 #include "pqProxySILModel.h"
49 #include "pqTreeViewSelectionHelper.h"
50 #include "pqTreeWidgetSelectionHelper.h"
52 #include <QHeaderView>
54 static const char ZE_SEP[]="@@][@@";
59 static const PixSingle &GetInstance();
60 QPixmap getPixFromStr(const std::string& st) const;
61 QPixmap getWholeMeshPix() const;
64 static const int NB_OF_DISCR=4;
65 static PixSingle *UNIQUE_PIX_SINGLE;
66 QPixmap _pixmaps[NB_OF_DISCR];
67 std::map<std::string,int> _ze_map;
71 PixSingle *PixSingle::UNIQUE_PIX_SINGLE=0;
73 const PixSingle &PixSingle::GetInstance()
75 if(!UNIQUE_PIX_SINGLE)
76 UNIQUE_PIX_SINGLE=new PixSingle;
77 return *UNIQUE_PIX_SINGLE;
80 PixSingle::PixSingle()
82 _pixmaps[0]=QPixmap(":/ParaViewResources/Icons/pqCellData16.png");
83 _pixmaps[1]=QPixmap(":/ParaViewResources/Icons/pqPointData16.png");
84 _pixmaps[2]=QPixmap(":/ParaViewResources/Icons/pqQuadratureData16.png");
85 _pixmaps[3]=QPixmap(":/ParaViewResources/Icons/pqElnoData16.png");
86 _ze_map[std::string("P0")]=0;
87 _ze_map[std::string("P1")]=1;
88 _ze_map[std::string("GAUSS")]=2;
89 _ze_map[std::string("GSSNE")]=3;
90 _whole_mesh=QPixmap(":/ParaViewResources/Icons/pqCellDataForWholeMesh16.png");
93 QPixmap PixSingle::getPixFromStr(const std::string& st) const
95 std::map<std::string,int>::const_iterator it(_ze_map.find(st));
97 return _pixmaps[(*it).second];
102 QPixmap PixSingle::getWholeMeshPix() const
107 class pqMEDReaderPanel::pqUI: public QObject, public Ui::MEDReaderPanel
110 pqUI(pqMEDReaderPanel *p):QObject(p)
112 this->VTKConnect = vtkSmartPointer<vtkEventQtSlotConnect>::New();
117 vtkSmartPointer<vtkEventQtSlotConnect> VTKConnect;
118 QMap<QTreeWidgetItem*, QString> TreeItemToPropMap;
121 pqMEDReaderPanel::pqMEDReaderPanel(pqProxy *object_proxy, QWidget *p):Superclass(object_proxy,p),_reload_req(false),_optional_widget(0)
126 void pqMEDReaderPanel::initAll()
129 this->UI=new pqUI(this);
130 this->UI->setupUi(this);
131 this->UI->Fields->setHeaderHidden(true);
134 vtkSMProxy *reader(this->referenceProxy()->getProxy());
135 vtkPVSILInformation *info(vtkPVSILInformation::New());
136 reader->GatherInformation(info);
137 vtkGraph *g(info->GetSIL());
138 vtkMutableDirectedGraph *g2(vtkMutableDirectedGraph::SafeDownCast(g));
140 vtkAbstractArray *verticesNames(g2->GetVertexData()->GetAbstractArray("Names",idNames));
141 vtkStringArray *verticesNames2(vtkStringArray::SafeDownCast(verticesNames));
144 for(int i=0;i<verticesNames2->GetNumberOfValues();i++)
146 vtkStdString &st(verticesNames2->GetValue(i));
147 if(st=="FieldsStatusTree")
154 std::cerr << "There is an internal error ! The tree on server side has not the expected look !" << std::endl;
155 vtkAdjacentVertexIterator *it0(vtkAdjacentVertexIterator::New());
156 g2->GetAdjacentVertices(id0,it0);
158 while(it0->HasNext())
160 vtkIdType idToolTipForTS(it0->Next());
161 QString toolTipName0(QString::fromStdString((const char *)verticesNames2->GetValue(idToolTipForTS)));
163 QList<QString> dts,its,tts;
165 vtkAdjacentVertexIterator *itForTS(vtkAdjacentVertexIterator::New());
166 g2->GetAdjacentVertices(idToolTipForTS,itForTS);
167 vtkIdType idForNbTS(itForTS->Next());
168 nbTS=QString::fromStdString((const char *)verticesNames2->GetValue(idForNbTS));
170 int nbTSInt(nbTS.toInt());
171 for(int ii=0;ii<nbTSInt;ii++)
173 dts.push_back(QString::fromStdString((const char *)verticesNames2->GetValue(idForNbTS+3*ii+1)));
174 its.push_back(QString::fromStdString((const char *)verticesNames2->GetValue(idForNbTS+3*ii+2)));
175 tts.push_back(QString::fromStdString((const char *)verticesNames2->GetValue(idForNbTS+3*ii+3)));
178 vtkIdType id1(it0->Next());
180 vtkSMProperty *SMProperty(this->proxy()->GetProperty("FieldsStatus"));
181 SMProperty->ResetToDefault();//this line is very important !
183 QString name0(QString::fromStdString((const char *)verticesNames2->GetValue(id1))); QList<QString> strs0; strs0.append(name0);
184 pqTreeWidgetItemObject *item0(new pqTreeWidgetItemObject(this->UI->Fields,strs0));
185 item0->setData(0,Qt::UserRole,name0);
186 item0->setData(0,Qt::ToolTipRole,toolTipName0);
188 QList<QVariant> modulesAct;
189 for(int i=0;i<nbTS.toInt();i++)
190 modulesAct.push_back(QVariant(true));
191 item0->setProperty("NbOfTS",nbTS);
192 item0->setProperty("DTS",QVariant(dts));
193 item0->setProperty("ITS",QVariant(its));
194 item0->setProperty("TTS",QVariant(tts));
195 item0->setProperty("ChosenTS",QVariant(modulesAct));
197 vtkAdjacentVertexIterator *it1(vtkAdjacentVertexIterator::New());//mesh
198 g2->GetAdjacentVertices(id1,it1);
199 while(it1->HasNext())
201 vtkIdType id2(it1->Next());
202 QString name1(QString::fromStdString((const char *)verticesNames2->GetValue(id2))); QList<QString> strs1; strs1.append(name1);
203 QString toolTipName1(name1);
204 pqTreeWidgetItemObject *item1(new pqTreeWidgetItemObject(item0,strs1));
205 item1->setData(0,Qt::UserRole,name1);
206 item1->setData(0,Qt::ToolTipRole,toolTipName1);
207 vtkAdjacentVertexIterator *it2(vtkAdjacentVertexIterator::New());//common support
208 g2->GetAdjacentVertices(id2,it2);
209 while(it2->HasNext())
211 vtkIdType id3(it2->Next());
212 QString name2(QString::fromStdString((const char *)verticesNames2->GetValue(id3))); QList<QString> strs2; strs2.append(name2);
213 QString toolTipName2(name2);
214 pqTreeWidgetItemObject *item2(new pqTreeWidgetItemObject(item1,strs2));
215 item2->setData(0,Qt::UserRole,name2);
216 item2->setData(0,Qt::ToolTipRole,toolTipName2);
217 item2->setData(0,Qt::CheckStateRole,0);
218 vtkAdjacentVertexIterator *it3(vtkAdjacentVertexIterator::New());//fields !
219 g2->GetAdjacentVertices(id3,it3);
220 while(it3->HasNext())
222 vtkIdType id4(it3->Next());
223 std::string name3CppFull((const char *)verticesNames2->GetValue(id4));
224 std::size_t pos(name3CppFull.find_first_of(ZE_SEP));
225 std::string name3Only(name3CppFull.substr(0,pos)); std::string spatialDiscr(name3CppFull.substr(pos+sizeof(ZE_SEP)-1));
226 QString name3(QString::fromStdString(name3Only)); QList<QString> strs3; strs3.append(name3);
227 QString toolTipName3(name3+QString(" (")+spatialDiscr.c_str()+QString(")"));
229 vtkAdjacentVertexIterator *it4(vtkAdjacentVertexIterator::New());// is it a special field ? A field mesh ?
230 g2->GetAdjacentVertices(id4,it4);
231 bool isSpecial(it4->HasNext());
234 pqTreeWidgetItemObject *item3(new pqTreeWidgetItemObject(item2,strs3));
235 _all_lev4.push_back(item3);
236 item3->setData(0,Qt::UserRole,name3);
237 item3->setData(0,Qt::CheckStateRole,0);
240 QFont font; font.setItalic(true); font.setUnderline(true);
241 item3->setData(0,Qt::FontRole,QVariant(font));
242 item3->setData(0,Qt::ToolTipRole,QString("Whole \"%1\" mesh").arg(name3));
243 item3->setData(0,Qt::DecorationRole,PixSingle::GetInstance().getWholeMeshPix());
247 item3->setData(0,Qt::ToolTipRole,toolTipName3);
248 item3->setData(0,Qt::DecorationRole,PixSingle::GetInstance().getPixFromStr(spatialDiscr));
250 this->propertyManager()->registerLink(item3,"checked",SIGNAL(checkedStateChanged(bool)),this->proxy(),SMProperty,ll);
251 _leaves.insert(std::pair<pqTreeWidgetItemObject *,int>(item3,ll));
252 connect(item2,SIGNAL(checkedStateChanged(bool)),item3,SLOT(setChecked(bool)));
253 connect(item3,SIGNAL(checkedStateChanged(bool)),this,SLOT(aLev4HasBeenFired()));
258 item2->setChecked(true);
266 this->UI->Fields->header()->setStretchLastSection(true);
267 this->UI->Fields->expandAll();
269 this->UI->stdMode->setChecked(true);
270 this->UI->VTKConnect->Connect(this->proxy(),vtkCommand::UpdateInformationEvent, this, SLOT(updateSIL()));
272 this->UI->Reload->setProperty("NbOfReloadDynProp",QVariant(1));
273 vtkSMProperty *SMProperty(this->proxy()->GetProperty("ReloadReq"));
274 connect(this->UI->Reload,SIGNAL(pressed()),this,SLOT(reloadFired()));
275 this->propertyManager()->registerLink(this->UI->Reload,"NbOfReloadDynProp",SIGNAL(pressed()),this->proxy(),SMProperty);
277 vtkSMProperty *SMProperty0(this->proxy()->GetProperty("GenerateVectors"));
278 this->propertyManager()->registerLink(this->UI->GenerateVects,"checked",SIGNAL(stateChanged(int)),this->proxy(),SMProperty0);
280 vtkSMProperty *SMProperty2(this->proxy()->GetProperty("TimeOrModal"));
281 SMProperty2->ResetToDefault();//this line is very important !
282 this->propertyManager()->registerLink(this->UI->modeMode,"checked",SIGNAL(toggled(bool)),this->proxy(),SMProperty2);
284 delete _optional_widget;
285 _optional_widget=new VectBoolWidget(this->UI->timeStepsInspector,getMaxNumberOfTS());
286 _optional_widget->hide();
287 this->UI->timeStepsInspector->setMinimumSize(QSize(0,0));
288 connect(this->UI->modeMode,SIGNAL(toggled(bool)),this,SLOT(vectOfBoolWidgetRequested(bool)));
289 vtkSMProperty *SMProperty3(this->proxy()->GetProperty("TimesFlagsStatus"));
290 SMProperty3->ResetToDefault();
291 const QVector<VectBoolItem *>& items(_optional_widget->getItems());
293 foreach(VectBoolItem *item,items)
295 this->propertyManager()->registerLink(item,"activated",SIGNAL(changed()),this->proxy(),SMProperty3,itt++);
299 pqMEDReaderPanel::~pqMEDReaderPanel()
301 delete _optional_widget;
304 void pqMEDReaderPanel::linkServerManagerProperties()
306 this->Superclass::linkServerManagerProperties();
309 void pqMEDReaderPanel::updateSIL()
314 this->UI->geometryGroupBox->hide();
316 foreach(QObject *child,children())
318 QLayout *layout(qobject_cast<QLayout *>(child));
326 void pqMEDReaderPanel::aLev4HasBeenFired()
328 pqTreeWidgetItemObject *zeItem(qobject_cast<pqTreeWidgetItemObject *>(sender()));
331 pqTreeWidgetItemObject *father(dynamic_cast<pqTreeWidgetItemObject *>(zeItem->QTreeWidgetItem::parent()));
332 QTreeWidgetItem *godFather(father->QTreeWidgetItem::parent()->parent());
335 if(zeItem->isChecked())
337 bool isActivatedTSChanged(false);
338 // This part garantees that all leaves having not the same father than zeItem are desactivated
339 foreach(pqTreeWidgetItemObject* elt,this->_all_lev4)
341 QTreeWidgetItem *testFath(elt->QTreeWidgetItem::parent());
345 elt->setChecked(false);
346 if(godFather!=testFath->parent()->parent())
347 isActivatedTSChanged=true;
350 //If all leaves are checked the father is check too
351 bool allItemsAreChked(true);
352 for(int i=0;i<father->childCount() && allItemsAreChked;i++)
354 pqTreeWidgetItemObject *elt(dynamic_cast<pqTreeWidgetItemObject *>(father->child(i)));
355 if(elt && !elt->isChecked())
356 allItemsAreChked=false;
358 if(allItemsAreChked && !father->isChecked())
359 father->setChecked(true);
360 // the user by clicking to a new entry has changed of TimeStepSeries -> notify it to thee time step selector widget
361 if(isActivatedTSChanged)
363 QStringList its,dts,tts;
364 getCurrentTS(its,dts,tts);
365 _optional_widget->setItems(its,dts,tts);
370 // if all are unchecked - check it again
371 bool allItemsAreUnChked(true);
372 foreach(pqTreeWidgetItemObject* elt,this->_all_lev4)
374 if(elt && elt->isChecked())
375 allItemsAreUnChked=false;
377 if(allItemsAreUnChked)
378 zeItem->setChecked(true);// OK zeItem was required to be unchecked but as it is the last one. Recheck it !
380 {// if all items are uncheked inside a same parent - uncheck the parent
381 allItemsAreUnChked=true;
382 for(int i=0;i<father->childCount() && allItemsAreUnChked;i++)
384 pqTreeWidgetItemObject *elt(dynamic_cast<pqTreeWidgetItemObject *>(father->child(i)));
385 if(elt && elt->isChecked())
386 allItemsAreUnChked=false;
388 if(allItemsAreUnChked && father->isChecked())
389 father->setChecked(false);
394 void pqMEDReaderPanel::reloadFired()
398 this->UI->Reload->setProperty("NbOfReloadDynProp",v);
400 for(std::set<std::pair<pqTreeWidgetItemObject *,int> >::const_iterator it=_leaves.begin();it!=_leaves.end();it++)
401 ((*it).first)->disconnect(SIGNAL(checkedStateChanged(bool)));
403 vtkSMProperty *SMProperty(this->proxy()->GetProperty("FieldsStatus"));
404 for(std::set<std::pair<pqTreeWidgetItemObject *,int> >::const_iterator it=_leaves.begin();it!=_leaves.end();it++)
405 this->propertyManager()->unregisterLink((*it).first,"checked",SIGNAL(checkedStateChanged(bool)),this->proxy(),SMProperty,(*it).second);
406 this->propertyManager()->propertyChanged();
407 vtkSMProperty *SMProperty3(this->proxy()->GetProperty("TimeOrModal"));
408 this->propertyManager()->unregisterLink(this->UI->modeMode,"checked",SIGNAL(toggled(bool)),this->proxy(),SMProperty3);
411 void pqMEDReaderPanel::vectOfBoolWidgetRequested(bool isMode)
415 this->UI->timeStepsInspector->setMinimumSize(QSize(200,250));
416 _optional_widget->show();
417 QStringList its,dts,tts;
418 getCurrentTS(its,dts,tts);
419 _optional_widget->setItems(its,dts,tts);
423 _optional_widget->hide();
424 this->UI->timeStepsInspector->setMinimumSize(QSize(0,0));
428 void pqMEDReaderPanel::getCurrentTS(QStringList& its, QStringList& dts, QStringList& tts) const
431 for(std::vector<pqTreeWidgetItemObject *>::const_iterator it=_all_lev4.begin();it!=_all_lev4.end();it++)
433 if((*it)->property("checked").toInt())
435 QTreeWidgetItem *obj((*it)->QTreeWidgetItem::parent()->QTreeWidgetItem::parent()->QTreeWidgetItem::parent());
436 pqTreeWidgetItemObject *objC(dynamic_cast<pqTreeWidgetItemObject *>(obj));
437 its=objC->property("ITS").toStringList();
438 dts=objC->property("DTS").toStringList();
439 tts=objC->property("TTS").toStringList();
443 std::cerr << "pqMEDReaderPanel::getCurrentTS : internal error ! Something is going wrong !" << std::endl;
446 int pqMEDReaderPanel::getMaxNumberOfTS() const
449 for(std::vector<pqTreeWidgetItemObject *>::const_iterator it=_all_lev4.begin();it!=_all_lev4.end();it++)
451 if((*it)->property("checked").toInt())
453 QTreeWidgetItem *obj((*it)->QTreeWidgetItem::parent()->QTreeWidgetItem::parent()->QTreeWidgetItem::parent());
454 pqTreeWidgetItemObject *objC(dynamic_cast<pqTreeWidgetItemObject *>(obj));
455 ret=std::max(ret,objC->property("NbOfTS").toInt());