1 // Copyright (C) 2014 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"
41 #include <SALOMEconfig.h>
42 #include CORBA_SERVER_HEADER(STLPlugin)
44 //=======================================================================
45 // function : STLPlugin_GUI()
46 // purpose : Constructor
47 //=======================================================================
48 STLPlugin_GUI::STLPlugin_GUI( GeometryGUI* parent ) : GEOMPluginGUI( parent )
52 //=======================================================================
53 // function : ~STLPlugin_GUI
54 // purpose : Destructor
55 //=======================================================================
56 STLPlugin_GUI::~STLPlugin_GUI()
60 //=======================================================================
61 // function : OnGUIEvent()
63 //=======================================================================
64 bool STLPlugin_GUI::OnGUIEvent( int theCommandID, SUIT_Desktop* parent )
67 switch ( theCommandID ) {
69 cmd = "Export_STL"; break;
71 cmd = "Import_STL"; break;
75 return OnGUIEvent( cmd, parent );
78 //=======================================================================
79 // function : OnGUIEvent()
81 //=======================================================================
82 bool STLPlugin_GUI::OnGUIEvent( const QString& theCommandID, SUIT_Desktop* parent )
86 if( theCommandID == "Export_STL" ) {
87 result = exportSTL( parent );
89 else if (theCommandID == "Import_STL") {
90 result = importSTL( parent );
93 getGeometryGUI()->getApp()->putInfo( tr("GEOM_PRP_COMMAND").arg( theCommandID ) );
99 //=======================================================================
100 // function : importSTL
102 //=======================================================================
103 bool STLPlugin_GUI::importSTL( SUIT_Desktop* parent )
105 SalomeApp_Application* app = getGeometryGUI()->getApp();
106 if ( !app ) return false;
107 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
108 if ( !study ) return false;
110 SALOMEDS::Study_var dsStudy = GeometryGUI::ClientStudyToStudy( study->studyDS() );
111 GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( dsStudy->StudyId(), "STLPluginEngine" );
112 GEOM::ISTLOperations_var stlOp = GEOM::ISTLOperations::_narrow( op );
113 if ( CORBA::is_nil( stlOp ) ) return false;
115 QStringList fileNames = app->getOpenFileNames( SUIT_FileDlg::getLastVisitedPath().isEmpty() ? QDir::currentPath() : QString(""),
117 tr( "IMPORT_TITLE" ),
119 if ( fileNames.count() > 0 )
121 QStringList entryList;
124 foreach( QString fileName, fileNames )
126 SUIT_OverrideCursor wc;
127 GEOM_Operation transaction( app, stlOp.in() );
131 app->putInfo( tr( "GEOM_PRP_LOADING" ).arg( fileName ) );
133 GEOM::ListOfGO_var result = stlOp->ImportSTL( fileName.toUtf8().constData() );
134 if ( result->length() > 0 && stlOp->IsDone() )
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( dsStudy,
139 SALOMEDS::SObject::_nil(),
141 publishName.toUtf8().constData() );
143 entryList.append( so->GetID() );
144 transaction.commit();
145 GEOM_Displayer( study ).Display( main.in() );
150 errors.append( QString( "%1 : %2" ).arg( fileName ).arg( stlOp->GetErrorCode() ) );
153 catch( const SALOME::SALOME_Exception& e )
158 getGeometryGUI()->updateObjBrowser( true );
159 app->browseObjects( entryList );
161 if ( errors.count() > 0 )
163 SUIT_MessageBox::critical( parent,
165 tr( "GEOM_IMPORT_ERRORS" ) + "\n" + errors.join( "\n" ) );
168 return fileNames.count() > 0;
171 //=======================================================================
172 // function : exportSTL
174 //=======================================================================
175 bool STLPlugin_GUI::exportSTL( SUIT_Desktop* parent )
177 SalomeApp_Application* app = getGeometryGUI()->getApp();
178 if ( !app ) return false;
179 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
180 if ( !study ) return false;
182 SALOMEDS::Study_var dsStudy = GeometryGUI::ClientStudyToStudy( study->studyDS() );
183 GEOM::GEOM_IOperations_var op = GeometryGUI::GetGeomGen()->GetPluginOperations( dsStudy->StudyId(), "STLPluginEngine" );
184 GEOM::ISTLOperations_var stlOp = GEOM::ISTLOperations::_narrow( op );
185 if ( CORBA::is_nil( stlOp ) ) return false;
187 LightApp_SelectionMgr* sm = app->selectionMgr();
188 if ( !sm ) return false;
190 SALOME_ListIO selectedObjects;
191 sm->selectedObjects( selectedObjects );
194 SALOME_ListIteratorOfListIO it( selectedObjects );
195 for ( ; it.More(); it.Next() )
197 Handle(SALOME_InteractiveObject) io = it.Value();
198 GEOM::GEOM_Object_var obj = GEOMBase::ConvertIOinGEOMObject( io );
200 if ( CORBA::is_nil( obj ) ) continue;
202 bool isASCII = false;
203 double deflection = 0.;
204 bool isRelative = false;
205 QString fileName = STLPlugin_ExportDlg::getFileName( io,
207 tr( "EXPORT_TITLE" ),
213 if ( fileName.isEmpty() )
216 SUIT_OverrideCursor wc;
218 GEOM_Operation transaction( app, stlOp.in() );
222 app->putInfo( tr( "GEOM_PRP_EXPORT" ).arg( fileName ) );
225 stlOp->ExportSTL( obj, fileName.toUtf8().constData(), isASCII, deflection, isRelative );
227 if ( stlOp->IsDone() )
229 transaction.commit();
234 SUIT_MessageBox::critical( parent,
236 tr( "GEOM_PRP_ABORT" ) + "\n" + tr( stlOp->GetErrorCode() ) );
240 catch ( const SALOME::SALOME_Exception& e )
250 SUIT_MessageBox::warning( parent,
252 tr( "GEOM_WRN_NO_APPROPRIATE_SELECTION" ) );
257 //=====================================================================================
259 //=====================================================================================
263 __declspec( dllexport )
265 GEOMGUI* GetLibGUI( GeometryGUI* parent )
267 return new STLPlugin_GUI( parent );