diff src/main/gov/nasa/jpf/search/heuristic/StaticPriorityQueue.java @ 0:61d41facf527

initial v8 import (history reset)
author Peter Mehlitz <Peter.C.Mehlitz@nasa.gov>
date Fri, 23 Jan 2015 10:14:01 -0800
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/main/gov/nasa/jpf/search/heuristic/StaticPriorityQueue.java	Fri Jan 23 10:14:01 2015 -0800
@@ -0,0 +1,91 @@
+/*
+ * Copyright (C) 2014, United States Government, as represented by the
+ * Administrator of the National Aeronautics and Space Administration.
+ * All rights reserved.
+ *
+ * The Java Pathfinder core (jpf-core) platform is licensed under the
+ * Apache License, Version 2.0 (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ * 
+ *        http://www.apache.org/licenses/LICENSE-2.0. 
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and 
+ * limitations under the License.
+ */
+
+package gov.nasa.jpf.search.heuristic;
+
+import gov.nasa.jpf.Config;
+
+import java.util.Iterator;
+import java.util.TreeSet;
+
+/**
+ * container for statically prioritized states, based on bounded
+ * RB trees
+ * 
+ * TreeSet is a better choice than PriorityQueue since the size
+ * constraint means we have to remove elements from both ends, which
+ * is inefficient with heaps. A simple (binary) sorted array using
+ * block copy can be more efficient, but that very much depends on
+ * queue size and processor, so it's not really worth the trouble
+ * 
+ * <2do> this needs to be re-evaluated in light of the stupid fact
+ * that pollFirst()/pollLast() are only in Java 1.6 and we can't switch
+ * yet (no Java 1.6 for PPC), which forces us to do some shuffling to
+ * make remove(last()) work as expected
+ */
+@SuppressWarnings("serial")
+public class StaticPriorityQueue extends TreeSet<PrioritizedState> {
+
+  int maxQueueSize;
+  
+  public StaticPriorityQueue (Config config) {
+    // unbound queues are pretty pointless - we could probably better use
+    // DFSearch in this case
+    maxQueueSize = config.getInt("search.heuristic.queue_limit", 1024);
+    if (maxQueueSize < 0){
+      maxQueueSize = Integer.MAX_VALUE;
+    }
+  }
+    
+  @Override
+  public boolean add (PrioritizedState s) {
+    if (size() < maxQueueSize) { // limit not yet reached
+      return super.add(s);
+      
+    } else {
+      PrioritizedState last = last();
+      if (s.compareTo(last) < 0) {
+        //pollLast();   // that's only Java 1.6
+        remove(last);
+        
+        return super.add(s);
+        
+      } else {
+        // don't add with this priority value
+        return false;
+      }
+    }
+  }
+  
+  public boolean isQueueLimitReached() {
+    return size() >= maxQueueSize;
+  }
+  
+  // for debugging purposes
+  void dump() {
+    int i=0;
+    System.err.print('[');
+    for (Iterator<PrioritizedState> it=iterator(); it.hasNext();) {
+      if (i++ > 0) {
+        System.err.print(',');
+      }
+      System.err.print(it.next());
+    }
+    System.err.println(']');
+  }
+}