xref: /libCEED/rust/libceed/src/vector.rs (revision eb07d68f27a4df7d950ee845c62e88e28e25414a)
1d275d636SJeremy L Thompson // Copyright (c) 2017-2025, 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     /// ```
48*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, VectorOpt};
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     /// ```
74*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, VectorOpt};
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     /// ```
100*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, VectorOpt};
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> {
128*eb07d68fSJeremy L Thompson     pub(crate) vector: 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>(
152*eb07d68fSJeremy L Thompson         vec: &'b mut 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         );
160656ef1e5SJeremy L Thompson         vec.check_error(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             )
167656ef1e5SJeremy L Thompson         })?;
16835afbe75SJeremy L Thompson         Ok(Self {
169*eb07d68fSJeremy L Thompson             vector: unsafe { Vector::from_raw(vec.ptr_copy_mut()?)? },
17035afbe75SJeremy L Thompson             _slice: slice,
17135afbe75SJeremy L Thompson         })
17235afbe75SJeremy L Thompson     }
17335afbe75SJeremy L Thompson }
17435afbe75SJeremy L Thompson 
17535afbe75SJeremy L Thompson // -----------------------------------------------------------------------------
1767ed177dbSJed Brown // Vector context wrapper
1779df49d7eSJed Brown // -----------------------------------------------------------------------------
1789df49d7eSJed Brown #[derive(Debug)]
1799df49d7eSJed Brown pub struct Vector<'a> {
1809df49d7eSJed Brown     pub(crate) ptr: bind_ceed::CeedVector,
1811142270cSJeremy L Thompson     _lifeline: PhantomData<&'a ()>,
1829df49d7eSJed Brown }
1839df49d7eSJed Brown impl From<&'_ Vector<'_>> for bind_ceed::CeedVector {
1849df49d7eSJed Brown     fn from(vec: &Vector) -> Self {
1859df49d7eSJed Brown         vec.ptr
1869df49d7eSJed Brown     }
1879df49d7eSJed Brown }
1889df49d7eSJed Brown 
1899df49d7eSJed Brown // -----------------------------------------------------------------------------
1909df49d7eSJed Brown // Destructor
1919df49d7eSJed Brown // -----------------------------------------------------------------------------
1929df49d7eSJed Brown impl<'a> Drop for Vector<'a> {
1939df49d7eSJed Brown     fn drop(&mut self) {
1949df49d7eSJed Brown         let not_none_and_active = self.ptr != unsafe { bind_ceed::CEED_VECTOR_NONE }
1959df49d7eSJed Brown             && self.ptr != unsafe { bind_ceed::CEED_VECTOR_ACTIVE };
1969df49d7eSJed Brown 
1979df49d7eSJed Brown         if not_none_and_active {
1989df49d7eSJed Brown             unsafe { bind_ceed::CeedVectorDestroy(&mut self.ptr) };
1999df49d7eSJed Brown         }
2009df49d7eSJed Brown     }
2019df49d7eSJed Brown }
2029df49d7eSJed Brown 
2039df49d7eSJed Brown // -----------------------------------------------------------------------------
2049df49d7eSJed Brown // Display
2059df49d7eSJed Brown // -----------------------------------------------------------------------------
2069df49d7eSJed Brown impl<'a> fmt::Display for Vector<'a> {
2079df49d7eSJed Brown     /// View a Vector
2089df49d7eSJed Brown     ///
2099df49d7eSJed Brown     /// ```
2109df49d7eSJed Brown     /// # use libceed::prelude::*;
2114d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
2129df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
213c68be7a2SJeremy L Thompson     /// let vec = libceed::vector::Vector::from_slice(&ceed, &[1., 2., 3.])?;
2149df49d7eSJed Brown     /// assert_eq!(
2159df49d7eSJed Brown     ///     vec.to_string(),
2169df49d7eSJed Brown     ///     "CeedVector length 3
2179df49d7eSJed Brown     ///     1.00000000
2189df49d7eSJed Brown     ///     2.00000000
2199df49d7eSJed Brown     ///     3.00000000
2209df49d7eSJed Brown     /// "
221c68be7a2SJeremy L Thompson     /// );
222c68be7a2SJeremy L Thompson     /// # Ok(())
223c68be7a2SJeremy L Thompson     /// # }
2249df49d7eSJed Brown     /// ```
2259df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
2269df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
2279df49d7eSJed Brown         let mut sizeloc = crate::MAX_BUFFER_LENGTH;
2289df49d7eSJed Brown         let format = CString::new("%12.8f").expect("CString::new failed");
2299df49d7eSJed Brown         let format_c: *const c_char = format.into_raw();
2309df49d7eSJed Brown         let cstring = unsafe {
2319df49d7eSJed Brown             let file = bind_ceed::open_memstream(&mut ptr, &mut sizeloc);
2329df49d7eSJed Brown             bind_ceed::CeedVectorView(self.ptr, format_c, file);
2339df49d7eSJed Brown             bind_ceed::fclose(file);
2349df49d7eSJed Brown             CString::from_raw(ptr)
2359df49d7eSJed Brown         };
2369df49d7eSJed Brown         cstring.to_string_lossy().fmt(f)
2379df49d7eSJed Brown     }
2389df49d7eSJed Brown }
2399df49d7eSJed Brown 
2409df49d7eSJed Brown // -----------------------------------------------------------------------------
2419df49d7eSJed Brown // Implementations
2429df49d7eSJed Brown // -----------------------------------------------------------------------------
2439df49d7eSJed Brown impl<'a> Vector<'a> {
2449df49d7eSJed Brown     // Constructors
245594ef120SJeremy L Thompson     pub fn create(ceed: &crate::Ceed, n: usize) -> crate::Result<Self> {
2461f9221feSJeremy L Thompson         let n = isize::try_from(n).unwrap();
2479df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
248656ef1e5SJeremy L Thompson         ceed.check_error(unsafe { bind_ceed::CeedVectorCreate(ceed.ptr, n, &mut ptr) })?;
2491142270cSJeremy L Thompson         Ok(Self {
2501142270cSJeremy L Thompson             ptr,
2511142270cSJeremy L Thompson             _lifeline: PhantomData,
2521142270cSJeremy L Thompson         })
2539df49d7eSJed Brown     }
2549df49d7eSJed Brown 
2552b671a0aSJeremy L Thompson     pub(crate) unsafe fn from_raw(ptr: bind_ceed::CeedVector) -> crate::Result<Self> {
2561142270cSJeremy L Thompson         Ok(Self {
2571142270cSJeremy L Thompson             ptr,
2581142270cSJeremy L Thompson             _lifeline: PhantomData,
2591142270cSJeremy L Thompson         })
2609df49d7eSJed Brown     }
2619df49d7eSJed Brown 
262e46a712dSJeremy L Thompson     fn ptr_copy_mut(&mut self) -> crate::Result<bind_ceed::CeedVector> {
263e46a712dSJeremy L Thompson         let mut ptr_copy = std::ptr::null_mut();
264656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorReferenceCopy(self.ptr, &mut ptr_copy) })?;
265e46a712dSJeremy L Thompson         Ok(ptr_copy)
266e46a712dSJeremy L Thompson     }
267e46a712dSJeremy L Thompson 
2685fb68f37SKaren (Ren) Stengel     /// Copy the array of self into vec_copy
2695fb68f37SKaren (Ren) Stengel     ///
2705fb68f37SKaren (Ren) Stengel     /// # arguments
2715fb68f37SKaren (Ren) Stengel     ///
2725fb68f37SKaren (Ren) Stengel     /// * `vec_source` - vector to copy array values from
27346de3253SJeremy L Thompson     ///
27446de3253SJeremy L Thompson     /// ```
275*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
27646de3253SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
27746de3253SJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
27846de3253SJeremy L Thompson     /// let a = ceed.vector_from_slice(&[1., 2., 3.])?;
27946de3253SJeremy L Thompson     /// let mut b = ceed.vector(3)?;
28046de3253SJeremy L Thompson     ///
28146de3253SJeremy L Thompson     /// b.copy_from(&a)?;
28246de3253SJeremy L Thompson     /// for (i, v) in b.view()?.iter().enumerate() {
28346de3253SJeremy L Thompson     ///     assert_eq!(*v, (i + 1) as Scalar, "Copy contents not set correctly");
28446de3253SJeremy L Thompson     /// }
28546de3253SJeremy L Thompson     /// # Ok(())
28646de3253SJeremy L Thompson     /// # }
28746de3253SJeremy L Thompson     /// ```
28846de3253SJeremy L Thompson     /// ```
289*eb07d68fSJeremy L Thompson     pub fn copy_from(&mut self, vec_source: &Vector) -> crate::Result<i32> {
290656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorCopy(vec_source.ptr, self.ptr) })
2915fb68f37SKaren (Ren) Stengel     }
2925fb68f37SKaren (Ren) Stengel 
2939df49d7eSJed Brown     /// Create a Vector from a slice
2949df49d7eSJed Brown     ///
2959df49d7eSJed Brown     /// # arguments
2969df49d7eSJed Brown     ///
2979df49d7eSJed Brown     /// * `slice` - values to initialize vector with
2989df49d7eSJed Brown     ///
2999df49d7eSJed Brown     /// ```
3009df49d7eSJed Brown     /// # use libceed::prelude::*;
3014d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3029df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
303*eb07d68fSJeremy L Thompson     /// let vec = libceed::Vector::from_slice(&ceed, &[1., 2., 3.])?;
3049df49d7eSJed Brown     /// assert_eq!(vec.length(), 3, "Incorrect length from slice");
305c68be7a2SJeremy L Thompson     /// # Ok(())
306c68be7a2SJeremy L Thompson     /// # }
3079df49d7eSJed Brown     /// ```
308594ef120SJeremy L Thompson     pub fn from_slice(ceed: &crate::Ceed, v: &[crate::Scalar]) -> crate::Result<Self> {
3099df49d7eSJed Brown         let mut x = Self::create(ceed, v.len())?;
3109df49d7eSJed Brown         x.set_slice(v)?;
3119df49d7eSJed Brown         Ok(x)
3129df49d7eSJed Brown     }
3139df49d7eSJed Brown 
3149df49d7eSJed Brown     /// Create a Vector from a mutable array reference
3159df49d7eSJed Brown     ///
3169df49d7eSJed Brown     /// # arguments
3179df49d7eSJed Brown     ///
3189df49d7eSJed Brown     /// * `slice` - values to initialize vector with
3199df49d7eSJed Brown     ///
3209df49d7eSJed Brown     /// ```
3219df49d7eSJed Brown     /// # use libceed::prelude::*;
3224d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3239df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
3249df49d7eSJed Brown     /// let mut rust_vec = vec![1., 2., 3.];
325c68be7a2SJeremy L Thompson     /// let vec = libceed::vector::Vector::from_array(&ceed, &mut rust_vec)?;
3269df49d7eSJed Brown     ///
3279df49d7eSJed Brown     /// assert_eq!(vec.length(), 3, "Incorrect length from slice");
328c68be7a2SJeremy L Thompson     /// # Ok(())
329c68be7a2SJeremy L Thompson     /// # }
3309df49d7eSJed Brown     /// ```
331594ef120SJeremy L Thompson     pub fn from_array(ceed: &crate::Ceed, v: &mut [crate::Scalar]) -> crate::Result<Self> {
3329df49d7eSJed Brown         let x = Self::create(ceed, v.len())?;
3339df49d7eSJed Brown         let (host, user_pointer) = (
3349df49d7eSJed Brown             crate::MemType::Host as bind_ceed::CeedMemType,
3359df49d7eSJed Brown             crate::CopyMode::UsePointer as bind_ceed::CeedCopyMode,
3369df49d7eSJed Brown         );
33780a9ef05SNatalie Beams         let v = v.as_ptr() as *mut crate::Scalar;
338656ef1e5SJeremy L Thompson         ceed.check_error(unsafe { bind_ceed::CeedVectorSetArray(x.ptr, host, user_pointer, v) })?;
3399df49d7eSJed Brown         Ok(x)
3409df49d7eSJed Brown     }
3419df49d7eSJed Brown 
34211544396SJeremy L Thompson     // Raw Ceed for error handling
34311544396SJeremy L Thompson     #[doc(hidden)]
34411544396SJeremy L Thompson     fn ceed(&self) -> bind_ceed::Ceed {
34511544396SJeremy L Thompson         unsafe { bind_ceed::CeedVectorReturnCeed(self.ptr) }
34611544396SJeremy L Thompson     }
34711544396SJeremy L Thompson 
3481142270cSJeremy L Thompson     // Error handling
3491142270cSJeremy L Thompson     #[doc(hidden)]
3501142270cSJeremy L Thompson     fn check_error(&self, ierr: i32) -> crate::Result<i32> {
35111544396SJeremy L Thompson         crate::check_error(|| self.ceed(), ierr)
3521142270cSJeremy L Thompson     }
3531142270cSJeremy L Thompson 
3547ed177dbSJed Brown     /// Returns the length of a Vector
3559df49d7eSJed Brown     ///
3569df49d7eSJed Brown     /// ```
3579df49d7eSJed Brown     /// # use libceed::prelude::*;
3584d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3599df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
360c68be7a2SJeremy L Thompson     /// let vec = ceed.vector(10)?;
3619df49d7eSJed Brown     ///
3629df49d7eSJed Brown     /// let n = vec.length();
3639df49d7eSJed Brown     /// assert_eq!(n, 10, "Incorrect length");
364c68be7a2SJeremy L Thompson     /// # Ok(())
365c68be7a2SJeremy L Thompson     /// # }
3669df49d7eSJed Brown     /// ```
3679df49d7eSJed Brown     pub fn length(&self) -> usize {
3689df49d7eSJed Brown         let mut n = 0;
3699df49d7eSJed Brown         unsafe { bind_ceed::CeedVectorGetLength(self.ptr, &mut n) };
3709df49d7eSJed Brown         usize::try_from(n).unwrap()
3719df49d7eSJed Brown     }
3729df49d7eSJed Brown 
3737ed177dbSJed Brown     /// Returns the length of a Vector
3749df49d7eSJed Brown     ///
3759df49d7eSJed Brown     /// ```
3769df49d7eSJed Brown     /// # use libceed::prelude::*;
3774d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3789df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
379c68be7a2SJeremy L Thompson     /// let vec = ceed.vector(10)?;
3809df49d7eSJed Brown     /// assert_eq!(vec.len(), 10, "Incorrect length");
381c68be7a2SJeremy L Thompson     /// # Ok(())
382c68be7a2SJeremy L Thompson     /// # }
3839df49d7eSJed Brown     /// ```
3849df49d7eSJed Brown     pub fn len(&self) -> usize {
3859df49d7eSJed Brown         self.length()
3869df49d7eSJed Brown     }
3879df49d7eSJed Brown 
3887ed177dbSJed Brown     /// Set the Vector to a constant value
3899df49d7eSJed Brown     ///
3909df49d7eSJed Brown     /// # arguments
3919df49d7eSJed Brown     ///
3929df49d7eSJed Brown     /// * `val` - Value to be used
3939df49d7eSJed Brown     ///
3949df49d7eSJed Brown     /// ```
3959df49d7eSJed Brown     /// # use libceed::prelude::*;
3964d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
3979df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
3989df49d7eSJed Brown     /// let len = 10;
399c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(len)?;
4009df49d7eSJed Brown     ///
4019df49d7eSJed Brown     /// let val = 42.0;
402c68be7a2SJeremy L Thompson     /// vec.set_value(val)?;
4039df49d7eSJed Brown     ///
404d3677ae8SJeremy L Thompson     /// for v in vec.view()?.iter() {
4059df49d7eSJed Brown     ///     assert_eq!(*v, val, "Value not set correctly");
406d3677ae8SJeremy L Thompson     /// }
407c68be7a2SJeremy L Thompson     /// # Ok(())
408c68be7a2SJeremy L Thompson     /// # }
4099df49d7eSJed Brown     /// ```
41080a9ef05SNatalie Beams     pub fn set_value(&mut self, value: crate::Scalar) -> crate::Result<i32> {
411656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorSetValue(self.ptr, value) })
4129df49d7eSJed Brown     }
4139df49d7eSJed Brown 
4149df49d7eSJed Brown     /// Set values from a slice of the same length
4159df49d7eSJed Brown     ///
4169df49d7eSJed Brown     /// # arguments
4179df49d7eSJed Brown     ///
4189df49d7eSJed Brown     /// * `slice` - values to into self; length must match
4199df49d7eSJed Brown     ///
4209df49d7eSJed Brown     /// ```
421*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
4224d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
4239df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
424c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(4)?;
425c68be7a2SJeremy L Thompson     /// vec.set_slice(&[10., 11., 12., 13.])?;
4269df49d7eSJed Brown     ///
427d3677ae8SJeremy L Thompson     /// for (i, v) in vec.view()?.iter().enumerate() {
42880a9ef05SNatalie Beams     ///     assert_eq!(*v, 10. + i as Scalar, "Slice not set correctly");
429d3677ae8SJeremy L Thompson     /// }
430c68be7a2SJeremy L Thompson     /// # Ok(())
431c68be7a2SJeremy L Thompson     /// # }
4329df49d7eSJed Brown     /// ```
43380a9ef05SNatalie Beams     pub fn set_slice(&mut self, slice: &[crate::Scalar]) -> crate::Result<i32> {
4349df49d7eSJed Brown         assert_eq!(self.length(), slice.len());
4359df49d7eSJed Brown         let (host, copy_mode) = (
4369df49d7eSJed Brown             crate::MemType::Host as bind_ceed::CeedMemType,
4379df49d7eSJed Brown             crate::CopyMode::CopyValues as bind_ceed::CeedCopyMode,
4389df49d7eSJed Brown         );
439656ef1e5SJeremy L Thompson         self.check_error(unsafe {
44080a9ef05SNatalie Beams             bind_ceed::CeedVectorSetArray(
44180a9ef05SNatalie Beams                 self.ptr,
44280a9ef05SNatalie Beams                 host,
44380a9ef05SNatalie Beams                 copy_mode,
44480a9ef05SNatalie Beams                 slice.as_ptr() as *mut crate::Scalar,
44580a9ef05SNatalie Beams             )
446656ef1e5SJeremy L Thompson         })
4479df49d7eSJed Brown     }
4489df49d7eSJed Brown 
449486868d3SJeremy L Thompson     /// Wrap a mutable slice in a Vector of the same length
45056e67e05SJeremy L Thompson     ///
45156e67e05SJeremy L Thompson     /// # arguments
45256e67e05SJeremy L Thompson     ///
453486868d3SJeremy L Thompson     /// * `slice` - values to wrap in self; length must match
45456e67e05SJeremy L Thompson     ///
45556e67e05SJeremy L Thompson     /// ```
456*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
45756e67e05SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
45856e67e05SJeremy L Thompson     /// # let ceed = libceed::Ceed::default_init();
45956e67e05SJeremy L Thompson     /// let mut vec = ceed.vector(4)?;
460486868d3SJeremy L Thompson     /// let mut array = [10., 11., 12., 13.];
46156e67e05SJeremy L Thompson     ///
46256e67e05SJeremy L Thompson     /// {
463486868d3SJeremy L Thompson     ///     // `wrapper` holds a mutable reference to the wrapped slice
464486868d3SJeremy L Thompson     ///     //   that is dropped when `wrapper` goes out of scope
465486868d3SJeremy L Thompson     ///     let wrapper = vec.wrap_slice_mut(&mut array)?;
466d3677ae8SJeremy L Thompson     ///     for (i, v) in vec.view()?.iter().enumerate() {
46756e67e05SJeremy L Thompson     ///         assert_eq!(*v, 10. + i as Scalar, "Slice not set correctly");
468d3677ae8SJeremy L Thompson     ///     }
46956e67e05SJeremy L Thompson     ///
470486868d3SJeremy L Thompson     ///     // This line will not compile, as the `wrapper` holds mutable
471486868d3SJeremy L Thompson     ///     //   access to the `array`
472486868d3SJeremy L Thompson     ///     // array[0] = 5.0;
47356e67e05SJeremy L Thompson     ///
474486868d3SJeremy L Thompson     ///     // Changes here are reflected in the `array`
47556e67e05SJeremy L Thompson     ///     vec.set_value(5.0)?;
476d3677ae8SJeremy L Thompson     ///     for v in vec.view()?.iter() {
477486868d3SJeremy L Thompson     ///         assert_eq!(*v, 5.0 as Scalar, "Value not set correctly");
478d3677ae8SJeremy L Thompson     ///     }
47956e67e05SJeremy L Thompson     /// }
48056e67e05SJeremy L Thompson     ///
481486868d3SJeremy L Thompson     /// // 'array' remains changed
482d3677ae8SJeremy L Thompson     /// for v in array.iter() {
483486868d3SJeremy L Thompson     ///     assert_eq!(*v, 5.0 as Scalar, "Array not mutated correctly");
484d3677ae8SJeremy L Thompson     /// }
48556e67e05SJeremy L Thompson     ///
486486868d3SJeremy L Thompson     /// // While changes to `vec` no longer affect `array`
487486868d3SJeremy L Thompson     /// vec.set_value(6.0)?;
488d3677ae8SJeremy L Thompson     /// for v in array.iter() {
489486868d3SJeremy L Thompson     ///     assert_eq!(*v, 5.0 as Scalar, "Array mutated without permission");
490d3677ae8SJeremy L Thompson     /// }
49156e67e05SJeremy L Thompson     /// # Ok(())
49256e67e05SJeremy L Thompson     /// # }
49356e67e05SJeremy L Thompson     /// ```
494486868d3SJeremy L Thompson     pub fn wrap_slice_mut<'b>(
49556e67e05SJeremy L Thompson         &mut self,
49656e67e05SJeremy L Thompson         slice: &'b mut [crate::Scalar],
497486868d3SJeremy L Thompson     ) -> crate::Result<VectorSliceWrapper<'b>> {
498*eb07d68fSJeremy L Thompson         VectorSliceWrapper::from_vector_and_slice_mut(self, slice)
49956e67e05SJeremy L Thompson     }
50056e67e05SJeremy L Thompson 
5017ed177dbSJed Brown     /// Sync the Vector to a specified memtype
5029df49d7eSJed Brown     ///
5039df49d7eSJed Brown     /// # arguments
5049df49d7eSJed Brown     ///
5059df49d7eSJed Brown     /// * `mtype` - Memtype to be synced
5069df49d7eSJed Brown     ///
5079df49d7eSJed Brown     /// ```
508*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, MemType};
5094d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5109df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
5119df49d7eSJed Brown     /// let len = 10;
512c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector(len)?;
5139df49d7eSJed Brown     ///
5149df49d7eSJed Brown     /// let val = 42.0;
5159df49d7eSJed Brown     /// vec.set_value(val);
516c68be7a2SJeremy L Thompson     /// vec.sync(MemType::Host)?;
5179df49d7eSJed Brown     ///
518d3677ae8SJeremy L Thompson     /// for v in vec.view()?.iter() {
5199df49d7eSJed Brown     ///     assert_eq!(*v, val, "Value not set correctly");
520d3677ae8SJeremy L Thompson     /// }
521c68be7a2SJeremy L Thompson     /// # Ok(())
522c68be7a2SJeremy L Thompson     /// # }
5239df49d7eSJed Brown     /// ```
5249df49d7eSJed Brown     pub fn sync(&self, mtype: crate::MemType) -> crate::Result<i32> {
525656ef1e5SJeremy L Thompson         self.check_error(unsafe {
526656ef1e5SJeremy L Thompson             bind_ceed::CeedVectorSyncArray(self.ptr, mtype as bind_ceed::CeedMemType)
527656ef1e5SJeremy L Thompson         })
5289df49d7eSJed Brown     }
5299df49d7eSJed Brown 
5309df49d7eSJed Brown     /// Create an immutable view
5319df49d7eSJed Brown     ///
5329df49d7eSJed Brown     /// ```
5339df49d7eSJed Brown     /// # use libceed::prelude::*;
5344d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5359df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
536c68be7a2SJeremy L Thompson     /// let vec = ceed.vector_from_slice(&[10., 11., 12., 13.])?;
5379df49d7eSJed Brown     ///
538e78171edSJeremy L Thompson     /// let v = vec.view()?;
5399df49d7eSJed Brown     /// assert_eq!(v[0..2], [10., 11.]);
5409df49d7eSJed Brown     ///
5419df49d7eSJed Brown     /// // It is valid to have multiple immutable views
542e78171edSJeremy L Thompson     /// let w = vec.view()?;
5439df49d7eSJed Brown     /// assert_eq!(v[1..], w[1..]);
544c68be7a2SJeremy L Thompson     /// # Ok(())
545c68be7a2SJeremy L Thompson     /// # }
5469df49d7eSJed Brown     /// ```
547e78171edSJeremy L Thompson     pub fn view(&self) -> crate::Result<VectorView> {
5489df49d7eSJed Brown         VectorView::new(self)
5499df49d7eSJed Brown     }
5509df49d7eSJed Brown 
5519df49d7eSJed Brown     /// Create an mutable view
5529df49d7eSJed Brown     ///
5539df49d7eSJed Brown     /// ```
5549df49d7eSJed Brown     /// # use libceed::prelude::*;
5554d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5569df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
557c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector_from_slice(&[10., 11., 12., 13.])?;
5589df49d7eSJed Brown     ///
5599df49d7eSJed Brown     /// {
560e78171edSJeremy L Thompson     ///     let mut v = vec.view_mut()?;
5619df49d7eSJed Brown     ///     v[2] = 9.;
5629df49d7eSJed Brown     /// }
5639df49d7eSJed Brown     ///
564e78171edSJeremy L Thompson     /// let w = vec.view()?;
5659df49d7eSJed Brown     /// assert_eq!(w[2], 9., "View did not mutate data");
566c68be7a2SJeremy L Thompson     /// # Ok(())
567c68be7a2SJeremy L Thompson     /// # }
5689df49d7eSJed Brown     /// ```
569e78171edSJeremy L Thompson     pub fn view_mut(&mut self) -> crate::Result<VectorViewMut> {
5709df49d7eSJed Brown         VectorViewMut::new(self)
5719df49d7eSJed Brown     }
5729df49d7eSJed Brown 
5737ed177dbSJed Brown     /// Return the norm of a Vector
5749df49d7eSJed Brown     ///
5759df49d7eSJed Brown     /// # arguments
5769df49d7eSJed Brown     ///
5779df49d7eSJed Brown     /// * `ntype` - Norm type One, Two, or Max
5789df49d7eSJed Brown     ///
5799df49d7eSJed Brown     /// ```
580*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, NormType};
5814d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
5829df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
583c68be7a2SJeremy L Thompson     /// let vec = ceed.vector_from_slice(&[1., 2., 3., 4.])?;
5849df49d7eSJed Brown     ///
585c68be7a2SJeremy L Thompson     /// let max_norm = vec.norm(NormType::Max)?;
5869df49d7eSJed Brown     /// assert_eq!(max_norm, 4.0, "Incorrect Max norm");
5879df49d7eSJed Brown     ///
588c68be7a2SJeremy L Thompson     /// let l1_norm = vec.norm(NormType::One)?;
5899df49d7eSJed Brown     /// assert_eq!(l1_norm, 10., "Incorrect L1 norm");
5909df49d7eSJed Brown     ///
591c68be7a2SJeremy L Thompson     /// let l2_norm = vec.norm(NormType::Two)?;
5929df49d7eSJed Brown     /// assert!((l2_norm - 5.477) < 1e-3, "Incorrect L2 norm");
593c68be7a2SJeremy L Thompson     /// # Ok(())
594c68be7a2SJeremy L Thompson     /// # }
5959df49d7eSJed Brown     /// ```
59680a9ef05SNatalie Beams     pub fn norm(&self, ntype: crate::NormType) -> crate::Result<crate::Scalar> {
59780a9ef05SNatalie Beams         let mut res: crate::Scalar = 0.0;
598656ef1e5SJeremy L Thompson         self.check_error(unsafe {
5999df49d7eSJed Brown             bind_ceed::CeedVectorNorm(self.ptr, ntype as bind_ceed::CeedNormType, &mut res)
600656ef1e5SJeremy L Thompson         })?;
6019df49d7eSJed Brown         Ok(res)
6029df49d7eSJed Brown     }
6039df49d7eSJed Brown 
6047ed177dbSJed Brown     /// Compute x = alpha x for a Vector
6059df49d7eSJed Brown     ///
6069df49d7eSJed Brown     /// # arguments
6079df49d7eSJed Brown     ///
6089df49d7eSJed Brown     /// * `alpha` - scaling factor
6099df49d7eSJed Brown     ///
6109df49d7eSJed Brown     /// ```
611*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
6124d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6139df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
614c68be7a2SJeremy L Thompson     /// let mut vec = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6159df49d7eSJed Brown     ///
616c68be7a2SJeremy L Thompson     /// vec = vec.scale(-1.0)?;
617d3677ae8SJeremy L Thompson     /// for (i, v) in vec.view()?.iter().enumerate() {
618d3677ae8SJeremy L Thompson     ///     assert_eq!(*v, -(i as Scalar), "Value not set correctly");
619d3677ae8SJeremy L Thompson     /// }
620c68be7a2SJeremy L Thompson     /// # Ok(())
621c68be7a2SJeremy L Thompson     /// # }
6229df49d7eSJed Brown     /// ```
6239df49d7eSJed Brown     #[allow(unused_mut)]
62480a9ef05SNatalie Beams     pub fn scale(mut self, alpha: crate::Scalar) -> crate::Result<Self> {
625656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorScale(self.ptr, alpha) })?;
6269df49d7eSJed Brown         Ok(self)
6279df49d7eSJed Brown     }
6289df49d7eSJed Brown 
6297ed177dbSJed Brown     /// Compute y = alpha x + y for a pair of Vectors
6309df49d7eSJed Brown     ///
6319df49d7eSJed Brown     /// # arguments
6329df49d7eSJed Brown     ///
6339df49d7eSJed Brown     /// * `alpha` - scaling factor
6349df49d7eSJed Brown     /// * `x`     - second vector, must be different than self
6359df49d7eSJed Brown     ///
6369df49d7eSJed Brown     /// ```
637*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
6384d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6399df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
640c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
641c68be7a2SJeremy L Thompson     /// let mut y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6429df49d7eSJed Brown     ///
643c68be7a2SJeremy L Thompson     /// y = y.axpy(-0.5, &x)?;
644d3677ae8SJeremy L Thompson     /// for (i, y) in y.view()?.iter().enumerate() {
645d3677ae8SJeremy L Thompson     ///     assert_eq!(*y, (i as Scalar) / 2.0, "Value not set correctly");
646d3677ae8SJeremy L Thompson     /// }
647c68be7a2SJeremy L Thompson     /// # Ok(())
648c68be7a2SJeremy L Thompson     /// # }
6499df49d7eSJed Brown     /// ```
6509df49d7eSJed Brown     #[allow(unused_mut)]
651*eb07d68fSJeremy L Thompson     pub fn axpy(mut self, alpha: crate::Scalar, x: &Vector) -> crate::Result<Self> {
652656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorAXPY(self.ptr, alpha, x.ptr) })?;
6539df49d7eSJed Brown         Ok(self)
6549df49d7eSJed Brown     }
6559df49d7eSJed Brown 
6565fb68f37SKaren (Ren) Stengel     /// Compute y = alpha x + beta y for a pair of Vectors
6575fb68f37SKaren (Ren) Stengel     ///
6585fb68f37SKaren (Ren) Stengel     /// # arguments
6595fb68f37SKaren (Ren) Stengel     ///
6605fb68f37SKaren (Ren) Stengel     /// * `alpha` - first scaling factor
6615fb68f37SKaren (Ren) Stengel     /// * `beta`  - second scaling factor
6625fb68f37SKaren (Ren) Stengel     /// * `x`     - second vector, must be different than self
6635fb68f37SKaren (Ren) Stengel     ///
6645fb68f37SKaren (Ren) Stengel     /// ```
665*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
6665fb68f37SKaren (Ren) Stengel     /// # fn main() -> libceed::Result<()> {
6675fb68f37SKaren (Ren) Stengel     /// # let ceed = libceed::Ceed::default_init();
6685fb68f37SKaren (Ren) Stengel     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6695fb68f37SKaren (Ren) Stengel     /// let mut y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
6705fb68f37SKaren (Ren) Stengel     ///
6715fb68f37SKaren (Ren) Stengel     /// y = y.axpby(-0.5, 1.0, &x)?;
6725fb68f37SKaren (Ren) Stengel     /// for (i, y) in y.view()?.iter().enumerate() {
673aa67b842SZach Atkins     ///     assert_eq!(*y, (i as Scalar) * 0.5, "Value not set correctly");
6745fb68f37SKaren (Ren) Stengel     /// }
6755fb68f37SKaren (Ren) Stengel     /// # Ok(())
6765fb68f37SKaren (Ren) Stengel     /// # }
6775fb68f37SKaren (Ren) Stengel     /// ```
6785fb68f37SKaren (Ren) Stengel     #[allow(unused_mut)]
6795fb68f37SKaren (Ren) Stengel     pub fn axpby(
6805fb68f37SKaren (Ren) Stengel         mut self,
6815fb68f37SKaren (Ren) Stengel         alpha: crate::Scalar,
6825fb68f37SKaren (Ren) Stengel         beta: crate::Scalar,
683*eb07d68fSJeremy L Thompson         x: &Vector,
6845fb68f37SKaren (Ren) Stengel     ) -> crate::Result<Self> {
685656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorAXPBY(self.ptr, alpha, beta, x.ptr) })?;
6865fb68f37SKaren (Ren) Stengel         Ok(self)
6875fb68f37SKaren (Ren) Stengel     }
6885fb68f37SKaren (Ren) Stengel 
6897ed177dbSJed Brown     /// Compute the pointwise multiplication w = x .* y for Vectors
6909df49d7eSJed Brown     ///
6919df49d7eSJed Brown     /// # arguments
6929df49d7eSJed Brown     ///
6939df49d7eSJed Brown     /// * `x` - first vector for product
6949df49d7eSJed Brown     /// * `y` - second vector for product
6959df49d7eSJed Brown     ///
6969df49d7eSJed Brown     /// ```
697*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
6984d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
6999df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
700c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
701c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
702c68be7a2SJeremy L Thompson     /// let y = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
7039df49d7eSJed Brown     ///
704c68be7a2SJeremy L Thompson     /// w = w.pointwise_mult(&x, &y)?;
705d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
706d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
707d3677ae8SJeremy L Thompson     /// }
708c68be7a2SJeremy L Thompson     /// # Ok(())
709c68be7a2SJeremy L Thompson     /// # }
7109df49d7eSJed Brown     /// ```
7119df49d7eSJed Brown     #[allow(unused_mut)]
712*eb07d68fSJeremy L Thompson     pub fn pointwise_mult(mut self, x: &Vector, y: &Vector) -> crate::Result<Self> {
713656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorPointwiseMult(self.ptr, x.ptr, y.ptr) })?;
7149df49d7eSJed Brown         Ok(self)
7159df49d7eSJed Brown     }
7169df49d7eSJed Brown 
7177ed177dbSJed Brown     /// Compute the pointwise multiplication w = w .* x for Vectors
7189df49d7eSJed Brown     ///
7199df49d7eSJed Brown     /// # arguments
7209df49d7eSJed Brown     ///
7219df49d7eSJed Brown     /// * `x` - second vector for product
7229df49d7eSJed Brown     ///
7239df49d7eSJed Brown     /// ```
724*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
7254d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
7269df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
727c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
728c68be7a2SJeremy L Thompson     /// let x = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
7299df49d7eSJed Brown     ///
730c68be7a2SJeremy L Thompson     /// w = w.pointwise_scale(&x)?;
731d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
732d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
733d3677ae8SJeremy L Thompson     /// }
734c68be7a2SJeremy L Thompson     /// # Ok(())
735c68be7a2SJeremy L Thompson     /// # }
7369df49d7eSJed Brown     /// ```
7379df49d7eSJed Brown     #[allow(unused_mut)]
738*eb07d68fSJeremy L Thompson     pub fn pointwise_scale(mut self, x: &Vector) -> crate::Result<Self> {
739656ef1e5SJeremy L Thompson         self.check_error(unsafe { bind_ceed::CeedVectorPointwiseMult(self.ptr, self.ptr, x.ptr) })?;
7409df49d7eSJed Brown         Ok(self)
7419df49d7eSJed Brown     }
7429df49d7eSJed Brown 
7437ed177dbSJed Brown     /// Compute the pointwise multiplication w = w .* w for a Vector
7449df49d7eSJed Brown     ///
7459df49d7eSJed Brown     /// ```
746*eb07d68fSJeremy L Thompson     /// # use libceed::{prelude::*, Scalar};
7474d27c890SJeremy L Thompson     /// # fn main() -> libceed::Result<()> {
7489df49d7eSJed Brown     /// # let ceed = libceed::Ceed::default_init();
749c68be7a2SJeremy L Thompson     /// let mut w = ceed.vector_from_slice(&[0., 1., 2., 3., 4.])?;
7509df49d7eSJed Brown     ///
751c68be7a2SJeremy L Thompson     /// w = w.pointwise_square()?;
752d3677ae8SJeremy L Thompson     /// for (i, w) in w.view()?.iter().enumerate() {
753d3677ae8SJeremy L Thompson     ///     assert_eq!(*w, (i as Scalar).powf(2.0), "Value not set correctly");
754d3677ae8SJeremy L Thompson     /// }
755c68be7a2SJeremy L Thompson     /// # Ok(())
756c68be7a2SJeremy L Thompson     /// # }
7579df49d7eSJed Brown     /// ```
7589df49d7eSJed Brown     #[allow(unused_mut)]
7599df49d7eSJed Brown     pub fn pointwise_square(mut self) -> crate::Result<Self> {
760656ef1e5SJeremy L Thompson         self.check_error(unsafe {
761656ef1e5SJeremy L Thompson             bind_ceed::CeedVectorPointwiseMult(self.ptr, self.ptr, self.ptr)
762656ef1e5SJeremy L Thompson         })?;
7639df49d7eSJed Brown         Ok(self)
7649df49d7eSJed Brown     }
7659df49d7eSJed Brown }
7669df49d7eSJed Brown 
7679df49d7eSJed Brown // -----------------------------------------------------------------------------
7689df49d7eSJed Brown // Vector Viewer
7699df49d7eSJed Brown // -----------------------------------------------------------------------------
7709df49d7eSJed Brown /// A (host) view of a Vector with Deref to slice.  We can't make
7719df49d7eSJed Brown /// Vector itself Deref to slice because we can't handle the drop to
7729df49d7eSJed Brown /// call bind_ceed::CeedVectorRestoreArrayRead().
7739df49d7eSJed Brown #[derive(Debug)]
7749df49d7eSJed Brown pub struct VectorView<'a> {
7759df49d7eSJed Brown     vec: &'a Vector<'a>,
77680a9ef05SNatalie Beams     array: *const crate::Scalar,
7779df49d7eSJed Brown }
7789df49d7eSJed Brown 
7799df49d7eSJed Brown impl<'a> VectorView<'a> {
7809df49d7eSJed Brown     /// Construct a VectorView from a Vector reference
7814d27c890SJeremy L Thompson     fn new(vec: &'a Vector) -> crate::Result<Self> {
7829df49d7eSJed Brown         let mut array = std::ptr::null();
783656ef1e5SJeremy L Thompson         vec.check_error(unsafe {
7849df49d7eSJed Brown             bind_ceed::CeedVectorGetArrayRead(
7859df49d7eSJed Brown                 vec.ptr,
7869df49d7eSJed Brown                 crate::MemType::Host as bind_ceed::CeedMemType,
7879df49d7eSJed Brown                 &mut array,
788e78171edSJeremy L Thompson             )
789656ef1e5SJeremy L Thompson         })?;
790e78171edSJeremy L Thompson         Ok(Self {
7919df49d7eSJed Brown             vec: vec,
7929df49d7eSJed Brown             array: array,
793e78171edSJeremy L Thompson         })
7949df49d7eSJed Brown     }
7959df49d7eSJed Brown }
7969df49d7eSJed Brown 
7979df49d7eSJed Brown // Destructor
7989df49d7eSJed Brown impl<'a> Drop for VectorView<'a> {
7999df49d7eSJed Brown     fn drop(&mut self) {
8009df49d7eSJed Brown         unsafe {
8019df49d7eSJed Brown             bind_ceed::CeedVectorRestoreArrayRead(self.vec.ptr, &mut self.array);
8029df49d7eSJed Brown         }
8039df49d7eSJed Brown     }
8049df49d7eSJed Brown }
8059df49d7eSJed Brown 
8069df49d7eSJed Brown // Data access
8079df49d7eSJed Brown impl<'a> Deref for VectorView<'a> {
80880a9ef05SNatalie Beams     type Target = [crate::Scalar];
80980a9ef05SNatalie Beams     fn deref(&self) -> &[crate::Scalar] {
8109df49d7eSJed Brown         unsafe { std::slice::from_raw_parts(self.array, self.vec.len()) }
8119df49d7eSJed Brown     }
8129df49d7eSJed Brown }
8139df49d7eSJed Brown 
8149df49d7eSJed Brown // Viewing
8159df49d7eSJed Brown impl<'a> fmt::Display for VectorView<'a> {
8169df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
8179df49d7eSJed Brown         write!(f, "VectorView({:?})", self.deref())
8189df49d7eSJed Brown     }
8199df49d7eSJed Brown }
8209df49d7eSJed Brown 
8219df49d7eSJed Brown // -----------------------------------------------------------------------------
8229df49d7eSJed Brown // Vector Viewer Mutable
8239df49d7eSJed Brown // -----------------------------------------------------------------------------
8249df49d7eSJed Brown /// A mutable (host) view of a Vector with Deref to slice.
8259df49d7eSJed Brown #[derive(Debug)]
8269df49d7eSJed Brown pub struct VectorViewMut<'a> {
8279df49d7eSJed Brown     vec: &'a Vector<'a>,
82880a9ef05SNatalie Beams     array: *mut crate::Scalar,
8299df49d7eSJed Brown }
8309df49d7eSJed Brown 
8319df49d7eSJed Brown impl<'a> VectorViewMut<'a> {
8329df49d7eSJed Brown     /// Construct a VectorViewMut from a Vector reference
8334d27c890SJeremy L Thompson     fn new(vec: &'a mut Vector) -> crate::Result<Self> {
8349df49d7eSJed Brown         let mut ptr = std::ptr::null_mut();
835656ef1e5SJeremy L Thompson         vec.check_error(unsafe {
8369df49d7eSJed Brown             bind_ceed::CeedVectorGetArray(
8379df49d7eSJed Brown                 vec.ptr,
8389df49d7eSJed Brown                 crate::MemType::Host as bind_ceed::CeedMemType,
8399df49d7eSJed Brown                 &mut ptr,
840e78171edSJeremy L Thompson             )
841656ef1e5SJeremy L Thompson         })?;
842e78171edSJeremy L Thompson         Ok(Self {
8439df49d7eSJed Brown             vec: vec,
8449df49d7eSJed Brown             array: ptr,
845e78171edSJeremy L Thompson         })
8469df49d7eSJed Brown     }
8479df49d7eSJed Brown }
8489df49d7eSJed Brown 
8499df49d7eSJed Brown // Destructor
8509df49d7eSJed Brown impl<'a> Drop for VectorViewMut<'a> {
8519df49d7eSJed Brown     fn drop(&mut self) {
8529df49d7eSJed Brown         unsafe {
8539df49d7eSJed Brown             bind_ceed::CeedVectorRestoreArray(self.vec.ptr, &mut self.array);
8549df49d7eSJed Brown         }
8559df49d7eSJed Brown     }
8569df49d7eSJed Brown }
8579df49d7eSJed Brown 
8589df49d7eSJed Brown // Data access
8599df49d7eSJed Brown impl<'a> Deref for VectorViewMut<'a> {
86080a9ef05SNatalie Beams     type Target = [crate::Scalar];
86180a9ef05SNatalie Beams     fn deref(&self) -> &[crate::Scalar] {
8629df49d7eSJed Brown         unsafe { std::slice::from_raw_parts(self.array, self.vec.len()) }
8639df49d7eSJed Brown     }
8649df49d7eSJed Brown }
8659df49d7eSJed Brown 
8669df49d7eSJed Brown // Mutable data access
8679df49d7eSJed Brown impl<'a> DerefMut for VectorViewMut<'a> {
86880a9ef05SNatalie Beams     fn deref_mut(&mut self) -> &mut [crate::Scalar] {
8699df49d7eSJed Brown         unsafe { std::slice::from_raw_parts_mut(self.array, self.vec.len()) }
8709df49d7eSJed Brown     }
8719df49d7eSJed Brown }
8729df49d7eSJed Brown 
8739df49d7eSJed Brown // Viewing
8749df49d7eSJed Brown impl<'a> fmt::Display for VectorViewMut<'a> {
8759df49d7eSJed Brown     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
8769df49d7eSJed Brown         write!(f, "VectorViewMut({:?})", self.deref())
8779df49d7eSJed Brown     }
8789df49d7eSJed Brown }
8799df49d7eSJed Brown 
8809df49d7eSJed Brown // -----------------------------------------------------------------------------
881