1 // Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #include <HYDROGUI_ProfileInterpolateOp.h>
25 #include "HYDROGUI_Tool.h"
26 #include "HYDROGUI_Module.h"
27 #include "HYDROGUI_DataObject.h"
28 #include "HYDROGUI_UpdateFlags.h"
29 #include "HYDROGUI_ProfileInterpolateDlg.h"
31 #include <HYDROData_Stream.h>
32 #include <HYDROData_Profile.h>
33 #include <HYDROData_Document.h>
34 #include <HYDROData_ProfileUZ.h>
35 #include <HYDROData_IPolyline.h>
36 #include <HYDROData_PolylineXY.h>
37 #include <HYDROData_InterpolatorsFactory.h>
38 #include <HYDROData_IProfilesInterpolator.h>
40 #include <LightApp_Application.h>
41 #include <LightApp_SelectionMgr.h>
42 #include <LightApp_UpdateFlags.h>
44 #include <OCCViewer_ViewModel.h>
45 #include <OCCViewer_ViewManager.h>
46 #include <OCCViewer_ViewWindow.h>
47 #include <OCCViewer_AISSelector.h>
49 #include <AIS_ListOfInteractive.hxx>
50 #include <AIS_ListIteratorOfListOfInteractive.hxx>
52 #include <BRep_Builder.hxx>
53 #include <TopoDS_Compound.hxx>
55 #include <QMessageBox>
57 HYDROGUI_ProfileInterpolateOp::HYDROGUI_ProfileInterpolateOp( HYDROGUI_Module* theModule )
58 : HYDROGUI_Operation( theModule )
60 setName( tr( "PROFILE_INTERPOLATION" ) );
63 HYDROGUI_ProfileInterpolateOp::~HYDROGUI_ProfileInterpolateOp()
67 void HYDROGUI_ProfileInterpolateOp::startOperation()
69 HYDROGUI_Operation::startOperation();
71 HYDROGUI_ProfileInterpolateDlg* aPanel = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
75 aPanel->setInterpolators( interpolators() );
79 void HYDROGUI_ProfileInterpolateOp::abortOperation()
81 if ( !myPreview.IsNull() )
83 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
84 if ( aDlg && !aDlg->getAISContext().IsNull() )
86 aDlg->getAISContext()->Remove( myPreview, false );
91 HYDROGUI_Operation::abortOperation();
94 void HYDROGUI_ProfileInterpolateOp::commitOperation()
96 if ( !myPreview.IsNull() )
98 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
99 if ( aDlg && !aDlg->getAISContext().IsNull() )
101 aDlg->getAISContext()->Remove( myPreview, false );
106 HYDROGUI_Operation::commitOperation();
109 HYDROGUI_InputPanel* HYDROGUI_ProfileInterpolateOp::createInputPanel() const
111 HYDROGUI_ProfileInterpolateDlg* aDlg = new HYDROGUI_ProfileInterpolateDlg( module(), getName() );
113 connect( aDlg, SIGNAL( riverChanged( const QString& ) ), this, SLOT( onRiverChanged( const QString& ) ) );
114 connect( aDlg, SIGNAL( interpolatorChanged( const QString& ) ), this, SLOT( onInterpolatorChanged( const QString& ) ) );
115 connect( aDlg, SIGNAL( interpolatorParametersChanged( const QString& ) ), this, SLOT( updatePreview() ) );
116 connect( aDlg, SIGNAL( profileNumberChanged( int ) ), this, SLOT( updatePreview() ) );
117 connect( aDlg, SIGNAL( profileStartChanged( const QString& ) ), this, SLOT( updatePreview() ) );
118 connect( aDlg, SIGNAL( profileFinishChanged( const QString& ) ), this, SLOT( updatePreview() ) );
123 bool HYDROGUI_ProfileInterpolateOp::processApply( int& theUpdateFlags, QString& theErrorMsg,
124 QStringList& theBrowseObjectsEntries )
126 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
131 HYDROData_IProfilesInterpolator* anIterp = interpolator( aDlg->interpolator() );
134 updateInterpolator( anIterp );
135 Handle(HYDROData_Stream) aRiver = Handle(HYDROData_Stream)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), aDlg->river(), KIND_STREAM ) );
136 if ( !aRiver.IsNull() )
139 aRiver->Interpolate( anIterp );
140 if ( anIterp->GetErrorCode() == OK )
141 commitDocOperation();
144 errMessage = tr( "CALCULATE_ERROR" ), QString( anIterp->GetErrorMessage().c_str() );
149 errMessage = tr( "CANT_GET_RIVER_OBJECT" ).arg( aDlg->river() );
151 if ( !errMessage.isEmpty() )
152 QMessageBox::critical( aDlg->topLevelWidget(), tr( "INTERPOLATION_ERROR" ), errMessage, QMessageBox::Ok );
157 bool res = errMessage.isEmpty();
160 theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
165 void HYDROGUI_ProfileInterpolateOp::updatePreview()
167 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
171 Handle(AIS_InteractiveContext) aCtx = aDlg->getAISContext();
175 if ( !myPreview.IsNull() )
176 aCtx->Remove( myPreview, false );
180 HYDROData_IProfilesInterpolator* anIterp = interpolator( aDlg->interpolator() );
183 updateInterpolator( anIterp );
184 anIterp->Calculate();
185 if ( anIterp->GetErrorCode() == OK )
187 TopoDS_Shape aShape = previewShape( anIterp );
188 if ( !aShape.IsNull() )
190 myPreview = new AIS_Shape( aShape );
191 myPreview->SetColor( Quantity_NOC_RED );
192 aCtx->Display( myPreview, 0, -1, false );
197 aCtx->UpdateCurrentViewer();
200 void HYDROGUI_ProfileInterpolateOp::updateInterpolator( HYDROData_IProfilesInterpolator* theInt )
202 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
203 if ( !aDlg || !theInt )
207 theInt->SetResultProfilesNumber( aDlg->profileNumber() );
208 theInt->SetProfiles( profile( aDlg->profileStart() ), profile( aDlg->profileFinish() ) );
209 ParamsList aParams = parameters( aDlg->interpolatorParameters() );
210 for ( ParamsList::Iterator it = aParams.begin(); it != aParams.end(); ++it )
211 theInt->SetParameter( (*it).first.toStdString(), (*it).second.toStdString() );
214 std::vector<double> HYDROGUI_ProfileInterpolateOp::profile( const QString& theName ) const
216 std::vector<double> aPoints;
217 Handle(HYDROData_Profile) aProf = Handle(HYDROData_Profile)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theName, KIND_PROFILE ) );
218 if ( !aProf.IsNull() )
220 HYDROData_Profile::ProfilePoints aPntList = aProf->GetProfilePoints();
221 for ( int i = aPntList.Lower(); i <= aPntList.Upper(); i++ )
223 HYDROData_Profile::ProfilePoint aPnt = aPntList.Value( i );
224 aPoints.push_back( aPnt.X() );
225 aPoints.push_back( aPnt.Y() );
226 aPoints.push_back( aPnt.Z() );
232 HYDROGUI_ProfileInterpolateOp::ParamsList HYDROGUI_ProfileInterpolateOp::parameters( const QString& theStr ) const
234 // Regular expression for parsing parameters string of kind: <param_name> <param_value> <param_name> <param_value> ..."
235 // QRegExp rx( "([A-Za-z_\\d]+)\\s+([A-Za-z_\\d]+)\\s*" );
237 // Regular expression for parsing parameters string of kind: <param_name> = <param_value> <param_name> = <param_value> ..."
238 QRegExp rx( "([A-Za-z_\\d]+)\\s*=\\s*([A-Za-z_\\d]+)\\s*" );
240 ParamsList aParamList;
241 for ( int i = 0; i >= 0; )
243 i = rx.indexIn( theStr, i );
246 aParamList.append( QPair<QString, QString>( rx.cap( 1 ), rx.cap( 2 ) ) );
247 i += rx.matchedLength();
254 QString HYDROGUI_ProfileInterpolateOp::parameters( const ParamsList& theParamList ) const
256 QStringList paramList;
257 for ( ParamsList::const_iterator it = theParamList.begin(); it != theParamList.end(); ++it )
258 paramList.append( QString( "%1 = %2" ).arg( (*it).first ).arg( (*it).second ) );
259 return paramList.join( " " );
262 QStringList HYDROGUI_ProfileInterpolateOp::interpolators() const
264 HYDROData_InterpolatorsFactory* anIFactory = 0;
265 Handle(HYDROData_Document) aDoc = doc();
266 if ( !aDoc.IsNull() )
267 anIFactory = aDoc->GetInterpolatorsFactory();
272 std::vector<std::string> iNames = anIFactory->GetInterpolatorNames();
273 for ( std::vector<std::string>::iterator it = iNames.begin(); it != iNames.end(); ++it )
274 aNames.append( (*it).c_str() );
280 HYDROData_IProfilesInterpolator* HYDROGUI_ProfileInterpolateOp::interpolator( const QString& theName ) const
282 HYDROData_InterpolatorsFactory* anIFactory = 0;
283 Handle(HYDROData_Document) aDoc = doc();
284 if ( !aDoc.IsNull() )
285 anIFactory = aDoc->GetInterpolatorsFactory();
287 HYDROData_IProfilesInterpolator* aRes = 0;
289 aRes = anIFactory->GetInterpolator( theName.toStdString() );
293 TopoDS_Shape HYDROGUI_ProfileInterpolateOp::previewShape( HYDROData_IProfilesInterpolator* theInterp ) const
295 TopoDS_Compound aPreviewShape;
298 BRep_Builder aBuilder;
299 aBuilder.MakeCompound( aPreviewShape );
300 for ( int i = 0; i < theInterp->GetCalculatedProfilesNumber(); i++ )
302 NCollection_Sequence<gp_XYZ> pointSeq;
303 std::vector<double> aPoints = theInterp->GetResultProfile( i );
304 for ( int i = 0; i < (int) aPoints.size(); i += 3 )
305 pointSeq.Append( gp_XYZ( aPoints[i], aPoints[i+1], aPoints[i+2] ) );
306 TopoDS_Shape aWire = HYDROData_PolylineXY::BuildWire( HYDROData_IPolyline::SECTION_SPLINE, false, pointSeq );
307 if ( !aWire.IsNull() )
308 aBuilder.Add( aPreviewShape, aWire );
312 return aPreviewShape;
315 void HYDROGUI_ProfileInterpolateOp::onInterpolatorChanged( const QString& theInterpName )
317 HYDROGUI_ProfileInterpolateDlg* aPanel = dynamic_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
318 HYDROData_IProfilesInterpolator* anInterp = interpolator( theInterpName );
319 if ( !aPanel || !anInterp )
322 aPanel->setInterpolatorDescription( QString( anInterp->GetDescription().c_str() ) );
327 void HYDROGUI_ProfileInterpolateOp::onRiverChanged( const QString& theRiver )
329 HYDROGUI_ProfileInterpolateDlg* aDlg = ::qobject_cast<HYDROGUI_ProfileInterpolateDlg*>( inputPanel() );
333 Handle(AIS_InteractiveContext) aCtx = aDlg->getAISContext();
334 if ( !aCtx.IsNull() )
336 AIS_ListOfInteractive aList;
337 aCtx->DisplayedObjects( aList );
338 for ( AIS_ListIteratorOfListOfInteractive it( aList ); it.More(); it.Next() )
340 Handle(AIS_Shape) anObj = Handle(AIS_Shape)::DownCast( it.Value() );
341 if ( !anObj.IsNull() )
342 aCtx->Remove( anObj, false );
345 Handle(HYDROData_Stream) aStream = Handle(HYDROData_Stream)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theRiver, KIND_STREAM ) );
346 if ( !aStream.IsNull() )
348 HYDROData_SequenceOfObjects aSeq = aStream->GetProfiles();
349 for ( int i = aSeq.Lower(); i <= aSeq.Upper(); i++ )
351 Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aSeq.Value( i ) );
352 if ( !aProfile.IsNull() )
354 Handle(AIS_Shape) aPrs = new AIS_Shape( aProfile->GetShape3D() );
355 aPrs->SetOwner( aProfile );
356 aPrs->SetColor( Quantity_NOC_BLACK );
357 aCtx->Display( aPrs, 0, 0, false );