src/HYDROGUI/HYDROGUI_LandCoverMapOp.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include "HYDROGUI_LandCoverMapOp.h"
  20
  21 #include "HYDROGUI_Module.h"
  22 #include "HYDROGUI_OCCDisplayer.h"
  23 #include "HYDROGUI_Operations.h"
  24 #include "HYDROGUI_LandCoverMapDlg.h"
  25 #include "HYDROGUI_Tool.h"
  26 #include "HYDROGUI_UpdateFlags.h"
  27 #include "HYDROGUI_DataObject.h"
  28
  29 #include <HYDROData_Iterator.h>
  30 #include <HYDROData_StricklerTable.h>
  31 #include <HYDROData_PolylineXY.h>
  32 #include <HYDROData_Object.h>
  33
  34 #include <TopoDS_Face.hxx>
  35 #include <TopTools_ListOfShape.hxx>
  36 #include <TopTools_ListIteratorOfListOfShape.hxx>
  37
  38 HYDROGUI_LandCoverMapOp::HYDROGUI_LandCoverMapOp( HYDROGUI_Module* theModule, const int theOperationId )
  39 : HYDROGUI_Operation( theModule ),
  40   myOperationId( theOperationId ),
  41   myPreviewPrs( 0 )
  42 {
  43   switch( myOperationId )
  44   {
  45     case CreateLandCoverMapId:
  46       setName( tr( "CREATE_LAND_COVER_MAP" ) );
  47       break;
  48     case AddLandCoverId:
  49       setName( tr( "ADD_LAND_COVER" ) );
  50       break;
  51     case RemoveLandCoverId:
  52       setName( tr( "REMOVE_LAND_COVER" ) );
  53       break;
  54     case SplitLandCoverId:
  55       setName( tr( "SPLIT_LAND_COVER" ) );
  56       break;
  57     case MergeLandCoverId:
  58       setName( tr( "MERGE_LAND_COVER" ) );
  59       break;
  60     case ChangeLandCoverTypeId:
  61       setName( tr( "CHANGE_LAND_COVER_TYPE" ) );
  62       break;
  63   }
  64 }
  65
  66 HYDROGUI_LandCoverMapOp::~HYDROGUI_LandCoverMapOp()
  67 {
  68   closePreview();
  69 }
  70
  71 void HYDROGUI_LandCoverMapOp::startOperation()
  72 {
  73   HYDROGUI_Operation::startOperation();
  74
  75   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
  76   if ( !aPanel )
  77     return;
  78
  79   aPanel->blockSignals( true );
  80
  81   aPanel->reset();
  82
  83   // Set name of the created/edited land cover map object
  84   QString anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_LAND_COVER_MAP_NAME" ) );
  85   if ( myOperationId != CreateLandCoverMapId )
  86   {
  87     if ( isApplyAndClose() )
  88       myEditedObject = Handle(HYDROData_LandCoverMap)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  89     if ( !myEditedObject.IsNull() )
  90     {
  91       anObjectName = myEditedObject->GetName();
  92
  93       // Construct a list of names of all land cover map objects defined within the data model
  94       QStringList aLandCoverMapNames;
  95       HYDROData_Iterator anIterator( doc(), KIND_LAND_COVER_MAP );
  96       for( ; anIterator.More(); anIterator.Next() )
  97       {
  98         Handle(HYDROData_LandCoverMap) aLandCoverObj =
  99           Handle(HYDROData_LandCoverMap)::DownCast( anIterator.Current() );
 100         if ( !aLandCoverObj.IsNull() )
 101           aLandCoverMapNames.append( aLandCoverObj->GetName() );
 102       }
 103
 104       //aLandCoverMapNames.sort();
 105       aPanel->setObjectNames( aLandCoverMapNames );
 106     }
 107   }
 108   aPanel->setObjectName( anObjectName );
 109
 110   aPanel->blockSignals( false );
 111 }
 112
 113 void HYDROGUI_LandCoverMapOp::abortOperation()
 114 {
 115   closePreview();
 116
 117   HYDROGUI_Operation::abortOperation();
 118 }
 119
 120 void HYDROGUI_LandCoverMapOp::commitOperation()
 121 {
 122   closePreview();
 123
 124   HYDROGUI_Operation::commitOperation();
 125 }
 126
 127 HYDROGUI_InputPanel* HYDROGUI_LandCoverMapOp::createInputPanel() const
 128 {
 129   HYDROGUI_LandCoverMapDlg* aPanel = new HYDROGUI_LandCoverMapDlg( module(), getName(), myOperationId );
 130   connect( aPanel, SIGNAL( landCoverMapChanged( const QString& ) ),
 131            this, SLOT( onLandCoverMapChanged( const QString& ) ) );
 132   connect( aPanel, SIGNAL( CreatePreview( const QStringList& ) ),
 133            this,   SLOT( onCreatePreview( const QStringList& ) ) );
 134   /*
 135   connect( aPanel, SIGNAL( addPolylines() ), SLOT( onAddPolylines() ) );
 136   connect( aPanel, SIGNAL( removePolylines() ), SLOT( onRemovePolylines() ) );
 137   */
 138   return aPanel;
 139 }
 140
 141 bool HYDROGUI_LandCoverMapOp::processApply( int& theUpdateFlags,
 142                                             QString& theErrorMsg,
 143                                             QStringList& theBrowseObjectsEntries )
 144 {
 145   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
 146   if ( !aPanel )
 147     return false;
 148
 149   // Check name of the created/edited object
 150   QString anObjectName = aPanel->getObjectName().simplified();
 151   if ( anObjectName.isEmpty() )
 152   {
 153     theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
 154     return false;
 155   }
 156
 157   if ( myOperationId == CreateLandCoverMapId || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anObjectName ) )
 158   {
 159     // check that there are no other objects with the same name in the document
 160     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName );
 161     if( !anObject.IsNull() )
 162     {
 163       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
 164       return false;
 165     }
 166   }
 167
 168   Handle(HYDROData_PolylineXY) aPolyline;
 169   Handle(HYDROData_Object) aFace;
 170
 171   TopTools_ListOfShape aFacesSelectedInViewer;
 172
 173   // Get polyline/face selected in combo-box
 174   if ( myOperationId == CreateLandCoverMapId || myOperationId == AddLandCoverId || myOperationId == SplitLandCoverId )
 175   {
 176     Handle(HYDROData_Entity) aPolylineFace = aPanel->getPolylineFace();
 177     aPolyline = Handle(HYDROData_PolylineXY)::DownCast( aPolylineFace );
 178     aFace = Handle(HYDROData_Object)::DownCast( aPolylineFace );
 179     if ( aPolyline.IsNull() && aFace.IsNull() )
 180     {
 181       theErrorMsg = tr( "POLYLINE_FACE_NOT_DEFINED" );
 182       return false;
 183     }
 184   }
 185   // Get face(s) selected in the 3d viewer
 186   else if ( myOperationId == RemoveLandCoverId ||
 187             myOperationId == MergeLandCoverId ||
 188             myOperationId == ChangeLandCoverTypeId )
 189   {
 190     // TODO:
 191     //Fill in aFacesSelectedInViewer list
 192   }
 193
 194   // Get selected Strickler type
 195   QString aSelectedStricklerType;
 196   if ( myOperationId == CreateLandCoverMapId ||
 197        myOperationId == AddLandCoverId ||
 198        myOperationId == MergeLandCoverId )
 199   {
 200     aSelectedStricklerType = aPanel->getSelectedStricklerTypeName();
 201     if ( aSelectedStricklerType.isEmpty() )
 202     {
 203       theErrorMsg = tr( "STRICKLER_TYPE_NOT_DEFINED" );
 204       return false;
 205     }
 206   }
 207
 208   // Create / find the new / edited land cover map object
 209   Handle(HYDROData_LandCoverMap) aLandCoverMapObj = myOperationId != CreateLandCoverMapId ? myEditedObject :
 210     Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
 211
 212   // Set land cover map name
 213   aLandCoverMapObj->SetName( anObjectName );
 214
 215   // Add land cover to new / edited land cover map
 216   if ( myOperationId == CreateLandCoverMapId || myOperationId == AddLandCoverId )
 217   {
 218     bool aLandCoverAdded = false;
 219     if ( !aPolyline.IsNull() )
 220       aLandCoverAdded = aLandCoverMapObj->Add( aPolyline, aSelectedStricklerType );
 221     else if ( !aFace.IsNull() )
 222       aLandCoverAdded = aLandCoverMapObj->Add( aFace, aSelectedStricklerType );
 223     if ( !aLandCoverAdded )
 224     {
 225       theErrorMsg = tr( "LAND_COVER_NOT_ADDED" );
 226       return false;
 227     }
 228   }
 229
 230   // Remove land cover from edited land cover map
 231   if ( myOperationId == RemoveLandCoverId )
 232   {
 233     bool aLandCoverRemoved = false;
 234     if ( !aFacesSelectedInViewer.IsEmpty() )
 235     {
 236       aLandCoverRemoved = aLandCoverMapObj->Remove( aFacesSelectedInViewer );
 237       if ( !aLandCoverRemoved )
 238       {
 239         theErrorMsg = tr( "LAND_COVER_NOT_REMOVED" );
 240         return false;
 241       }
 242     }
 243   }
 244
 245   // Split land cover(s) inside edited land cover map
 246   if ( myOperationId == SplitLandCoverId )
 247   {
 248     bool aLandCoverSplitted = false;
 249     if ( !aPolyline.IsNull() )
 250       aLandCoverSplitted = aLandCoverMapObj->Split( aPolyline );
 251     else if ( !aFace.IsNull() )
 252     {
 253       // TODO:
 254       //Get the complete boundary of the object face as the splitting polyline
 255       Handle(HYDROData_PolylineXY) aBoundaryPolyline;
 256       aLandCoverSplitted = aLandCoverMapObj->Split( aBoundaryPolyline );
 257     }
 258     if ( !aLandCoverSplitted )
 259     {
 260       theErrorMsg = tr( "LAND_COVER_NOT_SPLITTED" );
 261       return false;
 262     }
 263   }
 264
 265   // Merge land covers inside edited land cover map
 266   if ( myOperationId == MergeLandCoverId )
 267   {
 268     bool aLandCoverMerged = false;
 269     if ( !aFacesSelectedInViewer.IsEmpty() )
 270     {
 271       aLandCoverMerged = aLandCoverMapObj->Merge( aFacesSelectedInViewer, aSelectedStricklerType );
 272       if ( !aLandCoverMerged )
 273       {
 274         theErrorMsg = tr( "LAND_COVER_NOT_MERGED" );
 275         return false;
 276       }
 277     }
 278   }
 279
 280   /*if ( myOperationId == CreateLandCoverMapId )
 281   {
 282     aLandCoverMapObj->SetFillingColor( aLandCoverMapObj->DefaultFillingColor() );
 283     aLandCoverMapObj->SetBorderColor( aLandCoverMapObj->DefaultBorderColor() );
 284   }*/
 285
 286   // Update land cover map object and close preview
 287   aLandCoverMapObj->Update();
 288
 289   closePreview();
 290
 291   // Publish the newly created land cover map in the Object Browser
 292   module()->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( module() ), aLandCoverMapObj, true );
 293   if ( myOperationId == CreateLandCoverMapId )
 294   {
 295     QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aLandCoverMapObj );
 296     theBrowseObjectsEntries.append( anEntry );
 297   }
 298
 299   // Update presentation of land cover object in the 3d viewer
 300   module()->setIsToUpdate( aLandCoverMapObj );
 301   module()->getOCCDisplayer()->SetToUpdateColorScale();
 302
 303   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
 304
 305   return true;
 306 }
 307
 308 void HYDROGUI_LandCoverMapOp::onLandCoverMapChanged( const QString& theName )
 309 {
 310   // If the edited land cover map was changed in the combo-box, update myEditedObject
 311   if ( myOperationId != CreateLandCoverMapId )
 312   {
 313     myEditedObject = Handle(HYDROData_LandCoverMap)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theName ) );
 314     if ( !myEditedObject.IsNull() )
 315     {
 316       // Show preview of the newly selected land cover map
 317       QStringList aPolylineFaceNames;
 318       onCreatePreview( aPolylineFaceNames );
 319     }
 320   }
 321 }
 322
 323 void HYDROGUI_LandCoverMapOp::onCreatePreview( const QStringList& thePolylineFaceNames )
 324 {
 325   /*
 326   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
 327   if ( !aPanel )
 328     return;
 329
 330   QApplication::setOverrideCursor( Qt::WaitCursor );
 331
 332   HYDROData_SequenceOfObjects aZonePolylines;
 333   QStringList::const_iterator anIt = thePolylineNames.begin(), aLast = thePolylineNames.end();
 334   for( ; anIt!=aLast; anIt++ )
 335   {
 336     QString aPolylineName = *anIt;
 337     Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast(
 338       HYDROGUI_Tool::FindObjectByName( module(), aPolylineName, KIND_POLYLINEXY ) );
 339     if ( !aPolyline.IsNull() )
 340       aZonePolylines.Append( aPolyline );
 341   }
 342
 343   TCollection_AsciiString anError;
 344   TopoDS_Shape aZoneShape = HYDROData_LandCover::buildShape( aZonePolylines, anError );
 345
 346   LightApp_Application* anApp = module()->getApp();
 347   if ( !getPreviewManager() )
 348     setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>(
 349                        anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
 350   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 351   if ( aViewManager && !myPreviewPrs )
 352   {
 353     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
 354     {
 355       Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
 356       if ( !aCtx.IsNull() )
 357         myPreviewPrs = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
 358     }
 359   }
 360
 361   if ( aViewManager && myPreviewPrs )
 362   {
 363     QColor aFillingColor = Qt::magenta;
 364     QColor aBorderColor = Qt::transparent;
 365     if ( !myEditedObject.IsNull() ) {
 366       aFillingColor = myEditedObject->GetFillingColor();
 367       aBorderColor = myEditedObject->GetBorderColor();
 368     }
 369
 370     myPreviewPrs->setFillingColor( aFillingColor, false, false );
 371     myPreviewPrs->setBorderColor( aBorderColor, false, false );
 372
 373     if( !aZoneShape.IsNull() )
 374       myPreviewPrs->setShape( aZoneShape );
 375   }
 376
 377   QApplication::restoreOverrideCursor();
 378   */
 379 }
 380
 381 void HYDROGUI_LandCoverMapOp::closePreview()
 382 {
 383   if( myPreviewPrs )
 384   {
 385     delete myPreviewPrs;
 386     myPreviewPrs = 0;
 387   }
 388 }