+// Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
+//
+// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+#include "HYDROGUI_Poly3DOp.h"
+
+#include "HYDROGUI_Module.h"
+#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Poly3DDlg.h"
+#include "HYDROGUI_UpdateFlags.h"
+
+#include <HYDROData_Document.h>
+#include <HYDROData_Polyline3D.h>
+#include <HYDROData_Profile.h>
+#include <HYDROData_PolylineXY.h>
+
+#include <HYDROData_OperationsFactory.h>
+
+HYDROGUI_Poly3DOp::HYDROGUI_Poly3DOp( HYDROGUI_Module* theModule,
+ const bool theIsEdit )
+: HYDROGUI_Operation( theModule ),
+ myIsEdit( theIsEdit ),
+ myEditedObject( 0 )
+{
+ setName( theIsEdit ? tr( "EDIT_POLYLINE_3D" ) : tr( "CREATE_POLYLINE_3D" ) );
+}
+
+HYDROGUI_Poly3DOp::~HYDROGUI_Poly3DOp()
+{
+}
+
+HYDROGUI_InputPanel* HYDROGUI_Poly3DOp::createInputPanel() const
+{
+ return new HYDROGUI_Poly3DDlg( module(), getName() );
+}
+
+void HYDROGUI_Poly3DOp::startOperation()
+{
+ HYDROGUI_Operation::startOperation();
+
+ HYDROGUI_Poly3DDlg* aPanel = (HYDROGUI_Poly3DDlg*)inputPanel();
+ aPanel->reset();
+ aPanel->setMode( myIsEdit );
+
+ QString aPoly3DName;
+ if( myIsEdit )
+ {
+ myEditedObject = Handle(HYDROData_Polyline3D)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
+ if( !myEditedObject.IsNull() )
+ aPoly3DName = myEditedObject->GetName();
+ }
+ else
+ {
+ aPoly3DName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "POLYLINE_3D" ) );
+ }
+ aPanel->setResultName( aPoly3DName );
+
+ QString aSelectedName1, aSelectedName2;
+ if( myIsEdit && !myEditedObject.IsNull() )
+ {
+ Handle(HYDROData_Entity) anObject1 = myEditedObject->GetProfileUZ()->GetFatherObject();
+ if( !anObject1.IsNull() )
+ aSelectedName1 = anObject1->GetName();
+ Handle(HYDROData_Entity) anObject2 = myEditedObject->GetPolylineXY();
+ if( !anObject2.IsNull() )
+ aSelectedName2 = anObject2->GetName();
+ aPanel->setSelectedObjects( aSelectedName1, aSelectedName2 );
+ }
+ else if( !myIsEdit )
+ {
+ Handle(HYDROData_Profile) aSelectedProfile =
+ Handle(HYDROData_Profile)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
+ if( !aSelectedProfile.IsNull() )
+ {
+ QString aSelectedName = aSelectedProfile->GetName();
+ aPanel->setPreselectedObject( aSelectedName );
+ }
+ }
+}
+
+bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
+ QString& theErrorMsg )
+{
+ HYDROGUI_Poly3DDlg* aPanel = dynamic_cast<HYDROGUI_Poly3DDlg*>( inputPanel() );
+
+ QString aResultName = aPanel->getResultName();
+ if( aResultName.isEmpty() )
+ return false;
+
+ QString aSelectedName1, aSelectedName2;
+ if( !aPanel->getSelectedObjects( aSelectedName1, aSelectedName2 ) )
+ return false;
+
+ if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != aResultName ) )
+ {
+ // check that there are no other objects with the same name in the document
+ Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aResultName );
+ if( !anObject.IsNull() )
+ {
+ theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aResultName );
+ return false;
+ }
+ }
+
+ Handle(HYDROData_Entity) anObject1 =
+ HYDROGUI_Tool::FindObjectByName( module(), aSelectedName1, KIND_PROFILE ) ;
+ Handle(HYDROData_Entity) anObject2 =
+ HYDROGUI_Tool::FindObjectByName( module(), aSelectedName2, KIND_POLYLINEXY );
+ if( anObject1.IsNull() || anObject2.IsNull() )
+ return false;
+
+ Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( anObject1 );
+ Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( anObject2 );
+
+ if( aProfile.IsNull() || aPolyline.IsNull() )
+ return false;
+
+ Handle(HYDROData_ProfileUZ) aProfileUZ = aProfile->GetProfileUZ();
+ if( aProfileUZ.IsNull() )
+ return false;
+
+ Handle(HYDROData_Polyline3D) aResult;
+ if( myIsEdit )
+ {
+ aResult = myEditedObject;
+ aResult->RemoveProfileUZ();
+ aResult->RemovePolylineXY();
+ }
+ else
+ {
+ aResult = Handle(HYDROData_Polyline3D)::DownCast( doc()->CreateObject( KIND_POLYLINE ) );
+ }
+
+ if( aResult.IsNull() )
+ return false;
+
+ aResult->SetName( aResultName );
+ aResult->SetProfileUZ( aProfileUZ );
+ aResult->SetPolylineXY( aPolyline );
+
+ aResult->Update();
+
+ if( !myIsEdit )
+ {
+ size_t aViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
+ module()->setObjectVisible( aViewId, anObject1, false );
+ module()->setObjectVisible( aViewId, anObject2, false );
+ module()->setObjectVisible( aViewId, aResult, true );
+ }
+
+ theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced;
+ return true;
+}