Salome HOME
Issue 0020904: [CEA 411] export VTK in GEOM
[modules/geom.git] / src / MeasureGUI / MeasureGUI_AngleDlg.cxx
1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, 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.
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
20 // GEOM GEOMGUI : GUI for Geometry component
21 //  File   : MeasureGUI_AngleDlg.cxx
22 //  Author : Julia DOROVSKIKH, Open CASCADE S.A.S.
23 //
24 #include "MeasureGUI_AngleDlg.h"
25 #include "MeasureGUI_Widgets.h"
26
27 #include <DlgRef.h>
28 #include <GEOMBase.h>
29 #include <GeometryGUI.h>
30
31 #include <SUIT_Session.h>
32 #include <SUIT_Desktop.h>
33 #include <SUIT_ResourceMgr.h>
34 #include <SUIT_MessageBox.h>
35 #include <SUIT_ViewWindow.h>
36 #include <SUIT_ViewManager.h>
37 #include <SOCC_Prs.h>
38 #include <SOCC_ViewModel.h>
39 #include <SalomeApp_Tools.h>
40 #include <SalomeApp_Application.h>
41 #include <LightApp_SelectionMgr.h>
42
43 // OCCT Includes
44 #include <AIS_AngleDimension.hxx>
45 #include <AIS_ListIteratorOfListOfInteractive.hxx>
46 #include <BRep_Tool.hxx>
47 #include <BRepBuilderAPI_MakeEdge.hxx>
48 #include <BRepBuilderAPI_MakeVertex.hxx>
49 #include <TopExp.hxx>
50 #include <TopoDS.hxx>
51 #include <TopoDS_Edge.hxx>
52 #include <Geom_Plane.hxx>
53 #include <gce_MakePln.hxx>
54 #include <Precision.hxx>
55 #include <AIS.hxx>
56
57 // QT Includes
58 #include <qlineedit.h>
59 #include <qlabel.h>
60 #include <qlayout.h>
61 #include <qpushbutton.h>
62 #include <qradiobutton.h>
63 #include <qbuttongroup.h>
64
65 #include "GEOMImpl_Types.hxx"
66
67 #include "utilities.h"
68
69 #include <Standard_Failure.hxx>
70 #include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic : see Lucien PIGNOLONI / OCC
71
72 //=================================================================================
73 // class    : MeasureGUI_AngleDlg()
74 // purpose  : Constructs a MeasureGUI_AngleDlg which is a child of 'parent', with the
75 //            name 'name' and widget flags set to 'f'.
76 //            The dialog will by default be modeless, unless you set 'modal' to
77 //            TRUE to construct a modal dialog.
78 //=================================================================================
79 MeasureGUI_AngleDlg::MeasureGUI_AngleDlg (GeometryGUI* GUI, QWidget* parent)
80   : MeasureGUI_Skeleton(GUI, parent)
81 {
82   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
83   QPixmap image0 (aResMgr->loadPixmap("GEOM", tr("ICON_DLG_ANGLE")));
84   QPixmap image1 (aResMgr->loadPixmap("GEOM", tr("ICON_SELECT")));
85
86   setWindowTitle(tr("GEOM_MEASURE_ANGLE_TITLE"));
87
88   /***************************************************************/
89   mainFrame()->GroupConstructors->setTitle(tr("GEOM_MEASURE_ANGLE_ANGLE"));
90   mainFrame()->RadioButton1->setIcon(image0);
91
92   myGrp = new MeasureGUI_2Sel1LineEdit(centralWidget());
93   myGrp->GroupBox1->setTitle(tr("GEOM_MEASURE_ANGLE_OBJ"));
94   myGrp->TextLabel1->setText(tr("GEOM_OBJECT_I").arg("1"));
95   myGrp->TextLabel2->setText(tr("GEOM_OBJECT_I").arg("2"));
96   myGrp->TextLabel3->setText(tr("GEOM_MEASURE_ANGLE_IS"));
97   myGrp->LineEdit3->setReadOnly(true);
98   myGrp->PushButton1->setIcon(image1);
99   myGrp->PushButton2->setIcon(image1);
100   myGrp->LineEdit1->setReadOnly(true);
101   myGrp->LineEdit2->setReadOnly(true);
102
103   QVBoxLayout* layout = new QVBoxLayout(centralWidget());
104   layout->setMargin(0); layout->setSpacing(6);
105   layout->addWidget(myGrp);
106   /***************************************************************/
107
108   // Help page reference
109   myHelpFileName = "using_measurement_tools_page.html#angle_anchor";
110
111   // Initialisation
112   Init();
113 }
114
115 //=================================================================================
116 // function : ~MeasureGUI_AngleDlg()
117 // purpose  : Destroys the object and frees any allocated resources
118 //=================================================================================
119 MeasureGUI_AngleDlg::~MeasureGUI_AngleDlg()
120 {
121 }
122
123 //=================================================================================
124 // function : Init()
125 // purpose  :
126 //=================================================================================
127 void MeasureGUI_AngleDlg::Init()
128 {
129   // init variables
130   myGrp->LineEdit1->setText("");
131   myGrp->LineEdit2->setText("");
132   myObj = myObj2 = GEOM::GEOM_Object::_nil();
133
134   mySelBtn   = myGrp->PushButton1;
135   mySelEdit  = myGrp->LineEdit1;
136   mySelBtn2  = myGrp->PushButton2;
137   mySelEdit2 = myGrp->LineEdit2;
138
139   myEditCurrentArgument = mySelEdit;
140
141   // signals and slots connections
142   connect(mySelEdit2, SIGNAL(returnPressed()), this, SLOT(LineEditReturnPressed()));
143   connect(mySelBtn2,  SIGNAL(clicked()),       this, SLOT(SetEditCurrentArgument()));
144
145   globalSelection(GEOM_LINE);
146   MeasureGUI_Skeleton::Init();
147 }
148
149 //=================================================================================
150 // function : SelectionIntoArgument()
151 // purpose  : Called when selection has changed
152 //=================================================================================
153 void MeasureGUI_AngleDlg::SelectionIntoArgument()
154 {
155   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
156   SALOME_ListIO aSelList;
157   aSelMgr->selectedObjects(aSelList);
158
159   GEOM::GEOM_Object_var aSelectedObject = GEOM::GEOM_Object::_nil();
160
161   if (aSelList.Extent() > 0) {
162     Standard_Boolean testResult = Standard_False;
163     aSelectedObject = GEOMBase::ConvertIOinGEOMObject(aSelList.First(), testResult);
164     if (!testResult)
165       aSelectedObject = GEOM::GEOM_Object::_nil();
166   }
167
168   // clear selection
169   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
170   myGeomGUI->getApp()->selectionMgr()->clearSelected();
171   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
172           this, SLOT(SelectionIntoArgument()));
173
174   if (myEditCurrentArgument == mySelEdit) {
175     myObj = aSelectedObject;
176     if (!myObj->_is_nil() && myObj2->_is_nil())
177       myGrp->PushButton2->click();
178   }
179   else {
180     myObj2 = aSelectedObject;
181     if (!myObj2->_is_nil() && myObj->_is_nil())
182       myGrp->PushButton1->click();
183   }
184
185   processObject();
186 }
187
188 //=================================================================================
189 // function : processObject()
190 // purpose  : Fill dialogs fields in accordance with myObj and myObj2
191 //=================================================================================
192 void MeasureGUI_AngleDlg::processObject()
193 {
194   myGrp->LineEdit1->setText(!myObj->_is_nil()  ? GEOMBase::GetName(myObj ) : "");
195   myGrp->LineEdit2->setText(!myObj2->_is_nil() ? GEOMBase::GetName(myObj2) : "");
196
197   double anAngle = 0.;
198   if (getParameters(anAngle)) {
199     myGrp->LineEdit3->setText(DlgRef::PrintDoubleValue(anAngle));
200     redisplayPreview();
201   }
202   else {
203     myGrp->LineEdit3->setText("");
204     erasePreview();
205   }
206 }
207
208 //=================================================================================
209 // function : getParameters()
210 // purpose  : Get angle between objects
211 //=================================================================================
212 bool MeasureGUI_AngleDlg::getParameters (double& theAngle)
213 {
214   QString msg;
215   if (isValid(msg)) {
216     GEOM::GEOM_IMeasureOperations_var anOper = GEOM::GEOM_IMeasureOperations::_narrow( getOperation() );
217     try {
218       theAngle = anOper->GetAngle(myObj, myObj2);
219     }
220     catch(const SALOME::SALOME_Exception& e) {
221       SalomeApp_Tools::QtCatchCorbaException(e);
222       return false;
223     }
224
225     bool isDone = anOper->IsDone();
226     if (!isDone) {
227       CORBA::String_var aMsg = anOper->GetErrorCode();
228       SUIT_MessageBox::warning(this,
229                                QObject::tr("WRN_WARNING"),
230                                QObject::tr(aMsg.in()));
231     }
232     return isDone;
233   }
234
235   return false;
236 }
237
238 //=================================================================================
239 // function : SetEditCurrentArgument()
240 // purpose  :
241 //=================================================================================
242 void MeasureGUI_AngleDlg::SetEditCurrentArgument()
243 {
244   QPushButton* send = (QPushButton*)sender();
245
246   if (send == myGrp->PushButton1) {
247     myEditCurrentArgument = myGrp->LineEdit1;
248
249     myGrp->PushButton2->setDown(false);
250     myGrp->LineEdit2->setEnabled(false);
251   }
252   else {
253     myEditCurrentArgument = myGrp->LineEdit2;
254
255     myGrp->PushButton1->setDown(false);
256     myGrp->LineEdit1->setEnabled(false);
257   }
258
259   disconnect(myGeomGUI->getApp()->selectionMgr(), 0, this, 0);
260   globalSelection(GEOM_LINE);
261   connect(myGeomGUI->getApp()->selectionMgr(), SIGNAL(currentSelectionChanged()),
262           this, SLOT(SelectionIntoArgument()));
263
264   // enable line edit
265   myEditCurrentArgument->setEnabled(true);
266   myEditCurrentArgument->setFocus();
267   // after setFocus(), because it will be setDown(false) when loses focus
268   send->setDown(true);
269
270   // seems we need it only to avoid preview disappearing, caused by selection mode change
271   redisplayPreview();
272 }
273
274 //=================================================================================
275 // function : LineEditReturnPressed()
276 // purpose  :
277 //=================================================================================
278 void MeasureGUI_AngleDlg::LineEditReturnPressed()
279 {
280   QLineEdit* send = (QLineEdit*)sender();
281
282   if (send == mySelEdit)
283     myEditCurrentArgument = mySelEdit;
284   else
285     myEditCurrentArgument = mySelEdit2;
286
287   LightApp_SelectionMgr* aSelMgr = myGeomGUI->getApp()->selectionMgr();
288   SALOME_ListIO aSelList;
289   aSelMgr->selectedObjects(aSelList);
290
291   if (GEOMBase::SelectionByNameInDialogs(this, mySelEdit->text(), aSelList))
292     mySelEdit->setText(mySelEdit->text());
293 }
294
295 //=================================================================================
296 // function : buildPrs()
297 // purpose  :
298 //=================================================================================
299 SALOME_Prs* MeasureGUI_AngleDlg::buildPrs()
300 {
301   double anAngle = 0.;
302
303   SUIT_ViewWindow* vw = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
304
305   if (myObj->_is_nil() || myObj2->_is_nil() || !getParameters(anAngle) ||
306        vw->getViewManager()->getType() != OCCViewer_Viewer::Type())
307     return 0;
308
309   if (anAngle > Precision::Angular()) {
310     try {
311 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
312       OCC_CATCH_SIGNALS;
313 #endif
314       TopoDS_Shape S1, S2;
315       if (GEOMBase::GetShape(myObj , S1, TopAbs_EDGE) &&
316            GEOMBase::GetShape(myObj2, S2, TopAbs_EDGE)) {
317         TopoDS_Edge anEdge1 = TopoDS::Edge(S1);
318         TopoDS_Edge anEdge2 = TopoDS::Edge(S2);
319
320         // Build a plane for angle dimension presentation {P11, P12, P3}
321         TopoDS_Vertex V11, V12, V21, V22;
322         TopExp::Vertices(anEdge1, V11, V12);
323         TopExp::Vertices(anEdge2, V21, V22);
324
325         gp_Pnt aP11 = BRep_Tool::Pnt(TopExp::FirstVertex(anEdge1));
326         gp_Pnt aP12 = BRep_Tool::Pnt(TopExp::LastVertex (anEdge1));
327
328         gp_Pnt aP21 = BRep_Tool::Pnt(TopExp::FirstVertex(anEdge2));
329         gp_Pnt aP22 = BRep_Tool::Pnt(TopExp::LastVertex (anEdge2));
330
331         //        *P3
332         //         \
333         //          \
334         //           *P22
335         //          /
336         //         /
337         // P11    /         P12
338         // *-----/----------*
339         //  \   /
340         //   \ /
341         //    *P21
342         gp_Pnt aP3 (aP22.XYZ() + aP11.XYZ() - aP21.XYZ());
343
344         gce_MakePln gce_MP(aP11, aP12, aP3);
345         Handle(Geom_Plane) aPlane = new Geom_Plane(gce_MP.Value());
346
347         // Build the angle dimension presentation
348         QString aLabel;
349         aLabel.sprintf("%.1f", anAngle);
350
351         Handle(AIS_AngleDimension) anIO = new AIS_AngleDimension
352           (anEdge1, anEdge2, aPlane, anAngle * PI180,
353             TCollection_ExtendedString((Standard_CString)aLabel.toLatin1().data()));
354         Handle(Geom_Line) geom_lin1,geom_lin2;
355         gp_Pnt ptat11,ptat12,ptat21,ptat22;
356         Standard_Boolean isInfinite1,isInfinite2;
357         Handle(Geom_Curve) extCurv;
358         Standard_Integer extShape;
359         if (AIS::ComputeGeometry(anEdge1,
360                                   anEdge2,
361                                   extShape,
362                                   geom_lin1,
363                                   geom_lin2,
364                                   ptat11,
365                                   ptat12,
366                                   ptat21,
367                                   ptat22,
368                                   extCurv,
369                                   isInfinite1,
370                                   isInfinite2,
371                                   aPlane)) {
372           Standard_Real arrSize1 = anIO->ArrowSize();
373           Standard_Real arrSize2 = anIO->ArrowSize();
374           if (!isInfinite1) arrSize1 = ptat11.Distance(ptat12)/10.;
375           if (!isInfinite2) arrSize2 = ptat21.Distance(ptat22)/10.;
376           Standard_Real arrowSize = Max(arrSize1,arrSize2);
377           anIO->SetArrowSize(arrowSize);
378         }
379
380         SOCC_Prs* aPrs =
381           dynamic_cast<SOCC_Prs*>(((SOCC_Viewer*)(vw->getViewManager()->getViewModel()))->CreatePrs(0));
382
383         if (aPrs)
384           aPrs->AddObject(anIO);
385
386         return aPrs;
387       }
388     }
389     catch(Standard_Failure) {
390     }
391   }
392
393   return 0;
394 }
395
396 //=================================================================================
397 // function : isValid()
398 // purpose  :
399 //=================================================================================
400 bool MeasureGUI_AngleDlg::isValid (QString& msg)
401 {
402   return MeasureGUI_Skeleton::isValid(msg) && !myObj2->_is_nil();
403 }