01: /*
02: * <copyright>
03: *
04: * Copyright 1997-2004 BBNT Solutions, LLC
05: * under sponsorship of the Defense Advanced Research Projects
06: * Agency (DARPA).
07: *
08: * You can redistribute this software and/or modify it under the
09: * terms of the Cougaar Open Source License as published on the
10: * Cougaar Open Source Website (www.cougaar.org).
11: *
12: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
13: * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
14: * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
15: * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
16: * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
17: * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
18: * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
19: * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
20: * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
22: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23: *
24: * </copyright>
25: */
26:
27: package org.cougaar.planning.ldm.plan;
28:
29: import org.cougaar.core.util.UID;
30: import org.cougaar.core.util.UniqueObject;
31: import org.cougaar.util.KeyedSet;
32:
33: /**
34: * PlanElementSet is a custom container which maintains a hashtable-like
35: * association between pe.task.key and pe object. The supports the single
36: * most time-consuming operation in logplan lookups.
37: **/
38:
39: public class PlanElementSet extends KeyedSet {
40: protected Object getKey(Object o) {
41: if (o instanceof PlanElement) {
42: Task task = ((PlanElement) o).getTask();
43: if (task == null) {
44: throw new IllegalArgumentException(
45: "Invalid PlanElement (no task) added to a PlanElementSet: "
46: + o);
47: }
48: return ((UniqueObject) task).getUID();
49: } else {
50: return null;
51: }
52: }
53:
54: // special methods for PlanElement searches
55:
56: public PlanElement findPlanElement(Task task) {
57: UID sk = ((UniqueObject) task).getUID();
58: return (PlanElement) inner.get(sk);
59: }
60:
61: public PlanElement findPlanElement(UID key) {
62: return (PlanElement) inner.get(key);
63: }
64: }
|