Salome HOME
Corrections to handle adapation to shape
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_CreateDualMeshOp.cxx
1 // Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 // SMESH SMESHGUI : GUI for SMESH component
24 // File   : SMESHGUI_CreateDualMeshOp.cxx
25 // Author : Yoann AUDOUIN (EDF)
26 // SMESH includes
27 //
28 #include "SMESHGUI_CreateDualMeshOp.h"
29
30 #include "SMESHGUI.h"
31 #include "SMESHGUI_CreateDualMeshDlg.h"
32 #include "SMESHGUI_MeshEditPreview.h"
33 #include "SMESHGUI_Utils.h"
34 #include "SMESH_ActorUtils.h"
35 #include "SMESH_TypeFilter.hxx"
36 #include "SMDSAbs_ElementType.hxx"
37
38 // SALOME GUI includes
39 #include <LightApp_UpdateFlags.h>
40 #include <SUIT_MessageBox.h>
41 #include <SUIT_OverrideCursor.h>
42 #include <SalomeApp_Tools.h>
43 #include <SalomeApp_Application.h>
44 #include <SALOME_Actor.h>
45
46 // Qt includes
47 #include <QLineEdit>
48 #include <QCheckBox>
49
50 // IDL includes
51 #include <SALOMEconfig.h>
52 #include CORBA_SERVER_HEADER(SMESH_MeshEditor)
53
54 // VTK includes
55 #include <vtkProperty.h>
56
57 //================================================================================
58 /*!
59  * \brief Constructor
60  *
61  * Initialize operation
62 */
63 //================================================================================
64 SMESHGUI_CreateDualMeshOp::SMESHGUI_CreateDualMeshOp()
65   : SMESHGUI_SelectionOp(),
66     myDlg( 0 )
67 {
68 }
69
70 //================================================================================
71 /*!
72  * \brief Destructor
73 */
74 //================================================================================
75 SMESHGUI_CreateDualMeshOp::~SMESHGUI_CreateDualMeshOp()
76 {
77   if ( myDlg ) delete myDlg;
78 }
79
80 //================================================================================
81 /*!
82  * \brief Gets dialog of this operation
83   * \retval LightApp_Dialog* - pointer to dialog of this operation
84 */
85 //================================================================================
86 LightApp_Dialog* SMESHGUI_CreateDualMeshOp::dlg() const
87 {
88   return myDlg;
89 }
90
91 //================================================================================
92 /*!
93  * \brief Creates dialog if necessary and shows it
94  *
95  * Virtual method redefined from base class called when operation is started creates
96  * dialog if necessary and shows it, activates selection
97  */
98 //================================================================================
99 void SMESHGUI_CreateDualMeshOp::startOperation()
100 {
101   if( !myDlg )
102   {
103     myDlg = new SMESHGUI_CreateDualMeshDlg( );
104   }
105   connect( myDlg, SIGNAL( onClicked( int ) ), SLOT( ConnectRadioButtons( int ) ) );
106
107   myHelpFileName = "create_dual_mesh.html";
108
109   SMESHGUI_SelectionOp::startOperation();
110
111   myDlg->activateObject( 0 );
112   myDlg->ShowWarning( false );
113   myDlg->show();
114
115   selectionDone();
116 }
117
118 //================================================================================
119 /*!
120  * \brief Updates dialog's look and feel
121  *
122  * Virtual method redefined from the base class updates dialog's look and feel
123  */
124 //================================================================================
125 void SMESHGUI_CreateDualMeshOp::selectionDone()
126 {
127   if ( !dlg()->isVisible() )
128     return;
129
130   SMESHGUI_SelectionOp::selectionDone();
131   try
132   {
133     QString anObjEntry = myDlg->selectedObject( 0 );
134     _PTR(SObject) pObj = SMESH::getStudy()->FindObjectID( anObjEntry.toUtf8().data() );
135     if ( !pObj ) return;
136
137     SMESH::SMESH_IDSource_var idSource =
138       SMESH::SObjectToInterface<SMESH::SMESH_IDSource>( pObj );
139
140     myDlg->setButtonEnabled( true, QtxDialog::OK | QtxDialog::Apply );
141     if( idSource->_is_nil() )
142     {
143       myDlg->setButtonEnabled( false, QtxDialog::OK | QtxDialog::Apply );
144       return;
145     }
146     SMESH::SMESH_Mesh_var      mesh = idSource->GetMesh();
147
148     // show warning on non-conformal result mesh
149     if ( ! idSource->_is_nil() )
150     {
151       SMESH::SMESH_subMesh_var subMesh =
152         SMESH::SObjectToInterface<SMESH::SMESH_subMesh>( pObj );
153       // TODO: Check that mesh is only tetra
154       if (!checkMesh(idSource)){
155         myDlg->ShowWarning( true );
156       }
157     }
158     std::string mesh_name = "dual_" + pObj->GetName();
159     myDlg->myMeshName->setText(QString(mesh_name.c_str()));
160
161   }
162   catch ( const SALOME::SALOME_Exception& S_ex )
163   {
164     SalomeApp_Tools::QtCatchCorbaException( S_ex );
165   }
166   catch ( ... )
167   {
168   }
169
170 }
171
172 //================================================================================
173 /*!
174  * \brief Creates selection filter
175   * \param theId - identifier of current selection widget
176   * \retval SUIT_SelectionFilter* - pointer to the created filter or null
177  *
178  * Creates selection filter in accordance with identifier of current selection widget
179  */
180 //================================================================================
181 SUIT_SelectionFilter* SMESHGUI_CreateDualMeshOp::createFilter( const int theId ) const
182 {
183   if ( theId == 0 )
184     return new SMESH_TypeFilter( SMESH::MESHorSUBMESH );
185   else
186     return 0;
187 }
188
189 //================================================================================
190 /*!
191  * \brief Edits mesh
192  *
193  * Virtual slot redefined from the base class called when "Apply" button is clicked
194  */
195 //================================================================================
196 bool SMESHGUI_CreateDualMeshOp::onApply()
197 {
198   SUIT_OverrideCursor aWaitCursor;
199
200   QString aMess;
201   QStringList anEntryList;
202
203   QString anObjEntry = myDlg->selectedObject( 0 );
204   _PTR(SObject) pObj = SMESH::getStudy()->FindObjectID( anObjEntry.toUtf8().data() );
205   if ( !pObj )
206   {
207     dlg()->show();
208     SUIT_MessageBox::warning( myDlg,
209                               tr( "SMESH_WRN_WARNING" ), tr("MESH_IS_NOT_SELECTED") );
210     return false;
211   }
212
213   SMESH::SMESH_Mesh_var mesh;
214   SMESH::SMESH_IDSource_var idSource =
215     SMESH::SObjectToInterface<SMESH::SMESH_IDSource>( pObj );
216   if( !CORBA::is_nil(idSource) )
217     mesh = idSource->GetMesh();
218
219   if( CORBA::is_nil(mesh) )
220   {
221     SUIT_MessageBox::warning( myDlg,
222                               tr( "SMESH_WRN_WARNING" ), tr("REF_IS_NULL") );
223     return false;
224   }
225
226   bool aResult = false;
227   SMESH::SMESH_Gen_var gen = SMESHGUI::GetSMESHGen();
228   SMESH::SMESH_Mesh_var newMesh;
229   QByteArray newMeshName=myDlg->myMeshName->text().toUtf8();
230   bool adapt_to_shape=myDlg->myProjShape->isChecked();
231   try
232   {
233     newMesh = gen->CreateDualMesh(mesh, newMeshName.constData(), adapt_to_shape);
234
235     if ( !newMesh->_is_nil() )
236       if ( _PTR(SObject) aSObject = SMESH::ObjectToSObject( newMesh ) )
237       {
238         anEntryList.append( aSObject->GetID().c_str() );
239
240         SMESH::SetName( aSObject, newMeshName );
241       }
242     aResult = true;
243   }
244   catch ( const SALOME::SALOME_Exception& S_ex )
245   {
246     SalomeApp_Tools::QtCatchCorbaException( S_ex );
247     aResult = false;
248   }
249   catch ( ... )
250   {
251     aResult = false;
252   }
253   if( aResult )
254   {
255     SMESHGUI::Modified();
256     selectionDone();
257     update( UF_ObjBrowser | UF_Model | UF_Viewer );
258
259   }
260   SMESHGUI::GetSMESHGUI()->getApp()->updateObjectBrowser();
261
262   // updateObjBrowser(true);
263   // SMESHGUI::Modified();
264
265   // if( LightApp_Application* anApp =
266   //     dynamic_cast<LightApp_Application*>( SUIT_Session::session()->activeApplication() ) )
267   //   anApp->browseObjects( anEntryList, true );
268
269   return true;
270
271 }
272
273 //================================================================================
274 /*! checkMesh
275  *  Verify that mesh as only tetraheadrons as 3D elements
276  */
277 //================================================================================
278
279 bool
280 SMESHGUI_CreateDualMeshOp::checkMesh( const SMESH::SMESH_IDSource_var& idSource)
281 {
282   SMESH::smIdType_array_var nbElemOfType = idSource->GetMeshInfo();
283   // Checking that the mesh only has Tetrahedron
284   bool hasOnlyTetra  = (
285                     nbElemOfType[SMDSEntity_Tetra     ] &&
286                    !nbElemOfType[SMDSEntity_Hexa      ] &&
287                    !nbElemOfType[SMDSEntity_Pyramid   ] &&
288                    !nbElemOfType[SMDSEntity_Polygon   ] &&
289                    !nbElemOfType[SMDSEntity_Penta     ] );
290
291   return hasOnlyTetra;
292 }