#include "HYDROGUI_ImportImageOp.h"
+#include "HYDROGUI_DataModel.h"
#include "HYDROGUI_ImportImageDlg.h"
#include "HYDROGUI_Module.h"
#include "HYDROGUI_PrsImage.h"
#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_UpdateFlags.h"
-#include <HYDROData_Document.h>
-#include <HYDROData_Image.h>
+#include <HYDROData_Iterator.h>
+
+#include <HYDROOperations_Factory.h>
#include <GraphicsView_ViewManager.h>
#include <GraphicsView_ViewPort.h>
#include <LightApp_Application.h>
#include <LightApp_UpdateFlags.h>
-#include <SUIT_Desktop.h>
-#include <SUIT_MessageBox.h>
-
-HYDROGUI_ImportImageOp::HYDROGUI_ImportImageOp( HYDROGUI_Module* theModule )
+HYDROGUI_ImportImageOp::HYDROGUI_ImportImageOp( HYDROGUI_Module* theModule,
+ const bool theIsEdit )
: HYDROGUI_Operation( theModule ),
+ myIsEdit( theIsEdit ),
+ myEditedObject( 0 ),
myActiveViewManager( 0 ),
myPreviewViewManager( 0 ),
myPreviewPrs( 0 ),
+ myRefPreviewPrs( 0 ),
myPointType( HYDROGUI_PrsImage::None )
{
+ setName( theIsEdit ? tr( "EDIT_IMPORTED_IMAGE" ) : tr( "IMPORT_IMAGE" ) );
}
HYDROGUI_ImportImageOp::~HYDROGUI_ImportImageOp()
HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
aPanel->reset();
+ aPanel->setIsEdit( myIsEdit );
+
+ if( myIsEdit )
+ {
+ myEditedObject = Handle(HYDROData_Image)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
+ if( !myEditedObject.IsNull() )
+ {
+ QImage anImage = myEditedObject->Image();
+
+ QPoint aPointA1, aPointB1, aPointC1;
+ QPointF aPointA2, aPointB2, aPointC2;
+ myEditedObject->TrsfPoints( aPointA1, aPointB1, aPointC1,
+ aPointA2, aPointB2, aPointC2 );
+
+ onCreatePreview( anImage );
+
+ if( myPreviewPrs )
+ {
+ HYDROGUI_PrsImage::TransformationPointMap aPointMap =
+ myPreviewPrs->getTransformationPointMap();
+ if( !aPointMap.isEmpty() )
+ {
+ aPointMap[ HYDROGUI_PrsImage::PointA ].Point = aPointA1;
+ aPointMap[ HYDROGUI_PrsImage::PointB ].Point = aPointB1;
+ aPointMap[ HYDROGUI_PrsImage::PointC ].Point = aPointC1;
+ myPreviewPrs->setTransformationPointMap( aPointMap );
+ }
+ }
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap;
+ aDataMap[ HYDROGUI_PrsImage::PointA ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointA1, aPointA2 );
+ aDataMap[ HYDROGUI_PrsImage::PointB ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointB1, aPointB2 );
+ aDataMap[ HYDROGUI_PrsImage::PointC ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointC1, aPointC2 );
+ ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap );
+ }
+ }
+
+ // collect information about existing images
+ HYDROGUI_ImportImageDlg::PrsPointDataList aPrsPointDataList;
+ HYDROData_Iterator anIterator( doc(), KIND_IMAGE );
+ for( ; anIterator.More(); anIterator.Next() )
+ {
+ Handle(HYDROData_Image) anImageObj = Handle(HYDROData_Image)::DownCast( anIterator.Current() );
+ if( !anImageObj.IsNull() )
+ {
+ if( myIsEdit && HYDROGUI_Tool::IsEqual( anImageObj, myEditedObject ) )
+ continue;
+
+ QPoint aPointA1, aPointB1, aPointC1;
+ QPointF aPointA2, aPointB2, aPointC2;
+ anImageObj->TrsfPoints( aPointA1, aPointB1, aPointC1,
+ aPointA2, aPointB2, aPointC2 );
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap;
+ aDataMap[ HYDROGUI_PrsImage::PointA ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointA1, aPointA2 );
+ aDataMap[ HYDROGUI_PrsImage::PointB ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointB1, aPointB2 );
+ aDataMap[ HYDROGUI_PrsImage::PointC ] =
+ HYDROGUI_ImportImageDlg::TransformationData( aPointC1, aPointC2 );
+
+ HYDROGUI_ImportImageDlg::PrsPointData aPrsPointData( anImageObj->GetName(), aDataMap );
+ aPrsPointDataList.append( aPrsPointData );
+ }
+ }
+ ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setPrsPointDataList( aPrsPointDataList );
+}
+
+void HYDROGUI_ImportImageOp::abortOperation()
+{
+ closePreview();
+
+ HYDROGUI_Operation::abortOperation();
+}
+
+void HYDROGUI_ImportImageOp::commitOperation()
+{
+ closePreview();
+
+ HYDROGUI_Operation::commitOperation();
}
HYDROGUI_InputPanel* HYDROGUI_ImportImageOp::createInputPanel() const
{
- HYDROGUI_InputPanel* aPanel = new HYDROGUI_ImportImageDlg( module(), tr( "IMPORT_IMAGE" ) );
- connect( aPanel, SIGNAL( createPreview( QString ) ),
- this, SLOT( onCreatePreview( QString ) ) );
+ 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( refImageActivated( const QString& ) ),
+ this, SLOT( onRefImageActivated( const QString& ) ) );
return aPanel;
}
-void HYDROGUI_ImportImageOp::OnApply()
+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_Object) anObject = HYDROGUI_Tool::FindObjectByName( module(), anImageName );
+ if( !anObject.IsNull() )
+ {
+ theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anImageName );
+ return false;
+ }
+ }
+
+ if( !myPreviewPrs )
+ return false;
+
+ QImage anImage = myPreviewPrs->getImage();
+
+ bool anIsRefImage = aPanel->getTransformationMode() == HYDROGUI_ImportImageDlg::RefImage;
+
+ QTransform aRefTransform;
+ if( anIsRefImage && myRefPreviewPrs )
+ {
+ Handle(HYDROData_Image) aRefImageObj =
+ Handle(HYDROData_Image)::DownCast( myRefPreviewPrs->getObject() );
+ if( !aRefImageObj.IsNull() )
+ aRefTransform = aRefImageObj->Trsf();
+ }
+
HYDROGUI_ImportImageDlg::TransformationDataMap aMap;
- bool anIsOk = aPanel->getTransformationDataMap( aMap );
- if( !anIsOk || !myPreviewPrs )
+ if( !aPanel->getTransformationDataMap( aMap ) )
+ return false;
+
+ HYDROGUI_ImportImageDlg::TransformationDataMap aRefMap;
+ if( anIsRefImage && !aPanel->getTransformationDataMap( aRefMap, true ) )
+ return false;
+
+ QPoint aPointA1 = aMap[ HYDROGUI_PrsImage::PointA ].first;
+ QPoint aPointB1 = aMap[ HYDROGUI_PrsImage::PointB ].first;
+ QPoint aPointC1 = aMap[ HYDROGUI_PrsImage::PointC ].first;
+
+ QPointF aPointA2 = aMap[ HYDROGUI_PrsImage::PointA ].second;
+ QPointF aPointB2 = aMap[ HYDROGUI_PrsImage::PointB ].second;
+ QPointF aPointC2 = aMap[ HYDROGUI_PrsImage::PointC ].second;
+
+ QPoint aPointA3, aPointB3, aPointC3;
+ if( anIsRefImage )
{
- SUIT_MessageBox::critical( module()->getApp()->desktop(),
- tr( "INSUFFICIENT_INPUT_DATA" ),
- tr( "INPUT_VALID_DATA" ) );
- return;
+ aPointA3 = aRefMap[ HYDROGUI_PrsImage::PointA ].first;
+ aPointB3 = aRefMap[ HYDROGUI_PrsImage::PointB ].first;
+ aPointC3 = aRefMap[ HYDROGUI_PrsImage::PointC ].first;
}
- double xa1 = aMap[ HYDROGUI_PrsImage::PointA ].first.x();
- double ya1 = aMap[ HYDROGUI_PrsImage::PointA ].first.y();
- double xb1 = aMap[ HYDROGUI_PrsImage::PointB ].first.x();
- double yb1 = aMap[ HYDROGUI_PrsImage::PointB ].first.y();
- double xc1 = aMap[ HYDROGUI_PrsImage::PointC ].first.x();
- double yc1 = aMap[ HYDROGUI_PrsImage::PointC ].first.y();
+ int xa1 = aPointA1.x();
+ int ya1 = aPointA1.y();
+ int xb1 = aPointB1.x();
+ int yb1 = aPointB1.y();
+ int xc1 = aPointC1.x();
+ int yc1 = aPointC1.y();
+
+ double xa2 = aPointA2.x();
+ double ya2 = aPointA2.y();
+ double xb2 = aPointB2.x();
+ double yb2 = aPointB2.y();
+ double xc2 = aPointC2.x();
+ double yc2 = aPointC2.y();
+
+ int xa3 = aPointA3.x();
+ int ya3 = aPointA3.y();
+ int xb3 = aPointB3.x();
+ int yb3 = aPointB3.y();
+ int xc3 = aPointC3.x();
+ int yc3 = aPointC3.y();
+
+ // first, check that three input points don't belong to a single line
+ if( ( yb1 - ya1 ) * ( xc1 - xa1 ) == ( yc1 - ya1 ) * ( xb1 - xa1 ) )
+ {
+ theErrorMsg = tr( "POINTS_A_B_C_BELONG_TO_SINGLE_LINE" );
+ return false;
+ }
- double xa2 = aMap[ HYDROGUI_PrsImage::PointA ].second.x();
- double ya2 = aMap[ HYDROGUI_PrsImage::PointA ].second.y();
- double xb2 = aMap[ HYDROGUI_PrsImage::PointB ].second.x();
- double yb2 = aMap[ HYDROGUI_PrsImage::PointB ].second.y();
- double xc2 = aMap[ HYDROGUI_PrsImage::PointC ].second.x();
- double yc2 = aMap[ HYDROGUI_PrsImage::PointC ].second.y();
+ // the same check for the reference points
+ if( anIsRefImage && ( ( yb3 - ya3 ) * ( xc3 - xa3 ) == ( yc3 - ya3 ) * ( xb3 - xa3 ) ) )
+ {
+ theErrorMsg = tr( "REFERENCE_POINTS_A_B_C_BELONG_TO_SINGLE_LINE" );
+ return false;
+ }
QTransform aTransform1( xa1, ya1, 1, xb1, yb1, 1, xc1, yc1, 1 );
QTransform aTransform2( xa2, ya2, 1, xb2, yb2, 1, xc2, yc2, 1 );
+ QTransform aTransform3( xa3, ya3, 1, xb3, yb3, 1, xc3, yc3, 1 );
- QTransform aTransform = aTransform1.inverted() * aTransform2;
+ bool anIsInvertible = false;
+ QTransform aTransform1Inverted = aTransform1.inverted( &anIsInvertible );
+ if( !anIsInvertible )
+ {
+ theErrorMsg = tr( "TRANSFORMATION_MATRIX_CANNOT_BE_COMPUTED" );
+ return false;
+ }
- QImage anImage = myPreviewPrs->getImage();
+ QTransform aTransform;
+ if( anIsRefImage )
+ aTransform = aTransform1Inverted * aTransform3 * aRefTransform;
+ else
+ aTransform = aTransform1Inverted * aTransform2;
- closePreview();
+ Handle(HYDROData_Image) anImageObj;
+ if( myIsEdit )
+ anImageObj = myEditedObject;
+ else
+ anImageObj = Handle(HYDROData_Image)::DownCast( doc()->CreateObject( KIND_IMAGE ) );
- int aStudyId = module()->getStudyId();
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( aStudyId );
- if( !aDocument.IsNull() )
- {
- Handle(HYDROData_Image) anImageObj =
- Handle(HYDROData_Image)::DownCast( aDocument->CreateObject( KIND_IMAGE ) );
- if( !anImageObj.IsNull() )
- {
- static int ImageId = 0;
- anImageObj->SetName( QString( "Image_%1" ).arg( QString::number( ++ImageId ) ) );
+ if( anImageObj.IsNull() )
+ return false;
- anImageObj->SetImage( anImage );
- anImageObj->SetTrsf( aTransform );
+ anImageObj->SetName( anImageName );
- module()->update( UF_Model | UF_Viewer );
- }
+ anImageObj->SetImage( anImage );
+ anImageObj->SetTrsf( aTransform );
+
+ if( anIsRefImage )
+ {
+ aPointA2 = QPointF( aTransform.map( aPointA1 ) );
+ aPointB2 = QPointF( aTransform.map( aPointB1 ) );
+ aPointC2 = QPointF( aTransform.map( aPointC1 ) );
}
- commit();
-}
-void HYDROGUI_ImportImageOp::OnCancel()
-{
+ anImageObj->SetTrsfPoints( aPointA1, aPointB1, aPointC1,
+ aPointA2, aPointB2, aPointC2 );
+
+ 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();
- abort();
+
+ if( !myIsEdit )
+ module()->setObjectVisible( HYDROGUI_Tool::GetActiveGraphicsViewId( module() ), anImageObj, true );
+
+ if( myIsEdit )
+ if( HYDROOperations_Factory* aFactory = HYDROOperations_Factory::Factory() )
+ aFactory->UpdateImage( doc(), anImageObj );
+
+ theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced;
+ return true;
}
-void HYDROGUI_ImportImageOp::onCreatePreview( QString theFileName )
+void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage )
{
LightApp_Application* anApp = module()->getApp();
myActiveViewManager = anApp->activeViewManager();
- myPreviewPrs = new HYDROGUI_PrsImage( 0 ); // no data object
-
- QImage anImage( theFileName );
- myPreviewPrs->setImage( anImage );
-
+ myPreviewPrs = new HYDROGUI_PrsImage( myIsEdit ? myEditedObject : 0 );
+ myPreviewPrs->setImage( theImage );
myPreviewPrs->compute();
myPreviewViewManager =
dynamic_cast<GraphicsView_ViewManager*>( anApp->createViewManager( GraphicsView_Viewer::Type() ) );
if( myPreviewViewManager )
{
- myPreviewViewManager->setTitle( tr( "MAPPING" ) );
+ 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 );
+ //aViewPort->setMousePositionEnabled( true ); //ouv: temporarily commented
aViewPort->addItem( myPreviewPrs );
aViewPort->fitAll();
HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
- aPanel->initializePointSelection();
- onPointSelected();
+ QString anImageName;
+ if( myIsEdit )
+ {
+ if( !myEditedObject.IsNull() )
+ anImageName = myEditedObject->GetName();
+ }
+ else
+ anImageName = HYDROGUI_Tool::GenerateObjectName( module(), "Image" );
+ aPanel->setImageName( anImageName );
+
+ aPanel->setImageSize( theImage.size() );
- aPanel->synchronizeTransformedPoints();
+ aPanel->initializePointSelection();
+ onPointSelected( false );
}
void HYDROGUI_ImportImageOp::onActivatePointSelection( int thePointType )
{
myPointType = thePointType;
if( myPreviewPrs )
- myPreviewPrs->setTransformationPointMode( thePointType );
+ 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::onRefImageActivated( const QString& theName )
+{
+ GraphicsView_ViewPort* aViewPort = 0;
+ if( myPreviewViewManager )
+ if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() )
+ aViewPort = aViewer->getActiveViewPort();
+
+ if( !aViewPort )
+ return;
+
+ if( myPreviewPrs )
+ myPreviewPrs->setCaption( QString() );
+
+ if( myRefPreviewPrs )
+ {
+ myRefPreviewPrs->setCaption( QString() );
+ aViewPort->removeItem( myRefPreviewPrs );
+
+ delete myRefPreviewPrs;
+ myRefPreviewPrs = 0;
+ }
+
+ 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 );
+
+ // vertically shift the reference prs relatively to the main prs
+ if( myPreviewPrs )
+ {
+ myPreviewPrs->setCaption( tr( "IMPORTED_IMAGE" ) );
+
+ QImage anImage = myPreviewPrs->getImage();
+ myRefPreviewPrs->moveBy( 0, anImage.height() + 60 );
+ myRefPreviewPrs->setCaption( tr( "REFERENCE_IMAGE" ) );
+ }
+
+ aViewPort->addItem( myRefPreviewPrs );
+ }
+
+ aViewPort->fitAll();
+
+ HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+
+ aPanel->setImageSize( anImage.size(), true );
+
+ aPanel->initializePointSelection();
+ onPointSelected( true );
}
void HYDROGUI_ImportImageOp::onPointSelected()
{
+ onPointSelected( myRefPreviewPrs && myRefPreviewPrs->isSelected() );
+}
+
+void HYDROGUI_ImportImageOp::onPointSelected( bool theIsRefImage )
+{
+ HYDROGUI_PrsImage* aPrs = theIsRefImage ? myRefPreviewPrs : myPreviewPrs;
+ if( !aPrs )
+ return;
+
HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap;
const HYDROGUI_PrsImage::TransformationPointMap& aPointMap =
- myPreviewPrs->getTransformationPointMap();
+ aPrs->getTransformationPointMap();
HYDROGUI_PrsImage::TransformationPointMapIterator anIter( aPointMap );
while( anIter.hasNext() )
{
aDataMap[ aPointType ] = aData;
}
- ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap );
+ ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap, true, theIsRefImage );
}
void HYDROGUI_ImportImageOp::closePreview()
myPreviewPrs = 0;
}
+ if( myRefPreviewPrs )
+ {
+ delete myRefPreviewPrs;
+ myRefPreviewPrs = 0;
+ }
+
if( myPreviewViewManager )
{
module()->getApp()->removeViewManager( myPreviewViewManager ); // myPreviewViewManager is deleted here
