From ed97494a056ec176f095fc2973f907bfb36cbf2d Mon Sep 17 00:00:00 2001
From: Manuel Rego Casasnovas <mrego@igalia.com>
Date: Fri, 9 Oct 2009 17:34:54 +0200
Subject: [PATCH]  ItEr29S16CUCreacionProxectoCategorizacionTraballoItEr10S11:
 Added new method to Order to get the associated TaskElement. This will be
 needed in the future.

---
 .../navalplanner/business/orders/entities/Order.java  | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/Order.java b/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/Order.java
index 9a874a180..5232f0fe0 100644
--- a/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/Order.java
+++ b/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/Order.java
@@ -21,7 +21,6 @@
 package org.navalplanner.business.orders.entities;
 
 import java.util.ArrayList;
-import java.util.Date;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
@@ -31,6 +30,7 @@ import org.navalplanner.business.common.exceptions.ValidationException;
 import org.navalplanner.business.planner.entities.DayAssignment;
 import org.navalplanner.business.planner.entities.ResourceAllocation;
 import org.navalplanner.business.planner.entities.TaskElement;
+import org.navalplanner.business.planner.entities.TaskGroup;
 import org.navalplanner.business.resources.entities.Resource;
 
 /**
@@ -84,6 +84,15 @@ public class Order extends OrderLineGroup implements IValidable {
         return new ArrayList<OrderElement>(getChildren());
     }
 
+    public TaskGroup getAssociatedTaskElement() {
+        Set<TaskElement> taskElements = this.getTaskElements();
+        if (taskElements != null) {
+            return (TaskGroup) taskElements.iterator().next();
+        } else {
+            return null;
+        }
+    }
+
     public List<TaskElement> getAssociatedTasks() {
         ArrayList<TaskElement> result = new ArrayList<TaskElement>();
         for (OrderElement orderElement : getChildren()) {