1 // Copyright (C) 2014-2016 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 #include "STLPlugin_GUI.h"
22 #include "STLPlugin_ExportDlg.h"
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>
36 #include "GeometryGUI.h"
37 #include "GEOM_Operation.h"
39 #include "GEOM_Displayer.h"
40 #include "GEOM_GenericObjPtr.h"
42 #include <SALOMEconfig.h>
43 #include CORBA_SERVER_HEADER(STLPlugin)
45 typedef GEOM::GenericObjPtr<GEOM::ISTLOperations> STLOpPtr;
47 //=======================================================================
48 // function : STLPlugin_GUI()
49 // purpose : Constructor
50 //=======================================================================
51 STLPlugin_GUI::STLPlugin_GUI( GeometryGUI* parent ) : GEOMPluginGUI( parent )
55 //=======================================================================
56 // function : ~STLPlugin_GUI
57 // purpose : Destructor
58 //=======================================================================
59 STLPlugin_GUI::~STLPlugin_GUI()
63 //=======================================================================
64 // function : OnGUIEvent()
66 //=======================================================================
67 bool STLPlugin_GUI::OnGUIEvent( int theCommandID, SUIT_Desktop* parent )
70 switch ( theCommandID ) {
72 cmd = "Export_STL"; break;
74 cmd = "Import_STL"; break;
78 return OnGUIEvent( cmd, parent );
81 //=======================================================================
82 // function : OnGUIEvent()
84 //=======================================================================
85 bool STLPlugin_GUI::OnGUIEvent( const QString& theCommandID, SUIT_Desktop* parent )
89 if( theCommandID == "Export_STL" ) {
90 result = exportSTL( parent );
92 else if (theCommandID == "Import_STL") {
93 result = importSTL( parent );
96 getGeometryGUI()->getApp()->putInfo( tr("GEOM_PRP_COMMAND").arg( theCommandID ) );
102 //=======================================================================
103 // function : importSTL
105 //=======================================================================
106 bool STLPlugin_GUI::importSTL( SUIT_Desktop* parent )
108 SalomeApp_Application* app = getGeometryGUI()->getApp();
109 if ( !app ) return false;
110 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
111 if ( !study ) return false;
113 SALOMEDS::Study_var dsStudy = GeometryGUI::ClientStudyToStudy( study->studyDS() );
114 GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( dsStudy->StudyId(), "STLPluginEngine" );
115 STLOpPtr stlOp = GEOM::ISTLOperations::_narrow( op );
116 if ( stlOp.isNull() ) return false;
118 QStringList fileNames = app->getOpenFileNames( SUIT_FileDlg::getLastVisitedPath().isEmpty() ? QDir::currentPath() : QString(""),
120 tr( "IMPORT_TITLE" ),
122 if ( fileNames.count() > 0 )
124 QStringList entryList;
127 foreach( QString fileName, fileNames )
129 SUIT_OverrideCursor wc;
130 GEOM_Operation transaction( app, stlOp.get() );
134 app->putInfo( tr( "GEOM_PRP_LOADING" ).arg( fileName ) );
136 GEOM::ListOfGO_var result = stlOp->ImportSTL( fileName.toUtf8().constData() );
137 if ( result->length() > 0 && stlOp->IsDone() )
139 GEOM::GEOM_Object_var main = result[0];
140 QString publishName = GEOMBase::GetDefaultName( SUIT_Tools::file( fileName, true ) );
141 SALOMEDS::SObject_var so = GeometryGUI::GetGeomGen()->PublishInStudy( dsStudy,
142 SALOMEDS::SObject::_nil(),
144 publishName.toUtf8().constData() );
146 entryList.append( so->GetID() );
147 transaction.commit();
148 GEOM_Displayer( study ).Display( main.in() );
154 errors.append( QString( "%1 : %2" ).arg( fileName ).arg( stlOp->GetErrorCode() ) );
157 catch( const SALOME::SALOME_Exception& e )
162 getGeometryGUI()->updateObjBrowser( true );
163 app->browseObjects( entryList );
165 if ( errors.count() > 0 )
167 SUIT_MessageBox::critical( parent,
169 tr( "GEOM_IMPORT_ERRORS" ) + "\n" + errors.join( "\n" ) );
172 return fileNames.count() > 0;
175 //=======================================================================
176 // function : exportSTL
178 //=======================================================================
179 bool STLPlugin_GUI::exportSTL( SUIT_Desktop* parent )
181 SalomeApp_Application* app = getGeometryGUI()->getApp();
182 if ( !app ) return false;
183 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
184 if ( !study ) return false;
186 SALOMEDS::Study_var dsStudy = GeometryGUI::ClientStudyToStudy( study->studyDS() );
187 GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( dsStudy->StudyId(), "STLPluginEngine" );
188 STLOpPtr stlOp = GEOM::ISTLOperations::_narrow( op );
189 if ( stlOp.isNull() ) return false;
191 LightApp_SelectionMgr* sm = app->selectionMgr();
192 if ( !sm ) return false;
194 SALOME_ListIO selectedObjects;
195 sm->selectedObjects( selectedObjects );
198 SALOME_ListIteratorOfListIO it( selectedObjects );
199 for ( ; it.More(); it.Next() )
201 Handle(SALOME_InteractiveObject) io = it.Value();
202 GEOM::GEOM_Object_var obj = GEOMBase::ConvertIOinGEOMObject( io );
204 if ( CORBA::is_nil( obj ) ) continue;
206 bool isASCII = false;
207 double deflection = 0.;
208 bool isRelative = false;
209 QString fileName = STLPlugin_ExportDlg::getFileName( io,
211 tr( "EXPORT_TITLE" ),
217 if ( fileName.isEmpty() )
220 SUIT_OverrideCursor wc;
222 GEOM_Operation transaction( app, stlOp.get() );
226 app->putInfo( tr( "GEOM_PRP_EXPORT" ).arg( fileName ) );
229 stlOp->ExportSTL( obj, fileName.toUtf8().constData(), isASCII, deflection, isRelative );
231 if ( stlOp->IsDone() )
233 transaction.commit();
238 SUIT_MessageBox::critical( parent,
240 tr( "GEOM_PRP_ABORT" ) + "\n" + tr( stlOp->GetErrorCode() ) );
244 catch ( const SALOME::SALOME_Exception& e )
254 SUIT_MessageBox::warning( parent,
256 tr( "GEOM_WRN_NO_APPROPRIATE_SELECTION" ) );
261 //=====================================================================================
263 //=====================================================================================
267 __declspec( dllexport )
269 GEOMGUI* GetLibGUI( GeometryGUI* parent )
271 return new STLPlugin_GUI( parent );