PAL8616: the function Group should allow geometric shapes

This commit is contained in:
jfa 2005-04-15 09:11:26 +00:00
parent 14b3cb85a6
commit 078934d597

View File

@ -408,11 +408,23 @@ class Mesh:
Create a mesh group based on geometric object \a grp Create a mesh group based on geometric object \a grp
and give a \a name, if this parameter is not defined and give a \a name, if this parameter is not defined
the name is the same as the geometric group name the name is the same as the geometric group name
\param grp is a geometric group \param grp is a geometric group, a vertex, an edge, a face or a solid
\param name is the name of the mesh group \param name is the name of the mesh group
""" """
if name == "": if name == "":
name = grp.GetName() name = grp.GetName()
type = []
tgeo = str(grp.GetShapeType())
if tgeo == "VERTEX":
type = SMESH.NODE
elif tgeo == "EDGE":
type = SMESH.EDGE
elif tgeo == "FACE":
type = SMESH.FACE
elif tgeo == "SOLID":
type = SMESH.VOLUME
elif tgeo == "COMPOUND":
tgeo = geompy.GetType(grp) tgeo = geompy.GetType(grp)
if tgeo == geompy.ShapeType["VERTEX"]: if tgeo == geompy.ShapeType["VERTEX"]:
type = SMESH.NODE type = SMESH.NODE
@ -422,6 +434,11 @@ class Mesh:
type = SMESH.FACE type = SMESH.FACE
elif tgeo == geompy.ShapeType["SOLID"]: elif tgeo == geompy.ShapeType["SOLID"]:
type = SMESH.VOLUME type = SMESH.VOLUME
if type == []:
print "Mesh.Group: bad first argument: expected a group, a vertex, an edge, a face or a solid"
return 0
else:
return self.mesh.CreateGroupFromGEOM(type, name, grp) return self.mesh.CreateGroupFromGEOM(type, name, grp)
def ExportToMED(self, f, version, opt=0): def ExportToMED(self, f, version, opt=0):