def findIntersectPoints(curve, surface):
ps = []
tool = GeomAPI_IntCS(curve, surface)
pCount = tool.NbPoints()
for i in range(1, pCount + 1):
ps += [tool.Point(i)]
return ps
def getPntsCurveSurfaceIntersect(curve, surface):
pnts = []
tool = GeomAPI_IntCS(curve, surface)
pCount = tool.NbPoints();
for i in range(pCount):
pnts += [tool.Point(1)]
return pnts
def __init__(self, crv, srf):
super(IntersectCurveSurface, self).__init__(crv, srf)
# Perform
# OCC intersection.
csi = GeomAPI_IntCS(crv.object, srf.object)
results = []
for i in range(1, csi.NbPoints() + 1):
u, v, t = csi.Parameters(i, 0., 0., 0.)
pc = crv.eval(t)
ps = srf.eval(u, v)
pi = mean([pc, ps], axis=0)
pi = Point(*pi)
results.append([(t, u, v), pi])
npts = len(results)
self._set_results(npts, results)
def __init__(self):
plotocc.__init__(self)
self.b1 = self.get_face(gen_ellipsoid(rxyz=[100, 100, 105]))
self.b2 = self.get_face(gen_ellipsoid(rxyz=[210, 210, 210]))
self.beam = gp_Ax3(gp_Pnt(0, 150, 0), gp_Dir(1, 1.5, 0))
print(self.b1)
h_surf = BRep_Tool.Surface(self.b2)
ray = Geom_Line(self.beam.Axis())
self.RayTrace = GeomAPI_IntCS(ray, h_surf)
print(self.RayTrace.NbPoints())
self.num = 0
self.pts = [self.beam.Location()]
for i in range(5):
self.beam = self.reflect_b1(num=1)
self.beam = self.reflect_b1(num=2)
self.beam = self.reflect_b2(num=1)
self.beam = self.reflect_b2(num=2)
def intersections_with_line(self, line):
"""Compute the intersections with a line.
Parameters
----------
line : :class:`~compas.geometry.Line`
Returns
-------
list[:class:`~compas.geometry.Point`]
"""
intersection = GeomAPI_IntCS(Geom_Line(line.to_occ()),
self.occ_surface)
points = []
for index in range(intersection.NbPoints()):
pnt = intersection.Point(index + 1)
point = Point.from_occ(pnt)
points.append(point)
return points
def __init__(self, pnt=gp_Pnt(-10, 0, -10), vec=gp_Vec(10, 20, 100)):
plotocc.__init__(self)
self.solver = ode(self.newton).set_integrator('dopri5')
self.initial_conditions = self.gen_condition(pnt, vec)
self.m = 10.0
self.k = 0.0
ray = spl_2pnt()
px = np.linspace(-1, 1, 50) * 750
py = np.linspace(-1, 1, 50) * 750
mesh = np.meshgrid(px, py)
surf = mesh[0]**2 / 500 + mesh[1]**2 / 750
self.grd_axs = gp_Ax3(gp_Pnt(0, 0, 0), gp_Dir(-0.25, 0, 1))
#self.grd = pln_for_axs(self.grd_axs, [500, 500])
self.grd = spl_face(*mesh, surf)
self.h_surf = BRep_Tool.Surface(self.grd)
self.p_trce = GeomAPI_IntCS(ray, self.h_surf)
self.pts, self.vel = [], []
def reflect_axs2(beam, surf, axs=gp_Ax3(), indx=1):
p0, v0 = beam.Location(), dir_to_vec(beam.Direction())
h_surf = BRep_Tool.Surface(surf)
ray = Geom_Line(gp_Lin(p0, vec_to_dir(v0)))
if GeomAPI_IntCS(ray, h_surf).NbPoints() == 0:
return beam, beam, None
elif GeomAPI_IntCS(ray, h_surf).NbPoints() == 1:
return beam, beam, None
GeomAPI_IntCS(ray, h_surf).IsDone()
u, v, w = GeomAPI_IntCS(ray, h_surf).Parameters(indx)
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
vx.Normalize()
vy.Normalize()
vz.Normalize()
v1 = v0.Mirrored(gp_Ax2(p1, vec_to_dir(vz), vec_to_dir(vx)))
norm_ax = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
beam_ax = gp_Ax3(p1, vec_to_dir(v1), beam.XDirection().Reversed())
return beam_ax, norm_ax, 1
def reflect(beam, face, axs=gp_Ax3()):
p0, v0 = beam.Location(), dir_to_vec(beam.Direction())
h_surf = BRep_Tool.Surface(face)
ray = Geom_Line(gp_Lin(p0, vec_to_dir(v0)))
if GeomAPI_IntCS(ray, h_surf).NbPoints() == 0:
print("Out of Surface", axs.Location())
pln = make_plane(axs.Location(), dir_to_vec(axs.Direction()), 500,
-500, 500, -500)
h_surf = BRep_Tool.Surface(pln)
GeomAPI_IntCS(ray, h_surf).IsDone()
uvw = GeomAPI_IntCS(ray, h_surf).Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
vx.Normalize()
vy.Normalize()
vz.Normalize()
v1 = v0.Mirrored(gp_Ax2(p1, vec_to_dir(vz), vec_to_dir(vx)))
return gp_Ax3(p1, vec_to_dir(v1), beam.XDirection().Reversed())
def Reflect(self):
h_surf = BRep_Tool.Surface(self.tar.srf)
g_line = gp_Lin(self.ini.beam.Location(), self.ini.beam.Direction())
ray = Geom_Line(g_line)
if GeomAPI_IntCS(ray, h_surf).NbPoints():
self.tar.beam = reflect(self.ini.beam, self.tar.srf)
else:
pln = make_plane(self.tar.axs.Location(),
dir_to_vec(self.tar.axs.Direction()), 500, -500,
500, -500)
self.tar.beam = reflect(self.ini.beam, pln)
def __init__(self, theDisplay, sg: Sketch_QTGUI = None):
self.myCoordinateSystem = gp_Ax3(gp.XOY())
self.myContext: AIS_InteractiveContext = theDisplay.Context
self.myDisplay = theDisplay
self.myView: V3d_View = theDisplay.View
self.myGUI = sg
self.myGUI.SetAx3(self.myCoordinateSystem)
self.myGUI.SetContext(self.myContext)
self.myCurrentMethod = Sketch_ObjectTypeOfMethod.Nothing_Method
self.myCurrentDir: gp_Dir = gp.DZ()
self.myTempPnt: gp_Pnt = gp.Origin()
self.myCurrentPnt2d: gp_Pnt2d = gp.Origin2d()
self.myCurrentPlane = Geom_Plane(self.myCoordinateSystem)
self.myCurrentLine = Geom_Line(self.myTempPnt, self.myCurrentDir)
self.myIntCS = GeomAPI_IntCS()
self.PolylineFirstPoint = gp.Origin2d()
self.PolylineFirstPointExist = False
self.myData = []
self.myNode: SketchObjectNode = None
self.myModel = None
self.SnapToGridPoint = False
self.myCommands = []
self.myAnalyserSnap = Sketch_AnalyserSnap(self.myContext, self.myData,
self.myCoordinateSystem)
self.addCommand(Sketch_CommandPoint())
self.addCommand(Sketch_CommandLine2P())
self.addCommand(Sketch_CommandBezierCurve())
self.addCommand(Sketch_CommandArcCenter2P())
self.addCommand(Sketch_CommandCircleCenterRadius())
self.addCommand(Sketch_CommandBSpline())
self.addCommand(Sketch_CommandPointToBSpline())
self.addCommand(Sketch_CommandMoveObject())
self.addCommand(Sketch_CommandNurbCircleSquare())
self.addCommand(Sketch_CommandNurbCircleTriangle())
def Reflect(self, beam=gp_Ax3(), surf=make_plane()):
h_surf = BRep_Tool.Surface(surf)
ray = Geom_Line(beam.Location(), beam.Direction())
if GeomAPI_IntCS(ray, h_surf).NbPoints() == 0:
beam_v1 = beam
else:
GeomAPI_IntCS(ray, h_surf).IsDone()
uvw = GeomAPI_IntCS(ray, h_surf).Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
vx.Normalize()
vy.Normalize()
vz.Normalize()
norm = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
beam_v0 = beam
beam_v0.SetLocation(p1)
beam_v1 = beam_v0.Mirrored(norm.Ax2())
beam_v1.XReverse()
return beam_v1
def reflect(p0, v0, face):
h_surf = BRep_Tool.Surface(face)
ray = Geom_Line(gp_Lin(p0, vec_to_dir(v0)))
uvw = GeomAPI_IntCS(ray.GetHandle(), h_surf).Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
vx.Normalize()
vy.Normalize()
vz.Normalize()
v1 = v0.Mirrored(gp_Ax2(p1, vec_to_dir(vz)))
return p1, v1
def Reflect(self):
h_surf = BRep_Tool.Surface(self.tar.srf)
g_line = gp_Lin(self.ini.beam.Location(), self.ini.beam.Direction())
ray = Geom_Line(g_line)
if GeomAPI_IntCS(ray, h_surf).NbPoints():
self.tar.beam = reflect(self.ini.beam, self.tar.srf)
else:
pln = make_plane(self.tar.axs.Location(),
dir_to_vec(self.tar.axs.Direction()), 500, -500,
500, -500)
h_surf = BRep_Tool.Surface(pln)
self.tar.beam = reflect(self.ini.beam, pln)
print(self.ini.beam.Location())
print(self.tar.beam.Location())
GeomAPI_IntCS(ray, h_surf).IsDone()
uvw = GeomAPI_IntCS(ray, h_surf).Parameters(1)
u, v, w = uvw
print(u, v, w)
vz, v1, v2, r1, r2 = axs_curvature(h_surf, u, v)
tar_surf_axs = gp_Ax3(self.tar.beam.Location(), vec_to_dir(vz),
vec_to_dir(v1))
tar_surf = wavefront([r1, r2], tar_surf_axs)
self.display.DisplayShape(tar_surf, color="BLUE")
self.display.DisplayShape(axs_pln(tar_surf_axs))
self.GO_Prop(w)
h_tar_wave = BRep_Tool.Surface(self.ini_wave)
vz, v1, v2, r1, r2 = axs_curvature(h_tar_wave, u, v)
tar_wave_axs = self.tar.beam.Translated(gp_Vec(0, 0, 0))
tar_wave_axs.SetXDirection(vec_to_dir(v1))
self.tar.wave = wavefront([r1, r2], tar_wave_axs)
self.display.DisplayShape(self.tar.wave, color="RED")
self.display.DisplayShape(axs_pln(tar_wave_axs))
def __init__(self, theDisplay: Viewer3d, statusBar, sg=None):
self.myDisplay = theDisplay
self.myContext: AIS_InteractiveContext = theDisplay.Context
self.myView: V3d_View = theDisplay.View
self.myStatusBar = statusBar
self.myCurrentMethod = Part_ObjectTypeOfMethod.Nothing_Method
self.myIntCS = GeomAPI_IntCS()
self.myNode: SketchNode = None
self.myData = []
self.myCommands = []
self.addCommand(Part_CommandBezierSurface())
self.addCommand(Part_CommandRevolvedSurface())
self.addCommand(Part_CommandExtrudedSurface())
self.addCommand(Part_CommandSweepSurface())
def reflect_beam(self, beam0=gp_Ax3(), tr=0):
v0 = dir_to_vec(beam0.Direction())
v1 = dir_to_vec(beam0.XDirection())
surf = BRep_Tool.Surface(self.surf)
ray = Geom_Line(beam0.Axis())
uvw = GeomAPI_IntCS(ray, surf).Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
if vz.Dot(v0) > 0:
vz.Reverse()
vx.Normalize()
vy.Normalize()
vz.Normalize()
self.beam = gp_Ax3(p1, vec_to_dir(v0.Reversed()),
vec_to_dir(v1.Reversed()))
self.norm = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
if tr == 0:
self.beam.Mirror(self.norm.Ax2())
if self.beam.Direction().Dot(self.norm.Direction()) < 0:
self.beam.ZReverse()
elif tr == 1:
self.beam.ZReverse()
def proj_pnt_pln(self, pnt, surf, axs=gp_Ax3()):
lin = gp_Lin(pnt, axs.Direction())
sxy = GeomAPI_IntCS(Geom_Line(lin), BRep_Tool.Surface(surf)).Point(1)
return sxy
def Prj_pnt_to_face(axs, pnt, face):
lin = gp_Lin(pnt, axs.Direction())
sxy = GeomAPI_IntCS(Geom_Line(lin).GetHandle(),
BRep_Tool.Surface(face)).Point(1)
return sxy
from OCC.Core.gp import gp_Pln
from OCC.Core.BRep import BRep_Tool
from OCC.Core.GeomLProp import GeomLProp_SurfaceTool, GeomLProp_CurveTool
from OCC.Core.GeomAPI import GeomAPI_IntCS, GeomAPI_PointsToBSpline
from OCC.Core.TColgp import TColgp_Array1OfPnt
from OCCUtils.Construct import make_plane
from particle.base import spl_curv_pts
if __name__ == '__main__':
p0 = gp_Pnt(10, 10, 50)
p1 = gp_Pnt(-20, -20, -50)
pts, crv = spl_curv_pts([p0, p1])
pln = make_plane(vec_normal=gp_Vec(0, 1, 1))
h_surf = BRep_Tool.Surface(pln)
api = GeomAPI_IntCS(crv, h_surf)
print(api.IsDone())
print(api.NbSegments())
uvw = api.Parameters(1)
u, v, w = uvw
pnt_crv = gp_Pnt()
pnt_1st = gp_Pnt()
pnt_2nd = gp_Pnt()
pnt_srf = gp_Pnt()
GeomLProp_CurveTool.Value(crv, w, pnt_crv)
GeomLProp_CurveTool.Value(crv, 0, pnt_1st)
GeomLProp_CurveTool.Value(crv, 1, pnt_2nd)
GeomLProp_SurfaceTool.Value(h_surf, u, v, pnt_srf)
print(uvw)
print(pnt_crv)
class DBall(plotocc):
def __init__(self):
plotocc.__init__(self)
self.b1 = self.get_face(gen_ellipsoid(rxyz=[100, 100, 105]))
self.b2 = self.get_face(gen_ellipsoid(rxyz=[210, 210, 210]))
self.beam = gp_Ax3(gp_Pnt(0, 150, 0), gp_Dir(1, 1.5, 0))
print(self.b1)
h_surf = BRep_Tool.Surface(self.b2)
ray = Geom_Line(self.beam.Axis())
self.RayTrace = GeomAPI_IntCS(ray, h_surf)
print(self.RayTrace.NbPoints())
self.num = 0
self.pts = [self.beam.Location()]
for i in range(5):
self.beam = self.reflect_b1(num=1)
self.beam = self.reflect_b1(num=2)
self.beam = self.reflect_b2(num=1)
self.beam = self.reflect_b2(num=2)
def beam_run(self):
for i in range(3):
self.beam = self.reflect_b1()
self.beam = self.reflect_b2()
def get_face(self, sol):
top_api = Topo(sol)
print(top_api.number_of_faces())
for face in top_api.faces():
sol_face = face
return sol_face
def reflect_b1(self, num=1):
h_surf = BRep_Tool.Surface(self.b1)
ray = Geom_Line(self.beam.Axis())
self.RayTrace.Perform(ray, h_surf)
if self.RayTrace.NbPoints() == 0:
beam = self.beam
else:
self.num += 1
uvw = self.RayTrace.Parameters(num)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
norm = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
self.show_axs_pln(norm, scale=10)
beam = self.beam
beam.SetLocation(p1)
beam.SetDirection(beam.Direction().Reversed())
beam.Mirror(norm.Ax2())
print(self.num, self.b1, p1)
self.pts.append(p1)
# self.beam.XReverse()
# self.beam.Mirror(norm.Ax2())
return beam
def reflect_b2(self, num=1):
h_surf = BRep_Tool.Surface(self.b2)
ray = Geom_Line(self.beam.Axis())
self.RayTrace.Perform(ray, h_surf)
if self.RayTrace.NbPoints() == 0:
beam = self.beam
else:
self.num += 1
uvw = self.RayTrace.Parameters(num)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy).Reversed()
norm = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
self.show_axs_pln(norm, scale=10)
beam = self.beam
beam.SetLocation(p1)
beam.SetDirection(beam.Direction().Reversed())
beam.Mirror(norm.Axis())
print(self.num, self.b2, p1)
self.pts.append(p1)
# self.beam.XReverse()
# self.beam.Mirror(norm.Ax2())
return beam
# def reflect(self, p0, v0, face):
# h_surf = BRep_Tool.Surface(face)
# ray = Geom_Line(gp_Lin(p0, vec_to_dir(v0)))
# uvw = GeomAPI_IntCS(ray.GetHandle(), h_surf).Parameters(1)
# u, v, w = uvw
# p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
# GeomLProp_SurfaceTool.D1(h_surf, u, v, p1, vx, vy)
# vz = vx.Crossed(vy)
# vx.Normalize()
# vy.Normalize()
# vz.Normalize()
# v1 = v0.Mirrored(gp_Ax2(p1, vec_to_dir(vz)))
# return p1, v1
def export_file(self):
builder = BRep_Builder()
compound = TopoDS_Compound()
builder.MakeCompound(compound)
builder.Add(compound, self.b1)
builder.Add(compound, self.b2)
builder.Add(compound, make_polygon(self.pts))
self.export_stp(compound)
def display_ball(self):
#self.show_vec(self.beam, scale=50)
self.display.DisplayShape(self.b1, transparency=0.7, color="RED")
self.display.DisplayShape(self.b2, transparency=0.7, color="BLUE")
self.display.DisplayShape(self.beam.Location())
self.show_axs_pln(scale=100)
self.show_axs_pln(self.beam, scale=10)
self.show_pts(self.pts)
self.show()
self.export_file()
def check_ground(self):
if len(self.pts) > 2:
ray = make_edge(self.pts[-2], self.pts[-1])
h_line = BRep_Tool.Curve(ray)
h_surf = BRep_Tool.Surface(self.grd)
print(GeomAPI_IntCS(h_line, h_surf))
class Particle(plotocc):
def __init__(self, pnt=gp_Pnt(-10, 0, -10), vec=gp_Vec(10, 20, 100)):
plotocc.__init__(self)
self.solver = ode(self.newton).set_integrator('dopri5')
self.initial_conditions = self.gen_condition(pnt, vec)
self.m = 10.0
self.k = 0.0
ray = spl_2pnt()
px = np.linspace(-1, 1, 50) * 750
py = np.linspace(-1, 1, 50) * 750
mesh = np.meshgrid(px, py)
surf = mesh[0]**2 / 500 + mesh[1]**2 / 750
self.grd_axs = gp_Ax3(gp_Pnt(0, 0, 0), gp_Dir(-0.25, 0, 1))
#self.grd = pln_for_axs(self.grd_axs, [500, 500])
self.grd = spl_face(*mesh, surf)
self.h_surf = BRep_Tool.Surface(self.grd)
self.p_trce = GeomAPI_IntCS(ray, self.h_surf)
self.pts, self.vel = [], []
def gen_condition(self, pnt=gp_Pnt(), vec=gp_Vec(0, 0, 1)):
x, y, z = pnt.X(), pnt.Y(), pnt.Z()
u, v, w = vec.X(), vec.Y(), vec.Z()
return np.array([x, y, z, u, v, w])
def newton(self, t, Y):
"""
Computes the derivative of the state vector y according to the equation of motion:
Y is the state vector (x, y, z, u, v, w) === (position, velocity).
returns dY/dt.
"""
px, py, pz = Y[0], Y[1], Y[2]
vx, vy, vz = Y[3], Y[4], Y[5]
alpha = -self.k / self.m
ux, uy, uz = alpha * vx, alpha * vy, alpha * vz
#uy += -0.5 * py
#uz -= cnt.g / (pz)
uz -= cnt.g
return np.array([vx, vy, vz, ux, uy, uz])
def compute_trajectory(self, t0=0.0, t1=10.0):
self.r = ode(self.newton).set_integrator('dopri5')
self.r.set_initial_value(self.initial_conditions, t0)
positions = []
self.dt = 0.5
self.pts, self.vel = [], []
while self.r.successful() and self.r.t < t1:
txt = "{:03.2f} / {:03.2f} ".format(self.r.t, t1)
txt += "| {:03.2f} {:03.2f} {:03.2f} ".format(*self.r.y[:3])
txt += "| {:03.2f} {:03.2f} {:03.2f} ".format(*self.r.y[3:])
sys.stdout.write("\r" + txt)
sys.stdout.flush()
self.pts.append(gp_Pnt(*self.r.y[:3]))
self.vel.append(gp_Vec(*self.r.y[3:]))
self.check_ground()
self.r.integrate(self.r.t + self.dt)
print()
poly = make_polygon(self.pts)
self.display.DisplayShape(poly)
for pnt in self.pts[::10]:
self.display.DisplayShape(pnt)
self.display.DisplayShape(self.grd, transparency=0.8, color="BLUE")
def check_ground(self):
if len(self.pts) != 1:
ray = spl_2pnt(self.pts[-2], self.pts[-1])
vec = gp_Vec(self.pts[-2], self.pts[-1])
self.p_trce.Perform(ray, self.h_surf)
if self.p_trce.NbPoints() != 0 and vec.Z() < 0:
print()
uvw = self.p_trce.Parameters(1)
u, v, w = uvw
p0, v0 = gp_Pnt(), gp_Vec()
GeomLProp_CurveTool.D1(ray, w, p0, v0)
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(self.h_surf, u, v, p1, vx, vy)
vz = vx.Crossed(vy)
self.pts[-1] = p0
print(self.r.t, w, p0)
print(v0)
print(gp_Vec(self.pts[-2], self.pts[-1]))
norm = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
v1 = v0
v1.Mirror(norm.Ax2())
self.r.set_initial_value(self.gen_condition(p0, v1),
self.r.t + self.dt)
class HexPlane(plotocc):
def __init__(self):
plotocc.__init__(self)
self.compound = TopoDS_Compound()
self.builder = BRep_Builder()
self.builder.MakeCompound(self.compound)
self.beam = gp_Ax3()
self.beam.SetLocation(gp_Pnt(0.5, 0.5, 0.0))
self.beam.SetDirection(gp_Dir(0.0, 0.5, 1.0))
self.beam_line = line_from_axs(self.beam, length=20)
self.builder.Add(self.compound, self.beam_line)
ax = gp_Ax3(gp_Pnt(0, 0, 10), gp_Dir(0, 0, -1))
px = np.linspace(-1, 1, 10) * 10
py = np.linspace(-1, 1, 10) * 10
mesh = np.meshgrid(px, py)
surf = mesh[0]**2 / 100 + mesh[1]**2 / 150
self.surf = spl_face(*mesh, surf, ax)
self.surf_bound = self.make_PolySurf(radi=5, axs=ax)
self.beam_glin = Geom_Line(self.beam.Location(), self.beam.Direction())
self.ics = GeomAPI_IntCS(self.beam_glin, BRep_Tool.Surface(self.surf))
print(self.ics.NbPoints())
# print(self.ics.Point(1))
self.ics = GeomAPI_IntCS(self.beam_glin,
BRep_Tool.Surface(self.surf_bound))
print(self.ics.NbPoints())
#self.display.DisplayShape(self.surf, transparency=0.7)
self.display.DisplayShape(self.surf_bound, transparency=0.7)
self.plns = TopoDS_Shell()
self.builder.MakeShell(self.plns)
for ix in np.linspace(0, 1, 5):
for iy in np.linspace(0, 1, 5):
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(BRep_Tool.Surface(self.surf), ix, iy,
p1, vx, vy)
vz = vx.Crossed(vy)
axs = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
pln = self.make_PolyPlane(axs=axs, radi=2.5, shft=15.0)
print(pln)
self.builder.Add(self.compound, make_vertex(p1))
self.builder.Add(self.plns, pln)
self.builder.Add(self.compound, self.plns)
for face in Topo(self.plns).faces():
self.ics.Perform(self.beam_glin, BRep_Tool.Surface(face))
uvw = self.ics.Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(BRep_Tool.Surface(face), u, v, p1, vx, vy)
vz = vx.Crossed(vy)
if u > 0 and v > 0:
print(u, v)
print(p1)
print(self.ics.Point(1))
self.display.DisplayShape(p1)
self.display.DisplayShape(face, color="BLUE")
else:
print(u, v)
def make_PolyPlane(self, num=6, radi=1.0, shft=0.0, axs=gp_Ax3()):
lxy = radi - 0.1
pnts = []
angl = 360 / num
for i in range(num):
thet = np.deg2rad(i * angl) + np.deg2rad(shft)
x, y = radi * np.sin(thet), radi * np.cos(thet)
pnts.append(gp_Pnt(x, y, 0))
pnts.append(pnts[0])
poly = make_polygon(pnts)
brep = BRepBuilderAPI_MakeFace(gp_Pln(), poly)
brep.Add(poly)
face = brep.Face()
face.Location(set_loc(gp_Ax3(), axs))
return face
def make_PolySurf(self, num=6, radi=1.0, shft=0.0, axs=gp_Ax3()):
lxy = radi - 0.1
pnts = []
angl = 360 / num
for i in range(num):
thet = np.deg2rad(i * angl) + np.deg2rad(shft)
x, y = radi * np.sin(thet), radi * np.cos(thet)
pnts.append(gp_Pnt(x, y, 0))
pnts.append(pnts[0])
poly = make_polygon(pnts)
proj = BRepProj_Projection(poly, self.surf, gp_Pnt(0, 0, 10))
# print(proj.Current())
brep = BRepBuilderAPI_MakeFace(self.surf, poly)
# brep.Add(poly)
face = brep.Face()
face.Location(set_loc(gp_Ax3(), axs))
return face
def export_file(self):
write_step_file(self.compound, self.tempname + ".stp")
def display_Plane(self):
self.display.DisplayShape(self.beam_line)
self.display.DisplayShape(self.compound)
self.show_axs_pln(scale=1.0)
self.show()
class Sketch(object):
def __init__(self, theDisplay, sg: Sketch_QTGUI = None):
self.myCoordinateSystem = gp_Ax3(gp.XOY())
self.myContext: AIS_InteractiveContext = theDisplay.Context
self.myDisplay = theDisplay
self.myView: V3d_View = theDisplay.View
self.myGUI = sg
self.myGUI.SetAx3(self.myCoordinateSystem)
self.myGUI.SetContext(self.myContext)
self.myCurrentMethod = Sketch_ObjectTypeOfMethod.Nothing_Method
self.myCurrentDir: gp_Dir = gp.DZ()
self.myTempPnt: gp_Pnt = gp.Origin()
self.myCurrentPnt2d: gp_Pnt2d = gp.Origin2d()
self.myCurrentPlane = Geom_Plane(self.myCoordinateSystem)
self.myCurrentLine = Geom_Line(self.myTempPnt, self.myCurrentDir)
self.myIntCS = GeomAPI_IntCS()
self.PolylineFirstPoint = gp.Origin2d()
self.PolylineFirstPointExist = False
self.myData = []
self.myNode: SketchObjectNode = None
self.myModel = None
self.SnapToGridPoint = False
self.myCommands = []
self.myAnalyserSnap = Sketch_AnalyserSnap(self.myContext, self.myData,
self.myCoordinateSystem)
self.addCommand(Sketch_CommandPoint())
self.addCommand(Sketch_CommandLine2P())
self.addCommand(Sketch_CommandBezierCurve())
self.addCommand(Sketch_CommandArcCenter2P())
self.addCommand(Sketch_CommandCircleCenterRadius())
self.addCommand(Sketch_CommandBSpline())
self.addCommand(Sketch_CommandPointToBSpline())
self.addCommand(Sketch_CommandMoveObject())
self.addCommand(Sketch_CommandNurbCircleSquare())
self.addCommand(Sketch_CommandNurbCircleTriangle())
def SetContext(self, theContext: AIS_InteractiveContext):
self.myContext = theContext
self.myAnalyserSnap.SetContext(self.myContext)
self.myGUI.SetContext(theContext)
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetContext(self.myContext)
def SetData(self, thedata: list):
self.myData = thedata
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetData(self.myData)
def SetRootNode(self, theNode):
self.myNode = theNode
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetRootNode(self.myNode)
def SetModel(self, theModel):
self.myModel = theModel
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetModel(self.myModel)
def GetData(self):
return self.myData
def SetCoordinateSystem(self, theCS: gp_Ax3):
self.myCoordinateSystem = theCS
self.myCurrentPlane.SetPosition(self.myCoordinateSystem)
self.myAnalyserSnap.SetAx3(self.myCoordinateSystem)
self.myGUI.SetAx3(self.myCoordinateSystem)
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetAx3(self.myCoordinateSystem)
def GetCoordinateSystem(self):
return self.myCoordinateSystem
def SetPrecise(self, aPrecise: float):
if aPrecise > 0:
self.myAnalyserSnap.SetMinDistance(aPrecise)
def SetColor(self, theColor: Quantity_Color):
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetColor(theColor)
def SetType(self, theType: Sketch_ObjectType):
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetType(theType)
def SetStyle(self, theLineStyle: Prs3d_LineAspect):
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
self.CurCommand.SetStyle(theLineStyle)
def ObjectAction(self, theMethod: Sketch_ObjectTypeOfMethod):
self.myCurrentMethod = theMethod
self.SelectCurCommand()
self.CurCommand.Action()
if (self.myCurrentMethod == Sketch_ObjectTypeOfMethod.Line2P_Method
or self.myCurrentMethod
== Sketch_ObjectTypeOfMethod.ArcCenter2P_Method
) and self.PolylineFirstPointExist:
self.CurCommand.SetPolylineFirstPnt(self.PolylineFirstPoint)
else:
self.PolylineFirstPointExist = False
def GetStatus(self):
return self.myCurrentMethod
def ProjectPointOnPlane(self, v3dX, v3dY, v3dZ, projVx, projVy, projVz):
self.myTempPnt.SetCoord(v3dX, v3dY, v3dZ)
self.myCurrentDir.SetCoord(projVx, projVy, projVz)
self.myCurrentLine.SetDirection(self.myCurrentDir)
self.myCurrentLine.SetLocation(self.myTempPnt)
# pnt=self.myCurrentPlane.Pln().Location ()
# print(pnt.X(),pnt.Y(),pnt.Z())
self.myIntCS.Perform(
self.myCurrentLine,
self.myCurrentPlane) # perfrom intersection calculation
if self.myIntCS.NbPoints() >= 1:
self.myTempPnt = self.myIntCS.Point(1)
self.myCurrentPnt2d.SetX(
(self.myTempPnt.X() - self.myCoordinateSystem.Location().X()) *
self.myCoordinateSystem.XDirection().X() +
(self.myTempPnt.Y() - self.myCoordinateSystem.Location().Y()) *
self.myCoordinateSystem.XDirection().Y() +
(self.myTempPnt.Z() - self.myCoordinateSystem.Location().Z()) *
self.myCoordinateSystem.XDirection().Z())
self.myCurrentPnt2d.SetY(
(self.myTempPnt.X() - self.myCoordinateSystem.Location().X()) *
self.myCoordinateSystem.YDirection().X() +
(self.myTempPnt.Y() - self.myCoordinateSystem.Location().Y()) *
self.myCoordinateSystem.YDirection().Y() +
(self.myTempPnt.Z() - self.myCoordinateSystem.Location().Z()) *
self.myCoordinateSystem.YDirection().Z())
return True
else:
return False
def OnMouseInputEvent(self, *kargs):
theX, theY, buttons, modifier = kargs
aView: V3d_View = self.myView
v3dX, v3dY, v3dZ, projVx, projVy, projVz = aView.ConvertWithProj(
theX, theY)
if self.SnapToGridPoint:
# grid echo
v3dX, v3dY, v3dZ = self.myView.ConvertToGrid(theX, theY)
self.myDisplay.Viewer.ShowGridEcho(
self.myView, Graphic3d_Vertex(v3dX, v3dY, v3dZ))
self.myDisplay.Viewer.SetGridEcho(True)
if self.ProjectPointOnPlane(v3dX, v3dY, v3dZ, projVx, projVy, projVz):
self.SelectCurCommand()
if self.CurCommand.MouseInputEvent(self.myCurrentPnt2d, buttons,
modifier):
self.myCurrentMethod = Sketch_ObjectTypeOfMethod.Nothing_Method
def OnMouseMoveEvent(self, *kargs):
theX, theY, buttons, modifier = kargs
aView: V3d_View = self.myView
v3dX, v3dY, v3dZ, projVx, projVy, projVz = aView.ConvertWithProj(
theX, theY)
if self.SnapToGridPoint:
# grid echo
v3dX, v3dY, v3dZ = self.myView.ConvertToGrid(theX, theY)
self.myDisplay.Viewer.ShowGridEcho(
self.myView, Graphic3d_Vertex(v3dX, v3dY, v3dZ))
self.myDisplay.Viewer.SetGridEcho(True)
if self.ProjectPointOnPlane(v3dX, v3dY, v3dZ, projVx, projVy, projVz):
self.SelectCurCommand()
self.CurCommand.MouseMoveEvent(self.myCurrentPnt2d, buttons,
modifier)
def OnMouseReleaseEvent(self, *kargs):
buttons, modifier = kargs
self.SelectCurCommand()
self.CurCommand.MouseReleaseEvent(buttons, modifier)
def GetCurPoint2D(self):
return self.myCurrentPnt2d
def GetCurPoint3D(self):
return elclib.To3d(self.myCoordinateSystem.Ax2(), self.myCurrentPnt2d)
def SetPickTolerance(self, thePrecision):
# selector
selector_manager = self.myContext.MainSelector()
# self._display.Context.SetPixelTolerance(5)
selector_manager.SetPixelTolerance(thePrecision)
print(selector_manager.PixelTolerance())
def OnCancel(self):
self.SelectCurCommand()
self.myAnalyserSnap.Cancel()
if (self.myCurrentMethod == Sketch_ObjectTypeOfMethod.Line2P_Method
and self.myCurrentMethod
== Sketch_ObjectTypeOfMethod.ArcCenter2P_Method):
self.PolylineFirstPointExist = self.CurCommand.GetPolylineFirstPnt(
self.PolylineFirstPoint)
self.CurCommand.CancelEvent()
self.myCurrentMethod = Sketch_ObjectTypeOfMethod.Nothing_Method
# Acitivate selection automaticlly
# self.myContext.SetAutoActivateSelection(True)
# self.SetPickTolerance(20)
# self.myContext.ActivatedModes()
# # for all the sketch object selectable
# self.myContext.Deactivate()
self.myContext.Activate(0)
def DeleteSelectedObject(self):
index = 0
while index < self.myNode.childCount():
child = self.myNode.child(index)
assert isinstance(child, SketchObjectNode)
myCurObject: Sketch_Geometry = child.getSketchObject()
if self.myContext.IsSelected(myCurObject.GetAIS_Object()):
myCurObject.RemoveDisplay()
self.myNode.removeChild(index)
else:
index += 1
def ViewProperties(self):
# for idx in range(len(self.myData)):
# myCurObject: Sketch_Object = self.myData[idx]
# if self.myContext.IsSelected(myCurObject.GetAIS_Object()):
# self.myContext.ClearSelected(True)
# self.myGUI.SetSketch_Object(myCurObject)
# break
for child in self.myNode.children():
assert isinstance(child, SketchObjectNode)
myCurObject: Sketch_Geometry = child.getSketchObject()
if self.myContext.IsSelected(myCurObject.GetAIS_Object()):
self.myContext.ClearSelected(True)
self.myGUI.SetSketch_Object(myCurObject)
break
def RedrawAll(self):
for idx in range(len(self.myData)):
myCurObject: Sketch_Object = self.myData[idx]
if self.myContext.IsSelected(myCurObject.GetAIS_Object()):
self.myContext.Display(myCurObject.GetAIS_Object(), True)
def SetPolylineMode(self, amode):
for idx in range(len(self.myCommands)):
self.CurCommand = self.myCommands[idx]
self.CurCommand.SetPolylineMode(amode)
def SetSnap(self, theSnap):
self.myAnalyserSnap.SetSnapType(theSnap)
def GetSnap(self):
return self.myAnalyserSnap.GetSnapType()
def addCommand(self, theCommand: Sketch_Command):
theCommand.SetData(self.myData)
theCommand.SetContext(self.myContext)
theCommand.SetRootNode(self.myNode)
theCommand.SetAnalyserSnap(self.myAnalyserSnap)
theCommand.SetAx3(self.myCoordinateSystem)
self.myCommands.append(theCommand)
def SelectCurCommand(self):
for idx in range(len(self.myCommands)):
self.CurCommand: Sketch_Command = self.myCommands[idx]
if self.CurCommand.GetTypeOfMethod() == self.myCurrentMethod:
# Acitivate selection automaticlly
# This is a bug!! Remove the following line will lead to critical error when enable sketch command.
self.myContext.SetAutoActivateSelection(False)
break
points2 = []
points2.append(Point(-4, 0, 2))
points2.append(Point(-2, 2, 0))
points2.append(Point(2, 3, -1))
points2.append(Point(3, 7, -2))
points2.append(Point(4, 9, -1))
spline2 = BSplineCurve.from_points(points2)
surface = BSplineSurface.from_fill(spline1, spline2)
line = Geom_Line(gp_Pnt(0, 4, 0), gp_Dir(0, 0, 1))
# ==============================================================================
# Intersection
# ==============================================================================
intersection = GeomAPI_IntCS(line, surface.occ_surface)
pnt = intersection.Point(1)
point = Point.from_occ(pnt)
# ==============================================================================
# Viz
# ==============================================================================
mesh = surface.to_vizmesh()
boundary = Polyline(
mesh.vertices_attributes('xyz', keys=mesh.vertices_on_boundary()))
view = App()
view.add(mesh)
view.add(boundary, linewidth=2)
def __init__(self):
plotocc.__init__(self)
self.compound = TopoDS_Compound()
self.builder = BRep_Builder()
self.builder.MakeCompound(self.compound)
self.beam = gp_Ax3()
self.beam.SetLocation(gp_Pnt(0.5, 0.5, 0.0))
self.beam.SetDirection(gp_Dir(0.0, 0.5, 1.0))
self.beam_line = line_from_axs(self.beam, length=20)
self.builder.Add(self.compound, self.beam_line)
ax = gp_Ax3(gp_Pnt(0, 0, 10), gp_Dir(0, 0, -1))
px = np.linspace(-1, 1, 10) * 10
py = np.linspace(-1, 1, 10) * 10
mesh = np.meshgrid(px, py)
surf = mesh[0]**2 / 100 + mesh[1]**2 / 150
self.surf = spl_face(*mesh, surf, ax)
self.surf_bound = self.make_PolySurf(radi=5, axs=ax)
self.beam_glin = Geom_Line(self.beam.Location(), self.beam.Direction())
self.ics = GeomAPI_IntCS(self.beam_glin, BRep_Tool.Surface(self.surf))
print(self.ics.NbPoints())
# print(self.ics.Point(1))
self.ics = GeomAPI_IntCS(self.beam_glin,
BRep_Tool.Surface(self.surf_bound))
print(self.ics.NbPoints())
#self.display.DisplayShape(self.surf, transparency=0.7)
self.display.DisplayShape(self.surf_bound, transparency=0.7)
self.plns = TopoDS_Shell()
self.builder.MakeShell(self.plns)
for ix in np.linspace(0, 1, 5):
for iy in np.linspace(0, 1, 5):
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(BRep_Tool.Surface(self.surf), ix, iy,
p1, vx, vy)
vz = vx.Crossed(vy)
axs = gp_Ax3(p1, vec_to_dir(vz), vec_to_dir(vx))
pln = self.make_PolyPlane(axs=axs, radi=2.5, shft=15.0)
print(pln)
self.builder.Add(self.compound, make_vertex(p1))
self.builder.Add(self.plns, pln)
self.builder.Add(self.compound, self.plns)
for face in Topo(self.plns).faces():
self.ics.Perform(self.beam_glin, BRep_Tool.Surface(face))
uvw = self.ics.Parameters(1)
u, v, w = uvw
p1, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
GeomLProp_SurfaceTool.D1(BRep_Tool.Surface(face), u, v, p1, vx, vy)
vz = vx.Crossed(vy)
if u > 0 and v > 0:
print(u, v)
print(p1)
print(self.ics.Point(1))
self.display.DisplayShape(p1)
self.display.DisplayShape(face, color="BLUE")
else:
print(u, v)
# obj.show_ball()
axs = gp_Ax3(gp_Pnt(0, 0, 0), gp_Dir(1, 1, 1))
elp = gen_ellipsoid(axs, [10, 20, 30])
obj.display.DisplayShape(elp, transparency=0.7, color="BLUE")
obj.show_axs_pln(axs, scale=20)
axs = gp_Ax3(gp_Pnt(30, 0, 0), gp_Dir(1, 1, 0))
elp = gen_ellipsoid(axs, [10, 20, 30])
obj.display.DisplayShape(elp, transparency=0.7, color="BLUE")
obj.show_axs_pln(axs, scale=20)
#elp = gen_ellipsoid_geom(axs, [10, 20, 30])
top_api = Topo(elp)
print(top_api.number_of_faces())
for face in top_api.faces():
elp_face = face
print(elp_face)
elp_surf = BRep_Tool_Surface(elp_face)
print(elp_surf)
lin = Geom_Line(gp_Ax1(axs.Location(), gp_Dir(0, 1, 1)))
api = GeomAPI_IntCS(lin, elp_surf)
obj.display.DisplayShape(lin)
print(api.Point(1))
print(api.Point(2))
obj.show()
surf = BRep_Tool.Surface(face)
surf.Rotate(gp_Ax1(ax0.Location(), ax0.Direction()), np.deg2rad(15))
ax1 = gp_Ax3(gp_Pnt(150, 150, -20),
gp_Dir(0.5, 0.1, 1.0),
gp_Dir(0.0, 0.1, 1.0))
pnt = ax1.Location()
api = GeomAPI_ProjectPointOnSurf(pnt, surf)
print(api.NbPoints(), api.NearestPoint())
for i in range(api.NbPoints()):
idx = i + 1
u, v = api.Parameters(idx)
obj.display.DisplayShape(api.Point(idx))
print(idx, u, v)
print(GeomAPI_IntCS(Geom_Line(ax1.Axis()), surf).NbPoints())
u, v, w = GeomAPI_IntCS(Geom_Line(ax1.Axis()), surf).Parameters(1)
p, vx, vy = gp_Pnt(), gp_Vec(), gp_Vec()
api = GeomLProp_SLProps(surf, u, v, 1, 0.1E-03)
pnt = api.Value()
dst = pnt.Distance(ax1.Location())
ax2 = obj.prop_axs(ax1, dst)
rim_u = surf.UIso(u)
rim_v = surf.VIso(v)
print(v, rim_u.FirstParameter(), rim_u.LastParameter())
print(u, rim_v.FirstParameter(), rim_v.LastParameter())
obj.display.DisplayShape(rim_u, color="BLUE")
obj.display.DisplayShape(rim_v, color="BLUE")
print(api.GaussianCurvature())
print(api.MinCurvature(), api.MeanCurvature(), api.MaxCurvature())
print(dir_to_vec(api.Normal()))
