summaryrefslogtreecommitdiff
path: root/host/simu/model/distance_sensor.py
diff options
context:
space:
mode:
Diffstat (limited to 'host/simu/model/distance_sensor.py')
-rw-r--r--host/simu/model/distance_sensor.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/host/simu/model/distance_sensor.py b/host/simu/model/distance_sensor.py
index bdc8531d..e6c2c8b7 100644
--- a/host/simu/model/distance_sensor.py
+++ b/host/simu/model/distance_sensor.py
@@ -27,7 +27,8 @@ from simu.utils.trans_matrix import TransMatrix
class DistanceSensor:
- def __init__ (self, table, pos, angle, range, into = None, level = 0):
+ def __init__ (self, table, pos, angle, range, into = None, level = 0,
+ exclude = None):
self.table = table
self.pos = pos
self.angle = angle
@@ -36,6 +37,7 @@ class DistanceSensor:
pos[1] + sin (angle) * range)
self.into = into or ()
self.level = level
+ self.exclude = exclude
self.distance = None
def evaluate (self):
@@ -51,7 +53,7 @@ class DistanceSensor:
pos, target = m.apply (pos, target)
# Find intersection.
i = self.table.intersect (pos, target, level = self.level,
- comp = lambda a, b: a < b)
+ comp = lambda a, b: a < b, exclude = self.exclude)
if i is not None:
self.distance = i.distance
else: