servo/components/util/dlist.rs
Patrick Walton 1a28960bd2 util: Add a method to push all elements from another doubly-linked list
without any moves.

This is tracked upstream as Rust bug #18402.
2014-10-28 10:47:04 -07:00

94 lines
2.8 KiB
Rust

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
//! Utility functions for doubly-linked lists.
use std::collections::DList;
use std::mem;
use std::ptr;
struct RawDList<T> {
length: uint,
head: Option<Box<RawNode<T>>>,
tail: *mut RawNode<T>,
}
#[allow(dead_code)]
struct RawNode<T> {
next: Option<Box<RawNode<T>>>,
prev: *mut RawNode<T>,
value: T,
}
#[unsafe_destructor]
impl<T> Drop for RawDList<T> {
fn drop(&mut self) {
fail!("shouldn't happen")
}
}
/// Workaround for a missing method on Rust's `DList` type. Splits the head off a list in O(1)
/// time.
pub fn split<T>(list: &mut DList<T>) -> DList<T> {
let list = unsafe {
mem::transmute::<&mut DList<T>,&mut RawDList<T>>(list)
};
if list.length == 0 {
fail!("split_dlist(): empty list")
}
let mut head_node = mem::replace(&mut list.head, None);
let head_node_ptr: *mut RawNode<T> = &mut **head_node.as_mut().unwrap();
let mut head_list = RawDList {
length: 1,
head: head_node,
tail: head_node_ptr,
};
debug_assert!(list.head.is_none());
mem::swap(&mut head_list.head.as_mut().unwrap().next, &mut list.head);
debug_assert!(head_list.head.as_mut().unwrap().next.is_none());
debug_assert!(head_list.head.as_mut().unwrap().prev.is_null());
head_list.head.as_mut().unwrap().prev = ptr::null_mut();
list.length -= 1;
if list.length == 0 {
list.tail = ptr::null_mut()
} else {
if list.length == 1 {
list.tail = &mut **list.head.as_mut().unwrap() as *mut RawNode<T>
}
list.head.as_mut().unwrap().prev = ptr::null_mut()
}
unsafe {
mem::transmute::<RawDList<T>,DList<T>>(head_list)
}
}
/// Appends the items in the other list to this one, leaving the other list empty.
#[inline]
pub fn append_from<T>(this: &mut DList<T>, other: &mut DList<T>) {
unsafe {
let this = mem::transmute::<&mut DList<T>,&mut RawDList<T>>(this);
let other = mem::transmute::<&mut DList<T>,&mut RawDList<T>>(other);
if this.length == 0 {
this.head = mem::replace(&mut other.head, None);
this.tail = mem::replace(&mut other.tail, ptr::null_mut());
this.length = mem::replace(&mut other.length, 0);
return
}
(*this.tail).next = match mem::replace(&mut other.head, None) {
None => return,
Some(mut head) => {
head.prev = this.tail;
Some(head)
}
};
this.tail = mem::replace(&mut other.tail, ptr::null_mut());
this.length += other.length;
other.length = 0;
}
}