Skip to content

Commit 717f255

Browse files
committed
---
yaml --- r: 60282 b: refs/heads/master c: 7d81c80 h: refs/heads/master v: v3
1 parent d240339 commit 717f255

File tree

4 files changed

+126
-7
lines changed

4 files changed

+126
-7
lines changed

[refs]

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
---
2-
refs/heads/master: e1a199227635deddbdf010b3a79c2c96112909d7
2+
refs/heads/master: 7d81c80c65f8ed8f06d703b9d0635c8af521ea76
33
refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
44
refs/heads/snap-stage3: 2d28d645422c1617be58c8ca7ad9a457264ca850
55
refs/heads/try: c50a9d5b664478e533ba1d1d353213d70c8ad589

trunk/src/libcore/vec.rs

Lines changed: 31 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1681,8 +1681,8 @@ pub fn eachi_reverse<'r,T>(v: &'r [T],
16811681
*/
16821682
#[inline]
16831683
pub fn _each2<U, T>(v1: &[U], v2: &[T], f: &fn(u: &U, t: &T) -> bool) -> bool {
1684-
assert!(len(v1) == len(v2));
1685-
for uint::range(0u, len(v1)) |i| {
1684+
assert!(v1.len() == v2.len());
1685+
for uint::range(0u, v1.len()) |i| {
16861686
if !f(&v1[i], &v2[i]) {
16871687
return false;
16881688
}
@@ -1699,6 +1699,35 @@ pub fn each2<U, T>(v1: &[U], v2: &[T], f: &fn(u: &U, t: &T) -> bool) -> bool {
16991699
_each2(v1, v2, f)
17001700
}
17011701
1702+
/**
1703+
*
1704+
* Iterates over two vector with mutable.
1705+
*
1706+
* # Failure
1707+
*
1708+
* Both vectors must have the same length
1709+
*/
1710+
#[inline]
1711+
pub fn _each2_mut<U, T>(v1: &mut [U], v2: &mut [T], f: &fn(u: &mut U, t: &mut T) -> bool) -> bool {
1712+
assert!(v1.len() == v2.len());
1713+
for uint::range(0u, v1.len()) |i| {
1714+
if !f(&mut v1[i], &mut v2[i]) {
1715+
return false;
1716+
}
1717+
}
1718+
return true;
1719+
}
1720+
1721+
#[cfg(stage0)]
1722+
pub fn each2_mut<U, T>(v1: &mut [U], v2: &mut [T], f: &fn(u: &mut U, t: &mut T) -> bool) {
1723+
_each2_mut(v1, v2, f);
1724+
}
1725+
1726+
#[cfg(not(stage0))]
1727+
pub fn each2_mut<U, T>(v1: &mut [U], v2: &mut [T], f: &fn(u: &mut U, t: &mut T) -> bool) -> bool {
1728+
_each2_mut(v1, v2, f)
1729+
}
1730+
17021731
/**
17031732
* Iterate over all permutations of vector `v`.
17041733
*

trunk/src/libstd/priority_queue.rs

Lines changed: 56 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,14 @@
1212
1313
use core::old_iter::BaseIter;
1414
use core::util::{replace, swap};
15-
use core::unstable::intrinsics::{init, move_val_init};
15+
16+
#[abi = "rust-intrinsic"]
17+
extern "rust-intrinsic" {
18+
fn move_val_init<T>(dst: &mut T, src: T);
19+
fn init<T>() -> T;
20+
#[cfg(not(stage0))]
21+
fn uninit<T>() -> T;
22+
}
1623

1724
pub struct PriorityQueue<T> {
1825
priv data: ~[T],
@@ -134,13 +141,33 @@ pub impl <T:Ord> PriorityQueue<T> {
134141

135142
// The implementations of siftup and siftdown use unsafe blocks in
136143
// order to move an element out of the vector (leaving behind a
137-
// zeroed element), shift along the others and move it back into the
144+
// junk element), shift along the others and move it back into the
138145
// vector over the junk element. This reduces the constant factor
139146
// compared to using swaps, which involves twice as many moves.
140147

148+
#[cfg(not(stage0))]
149+
priv fn siftup(&mut self, start: uint, mut pos: uint) {
150+
unsafe {
151+
let new = *ptr::to_unsafe_ptr(&self.data[pos]);
152+
153+
while pos > start {
154+
let parent = (pos - 1) >> 1;
155+
if new > self.data[parent] {
156+
let x = replace(&mut self.data[parent], uninit());
157+
move_val_init(&mut self.data[pos], x);
158+
pos = parent;
159+
loop
160+
}
161+
break
162+
}
163+
move_val_init(&mut self.data[pos], new);
164+
}
165+
}
166+
167+
#[cfg(stage0)]
141168
priv fn siftup(&mut self, start: uint, mut pos: uint) {
142169
unsafe {
143-
let new = replace(&mut self.data[pos], init());
170+
let new = *ptr::to_unsafe_ptr(&self.data[pos]);
144171

145172
while pos > start {
146173
let parent = (pos - 1) >> 1;
@@ -156,10 +183,35 @@ pub impl <T:Ord> PriorityQueue<T> {
156183
}
157184
}
158185

186+
187+
#[cfg(not(stage0))]
188+
priv fn siftdown_range(&mut self, mut pos: uint, end: uint) {
189+
unsafe {
190+
let start = pos;
191+
let new = *ptr::to_unsafe_ptr(&self.data[pos]);
192+
193+
let mut child = 2 * pos + 1;
194+
while child < end {
195+
let right = child + 1;
196+
if right < end && !(self.data[child] > self.data[right]) {
197+
child = right;
198+
}
199+
let x = replace(&mut self.data[child], uninit());
200+
move_val_init(&mut self.data[pos], x);
201+
pos = child;
202+
child = 2 * pos + 1;
203+
}
204+
205+
move_val_init(&mut self.data[pos], new);
206+
self.siftup(start, pos);
207+
}
208+
}
209+
210+
#[cfg(stage0)]
159211
priv fn siftdown_range(&mut self, mut pos: uint, end: uint) {
160212
unsafe {
161213
let start = pos;
162-
let new = replace(&mut self.data[pos], init());
214+
let new = *ptr::to_unsafe_ptr(&self.data[pos]);
163215

164216
let mut child = 2 * pos + 1;
165217
while child < end {
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
2+
// file at the top-level directory of this distribution and at
3+
// http://rust-lang.org/COPYRIGHT.
4+
//
5+
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6+
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7+
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8+
// option. This file may not be copied, modified, or distributed
9+
// except according to those terms.
10+
11+
// -*- rust -*-
12+
fn main(){
13+
let mut t1 = ~[];
14+
t1.push('a');
15+
16+
let mut t2 = ~[];
17+
t2.push('b');
18+
19+
for vec::each2_mut(t1, t2) | i1, i2 | {
20+
assert!(*i1 == 'a');
21+
assert!(*i2 == 'b');
22+
}
23+
24+
for vec::each2(t1, t2) | i1, i2 | {
25+
io::println(fmt!("after t1: %?, t2: %?", i1, i2));
26+
}
27+
28+
for vec::each2_mut(t1, t2) | i1, i2 | {
29+
*i1 = 'b';
30+
*i2 = 'a';
31+
assert!(*i1 == 'b');
32+
assert!(*i2 == 'a');
33+
}
34+
35+
for vec::each2(t1, t2) | i1, i2 | {
36+
io::println(fmt!("before t1: %?, t2: %?", i1, i2));
37+
}
38+
}

0 commit comments

Comments
 (0)