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 PyQt4.QtGui import QMessageBox
  25
  26 import salome
  27 import SalomePyQt
  28 sgPyQt = SalomePyQt.SalomePyQt()
  29
  30 from salome.kernel.logger import Logger
  31 from salome.kernel import termcolor
  32 logger = Logger("salome.hydro.coupling1d2d.eficas.appli",
  33                 color = termcolor.GREEN_FG)
  34
  35 import eficasSalome
  36
  37 from salome.hydro.study import HydroStudyEditor
  38
  39 class SalomeEntry:
  40
  41     enable_salome_selection = True
  42     help_message = u"Une entrée de l'arbre d'étude de Salome est attendue"
  43
  44     def __init__(self, entryStr):
  45         self._entry = entryStr
  46
  47     @staticmethod
  48     def __convert__(entryStr):
  49         return SalomeEntry(entryStr)
  50
  51     @staticmethod
  52     def get_selected_value(selected_entry, study_editor):
  53         sobj = study_editor.study.FindObjectID(selected_entry)
  54         name = sobj.GetName()
  55         return "%s (%s)" % (name, selected_entry)
  56
  57 class EficasForCoupling1D2DAppli(eficasSalome.MyEficas):
  58     """
  59     This class launches Eficas and adds entries for the created files in
  60     HYDRO component in the study tree. The messages in this class are in
  61     french because they are displayed in Eficas interface.
  62
  63     :type  fichier: string
  64     :param fichier: path of an Eficas file to open
  65
  66     """
  67     def __init__(self, fichier = None, version = None):
  68         self.ed = HydroStudyEditor()
  69         self.codedir = os.path.dirname(__file__)
  70         sys.path[:0] = [self.codedir]
  71         eficasSalome.MyEficas.__init__(self, sgPyQt.getDesktop(), "coupling1d2d",
  72                                        fichier, version = version)
  73         sgPyQt.createView("Eficas Coupling 1D/2D", self)
  74
  75     def selectGroupFromSalome(self, kwType = None, editor=None):
  76         """
  77         Select an entry from Salome object browser
  78         """
  79         nbEntries = salome.sg.SelectedCount()
  80         if nbEntries < 1:
  81             msg = u"Veuillez sélectionner une entrée de l'arbre d'étude de Salome"
  82             QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(msg))
  83             return [], msg
  84         elif nbEntries > 1:
  85             msg = u"Une seule entrée doit être sélectionnée dans l'arbre d'étude de Salome"
  86             QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(msg))
  87             return [], msg
  88         else:
  89             try:
  90                 value = kwType.get_selected_value(salome.sg.getSelected(0), self.ed.editor)
  91                 msg = u"L'entrée de l'arbre d'étude de Salome a été sélectionnée"
  92                 return [value], msg
  93             except Exception, e:
  94                 QMessageBox.information(self, self.tr(u"Sélection depuis Salome"), self.tr(unicode(e)))
  95                 return [], unicode(e)
  96
  97     def addJdcInSalome(self, jdcPath):
  98         """
  99         Add the newly created file in Salome study
 100         """
 101         try:
 102             self.ed.find_or_create_coupling1d2d_case(jdcPath)
 103         except Exception, exc:
 104             msgError = "Can't add file to Salome study tree"
 105             logger.exception(msgError)
 106             QMessageBox.warning(self, self.tr("Warning"),
 107                                 self.tr("%s. Reason:\n%s\n\nSee logs for more details." % (msgError, exc)))
 108         salome.sg.updateObjBrowser(0)
 109
 110     def closeEvent(self, event):
 111         while self.codedir in sys.path:
 112             sys.path.remove(self.codedir)
 113         eficasSalome.MyEficas.closeEvent(self, event)