src/STLPlugin/STLPlugin_GUI.cxx

   1 // Copyright (C) 2014-2023  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 // internal includes
  21 #include "STLPlugin_GUI.h"
  22 #include "STLPlugin_ExportDlg.h"
  23
  24 // GUI includes
  25 #include <SUIT_Desktop.h>
  26 #include <SUIT_FileDlg.h>
  27 #include <SUIT_MessageBox.h>
  28 #include <SUIT_OverrideCursor.h>
  29 #include <SUIT_Tools.h>
  30 #include <LightApp_SelectionMgr.h>
  31 #include <SalomeApp_Application.h>
  32 #include <SalomeApp_Study.h>
  33 #include <SALOME_ListIO.hxx>
  34
  35 // GEOM includes
  36 #include "GeometryGUI.h"
  37 #include "GEOM_Operation.h"
  38 #include "GEOMBase.h"
  39 #include "GEOM_Displayer.h"
  40 #include "GEOM_GenericObjPtr.h"
  41
  42 #include <SALOMEconfig.h>
  43 #include CORBA_SERVER_HEADER(STLPlugin)
  44
  45 typedef GEOM::GenericObjPtr<GEOM::ISTLOperations> STLOpPtr;
  46
  47 //=======================================================================
  48 // function : STLPlugin_GUI()
  49 // purpose  : Constructor
  50 //=======================================================================
  51 STLPlugin_GUI::STLPlugin_GUI( GeometryGUI* parent ) : GEOMPluginGUI( parent )
  52 {
  53 }
  54
  55 //=======================================================================
  56 // function : ~STLPlugin_GUI
  57 // purpose  : Destructor
  58 //=======================================================================
  59 STLPlugin_GUI::~STLPlugin_GUI()
  60 {
  61 }
  62
  63 //=======================================================================
  64 // function : OnGUIEvent()
  65 // purpose  :
  66 //=======================================================================
  67 bool STLPlugin_GUI::OnGUIEvent( int theCommandID, SUIT_Desktop* parent )
  68 {
  69   QString cmd;
  70   switch ( theCommandID ) {
  71   case 1:
  72     cmd = "Export_STL"; break;
  73   case 2:
  74     cmd = "Import_STL"; break;
  75   default:
  76     break;
  77   }
  78   return OnGUIEvent( cmd, parent );
  79 }
  80
  81 //=======================================================================
  82 // function : OnGUIEvent()
  83 // purpose  :
  84 //=======================================================================
  85 bool STLPlugin_GUI::OnGUIEvent( const QString& theCommandID, SUIT_Desktop* parent )
  86 {
  87   bool result = false;
  88
  89   if( theCommandID == "Export_STL" ) {
  90     result = exportSTL( parent );
  91   }
  92   else if (theCommandID == "Import_STL") {
  93     result = importSTL( parent );
  94   }
  95   else {
  96     getGeometryGUI()->getApp()->putInfo( tr("GEOM_PRP_COMMAND").arg( theCommandID ) );
  97   }
  98
  99   return result;
 100 }
 101
 102 //=======================================================================
 103 // function : importSTL
 104 // purpose  :
 105 //=======================================================================
 106 bool STLPlugin_GUI::importSTL( SUIT_Desktop* parent )
 107 {
 108   SalomeApp_Application* app = getGeometryGUI()->getApp();
 109   if ( !app ) return false;
 110
 111   GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( "STLPluginEngine" );
 112   STLOpPtr stlOp = GEOM::ISTLOperations::_narrow( op );
 113   if ( stlOp.isNull() ) return false;
 114
 115   QStringList fileNames = app->getOpenFileNames( SUIT_FileDlg::getLastVisitedPath().isEmpty() ? QDir::currentPath() : QString(""),
 116                                                  tr( "STL_FILES" ),
 117                                                  tr( "IMPORT_TITLE" ),
 118                                                  parent );
 119   if ( fileNames.count() > 0 )
 120   {
 121     QStringList entryList;
 122     QStringList errors;
 123
 124     foreach( QString fileName, fileNames )
 125     {
 126       SUIT_OverrideCursor wc;
 127       GEOM_Operation transaction( app, stlOp.get() );
 128
 129       try
 130       {
 131         app->putInfo( tr( "GEOM_PRP_LOADING" ).arg( fileName ) );
 132         transaction.start();
 133         GEOM::ListOfGO_var result = stlOp->ImportSTL( fileName.toUtf8().constData() );
 134         if ( result->length() > 0 && stlOp->IsDone() )
 135         {
 136           GEOM::GEOM_Object_var main = result[0];
 137           QString publishName = GEOMBase::GetDefaultName( SUIT_Tools::file( fileName, true ) );
 138           SALOMEDS::SObject_var so = GeometryGUI::GetGeomGen()->PublishInStudy( SALOMEDS::SObject::_nil(),
 139                                                                                 main.in(),
 140                                                                                 publishName.toUtf8().constData() );
 141
 142           entryList.append( so->GetID() );
 143           transaction.commit();
 144           GEOM_Displayer().Display( main.in() );
 145           main->UnRegister();
 146         }
 147         else
 148         {
 149           transaction.abort();
 150           errors.append( QString( "%1 : %2" ).arg( fileName ).arg( stlOp->GetErrorCode() ) );
 151         }
 152       }
 153       catch( const SALOME::SALOME_Exception& e )
 154       {
 155         transaction.abort();
 156       }
 157     }
 158     getGeometryGUI()->updateObjBrowser();
 159     app->browseObjects( entryList );
 160
 161     if ( errors.count() > 0 )
 162     {
 163       SUIT_MessageBox::critical( parent,
 164                                  tr( "GEOM_ERROR" ),
 165                                  tr( "GEOM_IMPORT_ERRORS" ) + "\n" + errors.join( "\n" ) );
 166     }
 167   }
 168   return fileNames.count() > 0;
 169 }
 170
 171 //=======================================================================
 172 // function : exportSTL
 173 // purpose  :
 174 //=======================================================================
 175 bool STLPlugin_GUI::exportSTL( SUIT_Desktop* parent )
 176 {
 177   SalomeApp_Application* app = getGeometryGUI()->getApp();
 178   if ( !app ) return false;
 179
 180   GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( "STLPluginEngine" );
 181   STLOpPtr stlOp = GEOM::ISTLOperations::_narrow( op );
 182   if ( stlOp.isNull() ) return false;
 183
 184   LightApp_SelectionMgr* sm = app->selectionMgr();
 185   if ( !sm ) return false;
 186
 187   SALOME_ListIO selectedObjects;
 188   sm->selectedObjects( selectedObjects );
 189   bool ok = false;
 190
 191   SALOME_ListIteratorOfListIO it( selectedObjects );
 192   for ( ; it.More(); it.Next() )
 193   {
 194     Handle(SALOME_InteractiveObject) io = it.Value();
 195     GEOM::GEOM_Object_var obj = GEOMBase::ConvertIOinGEOMObject( io );
 196
 197     if ( CORBA::is_nil( obj ) ) continue;
 198
 199     bool isASCII = false;
 200     double deflection = 0.;
 201     bool isRelative = false;
 202     QString fileName = STLPlugin_ExportDlg::getFileName( io,
 203                                                          tr( "STL_FILES" ),
 204                                                          tr( "EXPORT_TITLE" ),
 205                                                          parent,
 206                                                          isASCII,
 207                                                          deflection,
 208                                                          isRelative );
 209
 210     if ( fileName.isEmpty() )
 211       return false;
 212
 213     SUIT_OverrideCursor wc;
 214
 215     GEOM_Operation transaction( app, stlOp.get() );
 216
 217     try
 218     {
 219       app->putInfo( tr( "GEOM_PRP_EXPORT" ).arg( fileName ) );
 220       transaction.start();
 221
 222       stlOp->ExportSTL( obj, fileName.toUtf8().constData(), isASCII, deflection, isRelative  );
 223
 224       if ( stlOp->IsDone() )
 225       {
 226         transaction.commit();
 227       }
 228       else
 229       {
 230         transaction.abort();
 231         SUIT_MessageBox::critical( parent,
 232                                    tr( "GEOM_ERROR" ),
 233                                    tr( "GEOM_PRP_ABORT" ) + "\n" + tr( stlOp->GetErrorCode() ) );
 234         return false;
 235       }
 236     }
 237     catch ( const SALOME::SALOME_Exception& e )
 238     {
 239       transaction.abort();
 240       return false;
 241     }
 242     ok = true;
 243   }
 244
 245   if ( !ok )
 246   {
 247     SUIT_MessageBox::warning( parent,
 248                               tr( "WRN_WARNING" ),
 249                               tr( "GEOM_WRN_NO_APPROPRIATE_SELECTION" ) );
 250   }
 251   return ok;
 252 }
 253
 254 //=====================================================================================
 255 // EXPORTED METHODS
 256 //=====================================================================================
 257 extern "C"
 258 {
 259 #ifdef WIN32
 260     __declspec( dllexport )
 261 #endif
 262   GEOMGUI* GetLibGUI( GeometryGUI* parent )
 263   {
 264     return new STLPlugin_GUI( parent );
 265   }
 266 }