src/HOMARDGUI/HOMARDGUI.cxx

   1 //  File   : HOMARDGUI.cxx
   2 //  Author : Paul RASCLE, EDF
   3 //  Module : HOMARD
   4
   5 using namespace std;
   6 #include "HOMARDGUI.h"
   7
   8 // SALOME Includes
   9 #include "Utils_ORB_INIT.hxx"
  10 #include "Utils_SINGLETON.hxx"
  11 #include "SALOME_LifeCycleCORBA.hxx"
  12
  13 #include "SUIT_ResourceMgr.h"
  14 #include "SUIT_MessageBox.h"
  15 #include "SUIT_Session.h"
  16 #include "SUIT_ViewWindow.h"
  17 #include "SUIT_ViewManager.h"
  18 #include <SUIT_Desktop.h>
  19
  20 #include "CAM_Module.h"
  21 #include "OB_Browser.h"
  22
  23 #include "SALOME_ListIO.hxx"
  24 #include "SALOME_ListIteratorOfListIO.hxx"
  25
  26 #include "SalomeApp_Application.h"
  27 #include "SalomeApp_DataModel.h"
  28 #include "SalomeApp_Study.h"
  29 #include "LightApp_SelectionMgr.h"
  30 #include "LightApp_Selection.h"
  31 #include <LightApp_Preferences.h>
  32 #include "SalomeApp_Module.h"
  33 #include "SALOMEconfig.h"
  34 #include <SALOME_LifeCycleCORBA.hxx>
  35
  36 #include <utilities.h>
  37
  38
  39 // QT Includes
  40 #include <QMenu>
  41 #include "MonCreateCase.h"
  42 #include "MonCreateIteration.h"
  43 #include "MonEditFile.h"
  44 #include "MonEditCase.h"
  45 #include "MonEditIteration.h"
  46 #include "MonEditHypothesis.h"
  47 #include "MonEditZone.h"
  48 #include "MonEditBoundaryDi.h"
  49 #include "HomardQtCommun.h"
  50
  51 // BOOST Includes
  52 #include <boost/shared_ptr.hpp>
  53
  54 //Pour le _CAST
  55 #include "SALOMEDS_Study.hxx"
  56 #include "HOMARDGUI_Utils.h"
  57
  58 static CORBA::ORB_var _orb;
  59
  60 //=======================================================================
  61 // function : HOMARDGUI()
  62 // purpose  : Constructor
  63 //=======================================================================
  64 HOMARDGUI::HOMARDGUI(const QString&) :
  65            SalomeApp_Module( "HOMARD" ), // default name
  66            LightApp_Module( "HOMARD" )
  67 {
  68 }
  69
  70
  71 //=======================================================================
  72 // function : ~HOMARDGUI()
  73 // purpose  : Destructor
  74 //=======================================================================
  75 HOMARDGUI::~HOMARDGUI()
  76 {
  77 }
  78
  79 //=======================================================================
  80 // function : InitHOMARDGen
  81 // launch HOMARD component and return a handle
  82 //=======================================================================
  83 HOMARD::HOMARD_Gen_var HOMARDGUI::InitHOMARDGen(SalomeApp_Application* app)
  84 {
  85   Engines::Component_var comp = app->lcc()->FindOrLoad_Component( "FactoryServer","HOMARD" );
  86   HOMARD::HOMARD_Gen_var clr = HOMARD::HOMARD_Gen::_narrow(comp);
  87   ASSERT(!CORBA::is_nil(clr));
  88   return clr;
  89 }
  90
  91 //=======================================================================
  92 // Module's initialization
  93 void HOMARDGUI::initialize( CAM_Application* app )
  94 //=======================================================================
  95 {
  96   SalomeApp_Module::initialize( app );
  97   InitHOMARDGen(dynamic_cast<SalomeApp_Application*>( app ));
  98   anId = 0;
  99   createActions();
 100   createMenus();
 101 }
 102
 103 //================================================
 104 // function : createHOMARDAction
 105 // create an item in status bar and Homard menu
 106 //================================================
 107 void HOMARDGUI::createHOMARDAction( const int id, const QString& po_id, const QString& icon_id, const int key, const bool toggle  )
 108 {
 109   QIcon icon;
 110   QWidget* parent = application()->desktop();
 111   SUIT_ResourceMgr* resMgr = application()->resourceMgr();
 112   QPixmap pix;
 113   if ( icon_id.length() )
 114      pix = resMgr->loadPixmap( "HOMARD", tr( icon_id .toLatin1().data()) );
 115   else
 116      pix = resMgr->loadPixmap( "HOMARD", tr( QString( "ICO_%1" ).arg( po_id ).toLatin1().data() ), false );
 117   if ( !pix.isNull() )
 118       icon = QIcon( pix );
 119
 120   QString tooltip    = tr( QString( "TOP_%1" ).arg( po_id ).toLatin1().data() ),
 121           menu       = tr( QString( "MEN_%1" ).arg( po_id ).toLatin1().data() ),
 122           status_bar = tr( QString( "STB_%1" ).arg( po_id ).toLatin1().data() );
 123
 124   createAction( id, tooltip, icon, menu, status_bar, key, parent, toggle, this, SLOT( OnGUIEvent() )  );
 125 }
 126
 127 //================================================
 128 // function : createAction
 129 // constructs Homard menu
 130 // calls createHOMARDAction for each item
 131 //================================================
 132 void HOMARDGUI::createActions(){
 133 //
 134   createHOMARDAction( 1101, "NEW_CASE",       "cas_calcule.png"  );
 135   createHOMARDAction( 1102, "NEW_ITERATION",  "iter_next.png"    );
 136   createHOMARDAction( 1111, "COMPUTE",        "mesh_compute.png" );
 137   createHOMARDAction( 1120, "EDIT_CASE",      "whatis.png"       );
 138   createHOMARDAction( 1121, "EDIT_ITERATION", "whatis.png"       );
 139   createHOMARDAction( 1122, "EDIT_HYPO",      "whatis.png"       );
 140   createHOMARDAction( 1123, "EDIT_ZONE",      "whatis.png"       );
 141   createHOMARDAction( 1124, "EDIT_BOUNDARY",  "whatis.png"       );
 142   createHOMARDAction( 1130, "EDIT_MESS_FILE", "texte.png"        );
 143 //
 144 }
 145
 146 //================================================
 147 // function : createPreferences
 148 // No preferences for Homard
 149 // Just a test
 150 //================================================
 151 void HOMARDGUI::createPreferences(){
 152    QString toto = tr( "PREF_GROUP_GENERAL" );
 153    int tabId  = addPreference( tr( "PREF_GROUP_GENERAL" ) );
 154    int genGroup = addPreference( tr( "PREF_TAB_SETTINGS" ), tabId );
 155    addPreference( tr( "PREF_TEST" ), genGroup, LightApp_Preferences::Color, "Homard", "shading_color" );
 156 }
 157
 158
 159 //================================================
 160 // function : createMenus
 161 //                Verifier l'avant dernier nombre passe en parametre
 162 //================================================
 163 void HOMARDGUI::createMenus(){
 164   int HOMARD_Id  = createMenu( tr( "HOMARD" ),  -1,  5, 10 );
 165   createMenu( 1101, HOMARD_Id, -1 ); //Create_Case
 166   createMenu( 1102, HOMARD_Id, -1 ); //Create_Iteration
 167   createMenu( 1111, HOMARD_Id, -1 ); //COMPUTE
 168   createMenu( separator(), HOMARD_Id,-1);
 169   createMenu( 1120, HOMARD_Id, -1 ); //Edit_Case
 170   createMenu( 1121, HOMARD_Id, -1 ); //Edit_Iteration
 171   createMenu( 1122, HOMARD_Id, -1 ); //Edit Hypo
 172   createMenu( 1123, HOMARD_Id, -1 ); //Edit_Zone
 173   createMenu( 1124, HOMARD_Id, -1 ); //Edit_Boundary
 174   createMenu( separator(), HOMARD_Id,-1);
 175   createMenu( 1130, HOMARD_Id, -1 ); //EditAsciiFile pour le fichier listeStd ou bilan
 176 }
 177
 178 //================================================
 179 void HOMARDGUI::OnGUIEvent()
 180 //================================================
 181 {
 182   MESSAGE("HOMARDGUI::OnGUIEvent()")
 183   setOrb();
 184   const QObject* obj = sender();
 185   if ( !obj || !obj->inherits( "QAction" ) )
 186        return;
 187   int id = actionId((QAction*)obj);
 188   bool ret;
 189   if ( id != -1 )
 190       ret = OnGUIEvent( id );
 191   MESSAGE("************** End of HOMARDGUI::OnGUIEvent()");
 192 }
 193
 194 //=======================================================================
 195 // Method OnGUIEvent pour Homard
 196 //=======================================================================
 197 bool HOMARDGUI::OnGUIEvent (int theCommandID)
 198 {
 199   MESSAGE("HOMARDGUI::OnGUIEvent(int)");
 200   SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( application() );
 201   if ( !app ) return false;
 202
 203   SalomeApp_Study* stud = dynamic_cast<SalomeApp_Study*> ( app->activeStudy() );
 204   if ( !stud ) {
 205     MESSAGE ( "FAILED to cast active study to SalomeApp_Study" );
 206     return false;
 207   }
 208
 209   SUIT_Desktop* parent = application()->desktop();
 210   HOMARD::HOMARD_Gen_var homardGen = HOMARDGUI::InitHOMARDGen(app);
 211
 212   if (!CORBA::is_nil(homardGen))
 213   {
 214     // Set current study
 215     SalomeApp_Study* aSAStudy =dynamic_cast<SalomeApp_Study*>(app->activeStudy());
 216     _PTR(Study) aStudy = aSAStudy->studyDS();
 217     SALOMEDS::Study_ptr aStudyDS;
 218     if (aStudy)
 219       aStudyDS = _CAST(Study,aStudy)->GetStudy();
 220       homardGen->SetCurrentStudy(aStudyDS);
 221    }
 222   getApp()->updateObjectBrowser();
 223
 224   SCRUTE(theCommandID);
 225   switch (theCommandID)
 226   {
 227     case 1101: // Creation d un Cas
 228     {
 229       MESSAGE("command " << theCommandID << " activated");
 230       MonCreateCase *aDlg = new MonCreateCase( parent, TRUE,
 231                             HOMARD::HOMARD_Gen::_duplicate(homardGen) ) ;
 232       aDlg->show();
 233       break;
 234     }
 235
 236     case 1102: // Creation d une Iteration
 237     {
 238       MESSAGE("command " << theCommandID << " activated");
 239       QString IterParentName=HOMARD_QT_COMMUN::SelectionArbreEtude(QString("IterationHomard"), 0);
 240       MESSAGE("IterParentName " << IterParentName.toStdString().c_str() << " choisi dans arbre");
 241       MonCreateIteration *IterDlg = new MonCreateIteration( parent, true,
 242                                      HOMARD::HOMARD_Gen::_duplicate(homardGen), IterParentName ) ;
 243       IterDlg->show();
 244       break;
 245     }
 246
 247     case 1111: // Compute une Iteration
 248     {
 249       MESSAGE("command " << theCommandID << " activated");
 250       QString monIter=HOMARD_QT_COMMUN::SelectionArbreEtude(QString("IterationHomard"), 1);
 251       if (monIter == QString("")) break;
 252       try
 253       {
 254         homardGen->Compute(monIter.toStdString().c_str(), 0);
 255       }
 256       catch( SALOME::SALOME_Exception& S_ex )
 257       {
 258         QMessageBox::information( 0, "Error",
 259                     QString(CORBA::string_dup(S_ex.details.text)),
 260                     QMessageBox::Ok + QMessageBox::Default );
 261         getApp()->updateObjectBrowser();
 262         return false;
 263       }
 264       break;
 265     }
 266
 267     case 1120: // Edition d un cas
 268     {
 269       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 270       _PTR(SObject) obj = chercheMonObjet();
 271       if ((obj) and (HOMARD_UTILS::isCase(obj)))
 272       {
 273           MonEditCase *aDlg = new MonEditCase(parent, true, HOMARD::HOMARD_Gen::_duplicate(homardGen), _ObjectName ) ;
 274           aDlg->show();
 275       }
 276     }
 277
 278     case 1121: // Edition d une iteration
 279     {
 280       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 281       _PTR(SObject) obj = chercheMonObjet();
 282       if ((obj) and (HOMARD_UTILS::isIter(obj)))
 283       {
 284           MonEditIteration *aDlg = new MonEditIteration(parent, true, HOMARD::HOMARD_Gen::_duplicate(homardGen), QString(""), _ObjectName ) ;
 285           aDlg->show();
 286       }
 287     }
 288
 289     case 1122: // Edition d une hypothese
 290     {
 291       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 292       _PTR(SObject) obj = chercheMonObjet();
 293       if ((obj) and (HOMARD_UTILS::isHypo(obj)))
 294       {
 295           MonEditHypothesis *aDlg = new MonEditHypothesis(0, true, HOMARD::HOMARD_Gen::_duplicate(homardGen),  _ObjectName, QString(""), QString("")) ;
 296           aDlg->show();
 297       }
 298     }
 299
 300     case 1123: // Edition d une zone
 301     {
 302       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 303       _PTR(SObject) obj = chercheMonObjet();
 304       if ((obj) and (HOMARD_UTILS::isZone(obj)))
 305       {
 306           MonEditZone *aDlg = new MonEditZone(0, TRUE, HOMARD::HOMARD_Gen::_duplicate(homardGen), QString(""), _ObjectName ) ;
 307           aDlg->show();
 308       }
 309     }
 310
 311     case 1124: // Edition d une frontiere
 312     {
 313       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 314       _PTR(SObject) obj = chercheMonObjet();
 315       if ((obj) and (HOMARD_UTILS::isBoundaryDi(obj)))
 316       {
 317           MESSAGE(".. Lancement de MonEditBoundaryDi" );
 318           MonEditBoundaryDi *aDlg = new MonEditBoundaryDi(0, TRUE, HOMARD::HOMARD_Gen::_duplicate(homardGen), QString(""), _ObjectName ) ;
 319           aDlg->show();
 320       }
 321     }
 322
 323     case 1130: // Edition du fichier mess
 324     {
 325       MESSAGE("command " << theCommandID << " activated avec objet " << _ObjectName.toStdString().c_str() );
 326       _PTR(SObject) obj = chercheMonObjet();
 327       if ((obj) and ((HOMARD_UTILS::isFileMess(obj) or HOMARD_UTILS::isFileSummary(obj))))
 328       {
 329           MonEditFile *aDlg = new MonEditFile( 0, true, HOMARD::HOMARD_Gen::_duplicate(homardGen), _ObjectName ) ;
 330           aDlg->show();
 331       }
 332      }
 333   }
 334   getApp()->updateObjectBrowser();
 335   return true;
 336 }
 337
 338 //=============================================================================
 339 /*!
 340  *
 341  */
 342 //=============================================================================
 343
 344 // Module's engine IOR
 345 //=============================================================================
 346 QString HOMARDGUI::engineIOR() const
 347 //=============================================================================
 348 {
 349   CORBA::String_var anIOR = getApp()->orb()->object_to_string( InitHOMARDGen(getApp()) );
 350   return QString( anIOR.in() );
 351 }
 352
 353 // Module's activation
 354 //=============================================================================
 355 bool HOMARDGUI::activateModule( SUIT_Study* theStudy )
 356 //=============================================================================
 357 {
 358   bool bOk = SalomeApp_Module::activateModule( theStudy );
 359
 360   setMenuShown( true );
 361   setToolShown( true );
 362
 363   return bOk;
 364 }
 365
 366 // Module's deactivation
 367 //=============================================================================
 368 bool HOMARDGUI::deactivateModule( SUIT_Study* theStudy )
 369 //=============================================================================
 370 {
 371   setMenuShown( false );
 372   setToolShown( false );
 373
 374   return SalomeApp_Module::deactivateModule( theStudy );
 375 }
 376
 377 // Default windows
 378 //=============================================================================
 379 void HOMARDGUI::windows( QMap<int, int>& theMap ) const
 380 //=============================================================================
 381 {
 382   theMap.clear();
 383   theMap.insert( SalomeApp_Application::WT_ObjectBrowser, Qt::LeftDockWidgetArea );
 384   theMap.insert( SalomeApp_Application::WT_PyConsole,     Qt::BottomDockWidgetArea );
 385 }
 386
 387 //=============================================================================
 388 void HOMARDGUI::setOrb()
 389 //=============================================================================
 390 {
 391   try
 392   {
 393      ORB_INIT &init = *SINGLETON_<ORB_INIT>::Instance();
 394      ASSERT(SINGLETON_<ORB_INIT>::IsAlreadyExisting());
 395      _orb = init( 0 , 0 );
 396   }
 397   catch (...)
 398   {
 399      INFOS("internal error : orb not found");
 400      _orb = 0;
 401   }
 402             ASSERT(! CORBA::is_nil(_orb));
 403 }
 404 //========================================
 405 _PTR(SObject) HOMARDGUI::chercheMonObjet()
 406 //========================================
 407 {
 408
 409     SALOMEDSClient_SObject* aSO = NULL;
 410     _PTR(SObject) obj;
 411     SALOME_ListIO lst;
 412     getApp()->selectionMgr()->selectedObjects( lst );
 413     if (  lst.Extent() == 1 )
 414     {
 415         Handle(SALOME_InteractiveObject) io = lst.First();
 416         SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( application()->activeStudy() );
 417         _PTR(Study) study = appStudy->studyDS();
 418         _PTR(SObject) obj = study->FindObjectID( io->getEntry() );
 419         _ObjectName = QString( obj->GetName().c_str() );
 420         return obj;
 421      }
 422      else
 423          return _PTR(SObject)(aSO);
 424 }
 425 //=============================================================================
 426 void HOMARDGUI::contextMenuPopup( const QString& client, QMenu* menu, QString& title )
 427 //=============================================================================
 428 {
 429   MESSAGE("Debut de contextMenuPopup");
 430   _PTR(SObject) obj = chercheMonObjet();
 431   if ( obj )
 432   {
 433     title = QString( obj->GetName().c_str() );
 434     _ObjectName = title;
 435     SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
 436 //
 437     if ( HOMARD_UTILS::isBoundaryDi(obj) )
 438     {
 439       QPixmap pix = resMgr->loadPixmap( "HOMARD", "whatis.png" );
 440       menu->addAction(QIcon(pix), "Edit boundary", this,SLOT(EditBoDi()));
 441     }
 442     else if ( HOMARD_UTILS::isZone(obj) )
 443     {
 444       QPixmap pix = resMgr->loadPixmap( "HOMARD", "whatis.png" );
 445       menu->addAction(QIcon(pix), "Edit zone", this,SLOT(EditZone()));
 446     }
 447     else if ( HOMARD_UTILS::isHypo(obj) )
 448     {
 449       QPixmap pix = resMgr->loadPixmap( "HOMARD", "whatis.png" );
 450       menu->addAction(QIcon(pix), "Edit hypothesis", this,SLOT(EditHypo()));
 451     }
 452     else if ( HOMARD_UTILS::isIter(obj) )
 453     {
 454       QPixmap pix = resMgr->loadPixmap( "HOMARD", "iter_next.png" );
 455       menu->addAction(QIcon(pix), "Next iteration", this,SLOT(NextIter()));
 456       QPixmap pix1 = resMgr->loadPixmap( "HOMARD", "whatis.png" );
 457       menu->addAction(QIcon(pix1), "Edit iteration", this,SLOT(EditIter()));
 458       QPixmap pix2 = resMgr->loadPixmap( "HOMARD", "mesh_compute.png" );
 459       menu->addAction(QIcon(pix2), "Compute", this,SLOT(LanceCalcul()));
 460     }
 461     else if ( HOMARD_UTILS::isCase(obj) )
 462     {
 463       QPixmap pix = resMgr->loadPixmap( "HOMARD", "whatis.png" );
 464       menu->addAction(QIcon(pix), "Edit case", this,SLOT(EditCase()));
 465     }
 466     else if ( HOMARD_UTILS::isFileMess(obj) or HOMARD_UTILS::isFileSummary(obj) )
 467     {
 468       QPixmap pix = resMgr->loadPixmap( "HOMARD", "texte.png" );
 469       menu->addAction(QIcon(pix), "Show file", this,SLOT(EditAsciiFile()));
 470     }
 471   }
 472 }
 473
 474 void HOMARDGUI::NextIter()
 475 {
 476   this->OnGUIEvent(1102);
 477 }
 478
 479 void HOMARDGUI::LanceCalcul()
 480 {
 481   this->OnGUIEvent(1111);
 482 }
 483
 484 void HOMARDGUI::EditCase()
 485 {
 486   this->OnGUIEvent(1120);
 487 }
 488
 489 void HOMARDGUI::EditIter()
 490 {
 491   this->OnGUIEvent(1121);
 492 }
 493
 494 void HOMARDGUI::EditHypo()
 495 {
 496   this->OnGUIEvent(1122);
 497 }
 498
 499 void HOMARDGUI::EditZone()
 500 {
 501   this->OnGUIEvent(1123);
 502 }
 503
 504 void HOMARDGUI::EditBoDi()
 505 {
 506   this->OnGUIEvent(1124);
 507 }
 508
 509 void HOMARDGUI::EditAsciiFile()
 510 {
 511   this->OnGUIEvent(1130);
 512 }
 513 //
 514 //=============================================================================
 515 // Export the module
 516 //=============================================================================
 517 extern "C" {
 518   Standard_EXPORT CAM_Module* createModule()
 519   {
 520     return new HOMARDGUI("");
 521   }
 522 }
 523