Salome HOME
Image update corrected.
[modules/hydro.git] / src / HYDROData / HYDROData_Image.cxx
index 362e151577e0dcf24ad2ba85377567fe0c440756..1a0bec79ebd2b0251d9023a07cfc1f42eb716887 100644 (file)
@@ -2,14 +2,14 @@
 #include "HYDROData_Image.h"
 
 #include "HYDROData_Document.h"
-#include "HYDROData_Tool.h"
-#include "HYDROOperations_Factory.h"
+#include "HYDROData_Lambert93.h"
+#include "HYDROData_OperationsFactory.h"
 
 #include <TDataStd_RealArray.hxx>
 #include <TDataStd_ByteArray.hxx>
+#include <TDataStd_Integer.hxx>
 #include <TDataStd_IntegerArray.hxx>
 #include <TDataStd_ReferenceList.hxx>
-#include <TDataStd_Name.hxx>
 #include <TDataStd_UAttribute.hxx>
 #include <TDataStd_AsciiString.hxx>
 
 
 #include <QStringList>
 
-static const Standard_GUID GUID_MUST_BE_UPDATED("80f2bb81-3873-4631-8ddd-940d2119f000");
 static const Standard_GUID GUID_SELF_SPLITTED("997995aa-5c19-40bf-9a60-ab4b70ad04d8");
+static const Standard_GUID GUID_HAS_LOCAL_POINTS("FD8841AA-FC44-42fa-B6A7-0F682CCC6F27");
+static const Standard_GUID GUID_HAS_GLOBAL_POINTS("330D0E81-742D-4ea3-92D4-484877CFA7C1");
 
 #define PYTHON_IMAGE_ID "KIND_IMAGE"
 
-IMPLEMENT_STANDARD_HANDLE(HYDROData_Image, HYDROData_Object)
-IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Image, HYDROData_Object)
+IMPLEMENT_STANDARD_HANDLE(HYDROData_Image, HYDROData_Entity)
+IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Image, HYDROData_Entity)
 
 HYDROData_Image::HYDROData_Image()
+: HYDROData_Entity()
 {
 }
 
