Salome HOME
Minor changes.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_GeoreferencementOp.cxx
index 58e29d0eeca7d28803639ad9cc013cd6e14276c2..d7b389e6bd0f3fee6e3287ec754888ee5302f01d 100644 (file)
 #include <LightApp_Application.h>
 #include <LightApp_UpdateFlags.h>
 
+#include <SUIT_Desktop.h>
+#include <SUIT_MessageBox.h>
+#include <SUIT_ViewWindow.h>
+
+#include <OCCViewer_ViewManager.h>
+
 #include <gp_XY.hxx>
 
 HYDROGUI_GeoreferencementOp::HYDROGUI_GeoreferencementOp( HYDROGUI_Module* theModule, const int theInitialMode )
@@ -61,23 +67,41 @@ void HYDROGUI_GeoreferencementOp::startOperation()
   aPanel->reset();
 
   if ( myInitialMode == All ) {
-    int anAllMode = HYDROGUI_GeoreferencementDlg::AllProfiles;
-    aPanel->setMode( anAllMode );
-    onModeActivated( anAllMode );
+    onModeActivated( HYDROGUI_GeoreferencementDlg::AllProfiles );
   } else if ( myInitialMode == Selected ) {
-    int aSelectionMode = HYDROGUI_GeoreferencementDlg::SelectedProfiles;
-    aPanel->setMode( aSelectionMode );
-    onModeActivated( aSelectionMode );
+    onModeActivated( HYDROGUI_GeoreferencementDlg::SelectedProfiles );
+  }
+
+  LightApp_Application* anApp = module()->getApp();
+  OCCViewer_ViewManager* aViewManager =
+    dynamic_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), false ) );
+  if ( aViewManager ) {
+    connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
+             aPanel, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ), 
+             Qt::UniqueConnection );
   }
+
+  connect( anApp->desktop(), SIGNAL( windowActivated( SUIT_ViewWindow* ) ),
+           this, SLOT( onWindowActivated( SUIT_ViewWindow* ) ), Qt::UniqueConnection );
 }
 
 void HYDROGUI_GeoreferencementOp::abortOperation()
 {
+  LightApp_Application* anApp = module()->getApp();
+  if ( anApp && anApp->desktop() ) {
+    anApp->desktop()->disconnect( this );
+  }
+
   HYDROGUI_Operation::abortOperation();
 }
 
 void HYDROGUI_GeoreferencementOp::commitOperation()
 {
+  LightApp_Application* anApp = module()->getApp();
+  if ( anApp && anApp->desktop() ) {
+    anApp->desktop()->disconnect( this );
+  }
+
   HYDROGUI_Operation::commitOperation();
 }
 
@@ -85,6 +109,7 @@ HYDROGUI_InputPanel* HYDROGUI_GeoreferencementOp::createInputPanel() const
 {
   HYDROGUI_InputPanel* aPanel = new HYDROGUI_GeoreferencementDlg( module(), getName() );
   connect( aPanel, SIGNAL( modeActivated( int ) ), SLOT( onModeActivated( int ) ) );
+  connect( aPanel, SIGNAL( updateSelection() ), SLOT( onUpdateSelection() ) );
 
   return aPanel;
 }
