#!/usr/bin/python
#
# Copyright (c) 2010 MakerBot Industries
# 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
"""Taste Tester

Generates GCode which will draw a raft of varying speeds in order to determine optimum speed.

Usage: python taste_tester.py [options]

Options:
  -h, --help						show this help
  --z-height						the Z height to start at in mm.  default 0.25
  --z-feedrate						the Z axis feedrate in mm/min.  default 150
  --feedrate-min					the XY axes feedrate minimum in mm/min. default 2000
  --feedrate-max					the XY axes feedrate minimum in mm/min. default 4000
  --size						the width/height of the raft in mm. default 80
  --spacing						spacing between lines in mm. default 5
  --start-delay						the delay after the pressure valve opens before movement in milliseconds.  default 25
  --stop-delay						the delay after the relief valve opens before movement in milliseconds.  default 150
"""

from math import *
import sys
import getopt
from PIL import Image
from PIL import ImageOps
from PIL import ImageEnhance

class TasteTester:
	"Class to handle generating frosting code."
	def __init__(self, start_z, z, min, max, width, spacing, start_delay, stop_delay):

		self.current_x = 0
		self.current_y = 0
		self.current_z = 0

		self.z_feedrate = z
		self.min_xy = min
		self.max_xy = max
		self.width = width
		self.spacing = spacing
		self.start_height = start_z
		self.start_delay = start_delay
		self.stop_delay = stop_delay
		
		self.rows = int(width/spacing)
		self.speed_increment = (max - min)/(self.rows-1)

	def getSpeed(self, line):
		return self.min_xy + self.speed_increment*line

	def generate(self):
		"Generate the actual GCode"
		
		print "(", " ".join(sys.argv), ")"
		print "(After you run the script, use the reference below to determine speeds.)"
		print "(Starting from the top, the lines were printed at the following speeds:)"
		
		for i in range(self.rows):
			line = i+1
			print "(#%d - %.2f mm/min)" % (line, self.getSpeed(i))

		print
		print "G21 (metric ftw)"
		print "G90 (absolute mode)"
		print "G92 X0 Y0 Z0 (zero all axes)"
		self.go_to_point(0, 0, self.start_height, self.z_feedrate)
		print		
		
		for i in range(self.rows):
			start = -self.width/2
			end = self.width/2
			speed = self.getSpeed(i)
			y_point = end - ((self.width / self.rows) * i)
			print "(line from %.2f to %.2f at %.2f)" % (start, end, speed)
			self.go_to_point(start, y_point, self.start_height, speed)
			print("M106 (pressure on)")
			print("G4 P%d (wait %dms)") % (self.start_delay, self.start_delay)
			self.go_to_point(end, y_point, self.start_height, speed)
			print("M107 (pressure off)");
			print("M126 (relief valve open)")
			print("G4 P%d (wait %dms)") % (self.stop_delay, self.stop_delay)
			print("M127 (relief valve close)")
			print

		print "M107 (pressure off)"
		print "M126 (relief open)"
		self.go_to_point(self.current_x, self.current_y, 10, self.z_feedrate)
		self.go_to_point(0, 0, 25, self.max_xy)
		print "M18 (drives off)"
		print "M127"
		
	def go_to_point(self, x, y, z, feedrate):
		"Output a move to a point"
		
		print "G1 X%.2f Y%.2f Z%.2f F%.1f" % (x, y, z, feedrate)
		self.current_x = x
		self.current_y = y
		self.current_z = z

def main(argv):
	
	try:
		opts, args = getopt.getopt(argv, '', [
			'help',
			'feedrate-min=',
			'feedrate-max=',
			'size=',
			'spacing=',
			'start-delay=',
			'stop-delay=',
			'z-height=',
			'z-feedrate='
		])
	except getopt.GetoptError:
		usage()
		sys.exit(2)
        
	z_height = 0.25
	z_feedrate = 150
	xy_min = 2000
	xy_max = 4000
	size = 80
	spacing = 5
	start_delay = 25
	stop_delay = 150

	for opt, arg in opts:
		if opt in ("-h", "--help"):
			usage()
			sys.exit()
		elif opt in ("--z-height"):
			z_height = float(arg)
		elif opt in ("--z-feedrate"):
			z_feedrate = float(arg)
		elif opt in ("--feedrate-min"):
			xy_min = float(arg)
		elif opt in ("--feedrate-max"):
			xy_max = float(arg)
		elif opt in ("--size"):
			size = float(arg)
		elif opt in ("--spacing"):
			spacing = float(arg)
		elif opt in ("--start-delay"):
			start_delay = float(arg)
		elif opt in ("--stop-delay"):
			stop_delay = float(arg)

	tasty = TasteTester(z_height, z_feedrate, xy_min, xy_max, size, spacing, start_delay, stop_delay)
	tasty.generate()

def usage():
    print __doc__

if __name__ == "__main__":
	main(sys.argv[1:])