diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/DayAssignment.java b/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/DayAssignment.java index e2e86792d..2498e6d4c 100644 --- a/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/DayAssignment.java +++ b/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/DayAssignment.java @@ -32,6 +32,7 @@ import java.util.Map; import java.util.Set; import java.util.Map.Entry; +import org.apache.commons.lang.ObjectUtils; import org.apache.commons.lang.Validate; import org.hibernate.validator.Min; import org.hibernate.validator.NotNull; @@ -145,6 +146,17 @@ public abstract class DayAssignment extends BaseEntity { return getOfType(GenericDayAssignment.class, dayAssignments); } + public static List withScenario( + Scenario scenario, Collection dayAssignments) { + List result = new ArrayList(); + for (T each : dayAssignments) { + if (ObjectUtils.equals(each.getScenario(), scenario)) { + result.add(each); + } + } + return result; + } + @Min(0) private int hours;