19df49d7eSJed Brown // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at 29df49d7eSJed Brown // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights 39df49d7eSJed Brown // reserved. See files LICENSE and NOTICE for details. 49df49d7eSJed Brown // 59df49d7eSJed Brown // This file is part of CEED, a collection of benchmarks, miniapps, software 69df49d7eSJed Brown // libraries and APIs for efficient high-order finite element and spectral 79df49d7eSJed Brown // element discretizations for exascale applications. For more information and 89df49d7eSJed Brown // source code availability see http://github.com/ceed. 99df49d7eSJed Brown // 109df49d7eSJed Brown // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 119df49d7eSJed Brown // a collaborative effort of two U.S. Department of Energy organizations (Office 129df49d7eSJed Brown // of Science and the National Nuclear Security Administration) responsible for 139df49d7eSJed Brown // the planning and preparation of a capable exascale ecosystem, including 149df49d7eSJed Brown // software, applications, hardware, advanced system engineering and early 159df49d7eSJed Brown // testbed platforms, in support of the nation's exascale computing imperative 169df49d7eSJed Brown 179df49d7eSJed Brown //! A Ceed Operator defines the finite/spectral element operator associated to a 189df49d7eSJed Brown //! Ceed QFunction. A Ceed Operator connects Ceed ElemRestrictions, 199df49d7eSJed Brown //! Ceed Bases, and Ceed QFunctions. 209df49d7eSJed Brown 219df49d7eSJed Brown use crate::prelude::*; 229df49d7eSJed Brown 239df49d7eSJed Brown // ----------------------------------------------------------------------------- 24*7ed177dbSJed Brown // Operator Field context wrapper 2508778c6fSJeremy L Thompson // ----------------------------------------------------------------------------- 2608778c6fSJeremy L Thompson #[derive(Debug)] 2708778c6fSJeremy L Thompson pub struct OperatorField<'a> { 2808778c6fSJeremy L Thompson pub(crate) ptr: bind_ceed::CeedOperatorField, 2908778c6fSJeremy L Thompson _lifeline: PhantomData<&'a ()>, 3008778c6fSJeremy L Thompson } 3108778c6fSJeremy L Thompson 3208778c6fSJeremy L Thompson // ----------------------------------------------------------------------------- 3308778c6fSJeremy L Thompson // Implementations 3408778c6fSJeremy L Thompson // ----------------------------------------------------------------------------- 3508778c6fSJeremy L Thompson impl<'a> OperatorField<'a> { 3608778c6fSJeremy L Thompson /// Get the name of an OperatorField 3708778c6fSJeremy L Thompson /// 3808778c6fSJeremy L Thompson /// ``` 3908778c6fSJeremy L Thompson /// # use libceed::prelude::*; 404d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 4108778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 4208778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 4308778c6fSJeremy L Thompson /// 4408778c6fSJeremy L Thompson /// // Operator field arguments 4508778c6fSJeremy L Thompson /// let ne = 3; 4608778c6fSJeremy L Thompson /// let q = 4 as usize; 4708778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 4808778c6fSJeremy L Thompson /// for i in 0..ne { 4908778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 5008778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 5108778c6fSJeremy L Thompson /// } 5208778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 5308778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 5408778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 5508778c6fSJeremy L Thompson /// 5608778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 5708778c6fSJeremy L Thompson /// 5808778c6fSJeremy L Thompson /// // Operator fields 5908778c6fSJeremy L Thompson /// let op = ceed 6008778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 6108778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 6208778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 6308778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 6408778c6fSJeremy L Thompson /// 6508778c6fSJeremy L Thompson /// let inputs = op.inputs()?; 6608778c6fSJeremy L Thompson /// 6708778c6fSJeremy L Thompson /// assert_eq!(inputs[0].name(), "dx", "Incorrect input name"); 6808778c6fSJeremy L Thompson /// assert_eq!(inputs[1].name(), "weights", "Incorrect input name"); 6908778c6fSJeremy L Thompson /// # Ok(()) 7008778c6fSJeremy L Thompson /// # } 7108778c6fSJeremy L Thompson /// ``` 7208778c6fSJeremy L Thompson pub fn name(&self) -> &str { 7308778c6fSJeremy L Thompson let mut name_ptr: *mut std::os::raw::c_char = std::ptr::null_mut(); 7408778c6fSJeremy L Thompson unsafe { 7508778c6fSJeremy L Thompson bind_ceed::CeedOperatorFieldGetName(self.ptr, &mut name_ptr); 7608778c6fSJeremy L Thompson } 7708778c6fSJeremy L Thompson unsafe { CStr::from_ptr(name_ptr) }.to_str().unwrap() 7808778c6fSJeremy L Thompson } 7908778c6fSJeremy L Thompson 8008778c6fSJeremy L Thompson /// Get the ElemRestriction of an OperatorField 8108778c6fSJeremy L Thompson /// 8208778c6fSJeremy L Thompson /// ``` 8308778c6fSJeremy L Thompson /// # use libceed::prelude::*; 844d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 8508778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 8608778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 8708778c6fSJeremy L Thompson /// 8808778c6fSJeremy L Thompson /// // Operator field arguments 8908778c6fSJeremy L Thompson /// let ne = 3; 9008778c6fSJeremy L Thompson /// let q = 4 as usize; 9108778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 9208778c6fSJeremy L Thompson /// for i in 0..ne { 9308778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 9408778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 9508778c6fSJeremy L Thompson /// } 9608778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 9708778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 9808778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 9908778c6fSJeremy L Thompson /// 10008778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 10108778c6fSJeremy L Thompson /// 10208778c6fSJeremy L Thompson /// // Operator fields 10308778c6fSJeremy L Thompson /// let op = ceed 10408778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 10508778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 10608778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 10708778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 10808778c6fSJeremy L Thompson /// 10908778c6fSJeremy L Thompson /// let inputs = op.inputs()?; 11008778c6fSJeremy L Thompson /// 11108778c6fSJeremy L Thompson /// assert!( 11208778c6fSJeremy L Thompson /// inputs[0].elem_restriction().is_some(), 11308778c6fSJeremy L Thompson /// "Incorrect field ElemRestriction" 11408778c6fSJeremy L Thompson /// ); 11508778c6fSJeremy L Thompson /// if let ElemRestrictionOpt::Some(r) = inputs[0].elem_restriction() { 11608778c6fSJeremy L Thompson /// assert_eq!( 11708778c6fSJeremy L Thompson /// r.num_elements(), 11808778c6fSJeremy L Thompson /// ne, 11908778c6fSJeremy L Thompson /// "Incorrect field ElemRestriction number of elements" 12008778c6fSJeremy L Thompson /// ); 12108778c6fSJeremy L Thompson /// } 12208778c6fSJeremy L Thompson /// 12308778c6fSJeremy L Thompson /// assert!( 12408778c6fSJeremy L Thompson /// inputs[1].elem_restriction().is_none(), 12508778c6fSJeremy L Thompson /// "Incorrect field ElemRestriction" 12608778c6fSJeremy L Thompson /// ); 12708778c6fSJeremy L Thompson /// # Ok(()) 12808778c6fSJeremy L Thompson /// # } 12908778c6fSJeremy L Thompson /// ``` 13008778c6fSJeremy L Thompson pub fn elem_restriction(&self) -> ElemRestrictionOpt { 13108778c6fSJeremy L Thompson let mut ptr = std::ptr::null_mut(); 13208778c6fSJeremy L Thompson unsafe { 13308778c6fSJeremy L Thompson bind_ceed::CeedOperatorFieldGetElemRestriction(self.ptr, &mut ptr); 13408778c6fSJeremy L Thompson } 13508778c6fSJeremy L Thompson if ptr == unsafe { bind_ceed::CEED_ELEMRESTRICTION_NONE } { 13608778c6fSJeremy L Thompson ElemRestrictionOpt::None 13708778c6fSJeremy L Thompson } else { 13808778c6fSJeremy L Thompson let slice = unsafe { 13908778c6fSJeremy L Thompson std::slice::from_raw_parts( 14008778c6fSJeremy L Thompson &ptr as *const bind_ceed::CeedElemRestriction as *const crate::ElemRestriction, 14108778c6fSJeremy L Thompson 1 as usize, 14208778c6fSJeremy L Thompson ) 14308778c6fSJeremy L Thompson }; 14408778c6fSJeremy L Thompson ElemRestrictionOpt::Some(&slice[0]) 14508778c6fSJeremy L Thompson } 14608778c6fSJeremy L Thompson } 14708778c6fSJeremy L Thompson 14808778c6fSJeremy L Thompson /// Get the Basis of an OperatorField 14908778c6fSJeremy L Thompson /// 15008778c6fSJeremy L Thompson /// ``` 15108778c6fSJeremy L Thompson /// # use libceed::prelude::*; 1524d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 15308778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 15408778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 15508778c6fSJeremy L Thompson /// 15608778c6fSJeremy L Thompson /// // Operator field arguments 15708778c6fSJeremy L Thompson /// let ne = 3; 15808778c6fSJeremy L Thompson /// let q = 4 as usize; 15908778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 16008778c6fSJeremy L Thompson /// for i in 0..ne { 16108778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 16208778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 16308778c6fSJeremy L Thompson /// } 16408778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 16508778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 16608778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 16708778c6fSJeremy L Thompson /// 16808778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 16908778c6fSJeremy L Thompson /// 17008778c6fSJeremy L Thompson /// // Operator fields 17108778c6fSJeremy L Thompson /// let op = ceed 17208778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 17308778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 17408778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 17508778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 17608778c6fSJeremy L Thompson /// 17708778c6fSJeremy L Thompson /// let inputs = op.inputs()?; 17808778c6fSJeremy L Thompson /// 17908778c6fSJeremy L Thompson /// assert!(inputs[0].basis().is_some(), "Incorrect field Basis"); 18008778c6fSJeremy L Thompson /// if let BasisOpt::Some(b) = inputs[0].basis() { 18108778c6fSJeremy L Thompson /// assert_eq!( 18208778c6fSJeremy L Thompson /// b.num_quadrature_points(), 18308778c6fSJeremy L Thompson /// q, 18408778c6fSJeremy L Thompson /// "Incorrect field Basis number of quadrature points" 18508778c6fSJeremy L Thompson /// ); 18608778c6fSJeremy L Thompson /// } 18708778c6fSJeremy L Thompson /// assert!(inputs[1].basis().is_some(), "Incorrect field Basis"); 18808778c6fSJeremy L Thompson /// if let BasisOpt::Some(b) = inputs[0].basis() { 18908778c6fSJeremy L Thompson /// assert_eq!( 19008778c6fSJeremy L Thompson /// b.num_quadrature_points(), 19108778c6fSJeremy L Thompson /// q, 19208778c6fSJeremy L Thompson /// "Incorrect field Basis number of quadrature points" 19308778c6fSJeremy L Thompson /// ); 19408778c6fSJeremy L Thompson /// } 19508778c6fSJeremy L Thompson /// 19608778c6fSJeremy L Thompson /// let outputs = op.outputs()?; 19708778c6fSJeremy L Thompson /// 19808778c6fSJeremy L Thompson /// assert!(outputs[0].basis().is_collocated(), "Incorrect field Basis"); 19908778c6fSJeremy L Thompson /// # Ok(()) 20008778c6fSJeremy L Thompson /// # } 20108778c6fSJeremy L Thompson /// ``` 20208778c6fSJeremy L Thompson pub fn basis(&self) -> BasisOpt { 20308778c6fSJeremy L Thompson let mut ptr = std::ptr::null_mut(); 20408778c6fSJeremy L Thompson unsafe { 20508778c6fSJeremy L Thompson bind_ceed::CeedOperatorFieldGetBasis(self.ptr, &mut ptr); 20608778c6fSJeremy L Thompson } 20708778c6fSJeremy L Thompson if ptr == unsafe { bind_ceed::CEED_BASIS_COLLOCATED } { 20808778c6fSJeremy L Thompson BasisOpt::Collocated 20908778c6fSJeremy L Thompson } else { 21008778c6fSJeremy L Thompson let slice = unsafe { 21108778c6fSJeremy L Thompson std::slice::from_raw_parts( 21208778c6fSJeremy L Thompson &ptr as *const bind_ceed::CeedBasis as *const crate::Basis, 21308778c6fSJeremy L Thompson 1 as usize, 21408778c6fSJeremy L Thompson ) 21508778c6fSJeremy L Thompson }; 21608778c6fSJeremy L Thompson BasisOpt::Some(&slice[0]) 21708778c6fSJeremy L Thompson } 21808778c6fSJeremy L Thompson } 21908778c6fSJeremy L Thompson 22008778c6fSJeremy L Thompson /// Get the Vector of an OperatorField 22108778c6fSJeremy L Thompson /// 22208778c6fSJeremy L Thompson /// ``` 22308778c6fSJeremy L Thompson /// # use libceed::prelude::*; 2244d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 22508778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 22608778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 22708778c6fSJeremy L Thompson /// 22808778c6fSJeremy L Thompson /// // Operator field arguments 22908778c6fSJeremy L Thompson /// let ne = 3; 23008778c6fSJeremy L Thompson /// let q = 4 as usize; 23108778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 23208778c6fSJeremy L Thompson /// for i in 0..ne { 23308778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 23408778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 23508778c6fSJeremy L Thompson /// } 23608778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 23708778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 23808778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 23908778c6fSJeremy L Thompson /// 24008778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 24108778c6fSJeremy L Thompson /// 24208778c6fSJeremy L Thompson /// // Operator fields 24308778c6fSJeremy L Thompson /// let op = ceed 24408778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 24508778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 24608778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 24708778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 24808778c6fSJeremy L Thompson /// 24908778c6fSJeremy L Thompson /// let inputs = op.inputs()?; 25008778c6fSJeremy L Thompson /// 25108778c6fSJeremy L Thompson /// assert!(inputs[0].vector().is_active(), "Incorrect field Vector"); 25208778c6fSJeremy L Thompson /// assert!(inputs[1].vector().is_none(), "Incorrect field Vector"); 25308778c6fSJeremy L Thompson /// # Ok(()) 25408778c6fSJeremy L Thompson /// # } 25508778c6fSJeremy L Thompson /// ``` 25608778c6fSJeremy L Thompson pub fn vector(&self) -> VectorOpt { 25708778c6fSJeremy L Thompson let mut ptr = std::ptr::null_mut(); 25808778c6fSJeremy L Thompson unsafe { 25908778c6fSJeremy L Thompson bind_ceed::CeedOperatorFieldGetVector(self.ptr, &mut ptr); 26008778c6fSJeremy L Thompson } 26108778c6fSJeremy L Thompson if ptr == unsafe { bind_ceed::CEED_VECTOR_ACTIVE } { 26208778c6fSJeremy L Thompson VectorOpt::Active 26308778c6fSJeremy L Thompson } else if ptr == unsafe { bind_ceed::CEED_VECTOR_NONE } { 26408778c6fSJeremy L Thompson VectorOpt::None 26508778c6fSJeremy L Thompson } else { 26608778c6fSJeremy L Thompson let slice = unsafe { 26708778c6fSJeremy L Thompson std::slice::from_raw_parts( 26808778c6fSJeremy L Thompson &ptr as *const bind_ceed::CeedVector as *const crate::Vector, 26908778c6fSJeremy L Thompson 1 as usize, 27008778c6fSJeremy L Thompson ) 27108778c6fSJeremy L Thompson }; 27208778c6fSJeremy L Thompson VectorOpt::Some(&slice[0]) 27308778c6fSJeremy L Thompson } 27408778c6fSJeremy L Thompson } 27508778c6fSJeremy L Thompson } 27608778c6fSJeremy L Thompson 27708778c6fSJeremy L Thompson // ----------------------------------------------------------------------------- 278*7ed177dbSJed Brown // Operator context wrapper 2799df49d7eSJed Brown // ----------------------------------------------------------------------------- 280c68be7a2SJeremy L Thompson #[derive(Debug)] 2819df49d7eSJed Brown pub(crate) struct OperatorCore<'a> { 2829df49d7eSJed Brown ptr: bind_ceed::CeedOperator, 2831142270cSJeremy L Thompson _lifeline: PhantomData<&'a ()>, 2849df49d7eSJed Brown } 2859df49d7eSJed Brown 286c68be7a2SJeremy L Thompson #[derive(Debug)] 2879df49d7eSJed Brown pub struct Operator<'a> { 2889df49d7eSJed Brown op_core: OperatorCore<'a>, 2899df49d7eSJed Brown } 2909df49d7eSJed Brown 291c68be7a2SJeremy L Thompson #[derive(Debug)] 2929df49d7eSJed Brown pub struct CompositeOperator<'a> { 2939df49d7eSJed Brown op_core: OperatorCore<'a>, 2949df49d7eSJed Brown } 2959df49d7eSJed Brown 2969df49d7eSJed Brown // ----------------------------------------------------------------------------- 2979df49d7eSJed Brown // Destructor 2989df49d7eSJed Brown // ----------------------------------------------------------------------------- 2999df49d7eSJed Brown impl<'a> Drop for OperatorCore<'a> { 3009df49d7eSJed Brown fn drop(&mut self) { 3019df49d7eSJed Brown unsafe { 3029df49d7eSJed Brown bind_ceed::CeedOperatorDestroy(&mut self.ptr); 3039df49d7eSJed Brown } 3049df49d7eSJed Brown } 3059df49d7eSJed Brown } 3069df49d7eSJed Brown 3079df49d7eSJed Brown // ----------------------------------------------------------------------------- 3089df49d7eSJed Brown // Display 3099df49d7eSJed Brown // ----------------------------------------------------------------------------- 3109df49d7eSJed Brown impl<'a> fmt::Display for OperatorCore<'a> { 3119df49d7eSJed Brown fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { 3129df49d7eSJed Brown let mut ptr = std::ptr::null_mut(); 3139df49d7eSJed Brown let mut sizeloc = crate::MAX_BUFFER_LENGTH; 3149df49d7eSJed Brown let cstring = unsafe { 3159df49d7eSJed Brown let file = bind_ceed::open_memstream(&mut ptr, &mut sizeloc); 3169df49d7eSJed Brown bind_ceed::CeedOperatorView(self.ptr, file); 3179df49d7eSJed Brown bind_ceed::fclose(file); 3189df49d7eSJed Brown CString::from_raw(ptr) 3199df49d7eSJed Brown }; 3209df49d7eSJed Brown cstring.to_string_lossy().fmt(f) 3219df49d7eSJed Brown } 3229df49d7eSJed Brown } 3239df49d7eSJed Brown 3249df49d7eSJed Brown /// View an Operator 3259df49d7eSJed Brown /// 3269df49d7eSJed Brown /// ``` 3279df49d7eSJed Brown /// # use libceed::prelude::*; 3284d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 3299df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 330c68be7a2SJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 3319df49d7eSJed Brown /// 3329df49d7eSJed Brown /// // Operator field arguments 3339df49d7eSJed Brown /// let ne = 3; 3349df49d7eSJed Brown /// let q = 4 as usize; 3359df49d7eSJed Brown /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 3369df49d7eSJed Brown /// for i in 0..ne { 3379df49d7eSJed Brown /// ind[2 * i + 0] = i as i32; 3389df49d7eSJed Brown /// ind[2 * i + 1] = (i + 1) as i32; 3399df49d7eSJed Brown /// } 340c68be7a2SJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 3419df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 342c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 3439df49d7eSJed Brown /// 344c68be7a2SJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 3459df49d7eSJed Brown /// 3469df49d7eSJed Brown /// // Operator fields 3479df49d7eSJed Brown /// let op = ceed 348c68be7a2SJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 349c68be7a2SJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 350c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 351c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 3529df49d7eSJed Brown /// 3539df49d7eSJed Brown /// println!("{}", op); 354c68be7a2SJeremy L Thompson /// # Ok(()) 355c68be7a2SJeremy L Thompson /// # } 3569df49d7eSJed Brown /// ``` 3579df49d7eSJed Brown impl<'a> fmt::Display for Operator<'a> { 3589df49d7eSJed Brown fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { 3599df49d7eSJed Brown self.op_core.fmt(f) 3609df49d7eSJed Brown } 3619df49d7eSJed Brown } 3629df49d7eSJed Brown 3639df49d7eSJed Brown /// View a composite Operator 3649df49d7eSJed Brown /// 3659df49d7eSJed Brown /// ``` 3669df49d7eSJed Brown /// # use libceed::prelude::*; 3674d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 3689df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 3699df49d7eSJed Brown /// 3709df49d7eSJed Brown /// // Sub operator field arguments 3719df49d7eSJed Brown /// let ne = 3; 3729df49d7eSJed Brown /// let q = 4 as usize; 3739df49d7eSJed Brown /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 3749df49d7eSJed Brown /// for i in 0..ne { 3759df49d7eSJed Brown /// ind[2 * i + 0] = i as i32; 3769df49d7eSJed Brown /// ind[2 * i + 1] = (i + 1) as i32; 3779df49d7eSJed Brown /// } 378c68be7a2SJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 3799df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 380c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 3819df49d7eSJed Brown /// 382c68be7a2SJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 3839df49d7eSJed Brown /// 384c68be7a2SJeremy L Thompson /// let qdata_mass = ceed.vector(q * ne)?; 385c68be7a2SJeremy L Thompson /// let qdata_diff = ceed.vector(q * ne)?; 3869df49d7eSJed Brown /// 387c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 3889df49d7eSJed Brown /// let op_mass = ceed 389c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 390c68be7a2SJeremy L Thompson /// .field("u", &r, &b, VectorOpt::Active)? 391c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_mass)? 392c68be7a2SJeremy L Thompson /// .field("v", &r, &b, VectorOpt::Active)?; 3939df49d7eSJed Brown /// 394c68be7a2SJeremy L Thompson /// let qf_diff = ceed.q_function_interior_by_name("Poisson1DApply")?; 3959df49d7eSJed Brown /// let op_diff = ceed 396c68be7a2SJeremy L Thompson /// .operator(&qf_diff, QFunctionOpt::None, QFunctionOpt::None)? 397c68be7a2SJeremy L Thompson /// .field("du", &r, &b, VectorOpt::Active)? 398c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_diff)? 399c68be7a2SJeremy L Thompson /// .field("dv", &r, &b, VectorOpt::Active)?; 4009df49d7eSJed Brown /// 4019df49d7eSJed Brown /// let op = ceed 402c68be7a2SJeremy L Thompson /// .composite_operator()? 403c68be7a2SJeremy L Thompson /// .sub_operator(&op_mass)? 404c68be7a2SJeremy L Thompson /// .sub_operator(&op_diff)?; 4059df49d7eSJed Brown /// 4069df49d7eSJed Brown /// println!("{}", op); 407c68be7a2SJeremy L Thompson /// # Ok(()) 408c68be7a2SJeremy L Thompson /// # } 4099df49d7eSJed Brown /// ``` 4109df49d7eSJed Brown impl<'a> fmt::Display for CompositeOperator<'a> { 4119df49d7eSJed Brown fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { 4129df49d7eSJed Brown self.op_core.fmt(f) 4139df49d7eSJed Brown } 4149df49d7eSJed Brown } 4159df49d7eSJed Brown 4169df49d7eSJed Brown // ----------------------------------------------------------------------------- 4179df49d7eSJed Brown // Core functionality 4189df49d7eSJed Brown // ----------------------------------------------------------------------------- 4199df49d7eSJed Brown impl<'a> OperatorCore<'a> { 4201142270cSJeremy L Thompson // Error handling 4211142270cSJeremy L Thompson #[doc(hidden)] 4221142270cSJeremy L Thompson fn check_error(&self, ierr: i32) -> crate::Result<i32> { 4231142270cSJeremy L Thompson let mut ptr = std::ptr::null_mut(); 4241142270cSJeremy L Thompson unsafe { 4251142270cSJeremy L Thompson bind_ceed::CeedOperatorGetCeed(self.ptr, &mut ptr); 4261142270cSJeremy L Thompson } 4271142270cSJeremy L Thompson crate::check_error(ptr, ierr) 4281142270cSJeremy L Thompson } 4291142270cSJeremy L Thompson 4309df49d7eSJed Brown // Common implementations 4316f97ff0aSJeremy L Thompson pub fn check(&self) -> crate::Result<i32> { 4326f97ff0aSJeremy L Thompson let ierr = unsafe { bind_ceed::CeedOperatorCheckReady(self.ptr) }; 4336f97ff0aSJeremy L Thompson self.check_error(ierr) 4346f97ff0aSJeremy L Thompson } 4356f97ff0aSJeremy L Thompson 4369df49d7eSJed Brown pub fn apply(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 4379df49d7eSJed Brown let ierr = unsafe { 4389df49d7eSJed Brown bind_ceed::CeedOperatorApply( 4399df49d7eSJed Brown self.ptr, 4409df49d7eSJed Brown input.ptr, 4419df49d7eSJed Brown output.ptr, 4429df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 4439df49d7eSJed Brown ) 4449df49d7eSJed Brown }; 4451142270cSJeremy L Thompson self.check_error(ierr) 4469df49d7eSJed Brown } 4479df49d7eSJed Brown 4489df49d7eSJed Brown pub fn apply_add(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 4499df49d7eSJed Brown let ierr = unsafe { 4509df49d7eSJed Brown bind_ceed::CeedOperatorApplyAdd( 4519df49d7eSJed Brown self.ptr, 4529df49d7eSJed Brown input.ptr, 4539df49d7eSJed Brown output.ptr, 4549df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 4559df49d7eSJed Brown ) 4569df49d7eSJed Brown }; 4571142270cSJeremy L Thompson self.check_error(ierr) 4589df49d7eSJed Brown } 4599df49d7eSJed Brown 4609df49d7eSJed Brown pub fn linear_assemble_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 4619df49d7eSJed Brown let ierr = unsafe { 4629df49d7eSJed Brown bind_ceed::CeedOperatorLinearAssembleDiagonal( 4639df49d7eSJed Brown self.ptr, 4649df49d7eSJed Brown assembled.ptr, 4659df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 4669df49d7eSJed Brown ) 4679df49d7eSJed Brown }; 4681142270cSJeremy L Thompson self.check_error(ierr) 4699df49d7eSJed Brown } 4709df49d7eSJed Brown 4719df49d7eSJed Brown pub fn linear_assemble_add_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 4729df49d7eSJed Brown let ierr = unsafe { 4739df49d7eSJed Brown bind_ceed::CeedOperatorLinearAssembleAddDiagonal( 4749df49d7eSJed Brown self.ptr, 4759df49d7eSJed Brown assembled.ptr, 4769df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 4779df49d7eSJed Brown ) 4789df49d7eSJed Brown }; 4791142270cSJeremy L Thompson self.check_error(ierr) 4809df49d7eSJed Brown } 4819df49d7eSJed Brown 4829df49d7eSJed Brown pub fn linear_assemble_point_block_diagonal( 4839df49d7eSJed Brown &self, 4849df49d7eSJed Brown assembled: &mut Vector, 4859df49d7eSJed Brown ) -> crate::Result<i32> { 4869df49d7eSJed Brown let ierr = unsafe { 4879df49d7eSJed Brown bind_ceed::CeedOperatorLinearAssemblePointBlockDiagonal( 4889df49d7eSJed Brown self.ptr, 4899df49d7eSJed Brown assembled.ptr, 4909df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 4919df49d7eSJed Brown ) 4929df49d7eSJed Brown }; 4931142270cSJeremy L Thompson self.check_error(ierr) 4949df49d7eSJed Brown } 4959df49d7eSJed Brown 4969df49d7eSJed Brown pub fn linear_assemble_add_point_block_diagonal( 4979df49d7eSJed Brown &self, 4989df49d7eSJed Brown assembled: &mut Vector, 4999df49d7eSJed Brown ) -> crate::Result<i32> { 5009df49d7eSJed Brown let ierr = unsafe { 5019df49d7eSJed Brown bind_ceed::CeedOperatorLinearAssembleAddPointBlockDiagonal( 5029df49d7eSJed Brown self.ptr, 5039df49d7eSJed Brown assembled.ptr, 5049df49d7eSJed Brown bind_ceed::CEED_REQUEST_IMMEDIATE, 5059df49d7eSJed Brown ) 5069df49d7eSJed Brown }; 5071142270cSJeremy L Thompson self.check_error(ierr) 5089df49d7eSJed Brown } 5099df49d7eSJed Brown } 5109df49d7eSJed Brown 5119df49d7eSJed Brown // ----------------------------------------------------------------------------- 5129df49d7eSJed Brown // Operator 5139df49d7eSJed Brown // ----------------------------------------------------------------------------- 5149df49d7eSJed Brown impl<'a> Operator<'a> { 5159df49d7eSJed Brown // Constructor 5169df49d7eSJed Brown pub fn create<'b>( 517594ef120SJeremy L Thompson ceed: &crate::Ceed, 5189df49d7eSJed Brown qf: impl Into<QFunctionOpt<'b>>, 5199df49d7eSJed Brown dqf: impl Into<QFunctionOpt<'b>>, 5209df49d7eSJed Brown dqfT: impl Into<QFunctionOpt<'b>>, 5219df49d7eSJed Brown ) -> crate::Result<Self> { 5229df49d7eSJed Brown let mut ptr = std::ptr::null_mut(); 5239df49d7eSJed Brown let ierr = unsafe { 5249df49d7eSJed Brown bind_ceed::CeedOperatorCreate( 5259df49d7eSJed Brown ceed.ptr, 5269df49d7eSJed Brown qf.into().to_raw(), 5279df49d7eSJed Brown dqf.into().to_raw(), 5289df49d7eSJed Brown dqfT.into().to_raw(), 5299df49d7eSJed Brown &mut ptr, 5309df49d7eSJed Brown ) 5319df49d7eSJed Brown }; 5329df49d7eSJed Brown ceed.check_error(ierr)?; 5339df49d7eSJed Brown Ok(Self { 5341142270cSJeremy L Thompson op_core: OperatorCore { 5351142270cSJeremy L Thompson ptr, 5361142270cSJeremy L Thompson _lifeline: PhantomData, 5371142270cSJeremy L Thompson }, 5389df49d7eSJed Brown }) 5399df49d7eSJed Brown } 5409df49d7eSJed Brown 5411142270cSJeremy L Thompson fn from_raw(ptr: bind_ceed::CeedOperator) -> crate::Result<Self> { 5429df49d7eSJed Brown Ok(Self { 5431142270cSJeremy L Thompson op_core: OperatorCore { 5441142270cSJeremy L Thompson ptr, 5451142270cSJeremy L Thompson _lifeline: PhantomData, 5461142270cSJeremy L Thompson }, 5479df49d7eSJed Brown }) 5489df49d7eSJed Brown } 5499df49d7eSJed Brown 5509df49d7eSJed Brown /// Apply Operator to a vector 5519df49d7eSJed Brown /// 5529df49d7eSJed Brown /// * `input` - Input Vector 5539df49d7eSJed Brown /// * `output` - Output Vector 5549df49d7eSJed Brown /// 5559df49d7eSJed Brown /// ``` 5569df49d7eSJed Brown /// # use libceed::prelude::*; 5574d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 5589df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 5599df49d7eSJed Brown /// let ne = 4; 5609df49d7eSJed Brown /// let p = 3; 5619df49d7eSJed Brown /// let q = 4; 5629df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 5639df49d7eSJed Brown /// 5649df49d7eSJed Brown /// // Vectors 565c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 566c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 5679df49d7eSJed Brown /// qdata.set_value(0.0); 568c68be7a2SJeremy L Thompson /// let u = ceed.vector_from_slice(&vec![1.0; ndofs])?; 569c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 5709df49d7eSJed Brown /// v.set_value(0.0); 5719df49d7eSJed Brown /// 5729df49d7eSJed Brown /// // Restrictions 5739df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 5749df49d7eSJed Brown /// for i in 0..ne { 5759df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 5769df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 5779df49d7eSJed Brown /// } 578c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 5799df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 5809df49d7eSJed Brown /// for i in 0..ne { 5819df49d7eSJed Brown /// indu[p * i + 0] = i as i32; 5829df49d7eSJed Brown /// indu[p * i + 1] = (i + 1) as i32; 5839df49d7eSJed Brown /// indu[p * i + 2] = (i + 2) as i32; 5849df49d7eSJed Brown /// } 585c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, 3, 1, 1, ndofs, MemType::Host, &indu)?; 5869df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 587c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 5889df49d7eSJed Brown /// 5899df49d7eSJed Brown /// // Bases 590c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 591c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 5929df49d7eSJed Brown /// 5939df49d7eSJed Brown /// // Build quadrature data 594c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 595c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 596c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 597c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 598c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 599c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 6009df49d7eSJed Brown /// 6019df49d7eSJed Brown /// // Mass operator 602c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 6039df49d7eSJed Brown /// let op_mass = ceed 604c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 605c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 606c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 607c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 6089df49d7eSJed Brown /// 6099df49d7eSJed Brown /// v.set_value(0.0); 610c68be7a2SJeremy L Thompson /// op_mass.apply(&u, &mut v)?; 6119df49d7eSJed Brown /// 6129df49d7eSJed Brown /// // Check 613e78171edSJeremy L Thompson /// let sum: Scalar = v.view()?.iter().sum(); 6149df49d7eSJed Brown /// assert!( 61580a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 6169df49d7eSJed Brown /// "Incorrect interval length computed" 6179df49d7eSJed Brown /// ); 618c68be7a2SJeremy L Thompson /// # Ok(()) 619c68be7a2SJeremy L Thompson /// # } 6209df49d7eSJed Brown /// ``` 6219df49d7eSJed Brown pub fn apply(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 6229df49d7eSJed Brown self.op_core.apply(input, output) 6239df49d7eSJed Brown } 6249df49d7eSJed Brown 6259df49d7eSJed Brown /// Apply Operator to a vector and add result to output vector 6269df49d7eSJed Brown /// 6279df49d7eSJed Brown /// * `input` - Input Vector 6289df49d7eSJed Brown /// * `output` - Output Vector 6299df49d7eSJed Brown /// 6309df49d7eSJed Brown /// ``` 6319df49d7eSJed Brown /// # use libceed::prelude::*; 6324d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 6339df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 6349df49d7eSJed Brown /// let ne = 4; 6359df49d7eSJed Brown /// let p = 3; 6369df49d7eSJed Brown /// let q = 4; 6379df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 6389df49d7eSJed Brown /// 6399df49d7eSJed Brown /// // Vectors 640c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 641c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 6429df49d7eSJed Brown /// qdata.set_value(0.0); 643c68be7a2SJeremy L Thompson /// let u = ceed.vector_from_slice(&vec![1.0; ndofs])?; 644c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 6459df49d7eSJed Brown /// 6469df49d7eSJed Brown /// // Restrictions 6479df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 6489df49d7eSJed Brown /// for i in 0..ne { 6499df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 6509df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 6519df49d7eSJed Brown /// } 652c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 6539df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 6549df49d7eSJed Brown /// for i in 0..ne { 6559df49d7eSJed Brown /// indu[p * i + 0] = i as i32; 6569df49d7eSJed Brown /// indu[p * i + 1] = (i + 1) as i32; 6579df49d7eSJed Brown /// indu[p * i + 2] = (i + 2) as i32; 6589df49d7eSJed Brown /// } 659c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, 3, 1, 1, ndofs, MemType::Host, &indu)?; 6609df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 661c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 6629df49d7eSJed Brown /// 6639df49d7eSJed Brown /// // Bases 664c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 665c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 6669df49d7eSJed Brown /// 6679df49d7eSJed Brown /// // Build quadrature data 668c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 669c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 670c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 671c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 672c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 673c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 6749df49d7eSJed Brown /// 6759df49d7eSJed Brown /// // Mass operator 676c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 6779df49d7eSJed Brown /// let op_mass = ceed 678c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 679c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 680c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 681c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 6829df49d7eSJed Brown /// 6839df49d7eSJed Brown /// v.set_value(1.0); 684c68be7a2SJeremy L Thompson /// op_mass.apply_add(&u, &mut v)?; 6859df49d7eSJed Brown /// 6869df49d7eSJed Brown /// // Check 687e78171edSJeremy L Thompson /// let sum: Scalar = v.view()?.iter().sum(); 6889df49d7eSJed Brown /// assert!( 68980a9ef05SNatalie Beams /// (sum - (2.0 + ndofs as Scalar)).abs() < 10.0 * libceed::EPSILON, 6909df49d7eSJed Brown /// "Incorrect interval length computed and added" 6919df49d7eSJed Brown /// ); 692c68be7a2SJeremy L Thompson /// # Ok(()) 693c68be7a2SJeremy L Thompson /// # } 6949df49d7eSJed Brown /// ``` 6959df49d7eSJed Brown pub fn apply_add(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 6969df49d7eSJed Brown self.op_core.apply_add(input, output) 6979df49d7eSJed Brown } 6989df49d7eSJed Brown 6999df49d7eSJed Brown /// Provide a field to a Operator for use by its QFunction 7009df49d7eSJed Brown /// 7019df49d7eSJed Brown /// * `fieldname` - Name of the field (to be matched with the name used by 7029df49d7eSJed Brown /// the QFunction) 7039df49d7eSJed Brown /// * `r` - ElemRestriction 7049df49d7eSJed Brown /// * `b` - Basis in which the field resides or `BasisCollocated` if 7059df49d7eSJed Brown /// collocated with quadrature points 7069df49d7eSJed Brown /// * `v` - Vector to be used by Operator or `VectorActive` if field 7079df49d7eSJed Brown /// is active or `VectorNone` if using `Weight` with the 7089df49d7eSJed Brown /// QFunction 7099df49d7eSJed Brown /// 7109df49d7eSJed Brown /// 7119df49d7eSJed Brown /// ``` 7129df49d7eSJed Brown /// # use libceed::prelude::*; 7134d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 7149df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 715c68be7a2SJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 716c68be7a2SJeremy L Thompson /// let mut op = ceed.operator(&qf, QFunctionOpt::None, QFunctionOpt::None)?; 7179df49d7eSJed Brown /// 7189df49d7eSJed Brown /// // Operator field arguments 7199df49d7eSJed Brown /// let ne = 3; 7209df49d7eSJed Brown /// let q = 4; 7219df49d7eSJed Brown /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 7229df49d7eSJed Brown /// for i in 0..ne { 7239df49d7eSJed Brown /// ind[2 * i + 0] = i as i32; 7249df49d7eSJed Brown /// ind[2 * i + 1] = (i + 1) as i32; 7259df49d7eSJed Brown /// } 726c68be7a2SJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 7279df49d7eSJed Brown /// 728c68be7a2SJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 7299df49d7eSJed Brown /// 7309df49d7eSJed Brown /// // Operator field 731c68be7a2SJeremy L Thompson /// op = op.field("dx", &r, &b, VectorOpt::Active)?; 732c68be7a2SJeremy L Thompson /// # Ok(()) 733c68be7a2SJeremy L Thompson /// # } 7349df49d7eSJed Brown /// ``` 7359df49d7eSJed Brown #[allow(unused_mut)] 7369df49d7eSJed Brown pub fn field<'b>( 7379df49d7eSJed Brown mut self, 7389df49d7eSJed Brown fieldname: &str, 7399df49d7eSJed Brown r: impl Into<ElemRestrictionOpt<'b>>, 7409df49d7eSJed Brown b: impl Into<BasisOpt<'b>>, 7419df49d7eSJed Brown v: impl Into<VectorOpt<'b>>, 7429df49d7eSJed Brown ) -> crate::Result<Self> { 7439df49d7eSJed Brown let fieldname = CString::new(fieldname).expect("CString::new failed"); 7449df49d7eSJed Brown let fieldname = fieldname.as_ptr() as *const i8; 7459df49d7eSJed Brown let ierr = unsafe { 7469df49d7eSJed Brown bind_ceed::CeedOperatorSetField( 7479df49d7eSJed Brown self.op_core.ptr, 7489df49d7eSJed Brown fieldname, 7499df49d7eSJed Brown r.into().to_raw(), 7509df49d7eSJed Brown b.into().to_raw(), 7519df49d7eSJed Brown v.into().to_raw(), 7529df49d7eSJed Brown ) 7539df49d7eSJed Brown }; 7541142270cSJeremy L Thompson self.op_core.check_error(ierr)?; 7559df49d7eSJed Brown Ok(self) 7569df49d7eSJed Brown } 7579df49d7eSJed Brown 75808778c6fSJeremy L Thompson /// Get a slice of Operator inputs 75908778c6fSJeremy L Thompson /// 76008778c6fSJeremy L Thompson /// ``` 76108778c6fSJeremy L Thompson /// # use libceed::prelude::*; 7624d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 76308778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 76408778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 76508778c6fSJeremy L Thompson /// 76608778c6fSJeremy L Thompson /// // Operator field arguments 76708778c6fSJeremy L Thompson /// let ne = 3; 76808778c6fSJeremy L Thompson /// let q = 4 as usize; 76908778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 77008778c6fSJeremy L Thompson /// for i in 0..ne { 77108778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 77208778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 77308778c6fSJeremy L Thompson /// } 77408778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 77508778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 77608778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 77708778c6fSJeremy L Thompson /// 77808778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 77908778c6fSJeremy L Thompson /// 78008778c6fSJeremy L Thompson /// // Operator fields 78108778c6fSJeremy L Thompson /// let op = ceed 78208778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 78308778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 78408778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 78508778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 78608778c6fSJeremy L Thompson /// 78708778c6fSJeremy L Thompson /// let inputs = op.inputs()?; 78808778c6fSJeremy L Thompson /// 78908778c6fSJeremy L Thompson /// assert_eq!(inputs.len(), 2, "Incorrect inputs array"); 79008778c6fSJeremy L Thompson /// # Ok(()) 79108778c6fSJeremy L Thompson /// # } 79208778c6fSJeremy L Thompson /// ``` 79308778c6fSJeremy L Thompson pub fn inputs(&self) -> crate::Result<&[crate::OperatorField]> { 79408778c6fSJeremy L Thompson // Get array of raw C pointers for inputs 79508778c6fSJeremy L Thompson let mut num_inputs = 0; 79608778c6fSJeremy L Thompson let mut inputs_ptr = std::ptr::null_mut(); 79708778c6fSJeremy L Thompson let ierr = unsafe { 79808778c6fSJeremy L Thompson bind_ceed::CeedOperatorGetFields( 79908778c6fSJeremy L Thompson self.op_core.ptr, 80008778c6fSJeremy L Thompson &mut num_inputs, 80108778c6fSJeremy L Thompson &mut inputs_ptr, 80208778c6fSJeremy L Thompson std::ptr::null_mut() as *mut bind_ceed::CeedInt, 80308778c6fSJeremy L Thompson std::ptr::null_mut() as *mut *mut bind_ceed::CeedOperatorField, 80408778c6fSJeremy L Thompson ) 80508778c6fSJeremy L Thompson }; 80608778c6fSJeremy L Thompson self.op_core.check_error(ierr)?; 80708778c6fSJeremy L Thompson // Convert raw C pointers to fixed length slice 80808778c6fSJeremy L Thompson let inputs_slice = unsafe { 80908778c6fSJeremy L Thompson std::slice::from_raw_parts( 81008778c6fSJeremy L Thompson inputs_ptr as *const crate::OperatorField, 81108778c6fSJeremy L Thompson num_inputs as usize, 81208778c6fSJeremy L Thompson ) 81308778c6fSJeremy L Thompson }; 81408778c6fSJeremy L Thompson Ok(inputs_slice) 81508778c6fSJeremy L Thompson } 81608778c6fSJeremy L Thompson 81708778c6fSJeremy L Thompson /// Get a slice of Operator outputs 81808778c6fSJeremy L Thompson /// 81908778c6fSJeremy L Thompson /// ``` 82008778c6fSJeremy L Thompson /// # use libceed::prelude::*; 8214d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 82208778c6fSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 82308778c6fSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 82408778c6fSJeremy L Thompson /// 82508778c6fSJeremy L Thompson /// // Operator field arguments 82608778c6fSJeremy L Thompson /// let ne = 3; 82708778c6fSJeremy L Thompson /// let q = 4 as usize; 82808778c6fSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 82908778c6fSJeremy L Thompson /// for i in 0..ne { 83008778c6fSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 83108778c6fSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 83208778c6fSJeremy L Thompson /// } 83308778c6fSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 83408778c6fSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 83508778c6fSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, 2, 1, q * ne, strides)?; 83608778c6fSJeremy L Thompson /// 83708778c6fSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 83808778c6fSJeremy L Thompson /// 83908778c6fSJeremy L Thompson /// // Operator fields 84008778c6fSJeremy L Thompson /// let op = ceed 84108778c6fSJeremy L Thompson /// .operator(&qf, QFunctionOpt::None, QFunctionOpt::None)? 84208778c6fSJeremy L Thompson /// .field("dx", &r, &b, VectorOpt::Active)? 84308778c6fSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &b, VectorOpt::None)? 84408778c6fSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 84508778c6fSJeremy L Thompson /// 84608778c6fSJeremy L Thompson /// let outputs = op.outputs()?; 84708778c6fSJeremy L Thompson /// 84808778c6fSJeremy L Thompson /// assert_eq!(outputs.len(), 1, "Incorrect outputs array"); 84908778c6fSJeremy L Thompson /// # Ok(()) 85008778c6fSJeremy L Thompson /// # } 85108778c6fSJeremy L Thompson /// ``` 85208778c6fSJeremy L Thompson pub fn outputs(&self) -> crate::Result<&[crate::OperatorField]> { 85308778c6fSJeremy L Thompson // Get array of raw C pointers for outputs 85408778c6fSJeremy L Thompson let mut num_outputs = 0; 85508778c6fSJeremy L Thompson let mut outputs_ptr = std::ptr::null_mut(); 85608778c6fSJeremy L Thompson let ierr = unsafe { 85708778c6fSJeremy L Thompson bind_ceed::CeedOperatorGetFields( 85808778c6fSJeremy L Thompson self.op_core.ptr, 85908778c6fSJeremy L Thompson std::ptr::null_mut() as *mut bind_ceed::CeedInt, 86008778c6fSJeremy L Thompson std::ptr::null_mut() as *mut *mut bind_ceed::CeedOperatorField, 86108778c6fSJeremy L Thompson &mut num_outputs, 86208778c6fSJeremy L Thompson &mut outputs_ptr, 86308778c6fSJeremy L Thompson ) 86408778c6fSJeremy L Thompson }; 86508778c6fSJeremy L Thompson self.op_core.check_error(ierr)?; 86608778c6fSJeremy L Thompson // Convert raw C pointers to fixed length slice 86708778c6fSJeremy L Thompson let outputs_slice = unsafe { 86808778c6fSJeremy L Thompson std::slice::from_raw_parts( 86908778c6fSJeremy L Thompson outputs_ptr as *const crate::OperatorField, 87008778c6fSJeremy L Thompson num_outputs as usize, 87108778c6fSJeremy L Thompson ) 87208778c6fSJeremy L Thompson }; 87308778c6fSJeremy L Thompson Ok(outputs_slice) 87408778c6fSJeremy L Thompson } 87508778c6fSJeremy L Thompson 8766f97ff0aSJeremy L Thompson /// Check if Operator is setup correctly 8776f97ff0aSJeremy L Thompson /// 8786f97ff0aSJeremy L Thompson /// ``` 8796f97ff0aSJeremy L Thompson /// # use libceed::prelude::*; 8806f97ff0aSJeremy L Thompson /// # fn main() -> libceed::Result<()> { 8816f97ff0aSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 8826f97ff0aSJeremy L Thompson /// let ne = 4; 8836f97ff0aSJeremy L Thompson /// let p = 3; 8846f97ff0aSJeremy L Thompson /// let q = 4; 8856f97ff0aSJeremy L Thompson /// let ndofs = p * ne - ne + 1; 8866f97ff0aSJeremy L Thompson /// 8876f97ff0aSJeremy L Thompson /// // Restrictions 8886f97ff0aSJeremy L Thompson /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 8896f97ff0aSJeremy L Thompson /// for i in 0..ne { 8906f97ff0aSJeremy L Thompson /// indx[2 * i + 0] = i as i32; 8916f97ff0aSJeremy L Thompson /// indx[2 * i + 1] = (i + 1) as i32; 8926f97ff0aSJeremy L Thompson /// } 8936f97ff0aSJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 8946f97ff0aSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 8956f97ff0aSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 8966f97ff0aSJeremy L Thompson /// 8976f97ff0aSJeremy L Thompson /// // Bases 8986f97ff0aSJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 8996f97ff0aSJeremy L Thompson /// 9006f97ff0aSJeremy L Thompson /// // Build quadrature data 9016f97ff0aSJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 9026f97ff0aSJeremy L Thompson /// let op_build = ceed 9036f97ff0aSJeremy L Thompson /// .operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 9046f97ff0aSJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 9056f97ff0aSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 9066f97ff0aSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 9076f97ff0aSJeremy L Thompson /// 9086f97ff0aSJeremy L Thompson /// // Check 9096f97ff0aSJeremy L Thompson /// assert!(op_build.check().is_ok(), "Operator setup incorrectly"); 9106f97ff0aSJeremy L Thompson /// # Ok(()) 9116f97ff0aSJeremy L Thompson /// # } 9126f97ff0aSJeremy L Thompson /// ``` 9136f97ff0aSJeremy L Thompson pub fn check(self) -> crate::Result<Self> { 9146f97ff0aSJeremy L Thompson self.op_core.check()?; 9156f97ff0aSJeremy L Thompson Ok(self) 9166f97ff0aSJeremy L Thompson } 9176f97ff0aSJeremy L Thompson 9183f2759cfSJeremy L Thompson /// Get the number of elements associated with an Operator 9193f2759cfSJeremy L Thompson /// 9203f2759cfSJeremy L Thompson /// 9213f2759cfSJeremy L Thompson /// ``` 9223f2759cfSJeremy L Thompson /// # use libceed::prelude::*; 9234d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 9243f2759cfSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 9253f2759cfSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 9263f2759cfSJeremy L Thompson /// let mut op = ceed.operator(&qf, QFunctionOpt::None, QFunctionOpt::None)?; 9273f2759cfSJeremy L Thompson /// 9283f2759cfSJeremy L Thompson /// // Operator field arguments 9293f2759cfSJeremy L Thompson /// let ne = 3; 9303f2759cfSJeremy L Thompson /// let q = 4; 9313f2759cfSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 9323f2759cfSJeremy L Thompson /// for i in 0..ne { 9333f2759cfSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 9343f2759cfSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 9353f2759cfSJeremy L Thompson /// } 9363f2759cfSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 9373f2759cfSJeremy L Thompson /// 9383f2759cfSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 9393f2759cfSJeremy L Thompson /// 9403f2759cfSJeremy L Thompson /// // Operator field 9413f2759cfSJeremy L Thompson /// op = op.field("dx", &r, &b, VectorOpt::Active)?; 9423f2759cfSJeremy L Thompson /// 9433f2759cfSJeremy L Thompson /// // Check number of elements 9443f2759cfSJeremy L Thompson /// assert_eq!(op.num_elements(), ne, "Incorrect number of elements"); 9453f2759cfSJeremy L Thompson /// # Ok(()) 9463f2759cfSJeremy L Thompson /// # } 9473f2759cfSJeremy L Thompson /// ``` 9483f2759cfSJeremy L Thompson pub fn num_elements(&self) -> usize { 9493f2759cfSJeremy L Thompson let mut nelem = 0; 9503f2759cfSJeremy L Thompson unsafe { bind_ceed::CeedOperatorGetNumElements(self.op_core.ptr, &mut nelem) }; 9513f2759cfSJeremy L Thompson usize::try_from(nelem).unwrap() 9523f2759cfSJeremy L Thompson } 9533f2759cfSJeremy L Thompson 9543f2759cfSJeremy L Thompson /// Get the number of quadrature points associated with each element of 9553f2759cfSJeremy L Thompson /// an Operator 9563f2759cfSJeremy L Thompson /// 9573f2759cfSJeremy L Thompson /// 9583f2759cfSJeremy L Thompson /// ``` 9593f2759cfSJeremy L Thompson /// # use libceed::prelude::*; 9604d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 9613f2759cfSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 9623f2759cfSJeremy L Thompson /// let qf = ceed.q_function_interior_by_name("Mass1DBuild")?; 9633f2759cfSJeremy L Thompson /// let mut op = ceed.operator(&qf, QFunctionOpt::None, QFunctionOpt::None)?; 9643f2759cfSJeremy L Thompson /// 9653f2759cfSJeremy L Thompson /// // Operator field arguments 9663f2759cfSJeremy L Thompson /// let ne = 3; 9673f2759cfSJeremy L Thompson /// let q = 4; 9683f2759cfSJeremy L Thompson /// let mut ind: Vec<i32> = vec![0; 2 * ne]; 9693f2759cfSJeremy L Thompson /// for i in 0..ne { 9703f2759cfSJeremy L Thompson /// ind[2 * i + 0] = i as i32; 9713f2759cfSJeremy L Thompson /// ind[2 * i + 1] = (i + 1) as i32; 9723f2759cfSJeremy L Thompson /// } 9733f2759cfSJeremy L Thompson /// let r = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &ind)?; 9743f2759cfSJeremy L Thompson /// 9753f2759cfSJeremy L Thompson /// let b = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 9763f2759cfSJeremy L Thompson /// 9773f2759cfSJeremy L Thompson /// // Operator field 9783f2759cfSJeremy L Thompson /// op = op.field("dx", &r, &b, VectorOpt::Active)?; 9793f2759cfSJeremy L Thompson /// 9803f2759cfSJeremy L Thompson /// // Check number of quadrature points 9813f2759cfSJeremy L Thompson /// assert_eq!( 9823f2759cfSJeremy L Thompson /// op.num_quadrature_points(), 9833f2759cfSJeremy L Thompson /// q, 9843f2759cfSJeremy L Thompson /// "Incorrect number of quadrature points" 9853f2759cfSJeremy L Thompson /// ); 9863f2759cfSJeremy L Thompson /// # Ok(()) 9873f2759cfSJeremy L Thompson /// # } 9883f2759cfSJeremy L Thompson /// ``` 9893f2759cfSJeremy L Thompson pub fn num_quadrature_points(&self) -> usize { 9903f2759cfSJeremy L Thompson let mut Q = 0; 9913f2759cfSJeremy L Thompson unsafe { bind_ceed::CeedOperatorGetNumQuadraturePoints(self.op_core.ptr, &mut Q) }; 9923f2759cfSJeremy L Thompson usize::try_from(Q).unwrap() 9933f2759cfSJeremy L Thompson } 9943f2759cfSJeremy L Thompson 9959df49d7eSJed Brown /// Assemble the diagonal of a square linear Operator 9969df49d7eSJed Brown /// 9979df49d7eSJed Brown /// This overwrites a Vector with the diagonal of a linear Operator. 9989df49d7eSJed Brown /// 9999df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 10009df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 10019df49d7eSJed Brown /// 10029df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 10039df49d7eSJed Brown /// * `assembled` - Vector to store assembled Operator diagonal 10049df49d7eSJed Brown /// 10059df49d7eSJed Brown /// ``` 10069df49d7eSJed Brown /// # use libceed::prelude::*; 10074d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 10089df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 10099df49d7eSJed Brown /// let ne = 4; 10109df49d7eSJed Brown /// let p = 3; 10119df49d7eSJed Brown /// let q = 4; 10129df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 10139df49d7eSJed Brown /// 10149df49d7eSJed Brown /// // Vectors 1015c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 1016c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 10179df49d7eSJed Brown /// qdata.set_value(0.0); 1018c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ndofs)?; 10199df49d7eSJed Brown /// u.set_value(1.0); 1020c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 10219df49d7eSJed Brown /// v.set_value(0.0); 10229df49d7eSJed Brown /// 10239df49d7eSJed Brown /// // Restrictions 10249df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 10259df49d7eSJed Brown /// for i in 0..ne { 10269df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 10279df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 10289df49d7eSJed Brown /// } 1029c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 10309df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 10319df49d7eSJed Brown /// for i in 0..ne { 10329df49d7eSJed Brown /// indu[p * i + 0] = (2 * i) as i32; 10339df49d7eSJed Brown /// indu[p * i + 1] = (2 * i + 1) as i32; 10349df49d7eSJed Brown /// indu[p * i + 2] = (2 * i + 2) as i32; 10359df49d7eSJed Brown /// } 1036c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, p, 1, 1, ndofs, MemType::Host, &indu)?; 10379df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1038c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 10399df49d7eSJed Brown /// 10409df49d7eSJed Brown /// // Bases 1041c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1042c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 10439df49d7eSJed Brown /// 10449df49d7eSJed Brown /// // Build quadrature data 1045c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1046c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1047c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1048c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1049c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1050c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 10519df49d7eSJed Brown /// 10529df49d7eSJed Brown /// // Mass operator 1053c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 10549df49d7eSJed Brown /// let op_mass = ceed 1055c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1056c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 1057c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1058c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 10599df49d7eSJed Brown /// 10609df49d7eSJed Brown /// // Diagonal 1061c68be7a2SJeremy L Thompson /// let mut diag = ceed.vector(ndofs)?; 10629df49d7eSJed Brown /// diag.set_value(0.0); 1063c68be7a2SJeremy L Thompson /// op_mass.linear_assemble_diagonal(&mut diag)?; 10649df49d7eSJed Brown /// 10659df49d7eSJed Brown /// // Manual diagonal computation 1066c68be7a2SJeremy L Thompson /// let mut true_diag = ceed.vector(ndofs)?; 10679c774eddSJeremy L Thompson /// true_diag.set_value(0.0)?; 10689df49d7eSJed Brown /// for i in 0..ndofs { 10699df49d7eSJed Brown /// u.set_value(0.0); 10709df49d7eSJed Brown /// { 1071e78171edSJeremy L Thompson /// let mut u_array = u.view_mut()?; 10729df49d7eSJed Brown /// u_array[i] = 1.; 10739df49d7eSJed Brown /// } 10749df49d7eSJed Brown /// 1075c68be7a2SJeremy L Thompson /// op_mass.apply(&u, &mut v)?; 10769df49d7eSJed Brown /// 10779df49d7eSJed Brown /// { 10789c774eddSJeremy L Thompson /// let v_array = v.view()?; 1079e78171edSJeremy L Thompson /// let mut true_array = true_diag.view_mut()?; 10809df49d7eSJed Brown /// true_array[i] = v_array[i]; 10819df49d7eSJed Brown /// } 10829df49d7eSJed Brown /// } 10839df49d7eSJed Brown /// 10849df49d7eSJed Brown /// // Check 1085e78171edSJeremy L Thompson /// diag.view()? 10869df49d7eSJed Brown /// .iter() 1087e78171edSJeremy L Thompson /// .zip(true_diag.view()?.iter()) 10889df49d7eSJed Brown /// .for_each(|(computed, actual)| { 10899df49d7eSJed Brown /// assert!( 109080a9ef05SNatalie Beams /// (*computed - *actual).abs() < 10.0 * libceed::EPSILON, 10919df49d7eSJed Brown /// "Diagonal entry incorrect" 10929df49d7eSJed Brown /// ); 10939df49d7eSJed Brown /// }); 1094c68be7a2SJeremy L Thompson /// # Ok(()) 1095c68be7a2SJeremy L Thompson /// # } 10969df49d7eSJed Brown /// ``` 10979df49d7eSJed Brown pub fn linear_assemble_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 10989df49d7eSJed Brown self.op_core.linear_assemble_diagonal(assembled) 10999df49d7eSJed Brown } 11009df49d7eSJed Brown 11019df49d7eSJed Brown /// Assemble the diagonal of a square linear Operator 11029df49d7eSJed Brown /// 11039df49d7eSJed Brown /// This sums into a Vector with the diagonal of a linear Operator. 11049df49d7eSJed Brown /// 11059df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 11069df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 11079df49d7eSJed Brown /// 11089df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 11099df49d7eSJed Brown /// * `assembled` - Vector to store assembled Operator diagonal 11109df49d7eSJed Brown /// 11119df49d7eSJed Brown /// 11129df49d7eSJed Brown /// ``` 11139df49d7eSJed Brown /// # use libceed::prelude::*; 11144d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 11159df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 11169df49d7eSJed Brown /// let ne = 4; 11179df49d7eSJed Brown /// let p = 3; 11189df49d7eSJed Brown /// let q = 4; 11199df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 11209df49d7eSJed Brown /// 11219df49d7eSJed Brown /// // Vectors 1122c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 1123c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 11249df49d7eSJed Brown /// qdata.set_value(0.0); 1125c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ndofs)?; 11269df49d7eSJed Brown /// u.set_value(1.0); 1127c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 11289df49d7eSJed Brown /// v.set_value(0.0); 11299df49d7eSJed Brown /// 11309df49d7eSJed Brown /// // Restrictions 11319df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 11329df49d7eSJed Brown /// for i in 0..ne { 11339df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 11349df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 11359df49d7eSJed Brown /// } 1136c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 11379df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 11389df49d7eSJed Brown /// for i in 0..ne { 11399df49d7eSJed Brown /// indu[p * i + 0] = (2 * i) as i32; 11409df49d7eSJed Brown /// indu[p * i + 1] = (2 * i + 1) as i32; 11419df49d7eSJed Brown /// indu[p * i + 2] = (2 * i + 2) as i32; 11429df49d7eSJed Brown /// } 1143c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, p, 1, 1, ndofs, MemType::Host, &indu)?; 11449df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1145c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 11469df49d7eSJed Brown /// 11479df49d7eSJed Brown /// // Bases 1148c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1149c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 11509df49d7eSJed Brown /// 11519df49d7eSJed Brown /// // Build quadrature data 1152c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1153c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1154c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1155c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1156c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1157c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 11589df49d7eSJed Brown /// 11599df49d7eSJed Brown /// // Mass operator 1160c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 11619df49d7eSJed Brown /// let op_mass = ceed 1162c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1163c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 1164c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1165c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 11669df49d7eSJed Brown /// 11679df49d7eSJed Brown /// // Diagonal 1168c68be7a2SJeremy L Thompson /// let mut diag = ceed.vector(ndofs)?; 11699df49d7eSJed Brown /// diag.set_value(1.0); 1170c68be7a2SJeremy L Thompson /// op_mass.linear_assemble_add_diagonal(&mut diag)?; 11719df49d7eSJed Brown /// 11729df49d7eSJed Brown /// // Manual diagonal computation 1173c68be7a2SJeremy L Thompson /// let mut true_diag = ceed.vector(ndofs)?; 11749c774eddSJeremy L Thompson /// true_diag.set_value(0.0)?; 11759df49d7eSJed Brown /// for i in 0..ndofs { 11769df49d7eSJed Brown /// u.set_value(0.0); 11779df49d7eSJed Brown /// { 1178e78171edSJeremy L Thompson /// let mut u_array = u.view_mut()?; 11799df49d7eSJed Brown /// u_array[i] = 1.; 11809df49d7eSJed Brown /// } 11819df49d7eSJed Brown /// 1182c68be7a2SJeremy L Thompson /// op_mass.apply(&u, &mut v)?; 11839df49d7eSJed Brown /// 11849df49d7eSJed Brown /// { 11859c774eddSJeremy L Thompson /// let v_array = v.view()?; 1186e78171edSJeremy L Thompson /// let mut true_array = true_diag.view_mut()?; 11879df49d7eSJed Brown /// true_array[i] = v_array[i] + 1.0; 11889df49d7eSJed Brown /// } 11899df49d7eSJed Brown /// } 11909df49d7eSJed Brown /// 11919df49d7eSJed Brown /// // Check 1192e78171edSJeremy L Thompson /// diag.view()? 11939df49d7eSJed Brown /// .iter() 1194e78171edSJeremy L Thompson /// .zip(true_diag.view()?.iter()) 11959df49d7eSJed Brown /// .for_each(|(computed, actual)| { 11969df49d7eSJed Brown /// assert!( 119780a9ef05SNatalie Beams /// (*computed - *actual).abs() < 10.0 * libceed::EPSILON, 11989df49d7eSJed Brown /// "Diagonal entry incorrect" 11999df49d7eSJed Brown /// ); 12009df49d7eSJed Brown /// }); 1201c68be7a2SJeremy L Thompson /// # Ok(()) 1202c68be7a2SJeremy L Thompson /// # } 12039df49d7eSJed Brown /// ``` 12049df49d7eSJed Brown pub fn linear_assemble_add_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 12059df49d7eSJed Brown self.op_core.linear_assemble_add_diagonal(assembled) 12069df49d7eSJed Brown } 12079df49d7eSJed Brown 12089df49d7eSJed Brown /// Assemble the point block diagonal of a square linear Operator 12099df49d7eSJed Brown /// 12109df49d7eSJed Brown /// This overwrites a Vector with the point block diagonal of a linear 12119df49d7eSJed Brown /// Operator. 12129df49d7eSJed Brown /// 12139df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 12149df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 12159df49d7eSJed Brown /// 12169df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 12179df49d7eSJed Brown /// * `assembled` - Vector to store assembled CeedOperator point block 12189df49d7eSJed Brown /// diagonal, provided in row-major form with an 12199df49d7eSJed Brown /// `ncomp * ncomp` block at each node. The dimensions of 12209df49d7eSJed Brown /// this vector are derived from the active vector for 12219df49d7eSJed Brown /// the CeedOperator. The array has shape 12229df49d7eSJed Brown /// `[nodes, component out, component in]`. 12239df49d7eSJed Brown /// 12249df49d7eSJed Brown /// ``` 12259df49d7eSJed Brown /// # use libceed::prelude::*; 12264d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 12279df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 12289df49d7eSJed Brown /// let ne = 4; 12299df49d7eSJed Brown /// let p = 3; 12309df49d7eSJed Brown /// let q = 4; 12319df49d7eSJed Brown /// let ncomp = 2; 12329df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 12339df49d7eSJed Brown /// 12349df49d7eSJed Brown /// // Vectors 1235c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 1236c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 12379df49d7eSJed Brown /// qdata.set_value(0.0); 1238c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ncomp * ndofs)?; 12399df49d7eSJed Brown /// u.set_value(1.0); 1240c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ncomp * ndofs)?; 12419df49d7eSJed Brown /// v.set_value(0.0); 12429df49d7eSJed Brown /// 12439df49d7eSJed Brown /// // Restrictions 12449df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 12459df49d7eSJed Brown /// for i in 0..ne { 12469df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 12479df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 12489df49d7eSJed Brown /// } 1249c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 12509df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 12519df49d7eSJed Brown /// for i in 0..ne { 12529df49d7eSJed Brown /// indu[p * i + 0] = (2 * i) as i32; 12539df49d7eSJed Brown /// indu[p * i + 1] = (2 * i + 1) as i32; 12549df49d7eSJed Brown /// indu[p * i + 2] = (2 * i + 2) as i32; 12559df49d7eSJed Brown /// } 1256c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, p, ncomp, ndofs, ncomp * ndofs, MemType::Host, &indu)?; 12579df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1258c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 12599df49d7eSJed Brown /// 12609df49d7eSJed Brown /// // Bases 1261c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1262c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, ncomp, p, q, QuadMode::Gauss)?; 12639df49d7eSJed Brown /// 12649df49d7eSJed Brown /// // Build quadrature data 1265c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1266c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1267c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1268c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1269c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1270c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 12719df49d7eSJed Brown /// 12729df49d7eSJed Brown /// // Mass operator 12739df49d7eSJed Brown /// let mut mass_2_comp = |[u, qdata, ..]: QFunctionInputs, [v, ..]: QFunctionOutputs| { 12749df49d7eSJed Brown /// // Number of quadrature points 12759df49d7eSJed Brown /// let q = qdata.len(); 12769df49d7eSJed Brown /// 12779df49d7eSJed Brown /// // Iterate over quadrature points 12789df49d7eSJed Brown /// for i in 0..q { 12799df49d7eSJed Brown /// v[i + 0 * q] = u[i + 1 * q] * qdata[i]; 12809df49d7eSJed Brown /// v[i + 1 * q] = u[i + 0 * q] * qdata[i]; 12819df49d7eSJed Brown /// } 12829df49d7eSJed Brown /// 12839df49d7eSJed Brown /// // Return clean error code 12849df49d7eSJed Brown /// 0 12859df49d7eSJed Brown /// }; 12869df49d7eSJed Brown /// 12879df49d7eSJed Brown /// let qf_mass = ceed 1288c68be7a2SJeremy L Thompson /// .q_function_interior(1, Box::new(mass_2_comp))? 1289c68be7a2SJeremy L Thompson /// .input("u", 2, EvalMode::Interp)? 1290c68be7a2SJeremy L Thompson /// .input("qdata", 1, EvalMode::None)? 1291c68be7a2SJeremy L Thompson /// .output("v", 2, EvalMode::Interp)?; 12929df49d7eSJed Brown /// 12939df49d7eSJed Brown /// let op_mass = ceed 1294c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1295c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 1296c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1297c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 12989df49d7eSJed Brown /// 12999df49d7eSJed Brown /// // Diagonal 1300c68be7a2SJeremy L Thompson /// let mut diag = ceed.vector(ncomp * ncomp * ndofs)?; 13019df49d7eSJed Brown /// diag.set_value(0.0); 1302c68be7a2SJeremy L Thompson /// op_mass.linear_assemble_point_block_diagonal(&mut diag)?; 13039df49d7eSJed Brown /// 13049df49d7eSJed Brown /// // Manual diagonal computation 1305c68be7a2SJeremy L Thompson /// let mut true_diag = ceed.vector(ncomp * ncomp * ndofs)?; 13069c774eddSJeremy L Thompson /// true_diag.set_value(0.0)?; 13079df49d7eSJed Brown /// for i in 0..ndofs { 13089df49d7eSJed Brown /// for j in 0..ncomp { 13099df49d7eSJed Brown /// u.set_value(0.0); 13109df49d7eSJed Brown /// { 1311e78171edSJeremy L Thompson /// let mut u_array = u.view_mut()?; 13129df49d7eSJed Brown /// u_array[i + j * ndofs] = 1.; 13139df49d7eSJed Brown /// } 13149df49d7eSJed Brown /// 1315c68be7a2SJeremy L Thompson /// op_mass.apply(&u, &mut v)?; 13169df49d7eSJed Brown /// 13179df49d7eSJed Brown /// { 13189c774eddSJeremy L Thompson /// let v_array = v.view()?; 1319e78171edSJeremy L Thompson /// let mut true_array = true_diag.view_mut()?; 13209df49d7eSJed Brown /// for k in 0..ncomp { 13219df49d7eSJed Brown /// true_array[i * ncomp * ncomp + k * ncomp + j] = v_array[i + k * ndofs]; 13229df49d7eSJed Brown /// } 13239df49d7eSJed Brown /// } 13249df49d7eSJed Brown /// } 13259df49d7eSJed Brown /// } 13269df49d7eSJed Brown /// 13279df49d7eSJed Brown /// // Check 1328e78171edSJeremy L Thompson /// diag.view()? 13299df49d7eSJed Brown /// .iter() 1330e78171edSJeremy L Thompson /// .zip(true_diag.view()?.iter()) 13319df49d7eSJed Brown /// .for_each(|(computed, actual)| { 13329df49d7eSJed Brown /// assert!( 133380a9ef05SNatalie Beams /// (*computed - *actual).abs() < 10.0 * libceed::EPSILON, 13349df49d7eSJed Brown /// "Diagonal entry incorrect" 13359df49d7eSJed Brown /// ); 13369df49d7eSJed Brown /// }); 1337c68be7a2SJeremy L Thompson /// # Ok(()) 1338c68be7a2SJeremy L Thompson /// # } 13399df49d7eSJed Brown /// ``` 13409df49d7eSJed Brown pub fn linear_assemble_point_block_diagonal( 13419df49d7eSJed Brown &self, 13429df49d7eSJed Brown assembled: &mut Vector, 13439df49d7eSJed Brown ) -> crate::Result<i32> { 13449df49d7eSJed Brown self.op_core.linear_assemble_point_block_diagonal(assembled) 13459df49d7eSJed Brown } 13469df49d7eSJed Brown 13479df49d7eSJed Brown /// Assemble the point block diagonal of a square linear Operator 13489df49d7eSJed Brown /// 13499df49d7eSJed Brown /// This sums into a Vector with the point block diagonal of a linear 13509df49d7eSJed Brown /// Operator. 13519df49d7eSJed Brown /// 13529df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 13539df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 13549df49d7eSJed Brown /// 13559df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 13569df49d7eSJed Brown /// * `assembled` - Vector to store assembled CeedOperator point block 13579df49d7eSJed Brown /// diagonal, provided in row-major form with an 13589df49d7eSJed Brown /// `ncomp * ncomp` block at each node. The dimensions of 13599df49d7eSJed Brown /// this vector are derived from the active vector for 13609df49d7eSJed Brown /// the CeedOperator. The array has shape 13619df49d7eSJed Brown /// `[nodes, component out, component in]`. 13629df49d7eSJed Brown /// 13639df49d7eSJed Brown /// ``` 13649df49d7eSJed Brown /// # use libceed::prelude::*; 13654d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 13669df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 13679df49d7eSJed Brown /// let ne = 4; 13689df49d7eSJed Brown /// let p = 3; 13699df49d7eSJed Brown /// let q = 4; 13709df49d7eSJed Brown /// let ncomp = 2; 13719df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 13729df49d7eSJed Brown /// 13739df49d7eSJed Brown /// // Vectors 1374c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 1375c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 13769df49d7eSJed Brown /// qdata.set_value(0.0); 1377c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ncomp * ndofs)?; 13789df49d7eSJed Brown /// u.set_value(1.0); 1379c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ncomp * ndofs)?; 13809df49d7eSJed Brown /// v.set_value(0.0); 13819df49d7eSJed Brown /// 13829df49d7eSJed Brown /// // Restrictions 13839df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 13849df49d7eSJed Brown /// for i in 0..ne { 13859df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 13869df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 13879df49d7eSJed Brown /// } 1388c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 13899df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 13909df49d7eSJed Brown /// for i in 0..ne { 13919df49d7eSJed Brown /// indu[p * i + 0] = (2 * i) as i32; 13929df49d7eSJed Brown /// indu[p * i + 1] = (2 * i + 1) as i32; 13939df49d7eSJed Brown /// indu[p * i + 2] = (2 * i + 2) as i32; 13949df49d7eSJed Brown /// } 1395c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, p, ncomp, ndofs, ncomp * ndofs, MemType::Host, &indu)?; 13969df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1397c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 13989df49d7eSJed Brown /// 13999df49d7eSJed Brown /// // Bases 1400c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1401c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, ncomp, p, q, QuadMode::Gauss)?; 14029df49d7eSJed Brown /// 14039df49d7eSJed Brown /// // Build quadrature data 1404c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1405c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1406c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1407c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1408c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1409c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 14109df49d7eSJed Brown /// 14119df49d7eSJed Brown /// // Mass operator 14129df49d7eSJed Brown /// let mut mass_2_comp = |[u, qdata, ..]: QFunctionInputs, [v, ..]: QFunctionOutputs| { 14139df49d7eSJed Brown /// // Number of quadrature points 14149df49d7eSJed Brown /// let q = qdata.len(); 14159df49d7eSJed Brown /// 14169df49d7eSJed Brown /// // Iterate over quadrature points 14179df49d7eSJed Brown /// for i in 0..q { 14189df49d7eSJed Brown /// v[i + 0 * q] = u[i + 1 * q] * qdata[i]; 14199df49d7eSJed Brown /// v[i + 1 * q] = u[i + 0 * q] * qdata[i]; 14209df49d7eSJed Brown /// } 14219df49d7eSJed Brown /// 14229df49d7eSJed Brown /// // Return clean error code 14239df49d7eSJed Brown /// 0 14249df49d7eSJed Brown /// }; 14259df49d7eSJed Brown /// 14269df49d7eSJed Brown /// let qf_mass = ceed 1427c68be7a2SJeremy L Thompson /// .q_function_interior(1, Box::new(mass_2_comp))? 1428c68be7a2SJeremy L Thompson /// .input("u", 2, EvalMode::Interp)? 1429c68be7a2SJeremy L Thompson /// .input("qdata", 1, EvalMode::None)? 1430c68be7a2SJeremy L Thompson /// .output("v", 2, EvalMode::Interp)?; 14319df49d7eSJed Brown /// 14329df49d7eSJed Brown /// let op_mass = ceed 1433c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1434c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 1435c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1436c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 14379df49d7eSJed Brown /// 14389df49d7eSJed Brown /// // Diagonal 1439c68be7a2SJeremy L Thompson /// let mut diag = ceed.vector(ncomp * ncomp * ndofs)?; 14409df49d7eSJed Brown /// diag.set_value(1.0); 1441c68be7a2SJeremy L Thompson /// op_mass.linear_assemble_add_point_block_diagonal(&mut diag)?; 14429df49d7eSJed Brown /// 14439df49d7eSJed Brown /// // Manual diagonal computation 1444c68be7a2SJeremy L Thompson /// let mut true_diag = ceed.vector(ncomp * ncomp * ndofs)?; 14459c774eddSJeremy L Thompson /// true_diag.set_value(0.0)?; 14469df49d7eSJed Brown /// for i in 0..ndofs { 14479df49d7eSJed Brown /// for j in 0..ncomp { 14489df49d7eSJed Brown /// u.set_value(0.0); 14499df49d7eSJed Brown /// { 1450e78171edSJeremy L Thompson /// let mut u_array = u.view_mut()?; 14519df49d7eSJed Brown /// u_array[i + j * ndofs] = 1.; 14529df49d7eSJed Brown /// } 14539df49d7eSJed Brown /// 1454c68be7a2SJeremy L Thompson /// op_mass.apply(&u, &mut v)?; 14559df49d7eSJed Brown /// 14569df49d7eSJed Brown /// { 14579c774eddSJeremy L Thompson /// let v_array = v.view()?; 1458e78171edSJeremy L Thompson /// let mut true_array = true_diag.view_mut()?; 14599df49d7eSJed Brown /// for k in 0..ncomp { 14609df49d7eSJed Brown /// true_array[i * ncomp * ncomp + k * ncomp + j] = v_array[i + k * ndofs]; 14619df49d7eSJed Brown /// } 14629df49d7eSJed Brown /// } 14639df49d7eSJed Brown /// } 14649df49d7eSJed Brown /// } 14659df49d7eSJed Brown /// 14669df49d7eSJed Brown /// // Check 1467e78171edSJeremy L Thompson /// diag.view()? 14689df49d7eSJed Brown /// .iter() 1469e78171edSJeremy L Thompson /// .zip(true_diag.view()?.iter()) 14709df49d7eSJed Brown /// .for_each(|(computed, actual)| { 14719df49d7eSJed Brown /// assert!( 147280a9ef05SNatalie Beams /// (*computed - 1.0 - *actual).abs() < 10.0 * libceed::EPSILON, 14739df49d7eSJed Brown /// "Diagonal entry incorrect" 14749df49d7eSJed Brown /// ); 14759df49d7eSJed Brown /// }); 1476c68be7a2SJeremy L Thompson /// # Ok(()) 1477c68be7a2SJeremy L Thompson /// # } 14789df49d7eSJed Brown /// ``` 14799df49d7eSJed Brown pub fn linear_assemble_add_point_block_diagonal( 14809df49d7eSJed Brown &self, 14819df49d7eSJed Brown assembled: &mut Vector, 14829df49d7eSJed Brown ) -> crate::Result<i32> { 14839df49d7eSJed Brown self.op_core 14849df49d7eSJed Brown .linear_assemble_add_point_block_diagonal(assembled) 14859df49d7eSJed Brown } 14869df49d7eSJed Brown 14879df49d7eSJed Brown /// Create a multigrid coarse Operator and level transfer Operators for a 14889df49d7eSJed Brown /// given Operator, creating the prolongation basis from the fine and 14899df49d7eSJed Brown /// coarse grid interpolation 14909df49d7eSJed Brown /// 14919df49d7eSJed Brown /// * `p_mult_fine` - Lvector multiplicity in parallel gather/scatter 14929df49d7eSJed Brown /// * `rstr_coarse` - Coarse grid restriction 14939df49d7eSJed Brown /// * `basis_coarse` - Coarse grid active vector basis 14949df49d7eSJed Brown /// 14959df49d7eSJed Brown /// ``` 14969df49d7eSJed Brown /// # use libceed::prelude::*; 14974d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 14989df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 14999df49d7eSJed Brown /// let ne = 15; 15009df49d7eSJed Brown /// let p_coarse = 3; 15019df49d7eSJed Brown /// let p_fine = 5; 15029df49d7eSJed Brown /// let q = 6; 15039df49d7eSJed Brown /// let ndofs_coarse = p_coarse * ne - ne + 1; 15049df49d7eSJed Brown /// let ndofs_fine = p_fine * ne - ne + 1; 15059df49d7eSJed Brown /// 15069df49d7eSJed Brown /// // Vectors 15079df49d7eSJed Brown /// let x_array = (0..ne + 1) 150880a9ef05SNatalie Beams /// .map(|i| 2.0 * i as Scalar / ne as Scalar - 1.0) 150980a9ef05SNatalie Beams /// .collect::<Vec<Scalar>>(); 1510c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&x_array)?; 1511c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 15129df49d7eSJed Brown /// qdata.set_value(0.0); 1513c68be7a2SJeremy L Thompson /// let mut u_coarse = ceed.vector(ndofs_coarse)?; 15149df49d7eSJed Brown /// u_coarse.set_value(1.0); 1515c68be7a2SJeremy L Thompson /// let mut u_fine = ceed.vector(ndofs_fine)?; 15169df49d7eSJed Brown /// u_fine.set_value(1.0); 1517c68be7a2SJeremy L Thompson /// let mut v_coarse = ceed.vector(ndofs_coarse)?; 15189df49d7eSJed Brown /// v_coarse.set_value(0.0); 1519c68be7a2SJeremy L Thompson /// let mut v_fine = ceed.vector(ndofs_fine)?; 15209df49d7eSJed Brown /// v_fine.set_value(0.0); 1521c68be7a2SJeremy L Thompson /// let mut multiplicity = ceed.vector(ndofs_fine)?; 15229df49d7eSJed Brown /// multiplicity.set_value(1.0); 15239df49d7eSJed Brown /// 15249df49d7eSJed Brown /// // Restrictions 15259df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1526c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 15279df49d7eSJed Brown /// 15289df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 15299df49d7eSJed Brown /// for i in 0..ne { 15309df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 15319df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 15329df49d7eSJed Brown /// } 1533c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 15349df49d7eSJed Brown /// 15359df49d7eSJed Brown /// let mut indu_coarse: Vec<i32> = vec![0; p_coarse * ne]; 15369df49d7eSJed Brown /// for i in 0..ne { 15379df49d7eSJed Brown /// for j in 0..p_coarse { 15389df49d7eSJed Brown /// indu_coarse[p_coarse * i + j] = (i + j) as i32; 15399df49d7eSJed Brown /// } 15409df49d7eSJed Brown /// } 1541c68be7a2SJeremy L Thompson /// let ru_coarse = ceed.elem_restriction( 15429df49d7eSJed Brown /// ne, 15439df49d7eSJed Brown /// p_coarse, 15449df49d7eSJed Brown /// 1, 15459df49d7eSJed Brown /// 1, 15469df49d7eSJed Brown /// ndofs_coarse, 15479df49d7eSJed Brown /// MemType::Host, 15489df49d7eSJed Brown /// &indu_coarse, 1549c68be7a2SJeremy L Thompson /// )?; 15509df49d7eSJed Brown /// 15519df49d7eSJed Brown /// let mut indu_fine: Vec<i32> = vec![0; p_fine * ne]; 15529df49d7eSJed Brown /// for i in 0..ne { 15539df49d7eSJed Brown /// for j in 0..p_fine { 15549df49d7eSJed Brown /// indu_fine[p_fine * i + j] = (i + j) as i32; 15559df49d7eSJed Brown /// } 15569df49d7eSJed Brown /// } 1557c68be7a2SJeremy L Thompson /// let ru_fine = ceed.elem_restriction(ne, p_fine, 1, 1, ndofs_fine, MemType::Host, &indu_fine)?; 15589df49d7eSJed Brown /// 15599df49d7eSJed Brown /// // Bases 1560c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1561c68be7a2SJeremy L Thompson /// let bu_coarse = ceed.basis_tensor_H1_Lagrange(1, 1, p_coarse, q, QuadMode::Gauss)?; 1562c68be7a2SJeremy L Thompson /// let bu_fine = ceed.basis_tensor_H1_Lagrange(1, 1, p_fine, q, QuadMode::Gauss)?; 15639df49d7eSJed Brown /// 15649df49d7eSJed Brown /// // Build quadrature data 1565c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1566c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1567c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1568c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1569c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1570c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 15719df49d7eSJed Brown /// 15729df49d7eSJed Brown /// // Mass operator 1573c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 15749df49d7eSJed Brown /// let op_mass_fine = ceed 1575c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1576c68be7a2SJeremy L Thompson /// .field("u", &ru_fine, &bu_fine, VectorOpt::Active)? 1577c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1578c68be7a2SJeremy L Thompson /// .field("v", &ru_fine, &bu_fine, VectorOpt::Active)?; 15799df49d7eSJed Brown /// 15809df49d7eSJed Brown /// // Multigrid setup 1581c68be7a2SJeremy L Thompson /// let (op_mass_coarse, op_prolong, op_restrict) = 1582c68be7a2SJeremy L Thompson /// op_mass_fine.create_multigrid_level(&multiplicity, &ru_coarse, &bu_coarse)?; 15839df49d7eSJed Brown /// 15849df49d7eSJed Brown /// // Coarse problem 15859df49d7eSJed Brown /// u_coarse.set_value(1.0); 1586c68be7a2SJeremy L Thompson /// op_mass_coarse.apply(&u_coarse, &mut v_coarse)?; 15879df49d7eSJed Brown /// 15889df49d7eSJed Brown /// // Check 1589e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 15909df49d7eSJed Brown /// assert!( 159180a9ef05SNatalie Beams /// (sum - 2.0).abs() < 50.0 * libceed::EPSILON, 15929df49d7eSJed Brown /// "Incorrect interval length computed" 15939df49d7eSJed Brown /// ); 15949df49d7eSJed Brown /// 15959df49d7eSJed Brown /// // Prolong 1596c68be7a2SJeremy L Thompson /// op_prolong.apply(&u_coarse, &mut u_fine)?; 15979df49d7eSJed Brown /// 15989df49d7eSJed Brown /// // Fine problem 1599c68be7a2SJeremy L Thompson /// op_mass_fine.apply(&u_fine, &mut v_fine)?; 16009df49d7eSJed Brown /// 16019df49d7eSJed Brown /// // Check 1602e78171edSJeremy L Thompson /// let sum: Scalar = v_fine.view()?.iter().sum(); 16039df49d7eSJed Brown /// assert!( 160480a9ef05SNatalie Beams /// (sum - 2.0).abs() < 50.0 * libceed::EPSILON, 16059df49d7eSJed Brown /// "Incorrect interval length computed" 16069df49d7eSJed Brown /// ); 16079df49d7eSJed Brown /// 16089df49d7eSJed Brown /// // Restrict 1609c68be7a2SJeremy L Thompson /// op_restrict.apply(&v_fine, &mut v_coarse)?; 16109df49d7eSJed Brown /// 16119df49d7eSJed Brown /// // Check 1612e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 16139df49d7eSJed Brown /// assert!( 161480a9ef05SNatalie Beams /// (sum - 2.0).abs() < 50.0 * libceed::EPSILON, 16159df49d7eSJed Brown /// "Incorrect interval length computed" 16169df49d7eSJed Brown /// ); 1617c68be7a2SJeremy L Thompson /// # Ok(()) 1618c68be7a2SJeremy L Thompson /// # } 16199df49d7eSJed Brown /// ``` 1620594ef120SJeremy L Thompson pub fn create_multigrid_level<'b>( 16219df49d7eSJed Brown &self, 16229df49d7eSJed Brown p_mult_fine: &Vector, 16239df49d7eSJed Brown rstr_coarse: &ElemRestriction, 16249df49d7eSJed Brown basis_coarse: &Basis, 1625594ef120SJeremy L Thompson ) -> crate::Result<(Operator<'b>, Operator<'b>, Operator<'b>)> { 16269df49d7eSJed Brown let mut ptr_coarse = std::ptr::null_mut(); 16279df49d7eSJed Brown let mut ptr_prolong = std::ptr::null_mut(); 16289df49d7eSJed Brown let mut ptr_restrict = std::ptr::null_mut(); 16299df49d7eSJed Brown let ierr = unsafe { 16309df49d7eSJed Brown bind_ceed::CeedOperatorMultigridLevelCreate( 16319df49d7eSJed Brown self.op_core.ptr, 16329df49d7eSJed Brown p_mult_fine.ptr, 16339df49d7eSJed Brown rstr_coarse.ptr, 16349df49d7eSJed Brown basis_coarse.ptr, 16359df49d7eSJed Brown &mut ptr_coarse, 16369df49d7eSJed Brown &mut ptr_prolong, 16379df49d7eSJed Brown &mut ptr_restrict, 16389df49d7eSJed Brown ) 16399df49d7eSJed Brown }; 16401142270cSJeremy L Thompson self.op_core.check_error(ierr)?; 16411142270cSJeremy L Thompson let op_coarse = Operator::from_raw(ptr_coarse)?; 16421142270cSJeremy L Thompson let op_prolong = Operator::from_raw(ptr_prolong)?; 16431142270cSJeremy L Thompson let op_restrict = Operator::from_raw(ptr_restrict)?; 16449df49d7eSJed Brown Ok((op_coarse, op_prolong, op_restrict)) 16459df49d7eSJed Brown } 16469df49d7eSJed Brown 16479df49d7eSJed Brown /// Create a multigrid coarse Operator and level transfer Operators for a 16489df49d7eSJed Brown /// given Operator with a tensor basis for the active basis 16499df49d7eSJed Brown /// 16509df49d7eSJed Brown /// * `p_mult_fine` - Lvector multiplicity in parallel gather/scatter 16519df49d7eSJed Brown /// * `rstr_coarse` - Coarse grid restriction 16529df49d7eSJed Brown /// * `basis_coarse` - Coarse grid active vector basis 16539df49d7eSJed Brown /// * `interp_c_to_f` - Matrix for coarse to fine 16549df49d7eSJed Brown /// 16559df49d7eSJed Brown /// ``` 16569df49d7eSJed Brown /// # use libceed::prelude::*; 16574d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 16589df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 16599df49d7eSJed Brown /// let ne = 15; 16609df49d7eSJed Brown /// let p_coarse = 3; 16619df49d7eSJed Brown /// let p_fine = 5; 16629df49d7eSJed Brown /// let q = 6; 16639df49d7eSJed Brown /// let ndofs_coarse = p_coarse * ne - ne + 1; 16649df49d7eSJed Brown /// let ndofs_fine = p_fine * ne - ne + 1; 16659df49d7eSJed Brown /// 16669df49d7eSJed Brown /// // Vectors 16679df49d7eSJed Brown /// let x_array = (0..ne + 1) 166880a9ef05SNatalie Beams /// .map(|i| 2.0 * i as Scalar / ne as Scalar - 1.0) 166980a9ef05SNatalie Beams /// .collect::<Vec<Scalar>>(); 1670c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&x_array)?; 1671c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 16729df49d7eSJed Brown /// qdata.set_value(0.0); 1673c68be7a2SJeremy L Thompson /// let mut u_coarse = ceed.vector(ndofs_coarse)?; 16749df49d7eSJed Brown /// u_coarse.set_value(1.0); 1675c68be7a2SJeremy L Thompson /// let mut u_fine = ceed.vector(ndofs_fine)?; 16769df49d7eSJed Brown /// u_fine.set_value(1.0); 1677c68be7a2SJeremy L Thompson /// let mut v_coarse = ceed.vector(ndofs_coarse)?; 16789df49d7eSJed Brown /// v_coarse.set_value(0.0); 1679c68be7a2SJeremy L Thompson /// let mut v_fine = ceed.vector(ndofs_fine)?; 16809df49d7eSJed Brown /// v_fine.set_value(0.0); 1681c68be7a2SJeremy L Thompson /// let mut multiplicity = ceed.vector(ndofs_fine)?; 16829df49d7eSJed Brown /// multiplicity.set_value(1.0); 16839df49d7eSJed Brown /// 16849df49d7eSJed Brown /// // Restrictions 16859df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1686c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 16879df49d7eSJed Brown /// 16889df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 16899df49d7eSJed Brown /// for i in 0..ne { 16909df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 16919df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 16929df49d7eSJed Brown /// } 1693c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 16949df49d7eSJed Brown /// 16959df49d7eSJed Brown /// let mut indu_coarse: Vec<i32> = vec![0; p_coarse * ne]; 16969df49d7eSJed Brown /// for i in 0..ne { 16979df49d7eSJed Brown /// for j in 0..p_coarse { 16989df49d7eSJed Brown /// indu_coarse[p_coarse * i + j] = (i + j) as i32; 16999df49d7eSJed Brown /// } 17009df49d7eSJed Brown /// } 1701c68be7a2SJeremy L Thompson /// let ru_coarse = ceed.elem_restriction( 17029df49d7eSJed Brown /// ne, 17039df49d7eSJed Brown /// p_coarse, 17049df49d7eSJed Brown /// 1, 17059df49d7eSJed Brown /// 1, 17069df49d7eSJed Brown /// ndofs_coarse, 17079df49d7eSJed Brown /// MemType::Host, 17089df49d7eSJed Brown /// &indu_coarse, 1709c68be7a2SJeremy L Thompson /// )?; 17109df49d7eSJed Brown /// 17119df49d7eSJed Brown /// let mut indu_fine: Vec<i32> = vec![0; p_fine * ne]; 17129df49d7eSJed Brown /// for i in 0..ne { 17139df49d7eSJed Brown /// for j in 0..p_fine { 17149df49d7eSJed Brown /// indu_fine[p_fine * i + j] = (i + j) as i32; 17159df49d7eSJed Brown /// } 17169df49d7eSJed Brown /// } 1717c68be7a2SJeremy L Thompson /// let ru_fine = ceed.elem_restriction(ne, p_fine, 1, 1, ndofs_fine, MemType::Host, &indu_fine)?; 17189df49d7eSJed Brown /// 17199df49d7eSJed Brown /// // Bases 1720c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1721c68be7a2SJeremy L Thompson /// let bu_coarse = ceed.basis_tensor_H1_Lagrange(1, 1, p_coarse, q, QuadMode::Gauss)?; 1722c68be7a2SJeremy L Thompson /// let bu_fine = ceed.basis_tensor_H1_Lagrange(1, 1, p_fine, q, QuadMode::Gauss)?; 17239df49d7eSJed Brown /// 17249df49d7eSJed Brown /// // Build quadrature data 1725c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1726c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1727c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1728c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1729c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1730c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 17319df49d7eSJed Brown /// 17329df49d7eSJed Brown /// // Mass operator 1733c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 17349df49d7eSJed Brown /// let op_mass_fine = ceed 1735c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1736c68be7a2SJeremy L Thompson /// .field("u", &ru_fine, &bu_fine, VectorOpt::Active)? 1737c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1738c68be7a2SJeremy L Thompson /// .field("v", &ru_fine, &bu_fine, VectorOpt::Active)?; 17399df49d7eSJed Brown /// 17409df49d7eSJed Brown /// // Multigrid setup 174180a9ef05SNatalie Beams /// let mut interp_c_to_f: Vec<Scalar> = vec![0.; p_coarse * p_fine]; 17429df49d7eSJed Brown /// { 1743c68be7a2SJeremy L Thompson /// let mut coarse = ceed.vector(p_coarse)?; 1744c68be7a2SJeremy L Thompson /// let mut fine = ceed.vector(p_fine)?; 1745c68be7a2SJeremy L Thompson /// let basis_c_to_f = 1746c68be7a2SJeremy L Thompson /// ceed.basis_tensor_H1_Lagrange(1, 1, p_coarse, p_fine, QuadMode::GaussLobatto)?; 17479df49d7eSJed Brown /// for i in 0..p_coarse { 17489df49d7eSJed Brown /// coarse.set_value(0.0); 17499df49d7eSJed Brown /// { 1750e78171edSJeremy L Thompson /// let mut array = coarse.view_mut()?; 17519df49d7eSJed Brown /// array[i] = 1.; 17529df49d7eSJed Brown /// } 1753c68be7a2SJeremy L Thompson /// basis_c_to_f.apply( 17549df49d7eSJed Brown /// 1, 17559df49d7eSJed Brown /// TransposeMode::NoTranspose, 17569df49d7eSJed Brown /// EvalMode::Interp, 17579df49d7eSJed Brown /// &coarse, 17589df49d7eSJed Brown /// &mut fine, 1759c68be7a2SJeremy L Thompson /// )?; 1760e78171edSJeremy L Thompson /// let array = fine.view()?; 17619df49d7eSJed Brown /// for j in 0..p_fine { 17629df49d7eSJed Brown /// interp_c_to_f[j * p_coarse + i] = array[j]; 17639df49d7eSJed Brown /// } 17649df49d7eSJed Brown /// } 17659df49d7eSJed Brown /// } 1766c68be7a2SJeremy L Thompson /// let (op_mass_coarse, op_prolong, op_restrict) = op_mass_fine.create_multigrid_level_tensor_H1( 1767c68be7a2SJeremy L Thompson /// &multiplicity, 1768c68be7a2SJeremy L Thompson /// &ru_coarse, 1769c68be7a2SJeremy L Thompson /// &bu_coarse, 1770c68be7a2SJeremy L Thompson /// &interp_c_to_f, 1771c68be7a2SJeremy L Thompson /// )?; 17729df49d7eSJed Brown /// 17739df49d7eSJed Brown /// // Coarse problem 17749df49d7eSJed Brown /// u_coarse.set_value(1.0); 1775c68be7a2SJeremy L Thompson /// op_mass_coarse.apply(&u_coarse, &mut v_coarse)?; 17769df49d7eSJed Brown /// 17779df49d7eSJed Brown /// // Check 1778e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 17799df49d7eSJed Brown /// assert!( 178080a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 17819df49d7eSJed Brown /// "Incorrect interval length computed" 17829df49d7eSJed Brown /// ); 17839df49d7eSJed Brown /// 17849df49d7eSJed Brown /// // Prolong 1785c68be7a2SJeremy L Thompson /// op_prolong.apply(&u_coarse, &mut u_fine)?; 17869df49d7eSJed Brown /// 17879df49d7eSJed Brown /// // Fine problem 1788c68be7a2SJeremy L Thompson /// op_mass_fine.apply(&u_fine, &mut v_fine)?; 17899df49d7eSJed Brown /// 17909df49d7eSJed Brown /// // Check 1791e78171edSJeremy L Thompson /// let sum: Scalar = v_fine.view()?.iter().sum(); 17929df49d7eSJed Brown /// assert!( 179380a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 17949df49d7eSJed Brown /// "Incorrect interval length computed" 17959df49d7eSJed Brown /// ); 17969df49d7eSJed Brown /// 17979df49d7eSJed Brown /// // Restrict 1798c68be7a2SJeremy L Thompson /// op_restrict.apply(&v_fine, &mut v_coarse)?; 17999df49d7eSJed Brown /// 18009df49d7eSJed Brown /// // Check 1801e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 18029df49d7eSJed Brown /// assert!( 180380a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 18049df49d7eSJed Brown /// "Incorrect interval length computed" 18059df49d7eSJed Brown /// ); 1806c68be7a2SJeremy L Thompson /// # Ok(()) 1807c68be7a2SJeremy L Thompson /// # } 18089df49d7eSJed Brown /// ``` 1809594ef120SJeremy L Thompson pub fn create_multigrid_level_tensor_H1<'b>( 18109df49d7eSJed Brown &self, 18119df49d7eSJed Brown p_mult_fine: &Vector, 18129df49d7eSJed Brown rstr_coarse: &ElemRestriction, 18139df49d7eSJed Brown basis_coarse: &Basis, 181480a9ef05SNatalie Beams interpCtoF: &Vec<Scalar>, 1815594ef120SJeremy L Thompson ) -> crate::Result<(Operator<'b>, Operator<'b>, Operator<'b>)> { 18169df49d7eSJed Brown let mut ptr_coarse = std::ptr::null_mut(); 18179df49d7eSJed Brown let mut ptr_prolong = std::ptr::null_mut(); 18189df49d7eSJed Brown let mut ptr_restrict = std::ptr::null_mut(); 18199df49d7eSJed Brown let ierr = unsafe { 18209df49d7eSJed Brown bind_ceed::CeedOperatorMultigridLevelCreateTensorH1( 18219df49d7eSJed Brown self.op_core.ptr, 18229df49d7eSJed Brown p_mult_fine.ptr, 18239df49d7eSJed Brown rstr_coarse.ptr, 18249df49d7eSJed Brown basis_coarse.ptr, 18259df49d7eSJed Brown interpCtoF.as_ptr(), 18269df49d7eSJed Brown &mut ptr_coarse, 18279df49d7eSJed Brown &mut ptr_prolong, 18289df49d7eSJed Brown &mut ptr_restrict, 18299df49d7eSJed Brown ) 18309df49d7eSJed Brown }; 18311142270cSJeremy L Thompson self.op_core.check_error(ierr)?; 18321142270cSJeremy L Thompson let op_coarse = Operator::from_raw(ptr_coarse)?; 18331142270cSJeremy L Thompson let op_prolong = Operator::from_raw(ptr_prolong)?; 18341142270cSJeremy L Thompson let op_restrict = Operator::from_raw(ptr_restrict)?; 18359df49d7eSJed Brown Ok((op_coarse, op_prolong, op_restrict)) 18369df49d7eSJed Brown } 18379df49d7eSJed Brown 18389df49d7eSJed Brown /// Create a multigrid coarse Operator and level transfer Operators for a 18399df49d7eSJed Brown /// given Operator with a non-tensor basis for the active basis 18409df49d7eSJed Brown /// 18419df49d7eSJed Brown /// * `p_mult_fine` - Lvector multiplicity in parallel gather/scatter 18429df49d7eSJed Brown /// * `rstr_coarse` - Coarse grid restriction 18439df49d7eSJed Brown /// * `basis_coarse` - Coarse grid active vector basis 18449df49d7eSJed Brown /// * `interp_c_to_f` - Matrix for coarse to fine 18459df49d7eSJed Brown /// 18469df49d7eSJed Brown /// ``` 18479df49d7eSJed Brown /// # use libceed::prelude::*; 18484d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 18499df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 18509df49d7eSJed Brown /// let ne = 15; 18519df49d7eSJed Brown /// let p_coarse = 3; 18529df49d7eSJed Brown /// let p_fine = 5; 18539df49d7eSJed Brown /// let q = 6; 18549df49d7eSJed Brown /// let ndofs_coarse = p_coarse * ne - ne + 1; 18559df49d7eSJed Brown /// let ndofs_fine = p_fine * ne - ne + 1; 18569df49d7eSJed Brown /// 18579df49d7eSJed Brown /// // Vectors 18589df49d7eSJed Brown /// let x_array = (0..ne + 1) 185980a9ef05SNatalie Beams /// .map(|i| 2.0 * i as Scalar / ne as Scalar - 1.0) 186080a9ef05SNatalie Beams /// .collect::<Vec<Scalar>>(); 1861c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&x_array)?; 1862c68be7a2SJeremy L Thompson /// let mut qdata = ceed.vector(ne * q)?; 18639df49d7eSJed Brown /// qdata.set_value(0.0); 1864c68be7a2SJeremy L Thompson /// let mut u_coarse = ceed.vector(ndofs_coarse)?; 18659df49d7eSJed Brown /// u_coarse.set_value(1.0); 1866c68be7a2SJeremy L Thompson /// let mut u_fine = ceed.vector(ndofs_fine)?; 18679df49d7eSJed Brown /// u_fine.set_value(1.0); 1868c68be7a2SJeremy L Thompson /// let mut v_coarse = ceed.vector(ndofs_coarse)?; 18699df49d7eSJed Brown /// v_coarse.set_value(0.0); 1870c68be7a2SJeremy L Thompson /// let mut v_fine = ceed.vector(ndofs_fine)?; 18719df49d7eSJed Brown /// v_fine.set_value(0.0); 1872c68be7a2SJeremy L Thompson /// let mut multiplicity = ceed.vector(ndofs_fine)?; 18739df49d7eSJed Brown /// multiplicity.set_value(1.0); 18749df49d7eSJed Brown /// 18759df49d7eSJed Brown /// // Restrictions 18769df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 1877c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 18789df49d7eSJed Brown /// 18799df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 18809df49d7eSJed Brown /// for i in 0..ne { 18819df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 18829df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 18839df49d7eSJed Brown /// } 1884c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 18859df49d7eSJed Brown /// 18869df49d7eSJed Brown /// let mut indu_coarse: Vec<i32> = vec![0; p_coarse * ne]; 18879df49d7eSJed Brown /// for i in 0..ne { 18889df49d7eSJed Brown /// for j in 0..p_coarse { 18899df49d7eSJed Brown /// indu_coarse[p_coarse * i + j] = (i + j) as i32; 18909df49d7eSJed Brown /// } 18919df49d7eSJed Brown /// } 1892c68be7a2SJeremy L Thompson /// let ru_coarse = ceed.elem_restriction( 18939df49d7eSJed Brown /// ne, 18949df49d7eSJed Brown /// p_coarse, 18959df49d7eSJed Brown /// 1, 18969df49d7eSJed Brown /// 1, 18979df49d7eSJed Brown /// ndofs_coarse, 18989df49d7eSJed Brown /// MemType::Host, 18999df49d7eSJed Brown /// &indu_coarse, 1900c68be7a2SJeremy L Thompson /// )?; 19019df49d7eSJed Brown /// 19029df49d7eSJed Brown /// let mut indu_fine: Vec<i32> = vec![0; p_fine * ne]; 19039df49d7eSJed Brown /// for i in 0..ne { 19049df49d7eSJed Brown /// for j in 0..p_fine { 19059df49d7eSJed Brown /// indu_fine[p_fine * i + j] = (i + j) as i32; 19069df49d7eSJed Brown /// } 19079df49d7eSJed Brown /// } 1908c68be7a2SJeremy L Thompson /// let ru_fine = ceed.elem_restriction(ne, p_fine, 1, 1, ndofs_fine, MemType::Host, &indu_fine)?; 19099df49d7eSJed Brown /// 19109df49d7eSJed Brown /// // Bases 1911c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 1912c68be7a2SJeremy L Thompson /// let bu_coarse = ceed.basis_tensor_H1_Lagrange(1, 1, p_coarse, q, QuadMode::Gauss)?; 1913c68be7a2SJeremy L Thompson /// let bu_fine = ceed.basis_tensor_H1_Lagrange(1, 1, p_fine, q, QuadMode::Gauss)?; 19149df49d7eSJed Brown /// 19159df49d7eSJed Brown /// // Build quadrature data 1916c68be7a2SJeremy L Thompson /// let qf_build = ceed.q_function_interior_by_name("Mass1DBuild")?; 1917c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build, QFunctionOpt::None, QFunctionOpt::None)? 1918c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 1919c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 1920c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 1921c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata)?; 19229df49d7eSJed Brown /// 19239df49d7eSJed Brown /// // Mass operator 1924c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 19259df49d7eSJed Brown /// let op_mass_fine = ceed 1926c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 1927c68be7a2SJeremy L Thompson /// .field("u", &ru_fine, &bu_fine, VectorOpt::Active)? 1928c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata)? 1929c68be7a2SJeremy L Thompson /// .field("v", &ru_fine, &bu_fine, VectorOpt::Active)?; 19309df49d7eSJed Brown /// 19319df49d7eSJed Brown /// // Multigrid setup 193280a9ef05SNatalie Beams /// let mut interp_c_to_f: Vec<Scalar> = vec![0.; p_coarse * p_fine]; 19339df49d7eSJed Brown /// { 1934c68be7a2SJeremy L Thompson /// let mut coarse = ceed.vector(p_coarse)?; 1935c68be7a2SJeremy L Thompson /// let mut fine = ceed.vector(p_fine)?; 1936c68be7a2SJeremy L Thompson /// let basis_c_to_f = 1937c68be7a2SJeremy L Thompson /// ceed.basis_tensor_H1_Lagrange(1, 1, p_coarse, p_fine, QuadMode::GaussLobatto)?; 19389df49d7eSJed Brown /// for i in 0..p_coarse { 19399df49d7eSJed Brown /// coarse.set_value(0.0); 19409df49d7eSJed Brown /// { 1941e78171edSJeremy L Thompson /// let mut array = coarse.view_mut()?; 19429df49d7eSJed Brown /// array[i] = 1.; 19439df49d7eSJed Brown /// } 1944c68be7a2SJeremy L Thompson /// basis_c_to_f.apply( 19459df49d7eSJed Brown /// 1, 19469df49d7eSJed Brown /// TransposeMode::NoTranspose, 19479df49d7eSJed Brown /// EvalMode::Interp, 19489df49d7eSJed Brown /// &coarse, 19499df49d7eSJed Brown /// &mut fine, 1950c68be7a2SJeremy L Thompson /// )?; 1951e78171edSJeremy L Thompson /// let array = fine.view()?; 19529df49d7eSJed Brown /// for j in 0..p_fine { 19539df49d7eSJed Brown /// interp_c_to_f[j * p_coarse + i] = array[j]; 19549df49d7eSJed Brown /// } 19559df49d7eSJed Brown /// } 19569df49d7eSJed Brown /// } 1957c68be7a2SJeremy L Thompson /// let (op_mass_coarse, op_prolong, op_restrict) = op_mass_fine.create_multigrid_level_H1( 1958c68be7a2SJeremy L Thompson /// &multiplicity, 1959c68be7a2SJeremy L Thompson /// &ru_coarse, 1960c68be7a2SJeremy L Thompson /// &bu_coarse, 1961c68be7a2SJeremy L Thompson /// &interp_c_to_f, 1962c68be7a2SJeremy L Thompson /// )?; 19639df49d7eSJed Brown /// 19649df49d7eSJed Brown /// // Coarse problem 19659df49d7eSJed Brown /// u_coarse.set_value(1.0); 1966c68be7a2SJeremy L Thompson /// op_mass_coarse.apply(&u_coarse, &mut v_coarse)?; 19679df49d7eSJed Brown /// 19689df49d7eSJed Brown /// // Check 1969e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 19709df49d7eSJed Brown /// assert!( 197180a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 19729df49d7eSJed Brown /// "Incorrect interval length computed" 19739df49d7eSJed Brown /// ); 19749df49d7eSJed Brown /// 19759df49d7eSJed Brown /// // Prolong 1976c68be7a2SJeremy L Thompson /// op_prolong.apply(&u_coarse, &mut u_fine)?; 19779df49d7eSJed Brown /// 19789df49d7eSJed Brown /// // Fine problem 1979c68be7a2SJeremy L Thompson /// op_mass_fine.apply(&u_fine, &mut v_fine)?; 19809df49d7eSJed Brown /// 19819df49d7eSJed Brown /// // Check 1982e78171edSJeremy L Thompson /// let sum: Scalar = v_fine.view()?.iter().sum(); 19839df49d7eSJed Brown /// assert!( 198480a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 19859df49d7eSJed Brown /// "Incorrect interval length computed" 19869df49d7eSJed Brown /// ); 19879df49d7eSJed Brown /// 19889df49d7eSJed Brown /// // Restrict 1989c68be7a2SJeremy L Thompson /// op_restrict.apply(&v_fine, &mut v_coarse)?; 19909df49d7eSJed Brown /// 19919df49d7eSJed Brown /// // Check 1992e78171edSJeremy L Thompson /// let sum: Scalar = v_coarse.view()?.iter().sum(); 19939df49d7eSJed Brown /// assert!( 199480a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 19959df49d7eSJed Brown /// "Incorrect interval length computed" 19969df49d7eSJed Brown /// ); 1997c68be7a2SJeremy L Thompson /// # Ok(()) 1998c68be7a2SJeremy L Thompson /// # } 19999df49d7eSJed Brown /// ``` 2000594ef120SJeremy L Thompson pub fn create_multigrid_level_H1<'b>( 20019df49d7eSJed Brown &self, 20029df49d7eSJed Brown p_mult_fine: &Vector, 20039df49d7eSJed Brown rstr_coarse: &ElemRestriction, 20049df49d7eSJed Brown basis_coarse: &Basis, 200580a9ef05SNatalie Beams interpCtoF: &[Scalar], 2006594ef120SJeremy L Thompson ) -> crate::Result<(Operator<'b>, Operator<'b>, Operator<'b>)> { 20079df49d7eSJed Brown let mut ptr_coarse = std::ptr::null_mut(); 20089df49d7eSJed Brown let mut ptr_prolong = std::ptr::null_mut(); 20099df49d7eSJed Brown let mut ptr_restrict = std::ptr::null_mut(); 20109df49d7eSJed Brown let ierr = unsafe { 20119df49d7eSJed Brown bind_ceed::CeedOperatorMultigridLevelCreateH1( 20129df49d7eSJed Brown self.op_core.ptr, 20139df49d7eSJed Brown p_mult_fine.ptr, 20149df49d7eSJed Brown rstr_coarse.ptr, 20159df49d7eSJed Brown basis_coarse.ptr, 20169df49d7eSJed Brown interpCtoF.as_ptr(), 20179df49d7eSJed Brown &mut ptr_coarse, 20189df49d7eSJed Brown &mut ptr_prolong, 20199df49d7eSJed Brown &mut ptr_restrict, 20209df49d7eSJed Brown ) 20219df49d7eSJed Brown }; 20221142270cSJeremy L Thompson self.op_core.check_error(ierr)?; 20231142270cSJeremy L Thompson let op_coarse = Operator::from_raw(ptr_coarse)?; 20241142270cSJeremy L Thompson let op_prolong = Operator::from_raw(ptr_prolong)?; 20251142270cSJeremy L Thompson let op_restrict = Operator::from_raw(ptr_restrict)?; 20269df49d7eSJed Brown Ok((op_coarse, op_prolong, op_restrict)) 20279df49d7eSJed Brown } 20289df49d7eSJed Brown } 20299df49d7eSJed Brown 20309df49d7eSJed Brown // ----------------------------------------------------------------------------- 20319df49d7eSJed Brown // Composite Operator 20329df49d7eSJed Brown // ----------------------------------------------------------------------------- 20339df49d7eSJed Brown impl<'a> CompositeOperator<'a> { 20349df49d7eSJed Brown // Constructor 2035594ef120SJeremy L Thompson pub fn create(ceed: &crate::Ceed) -> crate::Result<Self> { 20369df49d7eSJed Brown let mut ptr = std::ptr::null_mut(); 20379df49d7eSJed Brown let ierr = unsafe { bind_ceed::CeedCompositeOperatorCreate(ceed.ptr, &mut ptr) }; 20389df49d7eSJed Brown ceed.check_error(ierr)?; 20399df49d7eSJed Brown Ok(Self { 20401142270cSJeremy L Thompson op_core: OperatorCore { 20411142270cSJeremy L Thompson ptr, 20421142270cSJeremy L Thompson _lifeline: PhantomData, 20431142270cSJeremy L Thompson }, 20449df49d7eSJed Brown }) 20459df49d7eSJed Brown } 20469df49d7eSJed Brown 20479df49d7eSJed Brown /// Apply Operator to a vector 20489df49d7eSJed Brown /// 20499df49d7eSJed Brown /// * `input` - Input Vector 20509df49d7eSJed Brown /// * `output` - Output Vector 20519df49d7eSJed Brown /// 20529df49d7eSJed Brown /// ``` 20539df49d7eSJed Brown /// # use libceed::prelude::*; 20544d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 20559df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 20569df49d7eSJed Brown /// let ne = 4; 20579df49d7eSJed Brown /// let p = 3; 20589df49d7eSJed Brown /// let q = 4; 20599df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 20609df49d7eSJed Brown /// 20619df49d7eSJed Brown /// // Vectors 2062c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 2063c68be7a2SJeremy L Thompson /// let mut qdata_mass = ceed.vector(ne * q)?; 20649df49d7eSJed Brown /// qdata_mass.set_value(0.0); 2065c68be7a2SJeremy L Thompson /// let mut qdata_diff = ceed.vector(ne * q)?; 20669df49d7eSJed Brown /// qdata_diff.set_value(0.0); 2067c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ndofs)?; 20689df49d7eSJed Brown /// u.set_value(1.0); 2069c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 20709df49d7eSJed Brown /// v.set_value(0.0); 20719df49d7eSJed Brown /// 20729df49d7eSJed Brown /// // Restrictions 20739df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 20749df49d7eSJed Brown /// for i in 0..ne { 20759df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 20769df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 20779df49d7eSJed Brown /// } 2078c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 20799df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 20809df49d7eSJed Brown /// for i in 0..ne { 20819df49d7eSJed Brown /// indu[p * i + 0] = i as i32; 20829df49d7eSJed Brown /// indu[p * i + 1] = (i + 1) as i32; 20839df49d7eSJed Brown /// indu[p * i + 2] = (i + 2) as i32; 20849df49d7eSJed Brown /// } 2085c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, 3, 1, 1, ndofs, MemType::Host, &indu)?; 20869df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 2087c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 20889df49d7eSJed Brown /// 20899df49d7eSJed Brown /// // Bases 2090c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 2091c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 20929df49d7eSJed Brown /// 20939df49d7eSJed Brown /// // Build quadrature data 2094c68be7a2SJeremy L Thompson /// let qf_build_mass = ceed.q_function_interior_by_name("Mass1DBuild")?; 2095c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build_mass, QFunctionOpt::None, QFunctionOpt::None)? 2096c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 2097c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 2098c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 2099c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata_mass)?; 21009df49d7eSJed Brown /// 2101c68be7a2SJeremy L Thompson /// let qf_build_diff = ceed.q_function_interior_by_name("Poisson1DBuild")?; 2102c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build_diff, QFunctionOpt::None, QFunctionOpt::None)? 2103c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 2104c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 2105c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 2106c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata_diff)?; 21079df49d7eSJed Brown /// 21089df49d7eSJed Brown /// // Application operator 2109c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 21109df49d7eSJed Brown /// let op_mass = ceed 2111c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 2112c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 2113c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_mass)? 2114c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 21159df49d7eSJed Brown /// 2116c68be7a2SJeremy L Thompson /// let qf_diff = ceed.q_function_interior_by_name("Poisson1DApply")?; 21179df49d7eSJed Brown /// let op_diff = ceed 2118c68be7a2SJeremy L Thompson /// .operator(&qf_diff, QFunctionOpt::None, QFunctionOpt::None)? 2119c68be7a2SJeremy L Thompson /// .field("du", &ru, &bu, VectorOpt::Active)? 2120c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_diff)? 2121c68be7a2SJeremy L Thompson /// .field("dv", &ru, &bu, VectorOpt::Active)?; 21229df49d7eSJed Brown /// 21239df49d7eSJed Brown /// let op_composite = ceed 2124c68be7a2SJeremy L Thompson /// .composite_operator()? 2125c68be7a2SJeremy L Thompson /// .sub_operator(&op_mass)? 2126c68be7a2SJeremy L Thompson /// .sub_operator(&op_diff)?; 21279df49d7eSJed Brown /// 21289df49d7eSJed Brown /// v.set_value(0.0); 2129c68be7a2SJeremy L Thompson /// op_composite.apply(&u, &mut v)?; 21309df49d7eSJed Brown /// 21319df49d7eSJed Brown /// // Check 2132e78171edSJeremy L Thompson /// let sum: Scalar = v.view()?.iter().sum(); 21339df49d7eSJed Brown /// assert!( 213480a9ef05SNatalie Beams /// (sum - 2.0).abs() < 10.0 * libceed::EPSILON, 21359df49d7eSJed Brown /// "Incorrect interval length computed" 21369df49d7eSJed Brown /// ); 2137c68be7a2SJeremy L Thompson /// # Ok(()) 2138c68be7a2SJeremy L Thompson /// # } 21399df49d7eSJed Brown /// ``` 21409df49d7eSJed Brown pub fn apply(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 21419df49d7eSJed Brown self.op_core.apply(input, output) 21429df49d7eSJed Brown } 21439df49d7eSJed Brown 21449df49d7eSJed Brown /// Apply Operator to a vector and add result to output vector 21459df49d7eSJed Brown /// 21469df49d7eSJed Brown /// * `input` - Input Vector 21479df49d7eSJed Brown /// * `output` - Output Vector 21489df49d7eSJed Brown /// 21499df49d7eSJed Brown /// ``` 21509df49d7eSJed Brown /// # use libceed::prelude::*; 21514d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 21529df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 21539df49d7eSJed Brown /// let ne = 4; 21549df49d7eSJed Brown /// let p = 3; 21559df49d7eSJed Brown /// let q = 4; 21569df49d7eSJed Brown /// let ndofs = p * ne - ne + 1; 21579df49d7eSJed Brown /// 21589df49d7eSJed Brown /// // Vectors 2159c68be7a2SJeremy L Thompson /// let x = ceed.vector_from_slice(&[-1., -0.5, 0.0, 0.5, 1.0])?; 2160c68be7a2SJeremy L Thompson /// let mut qdata_mass = ceed.vector(ne * q)?; 21619df49d7eSJed Brown /// qdata_mass.set_value(0.0); 2162c68be7a2SJeremy L Thompson /// let mut qdata_diff = ceed.vector(ne * q)?; 21639df49d7eSJed Brown /// qdata_diff.set_value(0.0); 2164c68be7a2SJeremy L Thompson /// let mut u = ceed.vector(ndofs)?; 21659df49d7eSJed Brown /// u.set_value(1.0); 2166c68be7a2SJeremy L Thompson /// let mut v = ceed.vector(ndofs)?; 21679df49d7eSJed Brown /// v.set_value(0.0); 21689df49d7eSJed Brown /// 21699df49d7eSJed Brown /// // Restrictions 21709df49d7eSJed Brown /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 21719df49d7eSJed Brown /// for i in 0..ne { 21729df49d7eSJed Brown /// indx[2 * i + 0] = i as i32; 21739df49d7eSJed Brown /// indx[2 * i + 1] = (i + 1) as i32; 21749df49d7eSJed Brown /// } 2175c68be7a2SJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 21769df49d7eSJed Brown /// let mut indu: Vec<i32> = vec![0; p * ne]; 21779df49d7eSJed Brown /// for i in 0..ne { 21789df49d7eSJed Brown /// indu[p * i + 0] = i as i32; 21799df49d7eSJed Brown /// indu[p * i + 1] = (i + 1) as i32; 21809df49d7eSJed Brown /// indu[p * i + 2] = (i + 2) as i32; 21819df49d7eSJed Brown /// } 2182c68be7a2SJeremy L Thompson /// let ru = ceed.elem_restriction(ne, 3, 1, 1, ndofs, MemType::Host, &indu)?; 21839df49d7eSJed Brown /// let strides: [i32; 3] = [1, q as i32, q as i32]; 2184c68be7a2SJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 21859df49d7eSJed Brown /// 21869df49d7eSJed Brown /// // Bases 2187c68be7a2SJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 2188c68be7a2SJeremy L Thompson /// let bu = ceed.basis_tensor_H1_Lagrange(1, 1, p, q, QuadMode::Gauss)?; 21899df49d7eSJed Brown /// 21909df49d7eSJed Brown /// // Build quadrature data 2191c68be7a2SJeremy L Thompson /// let qf_build_mass = ceed.q_function_interior_by_name("Mass1DBuild")?; 2192c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build_mass, QFunctionOpt::None, QFunctionOpt::None)? 2193c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 2194c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 2195c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 2196c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata_mass)?; 21979df49d7eSJed Brown /// 2198c68be7a2SJeremy L Thompson /// let qf_build_diff = ceed.q_function_interior_by_name("Poisson1DBuild")?; 2199c68be7a2SJeremy L Thompson /// ceed.operator(&qf_build_diff, QFunctionOpt::None, QFunctionOpt::None)? 2200c68be7a2SJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 2201c68be7a2SJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 2202c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)? 2203c68be7a2SJeremy L Thompson /// .apply(&x, &mut qdata_diff)?; 22049df49d7eSJed Brown /// 22059df49d7eSJed Brown /// // Application operator 2206c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 22079df49d7eSJed Brown /// let op_mass = ceed 2208c68be7a2SJeremy L Thompson /// .operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)? 2209c68be7a2SJeremy L Thompson /// .field("u", &ru, &bu, VectorOpt::Active)? 2210c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_mass)? 2211c68be7a2SJeremy L Thompson /// .field("v", &ru, &bu, VectorOpt::Active)?; 22129df49d7eSJed Brown /// 2213c68be7a2SJeremy L Thompson /// let qf_diff = ceed.q_function_interior_by_name("Poisson1DApply")?; 22149df49d7eSJed Brown /// let op_diff = ceed 2215c68be7a2SJeremy L Thompson /// .operator(&qf_diff, QFunctionOpt::None, QFunctionOpt::None)? 2216c68be7a2SJeremy L Thompson /// .field("du", &ru, &bu, VectorOpt::Active)? 2217c68be7a2SJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, &qdata_diff)? 2218c68be7a2SJeremy L Thompson /// .field("dv", &ru, &bu, VectorOpt::Active)?; 22199df49d7eSJed Brown /// 22209df49d7eSJed Brown /// let op_composite = ceed 2221c68be7a2SJeremy L Thompson /// .composite_operator()? 2222c68be7a2SJeremy L Thompson /// .sub_operator(&op_mass)? 2223c68be7a2SJeremy L Thompson /// .sub_operator(&op_diff)?; 22249df49d7eSJed Brown /// 22259df49d7eSJed Brown /// v.set_value(1.0); 2226c68be7a2SJeremy L Thompson /// op_composite.apply_add(&u, &mut v)?; 22279df49d7eSJed Brown /// 22289df49d7eSJed Brown /// // Check 2229e78171edSJeremy L Thompson /// let sum: Scalar = v.view()?.iter().sum(); 22309df49d7eSJed Brown /// assert!( 223180a9ef05SNatalie Beams /// (sum - (2.0 + ndofs as Scalar)).abs() < 10.0 * libceed::EPSILON, 22329df49d7eSJed Brown /// "Incorrect interval length computed" 22339df49d7eSJed Brown /// ); 2234c68be7a2SJeremy L Thompson /// # Ok(()) 2235c68be7a2SJeremy L Thompson /// # } 22369df49d7eSJed Brown /// ``` 22379df49d7eSJed Brown pub fn apply_add(&self, input: &Vector, output: &mut Vector) -> crate::Result<i32> { 22389df49d7eSJed Brown self.op_core.apply_add(input, output) 22399df49d7eSJed Brown } 22409df49d7eSJed Brown 22419df49d7eSJed Brown /// Add a sub-Operator to a Composite Operator 22429df49d7eSJed Brown /// 22439df49d7eSJed Brown /// * `subop` - Sub-Operator 22449df49d7eSJed Brown /// 22459df49d7eSJed Brown /// ``` 22469df49d7eSJed Brown /// # use libceed::prelude::*; 22474d27c890SJeremy L Thompson /// # fn main() -> libceed::Result<()> { 22489df49d7eSJed Brown /// # let ceed = libceed::Ceed::default_init(); 2249c68be7a2SJeremy L Thompson /// let mut op = ceed.composite_operator()?; 22509df49d7eSJed Brown /// 2251c68be7a2SJeremy L Thompson /// let qf_mass = ceed.q_function_interior_by_name("MassApply")?; 2252c68be7a2SJeremy L Thompson /// let op_mass = ceed.operator(&qf_mass, QFunctionOpt::None, QFunctionOpt::None)?; 2253c68be7a2SJeremy L Thompson /// op = op.sub_operator(&op_mass)?; 22549df49d7eSJed Brown /// 2255c68be7a2SJeremy L Thompson /// let qf_diff = ceed.q_function_interior_by_name("Poisson1DApply")?; 2256c68be7a2SJeremy L Thompson /// let op_diff = ceed.operator(&qf_diff, QFunctionOpt::None, QFunctionOpt::None)?; 2257c68be7a2SJeremy L Thompson /// op = op.sub_operator(&op_diff)?; 2258c68be7a2SJeremy L Thompson /// # Ok(()) 2259c68be7a2SJeremy L Thompson /// # } 22609df49d7eSJed Brown /// ``` 22619df49d7eSJed Brown #[allow(unused_mut)] 22629df49d7eSJed Brown pub fn sub_operator(mut self, subop: &Operator) -> crate::Result<Self> { 22639df49d7eSJed Brown let ierr = 22649df49d7eSJed Brown unsafe { bind_ceed::CeedCompositeOperatorAddSub(self.op_core.ptr, subop.op_core.ptr) }; 22651142270cSJeremy L Thompson self.op_core.check_error(ierr)?; 22669df49d7eSJed Brown Ok(self) 22679df49d7eSJed Brown } 22689df49d7eSJed Brown 22696f97ff0aSJeremy L Thompson /// Check if CompositeOperator is setup correctly 22706f97ff0aSJeremy L Thompson /// 22716f97ff0aSJeremy L Thompson /// ``` 22726f97ff0aSJeremy L Thompson /// # use libceed::prelude::*; 22736f97ff0aSJeremy L Thompson /// # fn main() -> libceed::Result<()> { 22746f97ff0aSJeremy L Thompson /// # let ceed = libceed::Ceed::default_init(); 22756f97ff0aSJeremy L Thompson /// let ne = 4; 22766f97ff0aSJeremy L Thompson /// let p = 3; 22776f97ff0aSJeremy L Thompson /// let q = 4; 22786f97ff0aSJeremy L Thompson /// let ndofs = p * ne - ne + 1; 22796f97ff0aSJeremy L Thompson /// 22806f97ff0aSJeremy L Thompson /// // Restrictions 22816f97ff0aSJeremy L Thompson /// let mut indx: Vec<i32> = vec![0; 2 * ne]; 22826f97ff0aSJeremy L Thompson /// for i in 0..ne { 22836f97ff0aSJeremy L Thompson /// indx[2 * i + 0] = i as i32; 22846f97ff0aSJeremy L Thompson /// indx[2 * i + 1] = (i + 1) as i32; 22856f97ff0aSJeremy L Thompson /// } 22866f97ff0aSJeremy L Thompson /// let rx = ceed.elem_restriction(ne, 2, 1, 1, ne + 1, MemType::Host, &indx)?; 22876f97ff0aSJeremy L Thompson /// let strides: [i32; 3] = [1, q as i32, q as i32]; 22886f97ff0aSJeremy L Thompson /// let rq = ceed.strided_elem_restriction(ne, q, 1, q * ne, strides)?; 22896f97ff0aSJeremy L Thompson /// 22906f97ff0aSJeremy L Thompson /// // Bases 22916f97ff0aSJeremy L Thompson /// let bx = ceed.basis_tensor_H1_Lagrange(1, 1, 2, q, QuadMode::Gauss)?; 22926f97ff0aSJeremy L Thompson /// 22936f97ff0aSJeremy L Thompson /// // Build quadrature data 22946f97ff0aSJeremy L Thompson /// let qf_build_mass = ceed.q_function_interior_by_name("Mass1DBuild")?; 22956f97ff0aSJeremy L Thompson /// let op_build_mass = ceed 22966f97ff0aSJeremy L Thompson /// .operator(&qf_build_mass, QFunctionOpt::None, QFunctionOpt::None)? 22976f97ff0aSJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 22986f97ff0aSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 22996f97ff0aSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 23006f97ff0aSJeremy L Thompson /// 23016f97ff0aSJeremy L Thompson /// let qf_build_diff = ceed.q_function_interior_by_name("Poisson1DBuild")?; 23026f97ff0aSJeremy L Thompson /// let op_build_diff = ceed 23036f97ff0aSJeremy L Thompson /// .operator(&qf_build_diff, QFunctionOpt::None, QFunctionOpt::None)? 23046f97ff0aSJeremy L Thompson /// .field("dx", &rx, &bx, VectorOpt::Active)? 23056f97ff0aSJeremy L Thompson /// .field("weights", ElemRestrictionOpt::None, &bx, VectorOpt::None)? 23066f97ff0aSJeremy L Thompson /// .field("qdata", &rq, BasisOpt::Collocated, VectorOpt::Active)?; 23076f97ff0aSJeremy L Thompson /// 23086f97ff0aSJeremy L Thompson /// let op_build = ceed 23096f97ff0aSJeremy L Thompson /// .composite_operator()? 23106f97ff0aSJeremy L Thompson /// .sub_operator(&op_build_mass)? 23116f97ff0aSJeremy L Thompson /// .sub_operator(&op_build_diff)?; 23126f97ff0aSJeremy L Thompson /// 23136f97ff0aSJeremy L Thompson /// // Check 23146f97ff0aSJeremy L Thompson /// assert!(op_build.check().is_ok(), "Operator setup incorrectly"); 23156f97ff0aSJeremy L Thompson /// # Ok(()) 23166f97ff0aSJeremy L Thompson /// # } 23176f97ff0aSJeremy L Thompson /// ``` 23186f97ff0aSJeremy L Thompson pub fn check(self) -> crate::Result<Self> { 23196f97ff0aSJeremy L Thompson self.op_core.check()?; 23206f97ff0aSJeremy L Thompson Ok(self) 23216f97ff0aSJeremy L Thompson } 23226f97ff0aSJeremy L Thompson 23239df49d7eSJed Brown /// Assemble the diagonal of a square linear Operator 23249df49d7eSJed Brown /// 23259df49d7eSJed Brown /// This overwrites a Vector with the diagonal of a linear Operator. 23269df49d7eSJed Brown /// 23279df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 23289df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 23299df49d7eSJed Brown /// 23309df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 23319df49d7eSJed Brown /// * `assembled` - Vector to store assembled Operator diagonal 23329df49d7eSJed Brown pub fn linear_asssemble_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 23339df49d7eSJed Brown self.op_core.linear_assemble_add_diagonal(assembled) 23349df49d7eSJed Brown } 23359df49d7eSJed Brown 23369df49d7eSJed Brown /// Assemble the point block diagonal of a square linear Operator 23379df49d7eSJed Brown /// 23389df49d7eSJed Brown /// This overwrites a Vector with the point block diagonal of a linear 23399df49d7eSJed Brown /// Operator. 23409df49d7eSJed Brown /// 23419df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 23429df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 23439df49d7eSJed Brown /// 23449df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 23459df49d7eSJed Brown /// * `assembled` - Vector to store assembled Operator diagonal 23469df49d7eSJed Brown pub fn linear_assemble_add_diagonal(&self, assembled: &mut Vector) -> crate::Result<i32> { 23479df49d7eSJed Brown self.op_core.linear_assemble_add_diagonal(assembled) 23489df49d7eSJed Brown } 23499df49d7eSJed Brown 23509df49d7eSJed Brown /// Assemble the diagonal of a square linear Operator 23519df49d7eSJed Brown /// 23529df49d7eSJed Brown /// This overwrites a Vector with the diagonal of a linear Operator. 23539df49d7eSJed Brown /// 23549df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 23559df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 23569df49d7eSJed Brown /// 23579df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 23589df49d7eSJed Brown /// * `assembled` - Vector to store assembled CeedOperator point block 23599df49d7eSJed Brown /// diagonal, provided in row-major form with an 23609df49d7eSJed Brown /// `ncomp * ncomp` block at each node. The dimensions of 23619df49d7eSJed Brown /// this vector are derived from the active vector for 23629df49d7eSJed Brown /// the CeedOperator. The array has shape 23639df49d7eSJed Brown /// `[nodes, component out, component in]`. 23649df49d7eSJed Brown pub fn linear_assemble_point_block_diagonal( 23659df49d7eSJed Brown &self, 23669df49d7eSJed Brown assembled: &mut Vector, 23679df49d7eSJed Brown ) -> crate::Result<i32> { 23689df49d7eSJed Brown self.op_core.linear_assemble_add_diagonal(assembled) 23699df49d7eSJed Brown } 23709df49d7eSJed Brown 23719df49d7eSJed Brown /// Assemble the diagonal of a square linear Operator 23729df49d7eSJed Brown /// 23739df49d7eSJed Brown /// This sums into a Vector with the diagonal of a linear Operator. 23749df49d7eSJed Brown /// 23759df49d7eSJed Brown /// Note: Currently only non-composite Operators with a single field and 23769df49d7eSJed Brown /// composite Operators with single field sub-operators are supported. 23779df49d7eSJed Brown /// 23789df49d7eSJed Brown /// * `op` - Operator to assemble QFunction 23799df49d7eSJed Brown /// * `assembled` - Vector to store assembled CeedOperator point block 23809df49d7eSJed Brown /// diagonal, provided in row-major form with an 23819df49d7eSJed Brown /// `ncomp * ncomp` block at each node. The dimensions of 23829df49d7eSJed Brown /// this vector are derived from the active vector for 23839df49d7eSJed Brown /// the CeedOperator. The array has shape 23849df49d7eSJed Brown /// `[nodes, component out, component in]`. 23859df49d7eSJed Brown pub fn linear_assemble_add_point_block_diagonal( 23869df49d7eSJed Brown &self, 23879df49d7eSJed Brown assembled: &mut Vector, 23889df49d7eSJed Brown ) -> crate::Result<i32> { 23899df49d7eSJed Brown self.op_core.linear_assemble_add_diagonal(assembled) 23909df49d7eSJed Brown } 23919df49d7eSJed Brown } 23929df49d7eSJed Brown 23939df49d7eSJed Brown // ----------------------------------------------------------------------------- 2394