Fixes #134

citiususc · Aug 3, 2015 · bb67d88 · bb67d88
1 parent c4ed9c9
commit bb67d88
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/...core/src/main/java/es/usc/citius/hipster/model/function/impl/StateTransitionFunction.java b/...core/src/main/java/es/usc/citius/hipster/model/function/impl/StateTransitionFunction.java
@@ -19,6 +19,8 @@
 
 import es.usc.citius.hipster.model.Transition;
 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;
 
@@ -37,13 +39,13 @@ public abstract class StateTransitionFunction<S> implements TransitionFunction<V
 
     @Override
     public Iterable<Transition<Void, S>> transitionsFrom(final S state) {
-        ArrayList<Transition<Void, S>> transitions = new ArrayList<Transition<Void, S>>();
         //generate successor states
-        for(S current : successorsOf(state)){
-            //generate successor transitions from the states
-            transitions.add(new Transition<Void, S>(state, null, current));
-        }
-        return transitions;
+        return F.map(successorsOf(state), new Function<S, Transition<Void, S>>() {
+            @Override
+            public Transition<Void, S> apply(S current) {
+                return Transition.create(state, null, current);
+            }
+        });
     }
 
     /**