def opt(gfile, offset=(0.0,0.0,0.0), rotate=False):
'''Optimization core function:
Reads in gCode ascii file.
Processes gcode into toolpath list
figures out milling.
Reorders milling to get optimal
Writes out to new file.'''
start = datetime.now()
puts(colored.blue('Optimizing file: %s\n Started: %s'%(gfile.name,datetime.now())))
# Parse the gcode from the ascii to a list of command numbers and location
gcode = GCode(gfile)
gcode.parse()
# Take the list and make a toolpath out of it. A toolpath is a list of locations
# where the bit needs to be moved / milled : [ [x,y,z,t], ...]
tool = Tool(gcode)
tool.offset(offset)
tool.rotate(rotate)
tool.groupMills()
puts(colored.blue('Toolpath length: %.2f inches, (mill only: %.2f)'%(tool.length(),tool.millLength())))
if args.setMillHeight:
tool.setMillHeight(Z_MILL,Z_SPOT)
tool.uniqMills()
# This starts the optimization process:
# start at here, and go to the next path which is closest is the overall plan
puts(colored.blue('Starting Optimization:'))
here = [0.0]*3 # start at the origin
newMills = [] # accumulate mills here
k = 0
while len(tool.mills) > 0:
# No Optimization
# mill = tool.mills.pop(0)
# Basic optimization, find the next closest one and use it.
# mill = tool.getNextMill(here)
# Advanced Optimization: Assumes that each mill path closed, so finds
# the mill path which is close to the point and reorders it to be so
mill = tool.getClosestMill(here)
# you were here, now you are there
# move mills and update location
newMills.append(mill)
here = newMills[-1][-1]
k += 1
if (k%10) == 0:
sys.stdout.write('.')
sys.stdout.flush()
tool.mills.extend(newMills)
tool.reTool(Z_MOVE)
tool.uniq()
puts(colored.blue('Toolpath length: %.2f inches, (mill only: %.2f)'%(tool.length(),tool.millLength())))
# Save this with the _opt file ending.
output = tool.buildGcode()
outfile = FILEENDING.join(os.path.splitext(gfile.name))
puts(colored.green('Writing: %s'%outfile))
with open(outfile,'w') as f:
f.write(output)
# how long did this take?
puts(colored.green('Time to completion: %s'%(deltaTime(start))))
print
def zcorrect_file(gfile,surface_file_name = 'probe_test.out'):
# Load the correction surface
correction_surface = CorrectionSurface(surface_file_name)
# keep track of time
start = datetime.now()
name = gfile if isinstance(gfile,str) else gfile.name
puts(colored.blue('Z correcting the file: %s\n Started: %s'%(name,datetime.now())))
# Load the gcode.
gcode = GCode(gfile)
#parse the Gcode
gcode.parseAll()
# start an empty list
#out = []
# need to get rid of use of 'loc'
# loc = parse(args.move, getUnits=True) # only one move at a time.
# puts(colored.blue('Moving!\n (0,0) -> (%.3f,%.3f)'%(loc[0],loc[1])))
# create a tool object (toolpath object)
tool = Tool()
# load the gcode into the tool object
tool.build(gcode)
# adjust the z position at each point by the given amount
tool.zcorrect(correction_surface)
''' the follwing doe not work. is gcode.update(tool) broken?'''
# load the changes back into the gcode object
# append the modified g code to the empty list called out
# out.append([gcode])
# gcode.update(tool)
# out = gcode
# convert gcode to text format
# output = ''.join([o.getGcode(tag=args.name) for o in out])
# output = ''.join([out.getGcode()])
'''instead the following simgle lin suffices'''
'''is any info lost by doing it this way? E F M'''
# generate a gcode file from the tool object
output = tool.buildGcode()
# get an output file name
outfile = FILEENDING.join(os.path.splitext(gfile))
print "outfile is:"
print outfile
# tell the user
puts(colored.green('Writing: %s'%outfile))
# write to file
f = open(outfile,'w')
f.write(output)
'''
with open(outfile,'w') as f:
f.write(output)
'''
# how long did this take?
puts(colored.green('Time to completion: %s'%(deltaTime(start))))
print
