Salome HOME
modif pour MT
[tools/eficas.git] / InterfaceQT4 / gereIcones.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2007-2017   EDF R&D
3 #
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU Lesser General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License.
8 #
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 # Lesser General Public License for more details.
13 #
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 #
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #
20 # Modules Python
21 from __future__ import absolute_import
22 from __future__ import print_function
23 try :
24    from builtins import str
25    from builtins import object
26 except : pass
27
28 import types,os,re,sys
29 import traceback
30 import six
31 import inspect
32
33 from PyQt5.QtWidgets import QMessageBox, QFileDialog , QMenu, QPushButton, QTreeView ,QListView, QAbstractItemView
34 from PyQt5.QtGui import QIcon
35 from PyQt5.QtCore import  QFileInfo,  Qt, QSize, QVariant
36
37
38 from Extensions.i18n import tr
39 listeSuffixe=('bmp','png','jpg' ,'txt','med')
40
41
42 class FacultatifOuOptionnel(object):
43
44   def setReglesEtAide(self):
45       listeRegles=()
46       try :
47          listeRegles     = self.node.item.getRegles()
48       except :
49          pass
50       if hasattr(self,"RBRegle"):
51         if listeRegles==() : self.RBRegle.close() 
52         else :
53            icon3=QIcon(self.repIcon+"/lettreRblanc30.png")
54            self.RBRegle.setIcon(icon3)
55            self.RBRegle.clicked.connect(self.viewRegles)
56
57       cle_doc=None
58       if not hasattr(self,"RBInfo"):return
59       icon=QIcon(self.repIcon+"/point-interrogation30.png")
60       self.RBInfo.setIcon(icon)
61
62       from .monWidgetCommande import MonWidgetCommande
63       if isinstance(self,MonWidgetCommande) and self.editor.code =="MAP":
64          self.cle_doc = self.chercheDocMAP()
65       else :
66          self.cle_doc = self.node.item.getDocu()
67       if self.cle_doc == None  : self.RBInfo.close()
68       else : self.RBInfo.clicked.connect (self.viewDoc)
69
70
71   def chercheDocMAP(self):
72       try :
73         clef=self.editor.maConfiguration.adresse+"/"
74       except :
75         return None
76       for k in self.editor.readercata.cata.JdC.dict_groupes:
77           if self.obj.nom in self.editor.readercata.cata.JdC.dict_groupes[k]:
78              clef+=k
79              break
80       clef+="/"+ self.obj.nom[0:-5].lower()+"/spec_"+self.obj.nom[0:-5].lower()+".html"
81
82       return clef
83  
84   def viewDoc(self):
85       try :
86           if sys.platform[0:5]=="linux" : cmd="xdg-open "+self.cle_doc
87           else                          : cmd="start "+self.cle_doc
88           os.system(cmd)
89       except:
90           QMessageBox.warning( self,tr( "Aide Indisponible"),tr( "l'aide n est pas installee "))
91
92   def viewRegles(self):
93       self.node.appellebuildLBRegles()
94
95
96   def setIconePoubelle(self):
97       if not(hasattr(self,"RBPoubelle")):return
98
99       if self.node.item.object.isOblig()  and not( hasattr(self.node.item.object,'isDeletable') ): 
100          icon=QIcon(self.repIcon+"/deleteRondVide.png")
101          self.RBPoubelle.setIcon(icon)
102          return
103       icon=QIcon(self.repIcon+"/deleteRond.png")
104       self.RBPoubelle.setIcon(icon)
105       self.RBPoubelle.clicked.connect(self.aDetruire)
106
107   def setIconesSalome(self):
108        if not (hasattr(self,"RBSalome")): return
109        from Accas import SalomeEntry
110        mc = self.node.item.get_definition()
111        mctype = mc.type[0]
112        enable_salome_selection = self.editor.salome and \
113          (('grma' in repr(mctype)) or ('grno' in repr(mctype)) or ('SalomeEntry' in repr(mctype)) or \
114          (hasattr(mctype, "enable_salome_selection") and mctype.enable_salome_selection))
115
116        if enable_salome_selection:
117           icon=QIcon(self.repIcon+"/flecheSalome.png")
118           self.RBSalome.setIcon(icon)
119           self.RBSalome.pressed.connect(self.BSalomePressed)
120
121 #PNPN --> Telemac A revoir surement
122 # cela ou le catalogue grpma ou salomeEntry
123           if not(('grma' in repr(mctype)) or ('grno' in repr(mctype))) or not(self.editor.salome): 
124              if hasattr(self,"RBSalomeVue") : self.RBSalomeVue.close()
125           else : 
126              icon1=QIcon(self.repIcon+"/eye.png")
127              self.RBSalomeVue.setIcon(icon1)
128              self.RBSalomeVue.clicked.connect(self.BView2DPressed)
129        else:
130           self.RBSalome.close()
131           self.RBSalomeVue.close()
132
133      
134   def setIconesFichier(self):
135        if not ( hasattr(self,"BFichier")): return
136        mc = self.node.item.get_definition()
137        mctype = mc.type[0]
138        if mctype == "FichierOuRepertoire":
139           self.BFichierOuRepertoire=self.BFichier
140           self.BFichierOuRepertoire.clicked.connect(self.BFichierOuRepertoirePressed)
141           self.BVisuFichier.close()
142        elif mctype == "Repertoire":
143           self.BRepertoire=self.BFichier
144           self.BRepertoire.clicked.connect(self.BRepertoirePressed)
145           self.BVisuFichier.close()
146        else :
147           self.BFichier.clicked.connect(self.BFichierPressed)
148           self.BVisuFichier.clicked.connect(self.BFichierVisu)
149
150
151
152   def setIconesGenerales(self):
153       repIcon=self.node.editor.appliEficas.repIcon
154       if hasattr(self,"BVisuListe") :
155          fichier=os.path.join(repIcon, 'plusnode.png')
156          icon = QIcon(fichier)
157          self.BVisuListe.setIcon(icon)
158       if hasattr(self,"RBDeplie") :
159          fichier=os.path.join(repIcon, 'plusnode.png')
160          icon = QIcon(fichier)
161          self.RBDeplie.setIcon(icon)
162       if hasattr(self,"RBPlie") :
163          fichier=os.path.join(repIcon, 'minusnode.png')
164          icon = QIcon(fichier)
165          self.RBPlie.setIcon(icon)
166       if hasattr(self,"BVisuFichier") :
167          fichier=os.path.join(repIcon, 'visuFichier.png')
168          icon = QIcon(fichier)
169          self.BVisuFichier.setIcon(icon)
170
171       
172
173   def setRun(self):
174       if hasattr(self.editor.appliEficas, 'mesScripts') :
175          if self.editor.code in  self.editor.appliEficas.mesScripts :
176             self.dict_commandes_mesScripts=self.appliEficas.mesScripts[self.editor.code].dict_commandes
177             if self.obj.nom in self.dict_commandes_mesScripts :
178                self.ajoutScript()
179                icon=QIcon(self.repIcon+"/roue.png")
180                if hasattr(self,"RBRun"):self.RBRun.setIcon(icon)
181                return
182       if hasattr(self,"RBRun"): self.RBRun.close()
183       if hasattr(self,"CBScripts"): self.CBScripts.close()
184
185
186   def aDetruire(self):
187       self.node.delete()
188
189   def setValide(self):
190       if not(hasattr (self,'RBValide')) : return
191       couleur=self.node.item.getIconName()
192       monIcone = QIcon(self.repIcon+"/" + couleur + ".png")
193       self.RBValide.setIcon(monIcone)
194
195   # il faut chercher la bonne fenetre
196   def rendVisible(self):
197       #print "je passe par rendVisible de FacultatifOuOptionnel"
198       #print self
199       #print self.node.fenetre
200       #print "return pour etre sure"
201       return
202       #PNPN
203       newNode=self.node.treeParent.chercheNoeudCorrespondant(self.node.item.object)
204       #print newNode
205       self.editor.fenetreCentraleAffichee.scrollAreaCommandes.ensureWidgetVisible(newNode.fenetre)
206       #newNode.fenetre.setFocus()
207
208
209   def ajoutScript(self):
210         if not hasattr(self,'CBScripts') : return # Cas des Widgets Plies
211         self.dictCommandes={}
212         listeCommandes=self.dict_commandes_mesScripts[self.obj.nom]
213         if type(listeCommandes) != tuple: listeCommandes=(listeCommandes,)
214         i=0
215         for commande in listeCommandes :
216           conditionSalome=commande[3]
217           if (self.appliEficas.salome == 0 and conditionSalome == True): continue
218           self.CBScripts.addItem(commande[1])
219           self.dictCommandes[commande[1]]=i
220           i=i+1
221         self.CBScripts.activated.connect(self.choixSaisi)
222
223   def choixSaisi(self):
224       fction=str(self.CBScripts.currentText())
225       numero= self.dictCommandes[fction]
226       self.node.appelleFonction(numero,nodeTraite=self.node)
227       #self.reaffiche()
228
229 class ContientIcones(object):
230
231   def BFichierVisu(self):
232        fichier=self.lineEditVal.text()
233        if fichier == None or str(fichier)=="" : return
234        from .monViewTexte import ViewText
235        try :
236          if sys.platform[0:5]=="linux" :
237            #cmd="xdg-open "+ str(fichier)
238            #changer pour marcher dans l'EDC
239            #cmd="gedit "+ str(fichier)
240            from os.path import splitext
241            fileName,extension = splitext(fichier)
242            if extension in  self.parentQt.editor.appliEficas.maConfiguration.utilParExtensions:
243               cmd=self.parentQt.editor.appliEficas.maConfiguration.utilParExtensions[extension] + " " + str(fichier)
244            else:  cmd="xdg-open "+ str(fichier)
245            os.system(cmd)
246          else                          :
247            os.startfile(str(fichier)) 
248        except:
249          try :
250             fp=open(fichier)
251             txt=fp.read()
252             nomFichier=QFileInfo(fichier).baseName()
253             maVue=ViewText(self,entete=nomFichier)
254             maVue.setText(txt)
255             maVue.show()
256             fp.close()
257          except:
258             QMessageBox.warning( None,
259                        tr("Visualisation Fichier "),
260                        tr("Impossibilite d'afficher le Fichier"),)
261
262   def BFichierPressed(self):
263       mctype = self.node.item.get_definition().type
264       if len(mctype) > 1:
265           filters = mctype[1]
266       elif hasattr(mctype[0], "filters"):
267           filters = mctype[0].filters
268       else:
269           filters = ""
270       if len(mctype) > 2 and mctype[2] == "Sauvegarde":
271           fichier = QFileDialog.getSaveFileName(self.appliEficas,
272                               tr('Sauvegarder Fichier'),
273                               self.appliEficas.maConfiguration.savedir,
274                               filters)
275       else:
276           #print(filters)
277           fichier = QFileDialog.getOpenFileName(self.appliEficas,
278                               tr('Ouvrir Fichier'),
279                               self.appliEficas.maConfiguration.savedir,
280                               filters)
281
282       fichier=fichier[0]
283       if not(fichier == ""):
284          ulfile = os.path.abspath(six.text_type(fichier))
285          self.appliEficas.maConfiguration.savedir=os.path.split(ulfile)[0]
286          self.lineEditVal.setText(fichier)
287          self.editor.afficheCommentaire(tr("Fichier selectionne"))
288          self.LEvaleurPressed()
289          if (QFileInfo(fichier).suffix() in listeSuffixe ):
290              self.image=fichier
291              if (not hasattr(self,"BSelectInFile")):
292                try :
293                  self.BSelectInFile = QPushButton(self)
294                  self.BSelectInFile.setMinimumSize(QSize(140,40))
295                  self.BSelectInFile.setObjectName("BSelectInFile")
296                  self.gridLayout.addWidget(self.BSelectInFile,1,1,1,1)
297                  self.BSelectInFile.setText(tr("Selection"))
298                  self.BSelectInFile.clicked.connect(self.BSelectInFilePressed)
299                except :
300                 pass
301              else :
302                self.BSelectInFile.setVisible(1)
303                
304          elif hasattr(self, "BSelectInFile"):
305              self.BSelectInFile.setVisible(0)
306
307   def BFichierOuRepertoirePressed(self):
308       self.fileName=""
309       self.file_dialog=QFileDialog()
310       self.file_dialog.setFileMode(QFileDialog.Directory);
311       self.file_dialog.setFileMode(QFileDialog.Directory|QFileDialog.ExistingFiles)
312       self.file_dialog.setOption(QFileDialog.DontUseNativeDialog,True);
313       self.file_dialog.setWindowTitle('Choose File or Directory')
314       self.explore(self.file_dialog)
315       self.file_dialog.exec_()
316       if self.fileName == "" : return
317       self.lineEditVal.setText(self.fileName)
318       self.LEvaleurPressed()
319      
320
321   def explore(self,widget):
322       for c in widget.children() :
323           if isinstance(c, QTreeView) :
324              c.clicked.connect (self.changeBoutonOpen)
325              self.monTreeView=c
326           try :
327              if c.text() == "&Open" : self.monBoutonOpen=c
328           except : pass
329           self.explore(c) 
330
331
332   def changeBoutonOpen(self):
333       self.monBoutonOpen.setEnabled(True)
334       self.monBoutonOpen.setText("Choose")
335       self.monBoutonOpen.clicked.connect(self.monBoutonOpenClicked)
336       index = self.monTreeView.currentIndex();
337       self.fileName2 = self.monTreeView.model().data(index)
338
339   def monBoutonOpenClicked(self):
340       try :
341         self.fileName=self.file_dialog.selectedFiles()[0]
342       except :
343          self.fileName=self.file_dialog.directory().absolutePath()
344       self.file_dialog.close()
345       self.file_dialog=None
346
347   def BRepertoirePressed(self):
348       directory = QFileDialog.getExistingDirectory(self.appliEficas,
349             directory = self.appliEficas.maConfiguration.savedir,
350             options = QFileDialog.ShowDirsOnly)
351
352       if not (directory == "") :
353          absdir = os.path.abspath(six.text_type(directory))
354          self.appliEficas.maConfiguration.savedir = os.path.dirname(absdir)
355          self.lineEditVal.setText(directory)
356          self.LEvaleurPressed()
357
358   def BSelectInFilePressed(self):
359       from monSelectImage import MonSelectImage
360       MonSelectImage(file=self.image,parent=self).show()
361
362           
363
364   def BSalomePressed(self):
365         self.editor.afficheCommentaire("")
366         selection=[]
367         commentaire=""
368         genea=self.node.item.getGenealogie()
369         kwType = self.node.item.get_definition().type[0]
370         for e in genea:
371             if "GROUP_NO" in e: kwType = "GROUP_NO"
372             if "GROUP_MA" in e: kwType = "GROUP_MA"
373
374         if 'grno' in repr(kwType): kwType = "GROUP_NO"
375         if 'grma' in repr(kwType): kwType = "GROUP_MA"
376
377         if kwType in ("GROUP_NO","GROUP_MA"):
378            selection, commentaire = self.appliEficas.selectGroupFromSalome(kwType,editor=self.editor)
379
380         mc = self.node.item.get_definition()
381
382         if  (isinstance(mc.type,tuple) and len(mc.type) > 1 and "(*.med)" in mc.type[1] ):
383            selection, commentaire = self.appliEficas.selectMeshFile(editor=self.editor)
384            #print selection, commentaire
385            if commentaire != "" : 
386                   QMessageBox.warning( None,
387                   tr("Export Med vers Fichier "),
388                   tr("Impossibilite d exporter le Fichier"),)
389                   return
390            else :
391                   self.lineEditVal.setText(str(selection))
392                   return
393
394         from Accas import SalomeEntry
395         if inspect.isclass(kwType) and issubclass(kwType, SalomeEntry):
396            selection, commentaire = self.appliEficas.selectEntryFromSalome(kwType,editor=self.editor)
397
398         if commentaire !="" :
399             self.editor.afficheInfos(tr(str(commentaire)))
400         if selection == [] : return
401
402         min,max=self.node.item.getMinMax()
403         if max > 1 : 
404            self.ajoutNValeur(selection)
405            return
406
407         monTexte=""
408         for geomElt in selection: monTexte=geomElt+","
409         monTexte= monTexte[0:-1]
410         self.lineEditVal.setText(str(monTexte))
411         self.LEvaleurPressed()
412
413   def BView2DPressed(self):
414         try :
415           # cas d un Simp de base
416           valeur=self.lineEditVal.text()
417         except :
418           valeur=self.textSelected
419         valeur = str(valeur)
420         if valeur == str("") : return
421         if valeur :
422            ok, msgError = self.appliEficas.displayShape(valeur)
423            if not ok:
424               self.editor.afficheInfos(msgError,Qt.red)
425
426   def BParametresPressed(self):
427         liste=self.node.item.getListeParamPossible()
428         from monListeParamPanel import MonListeParamPanel
429         MonListeParamPanel(liste=liste,parent=self).show()
430