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_Tool2.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   startDocOperation();
 117   QApplication::setOverrideCursor(Qt::WaitCursor);
 118
 119   QStringList aPolygonsList;
 120   myPolygonFaces.Clear();
 121   myImporter.Free();
 122
 123   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( module()->getApp()->activeStudy() );
 124   if ( !aStudy )
 125     return;
 126
 127   erasePreview();
 128
 129   Handle(AIS_InteractiveContext) aCtx = NULL;
 130   int aShapeTypeOfFile = -1;
 131
 132   //Import polygons from SHP file as faces
 133   //This faces should be added to the new LCM object
 134
 135   int aStat = myImporter.ImportPolygons(doc(), myFileName, aPolygonsList, myPolygonFaces, aShapeTypeOfFile);
 136   if (aStat == 1)
 137   {
 138     aPanel->setPolygonNames(aPolygonsList);
 139
 140     LightApp_Application* anApp = module()->getApp();
 141     if ( !getPreviewManager() )
 142       setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
 143     OCCViewer_ViewManager* aViewManager = getPreviewManager();
 144
 145     if ( aViewManager )
 146     {
 147       if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
 148       {
 149         aCtx = aViewer->getAISContext();
 150         connect( aViewer, SIGNAL( selectionChanged() ), this, SLOT( onViewerSelectionChanged() ) );
 151       }
 152     }
 153
 154     for ( int i = 1; i <= myPolygonFaces.Length(); i++ )
 155     {
 156       TopoDS_Face aFace = TopoDS::Face(myPolygonFaces.Value( i ));
 157       if ( aViewManager && !aCtx.IsNull() )
 158       {
 159         HYDROGUI_Shape* aShape = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
 160         //Preview color is grey
 161         aShape->setFillingColor(QColor(50,50,50), false, false);
 162         aShape->setBorderColor(QColor(50,50,50), false, false);
 163         if( !aFace.IsNull() )
 164           aShape->setShape( aFace);
 165
 166         myPolygonName2PrsShape.insert( tr( "DEF_POLYGON_NAME" ) + "_" + QString::number(i), aShape);
 167       }
 168     }
 169
 170     if ( !aCtx.IsNull() )
 171     {
 172       UpdateZLayersOfHilightPresentationsOfDisplayedObjects( aCtx, Graphic3d_ZLayerId_TopOSD );
 173       aCtx->UpdateCurrentViewer();
 174     }
 175     //
 176     QApplication::restoreOverrideCursor();
 177     aPanel->setFirstPageState(true);
 178     //
 179     //Try to load DBF-database...
 180     QString aDBFFileName;
 181     aDBFFileName = myFileName.simplified().replace( myFileName.simplified().size() - 4, 4, ".dbf");
 182     bool DBF_Stat = myImporter.DBF_OpenDBF(aDBFFileName);
 183     // TODO:
 184     // add MSG BOX if stat is bad
 185     if (DBF_Stat)
 186     {
 187       myFieldList = myImporter.DBF_GetFieldList();
 188       aPanel->setAttributeNames(myFieldList);
 189     }
 190     aPanel->setDbfState(DBF_Stat);
 191   }
 192   else
 193   {
 194     erasePreview();
 195     aPanel->setPolygonNames(QStringList());
 196     aPanel->setObjectName("");
 197     QApplication::restoreOverrideCursor();
 198     QString aMess = tr( "CANNT_IMPORT_LCM" ) + "\n";
 199     if (aStat == -1)
 200       aMess += tr( "CANNT_OPEN_SHP" );
 201     else if (aStat == -2)
 202       aMess += tr( "CANNT_OPEN_SHX" );
 203     else
 204       aMess += tr ("SHP_TYPEFORMAT_MESS") + myImporter.GetShapeTypeName(aShapeTypeOfFile);
 205     SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "IMPORT_LANDCOVER_MAP" ), aMess);
 206     commitDocOperation();
 207     myImporter.Free();
 208     //abort();
 209     aPanel->setFirstPageState(false);
 210   }
 211
 212 }
 213
 214 HYDROGUI_InputPanel* HYDROGUI_ImportLandCoverMapOp::createInputPanel() const
 215 {
 216   HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportLandCoverMapDlg( module(), getName() );
 217
 218   connect( aPanel, SIGNAL( FileSelected( const QString& ) ), SLOT( onFileSelected() ) );
 219   connect( aPanel, SIGNAL( selectionChanged( const QStringList& ) ), this, SLOT( onSelectionChanged( const QStringList& ) ) );
 220   connect( aPanel, SIGNAL( Next( const int ) ), SLOT( onNext( const int ) ) );
 221
 222   return aPanel;
 223 }
 224
 225 bool HYDROGUI_ImportLandCoverMapOp::processApply( int& theUpdateFlags,
 226                                                   QString& theErrorMsg,
 227                                                   QStringList& theBrowseObjectsEntries )
 228 {
 229
 230   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 231   if ( !aPanel )
 232     return false;
 233
 234   myLCM = Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
 235   HYDROData_MapOfFaceToStricklerType aMapFace2ST;
 236
 237   QStringList aAttrV_T;
 238   QStringList aSTL;
 239   aPanel->getValAttr2StricklerTypeCorr(aAttrV_T, aSTL);
 240
 241   QVector<int> aSelIndices = aPanel->getSelectedPolygonIndices();
 242   foreach ( int Ind, aSelIndices )
 243   {
 244     TopoDS_Shape aShape = myPolygonFaces(Ind + 1);
 245     if ( aShape.IsNull() )
 246       continue;
 247     QString aST = "";
 248     if (aPanel->getAttrCheckBoxState())
 249     {
 250       HYDROData_ShapeFile::DBF_AttrValue aDataVal = myAttrValue[Ind];
 251       int aStricklerTypesInd = aAttrV_T.indexOf(QString(aDataVal.myStrVal));
 252       aST = aSTL.at(aStricklerTypesInd);
 253     }
 254     // else => ST is empty
 255     aMapFace2ST.Add( TopoDS::Face( aShape ), aST );
 256   }
 257
 258   //
 259   myLCM->StoreLandCovers(aMapFace2ST);
 260
 261   QString anObjName;
 262   if ( !aPanel->getFileName().isEmpty() )
 263     anObjName = aPanel->getObjectName();
 264
 265   Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
 266
 267   //check if name of LCM is already exists
 268   QSet<QString> aNameList;
 269   if (aDoc)
 270   {
 271     HYDROData_Iterator anIt( aDoc, KIND_LAND_COVER_MAP );
 272     for( ; anIt.More(); anIt.Next() )
 273       aNameList << anIt.Current()->GetName();
 274   }
 275
 276   QString aNewName = anObjName;
 277   for ( int i = 1; i < MAX_LCM_NAME_INDEX && aNameList.contains(aNewName); i++)
 278     aNewName = anObjName + "_" + QString::number(i);
 279
 280   if( !myLCM.IsNull() )
 281   {
 282     myLCM->SetName( aNewName );
 283     myLCM->SetColor( Qt::gray );
 284     myLCM->Show();
 285     module()->setIsToUpdate( myLCM );
 286   }
 287
 288   module()->update( UF_Model | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init );
 289
 290   erasePreview();
 291   myImporter.Free();
 292   myAttrValue.clear();
 293
 294   return true;
 295 }
 296
 297 void HYDROGUI_ImportLandCoverMapOp::onSelectionChanged( const QStringList& theSelectedNames )
 298 {
 299   Handle(AIS_InteractiveContext) aCtx = NULL;
 300
 301   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 302   if ( aViewManager ) {
 303     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
 304       aCtx = aViewer->getAISContext();
 305     }
 306   }
 307
 308   if ( !aCtx.IsNull() ) {
 309     bool bSelChanged = false;
 310     foreach ( QString aName, myPolygonName2PrsShape.keys() )
 311     {
 312       Handle(AIS_InteractiveObject) anObject =
 313         myPolygonName2PrsShape.value(aName)->getAISObject();
 314
 315       bool isSelected = theSelectedNames.contains( aName );
 316       if ( ( isSelected && !aCtx->IsSelected( anObject) ) || ( !isSelected && aCtx->IsSelected( anObject) ) )
 317       {
 318         aCtx->AddOrRemoveSelected( anObject, Standard_False );
 319         bSelChanged = true;
 320       }
 321     }
 322
 323     HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 324     if ( !aPanel )
 325       return;
 326
 327     if (bSelChanged && aPanel->getViewerState())
 328       aCtx->UpdateCurrentViewer();
 329   }
 330 }
 331
 332
 333 void HYDROGUI_ImportLandCoverMapOp::onViewerSelectionChanged()
 334 {
 335   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 336   if ( !aPanel )
 337     return;
 338
 339   int aCurIndex = -1;
 340   aCurIndex = aPanel->getCurrentWizardIndex();
 341
 342   if (aCurIndex != 0)
 343     return;
 344
 345   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 346   Handle(AIS_InteractiveContext) aCtx = NULL;
 347   if ( aViewManager )
 348     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
 349       aCtx = aViewer->getAISContext();
 350
 351   if ( !aCtx.IsNull() )
 352   {
 353     QStringList aSelectedNames;
 354     foreach ( QString aName, myPolygonName2PrsShape.keys() ) {
 355       bool isSelected = aCtx->IsSelected( myPolygonName2PrsShape.value(aName)->getAISObject() );
 356       if ( isSelected )
 357         aSelectedNames << aName;
 358     }
 359     aPanel->setViewerState(false);
 360     aPanel->setSelectedPolygonNames( aSelectedNames );
 361     aPanel->setViewerState(true);
 362   }
 363 }
 364
 365
 366 void HYDROGUI_ImportLandCoverMapOp::erasePreview()
 367 {
 368   foreach ( HYDROGUI_Shape* aShape, myPolygonName2PrsShape )
 369     delete aShape;
 370
 371   myPolygonName2PrsShape.clear();
 372 }
 373
 374
 375 void HYDROGUI_ImportLandCoverMapOp::abortOperation()
 376 {
 377   LightApp_Application* anApp = module()->getApp();
 378   if ( anApp )
 379     anApp->disconnect( this );
 380
 381   erasePreview();
 382   myImporter.Free();
 383   HYDROGUI_Operation::abortOperation();
 384 }
 385
 386 void HYDROGUI_ImportLandCoverMapOp::onNext( const int theIndex )
 387 {
 388   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 389   if ( !aPanel )
 390     return;
 391
 392   if (theIndex == 2)
 393   {
 394     //std::vector<HYDROData_ShapeFile::DBF_AttrValue> myAttrValue;
 395     int anIndOfSelField = myFieldList.indexOf(aPanel->getSelectedFieldName());
 396     if (anIndOfSelField == -1)
 397       return;
 398     //aPanel->setSecondPageState(true);
 399     myAttrValue.clear();
 400     myImporter.DBF_GetAttributeList(anIndOfSelField, myAttrValue );
 401
 402     mySetOfAttrValues.clear();
 403     for (size_t i = 0; i < myAttrValue.size(); i++)
 404     {
 405       HYDROData_ShapeFile::DBF_AttrValue aV = myAttrValue[i];
 406       mySetOfAttrValues << QString(aV.myStrVal);  //take myStrVal for now..
 407     }
 408
 409     //Collect all strickler_types
 410     QSet<QString> aSTSet;
 411     Handle_HYDROData_Document aDoc = HYDROData_Document::Document( application()->activeStudy()->id() );
 412     Handle(HYDROData_StricklerTable) DefStricklerTableObj;
 413     if ( aDoc )
 414     {
 415       HYDROData_Iterator It( aDoc, KIND_STRICKLER_TABLE );
 416       for( ; It.More(); It.Next() )
 417       {
 418         Handle(HYDROData_StricklerTable) aStricklerTableObj = Handle(HYDROData_StricklerTable)::DownCast( It.Current() );
 419         if (DefStricklerTableObj.IsNull())
 420           DefStricklerTableObj = aStricklerTableObj;
 421         if ( !aStricklerTableObj.IsNull())
 422         {
 423           const QStringList& aStricklerList = aStricklerTableObj->GetTypes();
 424           foreach (QString aStr, aStricklerList)
 425             aSTSet << aStr;
 426         }
 427       }
 428     }
 429
 430     QList<QString> aSetOfAttrValuesList = mySetOfAttrValues.toList();
 431     QList<QString> aSTSetList = aSTSet.toList();
 432     QVector<QColor> aSTColors;
 433     aSTColors.reserve(aSTSetList.size());
 434     foreach (QString str, aSTSetList)
 435     {
 436       QColor col = aDoc->GetAssociatedColor(str, NULL);
 437       aSTColors.append (col);
 438     }
 439
 440     //add an empty Strickler type
 441     aSTSetList.prepend("");
 442     aSTColors.prepend(QColor(Qt::gray));
 443
 444     QVector<int> aCurCBIndices(aSetOfAttrValuesList.size());
 445     if (DefStricklerTableObj->GetAttrName().trimmed() == aPanel->getSelectedFieldName().trimmed())
 446       for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
 447       {
 448         QString aST = DefStricklerTableObj->GetType(aSetOfAttrValuesList[i]);
 449         int anIndOfSt = aSTSetList.indexOf(aST);
 450         aCurCBIndices[i] = anIndOfSt;
 451       }
 452     else
 453       //TODO add warning ???
 454       for (int i = 0; i < aSetOfAttrValuesList.size(); i++)
 455         aCurCBIndices[i] = 0;
 456
 457     aPanel->FillCorrespondenceTable(aSetOfAttrValuesList, aSTSetList, aCurCBIndices, aSTColors);
 458   }
 459
 460 }
 461
 462
 463 void HYDROGUI_ImportLandCoverMapOp::onApply()
 464 {
 465   HYDROGUI_ImportLandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_ImportLandCoverMapDlg*>( inputPanel() );
 466   if ( !aPanel )
 467     return;
 468
 469   if (!aPanel->CheckFirstPageFilling())
 470     return;
 471
 472   QApplication::setOverrideCursor( Qt::WaitCursor );
 473   int anUpdateFlags = 0;
 474   QString anErrorMsg;
 475   QStringList aBrowseObjectsEntries;
 476
 477   bool aResult = false;
 478   try
 479   {
 480     aResult = processApply( anUpdateFlags, anErrorMsg, aBrowseObjectsEntries );
 481   }
 482   catch (...)
 483   {
 484     SUIT_MessageBox::critical( module()->getApp()->desktop(), tr( "LCM_IMPORT_ERROR" ), tr ("CANNT_IMPORT_POLYGONS_FROM_SHP"));
 485     aResult = false;
 486   }
 487
 488   QApplication::restoreOverrideCursor();
 489
 490   if ( aResult )
 491   {
 492     module()->update( anUpdateFlags );
 493     commit();
 494     browseObjects( aBrowseObjectsEntries );
 495   }
 496   else
 497   {
 498     Handle(HYDROData_Entity) LCM = Handle(HYDROData_Entity)::DownCast( myLCM );
 499     LCM->Remove();
 500     module()->setObjectRemoved( myLCM );
 501     abort();
 502   }
 503 }
 504