1*d275d636SJeremy 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 /// ``` 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 ); 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 { 1692b671a0aSJeremy L Thompson vector: unsafe { crate::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 /// ``` 27546de3253SJeremy L Thompson /// # use libceed::prelude::*; 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 /// ``` 28946de3253SJeremy L Thompson pub fn copy_from(&mut self, vec_source: &crate::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(); 303c68be7a2SJeremy L Thompson /// let vec = vector::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 /// ``` 4219df49d7eSJed Brown /// # use libceed::prelude::*; 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 /// ``` 45656e67e05SJeremy L Thompson /// # use libceed::prelude::*; 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>> { 49835afbe75SJeremy L Thompson crate::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 /// ``` 5089df49d7eSJed Brown /// # use libceed::prelude::*; 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 /// ``` 5809df49d7eSJed Brown /// # use libceed::prelude::*; 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 /// ``` 6119df49d7eSJed Brown /// # use libceed::prelude::*; 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 /// ``` 6379df49d7eSJed Brown /// # use libceed::prelude::*; 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)] 65180a9ef05SNatalie Beams pub fn axpy(mut self, alpha: crate::Scalar, x: &crate::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 /// ``` 6655fb68f37SKaren (Ren) Stengel /// # use libceed::prelude::*; 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, 6835fb68f37SKaren (Ren) Stengel x: &crate::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 /// ``` 6979df49d7eSJed Brown /// # use libceed::prelude::*; 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)] 7129df49d7eSJed Brown pub fn pointwise_mult(mut self, x: &crate::Vector, y: &crate::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 /// ``` 7249df49d7eSJed Brown /// # use libceed::prelude::*; 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)] 7389df49d7eSJed Brown pub fn pointwise_scale(mut self, x: &crate::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 /// ``` 7469df49d7eSJed Brown /// # use libceed::prelude::*; 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