356 lines
11 KiB
Python
Executable File
356 lines
11 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# This file is copied from GCoder.
|
|
#
|
|
# GCoder 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 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# GCoder 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 Printrun. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import sys
|
|
import re
|
|
import math
|
|
import datetime
|
|
|
|
gcode_parsed_args = ["x", "y", "e", "f", "z", "p", "i", "j"]
|
|
gcode_exp = re.compile("\([^\(\)]*\)|[/\*].*\n|[a-z][-+]?[0-9]*\.?[0-9]*")
|
|
m114_exp = re.compile("\([^\(\)]*\)|[/\*].*\n|[A-Z]:?[-+]?[0-9]*\.?[0-9]*")
|
|
move_gcodes = ["G0", "G1", "G2", "G3"]
|
|
|
|
class Line(object):
|
|
|
|
x = None
|
|
y = None
|
|
z = None
|
|
e = None
|
|
f = None
|
|
i = None
|
|
j = None
|
|
|
|
relative = False
|
|
relative_e = False
|
|
|
|
raw = None
|
|
split_raw = None
|
|
|
|
command = None
|
|
is_move = False
|
|
|
|
duration = None
|
|
|
|
def __init__(self, l):
|
|
self.raw = l.lower()
|
|
if ";" in self.raw:
|
|
self.raw = self.raw.split(";")[0].rstrip()
|
|
self.split_raw = gcode_exp.findall(self.raw)
|
|
self.command = self.split_raw[0].upper() if not self.split_raw[0].startswith("n") else self.split_raw[1].upper()
|
|
self.is_move = self.command in move_gcodes
|
|
|
|
def parse_coordinates(self, imperial):
|
|
# Not a G-line, we don't want to parse its arguments
|
|
if not self.command[0] == "G":
|
|
return
|
|
if imperial:
|
|
for bit in self.split_raw:
|
|
code = bit[0]
|
|
if code in gcode_parsed_args and len(bit) > 1:
|
|
setattr(self, code, 25.4*float(bit[1:]))
|
|
else:
|
|
for bit in self.split_raw:
|
|
code = bit[0]
|
|
if code in gcode_parsed_args and len(bit) > 1:
|
|
setattr(self, code, float(bit[1:]))
|
|
|
|
def __repr__(self):
|
|
return self.raw.upper()
|
|
|
|
class Layer(object):
|
|
|
|
lines = None
|
|
duration = None
|
|
|
|
def __init__(self, lines):
|
|
self.lines = lines
|
|
|
|
def measure(self):
|
|
xmin = float("inf")
|
|
ymin = float("inf")
|
|
zmin = 0
|
|
xmax = float("-inf")
|
|
ymax = float("-inf")
|
|
zmax = float("-inf")
|
|
relative = False
|
|
relative_e = False
|
|
|
|
current_x = 0
|
|
current_y = 0
|
|
current_z = 0
|
|
|
|
for line in self.lines:
|
|
if line.is_move:
|
|
x = line.x
|
|
y = line.y
|
|
z = line.z
|
|
|
|
if line.relative:
|
|
x = current_x + (x or 0)
|
|
y = current_y + (y or 0)
|
|
z = current_z + (z or 0)
|
|
|
|
if line.e:
|
|
if x:
|
|
xmin = min(xmin, x)
|
|
xmax = max(xmax, x)
|
|
if y:
|
|
ymin = min(ymin, y)
|
|
ymax = max(ymax, y)
|
|
if z:
|
|
zmin = min(zmin, z)
|
|
zmax = max(zmax, z)
|
|
|
|
current_x = x or current_x
|
|
current_y = y or current_y
|
|
current_z = z or current_z
|
|
|
|
elif line.command == "G92":
|
|
current_x = line.x or current_x
|
|
current_y = line.y or current_y
|
|
current_z = line.z or current_z
|
|
|
|
return (xmin, xmax), (ymin, ymax), (zmin, zmax)
|
|
|
|
class GCode(object):
|
|
|
|
lines = None
|
|
layers = None
|
|
all_layers = None
|
|
idxs = None
|
|
|
|
def __init__(self,data):
|
|
self.lines = [Line(l2) for l2 in
|
|
(l.strip() for l in data)
|
|
if l2 and not l2[0] == ";"]
|
|
self._preprocess()
|
|
self._create_layers()
|
|
|
|
def _preprocess(self):
|
|
"""Checks for G20, G21, G90 and G91, sets imperial and relative flags"""
|
|
imperial = False
|
|
relative = False
|
|
relative_e = False
|
|
for line in self.lines:
|
|
if line.is_move:
|
|
line.relative = relative
|
|
line.relative_e = relative_e
|
|
elif line.command == "G20":
|
|
imperial = True
|
|
elif line.command == "G21":
|
|
imperial = False
|
|
elif line.command == "G90":
|
|
relative = False
|
|
relative_e = False
|
|
elif line.command == "G91":
|
|
relative = True
|
|
relative_e = True
|
|
elif line.command == "M82":
|
|
relative_e = False
|
|
elif line.command == "M83":
|
|
relative_e = True
|
|
if line.command[0] == "G":
|
|
line.parse_coordinates(imperial)
|
|
|
|
# FIXME : looks like this needs to be tested with list Z on move
|
|
def _create_layers(self):
|
|
layers = {}
|
|
all_layers = []
|
|
idxs = []
|
|
|
|
layer_id = 0
|
|
layer_line = 0
|
|
|
|
prev_z = None
|
|
cur_z = 0
|
|
cur_lines = []
|
|
for line in self.lines:
|
|
if line.command == "G92" and line.z != None:
|
|
cur_z = line.z
|
|
elif line.is_move:
|
|
if line.z != None:
|
|
if line.relative:
|
|
cur_z += line.z
|
|
else:
|
|
cur_z = line.z
|
|
|
|
if cur_z != prev_z:
|
|
all_layers.append(Layer(cur_lines))
|
|
old_lines = layers.get(prev_z, [])
|
|
old_lines += cur_lines
|
|
layers[prev_z] = old_lines
|
|
cur_lines = []
|
|
layer_id += 1
|
|
layer_line = 0
|
|
|
|
cur_lines.append(line)
|
|
idxs.append((layer_id, layer_line))
|
|
layer_line += 1
|
|
prev_z = cur_z
|
|
|
|
if cur_lines:
|
|
all_layers.append(Layer(cur_lines))
|
|
old_lines = layers.pop(prev_z, [])
|
|
old_lines += cur_lines
|
|
layers[prev_z] = old_lines
|
|
|
|
for idx in layers.keys():
|
|
cur_lines = layers[idx]
|
|
has_movement = False
|
|
for l in layers[idx]:
|
|
if l.is_move and l.e != None:
|
|
has_movement = True
|
|
break
|
|
if has_movement:
|
|
layers[idx] = Layer(cur_lines)
|
|
else:
|
|
del layers[idx]
|
|
|
|
self.all_layers = all_layers
|
|
self.layers = layers
|
|
self.idxs = idxs
|
|
|
|
def num_layers(self):
|
|
return len(self.layers)
|
|
|
|
def measure(self):
|
|
xmin = float("inf")
|
|
ymin = float("inf")
|
|
zmin = 0
|
|
xmax = float("-inf")
|
|
ymax = float("-inf")
|
|
zmax = float("-inf")
|
|
|
|
for l in self.layers.values():
|
|
(xm, xM), (ym, yM), (zm, zM) = l.measure()
|
|
xmin = min(xm, xmin)
|
|
xmax = max(xM, xmax)
|
|
ymin = min(ym, ymin)
|
|
ymax = max(yM, ymax)
|
|
zmin = min(zm, zmin)
|
|
zmax = max(zM, zmax)
|
|
|
|
self.xmin = xmin
|
|
self.xmax = xmax
|
|
self.ymin = ymin
|
|
self.ymax = ymax
|
|
self.zmin = zmin
|
|
self.zmax = zmax
|
|
self.width = xmax - xmin
|
|
self.depth = ymax - ymin
|
|
self.height = zmax - zmin
|
|
|
|
def filament_length(self):
|
|
total_e = 0
|
|
max_e = 0
|
|
cur_e = 0
|
|
|
|
for line in self.lines:
|
|
if line.e == None:
|
|
continue
|
|
if line.is_move:
|
|
if line.relative_e:
|
|
total_e += line.e
|
|
else:
|
|
total_e += line.e - cur_e
|
|
cur_e = line.e
|
|
max_e = max(max_e, total_e)
|
|
elif line.command == "G92":
|
|
cur_e = line.e
|
|
|
|
return max_e
|
|
|
|
def estimate_duration(self):
|
|
lastx = lasty = lastz = laste = lastf = 0.0
|
|
x = y = z = e = f = 0.0
|
|
currenttravel = 0.0
|
|
totaltravel = 0.0
|
|
moveduration = 0.0
|
|
totalduration = 0.0
|
|
acceleration = 1500.0 #mm/s/s ASSUMING THE DEFAULT FROM SPRINTER !!!!
|
|
layerduration = 0.0
|
|
layerbeginduration = 0.0
|
|
layercount = 0
|
|
#TODO:
|
|
# get device caps from firmware: max speed, acceleration/axis (including extruder)
|
|
# calculate the maximum move duration accounting for above ;)
|
|
for layer in self.all_layers:
|
|
for line in layer.lines:
|
|
if line.command not in ["G1", "G0", "G4"]:
|
|
continue
|
|
if line.command == "G4":
|
|
moveduration = line.p
|
|
if not moveduration:
|
|
continue
|
|
else:
|
|
moveduration /= 1000.0
|
|
else:
|
|
x = line.x if line.x != None else lastx
|
|
y = line.y if line.y != None else lasty
|
|
e = line.e if line.e != None else laste
|
|
f = line.f / 60.0 if line.f != None else lastf # mm/s vs mm/m => divide by 60
|
|
|
|
# given last feedrate and current feedrate calculate the distance needed to achieve current feedrate.
|
|
# if travel is longer than req'd distance, then subtract distance to achieve full speed, and add the time it took to get there.
|
|
# then calculate the time taken to complete the remaining distance
|
|
|
|
currenttravel = math.hypot(x - lastx, y - lasty)
|
|
# FIXME: review this better
|
|
# this looks wrong : there's little chance that the feedrate we'll decelerate to is the previous feedrate
|
|
# shouldn't we instead look at three consecutive moves ?
|
|
distance = 2 * abs(((lastf + f) * (f - lastf) * 0.5) / acceleration) # multiply by 2 because we have to accelerate and decelerate
|
|
if distance <= currenttravel and lastf + f != 0 and f != 0:
|
|
# Unsure about this formula -- iXce reviewing this code
|
|
moveduration = 2 * distance / (lastf + f)
|
|
currenttravel -= distance
|
|
moveduration += currenttravel/f
|
|
else:
|
|
moveduration = math.sqrt(2 * distance / acceleration) # probably buggy : not taking actual travel into account
|
|
|
|
totalduration += moveduration
|
|
|
|
lastx = x
|
|
lasty = y
|
|
laste = e
|
|
lastf = f
|
|
|
|
layer.duration = totalduration - layerbeginduration
|
|
layerbeginduration = totalduration
|
|
|
|
return "%d layers, %s" % (len(self.layers), str(datetime.timedelta(seconds = int(totalduration))))
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print "usage: %s filename.gcode" % sys.argv[0]
|
|
return
|
|
|
|
gcode = GCode(open(sys.argv[1]))
|
|
|
|
gcode.measure()
|
|
|
|
print "Dimensions:"
|
|
print "\tX: %0.02f - %0.02f (%0.02f)" % (gcode.xmin,gcode.xmax,gcode.width)
|
|
print "\tY: %0.02f - %0.02f (%0.02f)" % (gcode.ymin,gcode.ymax,gcode.depth)
|
|
print "\tZ: %0.02f - %0.02f (%0.02f)" % (gcode.zmin,gcode.zmax,gcode.height)
|
|
print "Filament used: %0.02fmm" % gcode.filament_length()
|
|
print "Number of layers: %d" % gcode.num_layers()
|
|
print "Estimated duration: %s" % gcode.estimate_duration()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|