summaryrefslogtreecommitdiffhomepage
path: root/host/simu/robots/apbirthday/view/bag.py
diff options
context:
space:
mode:
Diffstat (limited to 'host/simu/robots/apbirthday/view/bag.py')
-rw-r--r--host/simu/robots/apbirthday/view/bag.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/host/simu/robots/apbirthday/view/bag.py b/host/simu/robots/apbirthday/view/bag.py
index a5d733e1..6a9e2d3d 100644
--- a/host/simu/robots/apbirthday/view/bag.py
+++ b/host/simu/robots/apbirthday/view/bag.py
@@ -23,6 +23,7 @@
# }}}
"""APBirthday bag of views."""
from simu.view.switch import Switch
+from simu.view.distance_sensor import DistanceSensor
from simu.view.distance_sensor_us import DistanceSensorUS
from simu.view.pos_report import PosReport
from simu.robots.apbirthday.view.robot import Robot
@@ -36,5 +37,7 @@ class Bag:
self.robot = Robot (table, model_bag.position, model_bag.cake_arm)
self.distance_sensor = [DistanceSensorUS (self.robot, ds)
for ds in model_bag.distance_sensor]
+ self.cake_front = DistanceSensor (self.robot, model_bag.cake_front)
+ self.cake_back = DistanceSensor (self.robot, model_bag.cake_back)
self.pos_report = PosReport (table, model_bag.pos_report)