@@ -38,7 +40,7 @@ QStringList HYDROData_Image::DumpToPython( MapOfTreatedObjects& theTreatedObject
 {
   QStringList aResList;
 
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( this );
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
   if ( aDocument.IsNull() )
     return aResList;
                              
@@ -56,50 +58,45 @@ QStringList HYDROData_Image::DumpToPython( MapOfTreatedObjects& theTreatedObject
     aResList << QString( "" );
     aResList << QString( "%1.LoadImage( \"%2\" );" )
                 .arg( anImageName ).arg( aFilePath );
-
-    // Dump transformation matrix for image
     aResList << QString( "" );
 
-    QTransform aTrsf = Trsf();
-
-    aResList << QString( "trsf = QTransform( %1, %2, %3," )
-                .arg( aTrsf.m11() ).arg( aTrsf.m12() ).arg( aTrsf.m13() );
-    aResList << QString( "                   %1, %2, %3," )
-                .arg( aTrsf.m21() ).arg( aTrsf.m22() ).arg( aTrsf.m23() );
-    aResList << QString( "                   %1, %2, %3 );" )
-                .arg( aTrsf.m31() ).arg( aTrsf.m32() ).arg( aTrsf.m33() );
-
-    aResList << QString( "%1.SetTrsf( trsf );" ).arg( anImageName );
-
     // Dump transformation points for image
-    aResList << QString( "" );
-
-    QPoint aPointAIn,  aPointBIn,  aPointCIn;
-    QPointF aPointAOut, aPointBOut, aPointCOut;
-    TrsfPoints( aPointAIn,  aPointBIn,  aPointCIn,
-                aPointAOut, aPointBOut, aPointCOut );
-
-    aResList << QString( "a_in = QPoint( %1, %2 );" )
-                .arg( aPointAIn.x() ).arg( aPointAIn.y() );
 
-    aResList << QString( "b_in = QPoint( %1, %2 );" )
-                .arg( aPointBIn.x() ).arg( aPointBIn.y() );
-
-    aResList << QString( "c_in = QPoint( %1, %2 );" )
-                .arg( aPointCIn.x() ).arg( aPointCIn.y() );
-
-    aResList << QString( "a_out = QPointF( %1, %2 );" )
-                .arg( aPointAOut.x() ).arg( aPointAOut.y() );
+    QString aGap = QString().fill( ' ', anImageName.size() + 16 );
 
-    aResList << QString( "b_out = QPointF( %1, %2 );" )
-                .arg( aPointBOut.x() ).arg( aPointBOut.y() );
+    bool anIsByTwoPoints = IsByTwoPoints();
 
-    aResList << QString( "c_out = QPointF( %1, %2 );" )
-                .arg( aPointCOut.x() ).arg( aPointCOut.y() );
+    QPoint aLocalPointA, aLocalPointB, aLocalPointC;
+    if ( GetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC ) )
+    {
+      aResList << QString( "%1.SetLocalPoints( QPoint( %2, %3 )," )
+                  .arg( anImageName ).arg( aLocalPointA.x() ).arg( aLocalPointA.y() );
+      aResList << QString( aGap             + "QPoint( %1, %2 )," )
+                  .arg( aLocalPointB.x() ).arg( aLocalPointB.y() );
+      aResList << QString( aGap             + "QPoint( %1, %2 ) );" )
+                  .arg( aLocalPointC.x() ).arg( aLocalPointC.y() );
+      aResList << QString( "" );
+    }
 
-    QString aGap = QString().fill( ' ', anImageName.size() + 16 );
-    aResList << QString( "%1.SetTrsfPoints( a_in,  b_in,  c_in," ).arg( anImageName );
-    aResList << QString( aGap            + "a_out, b_out, c_out );" );
+    HYDROData_Image::TransformationMode aTransformationMode;
+    QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC;
+    if ( GetGlobalPoints( aTransformationMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ) )
+    {
+      aResList << QString( "%1.SetGlobalPoints( %2," )
+                  .arg( anImageName ).arg( aTransformationMode );
+      aResList << QString( aGap             +  "QPointF( %1, %2 )," )
+                  .arg( aTrsfPointA.x() ).arg( aTrsfPointA.y() );
+      aResList << QString( aGap             +  "QPointF( %1, %2 )," )
+                  .arg( aTrsfPointB.x() ).arg( aTrsfPointB.y() );
+      aResList << QString( aGap             +  "QPointF( %1, %2 ) );" )
+                  .arg( aTrsfPointC.x() ).arg( aTrsfPointC.y() );
+
+      if ( aTransformationMode == ReferenceImage )
+      {
+        Handle(HYDROData_Image) aRefImg = GetTrsfReferenceImage();
+        setPythonReferenceObject( theTreatedObjects, aResList, aRefImg, "SetTrsfReferenceImage" );
+      }
+    }
   }
   else
   {
@@ -113,7 +110,8 @@ QStringList HYDROData_Image::DumpToPython( MapOfTreatedObjects& theTreatedObject
       aResList << QString( "%1.SetOperatorName( \"%2\" );" )
                   .arg( anImageName ).arg( anOperatorName );
 
-      ImageComposer_Operator* anImageOp = HYDROOperations_Factory::Factory()->Operator( this );
+      ImageComposer_Operator* anImageOp = 
+        HYDROData_OperationsFactory::Factory()->Operator( OperatorName() );
       if ( anImageOp )
       {
         // Dump operation arguments
@@ -139,57 +137,42 @@ QStringList HYDROData_Image::DumpToPython( MapOfTreatedObjects& theTreatedObject
       for ( int i = 0; i < aNbReferences; ++i )
       {
         Handle(HYDROData_Image) aRefImg = Handle(HYDROData_Image)::DownCast( Reference( i ) );
-        if ( aRefImg.IsNull() )
-          continue;
-
-        QString aRefImgName = aRefImg->GetName();
-
-        // The definition of reference image must be dumped before this
-        if ( !theTreatedObjects.contains( aRefImgName ) )
-        {
-          // Write definition of reference image
-          QStringList aRefImgDump = aRefImg->DumpToPython( theTreatedObjects );
-          if ( aRefImgDump.isEmpty() )
-            continue;
-
-          QStringList aTmpList = aResList;
-          aResList = aRefImgDump;
-
-          aResList.append( "" );
-          aResList.append( aTmpList );
-
-          theTreatedObjects.insert( aRefImgName, aRefImg );
-        }
-
-        aResList << QString( "%1.AppendReference( %2 );" )
-                    .arg( anImageName ).arg( aRefImgName );
+        setPythonReferenceObject( theTreatedObjects, aResList, aRefImg, "AppendReference" );
       }
     }
 
     // Necessary to update image in case of composed operator
     aResList << QString( "" );
-    aResList << QString( "%1.Update( False );" ).arg( anImageName );
+    aResList << QString( "%1.Update();" ).arg( anImageName );
   }
 
   return aResList;
 }
 
-void HYDROData_Image::Update( const bool theIsForce )
+void HYDROData_Image::Update()
 {
-  HYDROOperations_Factory* aFactory = HYDROOperations_Factory::Factory();
+  bool anIsToUpdate = IsMustBeUpdated();
+
+  HYDROData_Entity::Update();
+
+  if ( !anIsToUpdate )
+    return;
+
+  HYDROData_OperationsFactory* aFactory = HYDROData_OperationsFactory::Factory();
 
-  // Update image only if there is an operation
   ImageComposer_Operator* anOp = aFactory->Operator( OperatorName() );
-  if ( anOp ) 
+  if ( anOp ) // Update image if there is an operation
   {
     // Fill by arguments and process the operation
+    anOp->setBinArgs( Args() );
+
     QVariant anObj1, anObj2;
     int aNbReferences = NbReferences();
 
     if ( aNbReferences > 0 )
     {
       // First referenced object
-      Handle(HYDROData_Object) aRefObj = Reference( 0 );
+      Handle(HYDROData_Entity) aRefObj = Reference( 0 );
       if ( !aRefObj.IsNull() )
       {
         anObj1 = aRefObj->GetDataVariant();
@@ -205,26 +188,21 @@ void HYDROData_Image::Update( const bool theIsForce )
     if ( aNbReferences > 1 )
     {
       // Second referenced object
-      Handle(HYDROData_Object) aRefObj = Reference( 1 );
+      Handle(HYDROData_Entity) aRefObj = Reference( 1 );
       if ( !aRefObj.IsNull() )
         anObj2 = aRefObj->GetDataVariant();
     }
 
     ImageComposer_Image aResImg = anOp->process( anObj1, anObj2 );
     SetImage( aResImg );
+    SetTrsf( aResImg.transform() );
   }
-
-  if ( theIsForce )
+  else // Update image if it positioned relatively to other image
   {
-    Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( this );
-    if ( !aDocument.IsNull() )
-    {
-      // Change the states of this and all depended images
-      MustBeUpdated( true );
-      HYDROData_Tool::SetMustBeUpdatedImages( aDocument );
-      MustBeUpdated( false );
-    }
+    UpdateTrsf();
   }
+
+  SetToUpdate( false );
 }
 
 QVariant HYDROData_Image::GetDataVariant()
@@ -240,29 +218,48 @@ QVariant HYDROData_Image::GetDataVariant()
   return aVarData;
 }
 
+HYDROData_SequenceOfObjects HYDROData_Image::GetAllReferenceObjects() const
+{
+  HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
+
+  Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
+  if ( !aRefImage.IsNull() )
+    aResSeq.Append( aRefImage );
+
+  HYDROData_SequenceOfObjects aSeqRefObjects = GetReferenceObjects( 0 );
+  aResSeq.Append( aSeqRefObjects );
+
+  return aResSeq;
+}
+
 void HYDROData_Image::SetImage(const QImage& theImage)
 {
-  if (theImage.isNull()) {
+  if ( theImage.isNull() )
+  {
     // for empty image remove all previously stored attributes
     myLab.ForgetAttribute(TDataStd_IntegerArray::GetID());
     myLab.ForgetAttribute(TDataStd_ByteArray::GetID());
-    return;
   }
-  // store width, height, bytes per line and format in integer array 
-  Handle(TDataStd_IntegerArray) aParams;
-  if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams)) {
-    aParams = TDataStd_IntegerArray::Set(myLab, 1, 4);
+  else
+  {
+    // store width, height, bytes per line and format in integer array 
+    Handle(TDataStd_IntegerArray) aParams;
+    if (!myLab.FindAttribute(TDataStd_IntegerArray::GetID(), aParams)) {
+      aParams = TDataStd_IntegerArray::Set(myLab, 1, 4);
+    }
+    aParams->SetValue(1, theImage.width());
+    aParams->SetValue(2, theImage.height());
+    aParams->SetValue(3, theImage.bytesPerLine());
+    aParams->SetValue(4, (int)(theImage.format()));
+    // store data of image in byte array
+    const char* aData = (const char*)(theImage.bits());
+    SaveByteArray(0, aData, theImage.byteCount());
   }
-  aParams->SetValue(1, theImage.width());
-  aParams->SetValue(2, theImage.height());
-  aParams->SetValue(3, theImage.bytesPerLine());
-  aParams->SetValue(4, (int)(theImage.format()));
-  // store data of image in byte array
-  const char* aData = (const char*)(theImage.bits());
-  SaveByteArray(0, aData, theImage.byteCount());
+
+  SetToUpdate( true );
 }
 
-bool HYDROData_Image::LoadImage(const QString& theFilePath)
+bool HYDROData_Image::LoadImage( const QString& theFilePath )
 {
   QImage anImage( theFilePath );
   SetImage( anImage );
@@ -283,19 +280,25 @@ QImage HYDROData_Image::Image()
   return aResult;
 }
 
-void HYDROData_Image::SetFilePath(const QString& theFilePath)
+void HYDROData_Image::SetFilePath( const QString& theFilePath )
 {
   TCollection_AsciiString anAsciiStr( theFilePath.toStdString().c_str() );
   TDataStd_AsciiString::Set( myLab.FindChild( DataTag_FilePath ), anAsciiStr );
+
+  SetToUpdate( true );
 }
 
 QString HYDROData_Image::GetFilePath() const
 {
   QString aRes;
 
-  Handle(TDataStd_AsciiString) anAsciiStr;
-  if ( myLab.FindChild( DataTag_FilePath ).FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
-    aRes = QString( anAsciiStr->Get().ToCString() );
+  TDF_Label aLabel = myLab.FindChild( DataTag_FilePath, false );
+  if ( !aLabel.IsNull() )
+  {
+    Handle(TDataStd_AsciiString) anAsciiStr;
+    if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
+      aRes = QString( anAsciiStr->Get().ToCString() );
+  }
 
   return aRes;
 }
@@ -317,6 +320,8 @@ void HYDROData_Image::SetTrsf(const QTransform& theTrsf)
   anArray->SetValue(7, theTrsf.m31());
   anArray->SetValue(8, theTrsf.m32());
   anArray->SetValue(9, theTrsf.m33());
+
+  SetToUpdate( true );
 }
 
 QTransform HYDROData_Image::Trsf() const
@@ -332,58 +337,352 @@ QTransform HYDROData_Image::Trsf() const
   return aTrsf;
 }
 
-void HYDROData_Image::SetTrsfPoints(const QPoint& thePointAIn,
-                                    const QPoint& thePointBIn,
-                                    const QPoint& thePointCIn,
-                                    const QPointF& thePointAOut,
-                                    const QPointF& thePointBOut,
-                                    const QPointF& thePointCOut)
+void HYDROData_Image::UpdateTrsf()
 {
-  Handle(TDataStd_RealArray) anArray;
-  if (!myLab.FindChild(DataTag_TrsfPoints).FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
-    anArray = TDataStd_RealArray::Set(myLab.FindChild(DataTag_TrsfPoints), 1, 12);
+  QPoint aPointA, aPointB, aPointC;
+  if ( !GetLocalPoints( aPointA, aPointB, aPointC ) )
+    return;
+
+  TransformationMode aTrsfMode;
+  QPointF aTrsfPointA, aTrsfPointB, aTrsfPointC;
+  if ( !GetGlobalPoints( aTrsfMode, aTrsfPointA, aTrsfPointB, aTrsfPointC ) )
+    return;
+
+  QTransform aRefTransform;
+  Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
+
+  bool anIsRefImage = aTrsfMode == ReferenceImage;
+  if ( anIsRefImage )
+  {
+    if ( aRefImage.IsNull() )
+      return;
+
+    aRefTransform = aRefImage->Trsf();
+  }
+
+  bool anIsByTwoPoints = IsByTwoPoints();
+
+  // Convert lambert coordinates to cartesian
+  if ( aTrsfMode == ManualGeodesic )
+  {
+    double aXCart = 0, aYCart = 0;
+
+    HYDROData_Lambert93::toXY( aTrsfPointA.y(), aTrsfPointA.x(), aXCart, aYCart );
+    aTrsfPointA = QPointF( aXCart, aYCart );
+
+    HYDROData_Lambert93::toXY( aTrsfPointB.y(), aTrsfPointB.x(), aXCart, aYCart );
+    aTrsfPointB = QPointF( aXCart, aYCart );
+
+    if ( !anIsByTwoPoints )
+    {
+      HYDROData_Lambert93::toXY( aTrsfPointC.y(), aTrsfPointC.x(), aXCart, aYCart );
+      aTrsfPointC = QPointF( aXCart, aYCart );
+    }
+  }
+
+  // generate third points if needed
+  if ( anIsByTwoPoints )
+  {
+    aPointC = generateThirdPoint( aPointA, aPointB, true ).toPoint();
+    aTrsfPointC = generateThirdPoint( aTrsfPointA, aTrsfPointB, anIsRefImage );
   }
-  anArray->SetValue(1, thePointAIn.x());
-  anArray->SetValue(2, thePointAIn.y());
-  anArray->SetValue(3, thePointBIn.x());
-  anArray->SetValue(4, thePointBIn.y());
-  anArray->SetValue(5, thePointCIn.x());
-  anArray->SetValue(6, thePointCIn.y());
-  anArray->SetValue(7, thePointAOut.x());
-  anArray->SetValue(8, thePointAOut.y());
-  anArray->SetValue(9, thePointBOut.x());
-  anArray->SetValue(10, thePointBOut.y());
-  anArray->SetValue(11, thePointCOut.x());
-  anArray->SetValue(12, thePointCOut.y());
-}
-
-void HYDROData_Image::TrsfPoints(QPoint& thePointAIn,
-                                 QPoint& thePointBIn,
-                                 QPoint& thePointCIn,
-                                 QPointF& thePointAOut,
-                                 QPointF& thePointBOut,
-                                 QPointF& thePointCOut) const
+
+  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 xta = aTrsfPointA.x();
+  double yta = aTrsfPointA.y();
+  double xtb = aTrsfPointB.x();
+  double ytb = aTrsfPointB.y();
+  double xtc = aTrsfPointC.x();
+  double ytc = aTrsfPointC.y();
+
+  // first, check that three input points don't belong to a single line
+  if( ( yb - ya ) * ( xc - xa ) == ( yc - ya ) * ( xb - xa ) )
+    return;
+
+  // the same check for the reference points
+  if( anIsRefImage && ( ( ytb - yta ) * ( xtc - xta ) == ( ytc - yta ) * ( xtb - xta ) ) )
+    return;
+
+  QTransform aTransform1( xa, ya, 1, xb, yb, 1, xc, yc, 1 );
+  QTransform aTransform2( xta, yta, 1, xtb, ytb, 1, xtc, ytc, 1 );
+
+  bool anIsInvertible = false;
+  QTransform aTransform1Inverted = aTransform1.inverted( &anIsInvertible );
+  if( !anIsInvertible )
+    return;
+
+  QTransform aResTransform = aTransform1Inverted * aTransform2;
+  if( anIsRefImage )
+    aResTransform *= aRefTransform;
+
+  SetTrsf( aResTransform );
+}
+
+bool HYDROData_Image::IsByTwoPoints() const
 {
-  Handle(TDataStd_RealArray) anArray;
-  if (myLab.FindChild(DataTag_TrsfPoints).FindAttribute(TDataStd_RealArray::GetID(), anArray)) {
-    thePointAIn = QPointF( anArray->Value(1), anArray->Value(2) ).toPoint();
-    thePointBIn = QPointF( anArray->Value(3), anArray->Value(4) ).toPoint();
-    thePointCIn = QPointF( anArray->Value(5), anArray->Value(6) ).toPoint();
-    thePointAOut = QPointF( anArray->Value(7), anArray->Value(8) );
-    thePointBOut = QPointF( anArray->Value(9), anArray->Value(10) );
-    thePointCOut = QPointF( anArray->Value(11), anArray->Value(12) );
+  if ( !HasLocalPoints() || !HasGlobalPoints() )
+    return false;
+
+  QPoint aPointA, aPointB, aPointC;
+  GetLocalPoints( aPointA, aPointB, aPointC );
+
+  return aPointC.x() < 0 && aPointC.y() < 0; 
+}
+
+bool HYDROData_Image::HasReferences() const
+{
+  Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
+  int aNbReferences = NbReferences();
+
+  return !aRefImage.IsNull() || aNbReferences > 0;
+}
+
+void HYDROData_Image::RemoveAllReferences()
+{
+  if ( !HasReferences() )
+    return;
+
+  Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
+  if ( !aRefImage.IsNull() )
+  {
+    RemoveTrsfReferenceImage();
+  }
+  else
+  {
+    ClearReferences();
+    SetOperatorName( "" );
+    SetArgs( "" );
+    SetIsSelfSplitted( false );
+  }
+
+  bool anIsByTwoPoints = IsByTwoPoints();
+
+  QImage anImage = Image();
+  if ( anImage.isNull() )
+  {
+    SetToUpdate( false );
+    return;
   }
+
+  // Set local points to default position
+  QPoint aLocalPointA = QPoint( 0, 0 );
+  QPoint aLocalPointB = QPoint( anImage.width(), 0 );
+  QPoint aLocalPointC = anIsByTwoPoints ? QPoint( INT_MIN, INT_MIN ) : QPoint( 0, anImage.height() );
+
+  SetLocalPoints( aLocalPointA, aLocalPointB, aLocalPointC, false );
+
+  // Calculate global points
+  QTransform aTransform = Trsf();
+
+  QPointF aTrsfPointA = QPointF( aTransform.map( aLocalPointA ) );
+  QPointF aTrsfPointB = QPointF( aTransform.map( aLocalPointB ) );
+  QPointF aTrsfPointC = anIsByTwoPoints ? QPointF( INT_MIN, INT_MIN ) : 
+                                          QPointF( aTransform.map( aLocalPointC ) );
+
+  SetGlobalPoints( ManualCartesian, aTrsfPointA, aTrsfPointB, aTrsfPointC );
+
+  SetToUpdate( false );
+}
+
+void HYDROData_Image::SetLocalPoints( const QPoint& thePointA,
+                                      const QPoint& thePointB,
+                                      const QPoint& thePointC,
+                                      const bool    theIsUpdate )
+{
+  Handle(TDataStd_RealArray) anArray;
+  if ( !myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
+    anArray = TDataStd_RealArray::Set( myLab.FindChild( DataTag_TrsfPoints ), 1, 12 );
+
+  anArray->SetValue( 1, thePointA.x() );
+  anArray->SetValue( 2, thePointA.y() );
+  anArray->SetValue( 3, thePointB.x() );
+  anArray->SetValue( 4, thePointB.y() );
+  anArray->SetValue( 5, thePointC.x() );
+  anArray->SetValue( 6, thePointC.y() );
+
+  TDataStd_UAttribute::Set( myLab.FindChild( DataTag_TrsfPoints ), GUID_HAS_LOCAL_POINTS );
+
+  if ( theIsUpdate )
+    UpdateTrsf();
+
+  SetToUpdate( true );
+}
+
+bool HYDROData_Image::GetLocalPoints( QPoint& thePointA,
+                                      QPoint& thePointB,
+                                      QPoint& thePointC ) const
+{
+  if ( !HasLocalPoints() )
+    return false;
+
+  Handle(TDataStd_RealArray) anArray;
+  myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray );
+
+  thePointA = QPointF( anArray->Value( 1 ), anArray->Value( 2 ) ).toPoint();
+  thePointB = QPointF( anArray->Value( 3 ), anArray->Value( 4 ) ).toPoint();
+  thePointC = QPointF( anArray->Value( 5 ), anArray->Value( 6 ) ).toPoint();
+
+  return true;
+}
+
+bool HYDROData_Image::HasLocalPoints() const
+{
+  TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
+  if ( aLabel.IsNull() || !aLabel.IsAttribute( GUID_HAS_LOCAL_POINTS ) )
+    return false;
+
+  Handle(TDataStd_RealArray) anArray;
+  return aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray );
+}
+
+
+void HYDROData_Image::SetGlobalPoints( const TransformationMode& theMode,
+                                       const QPointF&            thePointA,
+                                       const QPointF&            thePointB,
+                                       const QPointF&            thePointC,
+                                       const bool                theIsUpdate )
+{
+  Handle(TDataStd_RealArray) anArray;
+  if ( !myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray ) )
+    anArray = TDataStd_RealArray::Set( myLab.FindChild( DataTag_TrsfPoints ), 1, 12 );
+
+  anArray->SetValue( 7,  thePointA.x() );
+  anArray->SetValue( 8,  thePointA.y() );
+  anArray->SetValue( 9,  thePointB.x() );
+  anArray->SetValue( 10, thePointB.y() );
+  anArray->SetValue( 11, thePointC.x() );
+  anArray->SetValue( 12, thePointC.y() );
+  
+  SetTrsfMode( theMode );
+
+  TDataStd_UAttribute::Set( myLab.FindChild( DataTag_TrsfPoints ), GUID_HAS_GLOBAL_POINTS );
+
+  if ( theIsUpdate )
+    UpdateTrsf();
+
+  SetToUpdate( true );
+}
+
+bool HYDROData_Image::GetGlobalPoints( TransformationMode& theMode,
+                                       QPointF&            thePointA,
+                                       QPointF&            thePointB,
+                                       QPointF&            thePointC ) const
+{
+  if ( !HasGlobalPoints() )
+    return false;
+
+  theMode = GetTrsfMode();
+
+  Handle(TDataStd_RealArray) anArray;
+  myLab.FindChild( DataTag_TrsfPoints ).FindAttribute( TDataStd_RealArray::GetID(), anArray );
+
+  thePointA = QPointF( anArray->Value( 7  ), anArray->Value( 8  ) );
+  thePointB = QPointF( anArray->Value( 9  ), anArray->Value( 10 ) );
+  thePointC = QPointF( anArray->Value( 11 ), anArray->Value( 12 ) );
+
+  return true;
 }
 
-bool HYDROData_Image::HasTrsfPoints() const
+bool HYDROData_Image::HasGlobalPoints() const
 {
+  TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
+  if ( aLabel.IsNull() || !aLabel.IsAttribute( GUID_HAS_GLOBAL_POINTS ) )
+    return false;
+
   Handle(TDataStd_RealArray) anArray;
-  return myLab.FindChild(DataTag_TrsfPoints).FindAttribute(TDataStd_RealArray::GetID(), anArray);
+  return aLabel.FindAttribute( TDataStd_RealArray::GetID(), anArray );
 }
 
-void HYDROData_Image::AppendReference( const Handle(HYDROData_Object)& theReferenced )
+void HYDROData_Image::SetReferencePoints( const Handle(HYDROData_Image)& theRefImage,
+                                          const QPointF&                 thePointA,
+                                          const QPointF&                 thePointB,
+                                          const QPointF&                 thePointC,
+                                          const bool                     theIsUpdate )
+{
+  SetTrsfReferenceImage( theRefImage );
+  SetGlobalPoints( ReferenceImage, thePointA, thePointB, thePointC, theIsUpdate );
+}
+
+bool HYDROData_Image::GetReferencePoints( Handle(HYDROData_Image)& theRefImage,
+                                          QPointF&                 thePointA,
+                                          QPointF&                 thePointB,
+                                          QPointF&                 thePointC ) const
+{
+  if ( !HasReferencePoints() )
+    return false;
+
+  theRefImage = GetTrsfReferenceImage();
+
+  TransformationMode aMode;
+  GetGlobalPoints( aMode, thePointA, thePointB, thePointC );
+
+  return true;
+}
+
+bool HYDROData_Image::HasReferencePoints() const
+{
+  if ( !HasGlobalPoints() )
+    return false;
+
+  Handle(HYDROData_Image) aRefImage = GetTrsfReferenceImage();
+  if ( aRefImage.IsNull() )
+    return false;
+
+  TransformationMode aTrsfMode = GetTrsfMode();
+  if ( aTrsfMode != ReferenceImage )
+    return false;
+
+  return true;
+}
+
+void HYDROData_Image::SetTrsfMode( const TransformationMode& theMode )
+{
+  TDataStd_Integer::Set( myLab.FindChild( DataTag_TrsfMode ), (int)theMode );
+  SetToUpdate( true );
+}
+
+HYDROData_Image::TransformationMode HYDROData_Image::GetTrsfMode() const
+{
+  TransformationMode aResMode = ManualGeodesic;
+
+  TDF_Label aLabel = myLab.FindChild( DataTag_TrsfPoints, false );
+  if ( !aLabel.IsNull() )
+  {
+    Handle(TDataStd_Integer) aMode;
+    if ( myLab.FindChild( DataTag_TrsfMode ).FindAttribute( TDataStd_Integer::GetID(), aMode ) )
+      aResMode = (TransformationMode)aMode->Get();
+  }
+
+  return aResMode;
+}
+
+void HYDROData_Image::SetTrsfReferenceImage( const Handle(HYDROData_Image)& theRefImage )
+{
+  SetReferenceObject( theRefImage, DataTag_TrsfImage );
+  SetToUpdate( true );
+}
+
+Handle(HYDROData_Image) HYDROData_Image::GetTrsfReferenceImage() const
+{
+  return Handle(HYDROData_Image)::DownCast( GetReferenceObject( DataTag_TrsfImage ) );
+}
+
+void HYDROData_Image::RemoveTrsfReferenceImage()
+{
+  RemoveReferenceObject( DataTag_TrsfImage );
+  SetToUpdate( true );
+}
+
+void HYDROData_Image::AppendReference( const Handle(HYDROData_Entity)& theReferenced )
 {
   AddReferenceObject( theReferenced, 0 );
+  SetToUpdate( true );
 }
 
 int HYDROData_Image::NbReferences() const
@@ -391,47 +690,56 @@ int HYDROData_Image::NbReferences() const
   return NbReferenceObjects( 0 );
 }
 
