Merge branch 'master' into behavior-factory

1 files changed, 7 insertions, 0 deletions

diff --git a/behaviors/Square.py b/behaviors/Square.py
index 9d3223a..5fdaeef 100644
--- a/behaviors/Square.py
+++ b/behaviors/Square.py
@@ -7,6 +7,7 @@ class Square(Behavior):
 
     def processResponse(self, sensorInputs, recursiveInputs):
         for sensory in sensorInputs:#TODO: consider replicating the dict
+            sensory['CenterLoc'] = list(sensory['Location'])
             xLoc = sensory['Location'][0]
             yLoc = sensory['Location'][1]
             width = self['Width']
@@ -15,3 +16,9 @@ class Square(Behavior):
                 '{x}<'+str(xLoc+width)+',{x}>'+str(xLoc-width)+\
                 ',{y}<'+str(yLoc+width)+',{y}>'+str(yLoc-width)
         return (sensorInputs, recursiveInputs)
+
+    def setLastOutput(self, output):
+        coutput = Behavior.deepCopyPacket(output)
+        for data in coutput:
+            data['Location'] = data['CenterLoc']
+        return coutput