Salome HOME
Feature 32: Image presentation in OCCViewer.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ImportImageOp.cxx
index bcc97ca11b17097b51fec7808b05292d95cf46ee..46cbd7ebaf560f42cf84b05f5b7c7dc1127924e3 100644 (file)
 
 #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 <HYDROData_Lambert93.h>
+
+#include <HYDROOperations_Factory.h>
 
 #include <GraphicsView_ViewManager.h>
 #include <GraphicsView_ViewPort.h>
 #include <LightApp_Application.h>
 #include <LightApp_UpdateFlags.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( tr( "IMPORT_IMAGE" ) );
+  setName( theIsEdit ? tr( "EDIT_IMPORTED_IMAGE" ) : tr( "IMPORT_IMAGE" ) );
 }
 
 HYDROGUI_ImportImageOp::~HYDROGUI_ImportImageOp()
@@ -57,130 +65,475 @@ void HYDROGUI_ImportImageOp::startOperation()
 
   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();
+
+      int aTransformationMode = myEditedObject->TrsfMode();
+      if( aTransformationMode == HYDROGUI_ImportImageDlg::RefImage )
+        aTransformationMode = HYDROGUI_ImportImageDlg::ManualCartesian;
+      aPanel->setTransformationMode( aTransformationMode );
+
+      QPoint aPointA1, aPointB1, aPointC1;
+      QPointF aPointA2, aPointB2, aPointC2;
+      QPointF aPointA3, aPointB3, aPointC3;
+      myEditedObject->TrsfPoints( aPointA1, aPointB1, aPointC1,
+                                  aPointA2, aPointB2, aPointC2,
+                                  aPointA3, aPointB3, aPointC3 );
+
+      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, aPointA3 );
+      aDataMap[ HYDROGUI_PrsImage::PointB ] =
+        HYDROGUI_ImportImageDlg::TransformationData( aPointB1, aPointB2, aPointB3 );
+      aDataMap[ HYDROGUI_PrsImage::PointC ] =
+        HYDROGUI_ImportImageDlg::TransformationData( aPointC1, aPointC2, aPointC3 );
+      aPanel->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;
+      QPointF aPointA3, aPointB3, aPointC3;
+      anImageObj->TrsfPoints( aPointA1, aPointB1, aPointC1,
+                              aPointA2, aPointB2, aPointC2,
+                              aPointA3, aPointB3, aPointC3 );
+
+      HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap;
+      aDataMap[ HYDROGUI_PrsImage::PointA ] =
+        HYDROGUI_ImportImageDlg::TransformationData( aPointA1, aPointA2, aPointA3 );
+      aDataMap[ HYDROGUI_PrsImage::PointB ] =
+        HYDROGUI_ImportImageDlg::TransformationData( aPointB1, aPointB2, aPointB3 );
+      aDataMap[ HYDROGUI_PrsImage::PointC ] =
+        HYDROGUI_ImportImageDlg::TransformationData( aPointC1, aPointC2, aPointC3 );
+
+      HYDROGUI_ImportImageDlg::PrsPointData aPrsPointData( anImageObj->GetName(), aDataMap );
+      aPrsPointDataList.append( aPrsPointData );
+    }
+  }
+  aPanel->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(), getName() );
-  connect( aPanel, SIGNAL( createPreview( QString ) ),
-           this, SLOT( onCreatePreview( QString ) ) );
+  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::processApply( int& theUpdateFlags )
+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;
+    }
+  }
+
+  int aTransformationMode = aPanel->getTransformationMode();
+  bool anIsRefImage = aTransformationMode == HYDROGUI_ImportImageDlg::RefImage;
+
   HYDROGUI_ImportImageDlg::TransformationDataMap aMap;
-  bool anIsOk = aPanel->getTransformationDataMap( aMap );
-  if( !anIsOk || !myPreviewPrs )
+  if( !aPanel->getTransformationDataMap( aMap ) )
     return false;
 