-Handle(HYDROData_Object) HYDROData_Image::Reference( const int theIndex ) const
+Handle(HYDROData_Entity) HYDROData_Image::Reference( const int theIndex ) const
 {
   return GetReferenceObject( 0, theIndex );
 }
 
 void HYDROData_Image::ChangeReference(
-    const int theIndex, Handle(HYDROData_Object) theReferenced)
+    const int theIndex, Handle(HYDROData_Entity) theReferenced)
 {
   SetReferenceObject( theReferenced, 0, theIndex );
+  SetToUpdate( true );
 }
 
 void HYDROData_Image::RemoveReference(const int theIndex)
 {
   RemoveReferenceObject( 0, theIndex );
+  SetToUpdate( true );
 }
 
 void HYDROData_Image::ClearReferences()
 {
   ClearReferenceObjects( 0 );
+  SetToUpdate( true );
 }
 
-void HYDROData_Image::SetOperatorName(const QString theOpName)
+void HYDROData_Image::SetOperatorName( const QString theOpName )
 {
-  TDataStd_Name::Set(myLab.FindChild(DataTag_Operator),
-    TCollection_ExtendedString(theOpName.toLatin1().constData()));
+  TCollection_AsciiString anAsciiStr( theOpName.toStdString().c_str() );
+  TDataStd_AsciiString::Set( myLab.FindChild( DataTag_Operator ), anAsciiStr );
+  SetToUpdate( true );
 }
 
 QString HYDROData_Image::OperatorName() const
 {
-  Handle(TDataStd_Name) aName;
-  if (myLab.FindChild(DataTag_Operator).
-        FindAttribute(TDataStd_Name::GetID(), aName)) {
-    TCollection_AsciiString aStr(aName->Get());
-    return QString(aStr.ToCString());
+  QString aRes;
+
+  TDF_Label aLabel = myLab.FindChild( DataTag_Operator, false );
+  if ( !aLabel.IsNull() )
+  {
+    Handle(TDataStd_AsciiString) anAsciiStr;
+    if ( aLabel.FindAttribute( TDataStd_AsciiString::GetID(), anAsciiStr ) )
+      aRes = QString( anAsciiStr->Get().ToCString() );
   }
-  return QString();
+
+  return aRes;
 }
 
 void HYDROData_Image::SetArgs(const QByteArray& theArgs)
 {
   SaveByteArray(DataTag_Operator, theArgs.constData(), theArgs.length());
+  SetToUpdate( true );
 }
 
 QByteArray HYDROData_Image::Args() const
@@ -443,20 +751,6 @@ QByteArray HYDROData_Image::Args() const
   return QByteArray(aData, aLen);
 }
 