@@ -92,6 +117,107 @@ HYDROGUI_InputPanel* HYDROGUI_GeoreferencementOp::createInputPanel() const
 bool HYDROGUI_GeoreferencementOp::processApply( int& theUpdateFlags,
                                                 QString& theErrorMsg )
 {
+  theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer | UF_VTK_Forced | UF_VTK_Init;
+
+  return store( theErrorMsg );
+}
+
+void HYDROGUI_GeoreferencementOp::onModeActivated( const int theActualMode )
+{
+  QString anErrorMsg;
+
+  // Get the panel
+  HYDROGUI_GeoreferencementDlg* aPanel = 
+    ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
+  if ( !aPanel ) {
+    return;
+  }
+
+  aPanel->setMode( theActualMode == HYDROGUI_GeoreferencementDlg::AllProfiles ?
+                   HYDROGUI_GeoreferencementDlg::SelectedProfiles :
+                   HYDROGUI_GeoreferencementDlg::AllProfiles);
+
+  // Store the dialog data to the data model
+  bool isToStore = false;
+  // Check if modifications exists
+  if ( aPanel->isModified() ) {
+    // Show confirmation dialog
+    SUIT_MessageBox::StandardButtons aButtons = 
+      SUIT_MessageBox::Yes | SUIT_MessageBox::No;
+
+    isToStore = SUIT_MessageBox::question( module()->getApp()->desktop(),
+                                           tr( "CONFIRMATION" ),
+                                           tr( "CONFIRM_STORE" ),
+                                           aButtons, 
+                                           SUIT_MessageBox::Yes) == SUIT_MessageBox::Yes;
+  }
+  // Store modifications if needed
+  if ( isToStore )
+  {
+    startDocOperation();
+    if ( !store( anErrorMsg ) )
+    {
+      abortDocOperation();
+
+      anErrorMsg.append( "\n" + tr( "INPUT_VALID_DATA" ) );
+      SUIT_MessageBox::critical( module()->getApp()->desktop(),
+                                 tr( "INSUFFICIENT_INPUT_DATA" ),
+                                 anErrorMsg );
+      return;
+    }
+    else
+    {
+      commitDocOperation();
+    }
+  }
+
+  aPanel->setMode( theActualMode );
+
+  // Get the list of profiles
+  HYDROData_SequenceOfObjects aSeqOfProfiles;
+  
+  if( theActualMode == HYDROGUI_GeoreferencementDlg::AllProfiles ) {
+    HYDROData_Iterator aProfilesIter( doc(), KIND_PROFILE );
+    while ( aProfilesIter.More() ) {
+      aSeqOfProfiles.Append( aProfilesIter.Current() );
+      aProfilesIter.Next();
+    }
+  } else if ( theActualMode == HYDROGUI_GeoreferencementDlg::SelectedProfiles ) {
+    aSeqOfProfiles = HYDROGUI_Tool::GetSelectedObjects( module() );
+  }
+
+  // Set the profiles to the panel
+  setPanelData( aSeqOfProfiles );
+}
+
+void HYDROGUI_GeoreferencementOp::onWindowActivated( SUIT_ViewWindow* theViewWindow )
+{
+  if ( !theViewWindow ) {
+    return;
+  }
+
+  OCCViewer_ViewManager* aViewManager =
+    dynamic_cast<OCCViewer_ViewManager*>( theViewWindow->getViewManager() );
+
+  if ( aViewManager ) {
+    HYDROGUI_GeoreferencementDlg* aPanel = 
+      ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
+    if ( aPanel ) {
+      connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
+               aPanel, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
+               Qt::UniqueConnection);
+    }
+  }
+}
+
+bool HYDROGUI_GeoreferencementOp::store( QString& theErrorMsg )
+{
+  // Clear the error string
+  theErrorMsg.clear();
+  
+  HYDROGUI_Module* aModule = module();
+
+  // Get the panel
   HYDROGUI_GeoreferencementDlg* aPanel = 
     ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
   if ( !aPanel ) {
@@ -99,52 +225,70 @@ bool HYDROGUI_GeoreferencementOp::processApply( int& theUpdateFlags,
   }
 
   // Get georeferencement data from the panel
-  HYDROGUI_GeoreferencementDlg::ProfilesGeoDataMap aGeoDataMap;
-  aPanel->getData( aGeoDataMap );
+  HYDROGUI_GeoreferencementDlg::ProfilesGeoDataList aGeoDataList;
+  aPanel->getData( aGeoDataList );
+  if ( aGeoDataList.empty() ) {
+    return true;
+  }
+
+  // Check the data validity
+  foreach ( const HYDROGUI_GeoreferencementDlg::ProfileGeoData& aGeoData, aGeoDataList ) {
+    if ( aGeoData.isIncomplete ) {
+      theErrorMsg = tr( "INCOMPLETE_DATA" ).arg( aGeoData.Name );
+      return false;
+    }
+  }
 
-  // Set the data
-  foreach ( const QString& aProfileName, aGeoDataMap.keys() ) {
+  // Store the data in the data model
+  foreach ( const HYDROGUI_GeoreferencementDlg::ProfileGeoData& aGeoData, aGeoDataList ) {
     Handle(HYDROData_Profile) aProfile = 
       Handle(HYDROData_Profile)::DownCast(
-        HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) );
+        HYDROGUI_Tool::FindObjectByName( module(), aGeoData.Name, KIND_PROFILE ) );
     if ( !aProfile.IsNull() ) {
-      HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData = 
-        aGeoDataMap.value( aProfileName );
-      if ( aGeoData.isValid ) {
-        aProfile->SetFirstPoint( gp_XY( aGeoData.Xg, aGeoData.Yg ) );
-        aProfile->SetLastPoint( gp_XY( aGeoData.Xd, aGeoData.Yd ) );
+      if ( !aGeoData.isEmpty ) {
+        if ( !aProfile->IsValid() ) // Show the profile after it became valid
+          aModule->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( aModule ), aProfile, true );
+
+        aProfile->SetLeftPoint( gp_XY( aGeoData.Xg, aGeoData.Yg ) );
+        aProfile->SetRightPoint( gp_XY( aGeoData.Xd, aGeoData.Yd ) );
       } else {
         aProfile->Invalidate();
+        aModule->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( aModule ), aProfile, false );
       }
+
+      aProfile->Update();
     }
   }
