X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fsensor.cpp;h=618a06b7b609c6ac8a930bc62c197bb4ac2db768;hb=5a02a4f42dfc5dc6b4daef5fc02431bdfb2968d8;hp=47e05f6068076063acc1e711fd97b480e66e406d;hpb=218f194f537dfda55d32ce8d8739e322a3d0c07d;p=r2c2.git

diff --git a/source/libr2c2/sensor.cpp b/source/libr2c2/sensor.cpp
index 47e05f6..618a06b 100644
--- a/source/libr2c2/sensor.cpp
+++ b/source/libr2c2/sensor.cpp
@@ -20,9 +20,21 @@ Sensor::Sensor(Layout &l):
 
 Sensor::~Sensor()
 {
+	if(layout.has_driver() && address)
+		layout.get_driver().remove_sensor(address);
 	layout.remove(*this);
 }
 
+void Sensor::set_address(unsigned a)
+{
+	Driver *driver = (layout.has_driver() ? &layout.get_driver() : 0);
+	if(driver && address)
+		driver->remove_sensor(address);
+	address = a;
+	if(driver && address)
+		driver->add_sensor(address);
+}
+
 void Sensor::tick(const Time::TimeDelta &dt)
 {
 	if(state_confirm_timeout)