-  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();
+  HYDROGUI_ImportImageDlg::TransformationDataMap aRefMap;
+  if( anIsRefImage && !aPanel->getTransformationDataMap( aRefMap, true ) )
+    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();
+  QPoint aPointA = aMap[ HYDROGUI_PrsImage::PointA ].ImagePoint;
+  QPoint aPointB = aMap[ HYDROGUI_PrsImage::PointB ].ImagePoint;
+  QPoint aPointC = aMap[ HYDROGUI_PrsImage::PointC ].ImagePoint;
 
-  QTransform aTransform1( xa1, ya1, 1, xb1, yb1, 1, xc1, yc1, 1 );
-  QTransform aTransform2( xa2, ya2, 1, xb2, yb2, 1, xc2, yc2, 1 );
+  QPointF aLPointA = aMap[ HYDROGUI_PrsImage::PointA ].LambertPoint;
+  QPointF aLPointB = aMap[ HYDROGUI_PrsImage::PointB ].LambertPoint;
+  QPointF aLPointC = aMap[ HYDROGUI_PrsImage::PointC ].LambertPoint;
 
-  QTransform aTransform = aTransform1.inverted() * aTransform2;
+  QPointF aCPointA = aMap[ HYDROGUI_PrsImage::PointA ].CartesianPoint;
+  QPointF aCPointB = aMap[ HYDROGUI_PrsImage::PointB ].CartesianPoint;
+  QPointF aCPointC = aMap[ HYDROGUI_PrsImage::PointC ].CartesianPoint;
 
-  QImage anImage = myPreviewPrs->getImage();
+  QPoint aRefPointA, aRefPointB, aRefPointC;
+  if( anIsRefImage )
+  {
+    aRefPointA = aRefMap[ HYDROGUI_PrsImage::PointA ].ImagePoint;
+    aRefPointB = aRefMap[ HYDROGUI_PrsImage::PointB ].ImagePoint;
+    aRefPointC = aRefMap[ HYDROGUI_PrsImage::PointC ].ImagePoint;
+  }
 
-  closePreview();
+  int xa = aPointA.x();
+  int ya = aPointA.y();
+  int xb = aPointB.x();
+  int yb = aPointB.y();
+  int xc = aPointC.x();
+  int yc = aPointC.y();
+
+  double xca = aCPointA.x();
+  double yca = aCPointA.y();
+  double xcb = aCPointB.x();
+  double ycb = aCPointB.y();
+  double xcc = aCPointC.x();
+  double ycc = aCPointC.y();
+
+  int xra = aRefPointA.x();
+  int yra = aRefPointA.y();
+  int xrb = aRefPointB.x();
+  int yrb = aRefPointB.y();
+  int xrc = aRefPointC.x();
+  int yrc = aRefPointC.y();
+
+  // first, check that three input points don't belong to a single line
+  if( ( yb - ya ) * ( xc - xa ) == ( yc - ya ) * ( xb - xa ) )
+  {
+    theErrorMsg = tr( "POINTS_A_B_C_BELONG_TO_SINGLE_LINE" );
+    return false;
+  }
 
-  int aStudyId = module()->getStudyId();
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( aStudyId );
-  if( aDocument.IsNull() )
+  // the same check for the reference points
+  if( anIsRefImage &&
+      ( ( yrb - yra ) * ( xrc - xra ) == ( yrc - yra ) * ( xrb - xra ) ) )
+  {
+    theErrorMsg = tr( "REFERENCE_POINTS_A_B_C_BELONG_TO_SINGLE_LINE" );
     return false;
+  }
+
+  QTransform aTransform1( xa, ya, 1, xb, yb, 1, xc, yc, 1 );
+  QTransform aTransform2( xca, yca, 1, xcb, ycb, 1, xcc, ycc, 1 );
+  QTransform aTransform3( xra, yra, 1, xrb, yrb, 1, xrc, yrc, 1 );
+
+  bool anIsInvertible = false;
+  QTransform aTransform1Inverted = aTransform1.inverted( &anIsInvertible );
+  if( !anIsInvertible )
+  {
+    theErrorMsg = tr( "TRANSFORMATION_MATRIX_CANNOT_BE_COMPUTED" );
+    return false;
+  }
+
+  QTransform aTransform;
+  if( anIsRefImage )
+    aTransform = aTransform1Inverted * aTransform3 * myRefTransform;
+  else
+    aTransform = aTransform1Inverted * aTransform2;
+
+  Handle(HYDROData_Image) anImageObj;
+  if( myIsEdit )
+    anImageObj = myEditedObject;
+  else
+    anImageObj = Handle(HYDROData_Image)::DownCast( doc()->CreateObject( KIND_IMAGE ) );
 
