Sergiu Toporjinschi
107 lines
4.7 KiB
Python
107 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
import re
|
|
from threading import Event
|
|
|
|
import flask
|
|
import octoprint.plugin
|
|
|
|
CMD_ESTEPS_LOAD_STEPS = "eSteps_load"
|
|
CMD_ESTEPS_START_EXTRUSION = "eSteps_startExtrusion"
|
|
CMD_ESTEPS_SAVE = "eSteps_save"
|
|
|
|
class API(octoprint.plugin.SimpleApiPlugin):
|
|
|
|
@staticmethod
|
|
def apiCommands():
|
|
return {
|
|
CMD_ESTEPS_SAVE: [],
|
|
CMD_ESTEPS_LOAD_STEPS: [],
|
|
CMD_ESTEPS_START_EXTRUSION: ['extrudeLength','extrudeSpeed','extrudeTemp']
|
|
}
|
|
|
|
def apiGateWay(self, command, data):
|
|
if command == CMD_ESTEPS_LOAD_STEPS:
|
|
self._logger.debug("Load steps from EEPROM")
|
|
if not self._printer.is_ready():
|
|
self._logger.warning("Printer not ready, operation canceled")
|
|
return flask.abort(503, 'Printer not ready, operation canceled')
|
|
|
|
# Register listener waiting response for M92 command
|
|
m92Event = Event()
|
|
self.registerGCodeWaiter("M92", self.m92GCodeResponse, m92Event)
|
|
|
|
# Issue M92 command
|
|
self._printer.commands("M92")
|
|
|
|
m92Event.wait(5)
|
|
|
|
return flask.jsonify({
|
|
"data": self.data["steps"]
|
|
})
|
|
if command == CMD_ESTEPS_START_EXTRUSION:
|
|
self._logger.debug("Heating the extruder [%s]", data)
|
|
if not self._printer.is_ready():
|
|
self._logger.warning("Printer not ready, operation canceled")
|
|
return flask.abort(503,"Printer not ready, operation canceled")
|
|
|
|
# Register event to be trigger when temp is achieved
|
|
self.registerEventTemp("T0", int(data["extrudeTemp"]), self.startExtrusion, data["extrudeLength"], data["extrudeSpeed"])
|
|
# Heating HotEnd
|
|
self._printer.commands("M104 S%(extrudeTemp)s" % data)
|
|
|
|
if command == CMD_ESTEPS_SAVE:
|
|
if (("newESteps" not in data or data["newESteps"] <= 0)
|
|
and
|
|
(("newXSteps" not in data or data["newXSteps"] <= 0) or
|
|
("newYSteps" not in data or data["newYSteps"] <= 0) or
|
|
("newZSteps" not in data or data["newZSteps"] <= 0))):
|
|
return flask.abort(400, "Invalid arguments. No value provided for X,Y, Z or E steps")
|
|
stopHeater = []
|
|
if "newESteps" in data:
|
|
eStepsSettings = self._settings.get(['eSteps'])
|
|
userControlsTemp = eStepsSettings.get("userControlsTemp")
|
|
turnOffHotend = eStepsSettings.get("turnOffHotend")
|
|
stopHeater = ["M104 S0"] if turnOffHotend and not userControlsTemp else []
|
|
|
|
steps = ("M92 " +
|
|
("E%(newESteps)s " % data if "newESteps" in data else "") +
|
|
("X%(newXSteps)s " % data if "newXSteps" in data else "") +
|
|
("Y%(newYSteps)s " % data if "newYSteps" in data else "") +
|
|
("Z%(newYSteps)s " % data if "newZSteps" in data else ""))
|
|
|
|
#save data to EEPROM and cool-down
|
|
self._printer.commands([steps, "M500"] + stopHeater)
|
|
return
|
|
|
|
|
|
############## HANDLERS ##############
|
|
@staticmethod
|
|
def startExtrusion(self, temps, extrudeLength, extrudeSpeed, *args):
|
|
self._logger.debug("Temperature achieved, extrusion started [temps:%s, extrudeLength:%s, extrudeSpeed:%s, args:%s]",
|
|
temps, extrudeLength, extrudeSpeed, args)
|
|
|
|
# Extrude
|
|
self._printer.extrude(amount=int(extrudeLength), speed=int(extrudeSpeed))
|
|
|
|
@staticmethod
|
|
def m92GCodeResponse(self, line, regex, event):
|
|
reg = re.compile(".*\s*(?P<command>(?P<gCode>M\d{1,3}) X(?P<xVal>\d{1,3}.\d{1,3}) Y(?P<yVal>\d{1,3}.\d{1,3}) Z(?P<zVal>\d{1,3}.\d{1,3}) E(?P<eVal>\d{1,3}.\d{1,3}))")
|
|
isM92command = reg.match(line)
|
|
if isM92command:
|
|
command = isM92command.group("command")
|
|
if isM92command.group("gCode") == "M92":
|
|
self.data["steps"]["X"] = float(isM92command.group("xVal"))
|
|
self.data["steps"]["Y"] = float(isM92command.group("yVal"))
|
|
self.data["steps"]["Z"] = float(isM92command.group("zVal"))
|
|
self.data["steps"]["E"] = float(isM92command.group("eVal"))
|
|
|
|
# Send the new data to the UI to be reloaded
|
|
self._logger.debug("gCode: %s", command)
|
|
self._logger.debug("X: %s, Y:%s, Z:%s, E:%s", self.data["steps"]["X"], self.data["steps"]["Y"], self.data["steps"]["Z"], self.data["steps"]["E"])
|
|
self._logger.debug("Finished data collection")
|
|
self.collectCommand = False
|
|
if event:
|
|
event.set()
|