diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/resources/entities/Resource.java b/navalplanner-business/src/main/java/org/navalplanner/business/resources/entities/Resource.java index eda7ebd4d..0a6b41c0b 100644 --- a/navalplanner-business/src/main/java/org/navalplanner/business/resources/entities/Resource.java +++ b/navalplanner-business/src/main/java/org/navalplanner/business/resources/entities/Resource.java @@ -61,6 +61,26 @@ import org.navalplanner.business.planner.entities.DayAssignment; */ public abstract class Resource extends BaseEntity{ + public static List machines( + Collection resources) { + return filter(Machine.class, resources); + } + + public static List workers(Collection resources) { + return filter(Worker.class, resources); + } + + public static List filter(Class klass, + Collection resources) { + List result = new ArrayList(); + for (Resource each : resources) { + if (klass.isInstance(each)) { + result.add(klass.cast(each)); + } + } + return result; + } + private ResourceCalendar calendar; private Set criterionSatisfactions = new HashSet(); @@ -788,4 +808,5 @@ public abstract class Resource extends BaseEntity{ public boolean isVirtual() { return false; } + }