paz_math/complex_vector.cpp at main · pazvirtual/paz_math · GitHub

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
118
119
120
121
#include "PAZ_Math"

paz::ComplexVec paz::ComplexVec::Constant(std::size_t rows, complex c)
{
    ComplexVec v(rows);
    std::fill(v.begin(), v.end(), c);
    return v;
}

paz::ComplexVec paz::ComplexVec::Zero(std::size_t rows)
{
    return Constant(rows, 0.);
}

paz::ComplexVec paz::ComplexVec::Ones(std::size_t rows)
{
    return Constant(rows, 1.);
}

paz::ComplexVec paz::ComplexVec::IdQuat()
{
    return {{0., 0., 0., 1.}};
}

paz::ComplexVec paz::ComplexVec::Randn(std::size_t rows)
{
    ComplexVec v(rows);
    for(auto& n : v)
    {
        n = randn();
    }
    return v;
}

paz::ComplexVec paz::ComplexVec::Cat(const ComplexMatRef& a, const
    ComplexMatRef& b)
{
    if(a.cols() != 1 || b.cols() != 1)
    {
        throw std::runtime_error("Matrices must be column vectors.");
    }
    ComplexVec v(a.size() + b.size());
    std::copy(a.begin(), a.end(), v.begin());
    std::copy(b.begin(), b.end(), v.begin() + a.size());
    return v;
}

paz::ComplexVec::ComplexVec(std::size_t rows) : ComplexMat(rows, 1) {}

paz::ComplexVec::ComplexVec(const ComplexMat& m) : ComplexMat(m)
{
    if(cols() != 1)
    {
        throw std::runtime_error("Matrix must be a column vector.");
    }
}

paz::ComplexVec::ComplexVec(const ComplexMatRef& m) : ComplexMat(m)
{
    if(cols() != 1)
    {
        throw std::runtime_error("Matrix must be a column vector.");
    }
}

paz::ComplexVec::ComplexVec(const std::initializer_list<std::initializer_list<
    complex>>& list)
{
    if(!list.size())
    {
        return;
    }
    if(list.size() != 1)
    {
        throw std::runtime_error("Vector initializer list is malformed.");
    }
    *this = ComplexVec(list.begin()->size());
    std::copy(list.begin()->begin(), list.begin()->end(), begin());
}

paz::ComplexMatRef paz::ComplexVec::segment(std::size_t start, std::size_t n)
    const
{
    return block(start, 0, n, 1);
}

void paz::ComplexVec::setSegment(std::size_t start, std::size_t n, const
    ComplexMatRef& rhs)
{
    setBlock(start, 0, n, 1, rhs);
}

paz::ComplexMatRef paz::ComplexVec::head(std::size_t n) const
{
    return segment(0, n);
}

void paz::ComplexVec::setHead(std::size_t n, const ComplexVec& rhs)
{
    setSegment(0, n, rhs);
}

paz::ComplexMatRef paz::ComplexVec::tail(std::size_t n) const
{
    return segment(size() - n, n);
}

void paz::ComplexVec::setTail(std::size_t n, const ComplexVec& rhs)
{
    setSegment(size() - n, n, rhs);
}

void paz::ComplexVec::resize(std::size_t newRows)
{
    resizeRows(newRows);
}

void paz::ComplexVec::resize(std::size_t newRows, complex c)
{
    resizeRows(newRows, c);
}