-  Handle(HYDROData_Image) anImageObj =
-    Handle(HYDROData_Image)::DownCast( aDocument->CreateObject( KIND_IMAGE ) );
   if( anImageObj.IsNull() )
     return false;
 
-  static int ImageId = 0;
-  anImageObj->SetName( QString( "Image_%1" ).arg( QString::number( ++ImageId ) ) );
+  anImageObj->SetName( anImageName );
 
-  anImageObj->SetImage( anImage );
+  anImageObj->SetImage( myImage );
   anImageObj->SetTrsf( aTransform );
 
-  theUpdateFlags = UF_Model | UF_Viewer;
-  return true;
-}
+  if( anIsRefImage )
+  {
+    aCPointA = QPointF( aTransform.map( aPointA ) );
+    aCPointB = QPointF( aTransform.map( aPointB ) );
+    aCPointC = QPointF( aTransform.map( aPointC ) );
+
+    // compute Lambert93 points
+    xca = aCPointA.x();
+    yca = aCPointA.y();
+    xcb = aCPointB.x();
+    ycb = aCPointB.y();
+    xcc = aCPointC.x();
+    ycc = aCPointC.y();
+
+    double xla = 0, yla = 0, xlb = 0, ylb = 0, xlc = 0, ylc = 0;
+    HYDROData_Lambert93::toGeo( xca, yca, yla, xla );
+    HYDROData_Lambert93::toGeo( xcb, ycb, ylb, xlb );
+    HYDROData_Lambert93::toGeo( xcc, ycc, ylc, xlc );
+
+    aLPointA = QPointF( xla * 3600.0, yla * 3600.0 ); // convert degrees to seconds
+    aLPointB = QPointF( xlb * 3600.0, ylb * 3600.0 ); // convert degrees to seconds
+    aLPointC = QPointF( xlc * 3600.0, ylc * 3600.0 ); // convert degrees to seconds
+  }
 
-void HYDROGUI_ImportImageOp::processCancel()
-{
+  anImageObj->SetTrsfMode( aTransformationMode );
+  anImageObj->SetTrsfPoints( aPointA, aPointB, aPointC,
+                             aLPointA, aLPointB, aLPointC,
+                             aCPointA, aCPointB, aCPointC );
+
+  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( QString theFileName )
+void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage )
 {
-  LightApp_Application* anApp = module()->getApp();
+  HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
 
-  myActiveViewManager = anApp->activeViewManager();
+  myImage = theImage;
 
-  myPreviewPrs = new HYDROGUI_PrsImage( 0 ); // no data object
+  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();
 
-  QImage anImage( theFileName );
-  myPreviewPrs->setImage( anImage );
+    myActiveViewManager = anApp->activeViewManager();
 
-  myPreviewPrs->compute();
+    myPreviewPrs = new HYDROGUI_PrsImage( myIsEdit ? myEditedObject : 0 );
+    myPreviewPrs->setImage( myImage );
+    myPreviewPrs->compute();
 
-  myPreviewViewManager =
-    dynamic_cast<GraphicsView_ViewManager*>( anApp->createViewManager( GraphicsView_Viewer::Type() ) );
-  if( myPreviewViewManager )
-  {
-    myPreviewViewManager->setTitle( tr( "MAPPING" ) );
-    if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() )
+    myPreviewViewManager =
+      dynamic_cast<GraphicsView_ViewManager*>( anApp->createViewManager( GraphicsView_Viewer::Type() ) );
+    if( myPreviewViewManager )
     {
-      if( GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort() )
+      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() )
       {
-        aViewPort->setMousePositionEnabled( true );
+        if( GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort() )
+        {
+          //aViewPort->setMousePositionEnabled( true ); //ouv: temporarily commented
 
-        aViewPort->addItem( myPreviewPrs );
-        aViewPort->fitAll();
+          aViewPort->addItem( myPreviewPrs );
+          aViewPort->fitAll();
 
-        myPreviewPrs->setIsTransformationPointPreview( true );
+          myPreviewPrs->setIsTransformationPointPreview( true );
+        }
+        connect( aViewer, SIGNAL( selectionChanged( GV_SelectionChangeStatus ) ),
+                 this, SLOT( onPointSelected() ) );
       }
-      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 );
   }
 
