Mercurial > hg > Members > kono > jpf-core
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(']'); + } +}