Function get_slice_shape_1d

ciphercore-base/src/slices.rs:127–167

(dimension: u64, slice_element: SliceElement)

}

fn get_slice_shape_1d(dimension: u64, slice_element: SliceElement) -> Result<Option<u64>> {
    match slice_element {
        SliceElement::SingleIndex(mut ind) => {
            if ind < 0 {
                ind += dimension as i64;
            }
            if ind < 0 || ind >= dimension as i64 {
                Err(runtime_error!("Slice is out of bounds (SingleIndex)"))
            } else {
                Ok(None)
            }
        }
        SliceElement::SubArray(_, _, _) => {
            let (begin, end, step) = normalize_subarray(dimension, slice_element)?;
            let mut current = begin;
            let mut counter = 0;
            loop {
                if (step > 0 && current >= end) || (step < 0 && current <= end) {
                    break;
                }
                if current < 0 || current >= dimension as i64 {
                    return Err(runtime_error!(
                        "Slicing index is out of bounds: {} not in [{}, {})",
                        current,
                        0,
                        dimension
                    ));
                }
                counter += 1;
                current += step;
            }
            if counter == 0 {
                return Err(runtime_error!("Empty slice"));
            }
            Ok(Some(counter))
        }
        SliceElement::Ellipsis => {
            panic!("Should not be here!");
        }
    }
}

#[cfg(test)]
mod tests {