]> SALOME platform Git repositories - modules/smesh.git/blob - src/Tools/blocFissure/ihm/fissureGenerale_plugin.py
Salome HOME
Simplification de insereFissureLongue
[modules/smesh.git] / src / Tools / blocFissure / ihm / fissureGenerale_plugin.py
1 # -*- coding: utf-8 -*-
2
3 # Copyright (C) 2006-2020  EDF R&D
4 #
5 # This library is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU Lesser General Public
7 # License as published by the Free Software Foundation; either
8 # version 2.1 of the License, or (at your option) any later version.
9 #
10 # This library is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 # Lesser General Public License for more details.
14 #
15 # You should have received a copy of the GNU Lesser General Public
16 # License along with this library; if not, write to the Free Software
17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
18 #
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #
21
22 # if you already have plugins defined in a salome_plugins.py file, add this file at the end.
23 # if not, copy this file as ${HOME}/Plugins/smesh_plugins.py or ${APPLI}/Plugins/smesh_plugins.py
24
25 import sys, traceback
26 import math
27 from blocFissure import gmu
28
29 def fissureGeneraleDlg(context):
30   # get context study, salomeGui
31   study = context.study
32   sg = context.sg
33
34   import os
35   #import subprocess
36   #import tempfile
37   from PyQt5 import QtCore
38   from PyQt5 import QtWidgets
39   from PyQt5 import QtGui
40   from PyQt5.QtWidgets import QFileDialog
41   from PyQt5.QtWidgets import QMessageBox
42   from PyQt5.QtGui import QPalette
43   from PyQt5.QtGui import QColor
44   from blocFissure.ihm.fissureGenerale_ui import Ui_Dialog
45
46   class fissureGeneraleDialog(QtWidgets.QDialog):
47
48     def __init__(self):
49       print("__init__")
50       QtWidgets.QDialog.__init__(self)
51       # Set up the user interface from Designer.
52       self.ui = Ui_Dialog()
53       self.ui.setupUi(self)
54
55       self.blackPalette = self.ui.dsb_influence.palette()
56       self.redPalette = QPalette()
57       self.redPalette.setColor(QPalette.Text, QColor(255,0,0))
58       self.NOK = False
59
60       self.initDefaut()
61       self.initDialog(self.defaut)
62       self.ui.lb_calcul.hide()
63
64       # Connect up the buttons.
65
66       self.ui.pb_exemple.clicked.connect(self.genereExemples)
67       self.ui.pb_valPrec.clicked.connect(self.readValPrec)
68       self.ui.pb_reset.clicked.connect(self.resetVal)
69       self.ui.pb_recharger.clicked.connect(self.recharger)
70       self.ui.pb_sauver.clicked.connect(self.sauver)
71       self.ui.pb_maillage.clicked.connect(self.selectMaillage)
72       self.ui.pb_facefiss.clicked.connect(self.selectFacefiss)
73       self.ui.pb_reptrav.clicked.connect(self.selectReptrav)
74       self.ui.bb_OkCancel.accepted.disconnect(self.accept)
75       self.ui.bb_OkCancel.accepted.connect(self.execute)
76
77     def initDefaut(self):
78       self.defaut = dict( \
79         nomCas            = "angleCube", \
80         maillageSain      = os.path.join(gmu.pathBloc, "materielCasTests", "CubeAngle.med"), \
81         brepFaceFissure   = os.path.join(gmu.pathBloc, "materielCasTests", "CubeAngleFiss.brep"), \
82         edgeFissIds       = [3], \
83         lgInfluence       = 20, \
84         meshBrep          = (5,10), \
85         rayonPipe         = 5, \
86         lenSegPipe        = 2.5, \
87         nbSegRad          = 5, \
88         nbSegCercle       = 32, \
89         areteFaceFissure  = 10, \
90         areteVives        = 0, \
91         reptrav           = os.curdir, \
92         nomres            = "maillage_avec_fissure", \
93         verbosite         = 0)
94
95
96     def initDialog(self, dico):
97       self.ui.le_maillage.setText(dico['maillageSain'])
98       self.ui.le_facefiss.setText(dico['brepFaceFissure'])
99       self.ui.le_fondfiss.setText(str(dico['edgeFissIds']))
100       self.ui.dsb_influence.setValue(dico['lgInfluence'])
101       self.ui.dsb_meshBrepMin.setValue(dico['meshBrep'][0])
102       self.ui.dsb_meshBrepMax.setValue(dico['meshBrep'][1])
103       self.ui.dsb_rayonPipe.setValue(dico['rayonPipe'])
104       self.ui.dsb_lenSegPipe.setValue(dico['lenSegPipe'])
105       self.ui.sb_couronnes.setValue(dico['nbSegRad'])
106       self.ui.sb_secteurs.setValue(dico['nbSegCercle'])
107       self.ui.dsb_areteFaceFissure.setValue(dico['areteFaceFissure'])
108       if 'aretesVives' in dico:
109         self.ui.dsb_aretesVives.setValue(dico['aretesVives'])
110       else:
111         self.ui.dsb_aretesVives.setValue(0)
112       self.ui.le_reptrav.setText(os.path.abspath(dico['reptrav']))
113       self.ui.le_nomres.setText(os.path.split(dico['nomres'])[1])
114       self.ui.cb_log.setCurrentIndex(dico['verbosite'])
115       incomplet = self.testval(dico)
116       pass
117
118     def testval(self, dico):
119       incomplet = False
120       if not os.path.lexists(dico['maillageSain']):
121         self.ui.le_maillage.setPalette(self.redPalette)
122         incomplet = True
123       else:
124         self.ui.le_maillage.setPalette(self.blackPalette)
125       cao_file = dico['brepFaceFissure']
126       if not os.path.lexists(cao_file):
127         self.ui.le_facefiss.setPalette(self.redPalette)
128         incomplet = True
129       else:
130         suffix = os.path.basename(cao_file).split(".")[-1]
131         if ( suffix.upper() not in ("BREP","XAO") ):
132           print ("Suffixe inconnu pour le fichier {}".format(cao_file))
133           self.ui.le_facefiss.setPalette(self.redPalette)
134           incomplet = True
135         else:
136           self.ui.le_facefiss.setPalette(self.blackPalette)
137       if dico['edgeFissIds']:
138         edgeFissIdsOK=True
139         param_0 = dico['edgeFissIds'][0]
140         type_param_id = type(param_0)
141         for param in dico['edgeFissIds'][1:]:
142           if not isinstance(param,type_param_id):
143             print ("La donnée {} n'est pas du même type que la première de la liste : {}.".format(param,type(param)))
144             incomplet = True
145             edgeFissIdsOK=False
146         if edgeFissIdsOK:
147           if isinstance(param_0, int):
148             pass
149           elif isinstance(param_0, str):
150             pass
151           else:
152             print("Il faut une liste d'IDs d'arêtes ou une liste de noms de groupes d'arêtes.")
153             incomplet = True
154             edgeFissIdsOK=False
155       else:
156         incomplet = True
157         edgeFissIdsOK=False
158       if edgeFissIdsOK:
159         self.ui.le_fondfiss.setPalette(self.blackPalette)
160       else:
161         self.ui.le_fondfiss.setPalette(self.redPalette)
162       if dico['meshBrep'][0] == 0:
163         self.ui.dsb_meshBrepMin.setPalette(self.redPalette)
164         incomplet = True
165       else:
166         self.ui.dsb_meshBrepMin.setPalette(self.blackPalette)
167       if dico['meshBrep'][1] == 0:
168         self.ui.dsb_meshBrepMax.setPalette(self.redPalette)
169         incomplet = True
170       else:
171         self.ui.dsb_meshBrepMax.setPalette(self.blackPalette)
172       if dico['rayonPipe'] == 0:
173         self.ui.dsb_rayonPipe.setPalette(self.redPalette)
174         incomplet = True
175       else:
176         self.ui.dsb_rayonPipe.setPalette(self.blackPalette)
177       if dico['lenSegPipe'] == 0:
178         self.ui.dsb_lenSegPipe.setPalette(self.redPalette)
179         incomplet = True
180       else:
181         self.ui.dsb_lenSegPipe.setPalette(self.blackPalette)
182       if dico['areteFaceFissure'] == 0:
183         self.ui.dsb_areteFaceFissure.setPalette(self.redPalette)
184         incomplet = True
185       else:
186         self.ui.dsb_areteFaceFissure.setPalette(self.blackPalette)
187
188       print("incomplet: ", incomplet)
189       return incomplet
190
191     def fileDefault(self):
192       filedef = os.path.expanduser( os.path.join(os.environ["HOME"],".config", "salome", "dialogFissureGenerale.dic") )
193       print(filedef)
194       return filedef
195
196     def writeDefault(self, dico):
197       filedef = self.fileDefault()
198       with open(filedef, 'w') as f:
199         f.write(str(dico))
200
201     def genereExemples(self):
202       maillageSain      = os.path.join(gmu.pathBloc, "materielCasTests", "CubeAngle.med")
203       brepFaceFissure   = os.path.join(gmu.pathBloc, "materielCasTests", "CubeAngleFiss.brep")
204       if (os.path.exists(maillageSain) and os.path.exists(brepFaceFissure)):
205         self.initDialog(self.defaut)
206       else:
207         self.ui.lb_calcul.setText("--- Generation exemples en cours ---")
208         self.ui.lb_calcul.show()
209         from blocFissure.materielCasTests import genereMateriel
210         self.ui.lb_calcul.hide()
211         self.initDialog(self.defaut)
212
213     def readValPrec(self):
214       filedef = self.fileDefault()
215       if os.path.exists(filedef):
216         with open(filedef, 'r') as f:
217           txt = f.read()
218         dico = eval(txt)
219         print(dico)
220         self.initDialog(dico)
221
222     def resetVal(self):
223       #self.initDefaut()
224       self.initDialog(self.defaut)
225
226     def setLogVerbosity(self, logfile):
227       from blocFissure.gmu import initLog # le mode de log s'initialise une seule fois
228       print("setLogVerbosity")
229       index = self.ui.cb_log.currentIndex()
230       print(index)
231       if index == 0:
232         initLog.setRelease(logfile)
233       elif index == 1:
234         initLog.setVerbose(logfile)
235       elif index == 2:
236         initLog.setDebug(logfile)
237
238
239     def sauver(self):
240       print("sauver")
241       fileDiag = QFileDialog(self)
242       fileDiag.setFileMode(QFileDialog.AnyFile)
243       fileDiag.setNameFilter("Parametres *.dic (*.dic)")
244       fileDiag.setViewMode(QFileDialog.List)
245       if fileDiag.exec_() :
246         fileNames = fileDiag.selectedFiles()
247         print(fileNames)
248         filedef = fileNames[0]
249         if filedef[-4:] not in ['.dic']:
250           filedef += '.dic'
251         dico = self.creeDico()
252         with open(filedef, 'w') as f:
253           f.write(str(dico))
254
255     def recharger(self):
256       print("recharger")
257       fileDiag = QFileDialog(self)
258       fileDiag.setFileMode(QFileDialog.ExistingFile)
259       fileDiag.setNameFilter("Parametres *.dic (*.dic)")
260       fileDiag.setViewMode(QFileDialog.Detail)
261       if fileDiag.exec_() :
262         fileNames = fileDiag.selectedFiles()
263         filedef = fileNames[0]
264         print(filedef)
265         if os.path.exists(filedef):
266           with open(filedef, 'r') as f:
267             txt = f.read()
268           dico = eval(txt)
269           print(dico)
270           self.initDialog(dico)
271
272     def selectMaillage(self):
273       fileDiag = QFileDialog(self)
274       fileDiag.setFileMode(QFileDialog.ExistingFile)
275       fileDiag.setNameFilter("Maillage *.med (*.med)")
276       fileDiag.setViewMode(QFileDialog.Detail)
277       if fileDiag.exec_() :
278         fileNames = fileDiag.selectedFiles()
279         filedef = fileNames[0]
280         print(filedef)
281         self.ui.le_maillage.setText(filedef)
282
283     def selectFacefiss(self):
284       fileDiag = QFileDialog(self)
285       fileDiag.setFileMode(QFileDialog.ExistingFile)
286       fileDiag.setNameFilter("Face fissure *.brep (*.brep)")
287       fileDiag.setViewMode(QFileDialog.Detail)
288       if fileDiag.exec_() :
289         fileNames = fileDiag.selectedFiles()
290         filedef = fileNames[0]
291         print(filedef)
292         self.ui.le_facefiss.setText(filedef)
293
294     def selectReptrav(self):
295       fileDiag = QFileDialog(self)
296       fileDiag.setFileMode(QFileDialog.Directory)
297       fileDiag.setViewMode(QFileDialog.Detail)
298       fileDiag.setDirectory(self.ui.le_reptrav.text())
299       if fileDiag.exec_() :
300         fileNames = fileDiag.selectedFiles()
301         reptrav = str(fileNames[0])
302         print("reptrav ", reptrav)
303         self.ui.le_reptrav.setText(os.path.abspath(reptrav))
304
305
306     def selectNomres(self):
307       fileDiag = QFileDialog(self)
308       fileDiag.setFileMode(QFileDialog.AnyFile)
309       fileDiag.setViewMode(QFileDialog.Detail)
310       nomres=str(os.path.split(str(self.ui.le_nomres.text()))[1])
311       fileDiag.setDirectory(self.ui.le_reptrav.text())
312       fileDiag.selectFile(nomres)
313       fileDiag.setNameFilter("Maillage *.med (*.med)")
314       self.ui.le_nomres.setText(nomres)
315       if fileDiag.exec_() :
316         fileNames = fileDiag.selectedFiles()
317         tempnom = os.path.split(str(fileNames[0]))[1]
318         print("nomres ", tempnom)
319         self.ui.le_nomres.setText(tempnom)
320       else:
321         self.ui.le_nomres.setText(nomres)
322
323     def creeDico(self):
324       dico = dict( \
325                   maillageSain     = str(self.ui.le_maillage.text()), \
326                   brepFaceFissure  = str(self.ui.le_facefiss.text()), \
327                   edgeFissIds      = eval(str(self.ui.le_fondfiss.text())), \
328                   lgInfluence      = self.ui.dsb_influence.value(), \
329                   meshBrep         = [self.ui.dsb_meshBrepMin.value(),self.ui.dsb_meshBrepMax.value()], \
330                   rayonPipe        = self.ui.dsb_rayonPipe.value(), \
331                   lenSegPipe       = self.ui.dsb_lenSegPipe.value(), \
332                   nbSegRad         = self.ui.sb_couronnes.value(), \
333                   nbSegCercle      = self.ui.sb_secteurs.value(), \
334                   areteFaceFissure = self.ui.dsb_areteFaceFissure.value(), \
335                   aretesVives      = self.ui.dsb_aretesVives.value(), \
336                   reptrav          = str(self.ui.le_reptrav.text()), \
337                   nomres           = str(self.ui.le_nomres.text()), \
338                   verbosite        = self.ui.cb_log.currentIndex() \
339                   )
340       print(dico)
341       return dico
342
343     def checkValues(self):
344       return self.NOK
345
346     def execute(self):
347       print("execute")
348       dico = self.creeDico()
349       NOK = self.testval(dico)
350       if not(NOK):
351         self.writeDefault(dico)
352         self.ui.lb_calcul.setText("--- Calcul en cours ---")
353         self.ui.lb_calcul.show()
354         logfile=os.path.join(dico['reptrav'], dico['nomres']+".log")
355         self.setLogVerbosity(logfile)
356         from blocFissure.gmu import geomsmesh               # après intialisation log dans setLogVerbosity
357         from blocFissure.gmu.casStandard import casStandard # après intialisation log dans setLogVerbosity
358         from blocFissure.gmu.fissError import fissError
359         try:
360           execInstance = casStandard(dico)
361         except fissError as erreur:
362           print('-'*60)
363           print(type(erreur))
364           print('-'*60)
365           print(erreur.msg)
366           print('-'*60)
367           for ligne in erreur.pile:
368             print(repr(ligne))
369           print('-'*60)
370           texte = erreur.msg
371 #           texte += +"<br>" +'-'*60 +"<br>"
372 #           for ligne in erreur.pile:
373 #             texte += repr(ligne) +"<br>"
374           mbox = QMessageBox(self)
375           mbox.setWindowTitle("erreur blocFissure")
376           mbox.setText(str(texte))
377           mbox.exec_()
378 #        except Exception as erreur:
379 #          print "exception non répertoriée"
380       self.NOK = NOK
381       self.ui.lb_calcul.hide()
382       #self.accept()
383
384     pass
385
386 # ----------------------------------------------------------------------------
387
388   print("main")
389   window = fissureGeneraleDialog()
390   retry = True
391   while(retry):
392     retry = False
393     window.exec_()
394     result = window.result()
395     if result:
396       # dialog accepted
397       print("dialog accepted, check")
398       retry = window.checkValues()
399     else:
400       print("dialog rejected, exit")
401   pass