-void HYDROData_Image::MustBeUpdated(bool theFlag)
-{
-  if (theFlag) {
-    TDataStd_UAttribute::Set(myLab, GUID_MUST_BE_UPDATED);
-  } else {
-    myLab.ForgetAttribute(GUID_MUST_BE_UPDATED);
-  }
-}
-
-bool HYDROData_Image::MustBeUpdated() const
-{
-  return myLab.IsAttribute(GUID_MUST_BE_UPDATED);
-}
-
 void HYDROData_Image::SetIsSelfSplitted(bool theFlag)
 {
   if (theFlag) {
@@ -464,9 +758,35 @@ void HYDROData_Image::SetIsSelfSplitted(bool theFlag)
   } else {
     myLab.ForgetAttribute(GUID_SELF_SPLITTED);
   }
+  SetToUpdate( true );
 }
 
 bool HYDROData_Image::IsSelfSplitted() const
 {
   return myLab.IsAttribute(GUID_SELF_SPLITTED);
 }
+
+QPointF HYDROData_Image::generateThirdPoint( const QPointF& thePointA,
+                                             const QPointF& thePointB,
+                                             const bool&    theIsLocal ) const
+{
+  // Rotate vector to 90 degrees : clockwise - for local
+  //                               counterclockwise - for global
+  const double aTheta = theIsLocal ? -M_PI_2 : M_PI_2;
+
+  QPointF aResPoint;
+
+  // Move to (0,0) for correct rotation
+  double x = thePointB.x() - thePointA.x();
+  double y = thePointB.y() - thePointA.y();
+
+  aResPoint.setX( x * cos( aTheta ) - y * sin( aTheta ) );
+  aResPoint.setY( x * sin( aTheta ) + y * cos( aTheta ) );
+
+  // Move back to origin position
+  aResPoint.setX( aResPoint.x() + thePointA.x() );
+  aResPoint.setY( aResPoint.y() + thePointA.y() );
+
+  return aResPoint;
+}
+