Added SplitBehavior and RunFinite behaviors

2 files changed, 70 insertions, 0 deletions

diff --git a/behaviors/RunFinite.py b/behaviors/RunFinite.py
new file mode 100644
index 0000000..de2ce27
--- /dev/null
+++ b/behaviors/RunFinite.py
@@ -0,0 +1,25 @@
+from operationscore.Behavior import *
+import util.ComponentRegistry as compReg
+
+class RunFinite(Behavior):
+    """RunFinite will just wire input to output, but only a finite number of
+    times as specified by the Iterations argument tag"""
+
+    def behaviorInit(self):
+        pass
+
+    def processResponse(self, inp, state):
+
+        if state:
+            iterations = state
+        else:
+            iterations = self['Iterations']
+
+        if iterations > 0:
+            out = inp
+        else:
+            out = []
+
+        iterations -= 1
+        
+        return (out, iterations)
diff --git a/behaviors/SplitBehavior.py b/behaviors/SplitBehavior.py
new file mode 100644
index 0000000..1892ad3
--- /dev/null
+++ b/behaviors/SplitBehavior.py
@@ -0,0 +1,45 @@
+from operationscore.Behavior import *
+import util.ComponentRegistry as compReg
+
+class SplitBehavior(Behavior):
+    """SplitBehavior takes a list of behaviors, runs the input on all behaviors
+    listed, and then returns the concantenation of all the behavior outputs.
+    Behavior list is given under tag <BehaviorList> as a list of Id's
+    
+    Example:
+    <Behavior>
+        <Class>behaviors.SplitBehavior</Class>
+        <Args>
+            <Id>splitbehavior</Id>
+            <BehaviorList>
+                <Id>behavior1Id</Id>
+                <Id>behavior2Id</Id>
+            </BehaviorList>
+        </Args>
+    </Behavior>
+    """
+
+    def behaviorInit(self):
+        pass
+        
+    def processResponse(self, inp, state):
+
+        out = []
+        newstate = {}
+        for behaviorId in self['BehaviorList']:
+
+            behavior = compReg.getComponent(behaviorId)
+            if behaviorId in state:
+                behaviorState = state[behaviorId]
+            else:
+                behaviorState = []
+
+            #print behaviorId, " ", str(inp), ",", str(behaviorState)
+            output = behavior.immediateProcessInput(inp, behaviorState)
+            (behaviorOutput, behaviorState) = output
+            #print "  -->", str(behaviorState), ",", str(behaviorOutput)
+            
+            newstate[behaviorId] = behaviorState
+            out.extend(behaviorOutput)
+
+        return (out, newstate)