array_object/convert/
from_complex.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
use crate::adaptor::*;
use crate::error::ArrayObjectError;
use crate::misc::Product;
use crate::storage::*;
use num_complex::Complex;

macro_rules! from_complex {
    ($($ty:ty),*) => {
        $(
            impl From<Pair<$ty>> for ArrayObject {
                fn from(Pair(val_re, val_im): Pair<$ty>) -> Self {
                    let data = [val_re.to_le_bytes().to_vec(), val_im.to_le_bytes().to_vec()].concat();
                    Self {
                        data,
                        shape: vec![],
                        datatype: DataType::Complex,
                    }
                }
            }
            impl From<Complex<$ty>> for ArrayObject {
                fn from(val: Complex<$ty>) -> Self {
                    Pair(val.re, val.im).into()
                }
            }
            impl<const N: usize> From<[Complex<$ty>; N]> for ArrayObject {
                fn from(val: [Complex<$ty>; N]) -> Self {
                    val.to_vec().into()
                }
            }
            impl From<&[Complex<$ty>]> for ArrayObject {
                fn from(val: &[Complex<$ty>]) -> Self {
                    val.to_vec().into()
                }
            }
            impl From<Vec<Pair<$ty>>> for ArrayObject {
                fn from(val: Vec<Pair<$ty>>) -> Self {
                    let shape = vec![val.len() as u64];
                    let mut data = Vec::<u8>::with_capacity(2 * val.len() * size_of::<$ty>());
                    for v in val {
                        let Pair(re, im) = v;
                        data.append(&mut re.to_le_bytes().to_vec());
                        data.append(&mut im.to_le_bytes().to_vec());
                    }
                    Self {
                        data,
                        shape,
                        datatype: DataType::Complex,
                    }
                }
            }
            impl From<Vec<Complex<$ty>>> for ArrayObject {
                fn from(val: Vec<Complex<$ty>>) -> Self {
                    let shape = vec![val.len() as u64];
                    let mut data = Vec::<u8>::with_capacity(2 * val.len() * size_of::<$ty>());
                    for v in val {
                        data.append(&mut v.re.to_le_bytes().to_vec());
                        data.append(&mut v.im.to_le_bytes().to_vec());
                    }
                    Self {
                        data,
                        shape,
                        datatype: DataType::Complex,
                    }
                }
            }
            impl TryFrom<VecVec<$ty>> for ArrayObject {
                type Error = ArrayObjectError;
                fn try_from(VecVec(val_re, val_im): VecVec<$ty>) -> Result<Self, Self::Error> {
                    if val_re.len() != val_im.len() {
                        return Err(ArrayObjectError::VectorLengthMismatch(val_re.len(), val_im.len()));
                    }
                    let shape = vec![val_re.len() as u64];
                    let mut data = Vec::<u8>::with_capacity(2 * val_re.len() * size_of::<$ty>());
                    for (re, im) in val_re.into_iter().zip(val_im.into_iter()) {
                        data.append(&mut re.to_le_bytes().to_vec());
                        data.append(&mut im.to_le_bytes().to_vec());
                    }
                    Ok(Self {
                        data,
                        shape,
                        datatype: DataType::Complex,
                    })
                }
            }
            impl TryFrom<VecShape<Complex<$ty>>> for ArrayObject {
                type Error = ArrayObjectError;
                fn try_from(VecShape(val, shape): VecShape<Complex<$ty>>) -> Result<Self, Self::Error> {
                    if val.len() != shape.product() as usize {
                        return Err(ArrayObjectError::NumberOfElementsMismatch(val.len(), shape.product() as usize));
                    }
                    if shape.len() > 15 {
                        return Err(ArrayObjectError::TooLargeDimension(shape.len()));
                    }
                    let mut temp: ArrayObject = val.into();
                    temp.shape = shape;
                    Ok(temp)
                }
            }
            impl TryFrom<VecVecShape<$ty>> for ArrayObject {
                type Error = ArrayObjectError;
                fn try_from(VecVecShape(val_re, val_im, shape): VecVecShape<$ty>) -> Result<Self, Self::Error> {
                    if val_re.len() != shape.product() as usize {
                        return Err(ArrayObjectError::NumberOfElementsMismatch(val_re.len(), shape.product() as usize));
                    }
                    if shape.len() > 15 {
                        return Err(ArrayObjectError::TooLargeDimension(shape.len()));
                    }
                    let mut temp: ArrayObject = VecVec(val_re, val_im).try_into()?;
                    temp.shape = shape;
                    Ok(temp)
                }
            }
        )*
    };
}

from_complex!(f32, f64);