X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROGUI%2FHYDROGUI_Tool.cxx;h=af92511763d2b47ad114139193fead1db1e44637;hb=fce3f1240b6161d7cb837344078e5943b90eca65;hp=6fa0a831763a9d609337a80d3af710112010d522;hpb=84f8b4a57d3cdad820bc1333a5066699cd1c8ae3;p=modules%2Fhydro.git

diff --git a/src/HYDROGUI/HYDROGUI_Tool.cxx b/src/HYDROGUI/HYDROGUI_Tool.cxx
index 6fa0a831..af925117 100644
--- a/src/HYDROGUI/HYDROGUI_Tool.cxx
+++ b/src/HYDROGUI/HYDROGUI_Tool.cxx
@@ -1,12 +1,8 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015  EDF-R&D
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -20,44 +16,29 @@
 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
 
-#include "HYDROGUI_Tool.h"
-
-#include "HYDROGUI_DataModel.h"
-#include "HYDROGUI_DataObject.h"
-#include "HYDROGUI_Module.h"
-#include "HYDROGUI_Prs.h"
-
+#include <HYDROGUI_Tool.h>
 #include <HYDROData_Document.h>
 #include <HYDROData_Iterator.h>
-#include <HYDROData_ImmersibleZone.h>
-#include <HYDROData_Tool.h>
-
-#include <GraphicsView_Viewer.h>
-
-#include <LightApp_Application.h>
-#include <LightApp_DataOwner.h>
-#include <LightApp_SelectionMgr.h>
-
-#include <OCCViewer_ViewModel.h>
-
-#include <QtxWorkstack.h>
-
-#include <STD_TabDesktop.h>
-
-#include <SUIT_Session.h>
-#include <SUIT_Study.h>
-#include <SUIT_ViewManager.h>
-#include <SUIT_ViewWindow.h>
-
+#include <HYDROData_Profile.h>
+#include <TCollection_ExtendedString.hxx>
+#include <TCollection_HAsciiString.hxx>
+#include <TCollection_HExtendedString.hxx>
+#include <Image_PixMap.hxx>
 #include <QDir>
-#include <QFileInfo>
 #include <QDockWidget>
+#include <QFile>
+#include <QFileInfo>
+#include <QLocale>
+#include <QStringList>
 #include <QTextCodec>
+#include <QWidget>
 
 // Definition of this id allows to use 'latin1' (Qt alias for 'ISO-8859-1')
 // encoding instead of default 'System'
 #define USE_LATIN1_ENCODING
 
+// #define DEB_GROUPS 1
+
 QString HYDROGUI_Tool::ToQString( const TCollection_AsciiString& src )
 {
 #ifdef USE_LATIN1_ENCODING
@@ -138,15 +119,6 @@ Handle(TCollection_HExtendedString) HYDROGUI_Tool::ToHExtString( const QString&
   return new TCollection_HExtendedString( ToExtString( src ) );
 }
 
-int HYDROGUI_Tool::GetActiveStudyId()
-{
-  if( SUIT_Session* aSession = SUIT_Session::session() )
-    if( SUIT_Application* anApp = aSession->activeApplication() )
-      if( SUIT_Study* aStudy = anApp->activeStudy() )
-        return aStudy->id();
-  return 0;
-}
-
 QString HYDROGUI_Tool::GetTempDir( const bool theToCreate )
 {
   QString aRes;
@@ -174,7 +146,7 @@ QString HYDROGUI_Tool::GetTempDir( const bool theToCreate )
 #ifdef WNT
     tmpdir = "C:\\";
 #else
-    tmpdir = "/tmp";
+    tmpdir = strdup( "/tmp" );
 #endif
   }
   aRes = tmpdir;
@@ -186,280 +158,87 @@ QString HYDROGUI_Tool::GetTempDir( const bool theToCreate )
   return aRes;
 }
 
