xref: /libCEED/rust/libceed/src/vector.rs (revision 5fb68f377259d3910de46d787b7c5d1587fd01e1)
13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
39df49d7eSJed Brown //
43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
59df49d7eSJed Brown //
63d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
79df49d7eSJed Brown 
89df49d7eSJed Brown //! A Ceed Vector constitutes the main data structure and serves as input/output
99df49d7eSJed Brown //! for Ceed Operators.
109df49d7eSJed Brown 
119df49d7eSJed Brown use std::{
129df49d7eSJed Brown     ops::{Deref, DerefMut},
139df49d7eSJed Brown     os::raw::c_char,
149df49d7eSJed Brown };
159df49d7eSJed Brown 
169df49d7eSJed Brown use crate::prelude::*;
179df49d7eSJed Brown 
189df49d7eSJed Brown // -----------------------------------------------------------------------------
197ed177dbSJed Brown // Vector option
209df49d7eSJed Brown // -----------------------------------------------------------------------------
21c68be7a2SJeremy L Thompson #[derive(Debug)]
229df49d7eSJed Brown pub enum VectorOpt<'a> {
239df49d7eSJed Brown     Some(&'a Vector<'a>),
249df49d7eSJed Brown     Active,
259df49d7eSJed Brown     None,
269df49d7eSJed Brown }
279df49d7eSJed Brown /// Construct a VectorOpt reference from a Vector reference
289df49d7eSJed Brown impl<'a> From<&'a Vector<'_>> for VectorOpt<'a> {
299df49d7eSJed Brown     fn from(vec: &'a Vector) -> Self {
309df49d7eSJed Brown         debug_assert!(vec.ptr != unsafe { bind_ceed::CEED_VECTOR_ACTIVE });
31c68be7a2SJeremy L Thompson         debug_assert!(vec.ptr != unsafe { bind_ceed::CEED_VECTOR_NONE });
329df49d7eSJed Brown         Self::Some(vec)
339df49d7eSJed Brown     }
349df49d7eSJed Brown }
359df49d7eSJed Brown impl<'a> VectorOpt<'a> {
369df49d7eSJed Brown     /// Transform a Rust libCEED VectorOpt into C libCEED CeedVector
379df49d7eSJed Brown     pub(crate) fn to_raw(self) -> bind_ceed::CeedVector {
389df49d7eSJed Brown         match self {
399df49d7eSJed Brown             Self::Some(vec) => vec.ptr,
409df49d7eSJed Brown             Self::Active => unsafe { bind_ceed::CEED_VECTOR_ACTIVE },
41c68be7a2SJeremy L Thompson             Self::None => unsafe { bind_ceed::CEED_VECTOR_NONE },
429df49d7eSJed Brown         }
439df49d7eSJed Brown     }
44e03682afSJeremy L Thompson 
45e03682afSJeremy L Thompson     /// Check if a VectorOpt is Some
46e03682afSJeremy L Thompson     ///
47e03682afSJeremy L Thompson     /// ```
48e03682afSJeremy L Thompson     /// # use libceed::prelude::*;
494d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
50e03682afSJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
51e03682afSJeremy L Thompson     /// let vec = libceed::vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
52e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::from(&vec);
53e03682afSJeremy L Thompson     /// assert!(vec_opt.is_some(), "Incorrect VectorOpt");
54e03682afSJeremy L Thompson     ///
55e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::Active;
56e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_some(), "Incorrect VectorOpt");
57e03682afSJeremy L Thompson     ///
58e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::None;
59e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_some(), "Incorrect VectorOpt");
60e03682afSJeremy L Thompson     /// # Ok(())
61e03682afSJeremy L Thompson     /// # }
62e03682afSJeremy L Thompson     /// ```
63e03682afSJeremy L Thompson     pub fn is_some(&self) -> bool {
64e03682afSJeremy L Thompson         match self {
65e03682afSJeremy L Thompson             Self::Some(_) => true,
66e03682afSJeremy L Thompson             Self::Active => false,
67e03682afSJeremy L Thompson             Self::None => false,
68e03682afSJeremy L Thompson         }
69e03682afSJeremy L Thompson     }
70e03682afSJeremy L Thompson 
71e03682afSJeremy L Thompson     /// Check if a VectorOpt is Active
72e03682afSJeremy L Thompson     ///
73e03682afSJeremy L Thompson     /// ```
74e03682afSJeremy L Thompson     /// # use libceed::prelude::*;
754d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
76e03682afSJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
77e03682afSJeremy L Thompson     /// let vec = libceed::vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
78e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::from(&vec);
79e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_active(), "Incorrect VectorOpt");
80e03682afSJeremy L Thompson     ///
81e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::Active;
82e03682afSJeremy L Thompson     /// assert!(vec_opt.is_active(), "Incorrect VectorOpt");
83e03682afSJeremy L Thompson     ///
84e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::None;
85e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_active(), "Incorrect VectorOpt");
86e03682afSJeremy L Thompson     /// # Ok(())
87e03682afSJeremy L Thompson     /// # }
88e03682afSJeremy L Thompson     /// ```
89e03682afSJeremy L Thompson     pub fn is_active(&self) -> bool {
90e03682afSJeremy L Thompson         match self {
91e03682afSJeremy L Thompson             Self::Some(_) => false,
92e03682afSJeremy L Thompson             Self::Active => true,
93e03682afSJeremy L Thompson             Self::None => false,
94e03682afSJeremy L Thompson         }
95e03682afSJeremy L Thompson     }
96e03682afSJeremy L Thompson 
97e03682afSJeremy L Thompson     /// Check if a VectorOpt is Some
98e03682afSJeremy L Thompson     ///
99e03682afSJeremy L Thompson     /// ```
100e03682afSJeremy L Thompson     /// # use libceed::prelude::*;
1014d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
102e03682afSJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
103e03682afSJeremy L Thompson     /// let vec = libceed::vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
104e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::from(&vec);
105e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_none(), "Incorrect VectorOpt");
106e03682afSJeremy L Thompson     ///
107e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::Active;
108e03682afSJeremy L Thompson     /// assert!(!vec_opt.is_none(), "Incorrect VectorOpt");
109e03682afSJeremy L Thompson     ///
110e03682afSJeremy L Thompson     /// let vec_opt = VectorOpt::None;
111e03682afSJeremy L Thompson     /// assert!(vec_opt.is_none(), "Incorrect VectorOpt");
112e03682afSJeremy L Thompson     /// # Ok(())
113e03682afSJeremy L Thompson     /// # }
114e03682afSJeremy L Thompson     /// ```
115e03682afSJeremy L Thompson     pub fn is_none(&self) -> bool {
116e03682afSJeremy L Thompson         match self {
117e03682afSJeremy L Thompson             Self::Some(_) => false,
118e03682afSJeremy L Thompson             Self::Active => false,
119e03682afSJeremy L Thompson             Self::None => true,
120e03682afSJeremy L Thompson         }
121e03682afSJeremy L Thompson     }
1229df49d7eSJed Brown }
1239df49d7eSJed Brown 
1249df49d7eSJed Brown // -----------------------------------------------------------------------------
1257ed177dbSJed Brown // Vector borrowed slice wrapper
12656e67e05SJeremy L Thompson // -----------------------------------------------------------------------------
127486868d3SJeremy L Thompson pub struct VectorSliceWrapper<'a> {
12856e67e05SJeremy L Thompson     pub(crate) vector: crate::Vector<'a>,
12956e67e05SJeremy L Thompson     pub(crate) _slice: &'a mut [crate::Scalar],
13056e67e05SJeremy L Thompson }
13156e67e05SJeremy L Thompson 
13256e67e05SJeremy L Thompson // -----------------------------------------------------------------------------
13356e67e05SJeremy L Thompson // Destructor
13456e67e05SJeremy L Thompson // -----------------------------------------------------------------------------
135486868d3SJeremy L Thompson impl<'a> Drop for VectorSliceWrapper<'a> {
13656e67e05SJeremy L Thompson     fn drop(&mut self) {
13756e67e05SJeremy L Thompson         unsafe {
13856e67e05SJeremy L Thompson             bind_ceed::CeedVectorTakeArray(
13956e67e05SJeremy L Thompson                 self.vector.ptr,
14056e67e05SJeremy L Thompson                 crate::MemType::Host as bind_ceed::CeedMemType,
14156e67e05SJeremy L Thompson                 std::ptr::null_mut(),
14256e67e05SJeremy L Thompson             )
14356e67e05SJeremy L Thompson         };
14456e67e05SJeremy L Thompson     }
14556e67e05SJeremy L Thompson }
14656e67e05SJeremy L Thompson 
14756e67e05SJeremy L Thompson // -----------------------------------------------------------------------------
14835afbe75SJeremy L Thompson // Convenience constructor
14935afbe75SJeremy L Thompson // -----------------------------------------------------------------------------
15035afbe75SJeremy L Thompson impl<'a> VectorSliceWrapper<'a> {
15135afbe75SJeremy L Thompson     fn from_vector_and_slice_mut<'b>(
15235afbe75SJeremy L Thompson         vec: &'b mut crate::Vector,
15335afbe75SJeremy L Thompson         slice: &'a mut [crate::Scalar],
15435afbe75SJeremy L Thompson     ) -> crate::Result<Self> {
15535afbe75SJeremy L Thompson         assert_eq!(vec.length(), slice.len());
15635afbe75SJeremy L Thompson         let (host, copy_mode) = (
15735afbe75SJeremy L Thompson             crate::MemType::Host as bind_ceed::CeedMemType,
15835afbe75SJeremy L Thompson             crate::CopyMode::UsePointer as bind_ceed::CeedCopyMode,
15935afbe75SJeremy L Thompson         );
16035afbe75SJeremy L Thompson         let ierr = unsafe {
16135afbe75SJeremy L Thompson             bind_ceed::CeedVectorSetArray(
16235afbe75SJeremy L Thompson                 vec.ptr,
16335afbe75SJeremy L Thompson                 host,
16435afbe75SJeremy L Thompson                 copy_mode,
16535afbe75SJeremy L Thompson                 slice.as_ptr() as *mut crate::Scalar,
16635afbe75SJeremy L Thompson             )
16735afbe75SJeremy L Thompson         };
16835afbe75SJeremy L Thompson         vec.check_error(ierr)?;
16935afbe75SJeremy L Thompson 
17035afbe75SJeremy L Thompson         Ok(Self {
171e46a712dSJeremy L Thompson             vector: crate::Vector::from_raw(vec.ptr_copy_mut()?)?,
17235afbe75SJeremy L Thompson             _slice: slice,
17335afbe75SJeremy L Thompson         })
17435afbe75SJeremy L Thompson     }
17535afbe75SJeremy L Thompson }
17635afbe75SJeremy L Thompson 
17735afbe75SJeremy L Thompson // -----------------------------------------------------------------------------
1787ed177dbSJed Brown // Vector context wrapper
1799df49d7eSJed Brown // -----------------------------------------------------------------------------
1809df49d7eSJed Brown #[derive(Debug)]
1819df49d7eSJed Brown pub struct Vector<'a> {
1829df49d7eSJed Brown     pub(crate) ptr: bind_ceed::CeedVector,
1831142270cSJeremy L Thompson     _lifeline: PhantomData<&'a ()>,
1849df49d7eSJed Brown }
1859df49d7eSJed Brown impl From<&'_ Vector<'_>> for bind_ceed::CeedVector {
1869df49d7eSJed Brown     fn from(vec: &Vector) -> Self {
1879df49d7eSJed Brown         vec.ptr
1889df49d7eSJed Brown     }
1899df49d7eSJed Brown }
1909df49d7eSJed Brown 
1919df49d7eSJed Brown // -----------------------------------------------------------------------------
1929df49d7eSJed Brown // Destructor
1939df49d7eSJed Brown // -----------------------------------------------------------------------------
1949df49d7eSJed Brown impl<'a> Drop for Vector<'a> {
1959df49d7eSJed Brown     fn drop(&mut self) {
1969df49d7eSJed Brown         let not_none_and_active = self.ptr != unsafe { bind_ceed::CEED_VECTOR_NONE }
1979df49d7eSJed Brown             && self.ptr != unsafe { bind_ceed::CEED_VECTOR_ACTIVE };
1989df49d7eSJed Brown 
1999df49d7eSJed Brown         if not_none_and_active {
2009df49d7eSJed Brown             unsafe { bind_ceed::CeedVectorDestroy(&mut self.ptr) };
2019df49d7eSJed Brown         }
2029df49d7eSJed Brown     }
2039df49d7eSJed Brown }
2049df49d7eSJed Brown 
2059df49d7eSJed Brown // -----------------------------------------------------------------------------
2069df49d7eSJed Brown // Display
2079df49d7eSJed Brown // -----------------------------------------------------------------------------
2089df49d7eSJed Brown impl<'a> fmt::Display for Vector<'a> {
2099df49d7eSJed Brown     /// View a Vector
2109df49d7eSJed Brown     ///
2119df49d7eSJed Brown     /// ```
2129df49d7eSJed Brown     /// # use libceed::prelude::*;
2134d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
2149df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
215c68be7a2SJeremy L Thompson     /// let vec = libceed::vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
2169df49d7eSJed Brown     /// assert_eq!(
2179df49d7eSJed Brown     ///     vec.to_string(),
2189df49d7eSJed Brown     ///     "CeedVector length 3
2199df49d7eSJed Brown     ///     1.00000000
2209df49d7eSJed Brown     ///     2.00000000
2219df49d7eSJed Brown     ///     3.00000000
2229df49d7eSJed Brown     /// "
223c68be7a2SJeremy L Thompson     /// );
224c68be7a2SJeremy L Thompson     /// # Ok(())
225c68be7a2SJeremy L Thompson     /// # }
2269df49d7eSJed Brown     /// ```
2279df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
2289df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
2299df49d7eSJed Brown         let mut sizeloc = crate::MAX_BUFFER_LENGTH;
2309df49d7eSJed Brown         let format = CString::new("%12.8f").expect("CString::new failed");
2319df49d7eSJed Brown         let format_c: *const c_char = format.into_raw();
2329df49d7eSJed Brown         let cstring = unsafe {
2339df49d7eSJed Brown             let file = bind_ceed::open_memstream(&mut ptr, &mut sizeloc);
2349df49d7eSJed Brown             bind_ceed::CeedVectorView(self.ptr, format_c, file);
2359df49d7eSJed Brown             bind_ceed::fclose(file);
2369df49d7eSJed Brown             CString::from_raw(ptr)
2379df49d7eSJed Brown         };
2389df49d7eSJed Brown         cstring.to_string_lossy().fmt(f)
2399df49d7eSJed Brown     }
2409df49d7eSJed Brown }
2419df49d7eSJed Brown 
2429df49d7eSJed Brown // -----------------------------------------------------------------------------
2439df49d7eSJed Brown // Implementations
2449df49d7eSJed Brown // -----------------------------------------------------------------------------
2459df49d7eSJed Brown impl<'a> Vector<'a> {
2469df49d7eSJed Brown     // Constructors
247594ef120SJeremy L Thompson     pub fn create(ceed: &crate::Ceed, n: usize) -> crate::Result<Self> {
2481f9221feSJeremy L Thompson         let n = isize::try_from(n).unwrap();
2499df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
2509df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorCreate(ceed.ptr, n, &mut ptr) };
2519df49d7eSJed Brown         ceed.check_error(ierr)?;
2521142270cSJeremy L Thompson         Ok(Self {
2531142270cSJeremy L Thompson             ptr,
2541142270cSJeremy L Thompson             _lifeline: PhantomData,
2551142270cSJeremy L Thompson         })
2569df49d7eSJed Brown     }
2579df49d7eSJed Brown 
2581142270cSJeremy L Thompson     pub(crate) fn from_raw(ptr: bind_ceed::CeedVector) -> crate::Result<Self> {
2591142270cSJeremy L Thompson         Ok(Self {
2601142270cSJeremy L Thompson             ptr,
2611142270cSJeremy L Thompson             _lifeline: PhantomData,
2621142270cSJeremy L Thompson         })
2639df49d7eSJed Brown     }
2649df49d7eSJed Brown 
265e46a712dSJeremy L Thompson     fn ptr_copy_mut(&mut self) -> crate::Result<bind_ceed::CeedVector> {
266e46a712dSJeremy L Thompson         let mut ptr_copy = std::ptr::null_mut();
267e46a712dSJeremy L Thompson         let ierr = unsafe { bind_ceed::CeedVectorReferenceCopy(self.ptr, &mut ptr_copy) };
268e46a712dSJeremy L Thompson         self.check_error(ierr)?;
269e46a712dSJeremy L Thompson         Ok(ptr_copy)
270e46a712dSJeremy L Thompson     }
271e46a712dSJeremy L Thompson 
272*5fb68f37SKaren (Ren) Stengel     /// Copy the array of self into vec_copy
273*5fb68f37SKaren (Ren) Stengel     ///
274*5fb68f37SKaren (Ren) Stengel     /// # arguments
275*5fb68f37SKaren (Ren) Stengel     ///
276*5fb68f37SKaren (Ren) Stengel     /// * `vec_source` - vector to copy array values from
277*5fb68f37SKaren (Ren) Stengel     #[allow(unused_mut)]
278*5fb68f37SKaren (Ren) Stengel     fn copy_from(mut self, vec_source: &crate::Vector) -> crate::Result<Self> {
279*5fb68f37SKaren (Ren) Stengel         let ierr = unsafe { bind_ceed::CeedVectorCopy(vec_source.ptr, self.ptr) };
280*5fb68f37SKaren (Ren) Stengel         self.check_error(ierr)?;
281*5fb68f37SKaren (Ren) Stengel         Ok(self)
282*5fb68f37SKaren (Ren) Stengel     }
283*5fb68f37SKaren (Ren) Stengel 
2849df49d7eSJed Brown     /// Create a Vector from a slice
2859df49d7eSJed Brown     ///
2869df49d7eSJed Brown     /// # arguments
2879df49d7eSJed Brown     ///
2889df49d7eSJed Brown     /// * `slice` - values to initialize vector with
2899df49d7eSJed Brown     ///
2909df49d7eSJed Brown     /// ```
2919df49d7eSJed Brown     /// # use libceed::prelude::*;
2924d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
2939df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
294c68be7a2SJeremy L Thompson     /// let vec = vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
2959df49d7eSJed Brown     /// assert_eq!(vec.length(), 3, "Incorrect length from slice");
296c68be7a2SJeremy L Thompson     /// # Ok(())
297c68be7a2SJeremy L Thompson     /// # }
2989df49d7eSJed Brown     /// ```
299594ef120SJeremy L Thompson     pub fn from_slice(ceed: &crate::Ceed, v: &[crate::Scalar]) -> crate::Result<Self> {
3009df49d7eSJed Brown         let mut x = Self::create(ceed, v.len())?;
3019df49d7eSJed Brown         x.set_slice(v)?;
3029df49d7eSJed Brown         Ok(x)
3039df49d7eSJed Brown     }
3049df49d7eSJed Brown 
3059df49d7eSJed Brown     /// Create a Vector from a mutable array reference
3069df49d7eSJed Brown     ///
3079df49d7eSJed Brown     /// # arguments
3089df49d7eSJed Brown     ///
3099df49d7eSJed Brown     /// * `slice` - values to initialize vector with
3109df49d7eSJed Brown     ///
3119df49d7eSJed Brown     /// ```
3129df49d7eSJed Brown     /// # use libceed::prelude::*;
3134d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3149df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
3159df49d7eSJed Brown     /// let mut rust_vec = vec![1., 2., 3.];
316c68be7a2SJeremy L Thompson     /// let vec = libceed::vector::Vector::from_array(&ceed, &mut rust_vec)?;
3179df49d7eSJed Brown     ///
3189df49d7eSJed Brown     /// assert_eq!(vec.length(), 3, "Incorrect length from slice");
319c68be7a2SJeremy L Thompson     /// # Ok(())
320c68be7a2SJeremy L Thompson     /// # }
3219df49d7eSJed Brown     /// ```
322594ef120SJeremy L Thompson     pub fn from_array(ceed: &crate::Ceed, v: &mut [crate::Scalar]) -> crate::Result<Self> {
3239df49d7eSJed Brown         let x = Self::create(ceed, v.len())?;
3249df49d7eSJed Brown         let (host, user_pointer) = (
3259df49d7eSJed Brown             crate::MemType::Host as bind_ceed::CeedMemType,
3269df49d7eSJed Brown             crate::CopyMode::UsePointer as bind_ceed::CeedCopyMode,
3279df49d7eSJed Brown         );
32880a9ef05SNatalie Beams         let v = v.as_ptr() as *mut crate::Scalar;
3299df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorSetArray(x.ptr, host, user_pointer, v) };
3309df49d7eSJed Brown         ceed.check_error(ierr)?;
3319df49d7eSJed Brown         Ok(x)
3329df49d7eSJed Brown     }
3339df49d7eSJed Brown 
3341142270cSJeremy L Thompson     // Error handling
3351142270cSJeremy L Thompson     #[doc(hidden)]
3361142270cSJeremy L Thompson     fn check_error(&self, ierr: i32) -> crate::Result<i32> {
3371142270cSJeremy L Thompson         let mut ptr = std::ptr::null_mut();
3381142270cSJeremy L Thompson         unsafe {
3391142270cSJeremy L Thompson             bind_ceed::CeedVectorGetCeed(self.ptr, &mut ptr);
3401142270cSJeremy L Thompson         }
3411142270cSJeremy L Thompson         crate::check_error(ptr, ierr)
3421142270cSJeremy L Thompson     }
3431142270cSJeremy L Thompson 
3447ed177dbSJed Brown     /// Returns the length of a Vector
3459df49d7eSJed Brown     ///
3469df49d7eSJed Brown     /// ```
3479df49d7eSJed Brown     /// # use libceed::prelude::*;
3484d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3499df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
350c68be7a2SJeremy L Thompson     /// let vec = ceed.vector(10)?;
3519df49d7eSJed Brown     ///
3529df49d7eSJed Brown     /// let n = vec.length();
3539df49d7eSJed Brown     /// assert_eq!(n, 10, "Incorrect length");
354c68be7a2SJeremy L Thompson     /// # Ok(())
355c68be7a2SJeremy L Thompson     /// # }
3569df49d7eSJed Brown     /// ```
3579df49d7eSJed Brown     pub fn length(&self) -> usize {
3589df49d7eSJed Brown         let mut n = 0;
3599df49d7eSJed Brown         unsafe { bind_ceed::CeedVectorGetLength(self.ptr, &mut n) };
3609df49d7eSJed Brown         usize::try_from(n).unwrap()
3619df49d7eSJed Brown     }
3629df49d7eSJed Brown 
3637ed177dbSJed Brown     /// Returns the length of a Vector
3649df49d7eSJed Brown     ///
3659df49d7eSJed Brown     /// ```
3669df49d7eSJed Brown     /// # use libceed::prelude::*;
3674d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3689df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
369c68be7a2SJeremy L Thompson     /// let vec = ceed.vector(10)?;
3709df49d7eSJed Brown     /// assert_eq!(vec.len(), 10, "Incorrect length");
371c68be7a2SJeremy L Thompson     /// # Ok(())
372c68be7a2SJeremy L Thompson     /// # }
3739df49d7eSJed Brown     /// ```
3749df49d7eSJed Brown     pub fn len(&self) -> usize {
3759df49d7eSJed Brown         self.length()
3769df49d7eSJed Brown     }
3779df49d7eSJed Brown 
3787ed177dbSJed Brown     /// Set the Vector to a constant value
3799df49d7eSJed Brown     ///
3809df49d7eSJed Brown     /// # arguments
3819df49d7eSJed Brown     ///
3829df49d7eSJed Brown     /// * `val` - Value to be used
3839df49d7eSJed Brown     ///
3849df49d7eSJed Brown     /// ```
3859df49d7eSJed Brown     /// # use libceed::prelude::*;
3864d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3879df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
3889df49d7eSJed Brown     /// let len = 10;
389c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(len)?;
3909df49d7eSJed Brown     ///
3919df49d7eSJed Brown     /// let val = 42.0;
392c68be7a2SJeremy L Thompson     /// vec.set_value(val)?;
3939df49d7eSJed Brown     ///
394d3677ae8SJeremy L Thompson     /// for v in vec.view()?.iter() {
3959df49d7eSJed Brown     ///     assert_eq!(*v, val, "Value not set correctly");
396d3677ae8SJeremy L Thompson     /// }
397c68be7a2SJeremy L Thompson     /// # Ok(())
398c68be7a2SJeremy L Thompson     /// # }
3999df49d7eSJed Brown     /// ```
40080a9ef05SNatalie Beams     pub fn set_value(&mut self, value: crate::Scalar) -> crate::Result<i32> {
4019df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorSetValue(self.ptr, value) };
4021142270cSJeremy L Thompson         self.check_error(ierr)
4039df49d7eSJed Brown     }
4049df49d7eSJed Brown 
4059df49d7eSJed Brown     /// Set values from a slice of the same length
4069df49d7eSJed Brown     ///
4079df49d7eSJed Brown     /// # arguments
4089df49d7eSJed Brown     ///
4099df49d7eSJed Brown     /// * `slice` - values to into self; length must match
4109df49d7eSJed Brown     ///
4119df49d7eSJed Brown     /// ```
4129df49d7eSJed Brown     /// # use libceed::prelude::*;
4134d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
4149df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
415c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(4)?;
416c68be7a2SJeremy L Thompson     /// vec.set_slice(&[10., 11., 12., 13.])?;
4179df49d7eSJed Brown     ///
418d3677ae8SJeremy L Thompson     /// for (i, v) in vec.view()?.iter().enumerate() {
41980a9ef05SNatalie Beams     ///     assert_eq!(*v, 10. + i as Scalar, "Slice not set correctly");
420d3677ae8SJeremy L Thompson     /// }
421c68be7a2SJeremy L Thompson     /// # Ok(())
422c68be7a2SJeremy L Thompson     /// # }
4239df49d7eSJed Brown     /// ```
42480a9ef05SNatalie Beams     pub fn set_slice(&mut self, slice: &[crate::Scalar]) -> crate::Result<i32> {
4259df49d7eSJed Brown         assert_eq!(self.length(), slice.len());
4269df49d7eSJed Brown         let (host, copy_mode) = (
4279df49d7eSJed Brown             crate::MemType::Host as bind_ceed::CeedMemType,
4289df49d7eSJed Brown             crate::CopyMode::CopyValues as bind_ceed::CeedCopyMode,
4299df49d7eSJed Brown         );
4309df49d7eSJed Brown         let ierr = unsafe {
43180a9ef05SNatalie Beams             bind_ceed::CeedVectorSetArray(
43280a9ef05SNatalie Beams                 self.ptr,
43380a9ef05SNatalie Beams                 host,
43480a9ef05SNatalie Beams                 copy_mode,
43580a9ef05SNatalie Beams                 slice.as_ptr() as *mut crate::Scalar,
43680a9ef05SNatalie Beams             )
4379df49d7eSJed Brown         };
4381142270cSJeremy L Thompson         self.check_error(ierr)
4399df49d7eSJed Brown     }
4409df49d7eSJed Brown 
441486868d3SJeremy L Thompson     /// Wrap a mutable slice in a Vector of the same length
44256e67e05SJeremy L Thompson     ///
44356e67e05SJeremy L Thompson     /// # arguments
44456e67e05SJeremy L Thompson     ///
445486868d3SJeremy L Thompson     /// * `slice` - values to wrap in self; length must match
44656e67e05SJeremy L Thompson     ///
44756e67e05SJeremy L Thompson     /// ```
44856e67e05SJeremy L Thompson     /// # use libceed::prelude::*;
44956e67e05SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
45056e67e05SJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
45156e67e05SJeremy L Thompson     /// let mut vec = ceed.vector(4)?;
452486868d3SJeremy L Thompson     /// let mut array = [10., 11., 12., 13.];
45356e67e05SJeremy L Thompson     ///
45456e67e05SJeremy L Thompson     /// {
455486868d3SJeremy L Thompson     ///     // `wrapper` holds a mutable reference to the wrapped slice
456486868d3SJeremy L Thompson     ///     //   that is dropped when `wrapper` goes out of scope
457486868d3SJeremy L Thompson     ///     let wrapper = vec.wrap_slice_mut(&mut array)?;
458d3677ae8SJeremy L Thompson     ///     for (i, v) in vec.view()?.iter().enumerate() {
45956e67e05SJeremy L Thompson     ///         assert_eq!(*v, 10. + i as Scalar, "Slice not set correctly");
460d3677ae8SJeremy L Thompson     ///     }
46156e67e05SJeremy L Thompson     ///
462486868d3SJeremy L Thompson     ///     // This line will not compile, as the `wrapper` holds mutable
463486868d3SJeremy L Thompson     ///     //   access to the `array`
464486868d3SJeremy L Thompson     ///     // array[0] = 5.0;
46556e67e05SJeremy L Thompson     ///
466486868d3SJeremy L Thompson     ///     // Changes here are reflected in the `array`
46756e67e05SJeremy L Thompson     ///     vec.set_value(5.0)?;
468d3677ae8SJeremy L Thompson     ///     for v in vec.view()?.iter() {
469486868d3SJeremy L Thompson     ///         assert_eq!(*v, 5.0 as Scalar, "Value not set correctly");
470d3677ae8SJeremy L Thompson     ///     }
47156e67e05SJeremy L Thompson     /// }
47256e67e05SJeremy L Thompson     ///
473486868d3SJeremy L Thompson     /// // 'array' remains changed
474d3677ae8SJeremy L Thompson     /// for v in array.iter() {
475486868d3SJeremy L Thompson     ///     assert_eq!(*v, 5.0 as Scalar, "Array not mutated correctly");
476d3677ae8SJeremy L Thompson     /// }
47756e67e05SJeremy L Thompson     ///
478486868d3SJeremy L Thompson     /// // While changes to `vec` no longer affect `array`
479486868d3SJeremy L Thompson     /// vec.set_value(6.0)?;
480d3677ae8SJeremy L Thompson     /// for v in array.iter() {
481486868d3SJeremy L Thompson     ///     assert_eq!(*v, 5.0 as Scalar, "Array mutated without permission");
482d3677ae8SJeremy L Thompson     /// }
48356e67e05SJeremy L Thompson     /// # Ok(())
48456e67e05SJeremy L Thompson     /// # }
48556e67e05SJeremy L Thompson     /// ```
486486868d3SJeremy L Thompson     pub fn wrap_slice_mut<'b>(
48756e67e05SJeremy L Thompson         &mut self,
48856e67e05SJeremy L Thompson         slice: &'b mut [crate::Scalar],
489486868d3SJeremy L Thompson     ) -> crate::Result<VectorSliceWrapper<'b>> {
49035afbe75SJeremy L Thompson         crate::VectorSliceWrapper::from_vector_and_slice_mut(self, slice)
49156e67e05SJeremy L Thompson     }
49256e67e05SJeremy L Thompson 
4937ed177dbSJed Brown     /// Sync the Vector to a specified memtype
4949df49d7eSJed Brown     ///
4959df49d7eSJed Brown     /// # arguments
4969df49d7eSJed Brown     ///
4979df49d7eSJed Brown     /// * `mtype` - Memtype to be synced
4989df49d7eSJed Brown     ///
4999df49d7eSJed Brown     /// ```
5009df49d7eSJed Brown     /// # use libceed::prelude::*;
5014d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5029df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
5039df49d7eSJed Brown     /// let len = 10;
504c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(len)?;
5059df49d7eSJed Brown     ///
5069df49d7eSJed Brown     /// let val = 42.0;
5079df49d7eSJed Brown     /// vec.set_value(val);
508c68be7a2SJeremy L Thompson     /// vec.sync(MemType::Host)?;
5099df49d7eSJed Brown     ///
510d3677ae8SJeremy L Thompson     /// for v in vec.view()?.iter() {
5119df49d7eSJed Brown     ///     assert_eq!(*v, val, "Value not set correctly");
512d3677ae8SJeremy L Thompson     /// }
513c68be7a2SJeremy L Thompson     /// # Ok(())
514c68be7a2SJeremy L Thompson     /// # }
5159df49d7eSJed Brown     /// ```
5169df49d7eSJed Brown     pub fn sync(&self, mtype: crate::MemType) -> crate::Result<i32> {
5179df49d7eSJed Brown         let ierr =
5189df49d7eSJed Brown             unsafe { bind_ceed::CeedVectorSyncArray(self.ptr, mtype as bind_ceed::CeedMemType) };
5191142270cSJeremy L Thompson         self.check_error(ierr)
5209df49d7eSJed Brown     }
5219df49d7eSJed Brown 
5229df49d7eSJed Brown     /// Create an immutable view
5239df49d7eSJed Brown     ///
5249df49d7eSJed Brown     /// ```
5259df49d7eSJed Brown     /// # use libceed::prelude::*;
5264d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5279df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
528c68be7a2SJeremy L Thompson     /// let vec = ceed.vector_from_slice(&[10., 11., 12., 13.])?;
5299df49d7eSJed Brown     ///
530e78171edSJeremy L Thompson     /// let v = vec.view()?;
5319df49d7eSJed Brown     /// assert_eq!(v[0..2], [10., 11.]);
5329df49d7eSJed Brown     ///
5339df49d7eSJed Brown     /// // It is valid to have multiple immutable views
534e78171edSJeremy L Thompson     /// let w = vec.view()?;
5359df49d7eSJed Brown     /// assert_eq!(v[1..], w[1..]);
536c68be7a2SJeremy L Thompson     /// # Ok(())
537c68be7a2SJeremy L Thompson     /// # }
5389df49d7eSJed Brown     /// ```
539e78171edSJeremy L Thompson     pub fn view(&self) -> crate::Result<VectorView> {
5409df49d7eSJed Brown         VectorView::new(self)
5419df49d7eSJed Brown     }
5429df49d7eSJed Brown 
5439df49d7eSJed Brown     /// Create an mutable view
5449df49d7eSJed Brown     ///
5459df49d7eSJed Brown     /// ```
5469df49d7eSJed Brown     /// # use libceed::prelude::*;
5474d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5489df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
549c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector_from_slice(&[10., 11., 12., 13.])?;
5509df49d7eSJed Brown     ///
5519df49d7eSJed Brown     /// {
552e78171edSJeremy L Thompson     ///     let mut v = vec.view_mut()?;
5539df49d7eSJed Brown     ///     v[2] = 9.;
5549df49d7eSJed Brown     /// }
5559df49d7eSJed Brown     ///
556e78171edSJeremy L Thompson     /// let w = vec.view()?;
5579df49d7eSJed Brown     /// assert_eq!(w[2], 9., "View did not mutate data");
558c68be7a2SJeremy L Thompson     /// # Ok(())
559c68be7a2SJeremy L Thompson     /// # }
5609df49d7eSJed Brown     /// ```
561e78171edSJeremy L Thompson     pub fn view_mut(&mut self) -> crate::Result<VectorViewMut> {
5629df49d7eSJed Brown         VectorViewMut::new(self)
5639df49d7eSJed Brown     }
5649df49d7eSJed Brown 
5657ed177dbSJed Brown     /// Return the norm of a Vector
5669df49d7eSJed Brown     ///
5679df49d7eSJed Brown     /// # arguments
5689df49d7eSJed Brown     ///
5699df49d7eSJed Brown     /// * `ntype` - Norm type One, Two, or Max
5709df49d7eSJed Brown     ///
5719df49d7eSJed Brown     /// ```
5729df49d7eSJed Brown     /// # use libceed::prelude::*;
5734d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5749df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
575c68be7a2SJeremy L Thompson     /// let vec = ceed.vector_from_slice(&[1., 2., 3., 4.])?;
5769df49d7eSJed Brown     ///
577c68be7a2SJeremy L Thompson     /// let max_norm = vec.norm(NormType::Max)?;
5789df49d7eSJed Brown     /// assert_eq!(max_norm, 4.0, "Incorrect Max norm");
5799df49d7eSJed Brown     ///
580c68be7a2SJeremy L Thompson     /// let l1_norm = vec.norm(NormType::One)?;
5819df49d7eSJed Brown     /// assert_eq!(l1_norm, 10., "Incorrect L1 norm");
5829df49d7eSJed Brown     ///
583c68be7a2SJeremy L Thompson     /// let l2_norm = vec.norm(NormType::Two)?;
5849df49d7eSJed Brown     /// assert!((l2_norm - 5.477) < 1e-3, "Incorrect L2 norm");
585c68be7a2SJeremy L Thompson     /// # Ok(())
586c68be7a2SJeremy L Thompson     /// # }
5879df49d7eSJed Brown     /// ```
58880a9ef05SNatalie Beams     pub fn norm(&self, ntype: crate::NormType) -> crate::Result<crate::Scalar> {
58980a9ef05SNatalie Beams         let mut res: crate::Scalar = 0.0;
5909df49d7eSJed Brown         let ierr = unsafe {
5919df49d7eSJed Brown             bind_ceed::CeedVectorNorm(self.ptr, ntype as bind_ceed::CeedNormType, &mut res)
5929df49d7eSJed Brown         };
5931142270cSJeremy L Thompson         self.check_error(ierr)?;
5949df49d7eSJed Brown         Ok(res)
5959df49d7eSJed Brown     }
5969df49d7eSJed Brown 
5977ed177dbSJed Brown     /// Compute x = alpha x for a Vector
5989df49d7eSJed Brown     ///
5999df49d7eSJed Brown     /// # arguments
6009df49d7eSJed Brown     ///
6019df49d7eSJed Brown     /// * `alpha` - scaling factor
6029df49d7eSJed Brown     ///
6039df49d7eSJed Brown     /// ```
6049df49d7eSJed Brown     /// # use libceed::prelude::*;
6054d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6069df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
607c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6089df49d7eSJed Brown     ///
609c68be7a2SJeremy L Thompson     /// vec = vec.scale(-1.0)?;
610d3677ae8SJeremy L Thompson     /// for (i, v) in vec.view()?.iter().enumerate() {
611d3677ae8SJeremy L Thompson     ///     assert_eq!(*v, -(i as Scalar), "Value not set correctly");
612d3677ae8SJeremy L Thompson     /// }
613c68be7a2SJeremy L Thompson     /// # Ok(())
614c68be7a2SJeremy L Thompson     /// # }
6159df49d7eSJed Brown     /// ```
6169df49d7eSJed Brown     #[allow(unused_mut)]
61780a9ef05SNatalie Beams     pub fn scale(mut self, alpha: crate::Scalar) -> crate::Result<Self> {
6189df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorScale(self.ptr, alpha) };
6191142270cSJeremy L Thompson         self.check_error(ierr)?;
6209df49d7eSJed Brown         Ok(self)
6219df49d7eSJed Brown     }
6229df49d7eSJed Brown 
6237ed177dbSJed Brown     /// Compute y = alpha x + y for a pair of Vectors
6249df49d7eSJed Brown     ///
6259df49d7eSJed Brown     /// # arguments
6269df49d7eSJed Brown     ///
6279df49d7eSJed Brown     /// * `alpha` - scaling factor
6289df49d7eSJed Brown     /// * `x`     - second vector, must be different than self
6299df49d7eSJed Brown     ///
6309df49d7eSJed Brown     /// ```
6319df49d7eSJed Brown     /// # use libceed::prelude::*;
6324d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6339df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
634c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
635c68be7a2SJeremy L Thompson     /// let mut y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6369df49d7eSJed Brown     ///
637c68be7a2SJeremy L Thompson     /// y = y.axpy(-0.5, &x)?;
638d3677ae8SJeremy L Thompson     /// for (i, y) in y.view()?.iter().enumerate() {
639d3677ae8SJeremy L Thompson     ///     assert_eq!(*y, (i as Scalar) / 2.0, "Value not set correctly");
640d3677ae8SJeremy L Thompson     /// }
641c68be7a2SJeremy L Thompson     /// # Ok(())
642c68be7a2SJeremy L Thompson     /// # }
6439df49d7eSJed Brown     /// ```
6449df49d7eSJed Brown     #[allow(unused_mut)]
64580a9ef05SNatalie Beams     pub fn axpy(mut self, alpha: crate::Scalar, x: &crate::Vector) -> crate::Result<Self> {
6469df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorAXPY(self.ptr, alpha, x.ptr) };
6471142270cSJeremy L Thompson         self.check_error(ierr)?;
6489df49d7eSJed Brown         Ok(self)
6499df49d7eSJed Brown     }
6509df49d7eSJed Brown 
651*5fb68f37SKaren (Ren) Stengel     /// Compute y = alpha x + beta y for a pair of Vectors
652*5fb68f37SKaren (Ren) Stengel     ///
653*5fb68f37SKaren (Ren) Stengel     /// # arguments
654*5fb68f37SKaren (Ren) Stengel     ///
655*5fb68f37SKaren (Ren) Stengel     /// * `alpha` - first scaling factor
656*5fb68f37SKaren (Ren) Stengel     /// * `beta`  - second scaling factor
657*5fb68f37SKaren (Ren) Stengel     /// * `x`     - second vector, must be different than self
658*5fb68f37SKaren (Ren) Stengel     ///
659*5fb68f37SKaren (Ren) Stengel     /// ```
660*5fb68f37SKaren (Ren) Stengel     /// # use libceed::prelude::*;
661*5fb68f37SKaren (Ren) Stengel     /// # fn main() -> libceed::Result<()> {
662*5fb68f37SKaren (Ren) Stengel     /// # let ceed = libceed::Ceed::default_init();
663*5fb68f37SKaren (Ren) Stengel     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
664*5fb68f37SKaren (Ren) Stengel     /// let mut y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
665*5fb68f37SKaren (Ren) Stengel     ///
666*5fb68f37SKaren (Ren) Stengel     /// y = y.axpby(-0.5, 1.0, &x)?;
667*5fb68f37SKaren (Ren) Stengel     /// for (i, y) in y.view()?.iter().enumerate() {
668*5fb68f37SKaren (Ren) Stengel     ///     assert_eq!(*y, (i as Scalar) * 1.5, "Value not set correctly");
669*5fb68f37SKaren (Ren) Stengel     /// }
670*5fb68f37SKaren (Ren) Stengel     /// # Ok(())
671*5fb68f37SKaren (Ren) Stengel     /// # }
672*5fb68f37SKaren (Ren) Stengel     /// ```
673*5fb68f37SKaren (Ren) Stengel     #[allow(unused_mut)]
674*5fb68f37SKaren (Ren) Stengel     pub fn axpby(
675*5fb68f37SKaren (Ren) Stengel         mut self,
676*5fb68f37SKaren (Ren) Stengel         alpha: crate::Scalar,
677*5fb68f37SKaren (Ren) Stengel         beta: crate::Scalar,
678*5fb68f37SKaren (Ren) Stengel         x: &crate::Vector,
679*5fb68f37SKaren (Ren) Stengel     ) -> crate::Result<Self> {
680*5fb68f37SKaren (Ren) Stengel         let ierr = unsafe { bind_ceed::CeedVectorAXPBY(self.ptr, alpha, beta, x.ptr) };
681*5fb68f37SKaren (Ren) Stengel         self.check_error(ierr)?;
682*5fb68f37SKaren (Ren) Stengel         Ok(self)
683*5fb68f37SKaren (Ren) Stengel     }
684*5fb68f37SKaren (Ren) Stengel 
6857ed177dbSJed Brown     /// Compute the pointwise multiplication w = x .* y for Vectors
6869df49d7eSJed Brown     ///
6879df49d7eSJed Brown     /// # arguments
6889df49d7eSJed Brown     ///
6899df49d7eSJed Brown     /// * `x` - first vector for product
6909df49d7eSJed Brown     /// * `y` - second vector for product
6919df49d7eSJed Brown     ///
6929df49d7eSJed Brown     /// ```
6939df49d7eSJed Brown     /// # use libceed::prelude::*;
6944d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6959df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
696c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
697c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
698c68be7a2SJeremy L Thompson     /// let y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6999df49d7eSJed Brown     ///
700c68be7a2SJeremy L Thompson     /// w = w.pointwise_mult(&x, &y)?;
701d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
702d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
703d3677ae8SJeremy L Thompson     /// }
704c68be7a2SJeremy L Thompson     /// # Ok(())
705c68be7a2SJeremy L Thompson     /// # }
7069df49d7eSJed Brown     /// ```
7079df49d7eSJed Brown     #[allow(unused_mut)]
7089df49d7eSJed Brown     pub fn pointwise_mult(mut self, x: &crate::Vector, y: &crate::Vector) -> crate::Result<Self> {
7099df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorPointwiseMult(self.ptr, x.ptr, y.ptr) };
7101142270cSJeremy L Thompson         self.check_error(ierr)?;
7119df49d7eSJed Brown         Ok(self)
7129df49d7eSJed Brown     }
7139df49d7eSJed Brown 
7147ed177dbSJed Brown     /// Compute the pointwise multiplication w = w .* x for Vectors
7159df49d7eSJed Brown     ///
7169df49d7eSJed Brown     /// # arguments
7179df49d7eSJed Brown     ///
7189df49d7eSJed Brown     /// * `x` - second vector for product
7199df49d7eSJed Brown     ///
7209df49d7eSJed Brown     /// ```
7219df49d7eSJed Brown     /// # use libceed::prelude::*;
7224d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
7239df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
724c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
725c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
7269df49d7eSJed Brown     ///
727c68be7a2SJeremy L Thompson     /// w = w.pointwise_scale(&x)?;
728d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
729d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
730d3677ae8SJeremy L Thompson     /// }
731c68be7a2SJeremy L Thompson     /// # Ok(())
732c68be7a2SJeremy L Thompson     /// # }
7339df49d7eSJed Brown     /// ```
7349df49d7eSJed Brown     #[allow(unused_mut)]
7359df49d7eSJed Brown     pub fn pointwise_scale(mut self, x: &crate::Vector) -> crate::Result<Self> {
7369df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorPointwiseMult(self.ptr, self.ptr, x.ptr) };
7371142270cSJeremy L Thompson         self.check_error(ierr)?;
7389df49d7eSJed Brown         Ok(self)
7399df49d7eSJed Brown     }
7409df49d7eSJed Brown 
7417ed177dbSJed Brown     /// Compute the pointwise multiplication w = w .* w for a Vector
7429df49d7eSJed Brown     ///
7439df49d7eSJed Brown     /// ```
7449df49d7eSJed Brown     /// # use libceed::prelude::*;
7454d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
7469df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
747c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
7489df49d7eSJed Brown     ///
749c68be7a2SJeremy L Thompson     /// w = w.pointwise_square()?;
750d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
751d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
752d3677ae8SJeremy L Thompson     /// }
753c68be7a2SJeremy L Thompson     /// # Ok(())
754c68be7a2SJeremy L Thompson     /// # }
7559df49d7eSJed Brown     /// ```
7569df49d7eSJed Brown     #[allow(unused_mut)]
7579df49d7eSJed Brown     pub fn pointwise_square(mut self) -> crate::Result<Self> {
7589df49d7eSJed Brown         let ierr = unsafe { bind_ceed::CeedVectorPointwiseMult(self.ptr, self.ptr, self.ptr) };
7591142270cSJeremy L Thompson         self.check_error(ierr)?;
7609df49d7eSJed Brown         Ok(self)
7619df49d7eSJed Brown     }
7629df49d7eSJed Brown }
7639df49d7eSJed Brown 
7649df49d7eSJed Brown // -----------------------------------------------------------------------------
7659df49d7eSJed Brown // Vector Viewer
7669df49d7eSJed Brown // -----------------------------------------------------------------------------
7679df49d7eSJed Brown /// A (host) view of a Vector with Deref to slice.  We can't make
7689df49d7eSJed Brown /// Vector itself Deref to slice because we can't handle the drop to
7699df49d7eSJed Brown /// call bind_ceed::CeedVectorRestoreArrayRead().
7709df49d7eSJed Brown #[derive(Debug)]
7719df49d7eSJed Brown pub struct VectorView<'a> {
7729df49d7eSJed Brown     vec: &'a Vector<'a>,
77380a9ef05SNatalie Beams     array: *const crate::Scalar,
7749df49d7eSJed Brown }
7759df49d7eSJed Brown 
7769df49d7eSJed Brown impl<'a> VectorView<'a> {
7779df49d7eSJed Brown     /// Construct a VectorView from a Vector reference
7784d27c890SJeremy L Thompson     fn new(vec: &'a Vector) -> crate::Result<Self> {
7799df49d7eSJed Brown         let mut array = std::ptr::null();
780e78171edSJeremy L Thompson         let ierr = unsafe {
7819df49d7eSJed Brown             bind_ceed::CeedVectorGetArrayRead(
7829df49d7eSJed Brown                 vec.ptr,
7839df49d7eSJed Brown                 crate::MemType::Host as bind_ceed::CeedMemType,
7849df49d7eSJed Brown                 &mut array,
785e78171edSJeremy L Thompson             )
786e78171edSJeremy L Thompson         };
787e78171edSJeremy L Thompson         vec.check_error(ierr)?;
788e78171edSJeremy L Thompson         Ok(Self {
7899df49d7eSJed Brown             vec: vec,
7909df49d7eSJed Brown             array: array,
791e78171edSJeremy L Thompson         })
7929df49d7eSJed Brown     }
7939df49d7eSJed Brown }
7949df49d7eSJed Brown 
7959df49d7eSJed Brown // Destructor
7969df49d7eSJed Brown impl<'a> Drop for VectorView<'a> {
7979df49d7eSJed Brown     fn drop(&mut self) {
7989df49d7eSJed Brown         unsafe {
7999df49d7eSJed Brown             bind_ceed::CeedVectorRestoreArrayRead(self.vec.ptr, &mut self.array);
8009df49d7eSJed Brown         }
8019df49d7eSJed Brown     }
8029df49d7eSJed Brown }
8039df49d7eSJed Brown 
8049df49d7eSJed Brown // Data access
8059df49d7eSJed Brown impl<'a> Deref for VectorView<'a> {
80680a9ef05SNatalie Beams     type Target = [crate::Scalar];
80780a9ef05SNatalie Beams     fn deref(&self) -> &[crate::Scalar] {
8089df49d7eSJed Brown         unsafe { std::slice::from_raw_parts(self.array, self.vec.len()) }
8099df49d7eSJed Brown     }
8109df49d7eSJed Brown }
8119df49d7eSJed Brown 
8129df49d7eSJed Brown // Viewing
8139df49d7eSJed Brown impl<'a> fmt::Display for VectorView<'a> {
8149df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
8159df49d7eSJed Brown         write!(f, "VectorView({:?})", self.deref())
8169df49d7eSJed Brown     }
8179df49d7eSJed Brown }
8189df49d7eSJed Brown 
8199df49d7eSJed Brown // -----------------------------------------------------------------------------
8209df49d7eSJed Brown // Vector Viewer Mutable
8219df49d7eSJed Brown // -----------------------------------------------------------------------------
8229df49d7eSJed Brown /// A mutable (host) view of a Vector with Deref to slice.
8239df49d7eSJed Brown #[derive(Debug)]
8249df49d7eSJed Brown pub struct VectorViewMut<'a> {
8259df49d7eSJed Brown     vec: &'a Vector<'a>,
82680a9ef05SNatalie Beams     array: *mut crate::Scalar,
8279df49d7eSJed Brown }
8289df49d7eSJed Brown 
8299df49d7eSJed Brown impl<'a> VectorViewMut<'a> {
8309df49d7eSJed Brown     /// Construct a VectorViewMut from a Vector reference
8314d27c890SJeremy L Thompson     fn new(vec: &'a mut Vector) -> crate::Result<Self> {
8329df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
833e78171edSJeremy L Thompson         let ierr = unsafe {
8349df49d7eSJed Brown             bind_ceed::CeedVectorGetArray(
8359df49d7eSJed Brown                 vec.ptr,
8369df49d7eSJed Brown                 crate::MemType::Host as bind_ceed::CeedMemType,
8379df49d7eSJed Brown                 &mut ptr,
838e78171edSJeremy L Thompson             )
839e78171edSJeremy L Thompson         };
840e78171edSJeremy L Thompson         vec.check_error(ierr)?;
841e78171edSJeremy L Thompson         Ok(Self {
8429df49d7eSJed Brown             vec: vec,
8439df49d7eSJed Brown             array: ptr,
844e78171edSJeremy L Thompson         })
8459df49d7eSJed Brown     }
8469df49d7eSJed Brown }
8479df49d7eSJed Brown 
8489df49d7eSJed Brown // Destructor
8499df49d7eSJed Brown impl<'a> Drop for VectorViewMut<'a> {
8509df49d7eSJed Brown     fn drop(&mut self) {
8519df49d7eSJed Brown         unsafe {
8529df49d7eSJed Brown             bind_ceed::CeedVectorRestoreArray(self.vec.ptr, &mut self.array);
8539df49d7eSJed Brown         }
8549df49d7eSJed Brown     }
8559df49d7eSJed Brown }
8569df49d7eSJed Brown 
8579df49d7eSJed Brown // Data access
8589df49d7eSJed Brown impl<'a> Deref for VectorViewMut<'a> {
85980a9ef05SNatalie Beams     type Target = [crate::Scalar];
86080a9ef05SNatalie Beams     fn deref(&self) -> &[crate::Scalar] {
8619df49d7eSJed Brown         unsafe { std::slice::from_raw_parts(self.array, self.vec.len()) }
8629df49d7eSJed Brown     }
8639df49d7eSJed Brown }
8649df49d7eSJed Brown 
8659df49d7eSJed Brown // Mutable data access
8669df49d7eSJed Brown impl<'a> DerefMut for VectorViewMut<'a> {
86780a9ef05SNatalie Beams     fn deref_mut(&mut self) -> &mut [crate::Scalar] {
8689df49d7eSJed Brown         unsafe { std::slice::from_raw_parts_mut(self.array, self.vec.len()) }
8699df49d7eSJed Brown     }
8709df49d7eSJed Brown }
8719df49d7eSJed Brown 
8729df49d7eSJed Brown // Viewing
8739df49d7eSJed Brown impl<'a> fmt::Display for VectorViewMut<'a> {
8749df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
8759df49d7eSJed Brown         write!(f, "VectorViewMut({:?})", self.deref())
8769df49d7eSJed Brown     }
8779df49d7eSJed Brown }
8789df49d7eSJed Brown 
8799df49d7eSJed Brown // -----------------------------------------------------------------------------
880