src/HYDROGUI/HYDROGUI_LandCoverOp.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_LandCoverOp.h"
  20
  21 #include "HYDROGUI_LandCoverDlg.h"
  22 #include "HYDROGUI_Module.h"
  23 #include "HYDROGUI_Shape.h"
  24 #include "HYDROGUI_Tool.h"
  25 #include "HYDROGUI_UpdateFlags.h"
  26 #include "HYDROGUI_DataObject.h"
  27
  28 #include <HYDROData_PolylineXY.h>
  29 #include <HYDROData_Document.h>
  30 #include <HYDROData_Iterator.h>
  31 #include <HYDROData_StricklerTable.h>
  32
  33 #include <OCCViewer_ViewManager.h>
  34 #include <OCCViewer_ViewModel.h>
  35
  36 #include <LightApp_Application.h>
  37
  38 #include <SUIT_MessageBox.h>
  39 #include <SUIT_Desktop.h>
  40
  41 #include <TopoDS.hxx>
  42 #include <TopoDS_Face.hxx>
  43 #include <TopoDS_Wire.hxx>
  44
  45 #include <QApplication>
  46
  47 HYDROGUI_LandCoverOp::HYDROGUI_LandCoverOp( HYDROGUI_Module* theModule,
  48                                             const bool theIsEdit )
  49 : HYDROGUI_Operation( theModule ),
  50   myIsEdit( theIsEdit ),
  51   myPreviewPrs( 0 )
  52 {
  53   setName( theIsEdit ? tr( "EDIT_LAND_COVER" ) : tr( "CREATE_LAND_COVER" ) );
  54 }
  55
  56 HYDROGUI_LandCoverOp::~HYDROGUI_LandCoverOp()
  57 {
  58   closePreview();
  59 }
  60
  61 void HYDROGUI_LandCoverOp::startOperation()
  62 {
  63   HYDROGUI_Operation::startOperation();
  64
  65   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
  66   if ( !aPanel )
  67     return;
  68
  69   aPanel->blockSignals( true );
  70
  71   aPanel->reset();
  72
  73   QString anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_LAND_COVER_NAME" ) );
  74
  75   HYDROGUI_ListModel::Object2VisibleList aSelectedPolylines;
  76   QString     aSelectedStricklerType;
  77
  78   if ( myIsEdit )
  79   {
  80     if ( isApplyAndClose() )
  81       myEditedObject = Handle(HYDROData_LandCover)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  82     if ( !myEditedObject.IsNull() )
  83     {
  84       anObjectName = myEditedObject->GetName();
  85
  86       HYDROData_SequenceOfObjects aRefPolylines = myEditedObject->GetPolylines();
  87       for ( int i = aRefPolylines.Lower(); i <= aRefPolylines.Upper(); i++ ) {
  88         Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( aRefPolylines.Value( i ) );
  89         if ( !aPolyline.IsNull() ) {
  90           aSelectedPolylines.append( HYDROGUI_ListModel::Object2Visible( aPolyline, true ) );
  91         }
  92       }
  93     }
  94
  95     aSelectedStricklerType = myEditedObject->GetStricklerType();
  96   }
  97
  98   aPanel->setObjectName( anObjectName );
  99   // Construct a list of unique names of all Strickler types defined within the data model
 100   QStringList aStricklerTypes;
 101   HYDROData_Iterator anIterator( doc(), KIND_STRICKLER_TABLE );
 102   for( ; anIterator.More(); anIterator.Next() )
 103   {
 104     Handle(HYDROData_StricklerTable) aStricklerTableObj =
 105       Handle(HYDROData_StricklerTable)::DownCast( anIterator.Current() );
 106     if ( !aStricklerTableObj.IsNull() )
 107     {
 108       // Get Strickler table data from the data model
 109       TColStd_SequenceOfExtendedString aTypes = aStricklerTableObj->GetTypes();
 110       for ( int i = 1; i <= aTypes.Length(); i++ )
 111       {
 112         QString aType = HYDROGUI_Tool::ToQString( aTypes.Value( i ) );
 113         if ( !aType.isEmpty() && !aStricklerTypes.contains( aType ))
 114           aStricklerTypes.append( aType );
 115       }
 116     }
 117   }
 118   aStricklerTypes.sort();
 119   aPanel->setAdditionalParams( aStricklerTypes );
 120
 121   aPanel->blockSignals( false );
 122
 123   aPanel->includePolylines( aSelectedPolylines );
 124   aPanel->setSelectedAdditionalParamName( aSelectedStricklerType );
 125 }
 126
 127 void HYDROGUI_LandCoverOp::abortOperation()
 128 {
 129   closePreview();
 130
 131   HYDROGUI_Operation::abortOperation();
 132 }
 133
 134 void HYDROGUI_LandCoverOp::commitOperation()
 135 {
 136   closePreview();
 137
 138   HYDROGUI_Operation::commitOperation();
 139 }
 140
 141 HYDROGUI_InputPanel* HYDROGUI_LandCoverOp::createInputPanel() const
 142 {
 143   HYDROGUI_LandCoverDlg* aPanel = new HYDROGUI_LandCoverDlg( module(), getName() );
 144   connect( aPanel, SIGNAL( CreatePreview( const QStringList& ) ),
 145            this,   SLOT( onCreatePreview( const QStringList& ) ) );
 146   connect( aPanel, SIGNAL( addPolylines() ), SLOT( onAddPolylines() ) );
 147   connect( aPanel, SIGNAL( removePolylines() ), SLOT( onRemovePolylines() ) );
 148   return aPanel;
 149 }
 150
 151 bool HYDROGUI_LandCoverOp::processApply( int& theUpdateFlags,
 152                                          QString& theErrorMsg,
 153                                          QStringList& theBrowseObjectsEntries )
 154 {
 155   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
 156   if ( !aPanel )
 157     return false;
 158
 159   QString anObjectName = aPanel->getObjectName().simplified();
 160   if ( anObjectName.isEmpty() )
 161   {
 162     theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
 163     return false;
 164   }
 165
 166   if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anObjectName ) )
 167   {
 168     // check that there are no other objects with the same name in the document
 169     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName );
 170     if( !anObject.IsNull() )
 171     {
 172       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
 173       return false;
 174     }
 175   }
 176
 177   HYDROData_SequenceOfObjects aZonePolylines;
 178   QString aStricklerType;
 179
 180   QStringList aSelectedPolylineNames = aPanel->getPolylineNames();
 181   QStringList::const_iterator anIt = aSelectedPolylineNames.begin(), aLast = aSelectedPolylineNames.end();
 182   for( ; anIt!=aLast; anIt++ )
 183   {
 184     QString aPolylineName = *anIt;
 185     if ( !aPolylineName.isEmpty() )
 186     {
 187       aZonePolylines.Append( Handle(HYDROData_PolylineXY)::DownCast(
 188         HYDROGUI_Tool::FindObjectByName( module(), aPolylineName, KIND_POLYLINEXY ) ) );
 189     }
 190   }
 191
 192   QString aSelectedStricklerType = aPanel->getSelectedAdditionalParamName();
 193
 194   // TODO: Generate TopoDS_Shape based on the set of polylines, implement generateTopShape data model method
 195   /*
 196   if ( HYDROData_LandCover::generateTopShape( aZonePolylines ).IsNull() )
 197   {
 198     theErrorMsg = tr( "ZONE_OBJECT_CANNOT_BE_CREATED" );
 199     return false;
 200   }
 201   */
 202
 203   Handle(HYDROData_LandCover) aZoneObj = myIsEdit ? myEditedObject :
 204     Handle(HYDROData_LandCover)::DownCast( doc()->CreateObject( KIND_LAND_COVER ) );
 205
 206   aZoneObj->SetName( anObjectName );
 207
 208   if ( !myIsEdit )
 209   {
 210     aZoneObj->SetFillingColor( HYDROData_LandCover::DefaultFillingColor() );
 211     aZoneObj->SetBorderColor( HYDROData_LandCover::DefaultBorderColor() );
 212   }
 213
 214   aZoneObj->SetPolylines( aZonePolylines );
 215   aZoneObj->SetStricklerType( aSelectedStricklerType );
 216   aZoneObj->Update();
 217
 218   closePreview();
 219
 220   if( !myIsEdit )
 221   {
 222     module()->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( module() ), aZoneObj, true );
 223     QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aZoneObj );
 224     theBrowseObjectsEntries.append( anEntry );
 225   }
 226
 227   module()->setIsToUpdate( aZoneObj );
 228
 229   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
 230
 231   return true;
 232 }
 233
 234 void HYDROGUI_LandCoverOp::onCreatePreview( const QStringList& thePolylineNames )
 235 {
 236   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
 237   if ( !aPanel )
 238     return;
 239
 240   QApplication::setOverrideCursor( Qt::WaitCursor );
 241
 242   QList<Handle(HYDROData_PolylineXY)> aZonePolylines;
 243   QStringList::const_iterator anIt = thePolylineNames.begin(), aLast = thePolylineNames.end();
 244   for( ; anIt!=aLast; anIt++ )
 245   {
 246     QString aPolylineName = *anIt;
 247     Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast(
 248       HYDROGUI_Tool::FindObjectByName( module(), aPolylineName, KIND_POLYLINEXY ) );
 249     if ( !aPolyline.IsNull() )
 250       aZonePolylines.append( aPolyline );
 251   }
 252
 253   // TODO: Generate TopoDS_Shape based on the set of polylines, implement generateTopShape data model method
 254   TopoDS_Shape aZoneShape;/* = HYDROData_LandCover::generateTopShape( aZonePolylines );
 255   if( aZoneShape.IsNull() )
 256     printErrorMessage( tr( "ZONE_OBJECT_CANNOT_BE_CREATED" ) );
 257   */
 258
 259   LightApp_Application* anApp = module()->getApp();
 260   if ( !getPreviewManager() )
 261     setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>(
 262                        anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
 263   OCCViewer_ViewManager* aViewManager = getPreviewManager();
 264   if ( aViewManager && !myPreviewPrs )
 265   {
 266     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
 267     {
 268       Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
 269       if ( !aCtx.IsNull() )
 270         myPreviewPrs = new HYDROGUI_Shape( aCtx, NULL, getPreviewZLayer() );
 271     }
 272   }
 273
 274   if ( aViewManager && myPreviewPrs )
 275   {
 276     QColor aFillingColor = HYDROData_LandCover::DefaultFillingColor();
 277     QColor aBorderColor = HYDROData_LandCover::DefaultBorderColor();
 278     if ( !myEditedObject.IsNull() ) {
 279       aFillingColor = myEditedObject->GetFillingColor();
 280       aBorderColor = myEditedObject->GetBorderColor();
 281     }
 282
 283     myPreviewPrs->setFillingColor( aFillingColor, false, false );
 284     myPreviewPrs->setBorderColor( aBorderColor, false, false );
 285
 286     TopoDS_Face aFace;
 287     if( !aZoneShape.IsNull() )
 288       aFace = TopoDS::Face( aZoneShape );
 289     myPreviewPrs->setFace( aFace, true, true, "" );
 290   }
 291
 292   QApplication::restoreOverrideCursor();
 293 }
 294
 295 void HYDROGUI_LandCoverOp::onAddPolylines()
 296 {
 297   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
 298   if ( !aPanel )
 299     return;
 300
 301   // Add polylines selected in the module browser
 302   Handle(HYDROData_PolylineXY) aPolyXY;
 303   HYDROGUI_ListModel::Object2VisibleList aSelectedPolylines;
 304   HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
 305
 306   if ( aSeq.IsEmpty() || !confirmPolylinesChange() )
 307     return;
 308
 309   for( int anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
 310   {
 311     aPolyXY = Handle(HYDROData_PolylineXY)::DownCast( aSeq.Value( anIndex ));
 312     if (!aPolyXY.IsNull())
 313       aSelectedPolylines.append( HYDROGUI_ListModel::Object2Visible( aPolyXY, true ) );
 314   }
 315
 316   aPanel->includePolylines( aSelectedPolylines );
 317   // TODO: create preview of included polylines
 318 }
 319
 320 void HYDROGUI_LandCoverOp::onRemovePolylines()
 321 {
 322   // Remove selected polylines from the calculation case
 323   HYDROGUI_LandCoverDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverDlg*>( inputPanel() );
 324   if ( !aPanel )
 325     return;
 326
 327   QStringList aSelectedList = aPanel->getSelectedPolylineNames();
 328   if ( aSelectedList.isEmpty() || !confirmPolylinesChange() )
 329     return;
 330
 331   HYDROGUI_ListModel::Object2VisibleList aSelectedPolylines;
 332   for (int i = 0; i < aSelectedList.length(); i++)
 333   {
 334     Handle(HYDROData_PolylineXY) anObject = Handle(HYDROData_PolylineXY)::DownCast(
 335       HYDROGUI_Tool::FindObjectByName( module(), aSelectedList.at(i) ) );
 336     if ( anObject.IsNull() )
 337       continue;
 338
 339     aSelectedPolylines.append( HYDROGUI_ListModel::Object2Visible( anObject, true ) );
 340   }
 341
 342   module()->update( UF_OCCViewer );
 343   aPanel->excludePolylines( aSelectedPolylines );
 344 }
 345
 346 void HYDROGUI_LandCoverOp::closePreview()
 347 {
 348   if( myPreviewPrs )
 349   {
 350     delete myPreviewPrs;
 351     myPreviewPrs = 0;
 352   }
 353 }
 354
 355 bool HYDROGUI_LandCoverOp::confirmPolylinesChange() const
 356 {
 357   if ( myEditedObject.IsNull() )
 358     return true;
 359
 360   // Check if the land cover object is already modified or not
 361   bool isConfirmed = myEditedObject->IsMustBeUpdated();
 362   if ( !isConfirmed )
 363   {
 364     // If not modified check if the land cover has already defined polylines
 365     HYDROData_SequenceOfObjects aSeq = myEditedObject->GetPolylines();
 366     if ( aSeq.Length() > 0 )
 367     {
 368       // If there are already defined polylines then ask a user to confirm land cover recalculation
 369       isConfirmed = ( SUIT_MessageBox::question( module()->getApp()->desktop(),
 370                       tr( "POLYLINES_CHANGED" ),
 371                       tr( "CONFIRM_LAND_COVER_RECALCULATION" ),
 372                       QMessageBox::Yes | QMessageBox::No,
 373                       QMessageBox::No ) == QMessageBox::Yes );
 374     }
 375     else
 376     {
 377       isConfirmed = true; // No polylines - nothing to recalculate
 378     }
 379   }
 380   return isConfirmed;
 381 }