Struct sparse_complex::ComplexMatrix

source · [−]

pub struct ComplexMatrix<T: Float> { /* private fields */ }

Expand description

The complex matrix struct

Implementations

source

impl<T: Float> ComplexMatrix<T>

source

pub fn new() -> Self

Create a new, initially empty ComplexMatrix

 use sparse_complex::ComplexMatrix;
 let mut m = ComplexMatrix::<f64>::new();

source

pub fn with_capacity(capacity: usize) -> Self

Create a new, initially empty ComplexMatrix with a given capacity

 use sparse_complex::ComplexMatrix;
 let mut m = ComplexMatrix::<f64>::with_capacity(5);

source

pub fn from_entries(entries: Vec<(usize, usize, Complex<T>)>) -> Self

Create a new ComplexMatrix from a vector of (row, col, Complex<T>) entries.

 use sparse_complex::ComplexMatrix;
 use num::Complex;
 let entries = vec![(0, 0, Complex::new(1., 1.)), (1, 1, Complex::new(1., 1.))];
 let mut m = ComplexMatrix::<f64>::from_entries(entries);

source

pub fn add_element(&mut self, row: usize, col: usize, value: Complex<T>)

Add or set an element at location (row, col) with value.

 use sparse_complex::ComplexMatrix;
 use num::Complex;
 
 let Z1: Complex<f64> = Complex { re: 1., im: -1. };
 let Z2: Complex<f64> = Complex { re: -1., im: 1. };
 
 let mut m = ComplexMatrix::new();
 m.add_element(0, 0, Z1);
 m.add_element(1, 1, Z2);
 
 assert_eq!(m.get(0, 0), Some(&Z1));
 assert_eq!(m.get(1, 1), Some(&Z2));

source

pub fn get(&self, row: usize, col: usize) -> Option<&Complex<T>>

Returns the Element-value at (row, col) if present, or None if not.

 use sparse_complex::ComplexMatrix;
 use num::Complex;
 
 let Z1: Complex<f64> = Complex { re: 1., im: -1. };
 let Z2: Complex<f64> = Complex { re: -1., im: 1. };
 
 let mut m = ComplexMatrix::new();
 m.add_element(0, 0, Z1);
 m.add_element(1, 1, Z2);
 
 assert_eq!(m.get(0, 0), Some(&Z1));
 assert_eq!(m.get(1, 1), Some(&Z2));

source

impl ComplexMatrix<f64>

source

pub fn solve(&self, b: &mut [Complex<f64>]) -> Result<(), &'static str>

Solve the system Ax=b, where:

A is a complex matrix
b is a complex vector

Returns a Result. Ok(()) if the system was solved successfully, Err(String) if not. The result is stored in b.

The solution use the Eigen::SparseLU.

 use sparse_complex::ComplexMatrix;
 use num::Complex;
 
 let Z1: Complex<f64> = Complex { re: 1., im: -1. };
 let Z2: Complex<f64> = Complex { re: -1., im: 1. };
 
 let mut m = ComplexMatrix::new();
 m.add_element(0, 0, Z1);
 m.add_element(1, 1, Z2);
 
 let mut b = vec![Complex::new(1., 0.), Complex::new(0., 1.)];
 m.solve(&mut b).unwrap();
 
 let expected = vec![Complex::new(0.5, 0.5), Complex::new(0.5, -0.5)];
 assert_eq!(b, expected);

source

impl ComplexMatrix<f32>

source

pub fn solve(&self, b: &mut [Complex<f32>]) -> Result<(), &'static str>

Solve the system Ax=b, where:

A is a complex matrix
b is a complex vector

Returns a Result. Ok(()) if the system was solved successfully, Err(String) if not. The result is stored in b.

This solution use the Eigen::SparseLU.

 use sparse_complex::ComplexMatrix;
 use num::Complex;
 
 let Z1: Complex<f32> = Complex { re: 1., im: -1. };
 let Z2: Complex<f32> = Complex { re: -1., im: 1. };
 
 let mut m = ComplexMatrix::new();
 m.add_element(0, 0, Z1);
 m.add_element(1, 1, Z2);
 
 let mut b = vec![Complex::new(1., 0.), Complex::new(0., 1.)];
 m.solve(&mut b).unwrap();
 
 let expected = vec![Complex::new(0.5, 0.5), Complex::new(0.5, -0.5)];
 assert_eq!(b, expected);