X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROGUI%2FHYDROGUI_ImportImageOp.cxx;h=bf837ade79e883bcac198a7b6be658d5d2aa1fab;hb=a53349567d67f4df0ef737798a25c24d9dc8f08e;hp=ffd893a79f48684993c27fa98144fc7deaf3b419;hpb=c874fabf131c86df3f867c094ed3a2a0813a784e;p=modules%2Fhydro.git diff --git a/src/HYDROGUI/HYDROGUI_ImportImageOp.cxx b/src/HYDROGUI/HYDROGUI_ImportImageOp.cxx index ffd893a7..bf837ade 100644 --- a/src/HYDROGUI/HYDROGUI_ImportImageOp.cxx +++ b/src/HYDROGUI/HYDROGUI_ImportImageOp.cxx @@ -1,12 +1,8 @@ -// 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 -// +// Copyright (C) 2014-2015 EDF-R&D // 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. +// version 2.1 of the License, or (at your option) any later version. // // This library is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -23,6 +19,7 @@ #include "HYDROGUI_ImportImageOp.h" #include "HYDROGUI_DataModel.h" +#include #include "HYDROGUI_ImportImageDlg.h" #include "HYDROGUI_Module.h" #include "HYDROGUI_PrsImage.h" @@ -41,8 +38,10 @@ #include #include +#include #include #include +#include HYDROGUI_ImportImageOp::HYDROGUI_ImportImageOp( HYDROGUI_Module* theModule, const bool theIsEdit ) @@ -75,7 +74,8 @@ void HYDROGUI_ImportImageOp::startOperation() if( myIsEdit ) { - myEditedObject = Handle(HYDROData_Image)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) ); + if ( isApplyAndClose() ) + myEditedObject = Handle(HYDROData_Image)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) ); if( !myEditedObject.IsNull() ) { QImage anImage = myEditedObject->Image(); @@ -84,79 +84,122 @@ void HYDROGUI_ImportImageOp::startOperation() QPoint aLocalPointA, aLocalPointB, aLocalPointC; myEditedObject->GetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC ); + // Create the edited image preview presentation in the viewer onCreatePreview( anImage ); - if( myPreviewPrs ) + // Set transformation local points A,B,C to the image preview presentation + setPresentationTrsfPoints( myPreviewPrs, anIsByTwoPoints, aLocalPointA, + aLocalPointB, aLocalPointC ); + + // Build the existing image local and global points mapping + // according to the current transformation mode. + HYDROData_Image::TransformationMode aTrsfMode = myEditedObject->GetTrsfMode(); + QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC; + HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap; + if ( aTrsfMode == HYDROData_Image::ReferenceImage ) { - HYDROGUI_PrsImage::TransformationPointMap aPointMap = - myPreviewPrs->getTransformationPointMap(); - if( !aPointMap.isEmpty() ) + // Compute global points using the transformation matrix of the reference image + Handle(HYDROData_Image) aRefImage = myEditedObject->GetTrsfReferenceImage(); + if ( !aRefImage.IsNull() ) { - aPointMap[ HYDROGUI_PrsImage::PointA ].Point = aLocalPointA; - aPointMap[ HYDROGUI_PrsImage::PointB ].Point = aLocalPointB; + QTransform aRefTrsf = aRefImage->Trsf(); // The reference image transformation matrix + aTrsfPointA = aRefTrsf.map( aLocalPointA ); // Compute the global point A + aTrsfPointB = aRefTrsf.map( aLocalPointB ); // Compute the global point B if ( !anIsByTwoPoints ) - aPointMap[ HYDROGUI_PrsImage::PointC ].Point = aLocalPointC; - - myPreviewPrs->setTransformationPointMap( aPointMap ); + { + aTrsfPointC = aRefTrsf.map( aLocalPointC ); // Compute the global point C if used + } + // Build the local-global points map + // Use the reference image transformation mode for interpreting global points + computeTrsfData( aRefImage->GetTrsfMode(), anIsByTwoPoints, aLocalPointA, aLocalPointB, aLocalPointC, + aTrsfPointA, aTrsfPointB, aTrsfPointC, aDataMap ); } } + else + { + // Get global points from the edited image + myEditedObject->GetGlobalPoints( aTrsfMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ); + // Build the local-global points map + computeTrsfData( aTrsfMode, anIsByTwoPoints, aLocalPointA, aLocalPointB, aLocalPointC, + aTrsfPointA, aTrsfPointB, aTrsfPointC, aDataMap ); + } - HYDROData_Image::TransformationMode aTrsfMode; - QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC; - myEditedObject->GetGlobalPoints( aTrsfMode, - aTrsfPointA, aTrsfPointB, aTrsfPointC ); - - HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap; - aDataMap[ HYDROGUI_PrsImage::PointA ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aTrsfMode, aLocalPointA, aTrsfPointA ); - aDataMap[ HYDROGUI_PrsImage::PointB ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aTrsfMode, aLocalPointB, aTrsfPointB ); - if ( !anIsByTwoPoints ) - aDataMap[ HYDROGUI_PrsImage::PointC ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aTrsfMode, aLocalPointC, aTrsfPointC ); - + // Initialize the dialog mode, local and global coordinates + // except coordinates on the reference image aPanel->setTransformationMode( aTrsfMode ); aPanel->setTransformationDataMap( aDataMap ); aPanel->setByTwoPoints( anIsByTwoPoints ); - // Set points on reference image - if ( aTrsfMode = HYDROData_Image::ReferenceImage ) + // Set points of the reference image + if ( aTrsfMode == HYDROData_Image::ReferenceImage ) { - Handle(HYDROData_Image) aRefImage = myEditedObject->GetTrsfReferenceImage(); + Handle(HYDROData_Image) aRefImage; + myEditedObject->GetReferencePoints( aRefImage, + aTrsfPointA, aTrsfPointB, aTrsfPointC ); if ( !aRefImage.IsNull() ) { aRefImageName = aRefImage->GetName(); + // Create the reference image presentation in the viewer onRefImageActivated( aRefImageName ); - if( myRefPreviewPrs ) - { - HYDROGUI_PrsImage::TransformationPointMap aRefPointMap = - myRefPreviewPrs->getTransformationPointMap(); - if( !aRefPointMap.isEmpty() ) - { - aRefPointMap[ HYDROGUI_PrsImage::PointA ].Point = aTrsfPointA.toPoint(); - aRefPointMap[ HYDROGUI_PrsImage::PointB ].Point = aTrsfPointB.toPoint(); - if ( !anIsByTwoPoints ) - aRefPointMap[ HYDROGUI_PrsImage::PointC ].Point = aTrsfPointC.toPoint(); - - myRefPreviewPrs->setTransformationPointMap( aRefPointMap ); - } - } + // Set transformation points A,B,C to the reference image presentation + setPresentationTrsfPoints( myRefPreviewPrs, anIsByTwoPoints, aTrsfPointA.toPoint(), + aTrsfPointB.toPoint(), aTrsfPointC.toPoint() ); + + // Prepare A, B, C points on the reference image HYDROGUI_ImportImageDlg::TransformationDataMap aRefDataMap; - aRefDataMap[ HYDROGUI_PrsImage::PointA ] = HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointA.toPoint() ); - aRefDataMap[ HYDROGUI_PrsImage::PointB ] = HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointB.toPoint() ); + aRefDataMap[ HYDROGUI_PrsImage::PointA ] = + HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointA.toPoint() ); + aRefDataMap[ HYDROGUI_PrsImage::PointB ] = + HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointB.toPoint() ); if ( !anIsByTwoPoints ) - aRefDataMap[ HYDROGUI_PrsImage::PointC ] = HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointC.toPoint() ); + aRefDataMap[ HYDROGUI_PrsImage::PointC ] = + HYDROGUI_ImportImageDlg::TransformationData( aTrsfPointC.toPoint() ); + // Initialize the dialog's A, B, C points coordinates of the reference image aPanel->setTransformationDataMap( aRefDataMap, true, true ); } } } } - // collect information about existing images HYDROGUI_ImportImageDlg::PrsPointDataList aPrsPointDataList; + getReferenceDataList( aPrsPointDataList ); + aPanel->setPrsPointDataList( aPrsPointDataList ); + // Select the current reference image in the dialog combobox + aPanel->setRefImageName( aRefImageName ); +} + +void HYDROGUI_ImportImageOp::setPresentationTrsfPoints( HYDROGUI_PrsImage* thePrs, + bool theIsByTwoPoints, + const QPoint theLocalPointA, + const QPoint theLocalPointB, + const QPoint theLocalPointC ) +{ + // Set transformation points A,B,C to the image presentation + if( thePrs ) + { + HYDROGUI_PrsImage::TransformationPointMap aPointMap = + thePrs->getTransformationPointMap(); + if( !aPointMap.isEmpty() ) + { + aPointMap[ HYDROGUI_PrsImage::PointA ].Point = theLocalPointA; + aPointMap[ HYDROGUI_PrsImage::PointB ].Point = theLocalPointB; + if ( !theIsByTwoPoints ) + aPointMap[ HYDROGUI_PrsImage::PointC ].Point = theLocalPointC; + + thePrs->setIsByTwoPoints( theIsByTwoPoints ); + thePrs->setTransformationPointMap( aPointMap ); + } + } +} + +void HYDROGUI_ImportImageOp::getReferenceDataList( + HYDROGUI_ImportImageDlg::PrsPointDataList& theList ) const +{ + // Collect information about existing images and initialize the combobox + // reference image selector in the dialog. HYDROData_Iterator anIterator( doc(), KIND_IMAGE ); for( ; anIterator.More(); anIterator.Next() ) { @@ -175,20 +218,36 @@ void HYDROGUI_ImportImageOp::startOperation() aTrsfPointA, aTrsfPointB, aTrsfPointC ); HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap; - aDataMap[ HYDROGUI_PrsImage::PointA ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aImgTrsfMode, aLocalPointA, aTrsfPointA ); - aDataMap[ HYDROGUI_PrsImage::PointB ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aImgTrsfMode, aLocalPointB, aTrsfPointB ); - aDataMap[ HYDROGUI_PrsImage::PointC ] = - HYDROGUI_ImportImageDlg::ComputeTrsfData( aImgTrsfMode, aLocalPointC, aTrsfPointC ); + computeTrsfData( aImgTrsfMode, false, aLocalPointA, aLocalPointB, aLocalPointC, + aTrsfPointA, aTrsfPointB, aTrsfPointC, aDataMap ); HYDROGUI_ImportImageDlg::PrsPointData aPrsPointData( anImageObj->GetName(), aDataMap ); - aPrsPointDataList.append( aPrsPointData ); + theList.append( aPrsPointData ); } } +} - aPanel->setPrsPointDataList( aPrsPointDataList ); - aPanel->setRefImageName( aRefImageName ); +void HYDROGUI_ImportImageOp::computeTrsfData( HYDROData_Image::TransformationMode theTrsfMode, + bool theIsByTwoPoints, + const QPoint& theLocalPointA, + const QPoint& theLocalPointB, + const QPoint& theLocalPointC, + const QPointF& theGlobalPointA, + const QPointF& theGlobalPointB, + const QPointF& theGlobalPointC, + HYDROGUI_ImportImageDlg::TransformationDataMap& theDataMap ) const +{ + // Build the local-global points map + HYDROGUI_ImportImageDlg::TransformationDataMap aDataMap; + theDataMap[ HYDROGUI_PrsImage::PointA ] = + HYDROGUI_ImportImageDlg::ComputeTrsfData( theTrsfMode, theLocalPointA, theGlobalPointA ); + theDataMap[ HYDROGUI_PrsImage::PointB ] = + HYDROGUI_ImportImageDlg::ComputeTrsfData( theTrsfMode, theLocalPointB, theGlobalPointB ); + if ( !theIsByTwoPoints ) + { + theDataMap[ HYDROGUI_PrsImage::PointC ] = + HYDROGUI_ImportImageDlg::ComputeTrsfData( theTrsfMode, theLocalPointC, theGlobalPointC ); + } } void HYDROGUI_ImportImageOp::abortOperation() @@ -200,7 +259,8 @@ void HYDROGUI_ImportImageOp::abortOperation() void HYDROGUI_ImportImageOp::commitOperation() { - closePreview(); + if ( isApplyAndClose() ) + closePreview(); HYDROGUI_Operation::commitOperation(); } @@ -208,19 +268,31 @@ void HYDROGUI_ImportImageOp::commitOperation() HYDROGUI_InputPanel* HYDROGUI_ImportImageOp::createInputPanel() const { 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( createPreview( QImage ) ), SLOT( onCreatePreview( QImage ) ) ); + connect( aPanel, SIGNAL( activatePointSelection( int ) ), SLOT( onActivatePointSelection( int ) ) ); + connect( aPanel, SIGNAL( pointCoordChanged( bool, int, bool, int ) ), + SLOT( onPointCoordChanged( bool, int, bool, int ) ) ); + connect( aPanel, SIGNAL( modeActivated( int ) ), SLOT( onModeActivated( int ) ) ); connect( aPanel, SIGNAL( refImageActivated( const QString& ) ), - this, SLOT( onRefImageActivated( const QString& ) ) ); + SLOT( onRefImageActivated( const QString& ) ) ); + connect( aPanel, SIGNAL( setCIsUsed( bool ) ), SLOT( onSetCIsUsed( bool ) ) ); + connect( aPanel, SIGNAL( filesSelected( const QStringList& ) ), + SLOT( onFilesSelected( const QStringList& ) ) ); return aPanel; } +void HYDROGUI_ImportImageOp::onSetCIsUsed( bool theCIsUsed ) +{ + if ( myPreviewPrs ) + { + myPreviewPrs->setIsByTwoPoints( !theCIsUsed ); + } + if ( myRefPreviewPrs ) + { + myRefPreviewPrs->setIsByTwoPoints( !theCIsUsed ); + } +} + bool HYDROGUI_ImportImageOp::checkPoints( const QPointF& thePointA, const QPointF& thePointB, const QPointF& thePointC, @@ -245,35 +317,36 @@ bool HYDROGUI_ImportImageOp::checkPoints( const QPointF& thePointA, theErrorMsg = theLineErrMsg; 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; + } + } } else { // check that two points are not identical - if ( ValuesEquals( xa, ya ) && ValuesEquals( xb, yb ) ) + if ( ValuesEquals( xa, xb ) && ValuesEquals( ya, 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 ) + QString& theErrorMsg, + QStringList& theBrowseObjectsEntries ) { HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); @@ -292,74 +365,82 @@ bool HYDROGUI_ImportImageOp::processApply( int& theUpdateFlags, } } - HYDROGUI_ImportImageDlg::TransformationDataMap aMap; - if( !aPanel->getTransformationDataMap( aMap ) ) - return false; + HYDROData_Image::TransformationMode aTransformationMode = + (HYDROData_Image::TransformationMode)aPanel->getTransformationMode(); + + QPoint aPointA, aPointB, aPointC; + QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC( INT_MIN, INT_MIN ); + Handle(HYDROData_Image) aRefImageObj; - bool anIsByTwoPoints = aPanel->isByTwoPoints(); + if ( aTransformationMode != HYDROData_Image::CartesianFromFile ) { + HYDROGUI_ImportImageDlg::TransformationDataMap aMap; + if( !aPanel->getTransformationDataMap( aMap ) ) + return false; - 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; + bool anIsByTwoPoints = aPanel->isByTwoPoints(); - // 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; + aPointA = aMap[ HYDROGUI_PrsImage::PointA ].ImagePoint; + aPointB = aMap[ HYDROGUI_PrsImage::PointB ].ImagePoint; + aPointC = anIsByTwoPoints ? QPoint( INT_MIN, INT_MIN ) : + aMap[ HYDROGUI_PrsImage::PointC ].ImagePoint; - Handle(HYDROData_Image) aRefImageObj; - QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC( INT_MIN, INT_MIN ); + // 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; - HYDROData_Image::TransformationMode aTransformationMode = - (HYDROData_Image::TransformationMode)aPanel->getTransformationMode(); - if ( aTransformationMode == HYDROData_Image::ReferenceImage ) - { - QString aRefImageName = aPanel->getRefImageName(); - if( aRefImageName.isEmpty() ) + if ( aTransformationMode == HYDROData_Image::ReferenceImage ) { - theErrorMsg = tr( "REFERENCE_IMAGE_IS_NOT_SELECTED" ); - return false; - } + 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; + 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; + aRefImageObj = Handle(HYDROData_Image)::DownCast( + HYDROGUI_Tool::FindObjectByName( module(), aRefImageName, KIND_IMAGE ) ); + if( aRefImageObj.IsNull() ) { + return false; + } + else if ( !isReferenceCorrect() ) { + aRefImageObj->RemoveAllReferences(); + } - aTrsfPointA = aRefMap[ HYDROGUI_PrsImage::PointA ].ImagePoint; - aTrsfPointB = aRefMap[ HYDROGUI_PrsImage::PointB ].ImagePoint; - if ( !anIsByTwoPoints ) + 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::ManualGeodesic ) - { - aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].GeodesicPoint; - aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].GeodesicPoint; - if ( !anIsByTwoPoints ) - aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].GeodesicPoint; + // 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 { - aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].CartesianPoint; - aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].CartesianPoint; - if ( !anIsByTwoPoints ) - aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].CartesianPoint; + if ( aTransformationMode == HYDROData_Image::ManualGeodesic ) + { + aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].GeodesicPoint; + aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].GeodesicPoint; + if ( !anIsByTwoPoints ) + aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].GeodesicPoint; + } + else + { + aTrsfPointA = aMap[ HYDROGUI_PrsImage::PointA ].CartesianPoint; + aTrsfPointB = aMap[ HYDROGUI_PrsImage::PointB ].CartesianPoint; + if ( !anIsByTwoPoints ) + aTrsfPointC = aMap[ HYDROGUI_PrsImage::PointC ].CartesianPoint; + } } } @@ -367,24 +448,45 @@ bool HYDROGUI_ImportImageOp::processApply( int& theUpdateFlags, if( myIsEdit ) anImageObj = myEditedObject; else + { anImageObj = Handle(HYDROData_Image)::DownCast( doc()->CreateObject( KIND_IMAGE ) ); + QString anEntry = HYDROGUI_DataObject::dataObjectEntry( anImageObj ); + theBrowseObjectsEntries.append( anEntry ); + } if( anImageObj.IsNull() ) return false; anImageObj->SetName( anImageName ); anImageObj->SetImage( myImage ); - anImageObj->SetLocalPoints( aPointA, aPointB, aPointC, false ); + + if ( aTransformationMode == HYDROData_Image::CartesianFromFile ) { + QString aGeoreferencementFileName = aPanel->getGeoreferencementFileName(); + if ( aGeoreferencementFileName.isEmpty() ) { + return false; + } - if ( aTransformationMode == HYDROData_Image::ReferenceImage ) - { - anImageObj->SetReferencePoints( aRefImageObj, + QPoint aLocalPointA( 0, 0 ), + aLocalPointB( anImageObj->Image().width(), 0 ), + aLocalPointC( INT_MIN, INT_MIN ); + anImageObj->SetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC, false ); + if ( !anImageObj->SetGlobalPointsFromFile( aGeoreferencementFileName ) ) { + theErrorMsg = tr( "CANT_LOAD_GEOREFERENCEMENT_FILE" ); + return false; + } + } else { + anImageObj->SetLocalPoints( aPointA, aPointB, aPointC, false ); + + if ( aTransformationMode == HYDROData_Image::ReferenceImage ) + { + anImageObj->SetReferencePoints( aRefImageObj, aTrsfPointA, aTrsfPointB, aTrsfPointC ); - } - else - { - anImageObj->SetGlobalPoints( aTransformationMode, + } + else + { + anImageObj->SetGlobalPoints( aTransformationMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ); + } } if( !myIsEdit ) @@ -395,18 +497,85 @@ bool HYDROGUI_ImportImageOp::processApply( int& theUpdateFlags, } // must be done after all checks and before calling SetVisible() method below - closePreview(); + if ( isApplyAndClose() ) + closePreview(); if( !myIsEdit ) module()->setObjectVisible( HYDROGUI_Tool::GetActiveGraphicsViewId( module() ), anImageObj, true ); - if( myIsEdit ) - anImageObj->Update(); + anImageObj->Update(); + + theUpdateFlags = UF_Model; + if ( isApplyAndClose() ) + theUpdateFlags |= UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced; + + if( isApplyAndClose() ) + { + commitDocOperation(); // to save the modifications in the data model + return true; + } - theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced; + if( SetNextFile() ) + { + theErrorMsg = ""; + module()->updateObjBrowser(); + return false; // and to continue the operation + } + + /*if( myFiles.count() > 1 ) + { + setIsApplyAndClose( true ); + }*/ return true; } +bool HYDROGUI_ImportImageOp::isReferenceCorrect() const +{ + bool isCorrect = true; + + if( myIsEdit && !myEditedObject.IsNull() ) + { + HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); + Handle(HYDROData_Image) aRefImageObj = Handle(HYDROData_Image)::DownCast( + HYDROGUI_Tool::FindObjectByName( module(), aPanel->getRefImageName(), KIND_IMAGE ) ); + if( !aRefImageObj.IsNull() ) + { + bool isFoundEdited = false; + HYDROData_SequenceOfObjects aRefSeq = aRefImageObj->GetAllReferenceObjects(); + for ( int i = 1, n = aRefSeq.Length(); i <= n && !isFoundEdited; ++i ) + { + Handle(HYDROData_Entity) anObject = aRefSeq.Value( i ); + isFoundEdited = anObject->GetName() == myEditedObject->GetName(); + } + isCorrect = !isFoundEdited; + } + } + return isCorrect; +} + +void HYDROGUI_ImportImageOp::apply() +{ + HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); + + bool aCanApply = isReferenceCorrect(); + if ( !aCanApply && !myEditedObject.IsNull() ) + { + Handle(HYDROData_Image) aRefImageObj = Handle(HYDROData_Image)::DownCast( + HYDROGUI_Tool::FindObjectByName( module(), aPanel->getRefImageName(), KIND_IMAGE ) ); + if ( !aRefImageObj.IsNull() ) + aCanApply = SUIT_MessageBox::question( module()->getApp()->desktop(), + tr( "CORRECT_INPUT_DATA" ), tr( "CONFIRM_REMOVE_REFERENCE_FROM_IMAGE" ). + arg( aRefImageObj->GetName() ).arg( myEditedObject->GetName() ), + QMessageBox::Yes | QMessageBox::No, QMessageBox::No ) == QMessageBox::Yes; + } + if ( aCanApply ) + HYDROGUI_Operation::apply(); + else { + aPanel->setRefImageName( "" ); + onRefImageActivated( aPanel->getRefImageName() ); + } +} + void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage ) { HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); @@ -415,9 +584,20 @@ void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage ) if( myPreviewPrs ) // if the image is changed by choosing another file { - myPreviewPrs->setImage( myImage ); - myPreviewPrs->compute(); - + myPreviewPrs->setImage( myImage, true ); + if ( sender() ) // reset the previous presentation settings + { + QString aFileName = aPanel->getFileName(); + + aPanel->reset(); + aPanel->setIsEdit( myIsEdit ); + // restore the file name + aPanel->setFileName( aFileName ); + // fill the reference list + HYDROGUI_ImportImageDlg::PrsPointDataList aPrsPointDataList; + getReferenceDataList( aPrsPointDataList ); + aPanel->setPrsPointDataList( aPrsPointDataList ); + } if( myPreviewViewManager ) { if( GraphicsView_Viewer* aViewer = myPreviewViewManager->getViewer() ) @@ -437,8 +617,7 @@ void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage ) myActiveViewManager = anApp->activeViewManager(); myPreviewPrs = new HYDROGUI_PrsImage( myIsEdit ? myEditedObject : 0 ); - myPreviewPrs->setImage( myImage ); - myPreviewPrs->compute(); + myPreviewPrs->setImage( myImage, true ); myPreviewViewManager = dynamic_cast( anApp->createViewManager( GraphicsView_Viewer::Type() ) ); @@ -457,29 +636,48 @@ void HYDROGUI_ImportImageOp::onCreatePreview( QImage theImage ) aViewPort->addItem( myPreviewPrs ); aViewPort->fitAll(); - + + if ( myEditedObject ) { + size_t aViewId = (size_t)aViewer; + module()->setObjectVisible( aViewId, myEditedObject, true ); + } + myPreviewPrs->setIsTransformationPointPreview( true ); + myPreviewPrs->setTransformationPointCursorShape( module()->getPrefEditCursor().shape() ); } connect( aViewer, SIGNAL( selectionChanged( GV_SelectionChangeStatus ) ), this, SLOT( onPointSelected() ) ); } } + } - QString anImageName; - if( myIsEdit ) - { - if( !myEditedObject.IsNull() ) - anImageName = myEditedObject->GetName(); + // Set the image name in the dialog + QString anImageName = aPanel->getImageName().simplified(); + // If edit mode and the name was not set yet then get from the edited object + if( myIsEdit && anImageName.isEmpty() ) + { + if( !myEditedObject.IsNull() ) + anImageName = myEditedObject->GetName(); + } + // If the name was not set then initialize it from the selected file name + if ( anImageName.isEmpty() ) + { + anImageName = aPanel->getFileName(); + if ( !anImageName.isEmpty() ) { + anImageName = QFileInfo( anImageName ).baseName(); + } + // If no file name then generate a new image name + if ( anImageName.isEmpty() ) { + anImageName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_IMAGE_NAME" ) ); } - else - anImageName = HYDROGUI_Tool::GenerateObjectName( module(), "Image" ); - aPanel->setImageName( anImageName ); } + aPanel->setImageName( anImageName ); aPanel->setImageSize( myImage.size() ); aPanel->initializePointSelection(); onPointSelected( false ); + onSetCIsUsed( !aPanel->isByTwoPoints() ); } void HYDROGUI_ImportImageOp::onActivatePointSelection( int thePointType ) @@ -525,6 +723,12 @@ void HYDROGUI_ImportImageOp::onModeActivated( int theMode ) void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName ) { + if( theName.isEmpty() ) { + if( myRefViewManager ) + closeView( myRefViewManager ); + return; + } + GraphicsView_ViewPort* aViewPort = 0; LightApp_Application* anApp = module()->getApp(); @@ -557,14 +761,17 @@ void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName ) } } } - qApp->processEvents(); if( !aViewPort ) return; + size_t aViewId = (size_t)myRefViewManager->getViewer(); + // Remove the old presentation of the reference image if any if( myRefPreviewPrs ) { + module()->setObjectVisible( aViewId, myRefPreviewPrs->getObject(), false ); + myRefPreviewPrs->setCaption( QString() ); aViewPort->removeItem( myRefPreviewPrs ); @@ -576,19 +783,25 @@ void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName ) QImage anImage; Handle(HYDROData_Image) anImageObj = Handle(HYDROData_Image)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theName, KIND_IMAGE ) ); + HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); if( !anImageObj.IsNull() ) { anImage = anImageObj->Image(); myRefPreviewPrs = new HYDROGUI_PrsImage( anImageObj ); - myRefPreviewPrs->setImage( anImage ); - myRefPreviewPrs->compute(); + myRefPreviewPrs->setImage( anImage, true ); myRefPreviewPrs->setIsTransformationPointPreview( true ); myRefPreviewPrs->setTransformationPointType( myPointType ); + myRefPreviewPrs->setIsByTwoPoints( aPanel->isByTwoPoints() ); + + myRefPreviewPrs->setTransformationPointCursorShape( module()->getPrefEditCursor().shape() ); + // Add the new reference image presentation to the appropriate view aViewPort->addItem( myRefPreviewPrs ); + + module()->setObjectVisible( aViewId, anImageObj, true ); } aViewPort->fitAll(); @@ -596,6 +809,7 @@ void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName ) // Split views horizontally if( anApp->desktop()->inherits( "STD_TabDesktop" ) ) { + qApp->processEvents(); QtxWorkstack* aWorkstack = ( (STD_TabDesktop*)anApp->desktop() )->workstack(); aViewPort->activateWindow(); aViewPort->show(); @@ -604,7 +818,6 @@ void HYDROGUI_ImportImageOp::onRefImageActivated( const QString& theName ) } // Initialize the dialog - HYDROGUI_ImportImageDlg* aPanel = (HYDROGUI_ImportImageDlg*)inputPanel(); aPanel->setImageSize( anImage.size(), true ); aPanel->initializePointSelection(); onPointSelected( true ); @@ -635,6 +848,7 @@ void HYDROGUI_ImportImageOp::onPointSelected( bool theIsRefImage ) const HYDROGUI_PrsImage::TransformationPointMap& aPointMap = aPrs->getTransformationPointMap(); + HYDROGUI_PrsImage::TransformationPointMapIterator anIter( aPointMap ); while( anIter.hasNext() ) { @@ -695,3 +909,22 @@ void HYDROGUI_ImportImageOp::closeView( GraphicsView_ViewManager* &aViewMgr ) aViewMgr = 0; } } + +void HYDROGUI_ImportImageOp::onFilesSelected( const QStringList& theFileNames ) +{ + myFiles = theFileNames; + myFileIndex = -1; + SetNextFile(); +} + +bool HYDROGUI_ImportImageOp::SetNextFile() +{ + myFileIndex++; + bool isEnabledEdit = myFiles.count()==1 || myFileIndex==myFiles.count(); + bool isValid = ( myFileIndex>=0 && myFileIndex( inputPanel() ); + aPanel->ActivateFile( aFile, isEnabledEdit ); + return isValid; +}