-void HYDROGUI_Tool::SetActiveViewManager( HYDROGUI_Module* theModule,
-                                          SUIT_ViewManager* theViewManager )
-{
-  if( theViewManager )
-    if( SUIT_ViewWindow* aViewWindow = theViewManager->getActiveView() )
-      if( STD_TabDesktop* aTabDesktop = dynamic_cast<STD_TabDesktop*>( theModule->getApp()->desktop() ) )
-        if( QtxWorkstack* aWorkstack = aTabDesktop->workstack() )
-          aWorkstack->setActiveWindow( aViewWindow );
-}
-
-void HYDROGUI_Tool::GetPrsSubObjects( HYDROGUI_Module* theModule,
-                                      HYDROData_SequenceOfObjects& theSeq )
+void HYDROGUI_Tool::GetObjectReferences( const Handle(HYDROData_Entity)& theObj,
+                                         HYDROData_SequenceOfObjects& theRefObjects,
+                                         QStringList& theRefNames )
 {
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  if( aDocument.IsNull() )
+  if( theObj.IsNull() )
     return;
 
-  HYDROData_Iterator anIterator( aDocument, KIND_IMAGE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Entity) anObject = anIterator.Current();
-    if( !anObject.IsNull() )
-      theSeq.Append( anObject );
-  }
+  HYDROData_SequenceOfObjects anAllRefObjects = theObj->GetAllReferenceObjects();
+  theRefObjects.Append( anAllRefObjects );
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_POLYLINE );
-  for( ; anIterator.More(); anIterator.Next() )
+  for( int i = 1, n = anAllRefObjects.Length(); i <= n; ++i )
   {
-    Handle(HYDROData_Entity) anObject = anIterator.Current();
-    if( !anObject.IsNull() )
-      theSeq.Append( anObject );
-  }
+    Handle(HYDROData_Entity) aRefObj = theRefObjects.Value( i );
+    if( aRefObj.IsNull() || aRefObj->IsRemoved() )
+      continue;
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_IMMERSIBLE_ZONE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Entity) anObject = anIterator.Current();
-    if( !anObject.IsNull() )
-      theSeq.Append( anObject );
-  }
-}
+    QString aRefObjectName = aRefObj->GetName();
+    if( theRefNames.contains( aRefObjectName ) )
+      continue;
 
-HYDROGUI_Prs* HYDROGUI_Tool::GetPresentation( const Handle(HYDROData_Entity)& theObj,
-                                              const GraphicsView_ObjectList& theObjects )
-{
-  if( !theObj.IsNull() )
-  {
-    GraphicsView_ObjectListIterator anIter( theObjects );
-    while( anIter.hasNext() )
-    {
-      if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
-      {
-        Handle(HYDROData_Entity) anObj = aPrs->getObject();
-        if( IsEqual( anObj, theObj ) )
-          return aPrs;
-      }
-    }
-  }
-  return NULL;
-}
+    theRefObjects.Append( aRefObj );
+    theRefNames.append( aRefObjectName );
 
-GraphicsView_ObjectList HYDROGUI_Tool::GetPrsList( GraphicsView_ViewPort* theViewPort )
-{
-  GraphicsView_ObjectList aList;
-  if( theViewPort )
-  {
-    GraphicsView_ObjectListIterator anIter( theViewPort->getObjects() );
-    while( anIter.hasNext() )
-      if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
-        aList.append( aPrs );
+    GetObjectReferences( aRefObj, theRefObjects, theRefNames );
   }
-  return aList;
 }
 
-HYDROData_SequenceOfObjects HYDROGUI_Tool::GetSelectedObjects( HYDROGUI_Module* theModule )
+HYDROData_SequenceOfObjects HYDROGUI_Tool::GetObjectBackReferences( const Handle(HYDROData_Entity)& theObj )
 {
-  HYDROData_SequenceOfObjects aSeq;
-
-  HYDROGUI_DataModel* aModel = theModule->getDataModel();
+  if( theObj.IsNull() )
+    return HYDROData_SequenceOfObjects();
 
-  SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
-  SUIT_DataOwnerPtrList anOwners;
-  aSelectionMgr->selected( anOwners );
+  QString anObjName = theObj->GetName();
 
-  QStringList aCollectedNameList; // to avoid duplication
-  foreach( SUIT_DataOwner* aSUITOwner, anOwners )
-  {
-    if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( aSUITOwner ) )
-    {
-      Handle(HYDROData_Entity) anObject = aModel->objectByEntry( anOwner->entry() );
-      if( !anObject.IsNull() )
-      {
-        QString aName = anObject->GetName();
-        if( !aCollectedNameList.contains( aName ) )
-        {
-          aSeq.Append( anObject );
-          aCollectedNameList.append( aName );
-        }
-      }
-    }
-  }
-  return aSeq;
-}
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document( theObj->Label() );
+  QMap<QString,HYDROData_SequenceOfObjects> aMapOfBackRefs =
+    GetObjectsBackReferences( aDoc, QStringList() << anObjName );
 
-Handle(HYDROData_Entity) HYDROGUI_Tool::GetSelectedObject( HYDROGUI_Module* theModule )
-{
-  HYDROData_SequenceOfObjects aSeq = GetSelectedObjects( theModule );
-  if( !aSeq.IsEmpty() )
-    return aSeq.First();
-  return NULL;
+  return aMapOfBackRefs[ anObjName ];
 }
 
-ObjectKind HYDROGUI_Tool::GetSelectedPartition( HYDROGUI_Module* theModule )
+QMap<QString,HYDROData_SequenceOfObjects> HYDROGUI_Tool::GetObjectsBackReferences(
+  const Handle(HYDROData_Document)& theDocument, const QStringList& theObjectNames )
 {
-  HYDROGUI_DataModel* aModel = theModule->getDataModel();
+  QMap<QString,HYDROData_SequenceOfObjects> aResMap;
 
-  SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
-  SUIT_DataOwnerPtrList anOwners;
-  aSelectionMgr->selected( anOwners );
+  if( theObjectNames.isEmpty() )
+    return aResMap;
 
-  if( anOwners.size() != 1 )
-    return KIND_UNKNOWN;
+  if( theDocument.IsNull() )
+    return aResMap;
 
-  if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( anOwners.first().operator->() ) )
+  HYDROData_Iterator anIterator( theDocument );
+  for( ; anIterator.More(); anIterator.Next() )
   {
-    QString anEntry = anOwner->entry();
-    QString aPrefix = HYDROGUI_DataObject::entryPrefix();
-    if( anEntry.left( aPrefix.length() ) == aPrefix )
-    {
-      anEntry.remove( aPrefix );
-      for( ObjectKind anObjectKind = KIND_UNKNOWN + 1; anObjectKind <= KIND_LAST; anObjectKind++ )
-        if( HYDROGUI_DataModel::partitionName( anObjectKind ) == anEntry )
-          return anObjectKind;
-    }
-  }
-  return KIND_UNKNOWN;
-}
-
-Handle(HYDROData_Entity) HYDROGUI_Tool::FindObjectByName( HYDROGUI_Module* theModule,
-                                                          const QString&   theName,
-                                                          const ObjectKind theObjectKind )
-{
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  return HYDROData_Tool::FindObjectByName( aDocument, theName, theObjectKind );
-}
-
-HYDROData_SequenceOfObjects HYDROGUI_Tool::FindObjectsByNames( HYDROGUI_Module*   theModule,
-                                                               const QStringList& theNames,
-                                                               const ObjectKind   theObjectKind )
-{
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  return HYDROData_Tool::FindObjectsByNames( aDocument, theNames, theObjectKind );
-}
-
-QString HYDROGUI_Tool::GenerateObjectName( HYDROGUI_Module*   theModule,
-                                           const QString&     thePrefix,
-                                           const QStringList& theUsedNames )
-{
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  return HYDROData_Tool::GenerateObjectName( aDocument, thePrefix, theUsedNames );
-}
-
-size_t HYDROGUI_Tool::GetActiveViewId( HYDROGUI_Module* theModule,
-                                       const QString&   theViewId )
-{
-  size_t aViewId = 0;
-  SUIT_ViewManager* aViewMgr = theModule->getApp()->activeViewManager();
-  if( !aViewMgr || ( !theViewId.isEmpty() && aViewMgr->getType() != theViewId ) )
-    return aViewId;
-
-  if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
-    aViewId = (size_t)aViewer;
-  return aViewId;
-}
-
-size_t HYDROGUI_Tool::GetActiveGraphicsViewId( HYDROGUI_Module* theModule )
-{
-  return GetActiveViewId( theModule, GraphicsView_Viewer::Type() );
-}
+    Handle(HYDROData_Entity) anObject = anIterator.Current();
+    if( anObject.IsNull() || anObject->IsRemoved() )
+      continue;
 
-size_t HYDROGUI_Tool::GetActiveOCCViewId( HYDROGUI_Module* theModule )
-{
-  return GetActiveViewId( theModule, OCCViewer_Viewer::Type() );
-}
+    QString anObjectName = anObject->GetName();
+    if ( theObjectNames.contains( anObjectName ) )
+      continue;
 
