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(getChildren()); } + public TaskGroup getAssociatedTaskElement() { + Set taskElements = this.getTaskElements(); + if (taskElements != null) { + return (TaskGroup) taskElements.iterator().next(); + } else { + return null; + } + } + public List getAssociatedTasks() { ArrayList result = new ArrayList(); for (OrderElement orderElement : getChildren()) {