#!/usr/bin/env python
import wx,time,random,threading,os,math
import stltool

class stlwrap:
	def __init__(self,obj,name=None):
		self.obj=obj
		self.name=name
		if name is None:
			self.name=obj.name
		
	def __repr__(self):
		return self.name
	

class showstl(wx.Window):
	def __init__(self,parent,size,pos):
		wx.Window.__init__(self,parent,size=size,pos=pos)
		self.l=wx.ListCtrl(self,style=wx.LC_LIST,size=(300,130),pos=(0,size[1]-130))
		self.eb=wx.Button(self,label="Export",pos=(300,size[1]-130))
		self.sb=wx.Button(self,label="Snap to Z=0",pos=(300,size[1]-105))
		self.cb=wx.Button(self,label="Put at 100,100",pos=(300,size[1]-80))
		self.db=wx.Button(self,label="Delete",pos=(300,size[1]-55))
		self.ab=wx.Button(self,label="Auto",pos=(300,size[1]-30))
		self.eb.Bind(wx.EVT_BUTTON,self.export)
		self.sb.Bind(wx.EVT_BUTTON,self.snap)
		self.cb.Bind(wx.EVT_BUTTON,self.center)
		self.db.Bind(wx.EVT_BUTTON,self.delete)
		self.ab.Bind(wx.EVT_BUTTON,self.autoplate)
		#self.SetBackgroundColour((0,0,0))
		#wx.FutureCall(200,self.paint)
		self.i=0
		self.previ=0
		self.Bind(wx.EVT_MOUSEWHEEL,self.rot)
		self.Bind(wx.EVT_MOUSE_EVENTS,self.move)
		self.Bind(wx.EVT_PAINT,self.repaint)
		#self.s=stltool.stl("sphere.stl").scale([2,1,1])
		self.triggered=0
		self.models={}
		self.basedir="."
		self.initpos=None
		self.prevsel=-1

	def center(self,event):
		i=self.l.GetFirstSelected()
		if i != -1:
				m=self.models[self.l.GetItemText(i)]
				m.offsets=[100,100,m.offsets[2]]
				self.Refresh()

	def snap(self,event):
		i=self.l.GetFirstSelected()
		if i != -1:
				m=self.models[self.l.GetItemText(i)]
				m.offsets[2]=-1.0*min(m.facetsminz)[0]
				#print m.offsets[2]
				self.Refresh()

	def delete(self,event):
		i=self.l.GetFirstSelected()
		if i != -1:
				del self.models[self.l.GetItemText(i)]
				self.l.DeleteItem(i)
				self.l.Select(self.l.GetItemCount()-1)
				self.Refresh()

	def export(self,event):
		dlg=wx.FileDialog(self,"Pick file to save to",self.basedir,style=wx.FD_SAVE)
		dlg.SetWildcard("STL files (;*.stl;)")
		if(dlg.ShowModal() == wx.ID_OK):
			name=dlg.GetPath()
			facets=[]
			for i in self.models.values():
				r=i.rot
				o=i.offsets
				if r != 0:
					i=i.rotate([0,0,-r])
				if o != [0,0,0]:
					i=i.translate([o[0],-o[1],o[2]])
				facets+=i.facets
			stltool.emitstl(name,facets,"plater_export")
			print "wrote ",name
			
	def testingloadfile(self,name):
		print name
		if not(os.path.exists(name)):
			return
		path = os.path.split(name)[0]
		self.basedir=path
		t=time.time()
		#print name
		if name.lower().endswith(".stl"):
			newname=name
			c=1
			while newname in self.models:
				newname=name+"(%d)"%c
				c+=1
			self.models[newname]=stltool.stl(name)
			self.models[newname].offsets=[0,0,0]
			self.models[newname].rot=0
			minx,miny,minz,maxx,maxy,maxz=(10000,10000,10000,0,0,0)
			for i in self.models[newname].facets:
				for j in i[1]:
					if j[0]<minx:
						minx=j[0]
					if j[1]<miny:
						miny=j[1]
					if j[2]<minz:
						minz=j[2]
					if j[0]>maxx:
						maxx=j[0]
					if j[1]>maxy:
						maxy=j[1]
					if j[2]>maxz:
						maxz=j[2]
			self.models[newname].dims=[minx,maxx,miny,maxy,minz,maxz]
			#if minx<0:
			#	self.models[newname].offsets[0]=-minx
			#if miny<0:
			#	self.models[newname].offsets[1]=-miny
			self.models[newname].bitmap=wx.EmptyBitmap(800,800,32)
			dc=wx.MemoryDC()
			dc.SelectObject(self.models[newname].bitmap)
			dc.SetBackground(wx.Brush((0,0,0,0)))
			dc.SetBrush(wx.Brush((0,0,0,255)))
			#dc.DrawRectangle(-1,-1,10000,10000)
			dc.SetBrush(wx.Brush(wx.Colour(128,255,128)))
			dc.SetPen(wx.Pen(wx.Colour(128,128,128)))
			m=self.models[newname]
			#m.offsets=[10,10,0]
			print m.offsets,m.dims
			scale=2
			for i in m.facets:#random.sample(m.facets,min(100000,len(m.facets))):
				dc.DrawPolygon([wx.Point(400+scale*p[0],(400+scale*p[1])) for p in i[1]])
				#if(time.time()-t)>5:
				#	break
			dc.SelectObject(wx.NullBitmap)
			m.bitmap.SetMask(wx.Mask(m.bitmap,wx.Colour(0,0,0,255)))
			
			#print time.time()-t
			self.l.Append([stlwrap(self.models[newname],newname)])
			i=self.l.GetFirstSelected()
			if i != -1:
				self.l.Select(i,0)
		
			self.l.Select(self.l.GetItemCount()-1)
			self.Refresh()
			
	def autoplate(self,event):
		print "Autoplating"
		
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\big_gear.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\big_gear.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\big_gear.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\big_gear.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\prusabits\\stl\\x-end-idler.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\prusabits\\stl\\x-end-idler.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\prusabits\\stl\\x-end-idler.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\prusabits\\stl\\pla-bushing.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\prusabits\\stl\\pla-bushing.stl')
		self.testingloadfile('C:\\Users\\tjhowse\\reprap\\models\\top.stl')
		separation = 2
		bedsize = [200,200,100]
		cursor = [0,0,0]
		newrow = 0
		for i in self.models:
			x = abs(self.models[i].dims[0] - self.models[i].dims[1])
			y = abs(self.models[i].dims[2] - self.models[i].dims[3])
			centre = [x/2, y/2]
			centreoffset = [self.models[i].dims[0] + centre[0], self.models[i].dims[2] + centre[1]]
			if (newrow == 0) or (newrow < y):
				newrow = y
			#To the person who works out why the offsets are applied differently here:
			# Good job, it confused the hell out of me.
			self.models[i].offsets[0] = cursor[0] + centre[0] - centreoffset[0]
			self.models[i].offsets[1] = cursor[1] + centre[1] + centreoffset[1]
			cursor[0] += x+separation
			if cursor[0] >= bedsize[0]:
				cursor[0] = 0
				cursor[1] += newrow+separation
				newrow = 0
				self.models[i].offsets[0] = cursor[0] + centre[0] - centreoffset[0]
				self.models[i].offsets[1] = cursor[1] + centre[1] + centreoffset[1]
				cursor[0] += x+separation
			if (cursor[1]+y) >= bedsize[1]:
				print "Bed full, sorry sir :("
				self.Refresh()
				return
		self.Refresh()
		
	
		
	def right(self,event):
		dlg=wx.FileDialog(self,"Open file to print",self.basedir,style=wx.FD_OPEN|wx.FD_FILE_MUST_EXIST)
		dlg.SetWildcard("STL files (;*.stl;)")
		if(dlg.ShowModal() == wx.ID_OK):
			name=dlg.GetPath()
			if not(os.path.exists(name)):
				return
			path = os.path.split(name)[0]
			self.basedir=path
			t=time.time()
			#print name
			if name.lower().endswith(".stl"):
				newname=name
				c=1
				while newname in self.models:
					newname=name+"(%d)"%c
					c+=1
				self.models[newname]=stltool.stl(name)
				self.models[newname].offsets=[0,0,0]
				self.models[newname].rot=0
				minx,miny,minz,maxx,maxy,maxz=(10000,10000,10000,0,0,0)
				for i in self.models[newname].facets:
					for j in i[1]:
						if j[0]<minx:
							minx=j[0]
						if j[1]<miny:
							miny=j[1]
						if j[2]<minz:
							minz=j[2]
						if j[0]>maxx:
							maxx=j[0]
						if j[1]>maxy:
							maxy=j[1]
						if j[2]>maxz:
							maxz=j[2]
				self.models[newname].dims=[minx,maxx,miny,maxy,minz,maxz]
				#if minx<0:
				#	self.models[newname].offsets[0]=-minx
				#if miny<0:
				#	self.models[newname].offsets[1]=-miny
				self.models[newname].bitmap=wx.EmptyBitmap(800,800,32)
				dc=wx.MemoryDC()
				dc.SelectObject(self.models[newname].bitmap)
				dc.SetBackground(wx.Brush((0,0,0,0)))
				dc.SetBrush(wx.Brush((0,0,0,255)))
				#dc.DrawRectangle(-1,-1,10000,10000)
				dc.SetBrush(wx.Brush(wx.Colour(128,255,128)))
				dc.SetPen(wx.Pen(wx.Colour(128,128,128)))
				m=self.models[newname]
				#m.offsets=[10,10,0]
				print m.offsets,m.dims
				scale=2
				for i in m.facets:#random.sample(m.facets,min(100000,len(m.facets))):
					dc.DrawPolygon([wx.Point(400+scale*p[0],(400+scale*p[1])) for p in i[1]])
					#if(time.time()-t)>5:
					#	break
				dc.SelectObject(wx.NullBitmap)
				m.bitmap.SetMask(wx.Mask(m.bitmap,wx.Colour(0,0,0,255)))
				
				#print time.time()-t
				self.l.Append([stlwrap(self.models[newname],newname)])
				i=self.l.GetFirstSelected()
				if i != -1:
					self.l.Select(i,0)
			
				self.l.Select(self.l.GetItemCount()-1)
			self.Refresh()
			#print time.time()-t
		
	def move(self,event):
		if event.ButtonUp(wx.MOUSE_BTN_LEFT):
			if(self.initpos is not None):
				i=self.l.GetFirstSelected()
				if i != -1:
					p=event.GetPositionTuple()
					#print (p[0]-self.initpos[0]),(p[1]-self.initpos[1])
					t=time.time()
					m=self.models[self.l.GetItemText(i)]
					m.offsets=[m.offsets[0]+0.5*(p[0]-self.initpos[0]),m.offsets[1]-0.5*(p[1]-self.initpos[1]),m.offsets[2]]
					#self.models[self.l.GetItemText(i)]=self.models[self.l.GetItemText(i)].translate([0.5*(p[0]-self.initpos[0]),0.5*(p[1]-self.initpos[1]),0])
					#print time.time()-t
				self.Refresh()
				self.initpos=None
		elif event.ButtonDown(wx.MOUSE_BTN_RIGHT):
			self.right(event)
		elif event.Dragging():
			if self.initpos is None:
				self.initpos=event.GetPositionTuple()
			self.Refresh()
			dc=wx.ClientDC(self)
			p=event.GetPositionTuple()
			dc.DrawLine(self.initpos[0],self.initpos[1],p[0],p[1])
			#print math.sqrt((p[0]-self.initpos[0])**2+(p[1]-self.initpos[1])**2)
					
			del dc
		else:
			event.Skip()
		
		
	def rotateafter(self):
		if(self.i!=self.previ):
			i=self.l.GetFirstSelected()
			if i != -1:
				#o=self.models[self.l.GetItemText(i)].offsets
				self.models[self.l.GetItemText(i)].rot+=5*(self.i-self.previ)
				#self.models[self.l.GetItemText(i)].offsets=o
			self.previ=self.i
			self.Refresh()
	def cr(self):
		time.sleep(0.01)
		wx.CallAfter(self.rotateafter)
		self.triggered=0
		
	def rot(self, event):
		z=event.GetWheelRotation()
		s=self.l.GetFirstSelected()
		if self.prevsel!=s:
			self.i=0
			self.prevsel=s
		if z > 0:
			self.i-=1
		else:
			self.i+=1
		if not self.triggered:
			self.triggered=1
			threading.Thread(target=self.cr).start()
	
	def repaint(self,event):
		dc=wx.PaintDC(self)
		self.paint(dc=dc)
		
	def paint(self,coord1="x",coord2="y",dc=None):
		coords={"x":0,"y":1,"z":2}
		if dc is None:
			dc=wx.ClientDC(self)
		offset=[0,0]
		scale=2
		dc.SetPen(wx.Pen(wx.Colour(100,100,100)))
		for i in xrange(20):
			dc.DrawLine(0,i*scale*10,400,i*scale*10)
			dc.DrawLine(i*scale*10,0,i*scale*10,400)
		dc.SetPen(wx.Pen(wx.Colour(0,0,0)))
		for i in xrange(4):
			dc.DrawLine(0,i*scale*50,400,i*scale*50)
			dc.DrawLine(i*scale*50,0,i*scale*50,400)
		dc.SetBrush(wx.Brush(wx.Colour(128,255,128)))
		dc.SetPen(wx.Pen(wx.Colour(128,128,128)))
		t=time.time()
		dcs=wx.MemoryDC()
		for m in self.models.values():
			b=m.bitmap
				#print b
			im=b.ConvertToImage()
				#print im
			imgc = wx.Point( im.GetWidth()/2,im.GetHeight()/2 )
				#print math.radians(5*(self.i-self.previ))
			im= im.Rotate( math.radians(m.rot), imgc, 0)
			bm=wx.BitmapFromImage(im)
			dcs.SelectObject(bm)
			bsz=bm.GetSize()
			dc.Blit(scale*m.offsets[0]-bsz[0]/2,400-(scale*m.offsets[1]+bsz[1]/2),bsz[0],bsz[1],dcs,0,0,useMask=1)
			#for i in m.facets:#random.sample(m.facets,min(100000,len(m.facets))):
			#	dc.DrawPolygon([wx.Point(offset[0]+scale*m.offsets[0]+scale*p[0],400-(offset[1]+scale*m.offsets[1]+scale*p[1])) for p in i[1]])
				#if(time.time()-t)>5:
				#	break
		del dc
		#print time.time()-t
		#s.export()
		
class stlwin(wx.Frame):
	def __init__(self,size=(400,530)):
		wx.Frame.__init__(self,None,title="Right-click to add a file",size=size)
		self.SetIcon(wx.Icon("plater.ico",wx.BITMAP_TYPE_ICO))
		self.SetClientSize(size)
		self.s=showstl(self,(400,530),(0,0))
		
if __name__ == '__main__':
	app = wx.App(False)
	main = stlwin()
	main.Show()
	app.MainLoop()