-QList<size_t> getViewIdList( HYDROGUI_Module* theModule,
-                             const QString&   theViewId )
-{
-  QList<size_t> aList;
-  ViewManagerList aViewMgrs;
-  theModule->getApp()->viewManagers( theViewId, aViewMgrs );
-  QListIterator<SUIT_ViewManager*> anIter( aViewMgrs );
-  while( anIter.hasNext() )
-  {
-    if( SUIT_ViewManager* aViewMgr = anIter.next() )
+    HYDROData_SequenceOfObjects aRefObjects = anObject->GetAllReferenceObjects();
+    for ( int i = 1, n = aRefObjects.Length(); i <= n; ++i )
     {
-      if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
-        aList.append( (size_t)aViewer );
-    }
-  }
-  return aList;
-}
-
-QList<size_t> HYDROGUI_Tool::GetGraphicsViewIdList( HYDROGUI_Module* theModule )
-{
-  return getViewIdList( theModule, GraphicsView_Viewer::Type() );
-}
-
-QList<size_t> HYDROGUI_Tool::GetOCCViewIdList( HYDROGUI_Module* theModule )
-{
-  return getViewIdList( theModule, OCCViewer_Viewer::Type() );
-}
+      Handle(HYDROData_Entity) aRefObject = aRefObjects.Value( i );
+      if( aRefObject.IsNull() || aRefObject->IsRemoved() )
+        continue;
 
-void HYDROGUI_Tool::GetObjectReferences( const Handle(HYDROData_Image)& theImage,
-                                         HYDROData_SequenceOfObjects& theRefObjects,
-                                         QStringList& theRefNames )
-{
-  if( theImage.IsNull() )
-    return;
+      QString aRefObjectName = aRefObject->GetName();
+      if ( !theObjectNames.contains( aRefObjectName ) )
+        continue;
 
-  for( int anIndex = 0, aNbRef = theImage->NbReferences(); anIndex < aNbRef; anIndex++ )
-  {
-    Handle(HYDROData_Entity) aRefObj = theImage->Reference( anIndex );
-    if( !aRefObj.IsNull() && !aRefObj->IsRemoved() )
-    {
-      QString aName = aRefObj->GetName();
-      if( !theRefNames.contains( aName ) )
-      {
-        theRefObjects.Append( aRefObj );
-        theRefNames.append( aRefObj->GetName() );
-        if( aRefObj->GetKind() == KIND_IMAGE )
-        {
-          Handle(HYDROData_Image) aRefImage = Handle(HYDROData_Image)::DownCast( aRefObj );
-          if( !aRefImage.IsNull() )
-            GetObjectReferences( aRefImage, theRefObjects, theRefNames );
-        }
-      }
+      aResMap[ aRefObjectName ].Append( anObject );
     }
   }
-}
-
-void HYDROGUI_Tool::GetObjectBackReferences( HYDROGUI_Module* theModule,
-                                             const Handle(HYDROData_Entity)& theObj,
-                                             HYDROData_SequenceOfObjects& theBackRefObjects,
-                                             QStringList& theBackRefNames )
-{
-  if( theObj.IsNull() )
-    return;
-
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  if( aDocument.IsNull() )
-    return;
 
-  QString aName = theObj->GetName();
-
-  HYDROData_Iterator anIterator( aDocument, KIND_IMAGE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Image) anImage = Handle(HYDROData_Image)::DownCast( anIterator.Current() );
-    if( !anImage.IsNull() )
-    {
-      HYDROData_SequenceOfObjects aRefObjects;
-      QStringList aRefNames;
-      GetObjectReferences( anImage, aRefObjects, aRefNames );
-      if( aRefNames.contains( aName ) )
-      {
-        theBackRefObjects.Append( anImage );
-        theBackRefNames.append( anImage->GetName() );
-      }
-    }
-  }
+  return aResMap;
 }
 
-
 QDockWidget* HYDROGUI_Tool::WindowDock( QWidget* wid )
 {
   if ( !wid )
@@ -475,42 +254,88 @@ QDockWidget* HYDROGUI_Tool::WindowDock( QWidget* wid )
   return dock;
 }
 
-QColor HYDROGUI_Tool::GenerateFillingColor( HYDROGUI_Module*   theModule,
-                                            const QStringList& theZoneNames )
+QStringList HYDROGUI_Tool::FindExistingObjectsNames( const Handle(HYDROData_Document)& theDoc, 
+                                                     const ObjectKind theObjectKind,
+                                                     bool isCheckValidProfile )
 {
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
-  return GenerateFillingColor( aDocument, theZoneNames );
-}
+  QStringList aNames;
 
