Skip to content

add a DeepClone trait for deep clones of shared ownership types #6498

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 3 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions src/libcore/cell.rs
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,17 @@ Similar to a mutable option type, but friendlier.
*/

#[mutable]
#[deriving(Clone)]
pub struct Cell<T> {
priv value: Option<T>
}

impl<T: DeepClone> DeepClone for Cell<T> {
fn deep_clone(&self) -> Cell<T> {
Cell{value: self.value.deep_clone()}
}
}

impl<T:cmp::Eq> cmp::Eq for Cell<T> {
fn eq(&self, other: &Cell<T>) -> bool {
(self.value) == (other.value)
Expand Down
57 changes: 48 additions & 9 deletions src/libcore/clone.rs
Original file line number Diff line number Diff line change
Expand Up @@ -23,17 +23,12 @@ by convention implementing the `Clone` trait and calling the
*/

pub trait Clone {
/// Return a deep copy of the owned object tree. Managed boxes are cloned with a shallow copy.
/// Return a deep copy of the owned object tree. Types with shared ownership like managed boxes
/// are cloned with a shallow copy.
fn clone(&self) -> Self;
}

impl Clone for () {
/// Return a copy of the value.
#[inline(always)]
fn clone(&self) -> () { () }
}

impl<T:Clone> Clone for ~T {
impl<T: Clone> Clone for ~T {
/// Return a deep copy of the owned box.
#[inline(always)]
fn clone(&self) -> ~T { ~(**self).clone() }
Expand All @@ -54,7 +49,7 @@ impl<T> Clone for @mut T {
macro_rules! clone_impl(
($t:ty) => {
impl Clone for $t {
/// Return a copy of the value.
/// Return a deep copy of the value.
#[inline(always)]
fn clone(&self) -> $t { *self }
}
Expand All @@ -77,9 +72,53 @@ clone_impl!(float)
clone_impl!(f32)
clone_impl!(f64)

clone_impl!(())
clone_impl!(bool)
clone_impl!(char)

pub trait DeepClone {
/// Return a deep copy of the object tree. Types with shared ownership are also copied via a
/// deep copy, unlike `Clone`. Note that this is currently unimplemented for managed boxes, as
/// it would need to handle cycles.
fn deep_clone(&self) -> Self;
}

macro_rules! deep_clone_impl(
($t:ty) => {
impl DeepClone for $t {
/// Return a deep copy of the value.
#[inline(always)]
fn deep_clone(&self) -> $t { *self }
}
}
)

impl<T: DeepClone> DeepClone for ~T {
/// Return a deep copy of the owned box.
#[inline(always)]
fn deep_clone(&self) -> ~T { ~(**self).deep_clone() }
}

deep_clone_impl!(int)
deep_clone_impl!(i8)
deep_clone_impl!(i16)
deep_clone_impl!(i32)
deep_clone_impl!(i64)

deep_clone_impl!(uint)
deep_clone_impl!(u8)
deep_clone_impl!(u16)
deep_clone_impl!(u32)
deep_clone_impl!(u64)

deep_clone_impl!(float)
deep_clone_impl!(f32)
deep_clone_impl!(f64)

deep_clone_impl!(())
deep_clone_impl!(bool)
deep_clone_impl!(char)

#[test]
fn test_owned_clone() {
let a: ~int = ~5i;
Expand Down
10 changes: 10 additions & 0 deletions src/libcore/option.rs
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,7 @@ use num::Zero;
use old_iter::{BaseIter, MutableIter, ExtendedIter};
use old_iter;
use str::StrSlice;
use clone::DeepClone;

#[cfg(test)] use str;

Expand All @@ -59,6 +60,15 @@ pub enum Option<T> {
Some(T),
}

impl<T: DeepClone> DeepClone for Option<T> {
fn deep_clone(&self) -> Option<T> {
match *self {
Some(ref x) => Some(x.deep_clone()),
None => None
}
}
}

impl<T:Ord> Ord for Option<T> {
fn lt(&self, other: &Option<T>) -> bool {
match (self, other) {
Expand Down
2 changes: 1 addition & 1 deletion src/libcore/prelude.rs
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ pub use io::{print, println};

/* Reexported types and traits */

pub use clone::Clone;
pub use clone::{Clone, DeepClone};
pub use cmp::{Eq, ApproxEq, Ord, TotalEq, TotalOrd, Ordering, Less, Equal, Greater, Equiv};
pub use container::{Container, Mutable, Map, Set};
pub use hash::Hash;
Expand Down
78 changes: 73 additions & 5 deletions src/libstd/rc.rs
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@ impl<T: Owned> Drop for Rc<T> {


impl<T: Owned> Clone for Rc<T> {
/// Return a shallow copy of the reference counted pointer.
#[inline]
fn clone(&self) -> Rc<T> {
unsafe {
Expand All @@ -85,9 +86,38 @@ impl<T: Owned> Clone for Rc<T> {
}
}

impl<T: Owned + DeepClone> DeepClone for Rc<T> {
/// Return a deep copy of the reference counted pointer.
#[inline]
fn deep_clone(&self) -> Rc<T> {
Rc::new(self.borrow().deep_clone())
}
}

#[cfg(test)]
mod test_rc {
use super::*;
use core::cell::Cell;

#[test]
fn test_clone() {
let x = Rc::new(Cell(5));
let y = x.clone();
do x.borrow().with_mut_ref |inner| {
*inner = 20;
}
assert_eq!(y.borrow().take(), 20);
}

#[test]
fn test_deep_clone() {
let x = Rc::new(Cell(5));
let y = x.deep_clone();
do x.borrow().with_mut_ref |inner| {
*inner = 20;
}
assert_eq!(y.borrow().take(), 5);
}

#[test]
fn test_simple() {
Expand All @@ -96,7 +126,7 @@ mod test_rc {
}

#[test]
fn test_clone() {
fn test_simple_clone() {
let x = Rc::new(5);
let y = x.clone();
assert_eq!(*x.borrow(), 5);
Expand Down Expand Up @@ -149,24 +179,26 @@ pub impl<T: Owned> RcMut<T> {

/// Fails if there is already a mutable borrow of the box
#[inline]
fn with_borrow(&self, f: &fn(&T)) {
fn with_borrow<U>(&self, f: &fn(&T) -> U) -> U {
unsafe {
assert!((*self.ptr).borrow != Mutable);
let previous = (*self.ptr).borrow;
(*self.ptr).borrow = Immutable;
f(&(*self.ptr).value);
let res = f(&(*self.ptr).value);
(*self.ptr).borrow = previous;
res
}
}

/// Fails if there is already a mutable or immutable borrow of the box
#[inline]
fn with_mut_borrow(&self, f: &fn(&mut T)) {
fn with_mut_borrow<U>(&self, f: &fn(&mut T) -> U) -> U {
unsafe {
assert!((*self.ptr).borrow == Nothing);
(*self.ptr).borrow = Mutable;
f(&mut (*self.ptr).value);
let res = f(&mut (*self.ptr).value);
(*self.ptr).borrow = Nothing;
res
}
}
}
Expand Down Expand Up @@ -200,6 +232,7 @@ impl<T: Owned> Drop for RcMut<T> {
}

impl<T: Owned> Clone for RcMut<T> {
/// Return a shallow copy of the reference counted pointer.
#[inline]
fn clone(&self) -> RcMut<T> {
unsafe {
Expand All @@ -209,10 +242,45 @@ impl<T: Owned> Clone for RcMut<T> {
}
}

impl<T: Owned + DeepClone> DeepClone for RcMut<T> {
/// Return a deep copy of the reference counted pointer.
#[inline]
fn deep_clone(&self) -> RcMut<T> {
do self.with_borrow |x| {
// FIXME: #6497: should avoid freeze (slow)
RcMut::new(x.deep_clone())
}
}
}

#[cfg(test)]
mod test_rc_mut {
use super::*;

#[test]
fn test_clone() {
let x = RcMut::new(5);
let y = x.clone();
do x.with_mut_borrow |value| {
*value = 20;
}
do y.with_borrow |value| {
assert_eq!(*value, 20);
}
}

#[test]
fn test_deep_clone() {
let x = RcMut::new(5);
let y = x.deep_clone();
do x.with_mut_borrow |value| {
*value = 20;
}
do y.with_borrow |value| {
assert_eq!(*value, 5);
}
}

#[test]
fn borrow_many() {
let x = RcMut::new(5);
Expand Down