src/HYDROGUI/HYDROGUI_Tool.cxx

   1 // Copyright (C) 2007-2013  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.
  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_Tool.h"
  24
  25 #include "HYDROGUI_DataModel.h"
  26 #include "HYDROGUI_DataObject.h"
  27 #include "HYDROGUI_Module.h"
  28 #include "HYDROGUI_Prs.h"
  29
  30 #include <HYDROData_Document.h>
  31 #include <HYDROData_Iterator.h>
  32 #include <HYDROData_ImmersibleZone.h>
  33 #include <HYDROData_Tool.h>
  34
  35 #include <GraphicsView_Viewer.h>
  36
  37 #include <LightApp_Application.h>
  38 #include <LightApp_DataOwner.h>
  39 #include <LightApp_SelectionMgr.h>
  40
  41 #include <OCCViewer_ViewModel.h>
  42
  43 #include <QtxWorkstack.h>
  44
  45 #include <STD_TabDesktop.h>
  46
  47 #include <SUIT_Session.h>
  48 #include <SUIT_Study.h>
  49 #include <SUIT_ViewManager.h>
  50 #include <SUIT_ViewWindow.h>
  51
  52 #include <QDir>
  53 #include <QFileInfo>
  54 #include <QDockWidget>
  55 #include <QTextCodec>
  56
  57 // Definition of this id allows to use 'latin1' (Qt alias for 'ISO-8859-1')
  58 // encoding instead of default 'System'
  59 #define USE_LATIN1_ENCODING
  60
  61 QString HYDROGUI_Tool::ToQString( const TCollection_AsciiString& src )
  62 {
  63 #ifdef USE_LATIN1_ENCODING
  64   QTextCodec* codec = QTextCodec::codecForName( "latin1" ); // alias for ISO-8859-1
  65 #else
  66   QTextCodec* codec = QTextCodec::codecForLocale();
  67 #endif
  68   QString res;
  69   if ( !src.IsEmpty() )
  70     res = codec ? codec->toUnicode( (char*)src.ToCString(), src.Length() ) :
  71       QString( (char*)src.ToCString() );
  72   return res;
  73 }
  74
  75 QString HYDROGUI_Tool::ToQString( const TCollection_ExtendedString& src )
  76 {
  77   return QString( (QChar*)src.ToExtString(), src.Length() );
  78 }
  79
  80 QString HYDROGUI_Tool::ToQString( const Handle(TCollection_HAsciiString)& src )
  81 {
  82   if( src.IsNull() )
  83     return QString();
  84   else
  85     return ToQString( src->String() );
  86 }
  87
  88 QString HYDROGUI_Tool::ToQString( const Handle(TCollection_HExtendedString)& src )
  89 {
  90   if( src.IsNull() )
  91     return QString();
  92   return ToQString( src->String() );
  93 }
  94
  95 TCollection_AsciiString HYDROGUI_Tool::ToAsciiString( const QString& src )
  96 {
  97   TCollection_AsciiString res;
  98   if( !src.isNull() )
  99   {
 100 #ifdef USE_LATIN1_ENCODING
 101     QTextCodec* codec = QTextCodec::codecForName( "latin1" ); // alias for ISO-8859-1
 102 #else
 103     QTextCodec* codec = QTextCodec::codecForLocale();
 104 #endif
 105     if( codec )
 106     {
 107       QByteArray str = codec->fromUnicode( src );
 108       res = TCollection_AsciiString( (Standard_CString)str.constData() );
 109     }
 110     else
 111       res = TCollection_AsciiString( src.toLatin1().data() );
 112   }
 113   return res;
 114 }
 115
 116 TCollection_ExtendedString HYDROGUI_Tool::ToExtString( const QString& src )
 117 {
 118   if( src.isEmpty() )
 119     return TCollection_ExtendedString();
 120
 121   Standard_Integer len = src.length();
 122   Standard_ExtString extStr = new Standard_ExtCharacter[ ( len + 1 ) * 2 ];
 123   memcpy( (void*)extStr, src.unicode(), len * 2 );
 124   ((short*)extStr)[ len ] = 0;
 125
 126   TCollection_ExtendedString trg( extStr );
 127   delete [] extStr;
 128   return trg;
 129 }
 130
 131 Handle(TCollection_HAsciiString) HYDROGUI_Tool::ToHAsciiString( const QString& src )
 132 {
 133   return new TCollection_HAsciiString( ToAsciiString( src ) );
 134 }
 135
 136 Handle(TCollection_HExtendedString) HYDROGUI_Tool::ToHExtString( const QString& src )
 137 {
 138   return new TCollection_HExtendedString( ToExtString( src ) );
 139 }
 140
 141 int HYDROGUI_Tool::GetActiveStudyId()
 142 {
 143   if( SUIT_Session* aSession = SUIT_Session::session() )
 144     if( SUIT_Application* anApp = aSession->activeApplication() )
 145       if( SUIT_Study* aStudy = anApp->activeStudy() )
 146         return aStudy->id();
 147   return 0;
 148 }
 149
 150 QString HYDROGUI_Tool::GetTempDir( const bool theToCreate )
 151 {
 152   QString aRes;
 153
 154   char* tmpdir = getenv ( "HYDRO_TMP_DIR" );
 155   if ( tmpdir )
 156   {
 157     // try to create folder if it does not exist
 158     QFileInfo fi( tmpdir );
 159     if ( !fi.exists() && theToCreate )
 160     {
 161       if ( QDir().mkdir( tmpdir ) )
 162         QFile::setPermissions( tmpdir, (QFile::Permissions)0x4FFFF );
 163        QFileInfo fi( tmpdir );
 164        if ( !fi.exists() || !fi.isWritable() )
 165          tmpdir = 0;
 166     }
 167   }
 168   if ( !tmpdir )
 169     tmpdir = getenv ( "TEMP" );
 170   if ( !tmpdir )
 171     tmpdir = getenv ( "TMP" );
 172   if ( !tmpdir )
 173   {
 174 #ifdef WNT
 175     tmpdir = "C:\\";
 176 #else
 177     tmpdir = "/tmp";
 178 #endif
 179   }
 180   aRes = tmpdir;
 181
 182   QFileInfo fi( aRes );
 183   if ( !fi.exists() || !fi.isWritable() )
 184     aRes = QString::null;
 185
 186   return aRes;
 187 }
 188
 189 void HYDROGUI_Tool::SetActiveViewManager( HYDROGUI_Module* theModule,
 190                                           SUIT_ViewManager* theViewManager )
 191 {
 192   if( theViewManager )
 193     if( SUIT_ViewWindow* aViewWindow = theViewManager->getActiveView() )
 194       if( STD_TabDesktop* aTabDesktop = dynamic_cast<STD_TabDesktop*>( theModule->getApp()->desktop() ) )
 195         if( QtxWorkstack* aWorkstack = aTabDesktop->workstack() )
 196           aWorkstack->setActiveWindow( aViewWindow );
 197 }
 198
 199 void HYDROGUI_Tool::GetPrsSubObjects( HYDROGUI_Module* theModule,
 200                                       HYDROData_SequenceOfObjects& theSeq )
 201 {
 202   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 203   if( aDocument.IsNull() )
 204     return;
 205
 206   HYDROData_Iterator anIterator( aDocument, KIND_IMAGE );
 207   for( ; anIterator.More(); anIterator.Next() )
 208   {
 209     Handle(HYDROData_Entity) anObject = anIterator.Current();
 210     if( !anObject.IsNull() )
 211       theSeq.Append( anObject );
 212   }
 213
 214   anIterator = HYDROData_Iterator( aDocument, KIND_POLYLINE );
 215   for( ; anIterator.More(); anIterator.Next() )
 216   {
 217     Handle(HYDROData_Entity) anObject = anIterator.Current();
 218     if( !anObject.IsNull() )
 219       theSeq.Append( anObject );
 220   }
 221
 222   anIterator = HYDROData_Iterator( aDocument, KIND_IMMERSIBLE_ZONE );
 223   for( ; anIterator.More(); anIterator.Next() )
 224   {
 225     Handle(HYDROData_Entity) anObject = anIterator.Current();
 226     if( !anObject.IsNull() )
 227       theSeq.Append( anObject );
 228   }
 229 }
 230
 231 HYDROGUI_Prs* HYDROGUI_Tool::GetPresentation( const Handle(HYDROData_Entity)& theObj,
 232                                               const GraphicsView_ObjectList& theObjects )
 233 {
 234   if( !theObj.IsNull() )
 235   {
 236     GraphicsView_ObjectListIterator anIter( theObjects );
 237     while( anIter.hasNext() )
 238     {
 239       if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
 240       {
 241         Handle(HYDROData_Entity) anObj = aPrs->getObject();
 242         if( IsEqual( anObj, theObj ) )
 243           return aPrs;
 244       }
 245     }
 246   }
 247   return NULL;
 248 }
 249
 250 GraphicsView_ObjectList HYDROGUI_Tool::GetPrsList( GraphicsView_ViewPort* theViewPort )
 251 {
 252   GraphicsView_ObjectList aList;
 253   if( theViewPort )
 254   {
 255     GraphicsView_ObjectListIterator anIter( theViewPort->getObjects() );
 256     while( anIter.hasNext() )
 257       if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
 258         aList.append( aPrs );
 259   }
 260   return aList;
 261 }
 262
 263 HYDROData_SequenceOfObjects HYDROGUI_Tool::GetSelectedObjects( HYDROGUI_Module* theModule )
 264 {
 265   HYDROData_SequenceOfObjects aSeq;
 266
 267   HYDROGUI_DataModel* aModel = theModule->getDataModel();
 268
 269   SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
 270   SUIT_DataOwnerPtrList anOwners;
 271   aSelectionMgr->selected( anOwners );
 272
 273   QStringList aCollectedNameList; // to avoid duplication
 274   foreach( SUIT_DataOwner* aSUITOwner, anOwners )
 275   {
 276     if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( aSUITOwner ) )
 277     {
 278       Handle(HYDROData_Entity) anObject = aModel->objectByEntry( anOwner->entry() );
 279       if( !anObject.IsNull() )
 280       {
 281         QString aName = anObject->GetName();
 282         if( !aCollectedNameList.contains( aName ) )
 283         {
 284           aSeq.Append( anObject );
 285           aCollectedNameList.append( aName );
 286         }
 287       }
 288     }
 289   }
 290   return aSeq;
 291 }
 292
 293 Handle(HYDROData_Entity) HYDROGUI_Tool::GetSelectedObject( HYDROGUI_Module* theModule )
 294 {
 295   HYDROData_SequenceOfObjects aSeq = GetSelectedObjects( theModule );
 296   if( !aSeq.IsEmpty() )
 297     return aSeq.First();
 298   return NULL;
 299 }
 300
 301 HYDROData_SequenceOfObjects HYDROGUI_Tool::GetGeometryObjects(
 302   HYDROGUI_Module* theModule )
 303 {
 304   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 305
 306   HYDROData_SequenceOfObjects aResSeq;
 307
 308   HYDROData_Iterator anIter( aDocument );
 309   for ( ; anIter.More(); anIter.Next() )
 310   {
 311     Handle(HYDROData_Entity) anObj = anIter.Current();
 312     if ( !HYDROData_Tool::IsGeometryObject( anObj ) )
 313       continue;
 314
 315     aResSeq.Append( anObj );
 316   }
 317
 318   return aResSeq;
 319 }
 320
 321 ObjectKind HYDROGUI_Tool::GetSelectedPartition( HYDROGUI_Module* theModule )
 322 {
 323   HYDROGUI_DataModel* aModel = theModule->getDataModel();
 324
 325   SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
 326   SUIT_DataOwnerPtrList anOwners;
 327   aSelectionMgr->selected( anOwners );
 328
 329   if( anOwners.size() != 1 )
 330     return KIND_UNKNOWN;
 331
 332   if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( anOwners.first().operator->() ) )
 333   {
 334     QString anEntry = anOwner->entry();
 335     QString aPrefix = HYDROGUI_DataObject::entryPrefix();
 336     if( anEntry.left( aPrefix.length() ) == aPrefix )
 337     {
 338       anEntry.remove( aPrefix );
 339       for( ObjectKind anObjectKind = KIND_UNKNOWN + 1; anObjectKind <= KIND_LAST; anObjectKind++ )
 340         if( HYDROGUI_DataModel::tr( HYDROGUI_DataModel::partitionName( anObjectKind ).toAscii() ) == anEntry )
 341           return anObjectKind;
 342     }
 343   }
 344   return KIND_UNKNOWN;
 345 }
 346
 347 Handle(HYDROData_Entity) HYDROGUI_Tool::FindObjectByName( HYDROGUI_Module* theModule,
 348                                                           const QString&   theName,
 349                                                           const ObjectKind theObjectKind )
 350 {
 351   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 352   return HYDROData_Tool::FindObjectByName( aDocument, theName, theObjectKind );
 353 }
 354
 355 HYDROData_SequenceOfObjects HYDROGUI_Tool::FindObjectsByNames( HYDROGUI_Module*   theModule,
 356                                                                const QStringList& theNames,
 357                                                                const ObjectKind   theObjectKind )
 358 {
 359   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 360   return HYDROData_Tool::FindObjectsByNames( aDocument, theNames, theObjectKind );
 361 }
 362
 363 QString HYDROGUI_Tool::GenerateObjectName( HYDROGUI_Module*   theModule,
 364                                            const QString&     thePrefix,
 365                                            const QStringList& theUsedNames )
 366 {
 367   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 368   return HYDROData_Tool::GenerateObjectName( aDocument, thePrefix, theUsedNames );
 369 }
 370
 371 size_t HYDROGUI_Tool::GetActiveViewId( HYDROGUI_Module* theModule,
 372                                        const QString&   theViewId )
 373 {
 374   size_t aViewId = 0;
 375   SUIT_ViewManager* aViewMgr = theModule->getApp()->activeViewManager();
 376   if( !aViewMgr || ( !theViewId.isEmpty() && aViewMgr->getType() != theViewId ) )
 377     return aViewId;
 378
 379   if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
 380     aViewId = (size_t)aViewer;
 381   return aViewId;
 382 }
 383
 384 size_t HYDROGUI_Tool::GetActiveGraphicsViewId( HYDROGUI_Module* theModule )
 385 {
 386   return GetActiveViewId( theModule, GraphicsView_Viewer::Type() );
 387 }
 388
 389 size_t HYDROGUI_Tool::GetActiveOCCViewId( HYDROGUI_Module* theModule )
 390 {
 391   return GetActiveViewId( theModule, OCCViewer_Viewer::Type() );
 392 }
 393
 394 QList<size_t> getViewIdList( HYDROGUI_Module* theModule,
 395                              const QString&   theViewId )
 396 {
 397   QList<size_t> aList;
 398   ViewManagerList aViewMgrs;
 399   theModule->getApp()->viewManagers( theViewId, aViewMgrs );
 400   QListIterator<SUIT_ViewManager*> anIter( aViewMgrs );
 401   while( anIter.hasNext() )
 402   {
 403     if( SUIT_ViewManager* aViewMgr = anIter.next() )
 404     {
 405       if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
 406         aList.append( (size_t)aViewer );
 407     }
 408   }
 409   return aList;
 410 }
 411
 412 QList<size_t> HYDROGUI_Tool::GetGraphicsViewIdList( HYDROGUI_Module* theModule )
 413 {
 414   return getViewIdList( theModule, GraphicsView_Viewer::Type() );
 415 }
 416
 417 QList<size_t> HYDROGUI_Tool::GetOCCViewIdList( HYDROGUI_Module* theModule )
 418 {
 419   return getViewIdList( theModule, OCCViewer_Viewer::Type() );
 420 }
 421
 422 void HYDROGUI_Tool::GetObjectReferences( const Handle(HYDROData_Image)& theImage,
 423                                          HYDROData_SequenceOfObjects& theRefObjects,
 424                                          QStringList& theRefNames )
 425 {
 426   if( theImage.IsNull() )
 427     return;
 428
 429   for( int anIndex = 0, aNbRef = theImage->NbReferences(); anIndex < aNbRef; anIndex++ )
 430   {
 431     Handle(HYDROData_Entity) aRefObj = theImage->Reference( anIndex );
 432     if( !aRefObj.IsNull() && !aRefObj->IsRemoved() )
 433     {
 434       QString aName = aRefObj->GetName();
 435       if( !theRefNames.contains( aName ) )
 436       {
 437         theRefObjects.Append( aRefObj );
 438         theRefNames.append( aRefObj->GetName() );
 439         if( aRefObj->GetKind() == KIND_IMAGE )
 440         {
 441           Handle(HYDROData_Image) aRefImage = Handle(HYDROData_Image)::DownCast( aRefObj );
 442           if( !aRefImage.IsNull() )
 443             GetObjectReferences( aRefImage, theRefObjects, theRefNames );
 444         }
 445       }
 446     }
 447   }
 448 }
 449
 450 void HYDROGUI_Tool::GetObjectBackReferences( HYDROGUI_Module* theModule,
 451                                              const Handle(HYDROData_Entity)& theObj,
 452                                              HYDROData_SequenceOfObjects& theBackRefObjects,
 453                                              QStringList& theBackRefNames )
 454 {
 455   if( theObj.IsNull() )
 456     return;
 457
 458   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 459   if( aDocument.IsNull() )
 460     return;
 461
 462   QString aName = theObj->GetName();
 463
 464   HYDROData_Iterator anIterator( aDocument, KIND_IMAGE );
 465   for( ; anIterator.More(); anIterator.Next() )
 466   {
 467     Handle(HYDROData_Image) anImage = Handle(HYDROData_Image)::DownCast( anIterator.Current() );
 468     if( !anImage.IsNull() )
 469     {
 470       HYDROData_SequenceOfObjects aRefObjects;
 471       QStringList aRefNames;
 472       GetObjectReferences( anImage, aRefObjects, aRefNames );
 473       if( aRefNames.contains( aName ) )
 474       {
 475         theBackRefObjects.Append( anImage );
 476         theBackRefNames.append( anImage->GetName() );
 477       }
 478     }
 479   }
 480 }
 481
 482
 483 QDockWidget* HYDROGUI_Tool::WindowDock( QWidget* wid )
 484 {
 485   if ( !wid )
 486     return 0;
 487
 488   QDockWidget* dock = 0;
 489   QWidget* w = wid->parentWidget();
 490   while ( w && !dock )
 491   {
 492     dock = ::qobject_cast<QDockWidget*>( w );
 493     w = w->parentWidget();
 494   }
 495   return dock;
 496 }
 497
 498 QColor HYDROGUI_Tool::GenerateFillingColor( HYDROGUI_Module*   theModule,
 499                                             const QStringList& theZoneNames )
 500 {
 501   Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
 502   return GenerateFillingColor( aDocument, theZoneNames );
 503 }
 504
 505 QColor HYDROGUI_Tool::GenerateFillingColor( const Handle(HYDROData_Document)& theDoc,
 506                                             const QStringList&                theZoneNames )
 507 {
 508   QColor aFillingColor( HYDROData_ImmersibleZone::DefaultFillingColor() );
 509
 510   int aCounter = 0;
 511   int aR = 0, aG = 0, aB = 0;
 512   QStringListIterator aZoneNameIter( theZoneNames );
 513   while( aZoneNameIter.hasNext() )
 514   {
 515     const QString& aZoneName = aZoneNameIter.next();
 516     Handle(HYDROData_ImmersibleZone) aRefZone =
 517       Handle(HYDROData_ImmersibleZone)::DownCast(
 518         HYDROData_Tool::FindObjectByName( theDoc, aZoneName, KIND_IMMERSIBLE_ZONE ) );
 519     if( !aRefZone.IsNull() )
 520     {
 521       QColor aRefColor = aRefZone->GetFillingColor();
 522       aR += aRefColor.red();
 523       aG += aRefColor.green();
 524       aB += aRefColor.blue();
 525       aCounter++;
 526     }
 527   }
 528
 529   if( aCounter > 0 )
 530   {
 531     aFillingColor = QColor( aR / aCounter, aG / aCounter, aB / aCounter );
 532   }
 533
 534   return aFillingColor;
 535 }
 536