src/HYDROGUI/HYDROGUI_ImportLandCoverMapOp.cxx

   1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "HYDROGUI_ImportLandCoverMapOp.h"
  24
  25 #include "HYDROGUI_DataModel.h"
  26 #include "HYDROGUI_Module.h"
  27 #include "HYDROGUI_UpdateFlags.h"
  28 #include "HYDROGUI_Tool.h"
  29 #include "HYDROGUI_ImportLandCoverMapDlg.h"
  30 #include "HYDROGUI_Shape.h"
  31 #include <HYDROGUI_ZLayers.h>
  32 #include <HYDROGUI_DataObject.h>
  33 #include <HYDROData_Iterator.h>
  34 #include <HYDROData_ShapeFile.h>
  35 #include <HYDROData_Profile.h>
  36 #include <HYDROData_LandCoverMap.h>
  37 #include <HYDROData_StricklerTable.h>
  38
  39 #include <SUIT_Desktop.h>
  40 #include <SUIT_FileDlg.h>
  41 #include <SUIT_MessageBox.h>
  42 #include <SUIT_Desktop.h>
  43 #include <SUIT_ViewManager.h>
  44
  45 #include <OCCViewer_ViewManager.h>
  46 #include <OCCViewer_ViewModel.h>
  47
  48 #include <SalomeApp_Study.h>
  49 #include <LightApp_Application.h>
  50 #include <LightApp_DataOwner.h>
  51 #include <LightApp_Displayer.h>
  52 #include <LightApp_SelectionMgr.h>
  53
  54 #include <BRep_Builder.hxx>
  55 #include <TopoDS.hxx>
  56 #include <TopoDS_Shape.hxx>
  57 #include <TopoDS_Wire.hxx>
  58
  59 #include <QSet>
  60 #include <QFile>
  61 #include <QFileInfo>
  62
  63 #include <QApplication>
  64
  65 #define MAX_LCM_NAME_INDEX 1000
  66
  67 HYDROGUI_ImportLandCoverMapOp::HYDROGUI_ImportLandCoverMapOp( HYDROGUI_Module* theModule )
  68 : HYDROGUI_Operation( theModule )
  69 {
  70   setName( tr( "IMPORT_LANDCOVER_MAP" ) );
  71 }
  72
  73 HYDROGUI_ImportLandCoverMapOp::~HYDROGUI_ImportLandCoverMapOp()
  74 {
  75   erasePreview();
  76   myImporter.Free();
  77   myAttrValue.clear();
  78 }
  79
  80 void HYDROGUI_ImportLandCoverMapOp::startOperation()
  81 {
  82   HYDROGUI_Operation::startOperation();
  83
  84   if ( !getPreviewManager() )
  85     setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( module()->getApp()->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
  86
  87   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
  88   if ( !aPanel )
  89     return;
  90
  91   aPanel->reset();
  92 }
  93
  94 void HYDROGUI_ImportLandCoverMapOp::onFileSelected()
  95 {
  96   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
  97   if ( !aPanel )
  98     return;
  99
 100   QString anObjectName = aPanel->getObjectName().simplified();
 101   anObjectName = aPanel->getFileName();
 102   if ( !anObjectName.isEmpty() )
 103       anObjectName = QFileInfo( anObjectName ).baseName();
 104
 105   if ( anObjectName.isEmpty() )
 106     anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_LANDCOVERMAP_NAME" ) );
 107   aPanel->setObjectName( anObjectName );
 108
 109   myFileName = aPanel->getFileName();
 110   if ( myFileName.isEmpty() )
 111   {
 112     abort();
 113     return;
 114   }
 115
 116   QString anExt = myFileName.split('.', QString::SkipEmptyParts).back();
 117
 118   if (anExt == "shp")
 119   {
 120     startDocOperation();
 121     QApplication::setOverrideCursor(Qt::WaitCursor);
 122
 123     QStringList aPolygonsList;
 124     myPolygonFaces.Clear();
 125     myImporter.Free();
 126
 127     SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
 128     if ( !aStudy )
 129       return;
 130
 131     erasePreview();
 132
 133     Handle(AIS_InteractiveContext) aCtx = NULL;
 134     int aShapeTypeOfFile = -1;
 135
 136     //Import polygons from SHP file as faces
 137     //This faces should be added to the new LCM object
 138
 139     int aStat = myImporter.ImportPolygons(myFileName, aPolygonsList, myPolygonFaces, aShapeTypeOfFile);
 140     if (aStat == 1)
 141     {
 142       aPanel->setPolygonNames(aPolygonsList);
 143
 144       LightApp_Application* anApp = module()->getApp();
 145       if ( !getPreviewManager() )
 146         setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
 147       OCCViewer_ViewManager* aViewManager = getPreviewManager();
 148
 149       if ( aViewManager )
 150       {
 151         if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
 152         {
 153           aCtx = aViewer->getAISContext();
 154           connect( aViewer, SIGNAL( selectionChanged() ), this, SLOT( onViewerSelectionChanged() ) );
 155         }
 156       }
 157
 158       for ( int i = 1; i <= myPolygonFaces.Length(); i++ )
 159       {
 160         TopoDS_Face aFace = TopoDS::Face(myPolygonFaces.Value( i ));
 161         if ( aViewManager && !aCtx.IsNull() )
 162         {
 163           HYDROGUI_Shape* aShape = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
 164           //Preview color is grey
 165           aShape->setFillingColor(QColor(50,50,50), false, false);
 166           aShape->setBorderColor(QColor(50,50,50), false, false);
 167           if( !aFace.IsNull() )
 168             aShape->setShape( aFace);
 169
 170           myPolygonName2PrsShape.insert( tr( "DEF_POLYGON_NAME" ) + "_" + QString::number(i), aShape);
 171         }
 172       }
 173
 174       if ( !aCtx.IsNull() )
 175       {
 176         UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
 177         aCtx->UpdateCurrentViewer();
 178       }
 179       //
 180       QApplication::restoreOverrideCursor();
 181       aPanel->setFirstPageState(true);
 182       //
 183       //Try to load DBF-database...
 184       QString aDBFFileName;
 185       aDBFFileName = myFileName.simplified().replace( myFileName.simplified().size() - 4, 4, ".dbf");
 186       bool DBF_Stat = myImporter.DBF_OpenDBF(aDBFFileName);
 187       // TODO:
 188       // add MSG BOX if stat is bad
 189       if (DBF_Stat)
 190       {
 191         myFieldList = myImporter.DBF_GetFieldList();
 192         aPanel->setAttributeNames(myFieldList);
 193       }
 194       aPanel->setDbfState(DBF_Stat);
 195     }
 196     else
 197     {
 198       erasePreview();
 199       aPanel->setPolygonNames(QStringList());
 200       aPanel->setObjectName("");
 201       QApplication::restoreOverrideCursor();
 202       QString aMess = tr( "CANNT_IMPORT_LCM" ) + "\n";
 203       if (aStat == -1)
 204         aMess += tr( "CANNT_OPEN_SHP" );
 205       else if (aStat == -2)
 206         aMess += tr( "CANNT_OPEN_SHX" );
 207       else
 208         aMess += tr ("SHP_TYPEFORMAT_MESS") + myImporter.GetShapeTypeName(aShapeTypeOfFile);
 209       SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_LANDCOVER_MAP" ), aMess);
 210       commitDocOperation();
 211       myImporter.Free();
 212       //abort();
 213       aPanel->setFirstPageState(false);
 214     }
 215   }
 216 }
 217
 218 HYDROGUI_InputPanel* HYDROGUI_ImportLandCoverMapOp::createInputPanel() const
 219 {
 220   HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportLandCoverMapDlg( module(), getName() );
 221
 222   connect( aPanel, SIGNAL( FileSelected( const QString& ) ), SLOT( onFileSelected() ) );
 223   connect( aPanel, SIGNAL( selectionChanged( const QStringList& ) ), this, SLOT( onSelectionChanged( const QStringList& ) ) );
 224   connect( aPanel, SIGNAL( Next( const int ) ), SLOT( onNext( const int ) ) );
 225
 226   return aPanel;
 227 }
 228
 229 bool HYDROGUI_ImportLandCoverMapOp::processApply( int& theUpdateFlags,
 230                                                   QString& theErrorMsg,
 231                                                   QStringList& theBrowseObjectsEntries )
 232 {
 233
 234   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 235   if ( !aPanel )
 236     return false;
 237
 238   myLCM = Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
 239   HYDROData_MapOfFaceToStricklerType aMapFace2ST;
 240
 241   QStringList aAttrV_T;
 242   QStringList aSTL;
 243   aPanel->getValAttr2StricklerTypeCorr(aAttrV_T, aSTL);
 244
 245   QVector<int> aSelIndices = aPanel->getSelectedPolygonIndices();
 246   foreach ( int Ind, aSelIndices )
 247   {
 248     TopoDS_Shape aShape = myPolygonFaces(Ind + 1);
 249     if ( aShape.IsNull() )
 250       continue;
 251     QString aST = "";
 252     if (aPanel->getAttrCheckBoxState())
 253     {
 254       HYDROData_ShapeFile::DBF_AttrValue aDataVal = myAttrValue[Ind];
 255       int aStricklerTypesInd = aAttrV_T.indexOf(QString(aDataVal.myStrVal));
 256       aST = aSTL.at(aStricklerTypesInd);
 257     }
 258     // else => ST is empty
 259     aMapFace2ST.Add( TopoDS::Face( aShape ), aST );
 260   }
 261
 262   //
 263   myLCM->StoreLandCovers(aMapFace2ST);
 264
 265   QString anObjName;
 266   if ( !aPanel->getFileName().isEmpty() )
 267     anObjName = aPanel->getObjectName();
 268
 269   Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
 270
 271   //check if name of LCM is already exists
 272   QSet<QString> aNameList;
 273   if (aDoc)
 274   {
 275     HYDROData_Iterator anIt( aDoc, KIND_LAND_COVER_MAP );
 276     for( ; anIt.More(); anIt.Next() )
 277       aNameList << anIt.Current()->GetName();
 278   }
 279
 280   QString aNewName = anObjName;
 281   for ( int i = 1; i < MAX_LCM_NAME_INDEX && aNameList.contains(aNewName); i++)
 282     aNewName = anObjName + "_" + QString::number(i);
 283
 284   if( !myLCM.IsNull() )
 285   {
 286     myLCM->SetName( aNewName );
 287     myLCM->SetColor( Qt::gray );
 288     myLCM->Show();
 289     module()->setIsToUpdate( myLCM );
 290   }
 291
 292   module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
 293
 294   erasePreview();
 295   myImporter.Free();
 296   myAttrValue.clear();
 297
 298   return true;
 299 }
 300
 301 void HYDROGUI_ImportLandCoverMapOp::onSelectionChanged( const QStringList& theSelectedNames )
 302 {
 303   Handle(AIS_InteractiveContext) aCtx = NULL;
 304
 305   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 306   if ( aViewManager ) {
 307     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
 308       aCtx = aViewer->getAISContext();
 309     }
 310   }
 311
 312   if ( !aCtx.IsNull() ) {
 313     foreach ( QString aName, myPolygonName2PrsShape.keys() ) {
 314       Handle(AIS_InteractiveObject) anObject =
 315         myPolygonName2PrsShape.value(aName)->getAISObject();
 316
 317       bool isSelected = theSelectedNames.contains( aName );
 318       if ( ( isSelected && !aCtx->IsSelected( anObject) ) ||
 319            ( !isSelected && aCtx->IsSelected( anObject) ) )
 320         aCtx->AddOrRemoveSelected( anObject, Standard_False );
 321     }
 322     aCtx->UpdateCurrentViewer();
 323   }
 324 }
 325
 326
 327 void HYDROGUI_ImportLandCoverMapOp::onViewerSelectionChanged()
 328 {
 329   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 330   if ( !aPanel )
 331     return;
 332
 333   int aCurIndex = -1;
 334   aCurIndex = aPanel->GetCurrentWizardIndex();
 335
 336   if (aCurIndex != 0)
 337     return;
 338
 339   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 340   Handle(AIS_InteractiveContext) aCtx = NULL;
 341   if ( aViewManager ) {
 342     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
 343       aCtx = aViewer->getAISContext();
 344     }
 345   }
 346
 347   if ( !aCtx.IsNull() )
 348   {
 349     QStringList aSelectedNames;
 350     foreach ( QString aName, myPolygonName2PrsShape.keys() ) {
 351       bool isSelected = aCtx->IsSelected( myPolygonName2PrsShape.value(aName)->getAISObject() );
 352       if ( isSelected )
 353         aSelectedNames << aName;
 354     }
 355     aPanel->setSelectedPolygonNames( aSelectedNames );
 356   }
 357 }
 358
 359
 360 void HYDROGUI_ImportLandCoverMapOp::erasePreview()
 361 {
 362   foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape )
 363     delete aShape;
 364
 365   myPolygonName2PrsShape.clear();
 366 }
 367
 368
 369 void HYDROGUI_ImportLandCoverMapOp::abortOperation()
 370 {
 371   LightApp_Application* anApp = module()->getApp();
 372   if ( anApp )
 373     anApp->disconnect( this );
 374
 375   erasePreview();
 376   myImporter.Free();
 377   HYDROGUI_Operation::abortOperation();
 378 }
 379
 380 void HYDROGUI_ImportLandCoverMapOp::onNext( const int theIndex )
 381 {
 382   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 383   if ( !aPanel )
 384     return;
 385
 386   if (theIndex == 2)
 387   {
 388     //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrValue;
 389     int anIndOfSelField = myFieldList.indexOf(aPanel->getSelectedFieldName());
 390     if (anIndOfSelField == -1)
 391       return;
 392     //aPanel->setSecondPageState(true);
 393     myAttrValue.clear();
 394     myImporter.DBF_GetAttributeList(anIndOfSelField, myAttrValue );
 395
 396     mySetOfAttrValues.clear();
 397     for (size_t i = 0; i < myAttrValue.size(); i++)
 398     {
 399       HYDROData_ShapeFile::DBF_AttrValue aV = myAttrValue[i];
 400       mySetOfAttrValues << QString(aV.myStrVal);  //take myStrVal for now..
 401     }
 402
 403     //Collect all strickler_types
 404     QSet<QString> aSTSet;
 405     Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
 406     Handle(HYDROData_StricklerTable) DefStricklerTableObj;
 407     if ( aDoc )
 408     {
 409       HYDROData_Iterator It( aDoc, KIND_STRICKLER_TABLE );
 410       for( ; It.More(); It.Next() )
 411       {
 412         Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( It.Current() );
 413         if (DefStricklerTableObj.IsNull())
 414           DefStricklerTableObj = aStricklerTableObj;
 415         if ( !aStricklerTableObj.IsNull())
 416         {
 417           const QStringList& aStricklerList = aStricklerTableObj->GetTypes();
 418           foreach (QString aStr, aStricklerList)
 419             aSTSet << aStr;
 420         }
 421       }
 422     }
 423
 424     QList<QString> aSetOfAttrValuesList = mySetOfAttrValues.toList();
 425     QList<QString> aSTSetList = aSTSet.toList();
 426     QVector<QColor> aSTColors;
 427     aSTColors.reserve(aSTSetList.size());
 428     foreach (QString str, aSTSetList)
 429     {
 430       QColor col = aDoc->GetAssociatedColor(str, NULL);
 431       aSTColors.append (col);
 432     }
 433
 434     //add an empty Strickler type
 435     aSTSetList.prepend("");
 436     aSTColors.prepend(QColor(Qt::gray));
 437
 438     QVector<int> aCurCBIndices(aSetOfAttrValuesList.size());
 439     if (DefStricklerTableObj->GetAttrName() == aPanel->getSelectedFieldName())
 440       for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
 441       {
 442         QString aST = DefStricklerTableObj->GetType(aSetOfAttrValuesList[i]);
 443         int anIndOfSt = aSTSetList.indexOf(aST);
 444         aCurCBIndices[i] = anIndOfSt;
 445       }
 446     else
 447       //TODO add warning ???
 448       for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
 449         aCurCBIndices[i] = 0;
 450
 451     aPanel->FillCorrespondenceTable(aSetOfAttrValuesList, aSTSetList, aCurCBIndices, aSTColors);
 452   }
 453
 454 }
 455
 456
 457 void HYDROGUI_ImportLandCoverMapOp::onApply()
 458 {
 459   QApplication::setOverrideCursor( Qt::WaitCursor );
 460   int anUpdateFlags = 0;
 461   QString anErrorMsg;
 462   QStringList aBrowseObjectsEntries;
 463
 464   bool aResult = false;
 465   try
 466   {
 467     aResult = processApply( anUpdateFlags, anErrorMsg, aBrowseObjectsEntries );
 468   }
 469   catch (...)
 470   {
 471     SUIT_MessageBox::critical( module()->getApp()->desktop(), tr( "LCM_IMPORT_ERROR" ), tr ("CANNT_IMPORT_POLYGONS_FROM_SHP"));
 472     aResult = false;
 473   }
 474
 475   QApplication::restoreOverrideCursor();
 476
 477   if ( aResult )
 478   {
 479     module()->update( anUpdateFlags );
 480     commit();
 481     browseObjects( aBrowseObjectsEntries );
 482   }
 483   else
 484   {
 485     Handle(HYDROData_Entity) LCM = Handle(HYDROData_Entity)::DownCast( myLCM );
 486     LCM->Remove();
 487     module()->setObjectRemoved( myLCM );
 488     abort();
 489   }
 490 }
 491