##// END OF EJS Templates
rust-ancestors: adjust branches and inline comments per previous change...
Yuya Nishihara -
r41170:17a19567 default
parent child Browse files
Show More
@@ -1,780 +1,775
1 // ancestors.rs
1 // ancestors.rs
2 //
2 //
3 // Copyright 2018 Georges Racinet <gracinet@anybox.fr>
3 // Copyright 2018 Georges Racinet <gracinet@anybox.fr>
4 //
4 //
5 // This software may be used and distributed according to the terms of the
5 // This software may be used and distributed according to the terms of the
6 // GNU General Public License version 2 or any later version.
6 // GNU General Public License version 2 or any later version.
7
7
8 //! Rust versions of generic DAG ancestors algorithms for Mercurial
8 //! Rust versions of generic DAG ancestors algorithms for Mercurial
9
9
10 use super::{Graph, GraphError, Revision, NULL_REVISION};
10 use super::{Graph, GraphError, Revision, NULL_REVISION};
11 use std::cmp::max;
11 use std::cmp::max;
12 use std::collections::{BinaryHeap, HashSet};
12 use std::collections::{BinaryHeap, HashSet};
13
13
14 /// Iterator over the ancestors of a given list of revisions
14 /// Iterator over the ancestors of a given list of revisions
15 /// This is a generic type, defined and implemented for any Graph, so that
15 /// This is a generic type, defined and implemented for any Graph, so that
16 /// it's easy to
16 /// it's easy to
17 ///
17 ///
18 /// - unit test in pure Rust
18 /// - unit test in pure Rust
19 /// - bind to main Mercurial code, potentially in several ways and have these
19 /// - bind to main Mercurial code, potentially in several ways and have these
20 /// bindings evolve over time
20 /// bindings evolve over time
21 pub struct AncestorsIterator<G: Graph> {
21 pub struct AncestorsIterator<G: Graph> {
22 graph: G,
22 graph: G,
23 visit: BinaryHeap<Revision>,
23 visit: BinaryHeap<Revision>,
24 seen: HashSet<Revision>,
24 seen: HashSet<Revision>,
25 stoprev: Revision,
25 stoprev: Revision,
26 }
26 }
27
27
28 /// Lazy ancestors set, backed by AncestorsIterator
28 /// Lazy ancestors set, backed by AncestorsIterator
29 pub struct LazyAncestors<G: Graph + Clone> {
29 pub struct LazyAncestors<G: Graph + Clone> {
30 graph: G,
30 graph: G,
31 containsiter: AncestorsIterator<G>,
31 containsiter: AncestorsIterator<G>,
32 initrevs: Vec<Revision>,
32 initrevs: Vec<Revision>,
33 stoprev: Revision,
33 stoprev: Revision,
34 inclusive: bool,
34 inclusive: bool,
35 }
35 }
36
36
37 pub struct MissingAncestors<G: Graph> {
37 pub struct MissingAncestors<G: Graph> {
38 graph: G,
38 graph: G,
39 bases: HashSet<Revision>,
39 bases: HashSet<Revision>,
40 }
40 }
41
41
42 impl<G: Graph> AncestorsIterator<G> {
42 impl<G: Graph> AncestorsIterator<G> {
43 /// Constructor.
43 /// Constructor.
44 ///
44 ///
45 /// if `inclusive` is true, then the init revisions are emitted in
45 /// if `inclusive` is true, then the init revisions are emitted in
46 /// particular, otherwise iteration starts from their parents.
46 /// particular, otherwise iteration starts from their parents.
47 pub fn new(
47 pub fn new(
48 graph: G,
48 graph: G,
49 initrevs: impl IntoIterator<Item = Revision>,
49 initrevs: impl IntoIterator<Item = Revision>,
50 stoprev: Revision,
50 stoprev: Revision,
51 inclusive: bool,
51 inclusive: bool,
52 ) -> Result<Self, GraphError> {
52 ) -> Result<Self, GraphError> {
53 let filtered_initrevs = initrevs.into_iter().filter(|&r| r >= stoprev);
53 let filtered_initrevs = initrevs.into_iter().filter(|&r| r >= stoprev);
54 if inclusive {
54 if inclusive {
55 let visit: BinaryHeap<Revision> = filtered_initrevs.collect();
55 let visit: BinaryHeap<Revision> = filtered_initrevs.collect();
56 let seen = visit.iter().map(|&x| x).collect();
56 let seen = visit.iter().map(|&x| x).collect();
57 return Ok(AncestorsIterator {
57 return Ok(AncestorsIterator {
58 visit: visit,
58 visit: visit,
59 seen: seen,
59 seen: seen,
60 stoprev: stoprev,
60 stoprev: stoprev,
61 graph: graph,
61 graph: graph,
62 });
62 });
63 }
63 }
64 let mut this = AncestorsIterator {
64 let mut this = AncestorsIterator {
65 visit: BinaryHeap::new(),
65 visit: BinaryHeap::new(),
66 seen: HashSet::new(),
66 seen: HashSet::new(),
67 stoprev: stoprev,
67 stoprev: stoprev,
68 graph: graph,
68 graph: graph,
69 };
69 };
70 this.seen.insert(NULL_REVISION);
70 this.seen.insert(NULL_REVISION);
71 for rev in filtered_initrevs {
71 for rev in filtered_initrevs {
72 for parent in this.graph.parents(rev)?.iter().cloned() {
72 for parent in this.graph.parents(rev)?.iter().cloned() {
73 this.conditionally_push_rev(parent);
73 this.conditionally_push_rev(parent);
74 }
74 }
75 }
75 }
76 Ok(this)
76 Ok(this)
77 }
77 }
78
78
79 #[inline]
79 #[inline]
80 fn conditionally_push_rev(&mut self, rev: Revision) {
80 fn conditionally_push_rev(&mut self, rev: Revision) {
81 if self.stoprev <= rev && !self.seen.contains(&rev) {
81 if self.stoprev <= rev && !self.seen.contains(&rev) {
82 self.seen.insert(rev);
82 self.seen.insert(rev);
83 self.visit.push(rev);
83 self.visit.push(rev);
84 }
84 }
85 }
85 }
86
86
87 /// Consumes partially the iterator to tell if the given target
87 /// Consumes partially the iterator to tell if the given target
88 /// revision
88 /// revision
89 /// is in the ancestors it emits.
89 /// is in the ancestors it emits.
90 /// This is meant for iterators actually dedicated to that kind of
90 /// This is meant for iterators actually dedicated to that kind of
91 /// purpose
91 /// purpose
92 pub fn contains(&mut self, target: Revision) -> Result<bool, GraphError> {
92 pub fn contains(&mut self, target: Revision) -> Result<bool, GraphError> {
93 if self.seen.contains(&target) && target != NULL_REVISION {
93 if self.seen.contains(&target) && target != NULL_REVISION {
94 return Ok(true);
94 return Ok(true);
95 }
95 }
96 for item in self {
96 for item in self {
97 let rev = item?;
97 let rev = item?;
98 if rev == target {
98 if rev == target {
99 return Ok(true);
99 return Ok(true);
100 }
100 }
101 if rev < target {
101 if rev < target {
102 return Ok(false);
102 return Ok(false);
103 }
103 }
104 }
104 }
105 Ok(false)
105 Ok(false)
106 }
106 }
107
107
108 pub fn peek(&self) -> Option<Revision> {
108 pub fn peek(&self) -> Option<Revision> {
109 self.visit.peek().map(|&r| r)
109 self.visit.peek().map(|&r| r)
110 }
110 }
111
111
112 /// Tell if the iterator is about an empty set
112 /// Tell if the iterator is about an empty set
113 ///
113 ///
114 /// The result does not depend whether the iterator has been consumed
114 /// The result does not depend whether the iterator has been consumed
115 /// or not.
115 /// or not.
116 /// This is mostly meant for iterators backing a lazy ancestors set
116 /// This is mostly meant for iterators backing a lazy ancestors set
117 pub fn is_empty(&self) -> bool {
117 pub fn is_empty(&self) -> bool {
118 if self.visit.len() > 0 {
118 if self.visit.len() > 0 {
119 return false;
119 return false;
120 }
120 }
121 if self.seen.len() > 1 {
121 if self.seen.len() > 1 {
122 return false;
122 return false;
123 }
123 }
124 // at this point, the seen set is at most a singleton.
124 // at this point, the seen set is at most a singleton.
125 // If not `self.inclusive`, it's still possible that it has only
125 // If not `self.inclusive`, it's still possible that it has only
126 // the null revision
126 // the null revision
127 self.seen.is_empty() || self.seen.contains(&NULL_REVISION)
127 self.seen.is_empty() || self.seen.contains(&NULL_REVISION)
128 }
128 }
129 }
129 }
130
130
131 /// Main implementation for the iterator
131 /// Main implementation for the iterator
132 ///
132 ///
133 /// The algorithm is the same as in `_lazyancestorsiter()` from `ancestors.py`
133 /// The algorithm is the same as in `_lazyancestorsiter()` from `ancestors.py`
134 /// with a few non crucial differences:
134 /// with a few non crucial differences:
135 ///
135 ///
136 /// - there's no filtering of invalid parent revisions. Actually, it should be
136 /// - there's no filtering of invalid parent revisions. Actually, it should be
137 /// consistent and more efficient to filter them from the end caller.
137 /// consistent and more efficient to filter them from the end caller.
138 /// - we don't have the optimization for adjacent revisions (i.e., the case
138 /// - we don't have the optimization for adjacent revisions (i.e., the case
139 /// where `p1 == rev - 1`), because it amounts to update the first element of
139 /// where `p1 == rev - 1`), because it amounts to update the first element of
140 /// the heap without sifting, which Rust's BinaryHeap doesn't let us do.
140 /// the heap without sifting, which Rust's BinaryHeap doesn't let us do.
141 /// - we save a few pushes by comparing with `stoprev` before pushing
141 /// - we save a few pushes by comparing with `stoprev` before pushing
142 impl<G: Graph> Iterator for AncestorsIterator<G> {
142 impl<G: Graph> Iterator for AncestorsIterator<G> {
143 type Item = Result<Revision, GraphError>;
143 type Item = Result<Revision, GraphError>;
144
144
145 fn next(&mut self) -> Option<Self::Item> {
145 fn next(&mut self) -> Option<Self::Item> {
146 let current = match self.visit.peek() {
146 let current = match self.visit.peek() {
147 None => {
147 None => {
148 return None;
148 return None;
149 }
149 }
150 Some(c) => *c,
150 Some(c) => *c,
151 };
151 };
152 let [p1, p2] = match self.graph.parents(current) {
152 let [p1, p2] = match self.graph.parents(current) {
153 Ok(ps) => ps,
153 Ok(ps) => ps,
154 Err(e) => return Some(Err(e)),
154 Err(e) => return Some(Err(e)),
155 };
155 };
156 if p1 < self.stoprev || self.seen.contains(&p1) {
156 if p1 < self.stoprev || self.seen.contains(&p1) {
157 self.visit.pop();
157 self.visit.pop();
158 } else {
158 } else {
159 *(self.visit.peek_mut().unwrap()) = p1;
159 *(self.visit.peek_mut().unwrap()) = p1;
160 self.seen.insert(p1);
160 self.seen.insert(p1);
161 };
161 };
162
162
163 self.conditionally_push_rev(p2);
163 self.conditionally_push_rev(p2);
164 Some(Ok(current))
164 Some(Ok(current))
165 }
165 }
166 }
166 }
167
167
168 impl<G: Graph + Clone> LazyAncestors<G> {
168 impl<G: Graph + Clone> LazyAncestors<G> {
169 pub fn new(
169 pub fn new(
170 graph: G,
170 graph: G,
171 initrevs: impl IntoIterator<Item = Revision>,
171 initrevs: impl IntoIterator<Item = Revision>,
172 stoprev: Revision,
172 stoprev: Revision,
173 inclusive: bool,
173 inclusive: bool,
174 ) -> Result<Self, GraphError> {
174 ) -> Result<Self, GraphError> {
175 let v: Vec<Revision> = initrevs.into_iter().collect();
175 let v: Vec<Revision> = initrevs.into_iter().collect();
176 Ok(LazyAncestors {
176 Ok(LazyAncestors {
177 graph: graph.clone(),
177 graph: graph.clone(),
178 containsiter: AncestorsIterator::new(
178 containsiter: AncestorsIterator::new(
179 graph,
179 graph,
180 v.iter().cloned(),
180 v.iter().cloned(),
181 stoprev,
181 stoprev,
182 inclusive,
182 inclusive,
183 )?,
183 )?,
184 initrevs: v,
184 initrevs: v,
185 stoprev: stoprev,
185 stoprev: stoprev,
186 inclusive: inclusive,
186 inclusive: inclusive,
187 })
187 })
188 }
188 }
189
189
190 pub fn contains(&mut self, rev: Revision) -> Result<bool, GraphError> {
190 pub fn contains(&mut self, rev: Revision) -> Result<bool, GraphError> {
191 self.containsiter.contains(rev)
191 self.containsiter.contains(rev)
192 }
192 }
193
193
194 pub fn is_empty(&self) -> bool {
194 pub fn is_empty(&self) -> bool {
195 self.containsiter.is_empty()
195 self.containsiter.is_empty()
196 }
196 }
197
197
198 pub fn iter(&self) -> AncestorsIterator<G> {
198 pub fn iter(&self) -> AncestorsIterator<G> {
199 // the arguments being the same as for self.containsiter, we know
199 // the arguments being the same as for self.containsiter, we know
200 // for sure that AncestorsIterator constructor can't fail
200 // for sure that AncestorsIterator constructor can't fail
201 AncestorsIterator::new(
201 AncestorsIterator::new(
202 self.graph.clone(),
202 self.graph.clone(),
203 self.initrevs.iter().cloned(),
203 self.initrevs.iter().cloned(),
204 self.stoprev,
204 self.stoprev,
205 self.inclusive,
205 self.inclusive,
206 )
206 )
207 .unwrap()
207 .unwrap()
208 }
208 }
209 }
209 }
210
210
211 impl<G: Graph> MissingAncestors<G> {
211 impl<G: Graph> MissingAncestors<G> {
212 pub fn new(graph: G, bases: impl IntoIterator<Item = Revision>) -> Self {
212 pub fn new(graph: G, bases: impl IntoIterator<Item = Revision>) -> Self {
213 let mut bases: HashSet<Revision> = bases.into_iter().collect();
213 let mut bases: HashSet<Revision> = bases.into_iter().collect();
214 if bases.is_empty() {
214 if bases.is_empty() {
215 bases.insert(NULL_REVISION);
215 bases.insert(NULL_REVISION);
216 }
216 }
217 MissingAncestors { graph, bases }
217 MissingAncestors { graph, bases }
218 }
218 }
219
219
220 pub fn has_bases(&self) -> bool {
220 pub fn has_bases(&self) -> bool {
221 self.bases.iter().any(|&b| b != NULL_REVISION)
221 self.bases.iter().any(|&b| b != NULL_REVISION)
222 }
222 }
223
223
224 /// Return a reference to current bases.
224 /// Return a reference to current bases.
225 ///
225 ///
226 /// This is useful in unit tests, but also setdiscovery.py does
226 /// This is useful in unit tests, but also setdiscovery.py does
227 /// read the bases attribute of a ancestor.missingancestors instance.
227 /// read the bases attribute of a ancestor.missingancestors instance.
228 pub fn get_bases<'a>(&'a self) -> &'a HashSet<Revision> {
228 pub fn get_bases<'a>(&'a self) -> &'a HashSet<Revision> {
229 &self.bases
229 &self.bases
230 }
230 }
231
231
232 pub fn add_bases(
232 pub fn add_bases(
233 &mut self,
233 &mut self,
234 new_bases: impl IntoIterator<Item = Revision>,
234 new_bases: impl IntoIterator<Item = Revision>,
235 ) {
235 ) {
236 self.bases.extend(new_bases);
236 self.bases.extend(new_bases);
237 }
237 }
238
238
239 /// Remove all ancestors of self.bases from the revs set (in place)
239 /// Remove all ancestors of self.bases from the revs set (in place)
240 pub fn remove_ancestors_from(
240 pub fn remove_ancestors_from(
241 &mut self,
241 &mut self,
242 revs: &mut HashSet<Revision>,
242 revs: &mut HashSet<Revision>,
243 ) -> Result<(), GraphError> {
243 ) -> Result<(), GraphError> {
244 revs.retain(|r| !self.bases.contains(r));
244 revs.retain(|r| !self.bases.contains(r));
245 // the null revision is always an ancestor
245 // the null revision is always an ancestor
246 revs.remove(&NULL_REVISION);
246 revs.remove(&NULL_REVISION);
247 if revs.is_empty() {
247 if revs.is_empty() {
248 return Ok(());
248 return Ok(());
249 }
249 }
250 // anything in revs > start is definitely not an ancestor of bases
250 // anything in revs > start is definitely not an ancestor of bases
251 // revs <= start need to be investigated
251 // revs <= start need to be investigated
252 // TODO optim: if a missingancestors is to be used several times,
252 // TODO optim: if a missingancestors is to be used several times,
253 // we shouldn't need to iterate each time on bases
253 // we shouldn't need to iterate each time on bases
254 let start = match self.bases.iter().cloned().max() {
254 let start = match self.bases.iter().cloned().max() {
255 Some(m) => m,
255 Some(m) => m,
256 None => {
256 None => {
257 // bases is empty (shouldn't happen, but let's be safe)
257 // bases is empty (shouldn't happen, but let's be safe)
258 return Ok(());
258 return Ok(());
259 }
259 }
260 };
260 };
261 // whatever happens, we'll keep at least keepcount of them
261 // whatever happens, we'll keep at least keepcount of them
262 // knowing this gives us a earlier stop condition than
262 // knowing this gives us a earlier stop condition than
263 // going all the way to the root
263 // going all the way to the root
264 let keepcount = revs.iter().filter(|r| **r > start).count();
264 let keepcount = revs.iter().filter(|r| **r > start).count();
265
265
266 let mut curr = start;
266 let mut curr = start;
267 while curr != NULL_REVISION && revs.len() > keepcount {
267 while curr != NULL_REVISION && revs.len() > keepcount {
268 if self.bases.contains(&curr) {
268 if self.bases.contains(&curr) {
269 revs.remove(&curr);
269 revs.remove(&curr);
270 self.add_parents(curr)?;
270 self.add_parents(curr)?;
271 }
271 }
272 curr -= 1;
272 curr -= 1;
273 }
273 }
274 Ok(())
274 Ok(())
275 }
275 }
276
276
277 /// Add rev's parents to self.bases
277 /// Add rev's parents to self.bases
278 #[inline]
278 #[inline]
279 fn add_parents(&mut self, rev: Revision) -> Result<(), GraphError> {
279 fn add_parents(&mut self, rev: Revision) -> Result<(), GraphError> {
280 // No need to bother the set with inserting NULL_REVISION over and
280 // No need to bother the set with inserting NULL_REVISION over and
281 // over
281 // over
282 for p in self.graph.parents(rev)?.iter().cloned() {
282 for p in self.graph.parents(rev)?.iter().cloned() {
283 if p != NULL_REVISION {
283 if p != NULL_REVISION {
284 self.bases.insert(p);
284 self.bases.insert(p);
285 }
285 }
286 }
286 }
287 Ok(())
287 Ok(())
288 }
288 }
289
289
290 /// Return all the ancestors of revs that are not ancestors of self.bases
290 /// Return all the ancestors of revs that are not ancestors of self.bases
291 ///
291 ///
292 /// This may include elements from revs.
292 /// This may include elements from revs.
293 ///
293 ///
294 /// Equivalent to the revset (::revs - ::self.bases). Revs are returned in
294 /// Equivalent to the revset (::revs - ::self.bases). Revs are returned in
295 /// revision number order, which is a topological order.
295 /// revision number order, which is a topological order.
296 pub fn missing_ancestors(
296 pub fn missing_ancestors(
297 &mut self,
297 &mut self,
298 revs: impl IntoIterator<Item = Revision>,
298 revs: impl IntoIterator<Item = Revision>,
299 ) -> Result<Vec<Revision>, GraphError> {
299 ) -> Result<Vec<Revision>, GraphError> {
300 // just for convenience and comparison with Python version
300 // just for convenience and comparison with Python version
301 let bases_visit = &mut self.bases;
301 let bases_visit = &mut self.bases;
302 let mut revs: HashSet<Revision> = revs
302 let mut revs: HashSet<Revision> = revs
303 .into_iter()
303 .into_iter()
304 .filter(|r| !bases_visit.contains(r))
304 .filter(|r| !bases_visit.contains(r))
305 .collect();
305 .collect();
306 let revs_visit = &mut revs;
306 let revs_visit = &mut revs;
307 let mut both_visit: HashSet<Revision> =
307 let mut both_visit: HashSet<Revision> =
308 revs_visit.intersection(&bases_visit).cloned().collect();
308 revs_visit.intersection(&bases_visit).cloned().collect();
309 if revs_visit.is_empty() {
309 if revs_visit.is_empty() {
310 return Ok(Vec::new());
310 return Ok(Vec::new());
311 }
311 }
312
312
313 let max_bases =
313 let max_bases =
314 bases_visit.iter().cloned().max().unwrap_or(NULL_REVISION);
314 bases_visit.iter().cloned().max().unwrap_or(NULL_REVISION);
315 let max_revs =
315 let max_revs =
316 revs_visit.iter().cloned().max().unwrap_or(NULL_REVISION);
316 revs_visit.iter().cloned().max().unwrap_or(NULL_REVISION);
317 let start = max(max_bases, max_revs);
317 let start = max(max_bases, max_revs);
318
318
319 // TODO heuristics for with_capacity()?
319 // TODO heuristics for with_capacity()?
320 let mut missing: Vec<Revision> = Vec::new();
320 let mut missing: Vec<Revision> = Vec::new();
321 for curr in (0..=start).rev() {
321 for curr in (0..=start).rev() {
322 if revs_visit.is_empty() {
322 if revs_visit.is_empty() {
323 break;
323 break;
324 }
324 }
325 if both_visit.contains(&curr) {
325 if both_visit.contains(&curr) {
326 // curr's parents might have made it into revs_visit through
326 // curr's parents might have made it into revs_visit through
327 // another path
327 // another path
328 // TODO optim: Rust's HashSet.remove returns a boolean telling
328 // TODO optim: Rust's HashSet.remove returns a boolean telling
329 // if it happened. This will spare us one set lookup
329 // if it happened. This will spare us one set lookup
330 both_visit.remove(&curr);
330 both_visit.remove(&curr);
331 for p in self.graph.parents(curr)?.iter().cloned() {
331 for p in self.graph.parents(curr)?.iter().cloned() {
332 if p == NULL_REVISION {
332 if p == NULL_REVISION {
333 continue;
333 continue;
334 }
334 }
335 revs_visit.remove(&p);
335 revs_visit.remove(&p);
336 bases_visit.insert(p);
336 bases_visit.insert(p);
337 both_visit.insert(p);
337 both_visit.insert(p);
338 }
338 }
339 continue;
339 } else if revs_visit.remove(&curr) {
340 }
341 if revs_visit.remove(&curr) {
342 missing.push(curr);
340 missing.push(curr);
343 for p in self.graph.parents(curr)?.iter().cloned() {
341 for p in self.graph.parents(curr)?.iter().cloned() {
344 if p == NULL_REVISION {
342 if p == NULL_REVISION {
345 continue;
343 continue;
346 }
344 }
347 if bases_visit.contains(&p) || both_visit.contains(&p) {
345 if bases_visit.contains(&p) || both_visit.contains(&p) {
348 // p is implicitely in this_visit.
346 // p is an ancestor of revs_visit, and is implicitly
349 // This means p is or should be in bothvisit
347 // in bases_visit, which means p is ::revs & ::bases.
350 // TODO optim: hence if bothvisit, we look up twice
348 // TODO optim: hence if bothvisit, we look up twice
351 revs_visit.remove(&p);
349 revs_visit.remove(&p);
352 bases_visit.insert(p);
350 bases_visit.insert(p);
353 both_visit.insert(p);
351 both_visit.insert(p);
354 } else {
352 } else {
355 // visit later
353 // visit later
356 revs_visit.insert(p);
354 revs_visit.insert(p);
357 }
355 }
358 }
356 }
359 } else if bases_visit.contains(&curr) {
357 } else if bases_visit.contains(&curr) {
360 for p in self.graph.parents(curr)?.iter().cloned() {
358 for p in self.graph.parents(curr)?.iter().cloned() {
361 if p == NULL_REVISION {
359 if p == NULL_REVISION {
362 continue;
360 continue;
363 }
361 }
364 if revs_visit.contains(&p) || both_visit.contains(&p) {
362 if revs_visit.contains(&p) || both_visit.contains(&p) {
365 // p is implicitely in this_visit.
363 // p is an ancestor of bases_visit, and is implicitly
366 // This means p is or should be in bothvisit
364 // in revs_visit, which means p is ::revs & ::bases.
367 // TODO optim: hence if bothvisit, we look up twice
365 // TODO optim: hence if bothvisit, we look up twice
368 revs_visit.remove(&p);
366 revs_visit.remove(&p);
369 bases_visit.insert(p);
367 bases_visit.insert(p);
370 both_visit.insert(p);
368 both_visit.insert(p);
371 } else {
369 } else {
372 // visit later
373 bases_visit.insert(p);
370 bases_visit.insert(p);
374 }
371 }
375 }
372 }
376 } else {
377 // not an ancestor of revs or bases: ignore
378 }
373 }
379 }
374 }
380 missing.reverse();
375 missing.reverse();
381 Ok(missing)
376 Ok(missing)
382 }
377 }
383 }
378 }
384
379
385 #[cfg(test)]
380 #[cfg(test)]
386 mod tests {
381 mod tests {
387
382
388 use super::*;
383 use super::*;
389 use std::iter::FromIterator;
384 use std::iter::FromIterator;
390
385
391 #[derive(Clone, Debug)]
386 #[derive(Clone, Debug)]
392 struct Stub;
387 struct Stub;
393
388
394 /// This is the same as the dict from test-ancestors.py
389 /// This is the same as the dict from test-ancestors.py
395 impl Graph for Stub {
390 impl Graph for Stub {
396 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
391 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
397 match rev {
392 match rev {
398 0 => Ok([-1, -1]),
393 0 => Ok([-1, -1]),
399 1 => Ok([0, -1]),
394 1 => Ok([0, -1]),
400 2 => Ok([1, -1]),
395 2 => Ok([1, -1]),
401 3 => Ok([1, -1]),
396 3 => Ok([1, -1]),
402 4 => Ok([2, -1]),
397 4 => Ok([2, -1]),
403 5 => Ok([4, -1]),
398 5 => Ok([4, -1]),
404 6 => Ok([4, -1]),
399 6 => Ok([4, -1]),
405 7 => Ok([4, -1]),
400 7 => Ok([4, -1]),
406 8 => Ok([-1, -1]),
401 8 => Ok([-1, -1]),
407 9 => Ok([6, 7]),
402 9 => Ok([6, 7]),
408 10 => Ok([5, -1]),
403 10 => Ok([5, -1]),
409 11 => Ok([3, 7]),
404 11 => Ok([3, 7]),
410 12 => Ok([9, -1]),
405 12 => Ok([9, -1]),
411 13 => Ok([8, -1]),
406 13 => Ok([8, -1]),
412 r => Err(GraphError::ParentOutOfRange(r)),
407 r => Err(GraphError::ParentOutOfRange(r)),
413 }
408 }
414 }
409 }
415 }
410 }
416
411
417 fn list_ancestors<G: Graph>(
412 fn list_ancestors<G: Graph>(
418 graph: G,
413 graph: G,
419 initrevs: Vec<Revision>,
414 initrevs: Vec<Revision>,
420 stoprev: Revision,
415 stoprev: Revision,
421 inclusive: bool,
416 inclusive: bool,
422 ) -> Vec<Revision> {
417 ) -> Vec<Revision> {
423 AncestorsIterator::new(graph, initrevs, stoprev, inclusive)
418 AncestorsIterator::new(graph, initrevs, stoprev, inclusive)
424 .unwrap()
419 .unwrap()
425 .map(|res| res.unwrap())
420 .map(|res| res.unwrap())
426 .collect()
421 .collect()
427 }
422 }
428
423
429 #[test]
424 #[test]
430 /// Same tests as test-ancestor.py, without membership
425 /// Same tests as test-ancestor.py, without membership
431 /// (see also test-ancestor.py.out)
426 /// (see also test-ancestor.py.out)
432 fn test_list_ancestor() {
427 fn test_list_ancestor() {
433 assert_eq!(list_ancestors(Stub, vec![], 0, false), vec![]);
428 assert_eq!(list_ancestors(Stub, vec![], 0, false), vec![]);
434 assert_eq!(
429 assert_eq!(
435 list_ancestors(Stub, vec![11, 13], 0, false),
430 list_ancestors(Stub, vec![11, 13], 0, false),
436 vec![8, 7, 4, 3, 2, 1, 0]
431 vec![8, 7, 4, 3, 2, 1, 0]
437 );
432 );
438 assert_eq!(list_ancestors(Stub, vec![1, 3], 0, false), vec![1, 0]);
433 assert_eq!(list_ancestors(Stub, vec![1, 3], 0, false), vec![1, 0]);
439 assert_eq!(
434 assert_eq!(
440 list_ancestors(Stub, vec![11, 13], 0, true),
435 list_ancestors(Stub, vec![11, 13], 0, true),
441 vec![13, 11, 8, 7, 4, 3, 2, 1, 0]
436 vec![13, 11, 8, 7, 4, 3, 2, 1, 0]
442 );
437 );
443 assert_eq!(list_ancestors(Stub, vec![11, 13], 6, false), vec![8, 7]);
438 assert_eq!(list_ancestors(Stub, vec![11, 13], 6, false), vec![8, 7]);
444 assert_eq!(
439 assert_eq!(
445 list_ancestors(Stub, vec![11, 13], 6, true),
440 list_ancestors(Stub, vec![11, 13], 6, true),
446 vec![13, 11, 8, 7]
441 vec![13, 11, 8, 7]
447 );
442 );
448 assert_eq!(list_ancestors(Stub, vec![11, 13], 11, true), vec![13, 11]);
443 assert_eq!(list_ancestors(Stub, vec![11, 13], 11, true), vec![13, 11]);
449 assert_eq!(list_ancestors(Stub, vec![11, 13], 12, true), vec![13]);
444 assert_eq!(list_ancestors(Stub, vec![11, 13], 12, true), vec![13]);
450 assert_eq!(
445 assert_eq!(
451 list_ancestors(Stub, vec![10, 1], 0, true),
446 list_ancestors(Stub, vec![10, 1], 0, true),
452 vec![10, 5, 4, 2, 1, 0]
447 vec![10, 5, 4, 2, 1, 0]
453 );
448 );
454 }
449 }
455
450
456 #[test]
451 #[test]
457 /// Corner case that's not directly in test-ancestors.py, but
452 /// Corner case that's not directly in test-ancestors.py, but
458 /// that happens quite often, as demonstrated by running the whole
453 /// that happens quite often, as demonstrated by running the whole
459 /// suite.
454 /// suite.
460 /// For instance, run tests/test-obsolete-checkheads.t
455 /// For instance, run tests/test-obsolete-checkheads.t
461 fn test_nullrev_input() {
456 fn test_nullrev_input() {
462 let mut iter =
457 let mut iter =
463 AncestorsIterator::new(Stub, vec![-1], 0, false).unwrap();
458 AncestorsIterator::new(Stub, vec![-1], 0, false).unwrap();
464 assert_eq!(iter.next(), None)
459 assert_eq!(iter.next(), None)
465 }
460 }
466
461
467 #[test]
462 #[test]
468 fn test_contains() {
463 fn test_contains() {
469 let mut lazy =
464 let mut lazy =
470 AncestorsIterator::new(Stub, vec![10, 1], 0, true).unwrap();
465 AncestorsIterator::new(Stub, vec![10, 1], 0, true).unwrap();
471 assert!(lazy.contains(1).unwrap());
466 assert!(lazy.contains(1).unwrap());
472 assert!(!lazy.contains(3).unwrap());
467 assert!(!lazy.contains(3).unwrap());
473
468
474 let mut lazy =
469 let mut lazy =
475 AncestorsIterator::new(Stub, vec![0], 0, false).unwrap();
470 AncestorsIterator::new(Stub, vec![0], 0, false).unwrap();
476 assert!(!lazy.contains(NULL_REVISION).unwrap());
471 assert!(!lazy.contains(NULL_REVISION).unwrap());
477 }
472 }
478
473
479 #[test]
474 #[test]
480 fn test_peek() {
475 fn test_peek() {
481 let mut iter =
476 let mut iter =
482 AncestorsIterator::new(Stub, vec![10], 0, true).unwrap();
477 AncestorsIterator::new(Stub, vec![10], 0, true).unwrap();
483 // peek() gives us the next value
478 // peek() gives us the next value
484 assert_eq!(iter.peek(), Some(10));
479 assert_eq!(iter.peek(), Some(10));
485 // but it's not been consumed
480 // but it's not been consumed
486 assert_eq!(iter.next(), Some(Ok(10)));
481 assert_eq!(iter.next(), Some(Ok(10)));
487 // and iteration resumes normally
482 // and iteration resumes normally
488 assert_eq!(iter.next(), Some(Ok(5)));
483 assert_eq!(iter.next(), Some(Ok(5)));
489
484
490 // let's drain the iterator to test peek() at the end
485 // let's drain the iterator to test peek() at the end
491 while iter.next().is_some() {}
486 while iter.next().is_some() {}
492 assert_eq!(iter.peek(), None);
487 assert_eq!(iter.peek(), None);
493 }
488 }
494
489
495 #[test]
490 #[test]
496 fn test_empty() {
491 fn test_empty() {
497 let mut iter =
492 let mut iter =
498 AncestorsIterator::new(Stub, vec![10], 0, true).unwrap();
493 AncestorsIterator::new(Stub, vec![10], 0, true).unwrap();
499 assert!(!iter.is_empty());
494 assert!(!iter.is_empty());
500 while iter.next().is_some() {}
495 while iter.next().is_some() {}
501 assert!(!iter.is_empty());
496 assert!(!iter.is_empty());
502
497
503 let iter = AncestorsIterator::new(Stub, vec![], 0, true).unwrap();
498 let iter = AncestorsIterator::new(Stub, vec![], 0, true).unwrap();
504 assert!(iter.is_empty());
499 assert!(iter.is_empty());
505
500
506 // case where iter.seen == {NULL_REVISION}
501 // case where iter.seen == {NULL_REVISION}
507 let iter = AncestorsIterator::new(Stub, vec![0], 0, false).unwrap();
502 let iter = AncestorsIterator::new(Stub, vec![0], 0, false).unwrap();
508 assert!(iter.is_empty());
503 assert!(iter.is_empty());
509 }
504 }
510
505
511 /// A corrupted Graph, supporting error handling tests
506 /// A corrupted Graph, supporting error handling tests
512 #[derive(Clone, Debug)]
507 #[derive(Clone, Debug)]
513 struct Corrupted;
508 struct Corrupted;
514
509
515 impl Graph for Corrupted {
510 impl Graph for Corrupted {
516 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
511 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
517 match rev {
512 match rev {
518 1 => Ok([0, -1]),
513 1 => Ok([0, -1]),
519 r => Err(GraphError::ParentOutOfRange(r)),
514 r => Err(GraphError::ParentOutOfRange(r)),
520 }
515 }
521 }
516 }
522 }
517 }
523
518
524 #[test]
519 #[test]
525 fn test_initrev_out_of_range() {
520 fn test_initrev_out_of_range() {
526 // inclusive=false looks up initrev's parents right away
521 // inclusive=false looks up initrev's parents right away
527 match AncestorsIterator::new(Stub, vec![25], 0, false) {
522 match AncestorsIterator::new(Stub, vec![25], 0, false) {
528 Ok(_) => panic!("Should have been ParentOutOfRange"),
523 Ok(_) => panic!("Should have been ParentOutOfRange"),
529 Err(e) => assert_eq!(e, GraphError::ParentOutOfRange(25)),
524 Err(e) => assert_eq!(e, GraphError::ParentOutOfRange(25)),
530 }
525 }
531 }
526 }
532
527
533 #[test]
528 #[test]
534 fn test_next_out_of_range() {
529 fn test_next_out_of_range() {
535 // inclusive=false looks up initrev's parents right away
530 // inclusive=false looks up initrev's parents right away
536 let mut iter =
531 let mut iter =
537 AncestorsIterator::new(Corrupted, vec![1], 0, false).unwrap();
532 AncestorsIterator::new(Corrupted, vec![1], 0, false).unwrap();
538 assert_eq!(iter.next(), Some(Err(GraphError::ParentOutOfRange(0))));
533 assert_eq!(iter.next(), Some(Err(GraphError::ParentOutOfRange(0))));
539 }
534 }
540
535
541 #[test]
536 #[test]
542 fn test_lazy_iter_contains() {
537 fn test_lazy_iter_contains() {
543 let mut lazy =
538 let mut lazy =
544 LazyAncestors::new(Stub, vec![11, 13], 0, false).unwrap();
539 LazyAncestors::new(Stub, vec![11, 13], 0, false).unwrap();
545
540
546 let revs: Vec<Revision> = lazy.iter().map(|r| r.unwrap()).collect();
541 let revs: Vec<Revision> = lazy.iter().map(|r| r.unwrap()).collect();
547 // compare with iterator tests on the same initial revisions
542 // compare with iterator tests on the same initial revisions
548 assert_eq!(revs, vec![8, 7, 4, 3, 2, 1, 0]);
543 assert_eq!(revs, vec![8, 7, 4, 3, 2, 1, 0]);
549
544
550 // contains() results are correct, unaffected by the fact that
545 // contains() results are correct, unaffected by the fact that
551 // we consumed entirely an iterator out of lazy
546 // we consumed entirely an iterator out of lazy
552 assert_eq!(lazy.contains(2), Ok(true));
547 assert_eq!(lazy.contains(2), Ok(true));
553 assert_eq!(lazy.contains(9), Ok(false));
548 assert_eq!(lazy.contains(9), Ok(false));
554 }
549 }
555
550
556 #[test]
551 #[test]
557 fn test_lazy_contains_iter() {
552 fn test_lazy_contains_iter() {
558 let mut lazy =
553 let mut lazy =
559 LazyAncestors::new(Stub, vec![11, 13], 0, false).unwrap(); // reminder: [8, 7, 4, 3, 2, 1, 0]
554 LazyAncestors::new(Stub, vec![11, 13], 0, false).unwrap(); // reminder: [8, 7, 4, 3, 2, 1, 0]
560
555
561 assert_eq!(lazy.contains(2), Ok(true));
556 assert_eq!(lazy.contains(2), Ok(true));
562 assert_eq!(lazy.contains(6), Ok(false));
557 assert_eq!(lazy.contains(6), Ok(false));
563
558
564 // after consumption of 2 by the inner iterator, results stay
559 // after consumption of 2 by the inner iterator, results stay
565 // consistent
560 // consistent
566 assert_eq!(lazy.contains(2), Ok(true));
561 assert_eq!(lazy.contains(2), Ok(true));
567 assert_eq!(lazy.contains(5), Ok(false));
562 assert_eq!(lazy.contains(5), Ok(false));
568
563
569 // iter() still gives us a fresh iterator
564 // iter() still gives us a fresh iterator
570 let revs: Vec<Revision> = lazy.iter().map(|r| r.unwrap()).collect();
565 let revs: Vec<Revision> = lazy.iter().map(|r| r.unwrap()).collect();
571 assert_eq!(revs, vec![8, 7, 4, 3, 2, 1, 0]);
566 assert_eq!(revs, vec![8, 7, 4, 3, 2, 1, 0]);
572 }
567 }
573
568
574 #[test]
569 #[test]
575 /// Test constructor, add/get bases
570 /// Test constructor, add/get bases
576 fn test_missing_bases() {
571 fn test_missing_bases() {
577 let mut missing_ancestors =
572 let mut missing_ancestors =
578 MissingAncestors::new(Stub, [5, 3, 1, 3].iter().cloned());
573 MissingAncestors::new(Stub, [5, 3, 1, 3].iter().cloned());
579 let mut as_vec: Vec<Revision> =
574 let mut as_vec: Vec<Revision> =
580 missing_ancestors.get_bases().iter().cloned().collect();
575 missing_ancestors.get_bases().iter().cloned().collect();
581 as_vec.sort();
576 as_vec.sort();
582 assert_eq!(as_vec, [1, 3, 5]);
577 assert_eq!(as_vec, [1, 3, 5]);
583
578
584 missing_ancestors.add_bases([3, 7, 8].iter().cloned());
579 missing_ancestors.add_bases([3, 7, 8].iter().cloned());
585 as_vec = missing_ancestors.get_bases().iter().cloned().collect();
580 as_vec = missing_ancestors.get_bases().iter().cloned().collect();
586 as_vec.sort();
581 as_vec.sort();
587 assert_eq!(as_vec, [1, 3, 5, 7, 8]);
582 assert_eq!(as_vec, [1, 3, 5, 7, 8]);
588 }
583 }
589
584
590 fn assert_missing_remove(
585 fn assert_missing_remove(
591 bases: &[Revision],
586 bases: &[Revision],
592 revs: &[Revision],
587 revs: &[Revision],
593 expected: &[Revision],
588 expected: &[Revision],
594 ) {
589 ) {
595 let mut missing_ancestors =
590 let mut missing_ancestors =
596 MissingAncestors::new(Stub, bases.iter().cloned());
591 MissingAncestors::new(Stub, bases.iter().cloned());
597 let mut revset: HashSet<Revision> = revs.iter().cloned().collect();
592 let mut revset: HashSet<Revision> = revs.iter().cloned().collect();
598 missing_ancestors
593 missing_ancestors
599 .remove_ancestors_from(&mut revset)
594 .remove_ancestors_from(&mut revset)
600 .unwrap();
595 .unwrap();
601 let mut as_vec: Vec<Revision> = revset.into_iter().collect();
596 let mut as_vec: Vec<Revision> = revset.into_iter().collect();
602 as_vec.sort();
597 as_vec.sort();
603 assert_eq!(as_vec.as_slice(), expected);
598 assert_eq!(as_vec.as_slice(), expected);
604 }
599 }
605
600
606 #[test]
601 #[test]
607 fn test_missing_remove() {
602 fn test_missing_remove() {
608 assert_missing_remove(
603 assert_missing_remove(
609 &[1, 2, 3, 4, 7],
604 &[1, 2, 3, 4, 7],
610 Vec::from_iter(1..10).as_slice(),
605 Vec::from_iter(1..10).as_slice(),
611 &[5, 6, 8, 9],
606 &[5, 6, 8, 9],
612 );
607 );
613 assert_missing_remove(&[10], &[11, 12, 13, 14], &[11, 12, 13, 14]);
608 assert_missing_remove(&[10], &[11, 12, 13, 14], &[11, 12, 13, 14]);
614 assert_missing_remove(&[7], &[1, 2, 3, 4, 5], &[3, 5]);
609 assert_missing_remove(&[7], &[1, 2, 3, 4, 5], &[3, 5]);
615 }
610 }
616
611
617 fn assert_missing_ancestors(
612 fn assert_missing_ancestors(
618 bases: &[Revision],
613 bases: &[Revision],
619 revs: &[Revision],
614 revs: &[Revision],
620 expected: &[Revision],
615 expected: &[Revision],
621 ) {
616 ) {
622 let mut missing_ancestors =
617 let mut missing_ancestors =
623 MissingAncestors::new(Stub, bases.iter().cloned());
618 MissingAncestors::new(Stub, bases.iter().cloned());
624 let missing = missing_ancestors
619 let missing = missing_ancestors
625 .missing_ancestors(revs.iter().cloned())
620 .missing_ancestors(revs.iter().cloned())
626 .unwrap();
621 .unwrap();
627 assert_eq!(missing.as_slice(), expected);
622 assert_eq!(missing.as_slice(), expected);
628 }
623 }
629
624
630 #[test]
625 #[test]
631 fn test_missing_ancestors() {
626 fn test_missing_ancestors() {
632 // examples taken from test-ancestors.py by having it run
627 // examples taken from test-ancestors.py by having it run
633 // on the same graph (both naive and fast Python algs)
628 // on the same graph (both naive and fast Python algs)
634 assert_missing_ancestors(&[10], &[11], &[3, 7, 11]);
629 assert_missing_ancestors(&[10], &[11], &[3, 7, 11]);
635 assert_missing_ancestors(&[11], &[10], &[5, 10]);
630 assert_missing_ancestors(&[11], &[10], &[5, 10]);
636 assert_missing_ancestors(&[7], &[9, 11], &[3, 6, 9, 11]);
631 assert_missing_ancestors(&[7], &[9, 11], &[3, 6, 9, 11]);
637 }
632 }
638
633
639 // A Graph represented by a vector whose indices are revisions
634 // A Graph represented by a vector whose indices are revisions
640 // and values are parents of the revisions
635 // and values are parents of the revisions
641 type VecGraph = Vec<[Revision; 2]>;
636 type VecGraph = Vec<[Revision; 2]>;
642
637
643 impl Graph for VecGraph {
638 impl Graph for VecGraph {
644 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
639 fn parents(&self, rev: Revision) -> Result<[Revision; 2], GraphError> {
645 Ok(self[rev as usize])
640 Ok(self[rev as usize])
646 }
641 }
647 }
642 }
648
643
649 /// An interesting case found by a random generator similar to
644 /// An interesting case found by a random generator similar to
650 /// the one in test-ancestor.py. An early version of Rust MissingAncestors
645 /// the one in test-ancestor.py. An early version of Rust MissingAncestors
651 /// failed this, yet none of the integration tests of the whole suite
646 /// failed this, yet none of the integration tests of the whole suite
652 /// catched it.
647 /// catched it.
653 #[test]
648 #[test]
654 fn test_remove_ancestors_from_case1() {
649 fn test_remove_ancestors_from_case1() {
655 let graph: VecGraph = vec![
650 let graph: VecGraph = vec![
656 [NULL_REVISION, NULL_REVISION],
651 [NULL_REVISION, NULL_REVISION],
657 [0, NULL_REVISION],
652 [0, NULL_REVISION],
658 [1, 0],
653 [1, 0],
659 [2, 1],
654 [2, 1],
660 [3, NULL_REVISION],
655 [3, NULL_REVISION],
661 [4, NULL_REVISION],
656 [4, NULL_REVISION],
662 [5, 1],
657 [5, 1],
663 [2, NULL_REVISION],
658 [2, NULL_REVISION],
664 [7, NULL_REVISION],
659 [7, NULL_REVISION],
665 [8, NULL_REVISION],
660 [8, NULL_REVISION],
666 [9, NULL_REVISION],
661 [9, NULL_REVISION],
667 [10, 1],
662 [10, 1],
668 [3, NULL_REVISION],
663 [3, NULL_REVISION],
669 [12, NULL_REVISION],
664 [12, NULL_REVISION],
670 [13, NULL_REVISION],
665 [13, NULL_REVISION],
671 [14, NULL_REVISION],
666 [14, NULL_REVISION],
672 [4, NULL_REVISION],
667 [4, NULL_REVISION],
673 [16, NULL_REVISION],
668 [16, NULL_REVISION],
674 [17, NULL_REVISION],
669 [17, NULL_REVISION],
675 [18, NULL_REVISION],
670 [18, NULL_REVISION],
676 [19, 11],
671 [19, 11],
677 [20, NULL_REVISION],
672 [20, NULL_REVISION],
678 [21, NULL_REVISION],
673 [21, NULL_REVISION],
679 [22, NULL_REVISION],
674 [22, NULL_REVISION],
680 [23, NULL_REVISION],
675 [23, NULL_REVISION],
681 [2, NULL_REVISION],
676 [2, NULL_REVISION],
682 [3, NULL_REVISION],
677 [3, NULL_REVISION],
683 [26, 24],
678 [26, 24],
684 [27, NULL_REVISION],
679 [27, NULL_REVISION],
685 [28, NULL_REVISION],
680 [28, NULL_REVISION],
686 [12, NULL_REVISION],
681 [12, NULL_REVISION],
687 [1, NULL_REVISION],
682 [1, NULL_REVISION],
688 [1, 9],
683 [1, 9],
689 [32, NULL_REVISION],
684 [32, NULL_REVISION],
690 [33, NULL_REVISION],
685 [33, NULL_REVISION],
691 [34, 31],
686 [34, 31],
692 [35, NULL_REVISION],
687 [35, NULL_REVISION],
693 [36, 26],
688 [36, 26],
694 [37, NULL_REVISION],
689 [37, NULL_REVISION],
695 [38, NULL_REVISION],
690 [38, NULL_REVISION],
696 [39, NULL_REVISION],
691 [39, NULL_REVISION],
697 [40, NULL_REVISION],
692 [40, NULL_REVISION],
698 [41, NULL_REVISION],
693 [41, NULL_REVISION],
699 [42, 26],
694 [42, 26],
700 [0, NULL_REVISION],
695 [0, NULL_REVISION],
701 [44, NULL_REVISION],
696 [44, NULL_REVISION],
702 [45, 4],
697 [45, 4],
703 [40, NULL_REVISION],
698 [40, NULL_REVISION],
704 [47, NULL_REVISION],
699 [47, NULL_REVISION],
705 [36, 0],
700 [36, 0],
706 [49, NULL_REVISION],
701 [49, NULL_REVISION],
707 [NULL_REVISION, NULL_REVISION],
702 [NULL_REVISION, NULL_REVISION],
708 [51, NULL_REVISION],
703 [51, NULL_REVISION],
709 [52, NULL_REVISION],
704 [52, NULL_REVISION],
710 [53, NULL_REVISION],
705 [53, NULL_REVISION],
711 [14, NULL_REVISION],
706 [14, NULL_REVISION],
712 [55, NULL_REVISION],
707 [55, NULL_REVISION],
713 [15, NULL_REVISION],
708 [15, NULL_REVISION],
714 [23, NULL_REVISION],
709 [23, NULL_REVISION],
715 [58, NULL_REVISION],
710 [58, NULL_REVISION],
716 [59, NULL_REVISION],
711 [59, NULL_REVISION],
717 [2, NULL_REVISION],
712 [2, NULL_REVISION],
718 [61, 59],
713 [61, 59],
719 [62, NULL_REVISION],
714 [62, NULL_REVISION],
720 [63, NULL_REVISION],
715 [63, NULL_REVISION],
721 [NULL_REVISION, NULL_REVISION],
716 [NULL_REVISION, NULL_REVISION],
722 [65, NULL_REVISION],
717 [65, NULL_REVISION],
723 [66, NULL_REVISION],
718 [66, NULL_REVISION],
724 [67, NULL_REVISION],
719 [67, NULL_REVISION],
725 [68, NULL_REVISION],
720 [68, NULL_REVISION],
726 [37, 28],
721 [37, 28],
727 [69, 25],
722 [69, 25],
728 [71, NULL_REVISION],
723 [71, NULL_REVISION],
729 [72, NULL_REVISION],
724 [72, NULL_REVISION],
730 [50, 2],
725 [50, 2],
731 [74, NULL_REVISION],
726 [74, NULL_REVISION],
732 [12, NULL_REVISION],
727 [12, NULL_REVISION],
733 [18, NULL_REVISION],
728 [18, NULL_REVISION],
734 [77, NULL_REVISION],
729 [77, NULL_REVISION],
735 [78, NULL_REVISION],
730 [78, NULL_REVISION],
736 [79, NULL_REVISION],
731 [79, NULL_REVISION],
737 [43, 33],
732 [43, 33],
738 [81, NULL_REVISION],
733 [81, NULL_REVISION],
739 [82, NULL_REVISION],
734 [82, NULL_REVISION],
740 [83, NULL_REVISION],
735 [83, NULL_REVISION],
741 [84, 45],
736 [84, 45],
742 [85, NULL_REVISION],
737 [85, NULL_REVISION],
743 [86, NULL_REVISION],
738 [86, NULL_REVISION],
744 [NULL_REVISION, NULL_REVISION],
739 [NULL_REVISION, NULL_REVISION],
745 [88, NULL_REVISION],
740 [88, NULL_REVISION],
746 [NULL_REVISION, NULL_REVISION],
741 [NULL_REVISION, NULL_REVISION],
747 [76, 83],
742 [76, 83],
748 [44, NULL_REVISION],
743 [44, NULL_REVISION],
749 [92, NULL_REVISION],
744 [92, NULL_REVISION],
750 [93, NULL_REVISION],
745 [93, NULL_REVISION],
751 [9, NULL_REVISION],
746 [9, NULL_REVISION],
752 [95, 67],
747 [95, 67],
753 [96, NULL_REVISION],
748 [96, NULL_REVISION],
754 [97, NULL_REVISION],
749 [97, NULL_REVISION],
755 [NULL_REVISION, NULL_REVISION],
750 [NULL_REVISION, NULL_REVISION],
756 ];
751 ];
757 let problem_rev = 28 as Revision;
752 let problem_rev = 28 as Revision;
758 let problem_base = 70 as Revision;
753 let problem_base = 70 as Revision;
759 // making the problem obvious: problem_rev is a parent of problem_base
754 // making the problem obvious: problem_rev is a parent of problem_base
760 assert_eq!(graph.parents(problem_base).unwrap()[1], problem_rev);
755 assert_eq!(graph.parents(problem_base).unwrap()[1], problem_rev);
761
756
762 let mut missing_ancestors: MissingAncestors<VecGraph> =
757 let mut missing_ancestors: MissingAncestors<VecGraph> =
763 MissingAncestors::new(
758 MissingAncestors::new(
764 graph,
759 graph,
765 [60, 26, 70, 3, 96, 19, 98, 49, 97, 47, 1, 6]
760 [60, 26, 70, 3, 96, 19, 98, 49, 97, 47, 1, 6]
766 .iter()
761 .iter()
767 .cloned(),
762 .cloned(),
768 );
763 );
769 assert!(missing_ancestors.bases.contains(&problem_base));
764 assert!(missing_ancestors.bases.contains(&problem_base));
770
765
771 let mut revs: HashSet<Revision> =
766 let mut revs: HashSet<Revision> =
772 [4, 12, 41, 28, 68, 38, 1, 30, 56, 44]
767 [4, 12, 41, 28, 68, 38, 1, 30, 56, 44]
773 .iter()
768 .iter()
774 .cloned()
769 .cloned()
775 .collect();
770 .collect();
776 missing_ancestors.remove_ancestors_from(&mut revs).unwrap();
771 missing_ancestors.remove_ancestors_from(&mut revs).unwrap();
777 assert!(!revs.contains(&problem_rev));
772 assert!(!revs.contains(&problem_rev));
778 }
773 }
779
774
780 }
775 }
General Comments 0
You need to be logged in to leave comments. Login now