heater: Rename add_sensor() to add_sensor_factory()

Signed-off-by: Douglas Hammond <wizhippo@gmail.com>
Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2019-02-25 20:20:37 -05:00
parent cf6c5e6f18
commit bb4fab5570
4 changed files with 12 additions and 12 deletions

View File

@ -191,7 +191,7 @@ def load_config(config):
for sensor_type, params in [("AD595", AD595), ("PT100 INA826", PT100)]:
func = (lambda config, params=params:
PrinterADCtoTemperature(config, LinearVoltage(config, params)))
pheater.add_sensor(sensor_type, func)
pheater.add_sensor_factory(sensor_type, func)
def load_config_prefix(config):
if config.get("resistance1", None) is None:
@ -199,4 +199,4 @@ def load_config_prefix(config):
else:
custom_sensor = CustomLinearResistance(config)
pheater = config.get_printer().lookup_object("heater")
pheater.add_sensor(custom_sensor.name, custom_sensor.create)
pheater.add_sensor_factory(custom_sensor.name, custom_sensor.create)

View File

@ -342,4 +342,4 @@ def load_config(config):
# Register sensors
pheater = config.get_printer().lookup_object("heater")
for name, klass in Sensors.items():
pheater.add_sensor(name, klass)
pheater.add_sensor_factory(name, klass)

View File

@ -117,9 +117,9 @@ def load_config(config):
pheater = config.get_printer().lookup_object("heater")
for sensor_type, params in Sensors.items():
func = (lambda config, params=params: PrinterThermistor(config, params))
pheater.add_sensor(sensor_type, func)
pheater.add_sensor_factory(sensor_type, func)
def load_config_prefix(config):
thermistor = CustomThermistor(config)
pheater = config.get_printer().lookup_object("heater")
pheater.add_sensor(thermistor.name, thermistor.create)
pheater.add_sensor_factory(thermistor.name, thermistor.create)

View File

@ -228,15 +228,15 @@ class ControlPID:
class PrinterHeaters:
def __init__(self, config):
self.printer = config.get_printer()
self.sensors = {}
self.sensor_factories = {}
self.heaters = {}
self.heaters_gcode_id = {}
# Register TURN_OFF_HEATERS command
gcode = self.printer.lookup_object('gcode')
gcode.register_command("TURN_OFF_HEATERS", self.cmd_TURN_OFF_HEATERS,
desc=self.cmd_TURN_OFF_HEATERS_help)
def add_sensor(self, sensor_type, sensor_factory):
self.sensors[sensor_type] = sensor_factory
def add_sensor_factory(self, sensor_type, sensor_factory):
self.sensor_factories[sensor_type] = sensor_factory
def setup_heater(self, config, gcode_id):
heater_name = config.get_name().split()[-1]
if heater_name == 'extruder':
@ -261,10 +261,10 @@ class PrinterHeaters:
self.printer.try_load_module(config, "adc_temperature")
self.printer.try_load_module(config, "spi_temperature")
sensor_type = config.get('sensor_type')
if sensor_type not in self.sensors:
raise self.printer.config_error("Unknown temperature sensor '%s'" % (
sensor_type,))
return self.sensors[sensor_type](config)
if sensor_type not in self.sensor_factories:
raise self.printer.config_error(
"Unknown temperature sensor '%s'" % (sensor_type,))
return self.sensor_factories[sensor_type](config)
def get_all_heaters(self):
return self.heaters.values()
cmd_TURN_OFF_HEATERS_help = "Turn off all heaters"