src/Plugins/MEDReader/ParaViewPlugin/pqMEDReaderPanel.cxx

   1 // Copyright (C) 2010-2014  CEA/DEN, EDF R&D
   2 //
   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.
   7 //
   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.
  12 //
  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
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 // Author : Anthony Geay
  20
  21 #include "pqMEDReaderPanel.h"
  22 #include "ui_MEDReaderPanel.h"
  23 #include "VectBoolWidget.h"
  24
  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"
  35 #include "vtkGraph.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"
  42
  43 #include "pqTreeWidgetItemObject.h"
  44 #include "pqSMAdaptor.h"
  45 #include "pqProxy.h"
  46 #include "pqPropertyManager.h"
  47 #include "pqSILModel.h"
  48 #include "pqProxySILModel.h"
  49 #include "pqTreeViewSelectionHelper.h"
  50 #include "pqTreeWidgetSelectionHelper.h"
  51
  52 #include <QHeaderView>
  53
  54 static const char ZE_SEP[]="@@][@@";
  55
  56 class PixSingle
  57 {
  58 public:
  59   static const PixSingle &GetInstance();
  60   QPixmap getPixFromStr(const std::string& st) const;
  61   QPixmap getWholeMeshPix() const;
  62   PixSingle();
  63 private:
  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;
  68   QPixmap _whole_mesh;
  69 };
  70
  71 PixSingle *PixSingle::UNIQUE_PIX_SINGLE=0;
  72
  73 const PixSingle &PixSingle::GetInstance()
  74 {
  75   if(!UNIQUE_PIX_SINGLE)
  76     UNIQUE_PIX_SINGLE=new PixSingle;
  77   return *UNIQUE_PIX_SINGLE;
  78 }
  79
  80 PixSingle::PixSingle()
  81 {
  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");
  91 }
  92
  93 QPixmap PixSingle::getPixFromStr(const std::string& st) const
  94 {
  95   std::map<std::string,int>::const_iterator it(_ze_map.find(st));
  96   if(it!=_ze_map.end())
  97     return _pixmaps[(*it).second];
  98   else
  99     return QPixmap();
 100 }
 101
 102 QPixmap PixSingle::getWholeMeshPix() const
 103 {
 104   return _whole_mesh;
 105 }
 106
 107 class pqMEDReaderPanel::pqUI: public QObject, public Ui::MEDReaderPanel
 108 {
 109 public:
 110   pqUI(pqMEDReaderPanel *p):QObject(p)
 111   {
 112     this->VTKConnect = vtkSmartPointer<vtkEventQtSlotConnect>::New();
 113   }
 114
 115   ~pqUI() { }
 116
 117   vtkSmartPointer<vtkEventQtSlotConnect> VTKConnect;
 118   QMap<QTreeWidgetItem*, QString> TreeItemToPropMap;
 119 };
 120
 121 pqMEDReaderPanel::pqMEDReaderPanel(pqProxy *object_proxy, QWidget *p):Superclass(object_proxy,p),_reload_req(false),_optional_widget(0)
 122 {
 123   initAll();
 124 }
 125
 126 void pqMEDReaderPanel::initAll()
 127 {
 128   _all_lev4.clear();
 129   this->UI=new pqUI(this);
 130   this->UI->setupUi(this);
 131   this->UI->Fields->setHeaderHidden(true);
 132   this->updateSIL();
 133   ////////////////////
 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));
 139   int idNames(0);
 140   vtkAbstractArray *verticesNames(g2->GetVertexData()->GetAbstractArray("Names",idNames));
 141   vtkStringArray *verticesNames2(vtkStringArray::SafeDownCast(verticesNames));
 142   vtkIdType id0;
 143   bool found(false);
 144   for(int i=0;i<verticesNames2->GetNumberOfValues();i++)
 145     {
 146       vtkStdString &st(verticesNames2->GetValue(i));
 147       if(st=="FieldsStatusTree")
 148         {
 149           id0=i;
 150           found=true;
 151         }
 152     }
 153   if(!found)
 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);
 157   int kk(0),ll(0);
 158   while(it0->HasNext())
 159     {
 160       vtkIdType idToolTipForTS(it0->Next());
 161       QString toolTipName0(QString::fromStdString((const char *)verticesNames2->GetValue(idToolTipForTS)));
 162       QString nbTS;
 163       QList<QString> dts,its,tts;
 164       {
 165         vtkAdjacentVertexIterator *itForTS(vtkAdjacentVertexIterator::New());
 166         g2->GetAdjacentVertices(idToolTipForTS,itForTS);
 167         vtkIdType idForNbTS(itForTS->Next());
 168         nbTS=QString::fromStdString((const char *)verticesNames2->GetValue(idForNbTS));
 169         itForTS->Delete();
 170         int nbTSInt(nbTS.toInt());
 171         for(int ii=0;ii<nbTSInt;ii++)
 172           {
 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)));
 176           }
 177       }
 178       vtkIdType id1(it0->Next());
 179       //
 180       vtkSMProperty *SMProperty(this->proxy()->GetProperty("FieldsStatus"));
 181       SMProperty->ResetToDefault();//this line is very important !
 182       //
 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);
 187       //
 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));
 196       //
 197       vtkAdjacentVertexIterator *it1(vtkAdjacentVertexIterator::New());//mesh
 198       g2->GetAdjacentVertices(id1,it1);
 199       while(it1->HasNext())
 200         {
 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())
 210             {
 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())
 221                 {
 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(")"));
 228                   //
 229                   vtkAdjacentVertexIterator *it4(vtkAdjacentVertexIterator::New());// is it a special field ? A field mesh ?
 230                   g2->GetAdjacentVertices(id4,it4);
 231                   bool isSpecial(it4->HasNext());
 232                   it4->Delete();
 233                   //
 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);
 238                   if(isSpecial)
 239                     {
 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());
 244                     }
 245                   else
 246                     {
 247                       item3->setData(0,Qt::ToolTipRole,toolTipName3);
 248                       item3->setData(0,Qt::DecorationRole,PixSingle::GetInstance().getPixFromStr(spatialDiscr));
 249                     }
 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()));
 254                   ll++;
 255                 }
 256               it3->Delete();
 257               if(kk==0)
 258                 item2->setChecked(true);
 259               kk++;
 260             }
 261           it2->Delete();
 262         }
 263       it1->Delete();
 264     }
 265   it0->Delete();
 266   this->UI->Fields->header()->setStretchLastSection(true);
 267   this->UI->Fields->expandAll();
 268   info->Delete();
 269   this->UI->stdMode->setChecked(true);
 270   this->UI->VTKConnect->Connect(this->proxy(),vtkCommand::UpdateInformationEvent, this, SLOT(updateSIL()));
 271   ///
 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);
 276   ///
 277   vtkSMProperty *SMProperty0(this->proxy()->GetProperty("GenerateVectors"));
 278   this->propertyManager()->registerLink(this->UI->GenerateVects,"checked",SIGNAL(stateChanged(int)),this->proxy(),SMProperty0);
 279   ///
 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);
 283   ///
 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());
 292   int itt(0);
 293   foreach(VectBoolItem *item,items)
 294     {
 295       this->propertyManager()->registerLink(item,"activated",SIGNAL(changed()),this->proxy(),SMProperty3,itt++);
 296     }
 297 }
 298
 299 pqMEDReaderPanel::~pqMEDReaderPanel()
 300 {
 301   delete _optional_widget;
 302 }
 303
 304 void pqMEDReaderPanel::linkServerManagerProperties()
 305 {
 306   this->Superclass::linkServerManagerProperties();
 307 }
 308
 309 void pqMEDReaderPanel::updateSIL()
 310 {
 311   if(_reload_req)
 312     {
 313       _reload_req=false;
 314       this->UI->geometryGroupBox->hide();
 315       delete this->UI;
 316       foreach(QObject *child,children())
 317         {
 318           QLayout *layout(qobject_cast<QLayout *>(child));
 319           if(layout)
 320             delete layout;
 321         }
 322       initAll();
 323     }
 324 }
 325
 326 void pqMEDReaderPanel::aLev4HasBeenFired()
 327 {
 328   pqTreeWidgetItemObject *zeItem(qobject_cast<pqTreeWidgetItemObject *>(sender()));
 329   if(!zeItem)
 330     return;
 331   pqTreeWidgetItemObject *father(dynamic_cast<pqTreeWidgetItemObject *>(zeItem->QTreeWidgetItem::parent()));
 332   QTreeWidgetItem *godFather(father->QTreeWidgetItem::parent()->parent());
 333   if(!father)
 334     return ;
 335   if(zeItem->isChecked())
 336     {
 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)
 340         {
 341           QTreeWidgetItem *testFath(elt->QTreeWidgetItem::parent());
 342           if(testFath!=father)
 343             if(elt->isChecked())
 344               {
 345                 elt->setChecked(false);
 346                 if(godFather!=testFath->parent()->parent())
 347                   isActivatedTSChanged=true;
 348               }
 349         }
 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++)
 353         {
 354           pqTreeWidgetItemObject *elt(dynamic_cast<pqTreeWidgetItemObject *>(father->child(i)));
 355           if(elt && !elt->isChecked())
 356             allItemsAreChked=false;
 357         }
 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)
 362         {
 363           QStringList its,dts,tts;
 364           getCurrentTS(its,dts,tts);
 365           _optional_widget->setItems(its,dts,tts);
 366         }
 367     }
 368   else
 369     {
 370       // if all are unchecked - check it again
 371       bool allItemsAreUnChked(true);
 372       foreach(pqTreeWidgetItemObject* elt,this->_all_lev4)
 373         {
 374           if(elt && elt->isChecked())
 375             allItemsAreUnChked=false;
 376         }
 377       if(allItemsAreUnChked)
 378         zeItem->setChecked(true);// OK zeItem was required to be unchecked but as it is the last one. Recheck it !
 379       else
 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++)
 383             {
 384               pqTreeWidgetItemObject *elt(dynamic_cast<pqTreeWidgetItemObject *>(father->child(i)));
 385               if(elt && elt->isChecked())
 386             allItemsAreUnChked=false;
 387             }
 388           if(allItemsAreUnChked && father->isChecked())
 389             father->setChecked(false);
 390         }
 391     }
 392 }
 393
 394 void pqMEDReaderPanel::reloadFired()
 395 {
 396   static int iii(1);
 397   QVariant v(iii++);
 398   this->UI->Reload->setProperty("NbOfReloadDynProp",v);
 399   _reload_req=true;
 400   for(std::set<std::pair<pqTreeWidgetItemObject *,int> >::const_iterator it=_leaves.begin();it!=_leaves.end();it++)
 401     ((*it).first)->disconnect(SIGNAL(checkedStateChanged(bool)));
 402   //
 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);
 409 }
 410
 411 void pqMEDReaderPanel::vectOfBoolWidgetRequested(bool isMode)
 412 {
 413   if(isMode)
 414     {
 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);
 420     }
 421   else
 422     {
 423       _optional_widget->hide();
 424       this->UI->timeStepsInspector->setMinimumSize(QSize(0,0));
 425     }
 426 }
 427
 428 void pqMEDReaderPanel::getCurrentTS(QStringList& its, QStringList& dts, QStringList& tts) const
 429 {
 430
 431   for(std::vector<pqTreeWidgetItemObject *>::const_iterator it=_all_lev4.begin();it!=_all_lev4.end();it++)
 432     {
 433       if((*it)->property("checked").toInt())
 434         {
 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();
 440           return;
 441         }
 442     }
 443   std::cerr << "pqMEDReaderPanel::getCurrentTS : internal error ! Something is going wrong !" << std::endl;
 444 }
 445
 446 int pqMEDReaderPanel::getMaxNumberOfTS() const
 447 {
 448   int ret(0);
 449   for(std::vector<pqTreeWidgetItemObject *>::const_iterator it=_all_lev4.begin();it!=_all_lev4.end();it++)
 450     {
 451       if((*it)->property("checked").toInt())
 452         {
 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());
 456         }
 457     }
 458   return ret;
 459 }