More scheduling and probabilisticity

1 files changed, 16 insertions, 0 deletions

diff --git a/src/lib/pqueue.c b/src/lib/pqueue.c
index 6f21e5f..c080bb2 100644
--- a/src/lib/pqueue.c
+++ b/src/lib/pqueue.c
@@ -157,3 +157,19 @@ void *pqueue_pop(struct pqueue *q)
 
 	return head;
 }
+
+void *pqueue_pop_at(struct pqueue *q, size_t p)
+{
+	void *head;
+	if (!q || q->size == 1 || p >= q->size)
+		return NULL;
+
+	head = q->d[p];
+	q->d[p] = q->d[--q->size];
+	if (q->cmppri(q->getpri(head), q->getpri(q->d[p])))
+		bubble_up(q, p);
+	else
+		percolate_down(q, p);
+
+	return head;
+}