M-Labs/nalgebra
3
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove use of unsafe for CsLaneIterMut

Browse Source
This commit is contained in:
Andreas Longva 2020-12-22 11:01:50 +01:00
parent b59c4a3216
commit e261e7c388
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
nalgebra-sparse/src/cs.rs
View File

@ -3,7 +3,7 @@ use crate::{SparseEntry, SparseEntryMut};
use std::sync::Arc;
use std::ops::Range;
use std::ptr::slice_from_raw_parts_mut;
use std::mem::replace;
/// An abstract compressed matrix.
///
@ -238,7 +238,7 @@ pub struct CsLaneIterMut<'a, T> {
// The index of the lane that will be returned on the next iteration
current_lane_idx: usize,
pattern: &'a SparsityPattern,
remaining_values: *mut T,
remaining_values: &'a mut [T],
}
impl<'a, T> CsLaneIterMut<'a, T> {
@ -246,7 +246,7 @@ impl<'a, T> CsLaneIterMut<'a, T> {
Self {
current_lane_idx: 0,
pattern,
remaining_values: values.as_mut_ptr()
remaining_values: values
}
}
}
@ -264,12 +264,9 @@ impl<'a, T> Iterator for CsLaneIterMut<'a, T>
if let Some(minor_indices) = lane {
let count = minor_indices.len();
// Note: I can't think of any way to construct this iterator without unsafe.
let values_in_lane;
unsafe {
values_in_lane = &mut *slice_from_raw_parts_mut(self.remaining_values, count);
self.remaining_values = self.remaining_values.add(count);
}
let remaining = replace(&mut self.remaining_values, &mut []);
let (values_in_lane, remaining) = remaining.split_at_mut(count);
self.remaining_values = remaining;
self.current_lane_idx += 1;
Some(CsLaneMut {