From 77459d43ed5e58f04e7d28929080a35c8560d2d6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=93scar=20Gonz=C3=A1lez=20Fern=C3=A1ndez?=
 <ogonzalez@igalia.com>
Date: Tue, 20 Oct 2009 00:11:57 +0200
Subject: [PATCH] ItEr31S12ValidacionEProbasFuncionaisItEr30S17: Adding support
 for modifying existent allocations

---
 .../planner/entities/ResourceAllocation.java   | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/ResourceAllocation.java b/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/ResourceAllocation.java
index ad30f3a12..11d8627f4 100644
--- a/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/ResourceAllocation.java
+++ b/navalplanner-business/src/main/java/org/navalplanner/business/planner/entities/ResourceAllocation.java
@@ -26,8 +26,10 @@ import java.util.Collection;
 import java.util.Collections;
 import java.util.Comparator;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import java.util.Map.Entry;
 
 import org.apache.commons.lang.Validate;
@@ -156,6 +158,22 @@ public abstract class ResourceAllocation<T extends DayAssignment> extends
             return new AllocationsAndResourcesCurried(task, resources,
                     resourceAllocations);
         }
+
+        public AllocationsAndResourcesCurried withExistentResources() {
+            return new AllocationsAndResourcesCurried(task,
+                    getResourcesFrom(ResourceAllocationWithDesiredResourcesPerDay
+                            .stripResourcesPerDay(resourceAllocations)),
+                    resourceAllocations);
+        }
+
+        private List<Resource> getResourcesFrom(
+                List<ResourceAllocation<?>> allocations) {
+            Set<Resource> resources = new HashSet<Resource>();
+            for (ResourceAllocation<?> resourceAllocation : allocations) {
+                resources.addAll(resourceAllocation.getAssociatedResources());
+            }
+            return new ArrayList<Resource>(resources);
+        }
     }
 
     public static class AllocationsAndResourcesCurried {