From 0b147a12bd9b9eeb69ef05c243dd51d8fa40d9d1 Mon Sep 17 00:00:00 2001
From: kliment <kliment.yanev@gmail.com>
Date: Tue, 21 Jun 2011 18:52:11 +0200
Subject: [PATCH] Added very basic gcode visualization

---
 gviz.py        | 110 +++++++++++++++++++++++++++++++++++++++++++++++++
 pronterface.py |  17 ++++++--
 2 files changed, 124 insertions(+), 3 deletions(-)
 create mode 100644 gviz.py

diff --git a/gviz.py b/gviz.py
new file mode 100644
index 0000000..7dec633
--- /dev/null
+++ b/gviz.py
@@ -0,0 +1,110 @@
+import wx,time
+
+class window(wx.Frame):
+    def __init__(self):
+        wx.Frame.__init__(self,None,title="Slicetest",size=(200,200))
+        self.p=gviz(self,size=(200,200),bedsize=(200,200))
+        s=time.time()
+        for i in open("/home/kliment/designs/spinner/gearend_export.gcode"):
+            self.p.addgcode(i)
+        print time.time()-s
+        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)):
+        wx.Panel.__init__(self,parent,-1,size=size)
+        self.bedsize=bedsize
+        self.lastpos=[0,0,0,0,0]
+        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))
+        self.fades=[wx.Pen(wx.Colour(150+20*i,150+20*i,150+20*i)) for i in xrange(6)]
+        self.showall=0
+        
+    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()
+        
+    def paint(self,event):
+        dc=wx.PaintDC(self)
+        dc.SetBackground(wx.Brush((250,250,200)))
+        dc.Clear()
+        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]])
+        del dc
+        
+    def showall(self,v):
+        self.showall=v
+        self.Refresh()
+        
+    def addgcode(self,gcode="M105"):
+        if "g1" in gcode.lower():
+            gcode=gcode.lower().split()
+            target=self.lastpos[:]
+            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
+            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.lastpos[1],target[0],target[1])]
+            self.pens[target[2]]+=[self.mainpen]
+            self.lastpos=target
+            
+            
+if __name__ == '__main__':
+    app = wx.App(False)
+    main = window()
+    main.Show()
+    app.MainLoop()
+
diff --git a/pronterface.py b/pronterface.py
index 1cd6b64..c8ef471 100644
--- a/pronterface.py
+++ b/pronterface.py
@@ -4,7 +4,7 @@ try:
 except:
     print "WX is not installed. This program requires WX to run."
     raise
-import printcore, os, sys, glob, time, threading, traceback, StringIO
+import printcore, os, sys, glob, time, threading, traceback, StringIO, gviz
 thread=threading.Thread
 winsize=(800,500)
 winssize=(800,120)
@@ -319,8 +319,9 @@ class PronterWindow(wx.Frame,pronsole.pronsole):
         self.zfeedc.Bind(wx.EVT_SPINCTRL,self.setfeeds)
         self.zfeedc.SetBackgroundColour((180,255,180))
         self.zfeedc.SetForegroundColour("black")
-        lls.Add((150,0),pos=(0,12),span=(14,1))
-        
+        lls.Add((10,0),pos=(0,11),span=(1,1))
+        self.gviz=gviz.gviz(self.panel,(200,200),(200,200))
+        lls.Add(self.gviz,pos=(0,10),span=(9,1))
         
         self.uppersizer=wx.BoxSizer(wx.VERTICAL)
         self.uppersizer.Add(self.uppertopsizer)
@@ -417,6 +418,7 @@ class PronterWindow(wx.Frame,pronsole.pronsole):
                 if self.p.printing:
                     string+= " Printing:%04.2f %%"%(100*float(self.p.queueindex)/len(self.p.mainqueue),)
                 wx.CallAfter(self.status.SetStatusText,string)
+                wx.CallAfter(self.gviz.Refresh)
                 if(self.monitor and self.p.online):
                     if self.sdprinting:
                         self.p.send_now("M27")
@@ -538,6 +540,7 @@ class PronterWindow(wx.Frame,pronsole.pronsole):
             self.filename=self.filename.replace(".stl","_export.gcode")
             self.f=[i.replace("\n","").replace("\r","") for i in open(self.filename)]
             wx.CallAfter(self.status.SetStatusText,"Loaded "+self.filename+", %d lines"%(len(self.f),))
+            threading.Thread(target=self.loadviz).start()
         except:
             self.filename=fn
         
@@ -574,6 +577,14 @@ class PronterWindow(wx.Frame,pronsole.pronsole):
                 self.f=[i.replace("\n","").replace("\r","") for i in open(name)]
                 self.filename=name
                 self.status.SetStatusText("Loaded "+name+", %d lines"%(len(self.f),))
+                threading.Thread(target=self.loadviz).start()
+                
+    def loadviz(self):
+        self.gviz.clear()
+        for i in self.f:
+            self.gviz.addgcode(i)
+        self.gviz.showall=1
+        wx.CallAfter(self.gviz.Refresh)
                 
     def printfile(self,event):
         if self.paused: