Skip to content
Fetching contributors…
Cannot retrieve contributors at this time
250 lines (216 sloc) 6.67 KB
import openvoronoi as ovd
import ovdvtk
import time
import vtk
import ttt
import math
def drawLine(myscreen, pt1, pt2, lineColor):
myscreen.addActor( ovdvtk.Line(p1=(pt1.x,pt1.y,0),p2=(pt2.x,pt2.y,0),color=lineColor) )
def drawArc(myscreen, pt1, pt2, r, cen,cw,arcColor):
# draw arc as many line-segments
start = pt1-cen
end = pt2-cen
theta1 = math.atan2(start.x,start.y)
theta2 = math.atan2(end.x,end.y)
alfa=[] # the list of angles
da=0.1
CIRCLE_FUZZ = 1e-9
# idea from emc2 / cutsim g-code interp G2/G3
if (cw == False ):
while ( (theta2 - theta1) > -CIRCLE_FUZZ):
theta2 -= 2*math.pi
else:
while( (theta2 - theta1) < CIRCLE_FUZZ):
theta2 += 2*math.pi
dtheta = theta2-theta1
arclength = r*dtheta
dlength = min(0.01, arclength/10)
steps = int( float(arclength) / float(dlength))
rsteps = float(1)/float(steps)
dc = math.cos(-dtheta*rsteps) # delta-cos
ds = math.sin(-dtheta*rsteps) # delta-sin
previous = pt1
tr = [start.x, start.y]
for i in range(steps):
#f = (i+1) * rsteps #; // varies from 1/rsteps..1 (?)
#theta = theta1 + i* dtheta
tr = rotate(tr[0], tr[1], dc, ds) #; // rotate center-start vector by a small amount
x = cen.x + tr[0]
y = cen.y + tr[1]
current = ovd.Point(x,y)
myscreen.addActor( ovdvtk.Line(p1=(previous.x,previous.y,0),p2=(current.x,current.y,0),color=arcColor) )
previous = current
# rotate by cos/sin. from emc2 gcodemodule.cc
def rotate(x, y, c, s):
tx = x * c - y * s;
y = x * s + y * c;
x = tx;
return [x,y]
def drawOffsets(myscreen, ofs):
# draw loops
nloop = 0
lineColor = ovdvtk.lgreen
arcColor = ovdvtk.green #grass
for lop in ofs:
n = 0
N = len(lop)
first_point=[]
previous=[]
for p in lop:
# p[0] is the Point
# p[1] is -1 for lines, and r for arcs
if n==0: # don't draw anything on the first iteration
previous=p[0]
#first_point = p[0]
else:
cw=p[3]
cen=p[2]
r=p[1]
p=p[0]
if r==-1:
drawLine(myscreen, previous, p, lineColor)
else:
drawArc(myscreen, previous, p, r,cen,cw, arcColor)
#myscreen.addActor( ovdvtk.Line(p1=(previous.x,previous.y,0),p2=(p.x,p.y,0),color=loopColor) )
previous=p
n=n+1
print "rendered loop ",nloop, " with ", len(lop), " points"
nloop = nloop+1
def insert_polygon_points(vd, polygon):
pts=[]
for p in polygon:
pts.append( ovd.Point( p[0], p[1] ) )
id_list = []
print "inserting ",len(pts)," point-sites:"
m=0
for p in pts:
id_list.append( vd.addVertexSite( p ) )
print " ",m," added vertex ", id_list[ len(id_list) -1 ]
m=m+1
return id_list
def insert_polygon_segments(vd,id_list):
j=0
print "inserting ",len(id_list)," line-segments:"
for n in range(len(id_list)):
n_nxt = n+1
if n==(len(id_list)-1):
n_nxt=0
print " ",j,"inserting segement ",id_list[n]," - ",id_list[n_nxt]
vd.addLineSite( id_list[n], id_list[n_nxt])
j=j+1
def insert_many_polygons(vd,segs):
polygon_ids =[]
t_before = time.time()
for poly in segs:
poly_id = insert_polygon_points(vd,poly)
polygon_ids.append(poly_id)
t_after = time.time()
pt_time = t_after-t_before
t_before = time.time()
for ids in polygon_ids:
insert_polygon_segments(vd,ids)
t_after = time.time()
seg_time = t_after-t_before
return [pt_time, seg_time]
def translate(segs,x,y):
out = []
for seg in segs:
seg2 = []
for p in seg:
p2 = []
p2.append(p[0] + x)
p2.append(p[1] + y)
seg2.append(p2)
out.append(seg2)
return out
def ttt_segments(text,scale):
wr = ttt.SEG_Writer()
wr.arc = False
wr.conic = False
wr.cubic = False
wr.scale = float(1)/float(scale)
ttt.ttt(text,wr)
segs = wr.get_segments()
return segs
def modify_segments(segs):
segs_mod =[]
for seg in segs:
first = seg[0]
last = seg[ len(seg)-1 ]
assert( first[0]==last[0] and first[1]==last[1] )
seg.pop()
seg.reverse() # to get interior or exterior offsets
segs_mod.append(seg)
#drawSegment(myscreen, seg)
return segs_mod
if __name__ == "__main__":
#print ocl.revision()
#w=2500
#h=1500
#w=1920
#h=1080
w=1024
h=1024
myscreen = ovdvtk.VTKScreen(width=w, height=h)
ovdvtk.drawOCLtext(myscreen, rev_text=ovd.version() )
scale=1
myscreen.render()
#random.seed(42)
far = 1
camPos = far
zmult = 3
# camPos/float(1000)
myscreen.camera.SetPosition(0, -camPos/float(1000), zmult*camPos)
myscreen.camera.SetClippingRange(-(zmult+1)*camPos,(zmult+1)*camPos)
myscreen.camera.SetFocalPoint(0.0, 0, 0)
vd = ovd.VoronoiDiagram(far,120)
print ovd.version()
# for vtk visualization
vod = ovdvtk.VD(myscreen,vd,float(scale), textscale=0.01, vertexradius=0.003)
vod.drawFarCircle()
vod.textScale = 0.02
vod.vertexRadius = 0.0031
vod.drawVertices=0
vod.drawVertexIndex=0
vod.drawGenerators=0
vod.offsetEdges = 0
vd.setEdgeOffset(0.05)
# segments from ttt
segs = ttt_segments( "E", 20000)
segs = translate(segs, -0.06, 0.05)
segs = modify_segments(segs)
times = insert_many_polygons(vd,segs)
print "all sites inserted. "
print "VD check: ", vd.check()
pi = ovd.PolygonInterior( True )
vd.filter_graph(pi)
of = ovd.FaceOffset( vd.getGraph() ) # pass the created graph to the Offset class
ofs_list=[]
t_before = time.time()
#for t in [0.002*x for x in range(1,10)]:
t=0.005
drawOffsets(myscreen, of.offset(t))
print of.str()
t=0.006
drawOffsets(myscreen, of.offset(t))
print of.str()
t=0.008
drawOffsets(myscreen, of.offset(t))
print of.str()
#ofs_list.append(ofs)
t_after = time.time()
oftime = t_after-t_before
"""
for ofs in ofs_list:
drawOffsets(myscreen, ofs)
"""
oftext = ovdvtk.Text()
oftext.SetPos( (50, 100) )
oftext_text = "Offset in {0:.3f} s CPU time.".format( oftime )
oftext.SetText( oftext_text )
myscreen.addActor(oftext)
vod.setVDText2(times)
vod.setAll()
print "PYTHON All DONE."
myscreen.render()
myscreen.iren.Start()
Something went wrong with that request. Please try again.