-  HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+  aPanel->setImageSize( myImage.size() );
 
   aPanel->initializePointSelection();
-  onPointSelected();
-
-  aPanel->synchronizeTransformedPoints();
+  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::onModeActivated( int theMode )
+{
+  HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel();
+
+  QString aRefImageName;
+  if( theMode == HYDROGUI_ImportImageDlg::RefImage )
+  {
+    aRefImageName = aPanel->getRefImageName();
+    if( aRefImageName.isEmpty() )
+      return; // do nothing in this case to avoid visual moving of preview prs
+  }
+  onRefImageActivated( aRefImageName );
+}
+
+void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName )
+{
+  myRefTransform.reset();
+
+  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();
+    myRefTransform = anImageObj->Trsf();
+
+    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::onLastViewClosed( SUIT_ViewManager* theViewManager )
+{
+  closePreview();
 }
 
 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() )
   {
@@ -188,23 +541,40 @@ void HYDROGUI_ImportImageOp::onPointSelected()
     const HYDROGUI_PrsImage::TransformationPoint& aTransformationPoint = anIter.value();
     const QPoint& aPoint = aTransformationPoint.Point;
 
-    HYDROGUI_ImportImageDlg::TransformationData aData( aPoint, QPointF() );
+    HYDROGUI_ImportImageDlg::TransformationData aData( aPoint, QPointF(), QPointF() );
     aDataMap[ aPointType ] = aData;
   }
 
-  ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap );
+  ( (HYDROGUI_ImportImageDlg*)inputPanel() )->setTransformationDataMap( aDataMap, true, theIsRefImage );
 }
 
 void HYDROGUI_ImportImageOp::closePreview()
 {
+  // It's very strange, but without calling this method (it's quite safe) a crash is stably reproduced.
+  // Scenario: create any non-Graphics view, activate import op, click apply.
+  // Result: a few SIGSEGVs coming from processEvents(), then crash.
+  if( myPreviewViewManager )
+    if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() )
+      if( GraphicsView_ViewPort* aViewPort = aViewer->getActiveViewPort() )
+        aViewPort->onBoundingRectChanged();
+
   if( myPreviewPrs )
   {
     delete myPreviewPrs;
     myPreviewPrs = 0;
   }
 
+  if( myRefPreviewPrs )
+  {
+    delete myRefPreviewPrs;
+    myRefPreviewPrs = 0;
+  }
+
   if( myPreviewViewManager )
   {
+    disconnect( myPreviewViewManager, SIGNAL( lastViewClosed( SUIT_ViewManager* ) ),
+                this, SLOT( onLastViewClosed( SUIT_ViewManager* ) ) );
+
     module()->getApp()->removeViewManager( myPreviewViewManager ); // myPreviewViewManager is deleted here
     myPreviewViewManager = 0;
   }