src/GEOMBase/GEOMBase.cxx

   1 // Copyright (C) 2007-2012  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 // GEOM GEOMGUI : GUI for Geometry component
  24 // File   : GEOMBase.cxx
  25 // Author : Damien COQUERET, Open CASCADE S.A.S.
  26 //
  27 #include "GEOMBase.h"
  28
  29 #include <GeometryGUI.h>
  30 #include <GEOM_Client.hxx>
  31
  32 ////  SALOME Includes
  33 #include <GEOM_Actor.h>
  34 #include <SVTK_ViewModel.h>
  35 #include <SVTK_ViewWindow.h>
  36 #include <OCCViewer_ViewPort3d.h>
  37 #include <OCCViewer_ViewModel.h>
  38 #include <OCCViewer_ViewWindow.h>
  39 #include <SOCC_ViewModel.h>
  40 #include <SOCC_Prs.h>
  41
  42 #include <SALOME_ListIO.hxx>
  43 #include <SALOME_ListIteratorOfListIO.hxx>
  44
  45 #include <SUIT_Desktop.h>
  46 #include <SUIT_Session.h>
  47 #include <SUIT_ViewManager.h>
  48 #include <SUIT_ViewWindow.h>
  49 #include <SUIT_MessageBox.h>
  50 #include <SalomeApp_Application.h>
  51 #include <SalomeApp_Study.h>
  52
  53 // // Open CASCADE Includes
  54 #include <BRep_Tool.hxx>
  55 #include <BRepAdaptor_Curve.hxx>
  56 #include <BRepAdaptor_Surface.hxx>
  57 #include <BRepPrimAPI_MakeCone.hxx>
  58
  59 #include <AIS_ListIteratorOfListOfInteractive.hxx>
  60 #include <AIS_ListOfInteractive.hxx>
  61
  62 #include <TopAbs.hxx>
  63 #include <TopExp.hxx>
  64 #include <TopTools_IndexedMapOfShape.hxx>
  65 #include <TopoDS.hxx>
  66
  67 #include <Precision.hxx>
  68
  69 #include <vtkRenderer.h>
  70
  71 #include <set>
  72
  73 //=====================================================================================
  74 // function : GetShapeFromIOR()
  75 // purpose  : Get shape data by the specified IOR
  76 //=====================================================================================
  77 TopoDS_Shape GEOMBase::GetShapeFromIOR( const QString& IOR )
  78 {
  79   GEOM::GEOM_Object_var geomObj = GEOMBase::GetObjectFromIOR( IOR );
  80   TopoDS_Shape shape;
  81   GetShape( geomObj, shape, TopAbs_SHAPE );
  82   return shape;
  83 }
  84
  85
  86 //=====================================================================================
  87 // function : GetIndex()
  88 // purpose  : Get the index of a sub-shape in a main shape : index start at 1
  89 //=====================================================================================
  90 int GEOMBase::GetIndex( const TopoDS_Shape& subshape, const TopoDS_Shape& shape )
  91 {
  92   int idx = -1;
  93   if ( !shape.IsNull() && !subshape.IsNull() ) {
  94     TopTools_IndexedMapOfShape anIndices;
  95     TopExp::MapShapes( shape, anIndices );
  96     if ( anIndices.Contains( subshape ) )
  97       idx = anIndices.FindIndex( subshape );
  98   }
  99   return idx;
 100 }
 101
 102
 103 //=======================================================================
 104 // function : GetTopoFromSelection()
 105 // purpose  : Define tds from a single selection and retuen true
 106 //=======================================================================
 107 TopoDS_Shape GEOMBase::GetTopoFromSelection( const SALOME_ListIO& IObjects )
 108 {
 109   TopoDS_Shape shape;
 110   if ( IObjects.Extent() == 1 ){
 111     Handle(SALOME_InteractiveObject) IO = IObjects.First();
 112     SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 113     if ( IO->hasEntry() && study ) {
 114       _PTR(Study) studyDS = study->studyDS();
 115       _PTR(SObject) obj( studyDS->FindObjectID( IO->getEntry() ) );
 116       _PTR(GenericAttribute) anAttr;
 117       if ( obj && obj->FindAttribute( anAttr, "AttributeIOR" ) ) {
 118         _PTR(AttributeIOR) anIOR( anAttr );
 119         shape = GetShapeFromIOR( anIOR->Value().c_str() );
 120       }
 121     }
 122   }
 123   return shape;
 124 }
 125
 126 //=======================================================================
 127 // function : GetNameOfSelectedIObjects()
 128 // purpose  : Define the name geom++ or other name of mono or multi sel.
 129 //=======================================================================
 130 int GEOMBase::GetNameOfSelectedIObjects( const SALOME_ListIO& IObjects,
 131                                          QString&             name,
 132                                          const bool           shapesOnly )
 133 {
 134   int nbSel = 0;
 135   name = ""; // clear output name
 136
 137   if ( !shapesOnly ) {
 138     nbSel = IObjects.Extent();
 139     if ( nbSel == 1 ) {
 140       Handle(SALOME_InteractiveObject) anIObj = IObjects.First();
 141       SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 142       if ( anIObj->hasEntry() && study ) {
 143         _PTR(Study) studyDS = study->studyDS();
 144         _PTR(SObject) obj( studyDS->FindObjectID( anIObj->getEntry() ) );
 145         _PTR(GenericAttribute) anAttr;
 146         if ( obj && obj->FindAttribute( anAttr, "AttributeName" ) ) {
 147           _PTR(AttributeName) aNameAttr ( anAttr );
 148           name = aNameAttr->Value().c_str();
 149         }
 150       }
 151     }
 152   }
 153   else {
 154     GEOM::ListOfGO anObjs;
 155     ConvertListOfIOInListOfGO( IObjects, anObjs, shapesOnly );
 156     nbSel = anObjs.length();
 157     if ( nbSel == 1 )
 158       name = GetName( anObjs[ 0 ] );
 159   }
 160
 161   if ( nbSel > 1 )
 162     name = QObject::tr( "%1_objects" ).arg( nbSel );
 163
 164   return nbSel;
 165 }
 166
 167
 168 //=================================================================================
 169 // function : GetShapeTypeString()
 170 // purpose  : for a single shape
 171 //=================================================================================
 172 QString GEOMBase::GetShapeTypeString(const TopoDS_Shape& shape)
 173 {
 174   QString aTypeString;
 175   if ( !shape.IsNull() ) {
 176     switch ( shape.ShapeType() ) {
 177     case TopAbs_COMPOUND:
 178       {
 179         aTypeString = QObject::tr( "GEOM_COMPOUND" );
 180         break;
 181       }
 182     case TopAbs_COMPSOLID:
 183       {
 184         aTypeString = QObject::tr( "GEOM_COMPOUNDSOLID" );
 185         break;
 186       }
 187     case TopAbs_SOLID:
 188       {
 189         aTypeString = QObject::tr( "GEOM_SOLID" );
 190         break;
 191       }
 192     case TopAbs_SHELL:
 193       {
 194         aTypeString = QObject::tr( "GEOM_SHELL" );
 195         break;
 196       }
 197     case TopAbs_FACE:
 198       {
 199         BRepAdaptor_Surface surf( TopoDS::Face( shape ) );
 200         switch ( surf.GetType() ) {
 201         case GeomAbs_Plane:
 202           {
 203             aTypeString = QObject::tr( "GEOM_PLANE" );
 204             break;
 205           }
 206         case GeomAbs_Cylinder:
 207           {
 208             aTypeString = QObject::tr( "GEOM_SURFCYLINDER" );
 209             break;
 210           }
 211         case GeomAbs_Sphere:
 212           {
 213             aTypeString = QObject::tr( "GEOM_SURFSPHERE" );
 214             break;
 215           }
 216         case GeomAbs_Torus:
 217           {
 218             aTypeString = QObject::tr( "GEOM_SURFTORUS" );
 219             break;
 220           }
 221         case GeomAbs_Cone:
 222           {
 223             aTypeString = QObject::tr( "GEOM_SURFCONE" );
 224             break;
 225           }
 226         default:
 227           {
 228             aTypeString = QObject::tr( "GEOM_FACE" );
 229             break;
 230           }
 231         }
 232         break;
 233       }
 234     case TopAbs_WIRE:
 235       {
 236         aTypeString = QObject::tr( "GEOM_WIRE" );
 237         break;
 238       }
 239     case TopAbs_EDGE:
 240       {
 241         BRepAdaptor_Curve curv( TopoDS::Edge( shape ) );
 242         switch ( curv.GetType() ) {
 243         case GeomAbs_Line:
 244           {
 245             aTypeString = ( qAbs( curv.FirstParameter() ) >= 1E6 || qAbs( curv.LastParameter() ) >= 1E6 ) ?
 246               QObject::tr( "GEOM_LINE" ) : QObject::tr( "GEOM_EDGE" );
 247             break;
 248           }
 249         case GeomAbs_Circle:
 250           {
 251             aTypeString = curv.IsClosed() ? QObject::tr( "GEOM_CIRCLE" ) : QObject::tr( "GEOM_ARC" );
 252             break;
 253           }
 254         default:
 255           {
 256             aTypeString = QObject::tr( "GEOM_EDGE" );
 257             break;
 258           }
 259         }
 260         break;
 261       }
 262     case TopAbs_VERTEX:
 263       {
 264         aTypeString = QObject::tr( "GEOM_VERTEX" );
 265         break;
 266       }
 267     case TopAbs_SHAPE:
 268       {
 269         aTypeString = QObject::tr( "GEOM_SHAPE" );
 270         break;
 271       }
 272     default:
 273       {
 274         break;
 275       }
 276     }
 277   }
 278   return aTypeString;
 279 }
 280
 281
 282 //=======================================================================
 283 // function : ConvertIORinGEOMAISShape()
 284 // purpose  :
 285 //=======================================================================
 286 Handle(GEOM_AISShape) GEOMBase::ConvertIORinGEOMAISShape(const QString& IOR, bool onlyInActiveView)
 287 {
 288   Handle(GEOM_AISShape) shape;
 289
 290   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 291   if ( study ) {
 292     _PTR(Study) studyDS = study->studyDS();
 293     _PTR(SObject) obj( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
 294     if ( obj ) {
 295       QList<SUIT_ViewWindow*> views;
 296       if ( onlyInActiveView )
 297         views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
 298       else
 299         views = SUIT_Session::session()->activeApplication()->desktop()->windows();
 300       foreach ( SUIT_ViewWindow* view, views ) {
 301         if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
 302           Handle(AIS_InteractiveContext) ic = ((OCCViewer_Viewer*)view->getViewManager()->getViewModel())->getAISContext();
 303
 304           AIS_ListOfInteractive displayed;
 305           ic->DisplayedObjects( displayed );
 306           ic->ObjectsInCollector( displayed );
 307
 308           AIS_ListIteratorOfListOfInteractive it( displayed );
 309           while ( it.More() && shape.IsNull() ) {
 310             if ( it.Value()->IsInstance( STANDARD_TYPE(GEOM_AISShape) ) ) {
 311               Handle(GEOM_AISShape) sh = Handle(GEOM_AISShape)::DownCast( it.Value() );
 312               if ( !sh.IsNull() && sh->hasIO() ) {
 313                 Handle(SALOME_InteractiveObject) IO = Handle(SALOME_InteractiveObject)::DownCast( sh->getIO() );
 314                 if ( !IO.IsNull() && IO->hasEntry() && obj->GetID() == IO->getEntry() )
 315                   shape = sh;
 316               }
 317             }
 318             it.Next();
 319           }
 320         }
 321         if ( !shape.IsNull() ) break;
 322       }
 323     }
 324   }
 325   return shape;
 326 }
 327
 328
 329 //=======================================================================
 330 // function : ConvertIORinGEOMActor()
 331 // purpose  :
 332 //=======================================================================
 333 GEOM_Actor* GEOMBase::ConvertIORinGEOMActor(const QString& IOR, bool onlyInActiveView)
 334 {
 335   GEOM_Actor* actor = 0;
 336
 337   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 338   if ( study ) {
 339     _PTR(Study) studyDS = study->studyDS();
 340     _PTR(SObject) obj( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
 341     if ( obj ) {
 342       QList<SUIT_ViewWindow*> views;
 343       if ( onlyInActiveView )
 344         views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
 345       else
 346         views = SUIT_Session::session()->activeApplication()->desktop()->windows();
 347       foreach ( SUIT_ViewWindow* view, views ) {
 348         if ( view && view->getViewManager()->getType() == SVTK_Viewer::Type() ) {
 349           SVTK_ViewWindow* aVTKViewWindow = dynamic_cast<SVTK_ViewWindow*>( view );
 350           if( !aVTKViewWindow )
 351             continue;
 352           vtkRenderer* Renderer = aVTKViewWindow->getRenderer();
 353           vtkActorCollection* theActors = Renderer->GetActors();
 354           theActors->InitTraversal();
 355           vtkActor* a = theActors->GetNextActor();
 356           while( a && !actor ) {
 357             if ( a->IsA( "GEOM_Actor" ) ) {
 358               GEOM_Actor* ga = GEOM_Actor::SafeDownCast( a );
 359               if ( ga && ga->hasIO() ) {
 360                 Handle(SALOME_InteractiveObject) IO = Handle(SALOME_InteractiveObject)::DownCast( ga->getIO() );
 361                 if ( !IO.IsNull() && IO->hasEntry() && obj->GetID() == IO->getEntry() )
 362                   actor = ga;
 363               }
 364             }
 365             a = theActors->GetNextActor();
 366           }
 367         }
 368         if ( actor ) break;
 369       }
 370     }
 371   }
 372   return actor;
 373 }
 374
 375 //=======================================================================
 376 // function : GetAIS()
 377 // purpose  :
 378 //=======================================================================
 379 Handle(AIS_InteractiveObject) GEOMBase::GetAIS( const Handle(SALOME_InteractiveObject)& IO,
 380                                                 bool onlyInActiveView, bool onlyGeom )
 381 {
 382   Handle(AIS_InteractiveObject) aisObject;
 383
 384   if ( !IO.IsNull() && IO->hasEntry() ) {
 385     QList<SUIT_ViewWindow*> views;
 386     if ( onlyInActiveView )
 387       views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
 388     else
 389       views = SUIT_Session::session()->activeApplication()->desktop()->windows();
 390
 391     foreach ( SUIT_ViewWindow* view, views ) {
 392       if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
 393         OCCViewer_Viewer* occViewer=(OCCViewer_Viewer*)view->getViewManager()->getViewModel();
 394         SOCC_Viewer* soccViewer = dynamic_cast<SOCC_Viewer*>(occViewer);
 395         if (soccViewer) {
 396           SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( soccViewer->CreatePrs( IO->getEntry() ) );
 397           if ( occPrs && !occPrs->IsNull() ) {
 398             AIS_ListOfInteractive shapes; occPrs->GetObjects( shapes );
 399             if( !shapes.Extent() ) continue;
 400             aisObject=shapes.First();
 401             delete occPrs;
 402           }
 403         }
 404       }
 405       if ( !aisObject.IsNull() ) break;
 406     } // foreach
 407   }
 408
 409   return aisObject;
 410 }
 411
 412
 413 //=======================================================================
 414 // function : ConvertIOinGEOMAISShape()
 415 // purpose  :
 416 //=======================================================================
 417 Handle(GEOM_AISShape) GEOMBase::ConvertIOinGEOMAISShape( const Handle(SALOME_InteractiveObject)& IO, bool onlyInActiveView )
 418 {
 419   return Handle(GEOM_AISShape)::DownCast( GetAIS( IO, onlyInActiveView, true ) );
 420 }
 421
 422
 423 //=======================================================================
 424 // function : ConvertListOfIOInListOfIOR()
 425 // purpose  :
 426 //=======================================================================
 427 QStringList GEOMBase::ConvertListOfIOInListOfIOR( const SALOME_ListIO& IObjects )
 428 {
 429   QStringList iors;
 430   SALOME_ListIteratorOfListIO it( IObjects );
 431   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 432   if ( study ) {
 433     _PTR(Study) studyDS = study->studyDS();
 434     for ( ; it.More(); it.Next() ) {
 435       GEOM::GEOM_Object_var geomObj = ConvertIOinGEOMObject( it.Value() );
 436       if ( !CORBA::is_nil( geomObj ) )
 437         iors.append( GetIORFromObject( geomObj ) );
 438     }
 439   }
 440   return iors;
 441 }
 442
 443
 444 //=======================================================================
 445 // function : ConvertIOinGEOMObject()
 446 // purpose  :
 447 //=======================================================================
 448 GEOM::GEOM_Object_ptr GEOMBase::ConvertIOinGEOMObject( const Handle(SALOME_InteractiveObject)& IO )
 449 {
 450   GEOM::GEOM_Object_var object;
 451
 452   if ( !IO.IsNull() && IO->hasEntry() ) {
 453     SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 454     if ( study ) {
 455       _PTR(Study) studyDS = study->studyDS();
 456       _PTR(SObject) obj( studyDS->FindObjectID( IO->getEntry() ) );
 457       if ( obj ) {
 458         CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject( obj );
 459         if ( !CORBA::is_nil( corbaObj ) )
 460           object = GEOM::GEOM_Object::_narrow( corbaObj );
 461       }
 462     }
 463   }
 464   return object._retn();
 465 }
 466
 467
 468 //=======================================================================
 469 // function : ConvertListOfIOInListOfGO()
 470 // purpose  :
 471 //=======================================================================
 472 void GEOMBase::ConvertListOfIOInListOfGO( const SALOME_ListIO& IObjects,
 473                                           GEOM::ListOfGO&      geomObjects,
 474                                           bool                 shapesOnly )
 475 {
 476   geomObjects.length( 0 );
 477
 478   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 479   if ( study ) {
 480     _PTR(Study) studyDS = study->studyDS();
 481
 482     geomObjects.length( IObjects.Extent() );
 483     SALOME_ListIteratorOfListIO it( IObjects );
 484
 485     int i = 0;
 486     for ( ; it.More(); it.Next() ) {
 487       GEOM::GEOM_Object_var geomObj = ConvertIOinGEOMObject( it.Value() );
 488       if ( !CORBA::is_nil( geomObj ) && ( !shapesOnly || IsShape( geomObj ) ) )
 489         geomObjects[ i++ ] = geomObj;
 490     }
 491     geomObjects.length( i );
 492   }
 493 }
 494
 495
 496 //=================================================================================
 497 // function : CreateArrowForLinearEdge()
 498 // purpose  : Create a cone topology to be used to display an arrow in the middle
 499 //          : of an edge showing its orientation. (For simulation and Viewer OCC only)
 500 //=================================================================================
 501 TopoDS_Shape GEOMBase::CreateArrowForLinearEdge( const TopoDS_Shape& shape )
 502 {
 503   TopoDS_Shape ArrowCone;
 504
 505   SUIT_ViewWindow* view = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
 506   if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() && shape.ShapeType() == TopAbs_EDGE ) {
 507     Handle(V3d_View) view3d = ((OCCViewer_ViewWindow*)view)->getViewPort()->getView();
 508     Standard_Real Width, Height;
 509     view3d->Size( Width, Height );
 510     const Standard_Real aHeight = (Width + Height) / 50.0;
 511
 512     try {
 513       Standard_Real first, last;
 514       Handle(Geom_Curve) curv = BRep_Tool::Curve( TopoDS::Edge( shape ), first, last );
 515       if ( !curv.IsNull() && curv->IsCN(1) ) {
 516         const Standard_Real param = ( first+last ) / 2.0;
 517         gp_Pnt middleParamPoint;
 518         gp_Vec V1;
 519         curv->D1( param, middleParamPoint, V1 );
 520         if ( V1.Magnitude() > Precision::Confusion() ) {
 521           /* Topology orientation not geom orientation */
 522           if ( shape.Orientation() == TopAbs_REVERSED )
 523             V1 *= -1.0;
 524
 525           gp_Ax2 anAxis( middleParamPoint, gp_Dir( V1 ) );
 526           const Standard_Real radius1 = aHeight / 5.0;
 527           if ( radius1 > 10.0 * Precision::Confusion() && aHeight > 10.0 * Precision::Confusion() )
 528             ArrowCone = BRepPrimAPI_MakeCone( anAxis, radius1, 0.0, aHeight ).Shape();
 529         }
 530       }
 531     }
 532     catch ( Standard_Failure ) {
 533       // OCC failures are hard to catch in GUI.
 534       // This is because of the position for #include <Standard_ErrorHandler.hxx> that is very critical to find
 535       // in SALOME environment : compilation error !
 536     }
 537   }
 538
 539   return ArrowCone;
 540 }
 541
 542
 543 //=================================================================================
 544 // function : VertexToPoint()
 545 // purpose  : If S can be converted in a gp_Pnt returns true and the result is P
 546 //=================================================================================
 547 bool GEOMBase::VertexToPoint( const TopoDS_Shape& shape, gp_Pnt& point )
 548 {
 549   if ( shape.IsNull() || shape.ShapeType() != TopAbs_VERTEX )
 550     return false;
 551   point = BRep_Tool::Pnt( TopoDS::Vertex( shape ) );
 552   return true;
 553 }
 554
 555
 556 //=================================================================================
 557 // function : GetBipointDxDyDz()
 558 // purpose  :
 559 //=================================================================================
 560 void GEOMBase::GetBipointDxDyDz( const gp_Pnt& point1, const gp_Pnt& point2, double& dx, double& dy, double& dz )
 561 {
 562   dx = point2.X() - point1.X();
 563   dy = point2.Y() - point1.Y();
 564   dz = point2.Z() - point1.Z();
 565 }
 566
 567
 568 //=================================================================================
 569 // function : LinearEdgeExtremities()
 570 // purpose  : If S can be converted in a linear edge and if initial an final points
 571 //          : distance is sufficient, returns true else returns false.
 572 //          : Resulting points are respectively P1 and P2
 573 //=================================================================================
 574 bool GEOMBase::LinearEdgeExtremities( const TopoDS_Shape& shape,  gp_Pnt& point1, gp_Pnt& point2 )
 575 {
 576   if ( shape.IsNull() || shape.ShapeType() != TopAbs_EDGE )
 577     return false;
 578
 579   BRepAdaptor_Curve curv( TopoDS::Edge( shape ) );
 580   if ( curv.GetType() != GeomAbs_Line )
 581     return false;
 582
 583   gp_Pnt p1, p2;
 584
 585   curv.D0( curv.FirstParameter(), p1 );
 586   curv.D0( curv.LastParameter(),  p2 );
 587
 588   if ( p1.Distance( p2 ) <= Precision::Confusion() )
 589     return false;
 590
 591   point1 = p1;
 592   point2 = p2;
 593   return true;
 594 }
 595
 596
 597 //=======================================================================
 598 // function : SelectionByNameInDialogs()
 599 // purpose  : Called when user has entered a name of object in a LineEdit.
 600 //          : The selection is changed. Dialog box will receive the
 601 //          : corresponding signal to manage this event.
 602 //=======================================================================
 603 bool GEOMBase::SelectionByNameInDialogs( QWidget* widget, const QString& objectUserName, const SALOME_ListIO& /*IObjects*/ )
 604 {
 605   /* Find SObject with name in component GEOM */
 606   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 607   if ( !study ) return false;
 608   _PTR(Study) studyDS = study->studyDS();
 609
 610   std::vector<_PTR(SObject)> listSO = studyDS->FindObjectByName( objectUserName.toStdString(), "GEOM" );
 611
 612   if ( listSO.size() < 1 ) {
 613     SUIT_MessageBox::critical( widget,
 614                                QObject::tr( "GEOM_WRN_WARNING" ),
 615                                QObject::tr( "GEOM_NAME_INCORRECT" ),
 616                                QObject::tr( "GEOM_BUT_OK" ) );
 617     return false;
 618   }
 619
 620   /* More than one object with same name */
 621   if ( listSO.size() > 1 ) {
 622     SUIT_MessageBox::critical( widget,
 623                                QObject::tr("GEOM_WRN_WARNING"),
 624                                QObject::tr("GEOM_IDENTICAL_NAMES_SELECT_BY_MOUSE"),
 625                                QObject::tr("GEOM_BUT_OK") );
 626     return false;
 627   }
 628
 629   /* Create a SALOME_InteractiveObject with a SALOME::SObject */
 630   Handle(SALOME_InteractiveObject) IO = new SALOME_InteractiveObject( listSO[0]->GetID().c_str(),
 631                                                                       "GEOM",
 632                                                                       objectUserName.toLatin1().constData() );
 633
 634   /* Add as a selected object       */
 635   /* Clear any previous selection : */
 636   /* Warning the LineEdit is purged because of signal currentSelectionChanged ! */
 637   // Sel->ClearIObjects(); //mzn
 638   // Sel->AddIObject(SI); //mzn
 639   return true;
 640 }
 641
 642
 643 //=======================================================================
 644 // function : DefineDlgPosition()
 645 // purpose  : Define x and y the default position for a dialog box
 646 //=======================================================================
 647 void GEOMBase::DefineDlgPosition( QWidget* dlg, int& x, int& y )
 648 {
 649   /* Here the position is on the bottom right corner - 10 */
 650   SUIT_Desktop* d = SUIT_Session::session()->activeApplication()->desktop();
 651   x = abs( d->x() + d->size().width()  - dlg->size().width()  - 10 );
 652   y = abs( d->y() + d->size().height() - dlg->size().height() - 10 );
 653 }
 654
 655
 656 //=======================================================================
 657 // function : GetDefaultName()
 658 // purpose  : Generates default names
 659 //=======================================================================
 660 QString GEOMBase::GetDefaultName( const QString& operation, bool extractPrefix )
 661 {
 662   QString aName = "";
 663
 664   // collect all object names of GEOM component
 665   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 666   if ( study ) {
 667     _PTR(Study) studyDS = study->studyDS();
 668
 669     std::set<std::string> names;
 670     _PTR(SComponent) component( studyDS->FindComponent( "GEOM" ) );
 671     if ( component ) {
 672       _PTR(ChildIterator) it( studyDS->NewChildIterator( component ) );
 673       for ( it->InitEx( true ); it->More(); it->Next() ) {
 674         names.insert( it->Value()->GetName() );
 675       }
 676     }
 677
 678     // build a unique name
 679     int aNumber = 0;
 680     bool isUnique = false;
 681     QString prefix = operation;
 682
 683     if ( extractPrefix ) {
 684       QStringList parts = prefix.split( "_", QString::KeepEmptyParts );
 685       if ( parts.count() > 1 ) {
 686         bool ok;
 687         aNumber = parts.last().toLong( &ok );
 688         if ( ok ) {
 689           parts.removeLast();
 690           prefix = parts.join( "_" );
 691           aNumber--;
 692         }
 693       }
 694     }
 695
 696     while ( !isUnique ) {
 697       aName = prefix + "_" + QString::number( ++aNumber );
 698       isUnique = ( names.count( aName.toStdString()) == 0 );
 699     }
 700   }
 701   return aName;
 702 }
 703
 704
 705 //=======================================================================
 706 // function : ShowErrorMessage()
 707 // purpose  : Shows message box with error code and comment
 708 //=======================================================================
 709 void GEOMBase::ShowErrorMessage( const QString& errorCode, const QString& comment )
 710 {
 711   QStringList text;
 712   text << QObject::tr( "GEOM_PRP_ABORT" );
 713   if ( !errorCode.isEmpty() )
 714     text << QObject::tr( errorCode.toLatin1().constData() );
 715   if ( !comment.isEmpty() )
 716     text << QObject::tr( comment.toLatin1().constData() );
 717
 718   SUIT_MessageBox::critical( SUIT_Session::session()->activeApplication()->desktop(),
 719                              QObject::tr( "GEOM_ERROR" ),
 720                              text.join( "\n" ),
 721                              QObject::tr( "GEOM_BUT_OK" ) );
 722 }
 723
 724
 725 //=======================================================================
 726 // function : GetObjectFromIOR()
 727 // purpose  : returns a GEOM_Object by given IOR (string)
 728 //=======================================================================
 729 GEOM::GEOM_Object_ptr GEOMBase::GetObjectFromIOR( const QString& IOR )
 730 {
 731   return GeometryGUI::GetObjectFromIOR (IOR);
 732 }
 733
 734 //=======================================================================
 735 // function : GetIORFromObject()
 736 // purpose  : returns IOR of a given GEOM_Object
 737 //=======================================================================
 738 QString GEOMBase::GetIORFromObject( GEOM::GEOM_Object_ptr object )
 739 {
 740   return GeometryGUI::GetIORFromObject (object);
 741 }
 742
 743 //=======================================================================
 744 // function : GetShape()
 745 // purpose  : returns a TopoDS_Shape stored in GEOM_Object
 746 //=======================================================================
 747 bool GEOMBase::GetShape( GEOM::GEOM_Object_ptr object, TopoDS_Shape& shape, const TopAbs_ShapeEnum type )
 748 {
 749   shape = TopoDS_Shape();
 750   if ( !CORBA::is_nil( object ) ) {
 751     TopAbs_ShapeEnum stype = (TopAbs_ShapeEnum)( object->GetShapeType() );
 752     if ( type == TopAbs_SHAPE || type == stype )
 753       shape = GEOM_Client::get_client().GetShape(  GeometryGUI::GetGeomGen(), object );
 754   }
 755   return !shape.IsNull();
 756 }
 757
 758 //=======================================================================
 759 // function : GetName()
 760 // purpose  : Get name of object
 761 //=======================================================================
 762 QString GEOMBase::GetName( GEOM::GEOM_Object_ptr object )
 763 {
 764   QString name;
 765   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 766
 767   if ( !CORBA::is_nil( object ) ) {
 768     // 1. search if object is already published in the study
 769     CORBA::String_var IOR = SalomeApp_Application::orb()->object_to_string( object );
 770     if ( study && strcmp( IOR.in(), "" ) != 0 ) {
 771       _PTR(SObject) aSObj( study->studyDS()->FindObjectIOR( std::string( IOR.in() ) ) );
 772       _PTR(GenericAttribute) anAttr;
 773       if ( aSObj && aSObj->FindAttribute( anAttr, "AttributeName") ) {
 774         _PTR(AttributeName) aNameAttr( anAttr );
 775         name = aNameAttr->Value().c_str();
 776       }
 777     }
 778
 779     // 2. if object is not found in the study, try default name
 780     if ( name.isEmpty() ) {
 781       if ( object->IsMainShape() ) {
 782         name = GetDefaultName( "geomObj" );
 783       }
 784       else {
 785         GEOM::GEOM_Object_var mainShape = object->GetMainShape();
 786         if ( !CORBA::is_nil( mainShape  ) ) {
 787           GEOM::ListOfLong_var indices = object->GetSubShapeIndices();
 788           if ( indices->length() > 0 ) {
 789             TopAbs_ShapeEnum type = (TopAbs_ShapeEnum)( object->GetShapeType() );
 790             name = QString( "%1:%2_%3" ).arg( GetName( mainShape.in() ) )
 791               .arg( TypeName( type ) ).arg( indices[0] );
 792           }
 793         }
 794       }
 795     }
 796   }
 797
 798   return name;
 799 }
 800
 801 //=======================================================================
 802 // function : IsShape()
 803 // purpose  : Return TRUE if object is valid and has shape
 804 //=======================================================================
 805 bool GEOMBase::IsShape( GEOM::GEOM_Object_ptr object )
 806 {
 807   return !object->_is_nil() && object->IsShape();
 808 }
 809
 810 //=======================================================================
 811 // function : TypeName()
 812 // purpose  : Get string representation for the shape type
 813 //=======================================================================
 814 QString GEOMBase::TypeName( TopAbs_ShapeEnum type )
 815 {
 816   QString name = "shape";
 817   switch( type ) {
 818   case TopAbs_COMPSOLID:
 819     name = "compsolid"; break;
 820   case TopAbs_COMPOUND:
 821     name = "compound";  break;
 822   case TopAbs_SOLID:
 823     name = "solid";     break;
 824   case TopAbs_SHELL:
 825     name = "shell";     break;
 826   case TopAbs_FACE:
 827     name = "face";      break;
 828   case TopAbs_WIRE:
 829     name = "wire";      break;
 830   case TopAbs_EDGE:
 831     name = "edge";      break;
 832   case TopAbs_VERTEX:
 833     name = "vertex";    break;
 834   default:
 835     break;
 836   }
 837   return name;
 838 }
 839
 840 //================================================================
 841 // Function : GetEntry
 842 // Purpose  : Get study entry for the given object (if it is published)
 843 //================================================================
 844 QString GEOMBase::GetEntry( GEOM::GEOM_Object_ptr object )
 845 {
 846   QString entry;
 847   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 848   if ( study && !CORBA::is_nil( object ) ) {
 849     QString objIOR = GetIORFromObject( object );
 850     if ( !objIOR.isEmpty() ) {
 851       _PTR(SObject) SO( study->studyDS()->FindObjectIOR( objIOR.toLatin1().constData() ) );
 852       if ( SO )
 853         entry = SO->GetID().c_str();
 854     }
 855   }
 856   return entry;
 857 }
 858
 859 //================================================================
 860 // Function : PublishSubObject
 861 // Purpose  : Publish sub-shape under the main object
 862 //================================================================
 863 void GEOMBase::PublishSubObject( GEOM::GEOM_Object_ptr object )
 864 {
 865   SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
 866   if ( study && !CORBA::is_nil( object ) ) {
 867     _PTR(Study) studyDS = study->studyDS();
 868     QString entry = GetEntry( object );
 869     GEOM::GEOM_Object_var father = object->GetMainShape();
 870     QString fatherEntry = GetEntry( father );
 871     if ( entry.isEmpty() && !CORBA::is_nil( father ) && !fatherEntry.isEmpty() ) {
 872       QString name = GetName( object );
 873       GeometryGUI::GetGeomGen()->AddInStudy( GeometryGUI::ClientStudyToStudy( studyDS ),
 874                                              object, name.toLatin1().data(), father.in() );
 875     }
 876   }
 877 }