src/test/ui/const-generics/type-dependent/issue-61936.rs

   1 // run-pass
   2 #![feature(const_generics)]
   3 #![allow(incomplete_features)]
   4
   5 trait SliceExt<T: Clone> {
   6     fn array_windows<'a, const N: usize>(&'a self) -> ArrayWindows<'a, T, N>;
   7 }
   8
   9 impl <T: Clone> SliceExt<T> for [T] {
  10    fn array_windows<'a, const N: usize>(&'a self) -> ArrayWindows<'a, T, N> {
  11        ArrayWindows{ idx: 0, slice: &self }
  12    }
  13 }
  14
  15 struct ArrayWindows<'a, T, const N: usize> {
  16     slice: &'a [T],
  17     idx: usize,
  18 }
  19
  20 impl <'a, T: Clone, const N: usize> Iterator for ArrayWindows<'a, T, N> {
  21     type Item = [T; N];
  22     fn next(&mut self) -> Option<Self::Item> {
  23         // Note: this is unsound for some `T` and not meant as an example
  24         // on how to implement `ArrayWindows`.
  25         let mut res = unsafe{ std::mem::zeroed() };
  26         let mut ptr = &mut res as *mut [T; N] as *mut T;
  27
  28         for i in 0..N {
  29             match self.slice[self.idx..].get(i) {
  30                 None => return None,
  31                 Some(elem) => unsafe { std::ptr::write_volatile(ptr, elem.clone())},
  32             };
  33             ptr = ptr.wrapping_add(1);
  34             self.idx += 1;
  35         }
  36
  37         Some(res)
  38     }
  39 }
  40
  41 const FOUR: usize = 4;
  42
  43 fn main() {
  44     let v: Vec<usize> = vec![0; 100];
  45
  46     for array in v.as_slice().array_windows::<FOUR>() {
  47         assert_eq!(array, [0, 0, 0, 0])
  48     }
  49 }