-QColor HYDROGUI_Tool::GenerateFillingColor( const Handle(HYDROData_Document)& theDoc,
-                                            const QStringList&                theZoneNames )
-{
-  QColor aFillingColor( HYDROData_ImmersibleZone::DefaultFillingColor() );
+  HYDROData_Iterator anIter( theDoc, theObjectKind );
+  for ( ; anIter.More(); anIter.Next() ) {
+    Handle(HYDROData_Entity) anObject = anIter.Current();
 
-  int aCounter = 0;
-  int aR = 0, aG = 0, aB = 0;
-  QStringListIterator aZoneNameIter( theZoneNames );
-  while( aZoneNameIter.hasNext() )
-  {
-    const QString& aZoneName = aZoneNameIter.next();
-    Handle(HYDROData_ImmersibleZone) aRefZone = 
-      Handle(HYDROData_ImmersibleZone)::DownCast(
-        HYDROData_Tool::FindObjectByName( theDoc, aZoneName, KIND_IMMERSIBLE_ZONE ) );
-    if( !aRefZone.IsNull() )
+    bool isOK = !anObject.IsNull();
+    
+    if( isOK && isCheckValidProfile )
     {
-      QColor aRefColor = aRefZone->GetFillingColor();
-      aR += aRefColor.red();
-      aG += aRefColor.green();
-      aB += aRefColor.blue();
-      aCounter++;
+      Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( anObject );
+      if( !aProfile.IsNull() && !aProfile->IsValid() )
+        isOK = false;
     }
+      
+    if( isOK )
+      aNames.append( anObject->GetName() );
   }
-  
-  if( aCounter > 0 )
+
+  return aNames;
+}
+
+QString HYDROGUI_Tool::GetCoordinateString( const double theNumber, bool isInLocale )
+{
+  if( isInLocale )
   {
-    aFillingColor = QColor( aR / aCounter, aG / aCounter, aB / aCounter );
+    static QLocale aLocale( QLocale::English, QLocale::France );
+    return aLocale.toString( theNumber, 'f', 2 );
   }
+  else
+    return QString::number( theNumber, 'f', 2 );
+}
+
+Handle(Image_PixMap) HYDROGUI_Tool::Pixmap( const QImage& theImage )
+{
+    Handle(Image_PixMap) pix;
+    if ( theImage.isNull() || theImage.format() == QImage::Format_Invalid )
+        return pix;
+
+    Handle(Image_PixMap) tmpPix = new Image_PixMap();
+    tmpPix->SetTopDown( false );
+    QImage anImage = theImage.mirrored();
+    if ( !anImage.hasAlphaChannel() && anImage.allGray() )
+    {
+        tmpPix->InitTrash( Image_PixMap::ImgGray, anImage.width(), anImage.height(), anImage.width() );
+        for ( int r = 0; r < anImage.height(); r++ )
+        {
+            Standard_Byte* aRowData = tmpPix->ChangeRow( anImage.height() - r - 1 );
+            for  ( int p = 0; p < anImage.width(); p++ )
+                aRowData[p] = qRed( anImage.pixel( p, r ) );
+        }
+    }
+    else
+    {
+        Image_PixMap::ImgFormat aFormat;
+        if ( anImage.hasAlphaChannel() )
+        {
+            if ( anImage.format() != QImage::Format_ARGB32 )
+                anImage = anImage.convertToFormat( QImage::Format_ARGB32 );
+            aFormat = Image_PixMap::ImgRGBA;
+        }
+        else
+        {
+            if ( anImage.format() != QImage::Format_RGB888 )
+                anImage = anImage.convertToFormat( QImage::Format_RGB888 );
+            aFormat = Image_PixMap::ImgRGB;
+        }
+
+        tmpPix->InitWrapper( aFormat, (Standard_Byte*)anImage.bits(), anImage.width(), anImage.height(), anImage.bytesPerLine() );
+    }
+
+    if ( !tmpPix.IsNull() )
+    {
+        pix = new Image_PixMap();
+        pix->InitCopy( *tmpPix.operator->() );
+        pix->SetTopDown( tmpPix->IsTopDown() );
+    }
 
-  return aFillingColor;
+    return pix;
 }