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