1 // Copyright (C) 2014-2015 EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 // Lesser General Public License for more details.
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include <HYDROGUI_ProfileInterpolateOp.h>
21 #include "HYDROGUI_Tool.h"
22 #include "HYDROGUI_Module.h"
23 #include "HYDROGUI_DataObject.h"
24 #include "HYDROGUI_UpdateFlags.h"
25 #include "HYDROGUI_ProfileInterpolateDlg.h"
27 #include <HYDROData_Stream.h>
28 #include <HYDROData_Profile.h>
29 #include <HYDROData_Document.h>
30 #include <HYDROData_ProfileUZ.h>
31 #include <HYDROData_IPolyline.h>
32 #include <HYDROData_PolylineXY.h>
33 #include <HYDROData_InterpolatorsFactory.h>
34 #include <HYDROData_IProfilesInterpolator.h>
36 #include <LightApp_Application.h>
37 #include <LightApp_SelectionMgr.h>
38 #include <LightApp_UpdateFlags.h>
40 #include <OCCViewer_ViewModel.h>
41 #include <OCCViewer_ViewPort3d.h>
42 #include <OCCViewer_ViewWindow.h>
43 #include <OCCViewer_ViewManager.h>
44 #include <OCCViewer_AISSelector.h>
46 #include <AIS_ListOfInteractive.hxx>
47 #include <AIS_ListIteratorOfListOfInteractive.hxx>
49 #include <BRep_Builder.hxx>
50 #include <TopoDS_Compound.hxx>
52 #include <QMessageBox>
54 HYDROGUI_ProfileInterpolateOp::HYDROGUI_ProfileInterpolateOp( HYDROGUI_Module* theModule )
55 : HYDROGUI_Operation( theModule )
57 setName( tr( "PROFILE_INTERPOLATION" ) );
60 HYDROGUI_ProfileInterpolateOp::~HYDROGUI_ProfileInterpolateOp()
64 void HYDROGUI_ProfileInterpolateOp::startOperation()
66 HYDROGUI_Operation::startOperation();
68 HYDROGUI_ProfileInterpolateDlg* aPanel = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
72 aPanel->setInterpolators( interpolators() );
76 void HYDROGUI_ProfileInterpolateOp::abortOperation()
78 if ( !myPreview.IsNull() )
80 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
81 if ( aDlg && !aDlg->getAISContext().IsNull() )
83 aDlg->getAISContext()->Remove( myPreview, false );
88 HYDROGUI_Operation::abortOperation();
91 void HYDROGUI_ProfileInterpolateOp::commitOperation()
93 if ( !myPreview.IsNull() )
95 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
96 if ( aDlg && !aDlg->getAISContext().IsNull() )
98 aDlg->getAISContext()->Remove( myPreview, false );
103 HYDROGUI_Operation::commitOperation();
106 HYDROGUI_InputPanel* HYDROGUI_ProfileInterpolateOp::createInputPanel() const
108 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
110 connect( aDlg, SIGNAL( riverChanged( const QString& ) ), this, SLOT( onRiverChanged( const QString& ) ) );
111 connect( aDlg, SIGNAL( interpolatorChanged( const QString& ) ), this, SLOT( onInterpolatorChanged( const QString& ) ) );
112 connect( aDlg, SIGNAL( interpolatorParametersChanged( const QString& ) ), this, SLOT( updatePreview() ) );
113 connect( aDlg, SIGNAL( profileNumberChanged( int ) ), this, SLOT( updatePreview() ) );
114 connect( aDlg, SIGNAL( profileStartChanged( const QString& ) ), this, SLOT( updatePreview() ) );
115 connect( aDlg, SIGNAL( profileFinishChanged( const QString& ) ), this, SLOT( updatePreview() ) );
120 bool HYDROGUI_ProfileInterpolateOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
121 QStringList& theBrowseObjectsEntries )
123 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
128 HYDROData_IProfilesInterpolator* anIterp = interpolator( aDlg->interpolator() );
131 updateInterpolator( anIterp );
132 Handle(HYDROData_Stream) aRiver = Handle(HYDROData_Stream)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), aDlg->river(), KIND_STREAM ) );
133 if ( !aRiver.IsNull() )
136 aRiver->Interpolate( anIterp );
137 if ( anIterp->GetErrorCode() == OK )
138 commitDocOperation();
141 errMessage = tr( "CALCULATE_ERROR" ), QString( HYDROGUI_Tool::ToQString( anIterp->GetErrorMessage() ) );
146 errMessage = tr( "CANT_GET_STREAM_OBJECT" ).arg( aDlg->river() );
148 if ( !errMessage.isEmpty() )
149 QMessageBox::critical( aDlg->topLevelWidget(), tr( "INTERPOLATION_ERROR" ), errMessage, QMessageBox::Ok );
154 bool res = errMessage.isEmpty();
157 theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
162 void HYDROGUI_ProfileInterpolateOp::updatePreview()
164 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
168 Handle(AIS_InteractiveContext) aCtx = aDlg->getAISContext();
172 if ( !myPreview.IsNull() )
173 aCtx->Remove( myPreview, false );
177 HYDROData_IProfilesInterpolator* anIterp = interpolator( aDlg->interpolator() );
180 updateInterpolator( anIterp );
181 anIterp->Calculate();
182 if ( anIterp->GetErrorCode() == OK )
184 TopoDS_Shape aShape = previewShape( anIterp );
185 if ( !aShape.IsNull() )
187 myPreview = new AIS_Shape( aShape );
188 myPreview->SetColor( Quantity_NOC_RED );
189 aCtx->Display( myPreview, 1, -1, false );
194 aCtx->UpdateCurrentViewer();
197 void HYDROGUI_ProfileInterpolateOp::updateInterpolator( HYDROData_IProfilesInterpolator* theInt )
199 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
200 if ( !aDlg || !theInt )
204 theInt->SetResultProfilesNumber( aDlg->profileNumber() );
205 theInt->SetProfiles( profile( aDlg->profileStart() ), profile( aDlg->profileFinish() ) );
206 ParamsList aParams = parameters( aDlg->interpolatorParameters() );
207 for ( ParamsList::Iterator it = aParams.begin(); it != aParams.end(); ++it ) {
208 theInt->SetParameter( HYDROGUI_Tool::ToAsciiString( (*it).first ),
209 HYDROGUI_Tool::ToAsciiString( (*it).second ) );
213 HYDROData_Profile::ProfilePoints HYDROGUI_ProfileInterpolateOp::profile( const QString& theName ) const
215 HYDROData_Profile::ProfilePoints aPoints;
216 Handle(HYDROData_Profile) aProf = Handle(HYDROData_Profile)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theName, KIND_PROFILE ) );
217 if ( !aProf.IsNull() )
219 aPoints = aProf->GetProfilePoints();
224 HYDROGUI_ProfileInterpolateOp::ParamsList HYDROGUI_ProfileInterpolateOp::parameters( const QString& theStr ) const
226 // Regular expression for parsing parameters string of kind: <param_name> <param_value> <param_name> <param_value> ..."
227 // QRegExp rx( "([A-Za-z_\\d]+)\\s+([A-Za-z_\\d]+)\\s*" );
229 // Regular expression for parsing parameters string of kind: <param_name> = <param_value> <param_name> = <param_value> ..."
230 QRegExp rx( "([A-Za-z_\\d]+)\\s*=\\s*([A-Za-z_\\d]+)\\s*" );
232 ParamsList aParamList;
233 for ( int i = 0; i >= 0; )
235 i = rx.indexIn( theStr, i );
238 aParamList.append( QPair<QString, QString>( rx.cap( 1 ), rx.cap( 2 ) ) );
239 i += rx.matchedLength();
246 QString HYDROGUI_ProfileInterpolateOp::parameters( const ParamsList& theParamList ) const
248 QStringList paramList;
249 for ( ParamsList::const_iterator it = theParamList.begin(); it != theParamList.end(); ++it )
250 paramList.append( QString( "%1 = %2" ).arg( (*it).first ).arg( (*it).second ) );
251 return paramList.join( " " );
254 QStringList HYDROGUI_ProfileInterpolateOp::interpolators() const
256 HYDROData_InterpolatorsFactory* anIFactory = 0;
257 Handle(HYDROData_Document) aDoc = doc();
258 if ( !aDoc.IsNull() )
259 anIFactory = aDoc->GetInterpolatorsFactory();
264 NCollection_Sequence<TCollection_AsciiString> iNames = anIFactory->GetInterpolatorNames();
265 for ( int i = 1, n = iNames.Size(); i <= n; ++i ) {
266 const TCollection_AsciiString& anInterpName = iNames.Value( i );
267 aNames.append( HYDROGUI_Tool::ToQString( anInterpName ) );
274 HYDROData_IProfilesInterpolator* HYDROGUI_ProfileInterpolateOp::interpolator( const QString& theName ) const
276 HYDROData_InterpolatorsFactory* anIFactory = 0;
277 Handle(HYDROData_Document) aDoc = doc();
278 if ( !aDoc.IsNull() )
279 anIFactory = aDoc->GetInterpolatorsFactory();
281 HYDROData_IProfilesInterpolator* aRes = 0;
283 aRes = anIFactory->GetInterpolator( HYDROGUI_Tool::ToAsciiString( theName ) );
287 TopoDS_Shape HYDROGUI_ProfileInterpolateOp::previewShape( HYDROData_IProfilesInterpolator* theInterp ) const
289 TopoDS_Compound aPreviewShape;
292 BRep_Builder aBuilder;
293 aBuilder.MakeCompound( aPreviewShape );
294 for ( int i = 0; i < theInterp->GetCalculatedProfilesNumber(); i++ )
296 NCollection_Sequence<gp_XYZ> pointSeq = theInterp->GetResultProfilePoints( i );
298 TopoDS_Shape aWire = HYDROData_PolylineXY::BuildWire( HYDROData_IPolyline::SECTION_SPLINE, false, pointSeq );
299 if ( !aWire.IsNull() )
300 aBuilder.Add( aPreviewShape, aWire );
304 return aPreviewShape;
307 void HYDROGUI_ProfileInterpolateOp::onInterpolatorChanged( const QString& theInterpName )
309 HYDROGUI_ProfileInterpolateDlg* aPanel = dynamic_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
310 HYDROData_IProfilesInterpolator* anInterp = interpolator( theInterpName );
311 if ( !aPanel || !anInterp )
314 aPanel->setInterpolatorDescription( HYDROGUI_Tool::ToQString( anInterp->GetDescription() ) );
319 void HYDROGUI_ProfileInterpolateOp::onRiverChanged( const QString& theRiver )
321 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
325 Handle(AIS_InteractiveContext) aCtx = aDlg->getAISContext();
326 if ( !aCtx.IsNull() )
328 AIS_ListOfInteractive aList;
329 aCtx->DisplayedObjects( aList );
330 for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
332 Handle(AIS_Shape) anObj = Handle(AIS_Shape)::DownCast( it.Value() );
333 if ( !anObj.IsNull() )
334 aCtx->Remove( anObj, false );
337 Handle(HYDROData_Stream) aStream = Handle(HYDROData_Stream)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theRiver, KIND_STREAM ) );
338 if ( !aStream.IsNull() )
340 HYDROData_SequenceOfObjects aSeq = aStream->GetProfiles();
341 for ( int i = aSeq.Lower(); i <= aSeq.Upper(); i++ )
343 Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aSeq.Value( i ) );
344 if ( !aProfile.IsNull() )
346 Handle(AIS_Shape) aPrs = new AIS_Shape( aProfile->GetShape3D() );
347 aPrs->SetOwner( aProfile );
348 aPrs->SetColor( Quantity_NOC_BLACK );
349 aCtx->Display( aPrs, 1, 0, false );
352 OCCViewer_ViewManager* vm = aDlg->viewManager();
355 QVector<SUIT_ViewWindow*> winList = vm->getViews();
356 for ( QVector<SUIT_ViewWindow*>::iterator it = winList.begin(); it != winList.end(); ++it )
358 OCCViewer_ViewWindow* occWin = ::qobject_cast<OCCViewer_ViewWindow*>( *it );
361 OCCViewer_ViewPort3d* vp = occWin->getViewPort();