def _unify_compound(proto):
builder = BRep_Builder()
comp = TopoDS_Compound()
builder.MakeCompound(comp)
explorer = TopExp_Explorer()
explorer.Init(proto.Shape(), TopAbs_SOLID)
while explorer.More():
builder.Add(comp, _unify_solid(Shape(explorer.Current())).Solid())
explorer.Next()
explorer.Init(proto.Shape(), TopAbs_SHELL, TopAbs_SOLID)
while explorer.More():
builder.Add(comp, _unify_shell(Shape(explorer.Current())).Shell())
explorer.Next()
faces = []
explorer.Init(proto.Shape(), TopAbs_FACE, TopAbs_SHELL)
while explorer.More():
faces.append(Shape(explorer.Current()))
explorer.Next()
faces_new = _unify_faces_array(faces)
for f in faces_new:
builder.Add(comp, f.Shape())
return Shape(comp)
def prop_soild(self, sol=TopoDS_Solid()):
self.sol_builder = TopoDS_Builder()
sol_exp = TopExp_Explorer(sol, TopAbs_FACE)
sol_top = TopologyExplorer(sol)
#print(self.cal_vol(sol), self.base_vol)
print(sol_top.number_of_faces())
self.face_lst = TopTools_ListOfShape()
self.face_cnt = []
self.face_num = 0
self.face_init(sol_exp.Current())
#self.sol_builder.Add(sol, sol_exp.Current())
sol_exp.Next()
while sol_exp.More():
face = sol_exp.Current()
self.face_expand(face)
sol_exp.Next()
if self.file == True:
stp_file = "./shp/shp_{:04d}.stp".format(self.sol_num)
write_step_file(sol, stp_file)
stp_file = "./shp/shp_{:04d}_exp.stp".format(self.sol_num)
new_shpe = TopoDS_Compound()
self.sol_builder.Add(gp_Pnt())
# self.sol_builder.MakeCompSolid(new_shpe)
#write_step_file(new_shpe, stp_file)
# if self.show == True:
# self.display.DisplayShape(self.face_cnt)
"""self.face_init(face)
def objToStr(obj) :
ret = dict()
ret['CLASS'] = str(obj.__class__.__name__)
if isinstance(obj,gp_Pnt):
ret['x,y,z'] = '(' + str(obj.X()) + ',' + str(obj.Y()) + ',' + str(obj.Z()) + ')'
elif isinstance(obj, AIS_Point):
ret['Component().Pnt()'] = obj.Component().Pnt()
elif isinstance(obj, AIS_Line):
pass
elif isinstance(obj, Geom_Point):
ret['Pnt()'] = obj.Pnt()
elif isinstance(obj, AIS_Shape):
ret['Shape()'] = obj.Shape()
elif isinstance(obj, TopoDS_Shape):
ret['Type'] = SHAPE_TYPES[obj.ShapeType()]
exp = TopExp_Explorer(obj, TopAbs_ShapeEnum(TopAbs_ShapeEnum.TopAbs_EDGE))
i = 0
while (exp.More()):
ret['Edge-'+str(i)] = exp.Current().__class__.__name__
i += 1
exp.Next()
exp = TopExp_Explorer(obj, TopAbs_ShapeEnum(TopAbs_ShapeEnum.TopAbs_VERTEX))
i = 0
while (exp.More()):
ret['Vertex-'+str(i)] = exp.Current().__class__.__name__
i += 1
exp.Next()
#todo QuantityColor
elif hasattr(obj,'__dict__'):
return vars(obj)
else:
ret['class'] = ' Unknown structure '
return ret
def topo_explorer(occtopo2explore, topotype2find):
"""
This function explores and fetches the specified topological type from the given OCCtopology.
e.g. find a list of OCCfaces in an OCCcompound.
Parameters
----------
occtopo2explore : OCCtopology
The OCCtopology to be explored.
OCCtopology includes: OCCshape, OCCcompound, OCCcompsolid, OCCsolid, OCCshell, OCCface, OCCwire, OCCedge, OCCvertex
topotype2find : str
The string describing the topology to find.
The strings can be e.g. "compound", "compsolid", "solid", "shell", "face", "wire", "edge", "vertex".
Returns
-------
list of topology : list of OCCtopology
The list of OCCtopology found in the specified OCCtopology.
"""
geom_list = []
if topotype2find == "compound":
shapetype2find_topABS = TopAbs_COMPOUND
if topotype2find == "compsolid":
shapetype2find_topABS = TopAbs_COMPSOLID
if topotype2find == "solid":
shapetype2find_topABS = TopAbs_SOLID
if topotype2find == "shell":
shapetype2find_topABS = TopAbs_SHELL
if topotype2find == "face":
shapetype2find_topABS = TopAbs_FACE
if topotype2find == "wire":
shapetype2find_topABS = TopAbs_WIRE
if topotype2find == "edge":
shapetype2find_topABS = TopAbs_EDGE
if topotype2find == "vertex":
shapetype2find_topABS = TopAbs_VERTEX
ex = TopExp_Explorer(occtopo2explore, shapetype2find_topABS)
while ex.More():
if shapetype2find_topABS == 0:
geom = topods_Compound(ex.Current())
if shapetype2find_topABS == 1:
geom = topods_CompSolid(ex.Current())
if shapetype2find_topABS == 2:
geom = topods_Solid(ex.Current())
if shapetype2find_topABS == 3:
geom = topods_Shell(ex.Current())
if shapetype2find_topABS == 4:
geom = topods_Face(ex.Current())
if shapetype2find_topABS == 5:
geom = topods_Wire(ex.Current())
if shapetype2find_topABS == 6:
geom = topods_Edge(ex.Current())
if shapetype2find_topABS == 7:
geom = topods_Vertex(ex.Current())
geom_list.append(geom)
ex.Next()
return geom_list
def prop_soild(self, sol=TopoDS_Solid()):
sol_exp = TopExp_Explorer(sol, TopAbs_FACE)
sol_top = TopologyExplorer(sol)
print()
print(sol, self.cal_vol(sol))
print(sol_top.number_of_faces())
self.face_init(sol_exp.Current())
sol_exp.Next()
while sol_exp.More():
face = sol_exp.Current()
self.face_expand(face)
sol_exp.Next()
self.tmp_face_n += 1
"""self.face_init(face)
def make_lens(body, face1, face2):
splitter = BOPAlgo_Splitter()
splitter.AddArgument(body)
splitter.AddTool(face1)
splitter.AddTool(face2)
splitter.Perform()
print(splitter.Arguments())
print(splitter.ShapesSD())
exp = TopExp_Explorer(splitter.Shape(), TopAbs_SOLID)
shp = []
while exp.More():
print(exp.Current())
shp.append(exp.Current())
exp.Next()
return shp
def split_edge_with_face(event=None):
display.EraseAll()
p0 = gp_Pnt()
vnorm = gp_Dir(1, 0, 0)
pln = gp_Pln(p0, vnorm)
face = BRepBuilderAPI_MakeFace(pln, -10, 10, -10, 10).Face()
p1 = gp_Pnt(0, 0, 15)
p2 = gp_Pnt(0, 0, -15)
edge = BRepBuilderAPI_MakeEdge(p1, p2).Edge()
# Initialize splitter
splitter = BOPAlgo_Splitter()
# Add the edge as an argument and the face as a tool. This will split
# the edge with the face.
splitter.AddArgument(edge)
splitter.AddTool(face)
splitter.Perform()
edges = []
exp = TopExp_Explorer(splitter.Shape(), TopAbs_EDGE)
while exp.More():
edges.append(exp.Current())
exp.Next()
print('Number of edges in split shape: ', len(edges))
display.DisplayShape(edges[0], color='red')
display.DisplayShape(edges[1], color='green')
display.DisplayShape(edges[2], color='yellow')
display.FitAll()
def prop_solids(self):
sol_exp = TopExp_Explorer(self.splitter.Shape(), TopAbs_SOLID)
self.sol_num = 0
while sol_exp.More():
self.sol_num += 1
self.prop_soild(sol_exp.Current())
sol_exp.Next()
def occ_triangle_mesh(event=None):
#
# Mesh the shape
#
BRepMesh_IncrementalMesh(aShape, 0.1)
builder = BRep_Builder()
Comp = TopoDS_Compound()
builder.MakeCompound(Comp)
ex = TopExp_Explorer(aShape, TopAbs_FACE)
while ex.More():
F = topods_Face(ex.Current())
L = TopLoc_Location()
facing = (BRep_Tool().Triangulation(F, L))
tab = facing.Nodes()
tri = facing.Triangles()
for i in range(1, facing.NbTriangles() + 1):
trian = tri.Value(i)
#print trian
index1, index2, index3 = trian.Get()
for j in range(1, 4):
if j == 1:
M = index1
N = index2
elif j == 2:
N = index3
elif j == 3:
M = index2
ME = BRepBuilderAPI_MakeEdge(tab.Value(M), tab.Value(N))
if ME.IsDone():
builder.Add(Comp, ME.Edge())
ex.Next()
display.DisplayShape(Comp, update=True)
def triangulate_solid(self):
mesh = BRepMesh_IncrementalMesh(self.shape, 0.3, False, 0.5, True)
mesh.Perform()
b = BRep_Tool()
ex = TopExp_Explorer(self.shape, TopAbs_FACE)
faces = []
verts = []
while ex.More():
face = topods_Face(ex.Current())
location = TopLoc_Location()
triang_face = (b.Triangulation(face, location))
if triang_face != None:
tab = triang_face.Nodes()
tri = triang_face.Triangles()
for i in range(1, triang_face.NbTriangles() + 1):
try:
verts.append(list(tab.Value(i).Coord()))
except:
continue
for i in range(1, triang_face.NbTriangles() + 1):
try:
index1, index2, index3 = tri.Value(i).Get()
faces.append([index1 - 1, index2 - 1, index3 - 1])
except:
continue
ex.Next()
return verts, faces
def getShapeItems(shape, topologyType):
items = []
ex = TopExp_Explorer(shape, topologyType)
while ex.More():
items.append(ex.Current())
ex.Next()
return items
def __init__(self):
plotocc.__init__(self)
self.shell = read_step_file(self.tmpdir + "SurfUV.stp")
print(self.shell)
top = TopExp_Explorer(self.shell, TopAbs_FACE)
self.face = top.Current()
print(top.Depth())
print(self.face)
self.surf = BRep_Tool.Surface(self.face)
u0, u1, v0, v1 = shapeanalysis_GetFaceUVBounds(self.face)
print(u0, u1, v0, v1)
sas = ShapeAnalysis_Surface(self.surf)
print(sas.Value(u0, v0))
print(sas.Value(u0, v1))
print(sas.Value(u1, v0))
print(sas.Value(u1, v1))
u = u0
while u <= u1:
v = v0
while v <= v1:
p = sas.Value(u, v)
self.display.DisplayShape(p, update=False)
v += 1 / 3
u += 1 / 4
def generate_face_framework(face, shape_maps):
def generate_wire_framework(wire):
wire_framework = []
ex = TopExp_Explorer(wire, TopAbs_EDGE)
while ex.More():
edge = ex.Current()
edge_info = (wire, edge)
wire_framework.append(edge_info)
ex.Next()
return wire_framework
_, wire_map, _ = shape_maps
wire_frameworks = []
outer_loop, inner_loops = [], []
ex = TopExp_Explorer(face, TopAbs_WIRE)
outer_wire_id = wire_map.FindIndex(shapeanalysis.OuterWire(face))
while ex.More():
wire = ex.Current()
wire_id = wire_map.FindIndex(wire)
if wire_id == outer_wire_id and INCLUDE_OUTER_WIRES:
outer_loop += generate_wire_framework(wire)
elif wire_id != outer_wire_id and INCLUDE_INNER_WIRES:
inner_loops.append(generate_wire_framework(wire))
else:
pass
ex.Next()
wire_frameworks.append(outer_loop)
wire_frameworks += inner_loops
return wire_frameworks, face
def face_mesh_triangle(comp=TopoDS_Shape(), isR=0.1, thA=0.1):
# Mesh the shape
BRepMesh_IncrementalMesh(comp, isR, True, thA, True)
bild1 = BRep_Builder()
comp1 = TopoDS_Compound()
bild1.MakeCompound(comp1)
bt = BRep_Tool()
ex = TopExp_Explorer(comp, TopAbs_FACE)
while ex.More():
face = topods_Face(ex.Current())
location = TopLoc_Location()
facing = bt.Triangulation(face, location)
tab = facing.Nodes()
tri = facing.Triangles()
print(facing.NbTriangles(), facing.NbNodes())
for i in range(1, facing.NbTriangles() + 1):
trian = tri.Value(i)
index1, index2, index3 = trian.Get()
for j in range(1, 4):
if j == 1:
m = index1
n = index2
elif j == 2:
n = index3
elif j == 3:
m = index2
me = BRepBuilderAPI_MakeEdge(tab.Value(m), tab.Value(n))
if me.IsDone():
bild1.Add(comp1, me.Edge())
ex.Next()
return comp1
def loops(self) -> List[BRepLoop]:
loops = []
explorer = TopExp_Explorer(self.shape, TopAbs_WIRE)
while explorer.More():
wire = explorer.Current()
loops.append(BRepLoop(wire))
explorer.Next()
return loops
def get_faces(shp):
ex = TopExp_Explorer(shp, TopAbs_FACE)
seq = []
while ex.More():
s1 = ex.Current()
seq.append(s1)
ex.Next()
return seq
def edges(self) -> List[BRepEdge]:
edges = []
explorer = TopExp_Explorer(self.shape, TopAbs_EDGE)
while explorer.More():
edge = explorer.Current()
edges.append(BRepEdge(edge))
explorer.Next()
return edges
def faces(self) -> List[BRepFace]:
faces = []
explorer = TopExp_Explorer(self.shape, TopAbs_FACE)
while explorer.More():
face = explorer.Current()
faces.append(BRepFace(face))
explorer.Next()
return faces
def reflection_elements(self, getter, topabs):
ret = []
ex = TopExp_Explorer(self.Shape(), topabs)
while ex.More():
obj = getter(ex.Current())
ret.append(Shape(obj))
ex.Next()
return ret
def vertices(self) -> List[BRepVertex]:
vertices = []
explorer = TopExp_Explorer(self.edge, TopAbs_VERTEX)
while explorer.More():
vertex = explorer.Current()
vertices.append(BRepVertex(vertex))
explorer.Next()
return vertices
def GetShapeEdges(shape):
#input opencad TopoDs_shape return lst of TopoDS_edges
edges = []
exp = TopExp_Explorer(shape, OCC.Core.TopAbs.TopAbs_EDGE)
while exp.More():
count = 0
edges.append(exp.Current())
exp.Next()
return edges
def generate_wire_framework(wire):
wire_framework = []
ex = TopExp_Explorer(wire, TopAbs_EDGE)
while ex.More():
edge = ex.Current()
edge_info = (wire, edge)
wire_framework.append(edge_info)
ex.Next()
return wire_framework
def _loop_topo(self,
topologyType: TopAbs_ShapeEnum,
topologicalEntity=None,
topologyTypeToAvoid=None) -> Iterator[Any]:
'''
this could be a faces generator for a python TopoShape class
that way you can just do:
for face in srf.faces:
processFace(face)
'''
topoTypes = {
TopAbs_VERTEX: TopoDS_Vertex,
TopAbs_EDGE: TopoDS_Edge,
TopAbs_FACE: TopoDS_Face,
TopAbs_WIRE: TopoDS_Wire,
TopAbs_SHELL: TopoDS_Shell,
TopAbs_SOLID: TopoDS_Solid,
TopAbs_COMPOUND: TopoDS_Compound,
TopAbs_COMPSOLID: TopoDS_CompSolid
}
topExp = TopExp_Explorer()
if topologyType not in topoTypes.keys():
raise AssertionError("%s not one of %s" %
(topologyType, topoTypes.keys()))
# use self.myShape if nothing is specified
if topologicalEntity is None and topologyTypeToAvoid is None:
topExp.Init(self.myShape, topologyType)
elif topologicalEntity is None and topologyTypeToAvoid is not None:
topExp.Init(self.myShape, topologyType, topologyTypeToAvoid)
elif topologyTypeToAvoid is None:
topExp.Init(topologicalEntity, topologyType)
elif topologyTypeToAvoid:
topExp.Init(topologicalEntity, topologyType, topologyTypeToAvoid)
seq = []
while topExp.More():
current_item = topExp.Current()
topo_to_add = self.topoFactory[topologyType](current_item)
seq.append(topo_to_add)
topExp.Next()
if self.ignore_orientation:
# filter out those entities that share the same TShape
# but do *not* share the same orientation
#filter_orientation_seq = []
filter_orientation_seq: List = []
for i in seq:
_present = False
for j in filter_orientation_seq:
if i.IsSame(j):
_present = True
break
if _present is False:
filter_orientation_seq.append(i)
return iter(filter_orientation_seq)
else:
return iter(seq)
def points_from_edge(edge):
vset = []
exp = TopExp_Explorer(edge, TopAbs_VERTEX)
while exp.More():
s = exp.Current()
exp.Next()
vert = topods.Vertex(s)
vset.append(as_list(vert))
return list(vset)
def color_the_edges(shp, display, color, width):
shapeList = []
Ex = TopExp_Explorer(shp, TopAbs_EDGE)
ctx = display.Context
while Ex.More():
aEdge = topods.Edge(Ex.Current())
# ais_shape = AIS_Shape(aEdge)
# ctx.SetColor(ais_shape, color, True)
# ctx.SetWidth(ais_shape, width, False)
# ctx.Display(ais_shape, False)
Ex.Next()
def show_split_solid(self):
colors = ["BLUE", "RED", "GREEN", "YELLOW", "BLACK", "WHITE"]
num = 0
sol_exp = TopExp_Explorer(self.splitter.Shape(), TopAbs_SOLID)
while sol_exp.More():
num += 1
self.display.DisplayShape(sol_exp.Current(),
color=colors[num % len(colors)],
transparency=0.5)
sol_exp.Next()
self.ShowOCC()
def fileout(self, dirname="./shp/"):
num = 0
stp_file = dirname + "shp_{:04d}.stp".format(num)
write_step_file(self.base, stp_file)
sol_exp = TopExp_Explorer(self.splitter.Shape(), TopAbs_SOLID)
while sol_exp.More():
num += 1
stp_file = dirname + "shp_{:04d}.stp".format(num)
write_step_file(sol_exp.Current(), stp_file)
sol_exp.Next()
def _unify_solid(proto):
mkSolid = BRepBuilderAPI_MakeSolid()
explorer = TopExp_Explorer()
explorer.Init(proto.Shape(), TopAbs_SHELL)
while explorer.More():
mkSolid.Add(_unify_shell(Shape(explorer.Current())).Shell())
explorer.Next()
mkSolid.Build()
return Shape(mkSolid.Shape())
def contourAsPoints(self) -> numpy.array:
# Create the wire
wire = self.countourAsWire()
pts = []
if self.done:
# Loop over the wire to get the edges
edge_explorer = TopExp_Explorer(wire, TopAbs_EDGE)
while edge_explorer.More():
edge = OCC.Core.TopoDS.topods_Edge(edge_explorer.Current())
curve = edge.Value()
# TODO: Get the coordinates from the edge curve
edge_explorer.Next()
return pts
def makeFillets(event=None):
newPrtName = 'bodyWithFillets'
workPart = win.activePart
wrkPrtUID = win.activePartUID
mkFillet = BRepFilletAPI_MakeFillet(workPart)
aEdgeExplorer = TopExp_Explorer(workPart, TopAbs_EDGE)
while aEdgeExplorer.More():
aEdge = topods_Edge(aEdgeExplorer.Current())
mkFillet.Add(thickness / 12., aEdge)
aEdgeExplorer.Next()
myBody = mkFillet.Shape()
win.getNewPartUID(myBody, name=newPrtName, ancestor=wrkPrtUID)
win.statusBar().showMessage('Bottle with fillets complete')
win.redraw()