]> SALOME platform Git repositories - modules/smesh.git/blob - src/SMESHGUI/SMESHGUI_MakeFull2DFrom3DOp.cxx
Salome HOME
bos #32735 [CEA] Create 2D Mesh from 3D elements.
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_MakeFull2DFrom3DOp.cxx
1 // Copyright (C) 2007-2023  CEA, EDF, 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 // File   : SMESHGUI_MakeFull2DFrom3DOp.cxx
20 // Author : Cesar Conopoima, Open CASCADE S.A.S. (cesar.conopoima@opencascade.com)
21
22 #include "SMESHGUI_MakeFull2DFrom3DOp.h"
23
24 #include "SMESHGUI.h"
25 #include "SMESHGUI_Utils.h"
26 #include "SMESHGUI_MeshUtils.h"
27 #include "SMESH_TypeFilter.hxx"
28 #include "SMESH_LogicalFilter.hxx"
29 #include "SMESHGUI_VTKUtils.h"
30 #include "SMESH_Actor.h"
31
32 // SALOME GUI includes
33 #include <LightApp_Application.h>
34 #include <LightApp_SelectionMgr.h>
35 #include <LightApp_UpdateFlags.h>
36 #include <SALOME_ListIO.hxx>
37 #include <SUIT_Desktop.h>
38 #include <SUIT_MessageBox.h>
39 #include <SUIT_OverrideCursor.h>
40 #include <SUIT_Session.h>
41 #include <SVTK_ViewModel.h>
42 //#include <SVTK_ViewWindow.h>
43 #include <SalomeApp_Study.h>
44 #include <SalomeApp_Tools.h>
45
46 // IDL includes
47 #include <SALOMEconfig.h>
48 #include CORBA_SERVER_HEADER(SMESH_Group)
49
50 // Qt includes
51 #include <QGroupBox>
52
53 #include <Standard_ErrorHandler.hxx>
54
55 /*!
56   \class SMESHGUI_Make2DFrom3DDlg
57   \brief Copy Mesh dialog box
58 */
59
60 SMESHGUI_MakeFull2DFrom3DDlg::SMESHGUI_MakeFull2DFrom3DDlg( QWidget* parent )
61   : SMESHGUI_Make2DFrom3DDlg( parent )
62 {
63   // title
64   setWindowTitle( tr("CAPTION") );
65   myModeGrp->setVisible(false);
66 }
67
68 SMESHGUI_MakeFull2DFrom3DDlg::~SMESHGUI_MakeFull2DFrom3DDlg()
69 {
70 }
71
72
73 /*!
74   \class SMESHGUI_MakeFull2DFrom3DOp
75   \brief Copy Mesh operation class
76 */
77 SMESHGUI_MakeFull2DFrom3DOp::SMESHGUI_MakeFull2DFrom3DOp()
78    : SMESHGUI_SelectionOp(),
79     myMeshFilter(SMESH::MESH),
80     myGroupFilter(SMESH::GROUP)
81 {
82 }
83
84 SMESHGUI_MakeFull2DFrom3DOp::~SMESHGUI_MakeFull2DFrom3DOp()
85 {
86 }
87
88 LightApp_Dialog* SMESHGUI_MakeFull2DFrom3DOp::dlg() const
89 {
90   return myDlg;
91 }
92
93 void SMESHGUI_MakeFull2DFrom3DOp::startOperation()
94 {
95   if( !myDlg )
96     myDlg = new SMESHGUI_MakeFull2DFrom3DDlg( desktop() );
97
98
99   myHelpFileName = "make_2dmesh_from_3d_elements.html";
100
101   SMESHGUI_SelectionOp::startOperation();
102
103   myDlg->setNewMeshName( SMESH::UniqueName( "Mesh_1" ) );
104   myDlg->setGroupName( SMESH::UniqueName( "Group" ) );
105   myDlg->show();
106
107   myDlg->activateObject( SMESHGUI_MakeFull2DFrom3DDlg::MeshOrGroups );
108   selectionDone();
109 }
110
111 void SMESHGUI_MakeFull2DFrom3DOp::selectionDone()
112 {
113   myDlg->clearSelection( SMESHGUI_MakeFull2DFrom3DDlg::MeshOrGroups );
114   mySrcMesh = SMESH::SMESH_Mesh::_nil();
115
116   if ( !dlg() ) return;
117   if ( dlg()->isVisible() ) {
118     try {
119       QStringList names, ids;
120       LightApp_Dialog::TypesList types;
121       selected( names, types, ids );
122       for ( int i = 0; i < names.count(); ++i )
123         names[i] = names[i].trimmed();
124       myDlg->selectObject( names, types, ids );
125
126       // enable/desable "new mesh" button
127       bool isMesh = true;
128       for ( int i = 0; i < ids.count() && isMesh; ++i )
129       {
130         _PTR(SObject) sobj = SMESH::getStudy()->FindObjectID( ids[i].toUtf8().constData() );
131         mySrcMesh = SMESH::SObjectToInterface<SMESH::SMESH_Mesh>( sobj );  
132       }
133       myDlg->setNewMeshEnabled( isMesh );
134     }
135     catch ( const SALOME::SALOME_Exception& S_ex ) {
136       SalomeApp_Tools::QtCatchCorbaException( S_ex );
137     }
138     catch ( ... ) {
139     }
140   }
141 }
142
143 SUIT_SelectionFilter* SMESHGUI_MakeFull2DFrom3DOp::createFilter( const int /*theId*/ ) const
144 {
145   SMESHGUI_MakeFull2DFrom3DOp* me = (SMESHGUI_MakeFull2DFrom3DOp*) this;
146
147   QList<SUIT_SelectionFilter*> subFilters;
148   subFilters.append( & me->myMeshFilter );
149   subFilters.append( & me->myGroupFilter );
150
151   SUIT_SelectionFilter* f = new SMESH_LogicalFilter( subFilters, SMESH_LogicalFilter::LO_OR );
152   return f;
153 }
154
155 bool SMESHGUI_MakeFull2DFrom3DOp::isValid( QString& msg ) const
156 {
157   if ( !dlg() ) return false;
158   
159   // check if a mesh is selected
160   if ( !myDlg->hasSelection( SMESHGUI_MakeFull2DFrom3DDlg::MeshOrGroups ))
161   {
162     msg = tr( "SMESH_ERR_NO_INPUT_MESH" );
163     return false;
164   }
165
166   QStringList entries;
167   dlg()->selectedObject( SMESHGUI_MakeFull2DFrom3DDlg::MeshOrGroups, entries );
168   const bool isMeshSelected = ( !mySrcMesh->_is_nil() );
169   if ( isMeshSelected )
170   {
171     // only one mesh is allowed
172     if ( entries.size() > 1 ) {
173       msg = tr( "SMESH_TOO_MANY_MESHES" );
174       return false;
175     }
176   }
177   else
178   {
179     // check if only groups are selected
180     for ( int i = 0; i < entries.count(); ++i )
181     {
182       SMESH::SMESH_GroupBase_var grp;
183       if ( _PTR(SObject) sobj = SMESH::getStudy()->FindObjectID( entries[i].toUtf8().constData() ))
184         grp = SMESH::SObjectToInterface<SMESH::SMESH_GroupBase>( sobj );
185       if ( grp->_is_nil() ) {
186         msg = tr( "SMESH_NOT_ONLY_GROUPS" );
187         return false;
188       }
189     }
190   }
191   // check if the selected objects contains elements of required type
192   bool hasVolumes = false;
193   
194   for ( int i = 0; i < entries.count(); ++i )
195   {
196     SMESH::SMESH_IDSource_var idSource;
197     if ( _PTR(SObject) sobj = SMESH::getStudy()->FindObjectID( entries[i].toUtf8().constData() ))
198       idSource = SMESH::SObjectToInterface<SMESH::SMESH_IDSource>( sobj );
199     if ( !idSource->_is_nil() ) {
200       SMESH::array_of_ElementType_var types = idSource->GetTypes();
201       for ( int j = 0; j < (int) types->length(); ++j )
202         if ( types[j] == SMESH::VOLUME )
203           hasVolumes = true;
204     }
205   }
206   if ( !hasVolumes ) {
207     msg = tr( "SMESH_ERR_NO_3D_ELEMENTS" );
208     return false;
209   } 
210
211   // check if new mesh name is specified
212   if ( myDlg->needNewMesh() && myDlg->getNewMeshName().isEmpty() ) {
213     msg = tr( "SMESH_ERR_MESH_NAME_NOT_SPECIFIED" );
214     return false;
215   }
216
217   // check if group name is specified
218   if ( myDlg->needGroup() && myDlg->getGroupName().isEmpty() ) {
219     msg = tr( "SMESH_ERR_GRP_NAME_NOT_SPECIFIED" );
220     return false;
221   }
222
223   return true;
224 }
225
226 bool SMESHGUI_MakeFull2DFrom3DOp::compute2DMesh( QStringList& theEntryList )
227 {
228   SUIT_OverrideCursor wc;
229
230   bool ok = false;
231   try {
232     QString meshName          = myDlg->needNewMesh() ? myDlg->getNewMeshName() : QString();
233     QString groupName         = myDlg->needGroup()   ? myDlg->getGroupName()   : QString();
234     bool copyAll              = myDlg->copySource();
235
236     QStringList entries;
237     dlg()->selectedObject( SMESHGUI_MakeFull2DFrom3DDlg::MeshOrGroups, entries );
238     SMESH::ListOfIDSources_var groups = new SMESH::ListOfIDSources;
239     QString wrongGroups = "";
240
241     if ( mySrcMesh->_is_nil() ) // get selected groups, find groups of wrong type
242     {
243       int nbGroups = 0;      
244       groups->length( entries.count() );
245       for ( int i = 0; i < entries.count(); ++i )
246       {
247         _PTR(SObject) sobj = SMESH::getStudy()->FindObjectID( entries[i].toUtf8().constData() );
248         SMESH::SMESH_IDSource_var grp = SMESH::SObjectToInterface<SMESH::SMESH_IDSource>( sobj );  
249         SMESH::array_of_ElementType_var types = grp->GetTypes();
250         if ( types->length() < 1 || types[0] != SMESH::VOLUME )
251         {
252           if ( !wrongGroups.isEmpty() )
253             wrongGroups += ", ";
254           wrongGroups += sobj->GetName().c_str();
255         }
256         else
257         {
258           groups[ nbGroups++ ] = grp;
259         }
260       }
261       groups->length( nbGroups );
262       mySrcMesh = groups[0]->GetMesh();
263     }
264
265     if ( !CORBA::is_nil( mySrcMesh ) ) {
266       SMESH::SMESH_MeshEditor_var aMeshEditor = mySrcMesh->GetMeshEditor();
267       SMESH::SMESH_Group_var newGrp;
268       SMESH::SMESH_Mesh_var newMesh;
269       CORBA::Long nbAdded = aMeshEditor->MakeBoundaryElements( SMESH::BND_2DFROM3D,
270                                                                groupName.toUtf8().constData(),
271                                                                meshName.toUtf8().constData(),
272                                                                copyAll,
273                                                                true,
274                                                                groups,
275                                                                newMesh.out(),
276                                                                newGrp.out() );
277       QString msg = tr("NB_ADDED").arg( nbAdded );
278       if ( !wrongGroups.isEmpty() )
279         msg += ".\n" + tr("WRONG_GROUPS").arg( wrongGroups );
280       SUIT_MessageBox::information( myDlg, tr("SMESH_INFORMATION"), msg);
281
282       if ( !newMesh->_is_nil() ) {
283         if( _PTR(SObject) aSObject = SMESH::ObjectToSObject( newMesh ) )
284           theEntryList.append( aSObject->GetID().c_str() );
285       }
286       ok = true;
287
288       for ( int i = 0; i < entries.count(); ++i )
289         if ( SMESH_Actor* actor = SMESH::FindActorByEntry( entries[i].toUtf8().constData() ))
290         {
291           actor->SetEntityMode( actor->GetEntityMode() | SMESH_Actor::eFaces );
292           SMESH::Update( actor->getIO(), actor->GetVisibility() );
293         }
294       SMESH::RepaintCurrentView();
295     }
296   }
297   catch ( ... ) {
298   }
299   return ok;
300 }
301
302 bool SMESHGUI_MakeFull2DFrom3DOp::onApply()
303 {
304   if ( SMESHGUI::isStudyLocked() )
305     return false;
306
307   QString msg;
308   if ( !isValid( msg ) ) {
309     dlg()->show();
310     if ( msg != "" )
311       SUIT_MessageBox::warning( myDlg, tr( "SMESH_ERROR" ), msg );
312     return false;
313   }
314
315   QStringList anEntryList;
316   bool res = false;
317   try {
318     res = compute2DMesh( anEntryList );
319   }
320   catch ( const SALOME::SALOME_Exception& S_ex ) {
321     SalomeApp_Tools::QtCatchCorbaException( S_ex );
322   }
323   catch ( ... ) {
324   }
325
326   if ( res ) {
327     SMESHGUI::Modified();
328     update( UF_ObjBrowser | UF_Model );
329     if( LightApp_Application* anApp =
330         dynamic_cast<LightApp_Application*>( SUIT_Session::session()->activeApplication() ) )
331       anApp->browseObjects( anEntryList, isApplyAndClose() );
332     myDlg->setNewMeshName( SMESH::UniqueName( "Mesh_1" ) );
333     myDlg->setGroupName( SMESH::UniqueName( "Group" ) );
334   }
335   else {
336     SUIT_MessageBox::warning( myDlg, tr( "SMESH_ERROR" ), tr( "SMESH_OPERATION_FAILED" ) );
337   }
338
339   return res;
340 }