Salome HOME
80f985933b726d8245dbc8f6d3a03241c8321f4b
[tools/eficas.git] / InterfaceQT4 / compooper.py
1 #-*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2013   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 import os
21 import tempfile
22 from PyQt4.QtGui import QMessageBox, QAction, QApplication, QCursor
23 from PyQt4.QtGui import *
24 from PyQt4.QtCore import Qt, SIGNAL, QVariant
25 from Extensions.i18n import tr
26 from Extensions.eficas_exception import EficasException
27
28 from Editeur     import Objecttreeitem
29 import browser
30 import typeNode
31
32 class Node(browser.JDCNode, typeNode.PopUpMenuNode):
33
34     def __init__(self,treeParent, item):
35         browser.JDCNode.__init__( self, treeParent, item)
36
37     def select(self):
38         browser.JDCNode.select(self)
39         #self.treeParent.tree.openPersistentEditor(self,1)
40         #self.monWidgetNom=self.treeParent.tree.itemWidget(self,1)
41         #self.treeParent.tree.connect(self.monWidgetNom,SIGNAL("returnPressed()"), self.nomme)
42         #if self.item.GetIconName() == "ast-red-square" : self.monWidgetNom.setDisabled(True)
43         #else : self.monWidgetNom.setFocus()  ;self.monWidgetNom.setDisabled(False)
44
45     def nomme(self):
46         nom=str(self.monWidgetNom.text())
47         self.editor.init_modif()
48         test,mess = self.item.nomme_sd(nom)
49         if (test== 0):
50            self.editor.affiche_infos(mess,Qt.red)
51            old=self.item.GetText()
52            self.monWidgetNom.setText(old)
53         else :
54            self.editor.affiche_infos(tr("Nommage du concept effectue"))
55            self.onValid()
56            try :
57                self.editor.panel.LENomConcept.setText(nom)
58            except :
59                pass
60
61     def getPanel( self ):
62         """
63         """
64         from monCommandePanel import MonCommandePanel
65         return MonCommandePanel(self,parent=self.editor)
66
67     def getPanel2(self):
68         from monWidgetCommande import MonWidgetCommande
69         return MonWidgetCommande(self,self.editor,self.item.object)
70
71     def createPopUpMenu(self):
72         typeNode.PopUpMenuNode.createPopUpMenu(self)
73         if ("AFFE_CARA_ELEM" in self.item.get_genealogie()) and self.editor.salome: 
74            self.ViewElt = QAction(tr('View3D'),self.tree)
75            self.tree.connect(self.ViewElt,SIGNAL("activated()"),self.view3D)
76            self.ViewElt.setStatusTip(tr("affiche dans Geom les elements de structure"))
77            self.menu.addAction(self.ViewElt)
78            if self.item.isvalid() :
79               self.ViewElt.setEnabled(1)
80            else:
81               self.ViewElt.setEnabled(0)
82         if  self.item.get_nom() == "DISTRIBUTION" :
83            self.Graphe = QAction(tr('Graphique'),self.tree)
84            self.tree.connect(self.Graphe,SIGNAL("activated()"),self.viewPng)
85            self.Graphe.setStatusTip(tr("affiche la distribution "))
86            self.menu.addAction(self.Graphe)
87            if self.item.isvalid() :
88               self.Graphe.setEnabled(1)
89            else:
90               self.Graphe.setEnabled(0)
91
92     def view3D(self) :
93         from Editeur import TroisDPal
94         troisD=TroisDPal.TroisDPilote(self.item,self.editor.appliEficas)
95         troisD.envoievisu()
96
97     def viewPng(self) :
98         from monPixmap import MonLabelPixmap
99         import generator
100         try:
101             QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
102             g = generator.plugins[self.appliEficas.format_fichier]()
103             g.gener(self.item.object, format='beautifie')
104             stdGener = g.getGenerateur()
105             loi = g.dictMCLois.keys()[0]
106             nomLoi = loi.get_name()
107             (fd, fichier) = tempfile.mkstemp(prefix = "openturns_graph_", suffix = ".png")
108             os.close(fd)
109             chemin = os.path.dirname(fichier)
110             base = os.path.splitext(os.path.basename(fichier))[0]
111             script = stdGener.GraphiquePDF(loi, chemin, base)
112             #print script
113             d = {}
114             exec script in d
115             widgetPng=MonLabelPixmap(self.appliEficas,fichier,nomLoi)
116             os.remove(fichier)
117             QApplication.restoreOverrideCursor()
118             widgetPng.show()
119         except:
120             QApplication.restoreOverrideCursor()
121             QMessageBox.warning(
122                 self.appliEficas,
123                 tr("Erreur interne"),
124                 tr("La PDF de la loi ne peut pas etre affichee."),
125                 tr("&Annuler"))
126
127 class EtapeTreeItem(Objecttreeitem.ObjectTreeItem):
128   """ La classe EtapeTreeItem est un adaptateur des objets ETAPE du noyau
129       Accas. Elle leur permet d'etre affichés comme des noeuds
130       d'un arbre graphique.
131       Cette classe a entre autres deux attributs importants :
132         - _object qui est un pointeur vers l'objet du noyau
133         - object qui pointe vers l'objet auquel sont délégués les
134           appels de méthode et les acces aux attributs
135       Dans le cas d'une ETAPE, _object et object pointent vers le 
136       meme objet.
137   """
138   itemNode=Node
139   
140   def IsExpandable(self):
141       return 1
142
143   def GetIconName(self):
144       """
145       Retourne le nom de l'icone a afficher dans l'arbre
146       Ce nom depend de la validite de l'objet
147       """
148       if not self.object.isactif():
149          return "ast-white-square"
150       elif self.object.isvalid():
151          return "ast-green-square"
152       else:
153          valid=self.valid_child()
154          valid=valid * self.valid_regles("non")
155          if self.reste_val != {}:
156             valid=0
157          if valid==0  :
158             return "ast-red-square"
159          else :
160             try :
161             # on traite ici le cas d include materiau
162             #  print self.object.definition.nom 
163               if  self.object.fichier_ini != self.object.nom_mater :
164                   return "ast-red-square"
165             except :
166               pass
167             return "ast-yellow-square"
168
169   def GetLabelText(self):
170       """ Retourne 3 valeurs :
171       - le texte a afficher dans le noeud représentant l'item
172       - la fonte dans laquelle afficher ce texte
173       - la couleur du texte
174       """
175       if self.object.isactif():
176         # None --> fonte et couleur par défaut
177         return self.labeltext,None,None
178       else:
179         return self.labeltext, None, None #CS_pbruno todo
180       
181   def get_objet(self,name) :
182       for v in self.object.mc_liste:
183           if v.nom == name : return v
184       return None
185       
186   def get_type_sd_prod(self):
187       """
188          Retourne le nom du type du concept résultat de l'étape
189       """
190       sd_prod=self.object.get_type_produit()
191       if sd_prod:
192          return sd_prod.__name__
193       else:
194          return ""
195
196   def additem(self,name,pos):      
197       mcent = self._object.addentite(name,pos)
198       return mcent
199       
200
201   def suppitem(self,item) :
202       # item : item du MOCLE de l'ETAPE a supprimer
203       # item.getObject() = MCSIMP, MCFACT, MCBLOC ou MCList 
204       itemobject=item.getObject()
205       if itemobject.isoblig() :
206           self.appli.affiche_infos(tr('Impossible de supprimer un mot-clef obligatoire '),Qt.red)
207           return 0
208       if self.object.suppentite(itemobject):
209           message = tr("Mot-clef %s supprime " , itemobject.nom)
210           self.appli.affiche_infos(message)
211           return 1
212       else :
213           self.appli.affiche_infos(tr('Pb interne : impossible de supprimer ce mot-clef'),Qt.red)
214           return 0
215
216   def GetText(self):
217       try:
218           return self.object.get_sdname()
219       except:
220           return ''
221
222   def keys(self):
223       keys=self.object.mc_dict.keys()
224       return keys
225
226   def GetSubList(self):
227       """
228          Reactualise la liste des items fils stockes dans self.sublist
229       """
230       if self.isactif():
231          liste=self.object.mc_liste
232       else:
233          liste=[]
234
235       sublist=[None]*len(liste)
236       # suppression des items lies aux objets disparus
237       for item in self.sublist:
238          old_obj=item.getObject()
239          if old_obj in liste:
240             pos=liste.index(old_obj)
241             sublist[pos]=item
242          else:
243             pass # objets supprimes ignores
244
245       # ajout des items lies aux nouveaux objets
246       pos=0
247       for obj in liste:
248          if sublist[pos] is None:
249             # nouvel objet : on cree un nouvel item
250             def setfunction(value, object=obj):
251                 object.setval(value)
252             item = self.make_objecttreeitem(self.appli, obj.nom + " : ", obj, setfunction)
253             sublist[pos]=item
254          pos=pos+1
255
256       self.sublist=sublist
257       return self.sublist
258
259   def isvalid(self):
260       return self.object.isvalid()
261
262   def iscopiable(self):
263       """
264       Retourne 1 si l'objet est copiable, 0 sinon
265       """
266       return 1
267
268   def updateDeplace(self,item):
269       if item.sd and item.sd.nom:
270          self.object.sd=item.sd
271          self.object.sd.nom=item.sd.nom
272
273   def update(self,item):
274       if item.sd and item.sd.nom:
275          self.nomme_sd(item.sd.nom)
276
277   def nomme_sd(self,nom):
278       """ Lance la méthode de nommage de la SD """
279       oldnom=""
280       if self.object.sd != None :
281          oldnom=self.object.sd.nom
282       test,mess= self.object.nomme_sd(nom)
283       if test:self.object.parent.reset_context()
284       if (test and self.appli.dict_reels.has_key(oldnom) ):
285               self.appli.dict_reels[nom]=self.appli.dict_reels[oldnom]
286       return test,mess
287
288   def is_reentrant(self):
289       return self.object.is_reentrant()
290     
291   def get_noms_sd_oper_reentrant(self):
292       return self.object.get_noms_sd_oper_reentrant()
293
294   def get_objet_commentarise(self):
295       """
296           Cette méthode retourne un objet commentarisé
297           représentatif de self.object
298       """
299       # Format de fichier utilisé
300       format=self.appli.appliEficas.format_fichier
301       return self.object.get_objet_commentarise(format)
302
303   def get_objet_commentarise_BAK(self):
304       """
305           Cette méthode retourne un objet commentarisé
306           représentatif de self.object
307       """
308       import generator,string,Accas
309       # Format de fichier utilisé
310       format=self.appli.appliEficas.format_fichier
311       g=generator.plugins[format]()
312       texte_commande = g.gener(self.object,format='beautifie')
313       # Il faut enlever la premiere ligne vide de texte_commande que 
314       # rajoute le generator
315       rebut,texte_commande = string.split(texte_commande,'\n',1)
316       # on construit l'objet COMMANDE_COMM repésentatif de self mais non 
317       # enregistré dans le jdc
318       commande_comment = Accas.COMMANDE_COMM(texte=texte_commande,reg='non',
319                                              parent=self.object.parent)
320       commande_comment.niveau = self.object.niveau
321       commande_comment.jdc = commande_comment.parent = self.object.jdc
322
323       pos=self.object.parent.etapes.index(self.object)
324       parent=self.object.parent
325       self.object.parent.suppentite(self.object)
326       parent.addentite(commande_comment,pos)
327
328       return commande_comment
329
330
331 import Accas
332 treeitem = EtapeTreeItem
333 objet = Accas.ETAPE    
334