Salome HOME
updated copyright message
[modules/geom.git] / src / MeasureGUI / MeasureGUI_CenterMassDlg.cxx
1 // Copyright (C) 2007-2023  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 // GEOM GEOMGUI : GUI for Geometry component
24 // File   : MeasureGUI_CenterMassDlg.cxx
25 // Author : Lucien PIGNOLONI, Open CASCADE S.A.S.
26 //
27 #include "MeasureGUI_CenterMassDlg.h"
28 #include "MeasureGUI_Widgets.h"
29
30 #include <DlgRef.h>
31 #include <GEOMBase.h>
32
33 #include <GeometryGUI.h>
34
35 #include <SUIT_Session.h>
36 #include <SUIT_ResourceMgr.h>
37 #include <SalomeApp_Application.h>
38 #include <LightApp_SelectionMgr.h>
39 #include <SalomeApp_Tools.h>
40
41 #include <BRep_Tool.hxx>
42 #include <TopoDS_Vertex.hxx>
43 #include <TopoDS.hxx>
44 #include <gp_Pnt.hxx>
45
46 //=================================================================================
47 // class    : MeasureGUI_CenterMassDlg()
48 // purpose  : Constructs a MeasureGUI_CenterMassDlg which is a child of 'parent', with the
49 //            name 'name' and widget flags set to 'f'.
50 //            The dialog will by default be modeless, unless you set 'modal' to
51 //            true to construct a modal dialog.
52 //=================================================================================
53 MeasureGUI_CenterMassDlg::MeasureGUI_CenterMassDlg( GeometryGUI* theGeometryGUI, QWidget* parent )
54   : GEOMBase_Skeleton( theGeometryGUI, parent, false )
55 {
56   QPixmap image0( SUIT_Session::session()->resourceMgr()->loadPixmap( "GEOM", tr( "ICON_DLG_CENTERMASS" ) ) );
57   QPixmap image1( SUIT_Session::session()->resourceMgr()->loadPixmap( "GEOM", tr( "ICON_SELECT" ) ) );
58
59   setWindowTitle( tr( "GEOM_CMASS_TITLE" ) );
60
61   /***************************************************************/
62   mainFrame()->GroupConstructors->setTitle( tr( "GEOM_CMASS" ) );
63   mainFrame()->RadioButton1->setIcon( image0 );
64   mainFrame()->RadioButton2->setAttribute( Qt::WA_DeleteOnClose );
65   mainFrame()->RadioButton2->close();
66   mainFrame()->RadioButton3->setAttribute( Qt::WA_DeleteOnClose );
67   mainFrame()->RadioButton3->close();
68
69   myGrp = new MeasureGUI_1Sel3LineEdit( centralWidget() );
70   myGrp->GroupBox1->setTitle( tr( "GEOM_CENTER" ) );
71   myGrp->TextLabel1->setText( tr( "GEOM_OBJECT" ) );
72   myGrp->TextLabel2->setText( tr( "GEOM_X" ) );
73   myGrp->TextLabel3->setText( tr( "GEOM_Y" ) );
74   myGrp->TextLabel4->setText( tr( "GEOM_Z" ) );
75   myGrp->LineEdit2->setReadOnly( true );
76   myGrp->LineEdit3->setReadOnly( true );
77   myGrp->LineEdit4->setReadOnly( true );
78   myGrp->PushButton1->setIcon( image1 );
79   myGrp->LineEdit1->setReadOnly( true );
80
81   QVBoxLayout* layout = new QVBoxLayout( centralWidget() );
82   layout->setMargin( 0 ); layout->setSpacing( 6 );
83   layout->addWidget( myGrp );
84
85   /***************************************************************/
86
87   myHelpFileName = "center_mass_page.html";
88
89   /* Initialisation */
90   Init();
91 }
92
93
94 //=================================================================================
95 // function : ~MeasureGUI_CenterMassDlg()
96 // purpose  : Destroys the object and frees any allocated resources
97 //=================================================================================
98 MeasureGUI_CenterMassDlg::~MeasureGUI_CenterMassDlg()
99 {
100 }
101
102
103 //=================================================================================
104 // function : Init()
105 // purpose  :
106 //=================================================================================
107 void MeasureGUI_CenterMassDlg::Init()
108 {
109   /* init variables */
110   myEditCurrentArgument = myGrp->LineEdit1;
111
112    /* signals and slots connections */
113   connect( buttonOk(),    SIGNAL( clicked() ), this, SLOT( ClickOnOk() ) );
114   connect( buttonApply(), SIGNAL( clicked() ), this, SLOT( ClickOnApply() ) );
115
116   connect( myGrp->LineEdit1,   SIGNAL( returnPressed() ), this, SLOT( LineEditReturnPressed() ) );
117   connect( myGrp->PushButton1, SIGNAL( clicked() ),       this, SLOT( SetEditCurrentArgument() ) );
118
119   connect( ( (SalomeApp_Application*)( SUIT_Session::session()->activeApplication() ) )->selectionMgr(),
120            SIGNAL( currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
121
122   initName( tr( "GEOM_POINT") );
123   activateSelection();
124   SelectionIntoArgument();
125 }
126
127 //=================================================================================
128 // function : activateSelection
129 // purpose  : 
130 //=================================================================================
131 void MeasureGUI_CenterMassDlg::activateSelection()
132 {
133   globalSelection( GEOM_ALLSHAPES );
134   localSelection( TopAbs_SHAPE );
135 }
136
137 //=================================================================================
138 // function : ClickOnOk()
139 // purpose  :
140 //=================================================================================
141 void MeasureGUI_CenterMassDlg::ClickOnOk()
142 {
143   if ( ClickOnApply() )
144     ClickOnCancel();
145 }
146
147
148 //=================================================================================
149 // function : ClickOnApply()
150 // purpose  :
151 //=================================================================================
152 bool MeasureGUI_CenterMassDlg::ClickOnApply()
153 {
154   if ( !onAccept() )
155     return false;
156
157   initName();
158   activateSelection();
159   return true;
160
161 //    myMeasureGUI->MakeCDGAndDisplay( myGeomShape );
162 }
163
164
165 //=================================================================================
166 // function : SelectionIntoArgument()
167 // purpose  : Called when selection as changed or other case
168 //=================================================================================
169 void MeasureGUI_CenterMassDlg::SelectionIntoArgument()
170 {
171   erasePreview();
172
173   myObj.nullify();
174   QList<TopAbs_ShapeEnum> aTypes;
175   aTypes << TopAbs_VERTEX << TopAbs_EDGE << TopAbs_WIRE << TopAbs_FACE << TopAbs_SHELL << TopAbs_SOLID << TopAbs_COMPSOLID << TopAbs_COMPOUND << TopAbs_SHAPE;
176   myObj = getSelected( aTypes );
177
178   if ( !myObj ) {
179     processObject();
180     return;
181   }
182
183   processObject();
184   displayPreview(true);
185 }
186
187 //=================================================================================
188 // function : SetEditCurrentArgument()
189 // purpose  :
190 //=================================================================================
191 void MeasureGUI_CenterMassDlg::SetEditCurrentArgument()
192 {
193   myGrp->LineEdit1->setFocus();
194   myEditCurrentArgument = myGrp->LineEdit1;
195   SelectionIntoArgument();
196 }
197
198
199 //=================================================================================
200 // function : LineEditReturnPressed()
201 // purpose  :
202 //=================================================================================
203 void MeasureGUI_CenterMassDlg::LineEditReturnPressed()
204 {
205   QLineEdit* send = ( QLineEdit* )sender();
206   if ( send == myGrp->LineEdit1 ) {
207     myEditCurrentArgument = myGrp->LineEdit1;
208     GEOMBase_Skeleton::LineEditReturnPressed();
209   }
210 }
211
212
213 //=================================================================================
214 // function : ActivateThisDialog()
215 // purpose  :
216 //=================================================================================
217 void MeasureGUI_CenterMassDlg::ActivateThisDialog()
218 {
219   GEOMBase_Skeleton::ActivateThisDialog();
220
221   connect( ( (SalomeApp_Application*)( SUIT_Session::session()->activeApplication() ) )->selectionMgr(), 
222            SIGNAL(currentSelectionChanged() ), this, SLOT( SelectionIntoArgument() ) );
223
224   activateSelection();
225   displayPreview(true);
226 }
227
228 //=================================================================================
229 // function : processObject()
230 // purpose  : Fill dialog fields in accordance with myObj
231 //=================================================================================
232 void MeasureGUI_CenterMassDlg::processObject()
233 {
234   if ( !myObj ) {
235     myGrp->LineEdit1->setText( "" );
236     myGrp->LineEdit2->setText( "" );
237     myGrp->LineEdit3->setText( "" );
238     myGrp->LineEdit4->setText( "" );
239     erasePreview();
240   }
241   else {
242     double x = 0, y = 0, z = 0;
243     
244     getParameters( x, y, z );
245     
246     myGrp->LineEdit1->setText( GEOMBase::GetName( myObj.get() ) );
247
248     SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
249     int aPrecision = resMgr->integerValue( "Geometry", "length_precision", 6 );
250     myGrp->LineEdit2->setText( DlgRef::PrintDoubleValue( x, aPrecision ) );
251     myGrp->LineEdit3->setText( DlgRef::PrintDoubleValue( y, aPrecision ) );
252     myGrp->LineEdit4->setText( DlgRef::PrintDoubleValue( z, aPrecision ) );
253
254     displayPreview(true);
255   }
256 }
257
258
259 //=================================================================================
260 // function : enterEvent()
261 // purpose  :
262 //=================================================================================
263 void MeasureGUI_CenterMassDlg::enterEvent( QEvent* )
264 {
265   if ( !mainFrame()->GroupConstructors->isEnabled() )
266     ActivateThisDialog();
267 }
268
269 //=================================================================================
270 // function : createOperation
271 // purpose  :
272 //=================================================================================
273 GEOM::GEOM_IOperations_ptr MeasureGUI_CenterMassDlg::createOperation()
274 {
275   return getGeomEngine()->GetIMeasureOperations();
276 }
277
278 //=================================================================================
279 // function : isValid
280 // purpose  :
281 //=================================================================================
282 bool MeasureGUI_CenterMassDlg::isValid( QString& )
283 {
284   return myObj;
285 }
286
287 //=================================================================================
288 // function : execute
289 // purpose  :
290 //=================================================================================
291 bool MeasureGUI_CenterMassDlg::getParameters( double& theX, double& theY, double& theZ )
292 {
293   if ( !myObj )
294     return false;
295   else {
296     try {
297       GEOM::GEOM_IMeasureOperations_var anOper = GEOM::GEOM_IMeasureOperations::_narrow( getOperation() );
298       GEOM::GEOM_Object_var anObj = anOper->GetCentreOfMass( myObj.get() );
299       if ( !anOper->IsDone() )
300         return false;
301
302       TopoDS_Shape aShape;
303       if ( !GEOMBase::GetShape( anObj, aShape ) ||
304            aShape.IsNull() ||
305            aShape.ShapeType() != TopAbs_VERTEX )
306         return false;
307
308       TopoDS_Vertex aVertex = TopoDS::Vertex( aShape );
309
310       gp_Pnt aPnt = BRep_Tool::Pnt( aVertex );
311
312       theX = aPnt.X();
313       theY = aPnt.Y();
314       theZ = aPnt.Z();
315
316       return true;
317     }
318     catch( const SALOME::SALOME_Exception& e ) {
319       SalomeApp_Tools::QtCatchCorbaException( e );
320       return false;
321     }
322   }
323 }
324
325 //=================================================================================
326 // function : execute
327 // purpose  :
328 //=================================================================================
329 bool MeasureGUI_CenterMassDlg::execute( ObjectList& objects )
330 {
331   GEOM::GEOM_IMeasureOperations_var anOper = GEOM::GEOM_IMeasureOperations::_narrow( getOperation() );
332   GEOM::GEOM_Object_var anObj = anOper->GetCentreOfMass( myObj.get() );
333
334   if ( !anObj->_is_nil() )
335     objects.push_back( anObj._retn() );
336
337   return true;
338 }
339
340 //=================================================================================
341 // function : addSubshapeToStudy
342 // purpose  : virtual method to add new SubObjects if local selection
343 //=================================================================================
344 void MeasureGUI_CenterMassDlg::addSubshapesToStudy()
345 {
346   GEOMBase::PublishSubObject( myObj.get() );
347 }
348
349 //=================================================================================
350 // function : getSourceObjects
351 // purpose  : virtual method to get source objects
352 //=================================================================================
353 QList<GEOM::GeomObjPtr> MeasureGUI_CenterMassDlg::getSourceObjects()
354 {
355   QList<GEOM::GeomObjPtr> res;
356   res << myObj;
357   return res;
358 }