--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/printrun-src/printrun/injectgcode.py Fri Jun 03 09:16:07 2016 +0200 @@ -0,0 +1,45 @@ +# This file is part of the Printrun suite. +# +# Printrun 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. +# +# Printrun 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 logging + +from .gui.widgets import MacroEditor + +from .utils import install_locale +install_locale('pronterface') + +def injector(gcode, viz_layer, layer_idx): + cb = lambda toadd: inject(gcode, viz_layer, layer_idx, toadd) + z = gcode.all_layers[layer_idx].z + z = z if z is not None else 0 + MacroEditor(_("Inject G-Code at layer %d (Z = %.03f)") % (viz_layer, z), "", cb, True) + +def injector_edit(gcode, viz_layer, layer_idx): + cb = lambda toadd: rewritelayer(gcode, viz_layer, layer_idx, toadd) + layer = gcode.all_layers[layer_idx] + z = layer.z + z = z if z is not None else 0 + lines = [line.raw for line in layer] + MacroEditor(_("Edit G-Code of layer %d (Z = %.03f)") % (viz_layer, z), lines, cb, True) + +def inject(gcode, viz_layer, layer_idx, toadd): + # TODO: save modified gcode after injection ? + nlines = len(gcode.prepend_to_layer(toadd, layer_idx)) + logging.info(_("Successfully injected %d lines at beginning of layer %d") % (nlines, viz_layer)) + +def rewritelayer(gcode, viz_layer, layer_idx, toadd): + # TODO: save modified gcode after edit ? + nlines = len(gcode.rewrite_layer(toadd, layer_idx)) + logging.info(_("Successfully edited layer %d (which now contains %d lines)") % (viz_layer, nlines))