Dump Image data to python script (Feature #13).

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_TwoImagesOp.cxx

diff --git a/src/HYDROGUI/HYDROGUI_TwoImagesOp.cxx b/src/HYDROGUI/HYDROGUI_TwoImagesOp.cxx
index dec0f974cc981af478bad21c80a2c52555d5fc29..74deaf716f1e178ad6f5c9d3fbd6803096467413 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_TwoImagesOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_TwoImagesOp.cxx
@@ -37,8 +37,29 @@
 
 HYDROGUI_TwoImagesOp::HYDROGUI_TwoImagesOp( HYDROGUI_Module* theModule, const int theType )
 : HYDROGUI_Operation( theModule ),
-  myType( theType )
+  myType( theType ),
+  myIsEdit( false ),
+  myEditedObject( 0 )
 {
+  if( myType == Edit )
+  {
+    myIsEdit = true;
+    myEditedObject = Handle(HYDROData_Image)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
+    if( !myEditedObject.IsNull() )
+    {
+      if( HYDROOperations_Factory* aFactory = HYDROOperations_Factory::Factory() )
+      {
+        if( ImageComposer_Operator* anOperator = aFactory->Operator( myEditedObject ) )
+        {
+          if( dynamic_cast<ImageComposer_FuseOperator*>( anOperator ) )
+            myType = Fuse;
+          else if( dynamic_cast<ImageComposer_CutOperator*>( anOperator ) )
+            myType = Cut;
+        }
+      }
+    }
+  }
+
   QString aName;
   switch( myType )
   {
@@ -65,8 +86,41 @@ void HYDROGUI_TwoImagesOp::startOperation()
   HYDROGUI_TwoImagesDlg* aPanel = (HYDROGUI_TwoImagesDlg*)inputPanel();
   aPanel->reset();
 
-  QString anImageName = HYDROGUI_Tool::GenerateObjectName( module(), getName() );
+  QString anImageName;
+  if( myIsEdit )
+  {
+    if( !myEditedObject.IsNull() )
+      anImageName = myEditedObject->GetName();
+  }
+  else
+    anImageName = HYDROGUI_Tool::GenerateObjectName( module(), getName() );
   aPanel->setImageName( anImageName );
+
+  QString aSelectedName1, aSelectedName2;
+  if( myIsEdit && !myEditedObject.IsNull() )
+  {
+    if( myEditedObject->NbReferences() > 0 )
+    {
+      Handle(HYDROData_Image) anImage1 = myEditedObject->Reference( 0 );
+      if( !anImage1.IsNull() )
+        aSelectedName1 = anImage1->GetName();
+    }
+    if( myEditedObject->NbReferences() > 1 )
+    {
+      Handle(HYDROData_Image) anImage2 = myEditedObject->Reference( 1 );
+      if( !anImage2.IsNull() )
+        aSelectedName2 = anImage2->GetName();
+    }
+    aPanel->setSelectedImages( aSelectedName1, aSelectedName2 );
+
+    HYDROOperations_Factory* aFactory = HYDROOperations_Factory::Factory();
+    if( ImageComposer_Operator* anOperator = aFactory->Operator( myEditedObject ) )
+    {
+      QColor aColor;
+      anOperator->getArgs( aColor );
+      aPanel->setColor( aColor );
+    }
+  }
 }
 
 bool HYDROGUI_TwoImagesOp::processApply( int& theUpdateFlags,
@@ -82,12 +136,15 @@ bool HYDROGUI_TwoImagesOp::processApply( int& theUpdateFlags,
   if( !aPanel->getSelectedImages( aSelectedName1, aSelectedName2 ) )
     return false;
 
-  // 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() )
+  if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anImageName ) )
   {
-    theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anImageName );
-    return false;
+    // 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;
+    }
   }
 
   Handle(HYDROData_Image) anImage1 = Handle(HYDROData_Image)::DownCast(
@@ -98,26 +155,51 @@ bool HYDROGUI_TwoImagesOp::processApply( int& theUpdateFlags,
     return false;
 
   HYDROOperations_Factory* aFactory = HYDROOperations_Factory::Factory();
-  ImageComposer_Operator* anOperator = 0;
-  switch( myType )
+  Handle(HYDROData_Image) aResult = 0;
+  if( myIsEdit )
   {
-    case Fuse: anOperator = new ImageComposer_FuseOperator(); break;
-    case Cut: anOperator = new ImageComposer_CutOperator(); break;
-    default: break;
+    aResult = myEditedObject;
+    aResult->ClearReferences();
+  }
+  else
+  {
+    ImageComposer_Operator* anOperator = 0;
+    switch( myType )
+    {
+      case Fuse: anOperator = new ImageComposer_FuseOperator(); break;
+      case Cut: anOperator = new ImageComposer_CutOperator(); break;
+      default: break;
+    }
+
+    if( !anOperator )
+      return false;
+
+    aResult = aFactory->CreateImage( doc(), anOperator );
   }
 
-  if( !anOperator )
-    return false;
-
-  Handle(HYDROData_Image) aResult = aFactory->CreateImage( doc(), anOperator );
   if( aResult.IsNull() )
     return false;
 
   aResult->SetName( anImageName );
   aResult->AppendReference( anImage1 );
   aResult->AppendReference( anImage2 );
+
+  if( ImageComposer_Operator* anOperator = aFactory->Operator( aResult ) )
+  {
+    anOperator->setArgs( aPanel->getColor() );
+    aResult->SetArgs( anOperator->getBinArgs() );
+  }
+
   aFactory->UpdateImage( doc(), aResult );
 
-  theUpdateFlags = UF_Model | UF_Viewer;
+  if( !myIsEdit )
+  {
+    size_t aViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
+    module()->setObjectVisible( aViewId, anImage1, false );
+    module()->setObjectVisible( aViewId, anImage2, false );
+    module()->setObjectVisible( aViewId, aResult, true );
+  }
+
+  theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced;
   return true;
 }