-  
-  theUpdateFlags = UF_Model;
+
+  aModule->update( UF_Model | UF_OCCViewer | UF_OCC_Forced );
+
   return true;
 }
 
-void HYDROGUI_GeoreferencementOp::onModeActivated( const int theActualMode )
+void HYDROGUI_GeoreferencementOp::onUpdateSelection()
+{
+  // Get the selected profiles
+  HYDROData_SequenceOfObjects aSeqOfProfiles = 
+    HYDROGUI_Tool::GetSelectedObjects( module() );
+
+  // Set them to the dialog
+  setPanelData( aSeqOfProfiles );
+}
+
+void HYDROGUI_GeoreferencementOp::setPanelData( 
+  const HYDROData_SequenceOfObjects& theProfiles )
 {
+  // Get the panel
   HYDROGUI_GeoreferencementDlg* aPanel = 
     ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
   if ( !aPanel ) {
     return;
   }
 
-  HYDROGUI_GeoreferencementDlg::ProfilesGeoDataMap aDataMap;
-  HYDROData_SequenceOfObjects aSeqOfProfiles;
-  
-  if( theActualMode == HYDROGUI_GeoreferencementDlg::AllProfiles ) {
-    HYDROData_Iterator aProfilesIter( doc(), KIND_PROFILE );
-    while ( aProfilesIter.More() ) {
-      aSeqOfProfiles.Append( aProfilesIter.Current() );
-      aProfilesIter.Next();
-    }
-  } else if ( theActualMode == HYDROGUI_GeoreferencementDlg::SelectedProfiles ) {
-    aSeqOfProfiles = HYDROGUI_Tool::GetSelectedObjects( module() );
-  }
+  // Get georeferencement data from the data model
+  HYDROGUI_GeoreferencementDlg::ProfilesGeoDataList aData;
 
-  HYDROData_SequenceOfObjects::Iterator anIter( aSeqOfProfiles );
+  HYDROData_SequenceOfObjects::Iterator anIter( theProfiles );
   for ( ; anIter.More(); anIter.Next() ) {
     Handle(HYDROData_Profile) aProfile =
         Handle(HYDROData_Profile)::DownCast( anIter.Value() );
@@ -152,20 +296,19 @@ void HYDROGUI_GeoreferencementOp::onModeActivated( const int theActualMode )
       continue;
     }
 
-    HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData;
+    HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData( aProfile->GetName() );
 
     gp_XY aFirstPoint, aLastPoint;
-    if ( aProfile->GetFirstPoint( aFirstPoint ) && aProfile->GetLastPoint( aLastPoint ) ) {
+    if ( aProfile->GetLeftPoint( aFirstPoint ) && aProfile->GetRightPoint( aLastPoint ) ) {
       aGeoData = 
-        HYDROGUI_GeoreferencementDlg::ProfileGeoData( aFirstPoint.X(), aFirstPoint.Y(),
+        HYDROGUI_GeoreferencementDlg::ProfileGeoData( aGeoData.Name, 
+                                                      aFirstPoint.X(), aFirstPoint.Y(),
                                                       aLastPoint.X(), aLastPoint.Y() );
     }
    
-    aDataMap.insert( aProfile->GetName(), aGeoData );
+    aData.append( aGeoData );
   }
 
-  aPanel->setData( aDataMap );
-}
-
-
-
+  // Set the collected data to the dialog
+  aPanel->setData( aData );
+}
\ No newline at end of file