def heightmap_from_equation(f, x_min=-1, x_max=1, y_min=-1, y_max=1):
""" takes an equation z= f(x,y)
and plot the related point cloud as a bspline surface
"""
print("compute surface")
n = 100
# initialize x axis
step_x = (x_max - x_min) / n
x_ = []
for i in range(n):
x_.append(x_min + i * step_x)
# initialize y axis
step_y = (y_max - y_min) / n
y_ = []
for i in range(n):
y_.append(y_min + i * step_y)
# compute z
array = TColgp_Array2OfPnt(1, len(x_), 1, len(y_))
i = 1
for x in x_:
j = 1
for y in y_:
z = f(x, y)
point_to_add = gp_Pnt(x, y, z)
array.SetValue(i, j, point_to_add)
j += 1
i += 1
print("bspline surface creation")
bspl_surface = GeomAPI_PointsToBSplineSurface(array, 3, 8, GeomAbs_C2,
0.001).Surface()
display.DisplayShape(bspl_surface, update=True)
def surf_spl(px, py, pz, axs=gp_Ax3()):
nx, ny = px.shape
pnt_2d = TColgp_Array2OfPnt(1, nx, 1, ny)
for row in range (pnt_2d.LowerRow(), pnt_2d.UpperRow()+1):
for col in range (pnt_2d.LowerCol(), pnt_2d.UpperCol()+1):
i,j = row-1, col-1
pnt = gp_Pnt (px[i, j], py[i, j], pz[i, j])
pnt_2d.SetValue(row, col, pnt)
curv = GeomAPI_PointsToBSplineSurface(pnt_2d, 3, 8, GeomAbs_G2, 0.001).Surface()
surf = BRepBuilderAPI_MakeFace(curv, 1e-6).Face()
surf = surf_trf (axs, surf)
return surf
def approximate_surface(points=[]):
#just to be safe, let's convert that list to Point objects
if points is not []:
new_points = []
for point in points:
if not isinstance(point, Point):
new_points.append(gp_Pnt(Point(point)._CSFDB_Getgp_Pntcoord()))
else:
new_points.append(point)
points = new_points
points = point_list_2(points)
print "approximate_surface: about to call GeomAPI_PointsToBSplineSurface"
surface = GeomAPI_PointsToBSplineSurface(points)
print "approximate_surface: about to call surface.Surface"
surface1 = surface.Surface()
surface2 = Handle_Geom_Surface(surface1)
print "approximate_surface: about to call BRepBuilderAPI_MakeFace"
face = BRepBuilderAPI_MakeFace(surface2)
#to get a TopoDS_Face, do: face.Face()
my_shape = face.Shape()
print "approximate_surface: returning"
return my_shape
def face(px, py, pz):
nx, ny = px.shape
pnt_2d = TColgp_Array2OfPnt(1, nx + 1, 1, ny + 1)
for row in range(pnt_2d.LowerRow(), pnt_2d.UpperRow()):
for col in range(pnt_2d.LowerCol(), pnt_2d.UpperCol()):
i, j = row - 1, col - 1
pnt = gp_Pnt(px[i, j], py[i, j], pz[i, j])
pnt_2d.SetValue(row, col, pnt)
print(i, j, px[i, j], py[i, j], pz[i, j])
curve = GeomAPI_PointsToBSplineSurface(pnt_2d, 3, 8, GeomAbs_G2,
0.001).Surface()
#surface = BRepBuilderAPI_MakeFace(curve, 1e-6)
#return surface.Face()
return curve
def face():
p1 = gp_Pnt()
p2 = gp_Pnt()
p3 = gp_Pnt()
p4 = gp_Pnt()
p5 = gp_Pnt()
p6 = gp_Pnt()
# The white Face
sphere = gp_Sphere(gp_Ax3(gp_Pnt(0, 0, 0), gp_Dir(1, 0, 0)), 150)
green_face = BRepBuilderAPI_MakeFace(sphere, 0.1, 0.7, 0.2, 0.9)
# The red face
p1.SetCoord(-15, 200, 10)
p2.SetCoord(5, 204, 0)
p3.SetCoord(15, 200, 0)
p4.SetCoord(-15, 20, 15)
p5.SetCoord(-5, 20, 0)
p6.SetCoord(15, 20, 35)
array = TColgp_Array2OfPnt(1, 3, 1, 2)
array.SetValue(1, 1, p1)
array.SetValue(2, 1, p2)
array.SetValue(3, 1, p3)
array.SetValue(1, 2, p4)
array.SetValue(2, 2, p5)
array.SetValue(3, 2, p6)
curve = GeomAPI_PointsToBSplineSurface(array, 3, 8, GeomAbs_C2, 0.001).Surface()
red_face = BRepBuilderAPI_MakeFace(curve, 1e-6)
#The brown face
circle = gp_Circ(gp_Ax2(gp_Pnt(0, 0, 0), gp_Dir(1, 0, 0)), 80)
Edge1 = BRepBuilderAPI_MakeEdge(circle, 0, math.pi)
Edge2 = BRepBuilderAPI_MakeEdge(gp_Pnt(0, 0, -80), gp_Pnt(0, -10, 40))
Edge3 = BRepBuilderAPI_MakeEdge(gp_Pnt(0, -10, 40), gp_Pnt(0, 0, 80))
##TopoDS_Wire YellowWire
MW1 = BRepBuilderAPI_MakeWire(Edge1.Edge(), Edge2.Edge(), Edge3.Edge())
if MW1.IsDone():
yellow_wire = MW1.Wire()
brown_face = BRepBuilderAPI_MakeFace(yellow_wire)
#The pink face
p1.SetCoord(35, -200, 40)
p2.SetCoord(50, -204, 30)
p3.SetCoord(65, -200, 30)
p4.SetCoord(35, -20, 45)
p5.SetCoord(45, -20, 30)
p6.SetCoord(65, -20, 65)
array2 = TColgp_Array2OfPnt(1, 3, 1, 2)
array2.SetValue(1, 1, p1)
array2.SetValue(2, 1, p2)
array2.SetValue(3, 1, p3)
array2.SetValue(1, 2, p4)
array2.SetValue(2, 2, p5)
array2.SetValue(3, 2, p6)
BSplineSurf = GeomAPI_PointsToBSplineSurface(array2, 3, 8, GeomAbs_C2,
0.001)
aFace = BRepBuilderAPI_MakeFace(BSplineSurf.Surface(), 1e-6).Face()
##
##//2d lines
P12d = gp_Pnt2d(0.9, 0.1)
P22d = gp_Pnt2d(0.2, 0.7)
P32d = gp_Pnt2d(0.02, 0.1)
##
line1 = Geom2d_Line(P12d, gp_Dir2d((0.2-0.9), (0.7-0.1)))
line2 = Geom2d_Line(P22d, gp_Dir2d((0.02-0.2), (0.1-0.7)))
line3 = Geom2d_Line(P32d, gp_Dir2d((0.9-0.02), (0.1-0.1)))
##
##//Edges are on the BSpline surface
Edge1 = BRepBuilderAPI_MakeEdge(line1.GetHandle(), BSplineSurf.Surface(),
0, P12d.Distance(P22d)).Edge()
Edge2 = BRepBuilderAPI_MakeEdge(line2.GetHandle(), BSplineSurf.Surface(),
0, P22d.Distance(P32d)).Edge()
Edge3 = BRepBuilderAPI_MakeEdge(line3.GetHandle(), BSplineSurf.Surface(),
0, P32d.Distance(P12d)).Edge()
##
Wire1 = BRepBuilderAPI_MakeWire(Edge1, Edge2, Edge3).Wire()
Wire1.Reverse()
pink_face = BRepBuilderAPI_MakeFace(aFace, Wire1).Face()
breplib_BuildCurves3d(pink_face)
display.DisplayColoredShape(green_face.Face(), 'GREEN')
display.DisplayColoredShape(red_face.Face(), 'RED')
display.DisplayColoredShape(pink_face, Quantity_Color(Quantity_NOC_PINK))
display.DisplayColoredShape(brown_face.Face(), 'BLUE')
display.DisplayColoredShape(yellow_wire, 'YELLOW', update=True)