src/PluginUtils/GeomSelectionTools.cxx

   1 // Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // ---
  21 // File    : GeomSelectionTools.cxx
  22 // Authors : Nicolas GEIMER (OCC)
  23 // ---
  24
  25 #include "GeomSelectionTools.h"
  26
  27 #include <LightApp_SelectionMgr.h>
  28 #include <SalomeApp_Application.h>
  29 #include <SUIT_Session.h>
  30
  31 #include <GEOMBase.h>
  32 #include <GEOMImpl_Types.hxx>
  33 #include <GEOM_Client.hxx>
  34 #include <GEOM_wrap.hxx>
  35 #include <SALOME_ListIO.hxx>
  36 #include <SMESHGUI_Utils.h>
  37
  38 #include <BRepAdaptor_Surface.hxx>
  39 #include <TopoDS.hxx>
  40 #include <TopoDS_Iterator.hxx>
  41
  42 #include "utilities.h"
  43
  44 #include "SALOME_LifeCycleCORBA.hxx"
  45 #include <sstream>
  46
  47 /*!
  48  * Constructor
  49  *
  50  */
  51 GeomSelectionTools::GeomSelectionTools()
  52 {
  53 }
  54
  55 /*!
  56  * Allows to get the Salome Application
  57  * @return A LightApp_SelectionMgr Pointer or 0 if it can't get it.
  58  */
  59 SalomeApp_Application*  GeomSelectionTools::GetSalomeApplication()
  60 {
  61   SalomeApp_Application* anApp = dynamic_cast<SalomeApp_Application*>( SUIT_Session::session()->activeApplication() );
  62   if (anApp)
  63     return anApp;
  64   else
  65     return 0;
  66 }
  67
  68 /*!
  69  * Allows to get the selection manager from LightApp
  70  * @return A LightApp_SelectionMgr Pointer or 0 if it can't get it.
  71  */
  72 LightApp_SelectionMgr*  GeomSelectionTools::selectionMgr()
  73 {
  74    SalomeApp_Application* anApp = GetSalomeApplication();
  75    if (anApp)
  76      return dynamic_cast<LightApp_SelectionMgr*>( anApp->selectionMgr() );
  77    else
  78      return 0;
  79 }
  80
  81 /*!
  82  * Return the list of the selected Salome Interactive Object (SALOME_ListIO*)
  83  * @return the list of the selected Salome Interactive Object
  84  */
  85 SALOME_ListIO* GeomSelectionTools::getSelectedSalomeObjects()
  86 {
  87   SALOME_ListIO* selected = new SALOME_ListIO;
  88   LightApp_SelectionMgr* aSel = selectionMgr();
  89   aSel->selectedObjects( *selected, NULL, false );
  90   return selected;
  91 }
  92
  93 /*!
  94  * Return the first selected Salome Interactive Object (Handle(Salome_InteractiveObject))
  95  * @return the first selected Salome Interactive Object
  96  */
  97 Handle(SALOME_InteractiveObject) GeomSelectionTools::getFirstSelectedSalomeObject()
  98 {
  99   SALOME_ListIO selected;
 100   LightApp_SelectionMgr* aSel = selectionMgr();
 101   aSel->selectedObjects( selected, NULL, false );
 102   if (!selected.IsEmpty()){
 103     SALOME_ListIteratorOfListIO anIt(selected);
 104     Handle(SALOME_InteractiveObject) anIO;
 105     anIO = anIt.Value();
 106     return anIO;
 107   }
 108   return NULL;
 109 }
 110
 111 /*!
 112  * Return the entry of the first selected Object
 113  * @return the entry of the first selected Object
 114  */
 115 std::string GeomSelectionTools::getFirstSelectedEntry()
 116 {
 117   Handle(SALOME_InteractiveObject) anIO;
 118   std::string entry="";
 119   anIO=GeomSelectionTools::getFirstSelectedSalomeObject();
 120   return GeomSelectionTools::getEntryOfObject(anIO);
 121 }
 122
 123 /*!
 124  * Return the entry of a Salome Interactive Object
 125  * @param anIO the Handle of the Salome Interactive Object
 126  * @return the entry of the Salome Interactive Object
 127  */
 128 std::string GeomSelectionTools::getEntryOfObject(Handle(SALOME_InteractiveObject) anIO){
 129   std::string entry="";
 130   _PTR(SObject) aSO = SalomeApp_Application::getStudy()->FindObjectID(anIO->getEntry());
 131   if (aSO){
 132     _PTR(SObject) aRefSObj;
 133     // If selected object is a reference
 134     if ( aSO->ReferencedObject( aRefSObj ))
 135       entry = aRefSObj->GetID();
 136     // If selected object is a reference is not a reference
 137     else
 138       entry= anIO->getEntry();
 139   }
 140   return entry;
 141 }
 142
 143 /*!
 144  * Retrieve the name from the entry of the object
 145  * @param entry the entry of the object
 146  * @return the name of the object
 147  */
 148 std::string GeomSelectionTools::getNameFromEntry(const std::string& entry)
 149 {
 150   std::string name = "";
 151   _PTR(SObject) aSO = SalomeApp_Application::getStudy()->FindObjectID(entry);
 152   if (aSO){
 153     _PTR(SObject) aRefSObj;
 154     // If selected object is a reference
 155     if ( aSO->ReferencedObject( aRefSObj ))
 156       name = aRefSObj->GetName();
 157     // If selected object is a reference is not a reference
 158     else
 159       name = aSO->GetName();
 160    }
 161   return name;
 162 }
 163
 164
 165 /*!
 166  * Retrieve the component type of the first selected object, it manages successfully references.
 167  * @return the component type of the first selected object
 168  */
 169 std::string GeomSelectionTools::getFirstSelectedComponentDataType()
 170 {
 171   Handle(SALOME_InteractiveObject) anIO;
 172   std::string DataType="";
 173   anIO=GeomSelectionTools::getFirstSelectedSalomeObject();
 174   _PTR(SObject) aSO = SalomeApp_Application::getStudy()->FindObjectID(anIO->getEntry());
 175   if (aSO){
 176     _PTR(SObject) aRefSObj;
 177     // If selected object is a reference
 178     if ( aSO->ReferencedObject( aRefSObj ))
 179       DataType= aRefSObj->GetFatherComponent()->ComponentDataType();
 180     // If selected object is a reference is not a reference
 181     else
 182       DataType=anIO->getComponentDataType();
 183  }
 184  return DataType;
 185 }
 186
 187 /*!
 188  * Retrieve the shape type from the entry
 189  * @return the shape type from the entry, return TopAbs_SHAPE if the object does not define a shape or a group.
 190  */
 191 TopAbs_ShapeEnum GeomSelectionTools::entryToShapeType(const std::string& entry)
 192 {
 193   TopAbs_ShapeEnum ShapeType = TopAbs_SHAPE;
 194   if ( _PTR(SObject) aSO = SalomeApp_Application::getStudy()->FindObjectID(entry))
 195   {
 196     _PTR(SObject) aRefSObj;
 197     if ( aSO->ReferencedObject( aRefSObj ))    // If selected object is a reference
 198       aSO = aRefSObj;
 199     GEOM::GEOM_Object_var aShape = SMESH::SObjectToInterface<GEOM::GEOM_Object>(aSO);
 200     if ( !aShape->_is_nil() )
 201     {
 202       ShapeType= (TopAbs_ShapeEnum)aShape->GetShapeType();
 203       // if the Geom Object is a group
 204       if (aShape->GetType() == GEOM_GROUP)
 205       {
 206         GEOM::GEOM_Gen_var _geomEngine = aShape->GetGen();
 207         GEOM::GEOM_IGroupOperations_wrap aGroupOp = _geomEngine->GetIGroupOperations();
 208         ShapeType= (TopAbs_ShapeEnum)aGroupOp->GetType(aShape);
 209       }
 210       // if not
 211       else if ( ShapeType == TopAbs_COMPOUND )
 212       {
 213         TopoDS_Shape shape;
 214         if (GEOMBase::GetShape(aShape, shape))
 215         {
 216           TopoDS_Iterator it( shape );
 217           if ( it.More() )
 218             ShapeType = it.Value().ShapeType();
 219         }
 220       }
 221     }
 222   }
 223   return ShapeType;
 224 }
 225
 226 /*!
 227  * Gives the ShapeType of the first Selected Object, return TopAbs_SHAPE if the first selected object does not define a shape.
 228  * @return the ShapeType of the first Selected Object, return TopAbs_SHAPE if the first selected object does not define a shape.
 229  */
 230 TopAbs_ShapeEnum GeomSelectionTools:: getFirstSelectedShapeType()
 231 {
 232   Handle(SALOME_InteractiveObject) anIO=GeomSelectionTools::getFirstSelectedSalomeObject();
 233   return entryToShapeType(anIO->getEntry());
 234 }
 235
 236 /*!
 237  *  Print information to std output of the face
 238  *  and return the OCC type of face: Plane, Cylinder,Cone, Sphere, Torus, BezierSurface,BSplineSurface, SurfaceOfRevolution,SurfaceOfExtrusion, OtherSurface
 239  *  @param TopoDS_Shape S Face we want information about.
 240  *  @return the OCC type of face: Plane, Cylinder,Cone, Sphere, Torus, BezierSurface,BSplineSurface, SurfaceOfRevolution,SurfaceOfExtrusion, OtherSurface
 241  *  return Other_Surface if the selected face is not a face.
 242  *  Information printed is :
 243  *  U and V degrees
 244  *  U and V number of poles
 245  *  U and V number of knots
 246  *  U or V is Rational ?
 247  *
 248  */
 249 GeomAbs_SurfaceType GeomSelectionTools::getFaceInformation(TopoDS_Shape S)
 250 {
 251   GeomAbs_SurfaceType surf_type=GeomAbs_OtherSurface ;
 252   if (!S.IsNull() &&  S.ShapeType()==TopAbs_FACE){
 253     TopoDS_Face f=TopoDS::Face(S);
 254     BRepAdaptor_Surface surf_adap(f);
 255
 256     /* Global Information */
 257     std::cout << "GLOBAL INFORMATION" << std::endl;
 258     std::cout << "******************" << std::endl;
 259     std::stringstream buffer;
 260     buffer << "Degre U : " <<  surf_adap.UDegree();
 261    //conversion nécessaire pour affichage
 262     std::cout << buffer.str() << std::endl;
 263     std::cout <<  " Degre V : " <<  surf_adap.VDegree() << std::endl;
 264     std::cout <<  " Nb Poles U : " <<  surf_adap.NbUPoles() << std::endl;
 265     std::cout <<  " Nb Poles V : " <<  surf_adap.NbVPoles() << std::endl;
 266     std::cout <<  " Nb Noeuds U : " <<  surf_adap.NbUKnots() << std::endl;
 267     std::cout <<  " Nb Noeuds V : " <<  surf_adap.NbVKnots() << std::endl;
 268     std::cout <<  " U Rationnel ? " <<  surf_adap.IsURational() << std::endl;
 269     std::cout <<  " V Rationnel ? " <<  surf_adap.IsVRational() << std::endl;
 270
 271     surf_type=surf_adap.GetType();
 272   }
 273   return surf_type;
 274 }
 275
 276
 277 //////////////////////////////////////////
 278 // Utility functions
 279 //////////////////////////////////////////
 280 #include <QLocale>
 281 #include <QRegExp>
 282
 283 QString PluginUtils::PrintDoubleValue( double theValue, int thePrecision )
 284 {
 285   const double prec = 1e-12;
 286
 287   if ( qAbs(theValue) < prec )
 288     return "0";
 289
 290   QString aRes = QLocale().toString( theValue, thePrecision >= 0 ? 'f' : 'g', qAbs( thePrecision ) );
 291
 292   if ( prec > 0 ) {
 293     int p = 0;
 294     while ( p < thePrecision ) {
 295       QString aRes = QLocale().toString( theValue, thePrecision >= 0 ? 'f' : 'g', qAbs( p++ ) );
 296       double v = aRes.toDouble();
 297       double err = qAbs( theValue - v );
 298       if ( err > 0 && err <= prec )
 299         break;
 300     }
 301   }
 302
 303   // remove trailing zeroes
 304
 305   QRegExp expre( QString( "(%1|%2)[+-]?[0-9]+$" ).arg( QLocale().exponential().toLower(),
 306                                QLocale().exponential().toUpper() ) );
 307
 308   int idx = aRes.indexOf( expre );
 309   QString aResExp = "";
 310   if ( idx >= 0 ) {
 311     aResExp = aRes.mid( idx );
 312     aRes = aRes.left( idx );
 313   }
 314
 315   if ( aRes.contains( QLocale().decimalPoint() ) )
 316     aRes.remove( QRegExp( QString( "(\\%1|0)0*$" ).arg( QLocale().decimalPoint() ) ) );
 317
 318   return aRes == "-0" ? QString( "0" ) : aRes + aResExp;
 319 }