src/salome_hydro/coupling1d2d/eficas/appli.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 #  Copyright (C) 2012-2013 EDF
   4 #
   5 #  This file is part of SALOME HYDRO module.
   6 #
   7 #  SALOME HYDRO module is free software: you can redistribute it and/or modify
   8 #  it under the terms of the GNU General Public License as published by
   9 #  the Free Software Foundation, either version 3 of the License, or
  10 #  (at your option) any later version.
  11 #
  12 #  SALOME HYDRO module is distributed in the hope that it will be useful,
  13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #  GNU General Public License for more details.
  16 #
  17 #  You should have received a copy of the GNU General Public License
  18 #  along with SALOME HYDRO module.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 import os
  21 import sys
  22 import re
  23
  24 from PyQt5.QtGui import *
  25 from PyQt5.QtWidgets import *
  26
  27 import salome
  28 import SalomePyQt
  29 sgPyQt = SalomePyQt.SalomePyQt()
  30
  31 from salome.kernel.logger import Logger
  32 from salome.kernel import termcolor
  33 logger = Logger("salome.hydro.coupling1d2d.eficas.appli",
  34                 color = termcolor.GREEN_FG)
  35
  36 import eficasSalome
  37
  38 from salome.hydro.study import HydroStudyEditor
  39
  40 class SalomeEntry:
  41
  42     enable_salome_selection = True
  43     help_message = u"Une entrée de l'arbre d'étude de Salome est attendue"
  44
  45     def __init__(self, entryStr):
  46         self._entry = entryStr
  47
  48     @staticmethod
  49     def __convert__(entryStr):
  50         return SalomeEntry(entryStr)
  51
  52     @staticmethod
  53     def get_selected_value(selected_entry, study_editor):
  54         sobj = study_editor.study.FindObjectID(selected_entry)
  55         name = sobj.GetName()
  56         return "%s (%s)" % (name, selected_entry)
  57
  58 class EficasForCoupling1D2DAppli(eficasSalome.MyEficas):
  59     """
  60     This class launches Eficas and adds entries for the created files in
  61     HYDRO component in the study tree. The messages in this class are in
  62     french because they are displayed in Eficas interface.
  63
  64     :type  fichier: string
  65     :param fichier: path of an Eficas file to open
  66
  67     """
  68     def __init__(self, fichier = None, version = None):
  69         self.ed = HydroStudyEditor()
  70         self.codedir = os.path.dirname(__file__)
  71         sys.path[:0] = [self.codedir]
  72         eficasSalome.MyEficas.__init__(self, sgPyQt.getDesktop(), "coupling1d2d",
  73                                        fichier, version = version)
  74         sgPyQt.createView("Eficas Coupling 1D/2D", self)
  75
  76     def selectGroupFromSalome(self, kwType = None, editor=None):
  77         """
  78         Select an entry from Salome object browser
  79         """
  80         nbEntries = salome.sg.SelectedCount()
  81         if nbEntries < 1:
  82             msg = u"Veuillez sélectionner une entrée de l'arbre d'étude de Salome"
  83             QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(msg))
  84             return [], msg
  85         elif nbEntries > 1:
  86             msg = u"Une seule entrée doit être sélectionnée dans l'arbre d'étude de Salome"
  87             QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(msg))
  88             return [], msg
  89         else:
  90             try:
  91                 value = kwType.get_selected_value(salome.sg.getSelected(0), self.ed.editor)
  92                 msg = u"L'entrée de l'arbre d'étude de Salome a été sélectionnée"
  93                 return [value], msg
  94             except Exception, e:
  95                 QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(unicode(e)))
  96                 return [], unicode(e)
  97
  98     def addJdcInSalome(self, jdcPath):
  99         """
 100         Add the newly created file in Salome study
 101         """
 102         try:
 103             self.ed.find_or_create_coupling1d2d_case(jdcPath)
 104         except Exception, exc:
 105             msgError = "Can't add file to Salome study tree"
 106             logger.exception(msgError)
 107             QMessageBox.warning(self, self.tr("Warning"),
 108                                 self.tr("%s. Reason:\n%s\n\nSee logs for more details." % (msgError, exc)))
 109         salome.sg.updateObjBrowser(0)
 110
 111     def closeEvent(self, event):
 112         while self.codedir in sys.path:
 113             sys.path.remove(self.codedir)
 114         eficasSalome.MyEficas.closeEvent(self, event)