Salome HOME
Update copyrights
[modules/smesh.git] / src / Tools / blocFissure / ihm / fissureCoude_plugin.py
1 # -*- coding: utf-8 -*-
2
3 # Copyright (C) 2006-2019  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 math
26 import sys
27 import traceback
28
29 from blocFissure import gmu
30
31
32 def fissureCoudeDlg(context):
33   # get context study, salomeGui
34   study = context.study
35   sg = context.sg
36   
37   import os
38   #import subprocess
39   #import tempfile
40   from qtsalome import QFileDialog, QMessageBox, QPalette, QColor, QDialog
41   from blocFissure.ihm.fissureCoude_ui import Ui_Dialog
42   
43   class fissureCoudeDialog(QDialog):
44     
45     def __init__(self):
46       QDialog.__init__(self)
47       # Set up the user interface from Designer.
48       self.ui = Ui_Dialog()
49       self.ui.setupUi(self)
50       
51       self.blackPalette = self.ui.dsb_angle.palette()
52       self.redPalette = QPalette()
53       self.redPalette.setColor(QPalette.Text, QColor(255,0,0))
54       self.NOK = False
55       
56       self.initDefaut()
57       self.initDialog(self.defaut)
58       self.ui.dsb_angle.setSpecialValueText("saisie_obligatoire")
59       self.ui.dsb_rCintr.setSpecialValueText("saisie_obligatoire")
60       self.ui.dsb_lTubeP1.setSpecialValueText("saisie_obligatoire")
61       self.ui.dsb_lTubeP2.setSpecialValueText("saisie_obligatoire")
62       self.ui.dsb_epais.setSpecialValueText("saisie_obligatoire")
63       self.ui.dsb_dext.setSpecialValueText("saisie_obligatoire")
64       self.ui.dsb_profondeur.setSpecialValueText("saisie_obligatoire")
65       self.ui.dsb_longueur.setSpecialValueText("saisie_obligatoire")
66       self.ui.dsb_azimut.setSpecialValueText("saisie_obligatoire")
67       self.ui.dsb_orientation.setSpecialValueText("saisie_obligatoire")
68       self.ui.dsb_posiAngul.setSpecialValueText("saisie_obligatoire")
69       self.ui.dsb_absCurv.setSpecialValueText("saisie_obligatoire")
70       self.ui.sb_nbTranches.setSpecialValueText("saisie_obligatoire")
71       self.ui.sb_nbCouronne.setSpecialValueText("saisie_obligatoire")
72       self.ui.sb_nbSecteur.setSpecialValueText("saisie_obligatoire")
73       self.ui.dsb_aretesFaceFissure.setSpecialValueText("automatique")
74       self.ui.dsb_influence.setSpecialValueText("automatique")
75       self.ui.lb_calcul.hide()
76       
77       # Connect up the buttons.
78       self.ui.pb_valPrec.clicked.connect(self.readValPrec)
79       self.ui.pb_reset.clicked.connect(self.resetVal)
80       self.ui.pb_recharger.clicked.connect(self.recharger)
81       self.ui.pb_sauver.clicked.connect(self.sauver)
82       self.ui.buttonBox.accepted.disconnect(self.accept)
83       self.ui.buttonBox.accepted.connect(self.execute)
84     
85     def initDefaut(self):
86       self.defaut = dict(
87         angle             = -181.0,
88         rCintr            = 0.0,
89         lTubeP1           = 0.0,
90         lTubeP2           = 0.0,
91         epais             = 0.0,
92         dext              = 0.0,
93         profondeur        = 0.0,
94         longueur          = 0.0,
95         azimut            = -181.0,
96         orientation       = -1.0,
97         posiAngul         = -181.0,
98         absCurv           = 0.0,
99         nbTranches        = 7,
100         nbCouronnes       = 1,
101         nbSecteurs        = 3,
102         cbOptDiscrSain    = False,
103         cbOptDiscrFiss    = False,
104         rbPosiAngul       = True,
105         rbFissExt         = True,
106         cbForceEllipse    = False,
107         nbAxeTubeP1       = 15,
108         nbAxeTubeP2       = 15,
109         nbAxeCoude        = 10,
110         nbCirconf         = 20,
111         nbEpaisseur       = 3,
112         rayonTore         = 2.0,
113         aretesFaceFissure = 0.0,
114         influence         = 0.0,
115         )
116       
117     def initDialog(self, dico):
118       self.ui.dsb_angle.setValue(dico['angle'])
119       self.ui.dsb_rCintr.setValue(dico['rCintr'])
120       self.ui.dsb_lTubeP1.setValue(dico['lTubeP1'])
121       self.ui.dsb_lTubeP2.setValue(dico['lTubeP2'])
122       self.ui.dsb_epais.setValue(dico['epais'])
123       self.ui.dsb_dext.setValue(dico['dext'])
124       self.ui.dsb_profondeur.setValue(dico['profondeur'])
125       self.ui.dsb_longueur.setValue(dico['longueur'])
126       self.ui.dsb_azimut.setValue(dico['azimut'])
127       self.ui.dsb_orientation.setValue(dico['orientation'])
128       self.ui.dsb_posiAngul.setValue(dico['posiAngul'])
129       self.ui.dsb_absCurv.setValue(dico['absCurv'])
130       self.ui.sb_nbTranches.setValue(dico['nbTranches'])
131       self.ui.sb_nbCouronne.setValue(dico['nbCouronnes'])
132       self.ui.sb_nbSecteur.setValue(dico['nbSecteurs'])
133       self.ui.dsb_aretesFaceFissure.setValue(dico['aretesFaceFissure'])
134       self.ui.dsb_influence.setValue(dico['influence'])
135       self.ui.sb_nbAxeTubeP1.setValue(dico['nbAxeTubeP1'])
136       self.ui.sb_nbAxeTubeP2.setValue(dico['nbAxeTubeP2'])
137       self.ui.sb_nbAxeCoude.setValue(dico['nbAxeCoude'])
138       self.ui.sb_nbCirconf.setValue(dico['nbCirconf'])
139       self.ui.sb_nbEpaisseur.setValue(dico['nbEpaisseur'])
140       self.ui.dsb_rayonTore.setValue(dico['rayonTore'])
141       #self.ui.cb_optDiscrSain.setChecked(False)
142       #self.ui.gb_discrSain.setShown(False)
143       self.ui.cb_optDiscrSain.setChecked(not(dico['cbOptDiscrSain']))
144       self.ui.cb_optDiscrSain.click()
145       self.ui.cb_optDiscrFiss.setChecked(not(dico['cbOptDiscrFiss']))
146       self.ui.cb_optDiscrFiss.click()
147       if dico['rbPosiAngul']:
148         self.ui.dsb_absCurv.setEnabled(False)
149         self.ui.dsb_posiAngul.setEnabled(True)
150         self.ui.rb_posiAngul.setChecked(True)
151         #self.ui.rb_posiAngul.click()
152       else:
153         self.ui.dsb_absCurv.setEnabled(True)
154         self.ui.dsb_posiAngul.setEnabled(False)
155         self.ui.rb_absCurv.setChecked(True)
156         #self.ui.rb_absCurv.click()
157       self.ui.rb_fissExt.setChecked(dico['rbFissExt'])
158       self.ui.cb_forceEllipse.setChecked(dico['cbForceEllipse'])
159       incomplet = self.testval(dico)
160       pass
161     
162     def testval(self, dico):
163       incomplet = False
164       if dico['angle'] < -180.0:
165         self.ui.dsb_angle.setPalette(self.redPalette)
166         incomplet = True
167       else:
168         self.ui.dsb_angle.setPalette(self.blackPalette)
169         
170       if dico['rCintr'] == 0.0:
171         self.ui.dsb_rCintr.setPalette(self.redPalette)
172         incomplet = True
173       else:
174         self.ui.dsb_rCintr.setPalette(self.blackPalette)
175         
176       if dico['lTubeP1'] == 0.0:  
177         self.ui.dsb_lTubeP1.setPalette(self.redPalette)
178         incomplet = True
179       else:
180         self.ui.dsb_lTubeP1.setPalette(self.blackPalette)
181         
182       if dico['lTubeP2'] == 0.0:  
183         self.ui.dsb_lTubeP2.setPalette(self.redPalette)
184         incomplet = True
185       else:
186         self.ui.dsb_lTubeP2.setPalette(self.blackPalette)
187         
188       if dico['epais'] == 0.0:  
189         self.ui.dsb_epais.setPalette(self.redPalette)
190         incomplet = True
191       else:
192         self.ui.dsb_epais.setPalette(self.blackPalette)
193         
194       if dico['dext'] == 0.0:  
195         self.ui.dsb_dext.setPalette(self.redPalette)
196         incomplet = True
197       else:
198         self.ui.dsb_dext.setPalette(self.blackPalette)
199         
200       if dico['profondeur'] == 0.0:  
201         self.ui.dsb_profondeur.setPalette(self.redPalette)
202         incomplet = True
203       else:
204         self.ui.dsb_profondeur.setPalette(self.blackPalette)
205         
206       if dico['longueur'] == 0.0:  
207         self.ui.dsb_longueur.setPalette(self.redPalette)
208         incomplet = True
209       else:
210         self.ui.dsb_longueur.setPalette(self.blackPalette)
211         
212       if dico['azimut'] < -180.0:  
213         self.ui.dsb_azimut.setPalette(self.redPalette)
214         incomplet = True
215       else:
216         self.ui.dsb_azimut.setPalette(self.blackPalette)
217         
218       if dico['orientation'] < 0.0:  
219         self.ui.dsb_orientation.setPalette(self.redPalette)
220         incomplet = True
221       else:
222         self.ui.dsb_orientation.setPalette(self.blackPalette)
223         
224       if dico['posiAngul'] < -180.0 and dico['rbPosiAngul'] == True:  
225         self.ui.dsb_posiAngul.setPalette(self.redPalette)
226         incomplet = True
227       else:
228         self.ui.dsb_posiAngul.setPalette(self.blackPalette)
229         
230       if dico['absCurv'] == 0.0 and dico['rbPosiAngul'] == False:  
231         self.ui.dsb_absCurv.setPalette(self.redPalette)
232         incomplet = True
233       else:
234         self.ui.dsb_absCurv.setPalette(self.blackPalette)
235         
236       if dico['nbTranches'] == 7:  
237         self.ui.sb_nbTranches.setPalette(self.redPalette)
238         incomplet = True
239       else:
240         self.ui.sb_nbTranches.setPalette(self.blackPalette)
241         
242       if dico['nbCouronnes'] == 1:  
243         self.ui.sb_nbCouronne.setPalette(self.redPalette)
244         incomplet = True
245       else:
246         self.ui.sb_nbCouronne.setPalette(self.blackPalette)
247         
248       if dico['nbSecteurs'] == 3:  
249         self.ui.sb_nbSecteur.setPalette(self.redPalette)
250         incomplet = True
251       else:
252         self.ui.sb_nbSecteur.setPalette(self.blackPalette)
253         
254       print("incomplet: ", incomplet)
255       return incomplet
256     
257     def fileDefault(self):
258       filedef = os.path.expanduser("~/.config/salome/dialogFissureCoude.dic")
259       print(filedef)
260       return filedef
261     
262     def writeDefault(self, dico):
263       filedef = self.fileDefault()
264       with open(filedef, 'w') as f:
265           f.write(str(dico))
266     
267     def readValPrec(self):
268       filedef = self.fileDefault()
269       if os.path.exists(filedef):
270         with open(filedef, 'r') as f:
271             txt = f.read()
272         dico = eval(txt)
273         print(dico)
274         self.initDialog(dico)
275
276     def resetVal(self):
277       #self.initDefaut()
278       self.initDialog(self.defaut)
279       
280     def sauver(self):
281       print("sauver")
282       fileDiag = QFileDialog(self)
283       fileDiag.setFileMode(QFileDialog.AnyFile)
284       fileDiag.setNameFilter("Parametres *.dic (*.dic)")
285       fileDiag.setViewMode(QFileDialog.List)
286       if fileDiag.exec_() :
287         fileNames = fileDiag.selectedFiles()
288         filedef = fileNames[0]
289         dico = self.creeDico()
290         with open(filedef, 'w') as f:
291             f.write(str(dico))
292         
293     def recharger(self):
294       print("recharger")
295       fileDiag = QFileDialog(self)
296       fileDiag.setFileMode(QFileDialog.ExistingFile)
297       fileDiag.setNameFilter("Parametres *.dic (*.dic)")
298       fileDiag.setViewMode(QFileDialog.Detail)
299       if fileDiag.exec_() :
300         fileNames = fileDiag.selectedFiles()
301         filedef = fileNames[0]
302         print(filedef)
303         if os.path.exists(filedef):
304           with open(filedef, 'r') as f:
305               txt = f.read()
306           dico = eval(txt)
307           print(dico)
308           self.initDialog(dico)
309          
310     def creeDico(self):
311       dico = dict(
312         angle             = self.ui.dsb_angle.value(),
313         rCintr            = self.ui.dsb_rCintr.value(),
314         lTubeP1           = self.ui.dsb_lTubeP1.value(),
315         lTubeP2           = self.ui.dsb_lTubeP2.value(),
316         epais             = self.ui.dsb_epais.value(),
317         dext              = self.ui.dsb_dext.value(),
318         profondeur        = self.ui.dsb_profondeur.value(),
319         longueur          = self.ui.dsb_longueur.value(),
320         azimut            = self.ui.dsb_azimut.value(),
321         orientation       = self.ui.dsb_orientation.value(),
322         posiAngul         = self.ui.dsb_posiAngul.value(),
323         absCurv           = self.ui.dsb_absCurv.value(),
324         nbTranches        = self.ui.sb_nbTranches.value(),
325         nbCouronnes       = self.ui.sb_nbCouronne.value(),
326         nbSecteurs        = self.ui.sb_nbSecteur.value(),
327         cbOptDiscrSain    = self.ui.cb_optDiscrSain.isChecked(),
328         cbOptDiscrFiss    = self.ui.cb_optDiscrFiss.isChecked(),
329         rbPosiAngul       = self.ui.rb_posiAngul.isChecked(),
330         rbFissExt         = self.ui.rb_fissExt.isChecked(),
331         cbForceEllipse    = self.ui.cb_forceEllipse.isChecked(),
332         nbAxeTubeP1       = self.ui.sb_nbAxeTubeP1.value(),
333         nbAxeTubeP2       = self.ui.sb_nbAxeTubeP2.value(),
334         nbAxeCoude        = self.ui.sb_nbAxeCoude.value(),
335         nbCirconf         = self.ui.sb_nbCirconf.value(),
336         nbEpaisseur       = self.ui.sb_nbEpaisseur.value(),
337         rayonTore         = self.ui.dsb_rayonTore.value(),
338         aretesFaceFissure = self.ui.dsb_aretesFaceFissure.value(),
339         influence         = self.ui.dsb_influence.value(),
340         )
341       print(dico)
342       return dico
343       
344     def checkValues(self):
345       return self.NOK
346
347     def execute(self):
348       from blocFissure.gmu import initLog
349       #initLog.setDebug()
350       initLog.setVerbose() # don't set the level too early, to be able to modify it
351       from blocFissure.gmu import geomsmesh
352       from blocFissure.gmu.casStandard import casStandard
353       from blocFissure.ihm.fissureCoude_ihm import fissureCoude_ihm
354       
355       dico = self.creeDico()
356       NOK = self.testval(dico)
357       if not(NOK):
358         dico['lenSegPipe'] = (dico['longueur'] + math.pi*dico['profondeur'])/dico['nbTranches']
359         print('lenSegPipe', dico['lenSegPipe'])
360         areteMinAngle = (dico['rCintr'] -dico['dext']/2.0)*(dico['angle']*math.pi/180.0)/dico['nbAxeCoude']
361         print('areteMinAngle', areteMinAngle)
362         areteMinCirco = dico['dext']*math.pi/(2*dico['nbCirconf'])
363         print('areteMinCirco', areteMinCirco)
364         areteMinEpais = dico['epais']/dico['nbEpaisseur']
365         print('areteMinEpais', areteMinEpais)
366         if dico['influence'] == 0:
367           dico['influence'] = max(areteMinAngle, areteMinCirco, areteMinEpais)
368           print('influence', dico['influence'])
369         if dico['aretesFaceFissure'] == 0:
370           dico['aretesFaceFissure'] = (areteMinAngle + areteMinCirco)/2.0
371           print('aretesFaceFissure', dico['aretesFaceFissure'])
372         if dico['rbPosiAngul'] == False:
373           rmoy = (dico['dext'] - dico['epais'])/2.0
374           eta = 1
375           if dico['rbFissExt'] == False:
376             eta = -1
377           dico['posiAngul'] = (180.0/math.pi)*dico['absCurv']/(dico['rCintr']+(rmoy+eta*dico['epais']/2.0)*math.cos(math.pi*dico['azimut']/180.))
378           print('posiAngul' , dico['posiAngul'])
379         
380         self.writeDefault(dico)
381         self.ui.lb_calcul.show()
382         probleme = fissureCoude_ihm(0)
383         probleme.setDicoParams(dico)
384         probleme.executeProbleme()
385       self.NOK = NOK
386       self.accept()
387     
388     pass 
389
390 # ----------------------------------------------------------------------------
391                      
392   window = fissureCoudeDialog()
393 #  window.ui.dsb_tolerance.setValue(0.01)
394   retry = True
395   while(retry):
396     retry = False
397     window.exec_()
398     result = window.result()
399     if result:
400       # dialog accepted
401       print("dialog accepted, check")
402       retry = window.checkValues()
403     else:
404       print("dialog rejected, exit")
405   pass
406