Fixes #133

citiususc · Aug 3, 2015 · c4ed9c9 · c4ed9c9
1 parent abb4a1e
commit c4ed9c9
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/...ain/java/es/usc/citius/hipster/model/function/impl/LazyActionStateTransitionFunction.java b/...ain/java/es/usc/citius/hipster/model/function/impl/LazyActionStateTransitionFunction.java
@@ -21,6 +21,8 @@
 import es.usc.citius.hipster.model.function.ActionFunction;
 import es.usc.citius.hipster.model.function.ActionStateTransitionFunction;
 import es.usc.citius.hipster.model.function.TransitionFunction;
+import es.usc.citius.hipster.util.F;
+import es.usc.citius.hipster.util.Function;
 
 import java.util.ArrayList;
 
@@ -52,12 +54,11 @@ public LazyActionStateTransitionFunction(ActionFunction<A, S> af, ActionStateTra
 
     @Override
     public Iterable<Transition<A, S>> transitionsFrom(final S state) {
-        ArrayList<Transition<A, S>> transitions = new ArrayList<Transition<A, S>>();
-        //generate set of actions
-        for(A applicableAction : af.actionsFor(state)){
-            //generate transition for each action
-            transitions.add(new Transition<A, S>(state, applicableAction, tf.apply(applicableAction, state)));
-        }
-        return transitions;
+        return F.map(af.actionsFor(state), new Function<A, Transition<A, S>>() {
+            @Override
+            public Transition<A, S> apply(A a) {
+                return Transition.create(state, a, tf.apply(a, state));
+            }
+        });
     }
 }