2011-06-21 16:52:11 +00:00
|
|
|
import wx,time
|
|
|
|
|
|
|
|
class window(wx.Frame):
|
2011-06-22 12:54:59 +00:00
|
|
|
def __init__(self,f,size=(600,600),bedsize=(200,200)):
|
|
|
|
wx.Frame.__init__(self,None,title="Slicetest",size=(size[0],size[1]))
|
|
|
|
self.p=gviz(self,size=size,bedsize=bedsize)
|
2011-06-21 16:52:11 +00:00
|
|
|
s=time.time()
|
2011-06-22 12:54:59 +00:00
|
|
|
for i in f:
|
2011-06-21 16:52:11 +00:00
|
|
|
self.p.addgcode(i)
|
2011-06-22 12:54:59 +00:00
|
|
|
#print time.time()-s
|
2011-06-21 16:52:11 +00:00
|
|
|
self.p.Bind(wx.EVT_KEY_DOWN,self.key)
|
|
|
|
def key(self, event):
|
|
|
|
x=event.GetKeyCode()
|
|
|
|
#print x
|
|
|
|
if x==wx.WXK_UP:
|
|
|
|
self.p.layerup()
|
|
|
|
if x==wx.WXK_DOWN:
|
|
|
|
self.p.layerdown()
|
|
|
|
|
|
|
|
#print p.lines.keys()
|
|
|
|
|
|
|
|
class gviz(wx.Panel):
|
|
|
|
def __init__(self,parent,size=(200,200),bedsize=(200,200)):
|
2011-06-22 12:54:59 +00:00
|
|
|
wx.Panel.__init__(self,parent,-1,size=(size[0],size[1]))
|
|
|
|
self.size=size
|
2011-06-21 16:52:11 +00:00
|
|
|
self.bedsize=bedsize
|
|
|
|
self.lastpos=[0,0,0,0,0]
|
2011-06-22 11:40:27 +00:00
|
|
|
self.hilightpos=self.lastpos[:]
|
2011-06-21 16:52:11 +00:00
|
|
|
self.Bind(wx.EVT_PAINT,self.paint)
|
|
|
|
self.lines={}
|
|
|
|
self.pens={}
|
|
|
|
self.layers=[]
|
|
|
|
self.layerindex=0
|
|
|
|
self.scale=[min(float(size[0])/bedsize[0],float(size[1])/bedsize[1])]*2
|
|
|
|
self.mainpen=wx.Pen(wx.Colour(0,0,0))
|
2011-06-22 11:40:27 +00:00
|
|
|
self.hlpen=wx.Pen(wx.Colour(200,50,50))
|
2011-06-21 16:52:11 +00:00
|
|
|
self.fades=[wx.Pen(wx.Colour(150+20*i,150+20*i,150+20*i)) for i in xrange(6)]
|
|
|
|
self.showall=0
|
2011-06-22 10:21:04 +00:00
|
|
|
self.hilight=[]
|
2011-06-21 16:52:11 +00:00
|
|
|
|
|
|
|
def clear(self):
|
|
|
|
self.lastpos=[0,0,0,0,0]
|
|
|
|
self.Bind(wx.EVT_PAINT,self.paint)
|
|
|
|
self.lines={}
|
|
|
|
self.pens={}
|
|
|
|
self.layers=[]
|
|
|
|
self.layerindex=0
|
|
|
|
self.showall=0
|
|
|
|
|
|
|
|
def layerup(self):
|
|
|
|
if(self.layerindex+1<len(self.layers)):
|
|
|
|
self.layerindex+=1
|
|
|
|
self.Refresh()
|
|
|
|
|
|
|
|
def layerdown(self):
|
|
|
|
if(self.layerindex>0):
|
|
|
|
self.layerindex-=1
|
|
|
|
self.Refresh()
|
2011-06-22 10:21:04 +00:00
|
|
|
|
|
|
|
def setlayer(self,layer):
|
|
|
|
try:
|
|
|
|
self.layerindex=self.layers.index(layer)
|
|
|
|
wx.CallAfter(self.Refresh)
|
|
|
|
self.showall=0
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
|
2011-06-21 16:52:11 +00:00
|
|
|
|
|
|
|
def paint(self,event):
|
|
|
|
dc=wx.PaintDC(self)
|
|
|
|
dc.SetBackground(wx.Brush((250,250,200)))
|
|
|
|
dc.Clear()
|
2011-06-22 12:54:59 +00:00
|
|
|
dc.SetBrush(wx.Brush((0,0,0)))
|
|
|
|
dc.DrawRectangle(self.size[0]-15,0,15,self.size[1])
|
|
|
|
dc.SetBrush(wx.Brush((0,255,0)))
|
|
|
|
if len(self.layers):
|
|
|
|
dc.DrawRectangle(self.size[0]-14,(1.0-(1.0*self.layerindex)/len(self.layers))*self.size[1],13,self.size[1]-1)
|
2011-06-21 16:52:11 +00:00
|
|
|
if self.showall:
|
|
|
|
l=[]
|
|
|
|
for i in self.layers:
|
|
|
|
dc.DrawLineList(l,self.fades[0])
|
|
|
|
l=map(lambda x:(self.scale[0]*x[0],self.scale[1]*x[1],self.scale[0]*x[2],self.scale[1]*x[3],) ,self.lines[i])
|
|
|
|
dc.DrawLineList(l,self.pens[i])
|
|
|
|
return
|
|
|
|
if self.layerindex<len(self.layers) and self.layers[self.layerindex] in self.lines.keys():
|
|
|
|
for i in range(min(self.layerindex,6))[-6:]:
|
|
|
|
#print i, self.layerindex, self.layerindex-i
|
|
|
|
l=map(lambda x:(self.scale[0]*x[0],self.scale[1]*x[1],self.scale[0]*x[2],self.scale[1]*x[3],) ,self.lines[self.layers[self.layerindex-i-1]])
|
|
|
|
dc.DrawLineList(l,self.fades[i])
|
|
|
|
l=map(lambda x:(self.scale[0]*x[0],self.scale[1]*x[1],self.scale[0]*x[2],self.scale[1]*x[3],) ,self.lines[self.layers[self.layerindex]])
|
|
|
|
dc.DrawLineList(l,self.pens[self.layers[self.layerindex]])
|
2011-06-22 11:40:27 +00:00
|
|
|
l=map(lambda x:(self.scale[0]*x[0],self.scale[1]*x[1],self.scale[0]*x[2],self.scale[1]*x[3],) ,self.hilight)
|
|
|
|
dc.DrawLineList(l,self.hlpen)
|
2011-06-21 16:52:11 +00:00
|
|
|
del dc
|
|
|
|
|
2011-06-22 11:40:27 +00:00
|
|
|
def addgcode(self,gcode="M105",hilight=0):
|
|
|
|
gcode=gcode.split("*")[0]
|
2011-06-21 16:52:11 +00:00
|
|
|
if "g1" in gcode.lower():
|
|
|
|
gcode=gcode.lower().split()
|
|
|
|
target=self.lastpos[:]
|
2011-06-22 11:40:27 +00:00
|
|
|
if hilight:
|
|
|
|
target=self.hilightpos[:]
|
2011-06-21 16:52:11 +00:00
|
|
|
for i in gcode:
|
|
|
|
if i[0]=="x":
|
|
|
|
target[0]=float(i[1:])
|
|
|
|
elif i[0]=="y":
|
|
|
|
target[1]=float(i[1:])
|
|
|
|
elif i[0]=="z":
|
|
|
|
target[2]=float(i[1:])
|
|
|
|
elif i[0]=="e":
|
|
|
|
target[3]=float(i[1:])
|
|
|
|
elif i[0]=="f":
|
|
|
|
target[4]=float(i[1:])
|
|
|
|
#draw line
|
2011-06-22 11:40:27 +00:00
|
|
|
if not hilight:
|
|
|
|
if not target[2] in self.lines.keys():
|
|
|
|
self.lines[target[2]]=[]
|
|
|
|
self.pens[target[2]]=[]
|
|
|
|
self.layers+=[target[2]]
|
|
|
|
self.lines[target[2]]+=[(self.lastpos[0],self.bedsize[1]-self.lastpos[1],target[0],self.bedsize[1]-target[1])]
|
|
|
|
self.pens[target[2]]+=[self.mainpen]
|
|
|
|
self.lastpos=target
|
|
|
|
else:
|
|
|
|
self.hilight+=[(self.hilightpos[0],self.bedsize[1]-self.hilightpos[1],target[0],self.bedsize[1]-target[1])]
|
|
|
|
self.hilightpos=target
|
|
|
|
|
2011-06-21 16:52:11 +00:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
app = wx.App(False)
|
2011-06-22 12:54:59 +00:00
|
|
|
main = window(open("/home/kliment/designs/spinner/gearend_export.gcode"))
|
2011-06-21 16:52:11 +00:00
|
|
|
main.Show()
|
|
|
|
app.MainLoop()
|
|
|
|
|