1 # -*- coding: utf-8 -*-
3 # Copyright (C) 2007-2011 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 This module provides tools to facilitate the use of geom engine and geom
27 GEOM = None # GEOM module is loaded only when needed
29 from salome.kernel.logger import Logger
30 from salome.kernel import termcolor
31 logger = Logger("salome.geom.geomtools", color = termcolor.RED)
33 from salome.kernel.studyedit import getActiveStudyId, getStudyEditor
37 def getGeompy(studyId = None):
39 Return an object behaving exactly like geompy module, except that it is
40 associated with the study `studyId`. If `studyId` is :const:`None`, return
41 a pseudo geompy object for the current study.
43 # We can't use geompy module because it initializes GEOM with
44 # salome.myStudy, which may not exist. So we use this trick to create
45 # a pseudo geompy module.
48 studyId = getActiveStudyId()
49 if not _geompys.has_key(studyId):
51 _geompys[studyId] = salome.lcc.FindOrLoadComponent("FactoryServer",
53 _geompys[studyId].ShapeType = geompyDC.ShapeType
54 _geompys[studyId].GEOM = geompyDC.GEOM
55 _geompys[studyId].kind = geompyDC.kind
56 _geompys[studyId].info = geompyDC.info
57 _geompys[studyId].PackData = geompyDC.PackData
58 _geompys[studyId].ReadTexture = geompyDC.ReadTexture
59 study = salome.myStudyManager.GetStudyByID(studyId)
60 _geompys[studyId].init_geom(study)
61 return _geompys[studyId]
66 This class provides several methods to manipulate geom objects in Salome
67 study. The parameter `studyEditor` defines a
68 :class:`~salome.kernel.studyedit.StudyEditor` object used to access the study. If
69 :const:`None`, the method returns a :class:`~salome.kernel.studyedit.StudyEditor`
70 object on the current study.
74 This instance attribute contains the underlying
75 :class:`~salome.kernel.studyedit.StudyEditor` object. It can be used to access
76 the study but the attribute itself should not be modified.
80 def __init__(self, studyEditor = None):
83 GEOM = __import__("GEOM")
84 if studyEditor is None:
85 studyEditor = getStudyEditor()
86 self.editor = studyEditor
88 def displayShapeByName(self, shapeName, color = None):
90 Display the geometrical shape whose name in the study is `shapeName`.
92 :type shapeName: string
93 :param shapeName: name of the geometrical shape
95 :type color: tuple (triplet)
96 :param color: RGB components of the color of the shape
98 :return: True if the shape was found, False otherwise
100 logger.debug("displayShapeByName in PAL: %s with color %s" %
102 listSO = self.editor.study.FindObjectByName(shapeName, "GEOM")
105 geomObj = self.editor.getOrLoadObject(sObj)
107 shape = geomObj._narrow(GEOM.GEOM_Object)
109 geomgui = salome.ImportComponentGUI("GEOM")
110 geomgui.createAndDisplayGO(entry)
111 geomgui.setDisplayMode(entry, 1)
112 if color is not None:
113 geomgui.setColor(entry, color[0], color[1], color[2])