Skip to content

Commit

Permalink
minor improvement
Browse files Browse the repository at this point in the history
  • Loading branch information
antonmarsden committed Aug 7, 2023
1 parent 68f3a0e commit d85b634
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions src/view.rs
Original file line number Diff line number Diff line change
Expand Up @@ -46,10 +46,11 @@ trait TooDeeViewCommon<T>: TooDeeOps<T> {
let stride = self.stride();
let start = col;
let end = {
if self.num_rows() == 0 {
let num_rows = self.num_rows();
if num_rows == 0 {
start
} else {
start + (self.num_rows() - 1) * stride + 1
start + (num_rows - 1) * stride + 1
}
};
(start..end, stride - 1)
Expand All @@ -59,14 +60,18 @@ trait TooDeeViewCommon<T>: TooDeeOps<T> {
}

impl<T> TooDeeViewCommon<T> for TooDeeView<'_, T> {
#[inline]
fn data(&self) -> &[T] { &self.data }
#[inline]
fn stride(&self) -> usize {
self.stride
}
}

impl<T> TooDeeViewCommon<T> for TooDeeViewMut<'_, T> {
#[inline]
fn data(&self) -> &[T] { &self.data }
#[inline]
fn stride(&self) -> usize {
self.stride
}
Expand Down

0 comments on commit d85b634

Please sign in to comment.