Move entity into a separate crate (#297)
This commit is contained in:
committed by
Dan Gohman
parent
18b2f12150
commit
9e17e62d68
55
lib/entity/src/keys.rs
Normal file
55
lib/entity/src/keys.rs
Normal file
@@ -0,0 +1,55 @@
|
||||
//! A double-ended iterator over entity references.
|
||||
|
||||
use EntityRef;
|
||||
use std::marker::PhantomData;
|
||||
|
||||
/// Iterate over all keys in order.
|
||||
pub struct Keys<K: EntityRef> {
|
||||
pos: usize,
|
||||
rev_pos: usize,
|
||||
unused: PhantomData<K>,
|
||||
}
|
||||
|
||||
impl<K: EntityRef> Keys<K> {
|
||||
/// Create a `Keys` iterator that visits `count` entities starting from 0.
|
||||
pub fn new(count: usize) -> Self {
|
||||
Self {
|
||||
pos: 0,
|
||||
rev_pos: count,
|
||||
unused: PhantomData,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<K: EntityRef> Iterator for Keys<K> {
|
||||
type Item = K;
|
||||
|
||||
fn next(&mut self) -> Option<Self::Item> {
|
||||
if self.pos < self.rev_pos {
|
||||
let k = K::new(self.pos);
|
||||
self.pos += 1;
|
||||
Some(k)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
let size = self.rev_pos - self.pos;
|
||||
(size, Some(size))
|
||||
}
|
||||
}
|
||||
|
||||
impl<K: EntityRef> DoubleEndedIterator for Keys<K> {
|
||||
fn next_back(&mut self) -> Option<Self::Item> {
|
||||
if self.rev_pos > self.pos {
|
||||
let k = K::new(self.rev_pos - 1);
|
||||
self.rev_pos -= 1;
|
||||
Some(k)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<K: EntityRef> ExactSizeIterator for Keys<K> {}
|
||||
Reference in New Issue
Block a user