1 files changed, 68 insertions, 0 deletions
diff --git a/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/TraceUnit.java b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/TraceUnit.java
new file mode 100644
index 00000000000..906e5d79be8
--- /dev/null
+++ b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/TraceUnit.java
@@ -0,0 +1,68 @@
+/*******************************************************************************
+ * Copyright (c) 2004, 2007 Boeing.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     Boeing - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.osee.define.traceability.data;
+
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.Set;
+
+/**
+ * @author Roberto E. Escobar
+ */
+public class TraceUnit {
+   private final String traceUnitType;
+   private final String name;
+   private final Collection<TraceMark> traceMarks;
+
+   public TraceUnit(String traceUnitType, String name) {
+      this.name = name;
+      this.traceUnitType = traceUnitType;
+      this.traceMarks = new HashSet<TraceMark>();
+   }
+
+   public String getTraceUnitType() {
+      return traceUnitType;
+   }
+
+   public String getName() {
+      return name;
+   }
+
+   public void addAllTraceMarks(Collection<TraceMark> traceItems) {
+      traceMarks.addAll(traceItems);
+   }
+
+   public void addTraceMark(TraceMark traceMark) {
+      traceMarks.add(traceMark);
+   }
+
+   public Collection<TraceMark> getTraceMarks() {
+      return traceMarks;
+   }
+
+   public Set<String> getTraceMarkTypes() {
+      Set<String> toReturn = new HashSet<String>();
+      for (TraceMark traceMark : traceMarks) {
+         toReturn.add(traceMark.getTraceType());
+      }
+      return toReturn;
+   }
+
+   public Collection<TraceMark> getTraceMarksByType(String type) {
+      Set<TraceMark> toReturn = new HashSet<TraceMark>();
+      for (TraceMark traceMark : traceMarks) {
+         if (traceMark.getTraceType().equalsIgnoreCase(type)) {
+            toReturn.add(traceMark);
+         }
+      }
+      return toReturn;
+   }
+}