src/HYDROGUI/HYDROGUI_GeoreferencementOp.cxx

   1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "HYDROGUI_GeoreferencementOp.h"
  24
  25 #include "HYDROGUI_GeoreferencementDlg.h"
  26 #include "HYDROGUI_DataModel.h"
  27 #include "HYDROGUI_Module.h"
  28 #include "HYDROGUI_Tool.h"
  29 #include "HYDROGUI_UpdateFlags.h"
  30
  31 #include <HYDROData_Profile.h>
  32 #include <HYDROData_Iterator.h>
  33 #include <HYDROData_Entity.h>
  34
  35 #include <LightApp_Application.h>
  36 #include <LightApp_UpdateFlags.h>
  37
  38 #include <SUIT_Desktop.h>
  39 #include <SUIT_MessageBox.h>
  40 #include <SUIT_ViewWindow.h>
  41
  42 #include <OCCViewer_ViewManager.h>
  43
  44 #include <gp_XY.hxx>
  45
  46 HYDROGUI_GeoreferencementOp::HYDROGUI_GeoreferencementOp( HYDROGUI_Module* theModule, const int theInitialMode )
  47 : HYDROGUI_Operation( theModule ),
  48   myInitialMode( theInitialMode )
  49 {
  50   setName( tr( "PROFILES_GEOREFERENCEMENT" ) );
  51 }
  52
  53 HYDROGUI_GeoreferencementOp::~HYDROGUI_GeoreferencementOp()
  54 {
  55 }
  56
  57 void HYDROGUI_GeoreferencementOp::startOperation()
  58 {
  59   HYDROGUI_Operation::startOperation();
  60
  61   HYDROGUI_GeoreferencementDlg* aPanel =
  62     ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
  63   if ( !aPanel ) {
  64     return;
  65   }
  66
  67   aPanel->reset();
  68
  69   if ( myInitialMode == All ) {
  70     int anAllMode = HYDROGUI_GeoreferencementDlg::AllProfiles;
  71     aPanel->setMode( anAllMode );
  72     onModeActivated( anAllMode );
  73   } else if ( myInitialMode == Selected ) {
  74     int aSelectionMode = HYDROGUI_GeoreferencementDlg::SelectedProfiles;
  75     aPanel->setMode( aSelectionMode );
  76     onModeActivated( aSelectionMode );
  77   }
  78
  79   LightApp_Application* anApp = module()->getApp();
  80   OCCViewer_ViewManager* aViewManager =
  81     dynamic_cast<OCCViewer_ViewManager*>( anApp->getViewManager( OCCViewer_Viewer::Type(), false ) );
  82   if ( aViewManager ) {
  83     connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
  84              aPanel, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
  85   }
  86
  87   connect( anApp->desktop(), SIGNAL( windowActivated( SUIT_ViewWindow* ) ),
  88            this, SLOT( onWindowActivated( SUIT_ViewWindow* ) ) );
  89 }
  90
  91 void HYDROGUI_GeoreferencementOp::abortOperation()
  92 {
  93   HYDROGUI_Operation::abortOperation();
  94 }
  95
  96 void HYDROGUI_GeoreferencementOp::commitOperation()
  97 {
  98   HYDROGUI_Operation::commitOperation();
  99 }
 100
 101 HYDROGUI_InputPanel* HYDROGUI_GeoreferencementOp::createInputPanel() const
 102 {
 103   HYDROGUI_InputPanel* aPanel = new HYDROGUI_GeoreferencementDlg( module(), getName() );
 104   connect( aPanel, SIGNAL( modeActivated( int ) ), SLOT( onModeActivated( int ) ) );
 105
 106   return aPanel;
 107 }
 108
 109 bool HYDROGUI_GeoreferencementOp::processApply( int& theUpdateFlags,
 110                                                 QString& theErrorMsg )
 111 {
 112   theUpdateFlags = UF_Model;
 113
 114   return store( theErrorMsg );
 115 }
 116
 117 void HYDROGUI_GeoreferencementOp::onModeActivated( const int theActualMode )
 118 {
 119   QString anErrorMsg;
 120
 121   // Get the panel
 122   HYDROGUI_GeoreferencementDlg* aPanel =
 123     ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
 124   if ( !aPanel ) {
 125     return;
 126   }
 127
 128   // Store the dialog data to the data model
 129   if ( !store( anErrorMsg ) ) {
 130     aPanel->setMode( theActualMode == HYDROGUI_GeoreferencementDlg::AllProfiles ?
 131                      HYDROGUI_GeoreferencementDlg::SelectedProfiles :
 132                      HYDROGUI_GeoreferencementDlg::AllProfiles );
 133
 134     if ( anErrorMsg.isEmpty() ) {
 135       anErrorMsg = tr( "INSUFFICIENT_INPUT_DATA" );
 136     }
 137     SUIT_MessageBox::critical( module()->getApp()->desktop(),
 138                                tr( "ERROR" ),
 139                                anErrorMsg );
 140     return;
 141   }
 142
 143   // Get georeferencement data from the data model
 144   HYDROGUI_GeoreferencementDlg::ProfilesGeoDataMap aDataMap;
 145   HYDROData_SequenceOfObjects aSeqOfProfiles;
 146
 147   if( theActualMode == HYDROGUI_GeoreferencementDlg::AllProfiles ) {
 148     HYDROData_Iterator aProfilesIter( doc(), KIND_PROFILE );
 149     while ( aProfilesIter.More() ) {
 150       aSeqOfProfiles.Append( aProfilesIter.Current() );
 151       aProfilesIter.Next();
 152     }
 153   } else if ( theActualMode == HYDROGUI_GeoreferencementDlg::SelectedProfiles ) {
 154     aSeqOfProfiles = HYDROGUI_Tool::GetSelectedObjects( module() );
 155   }
 156
 157   HYDROData_SequenceOfObjects::Iterator anIter( aSeqOfProfiles );
 158   for ( ; anIter.More(); anIter.Next() ) {
 159     Handle(HYDROData_Profile) aProfile =
 160         Handle(HYDROData_Profile)::DownCast( anIter.Value() );
 161     if ( aProfile.IsNull() ) {
 162       continue;
 163     }
 164
 165     HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData;
 166
 167     gp_XY aFirstPoint, aLastPoint;
 168     if ( aProfile->GetFirstPoint( aFirstPoint ) && aProfile->GetLastPoint( aLastPoint ) ) {
 169       aGeoData =
 170         HYDROGUI_GeoreferencementDlg::ProfileGeoData( aFirstPoint.X(), aFirstPoint.Y(),
 171                                                       aLastPoint.X(), aLastPoint.Y() );
 172     }
 173
 174     aDataMap.insert( aProfile->GetName(), aGeoData );
 175   }
 176
 177   // Set the collected data to the dialog
 178   aPanel->setData( aDataMap );
 179 }
 180
 181 void HYDROGUI_GeoreferencementOp::onWindowActivated( SUIT_ViewWindow* theViewWindow )
 182 {
 183   if ( !theViewWindow ) {
 184     return;
 185   }
 186
 187   OCCViewer_ViewManager* aViewManager =
 188     dynamic_cast<OCCViewer_ViewManager*>( theViewWindow->getViewManager() );
 189
 190   if ( aViewManager ) {
 191     HYDROGUI_GeoreferencementDlg* aPanel =
 192       ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
 193     if ( aPanel ) {
 194       connect( aViewManager, SIGNAL( mousePress( SUIT_ViewWindow*, QMouseEvent* ) ),
 195                aPanel, SLOT( onMousePress( SUIT_ViewWindow*, QMouseEvent* ) ) );
 196     }
 197   }
 198 }
 199
 200 bool HYDROGUI_GeoreferencementOp::store( QString& theErrorMsg )
 201 {
 202   // Clear the error string
 203   theErrorMsg.clear();
 204
 205   // Get the panel
 206   HYDROGUI_GeoreferencementDlg* aPanel =
 207     ::qobject_cast<HYDROGUI_GeoreferencementDlg*>( inputPanel() );
 208   if ( !aPanel ) {
 209     return false;
 210   }
 211
 212   // Get georeferencement data from the panel
 213   HYDROGUI_GeoreferencementDlg::ProfilesGeoDataMap aGeoDataMap;
 214   aPanel->getData( aGeoDataMap );
 215
 216   if ( aGeoDataMap.empty() ) {
 217     return true;
 218   }
 219
 220   // Check the data validity
 221   foreach ( const QString& aProfileName, aGeoDataMap.keys() ) {
 222     HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData =
 223       aGeoDataMap.value( aProfileName );
 224     if ( aGeoData.isIncomplete ) {
 225       theErrorMsg = tr( "INCOMPLETE_DATA" ).arg( aProfileName );
 226       return false;
 227     }
 228   }
 229
 230   // Store the data in the data model
 231   foreach ( const QString& aProfileName, aGeoDataMap.keys() ) {
 232     Handle(HYDROData_Profile) aProfile =
 233       Handle(HYDROData_Profile)::DownCast(
 234         HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) );
 235     if ( !aProfile.IsNull() ) {
 236       HYDROGUI_GeoreferencementDlg::ProfileGeoData aGeoData =
 237         aGeoDataMap.value( aProfileName );
 238       if ( !aGeoData.isEmpty ) {
 239         aProfile->SetFirstPoint( gp_XY( aGeoData.Xg, aGeoData.Yg ) );
 240         aProfile->SetLastPoint( gp_XY( aGeoData.Xd, aGeoData.Yd ) );
 241       } else {
 242         aProfile->Invalidate();
 243       }
 244     }
 245   }
 246
 247   return true;
 248 }