+ HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportImageDlg( module(), getName() );
+ connect( aPanel, SIGNAL( createPreview( QImage ) ),
+ this, SLOT( onCreatePreview( QImage ) ) );
+ connect( aPanel, SIGNAL( activatePointSelection( int ) ),
+ this, SLOT( onActivatePointSelection( int ) ) );
+ connect( aPanel, SIGNAL( pointCoordChanged( bool, int, bool, int ) ),
+ this, SLOT( onPointCoordChanged( bool, int, bool, int ) ) );
+ connect( aPanel, SIGNAL( modeActivated( int ) ),
+ this, SLOT( onModeActivated( int ) ) );
+ connect( aPanel, SIGNAL( refImageActivated( const QString& ) ),
+ this, SLOT( onRefImageActivated( const QString& ) ) );
+ return aPanel;
+}
+
+bool HYDROGUI_ImportImageOp::checkPoints( const QPointF& thePointA,
+ const QPointF& thePointB,
+ const QPointF& thePointC,
+ const bool theIsByTwoPoints,
+ const QString& theLineErrMsg,
+ const QString& thePoinErrMsg,
+ QString& theErrorMsg,
+ const bool theIsToCheckInvertibles ) const
+{
+ double xa = thePointA.x();
+ double ya = thePointA.y();
+ double xb = thePointB.x();
+ double yb = thePointB.y();
+ double xc = thePointC.x();
+ double yc = thePointC.y();
+
+ if ( !theIsByTwoPoints )
+ {
+ // check that three input points don't belong to a single line
+ if ( ValuesEquals( ( yb - ya ) * ( xc - xa ), ( yc - ya ) * ( xb - xa ) ) )
+ {
+ theErrorMsg = theLineErrMsg;
+ return false;
+ }
+ }
+ else
+ {
+ // check that two points are not identical
+ if ( ValuesEquals( xa, ya ) && ValuesEquals( xb, yb ) )
+ {
+ theErrorMsg = thePoinErrMsg;
+ return false;
+ }
+ }
+
+ if ( theIsToCheckInvertibles )
+ {
+ QTransform aTransform1( xa, ya, 1, xb, yb, 1, xc, yc, 1 );
+
+ bool anIsInvertible = false;
+ QTransform aTransform1Inverted = aTransform1.inverted( &anIsInvertible );
+ if( !anIsInvertible )
+ {
+ theErrorMsg = tr( "TRANSFORMATION_MATRIX_CANNOT_BE_COMPUTED" );
+ return false;
+ }
+ }
+
+ return true;
+}
+
+bool HYDROGUI_ImportImageOp::processApply( int& theUpdateFlags,
+ QString& theErrorMsg )
+{
+ HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+
+ QString anImageName = aPanel->getImageName();
+ if( anImageName.isEmpty() )
+ return false;
+
+ if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anImageName ) )
+ {
+ // check that there are no other objects with the same name in the document
+ Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anImageName );
+ if( !anObject.IsNull() )
+ {
+ theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anImageName );
+ return false;
+ }
+ }
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aMap;
+ if( !aPanel->getTransformationDataMap( aMap ) )
+ return false;
+
+ bool anIsByTwoPoints = aPanel->isByTwoPoints();
+
+ QPoint aPointA = aMap[ HYDROGUI_PrsImage::PointA ].ImagePoint;
+ QPoint aPointB = aMap[ HYDROGUI_PrsImage::PointB ].ImagePoint;
+ QPoint aPointC = anIsByTwoPoints ? QPoint( INT_MIN, INT_MIN ) :
+ aMap[ HYDROGUI_PrsImage::PointC ].ImagePoint;
+
+ // first, we check correctness of image points
+ if ( !checkPoints( aPointA, aPointB, aPointC, anIsByTwoPoints,
+ tr( "POINTS_A_B_C_BELONG_TO_SINGLE_LINE" ),
+ tr( "POINTS_A_B_ARE_IDENTICAL" ),
+ theErrorMsg, true ) )
+ return false;
+
+ Handle(HYDROData_Image) aRefImageObj;
+ QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC( INT_MIN, INT_MIN );
+
+ HYDROData_Image::TransformationMode aTransformationMode =
+ (HYDROData_Image::TransformationMode)aPanel->getTransformationMode();
+ if ( aTransformationMode == HYDROData_Image::ReferenceImage )
+ {
+ QString aRefImageName = aPanel->getRefImageName();
+ if( aRefImageName.isEmpty() )
+ {
+ theErrorMsg = tr( "REFERENCE_IMAGE_IS_NOT_SELECTED" );
+ return false;
+ }
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aRefMap;
+ if( !aPanel->getTransformationDataMap( aRefMap, true ) )
+ return false;
+
+ aRefImageObj = Handle(HYDROData_Image)::DownCast(
+ HYDROGUI_Tool::FindObjectByName( module(), aRefImageName, KIND_IMAGE ) );
+ if( aRefImageObj.IsNull() )
+ return false;
+
+ aTrsfPointA = aRefMap[ HYDROGUI_PrsImage::PointA ].ImagePoint;
+ aTrsfPointB = aRefMap[ HYDROGUI_PrsImage::PointB ].ImagePoint;
+ if ( !anIsByTwoPoints )
+ aTrsfPointC = aRefMap[ HYDROGUI_PrsImage::PointC ].ImagePoint;
+
+ // the same check of correctness for the reference points
+ if ( !checkPoints( aTrsfPointA, aTrsfPointB, aTrsfPointC, anIsByTwoPoints,
+ tr( "REFERENCE_POINTS_A_B_C_BELONG_TO_SINGLE_LINE" ),
+ tr( "REFERENCE_POINTS_A_B_ARE_IDENTICAL" ),
+ theErrorMsg, false ) )
+ return false;
+ }
+ else
+ {
+ if ( aTransformationMode == HYDROData_Image::ManualLambert )
+ {
+ aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].LambertPoint;
+ aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].LambertPoint;
+ if ( !anIsByTwoPoints )
+ aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].LambertPoint;
+ }
+ else
+ {
+ aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].CartesianPoint;
+ aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].CartesianPoint;
+ if ( !anIsByTwoPoints )
+ aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].CartesianPoint;
+ }
+ }
+
+ Handle(HYDROData_Image) anImageObj;
+ if( myIsEdit )
+ anImageObj = myEditedObject;
+ else
+ anImageObj = Handle(HYDROData_Image)::DownCast( doc()->CreateObject( KIND_IMAGE ) );
+
+ if( anImageObj.IsNull() )
+ return false;
+
+ anImageObj->SetName( anImageName );
+ anImageObj->SetImage( myImage );
+ anImageObj->SetLocalPoints( aPointA, aPointB, aPointC );
+
+ if ( aTransformationMode == HYDROData_Image::ReferenceImage )
+ {
+ anImageObj->SetReferencePoints( aRefImageObj,
+ aTrsfPointA, aTrsfPointB, aTrsfPointC );
+ }
+ else
+ {
+ anImageObj->SetGlobalPoints( aTransformationMode,
+ aTrsfPointA, aTrsfPointB, aTrsfPointC );
+ }
+
+ if( !myIsEdit )
+ {
+ // Set imported file name for image
+ QString aFilePath = aPanel->getFileName();
+ anImageObj->SetFilePath( aFilePath );
+ }
+
+ // must be done after all checks and before calling SetVisible() method below
+ closePreview();
+
+ if( !myIsEdit )
+ module()->setObjectVisible( HYDROGUI_Tool::GetActiveGraphicsViewId( module() ), anImageObj, true );
+
+ if( myIsEdit )
+ anImageObj->Update();
+
+ theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced;
+ return true;
+}
+
+void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage )
+{
+ HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+
+ myImage = theImage;
+
+ if( myPreviewPrs ) // if the image is changed by choosing another file
+ {
+ myPreviewPrs->setImage( myImage );
+ myPreviewPrs->compute();
+
+ if( myPreviewViewManager )
+ {
+ if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() )
+ {
+ if( GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort() )
+ {
+ aViewPort->onBoundingRectChanged();
+ aViewPort->fitAll();
+ }
+ }
+ }
+ }
+ else
+ {
+ LightApp_Application* anApp = module()->getApp();
+
+ myActiveViewManager = anApp->activeViewManager();
+
+ myPreviewPrs = new HYDROGUI_PrsImage( myIsEdit ? myEditedObject : 0 );
+ myPreviewPrs->setImage( myImage );
+ myPreviewPrs->compute();
+
+ myPreviewViewManager =
+ dynamic_cast<GraphicsView_ViewManager*>( anApp->createViewManager( GraphicsView_Viewer::Type() ) );
+ if( myPreviewViewManager )
+ {
+ connect( myPreviewViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
+ this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
+
+ module()->setViewManagerRole( myPreviewViewManager, HYDROGUI_Module::VMR_TransformImage );
+ myPreviewViewManager->setTitle( tr( "TRANSFORM_IMAGE" ) );
+ if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() )
+ {
+ if( GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort() )
+ {
+ //aViewPort->setMousePositionEnabled( true ); //ouv: temporarily commented
+
+ aViewPort->addItem( myPreviewPrs );
+ aViewPort->fitAll();
+
+ myPreviewPrs->setIsTransformationPointPreview( true );
+ }
+ connect( aViewer, SIGNAL( selectionChanged( GV_SelectionChangeStatus ) ),
+ this, SLOT( onPointSelected() ) );
+ }
+ }
+
+ QString anImageName;
+ if( myIsEdit )
+ {
+ if( !myEditedObject.IsNull() )
+ anImageName = myEditedObject->GetName();
+ }
+ else
+ anImageName = HYDROGUI_Tool::GenerateObjectName( module(), "Image" );
+ aPanel->setImageName( anImageName );
+ }
+
+ aPanel->setImageSize( myImage.size() );
+
+ aPanel->initializePointSelection();
+ onPointSelected( false );
+}
+
+void HYDROGUI_ImportImageOp::onActivatePointSelection( int thePointType )
+{
+ myPointType = thePointType;
+ if( myPreviewPrs )
+ myPreviewPrs->setTransformationPointType( thePointType );
+ if( myRefPreviewPrs )
+ myRefPreviewPrs->setTransformationPointType( thePointType );
+}
+
+void HYDROGUI_ImportImageOp::onPointCoordChanged( bool theIsRef,
+ int thePointType,
+ bool theIsY,
+ int theValue )
+{
+ if( !theIsRef && myPreviewPrs )
+ myPreviewPrs->updateTransformationPoint( thePointType, theIsY, theValue );
+ else if( theIsRef && myRefPreviewPrs )
+ myRefPreviewPrs->updateTransformationPoint( thePointType, theIsY, theValue );
+}
+
+void HYDROGUI_ImportImageOp::onModeActivated( int theMode )
+{
+ HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+
+ QString aRefImageName;
+ if( theMode == HYDROData_Image::ReferenceImage )
+ {
+ aRefImageName = aPanel->getRefImageName();
+ if( aRefImageName.isEmpty() )
+ return; // do nothing in this case to avoid visual moving of preview prs
+ onRefImageActivated( aRefImageName );
+ }
+ else
+ {
+ if( myRefViewManager )
+ {
+ closeView( myRefViewManager );
+ }
+ }
+}
+
+void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName )
+{
+ GraphicsView_ViewPort* aViewPort = 0;
+
+ LightApp_Application* anApp = module()->getApp();
+ ///// Get a view port for the reference image preview
+ if( myRefViewManager )
+ {
+ if( GraphicsView_Viewer* aViewer = myRefViewManager->getViewer() )
+ {
+ aViewPort = aViewer->getActiveViewPort();
+ }
+ }
+ else
+ {
+ anApp = module()->getApp();
+ // Init reference image preview
+ myRefViewManager =
+ dynamic_cast<GraphicsView_ViewManager*>( anApp->createViewManager( GraphicsView_Viewer::Type() ) );
+ if( myRefViewManager )
+ {
+ connect( myRefViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
+ this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
+
+ module()->setViewManagerRole( myRefViewManager, HYDROGUI_Module::VMR_ReferenceImage );
+ myRefViewManager->setTitle( tr( "REFERENCE_IMAGE" ) );
+ if( GraphicsView_Viewer* aViewer = myRefViewManager->getViewer() )
+ {
+ aViewPort = aViewer->getActiveViewPort();
+ connect( aViewer, SIGNAL( selectionChanged( GV_SelectionChangeStatus ) ),
+ this, SLOT( onRefPointSelected() ) );
+ }
+ }
+ }
+ qApp->processEvents();
+
+ if( !aViewPort )
+ return;
+
+ // Remove the old presentation of the reference image if any
+ if( myRefPreviewPrs )
+ {
+ myRefPreviewPrs->setCaption( QString() );
+ aViewPort->removeItem( myRefPreviewPrs );
+
+ delete myRefPreviewPrs;
+ myRefPreviewPrs = 0;
+ }
+
+ // Create a new reference image presentation
+ QImage anImage;
+ Handle(HYDROData_Image) anImageObj = Handle(HYDROData_Image)::DownCast(
+ HYDROGUI_Tool::FindObjectByName( module(), theName, KIND_IMAGE ) );
+ if( !anImageObj.IsNull() )
+ {
+ anImage = anImageObj->Image();
+
+ myRefPreviewPrs = new HYDROGUI_PrsImage( anImageObj );
+ myRefPreviewPrs->setImage( anImage );
+ myRefPreviewPrs->compute();
+
+ myRefPreviewPrs->setIsTransformationPointPreview( true );
+ myRefPreviewPrs->setTransformationPointType( myPointType );
+
+ // Add the new reference image presentation to the appropriate view
+ aViewPort->addItem( myRefPreviewPrs );
+ }
+
+ aViewPort->fitAll();
+
+ // Split views horizontally
+ if( anApp->desktop()->inherits( "STD_TabDesktop" ) )
+ {
+ QtxWorkstack* aWorkstack = ( (STD_TabDesktop*)anApp->desktop() )->workstack();
+ aViewPort->activateWindow();
+ aViewPort->show();
+ aViewPort->setFocus(Qt::ActiveWindowFocusReason);
+ aWorkstack->splitHorizontal();
+ }
+
+ // Initialize the dialog
+ HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+ aPanel->setImageSize( anImage.size(), true );
+ aPanel->initializePointSelection();
+ onPointSelected( true );
+}
+
+void HYDROGUI_ImportImageOp::onLastViewClosed( SUIT_ViewManager* theViewManager )
+{
+ closePreview();
+}
+
+void HYDROGUI_ImportImageOp::onPointSelected()
+{
+ onPointSelected( myRefPreviewPrs && myRefPreviewPrs->isSelected() );
+}
+
+void HYDROGUI_ImportImageOp::onRefPointSelected()
+{
+ onPointSelected( true );
+}
+
+void HYDROGUI_ImportImageOp::onPointSelected( bool theIsRefImage )
+{
+ HYDROGUI_PrsImage* aPrs = theIsRefImage ? myRefPreviewPrs : myPreviewPrs;
+ if( !aPrs )
+ return;
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap;
+
+ const HYDROGUI_PrsImage::TransformationPointMap& aPointMap =
+ aPrs->getTransformationPointMap();
+ HYDROGUI_PrsImage::TransformationPointMapIterator anIter( aPointMap );
+ while( anIter.hasNext() )
+ {
+ int aPointType = anIter.next().key();
+ const HYDROGUI_PrsImage::TransformationPoint& aTransformationPoint = anIter.value();
+ const QPoint& aPoint = aTransformationPoint.Point;
+
+ HYDROGUI_ImportImageDlg::TransformationData aData( aPoint, QPointF(), QPointF() );
+ aDataMap[ aPointType ] = aData;
+ }
+
+ ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap, true, theIsRefImage );
+}
+
+void HYDROGUI_ImportImageOp::closePreview()
+{
+ closeView( myPreviewViewManager );
+ closeView( myRefViewManager );
+
+ if( myActiveViewManager )
+ HYDROGUI_Tool::SetActiveViewManager( module(), myActiveViewManager );
+}
+
+void HYDROGUI_ImportImageOp::closeView( GraphicsView_ViewManager* &aViewMgr )
+{
+ if( aViewMgr )
+ {
+ GraphicsView_ViewPort* aViewPort = 0;
+ if( GraphicsView_Viewer* aViewer = aViewMgr->getViewer() )
+ {
+ aViewPort = aViewer->getActiveViewPort();
+ }
+ disconnect( aViewMgr, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
+ this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
+
+ // Nullify appropriate presentation pointer
+ HYDROGUI_PrsImage* aPrs = 0;
+ switch ( module()->getViewManagerRole( aViewMgr ) )
+ {
+ case HYDROGUI_Module::VMR_ReferenceImage:
+ aPrs = myRefPreviewPrs;
+ myRefPreviewPrs = 0;
+ break;
+ case HYDROGUI_Module::VMR_TransformImage:
+ aPrs = myPreviewPrs;
+ myPreviewPrs = 0;
+ }
+
+ // Remove the appropriate presentation from the view
+ if( aPrs && aViewPort )
+ {
+ aViewPort->removeItem( aPrs );
+ delete aPrs;
+ }
+
+ // Delete the view
+ module()->getApp()->removeViewManager( aViewMgr ); // aViewMgr is deleted here
+ aViewMgr = 0;
+ }