fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-06-09 23:08:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

compiler/rust/bitset: Generalize BitSetIterator

Browse source 
It now iterates over a slice instead of a BitSet

Reviewed-by: Mel Henning <mhenning@darkrefraction.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/41915>
This commit is contained in:
Faith Ekstrand 2026-05-21 13:51:27 -04:00 • committed by Marge Bot
parent f43e57b3c0
commit 63d2ccd64b
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/compiler/rust/bitset.rs
View file

@ -245,7 +245,7 @@ impl<K: IntoBitIndex> BitSet<K> {
impl<K: FromBitIndex> BitSet<K> {
pub fn iter(&self) -> impl '_ + Iterator<Item = K> {
BitSetIter::new(self)
BitSetIter::new(&self.words)
}
}
@ -550,15 +550,17 @@ binop!(
);
struct BitSetIter<'a, K> {
set: &'a BitSet<K>,
words: &'a [u32],
idx: BitIndex,
phantom: PhantomData<K>,
}
impl<'a, K> BitSetIter<'a, K> {
fn new(set: &'a BitSet<K>) -> Self {
fn new(words: &'a [u32]) -> Self {
Self {
set,
words,
idx: BitIndex::ZERO,
phantom: PhantomData,
}
}
}
@ -567,11 +569,11 @@ impl<'a, K: FromBitIndex> Iterator for BitSetIter<'a, K> {
type Item = K;
fn next(&mut self) -> Option<K> {
if let Some(idx) = find_next_set(&self.set.words, self.idx) {
if let Some(idx) = find_next_set(self.words, self.idx) {
self.idx = idx + 1;
Some(K::from_bit_index(idx.into()))
} else {
self.idx = BitIndex::from_word(self.set.words.len());
self.idx = BitIndex::from_word(self.words.len());
None
}
}