]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.utils.datastructures/src/org/simantics/utils/datastructures/prioritystack/PriorityStack.java
Migrated source code from Simantics SVN
[simantics/platform.git] / bundles / org.simantics.utils.datastructures / src / org / simantics / utils / datastructures / prioritystack / PriorityStack.java
diff --git a/bundles/org.simantics.utils.datastructures/src/org/simantics/utils/datastructures/prioritystack/PriorityStack.java b/bundles/org.simantics.utils.datastructures/src/org/simantics/utils/datastructures/prioritystack/PriorityStack.java
new file mode 100644 (file)
index 0000000..d35362a
--- /dev/null
@@ -0,0 +1,201 @@
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+\r
+package org.simantics.utils.datastructures.prioritystack;\r
+\r
+import java.lang.reflect.Array;\r
+import java.lang.reflect.Method;\r
+import java.util.ArrayList;\r
+import java.util.HashMap;\r
+import java.util.LinkedList;\r
+import java.util.List;\r
+import java.util.ListIterator;\r
+import java.util.Map;\r
+\r
+import org.simantics.utils.strings.EString;\r
+import org.simantics.utils.threads.IThreadWorkQueue;\r
+import org.simantics.utils.threads.SyncListenerList;\r
+\r
+\r
+/**\r
+ * Implementation to IPriorityStack.\r
+ * \r
+ * <p>\r
+ * Note!\r
+ * getAllItems() is optimized for result with the penalty of slower add/remove methods. \r
+ *\r
+ * @author Toni Kalajainen\r
+ * @param <E>\r
+ */\r
+public class PriorityStack<E> implements IPriorityStack<E> {\r
+       \r
+       @SuppressWarnings({ "rawtypes" })\r
+    private SyncListenerList<IPriorityStackListener> listeners =\r
+               new SyncListenerList<IPriorityStackListener>(IPriorityStackListener.class);\r
+       \r
+       private LinkedList<E> list = \r
+               new LinkedList<E>(); \r
+       \r
+       private E[] snapshotArray;\r
+       \r
+       private Map<E, Integer> priorities = \r
+               new HashMap<E, Integer>(); \r
+       \r
+       final Class<E> clazz;\r
+       \r
+       public PriorityStack(Class<E> clazz) {\r
+               this.clazz = clazz;\r
+               snapshotArray = createArray(0);\r
+       }\r
+       \r
+       @Override\r
+       public void add(E interactor, int priority) {\r
+               abu:\r
+               synchronized(this) {\r
+                       if (list.contains(interactor))\r
+                               throw new IllegalArgumentException("InteractorStack already contains item "+interactor);\r
+               \r
+                       priorities.put(interactor, priority);\r
+               \r
+                       if (list.size()==0)\r
+                       {\r
+                               list.add(interactor);\r
+                               snapshotArray = createSnapshot(list);\r
+                               break abu;\r
+                       }\r
+                               \r
+                       ListIterator<E> li = list.listIterator();\r
+                       while (li.hasNext()) {\r
+                               E i = li.next();\r
+                               double w = priorities.get( i );\r
+                               if (w > priority) {\r
+                                       li.previous();\r
+                                       li.add(interactor);\r
+                                       snapshotArray = createSnapshot(list);\r
+                                       break abu;\r
+                               }\r
+                       }\r
+                       list.addLast(interactor);\r
+                       snapshotArray = createSnapshot(list);\r
+               }\r
+               fireInteractorAdded(interactor);                \r
+       }\r
+       \r
+       @Override\r
+       public boolean remove(E interactor) {\r
+               synchronized(this) {\r
+                       if (!priorities.containsKey(interactor)) \r
+                               return false;\r
+                       priorities.remove(interactor);\r
+                       list.remove(interactor);\r
+                       snapshotArray = createSnapshot(list);\r
+               }\r
+               fireInteractorRemoved(interactor);\r
+               return true;\r
+       }       \r
+\r
+       @Override\r
+       public synchronized Integer getPriority(E interactor) {\r
+               return priorities.get(interactor);\r
+       }       \r
+\r
+       @SuppressWarnings({ "unchecked" })\r
+       private E[] createArray(int length)\r
+       {\r
+               return (E[]) Array.newInstance(clazz, length);\r
+       }\r
+\r
+       E[] createSnapshot(List<E> list) {\r
+               E[] result = createArray(list.size());\r
+               int index = 0;\r
+               for (E i : list)\r
+                       result[index++] = i; \r
+               return result;\r
+       }\r
+               \r
+       public synchronized int indexOf(E item)\r
+       {\r
+               for (int i=0; i<snapshotArray.length; i++)\r
+                       if (snapshotArray[i]==item)\r
+                               return i;\r
+               return -1;\r
+       }       \r
+       \r
+       @Override\r
+       public synchronized E[] toArray() {\r
+               return snapshotArray;\r
+       }\r
+\r
+       public synchronized <R extends E> R getSingleItem(Class<R> clazz)\r
+       {\r
+               R array[] = getItemsByClass(clazz);\r
+               if (array.length!=1)\r
+                       throw new RuntimeException("one "+clazz.getName()+" expected in PriorityStack, got "+array.length);\r
+               return (R) array[0];\r
+       }\r
+       \r
+       @SuppressWarnings("unchecked")\r
+    @Override\r
+       public synchronized <R extends E> R[] getItemsByClass(Class<R> clazz)\r
+       {\r
+               List<E> result = new ArrayList<E>(list.size());\r
+               for (E i : list)\r
+                       if (clazz.isAssignableFrom(i.getClass()))\r
+                               result.add(i);\r
+               return (R[])result.toArray(createArray(result.size()));\r
+       }\r
+\r
+       private static Method itemAdded = SyncListenerList.getMethod(IPriorityStackListener.class, "itemAdded");\r
+       private void fireInteractorAdded(E interactor)\r
+       {\r
+               listeners.fireEventSync(itemAdded, this, interactor);\r
+       }\r
+       \r
+       private static Method itemRemoved = SyncListenerList.getMethod(IPriorityStackListener.class, "itemRemoved");\r
+       private void fireInteractorRemoved(E interactor)\r
+       {\r
+               listeners.fireEventSync(itemRemoved, this, interactor);\r
+       }\r
+\r
+       @Override\r
+       public synchronized void addStackListener(IPriorityStackListener<E> listener) {\r
+               listeners.add(listener);\r
+       }\r
+\r
+       @Override\r
+       public synchronized void removeStackListener(IPriorityStackListener<E> listener) {\r
+               listeners.remove(listener);\r
+       }\r
+\r
+       @Override\r
+       public synchronized boolean contains(E interactor) {\r
+               return list.contains(interactor);\r
+       }\r
+\r
+       @Override\r
+       public void addStackListener(IThreadWorkQueue thread,\r
+                       IPriorityStackListener<E> listener) {\r
+               listeners.add(thread, listener);\r
+       }\r
+\r
+       @Override\r
+       public void removeStackListener(IThreadWorkQueue thread,\r
+                       IPriorityStackListener<E> listener) {\r
+               listeners.remove(thread, listener);\r
+       }\r
+\r
+       @Override\r
+       public String toString() {\r
+           return EString.implode(snapshotArray, "\n");\r
+       }\r
+\r
+}\r