1 | 'use strict'
|
2 |
|
3 | const array = require('../../utils/array')
|
4 | const isInteger = require('../../utils/number').isInteger
|
5 |
|
6 | function factory (type, config, load, typed) {
|
7 | const matrix = load(require('../../type/matrix/function/matrix'))
|
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 | const diag = typed('diag', {
|
47 |
|
48 |
|
49 | 'Array': function (x) {
|
50 | return _diag(x, 0, array.size(x), null)
|
51 | },
|
52 |
|
53 | 'Array, number': function (x, k) {
|
54 | return _diag(x, k, array.size(x), null)
|
55 | },
|
56 |
|
57 | 'Array, BigNumber': function (x, k) {
|
58 | return _diag(x, k.toNumber(), array.size(x), null)
|
59 | },
|
60 |
|
61 | 'Array, string': function (x, format) {
|
62 | return _diag(x, 0, array.size(x), format)
|
63 | },
|
64 |
|
65 | 'Array, number, string': function (x, k, format) {
|
66 | return _diag(x, k, array.size(x), format)
|
67 | },
|
68 |
|
69 | 'Array, BigNumber, string': function (x, k, format) {
|
70 | return _diag(x, k.toNumber(), array.size(x), format)
|
71 | },
|
72 |
|
73 | 'Matrix': function (x) {
|
74 | return _diag(x, 0, x.size(), x.storage())
|
75 | },
|
76 |
|
77 | 'Matrix, number': function (x, k) {
|
78 | return _diag(x, k, x.size(), x.storage())
|
79 | },
|
80 |
|
81 | 'Matrix, BigNumber': function (x, k) {
|
82 | return _diag(x, k.toNumber(), x.size(), x.storage())
|
83 | },
|
84 |
|
85 | 'Matrix, string': function (x, format) {
|
86 | return _diag(x, 0, x.size(), format)
|
87 | },
|
88 |
|
89 | 'Matrix, number, string': function (x, k, format) {
|
90 | return _diag(x, k, x.size(), format)
|
91 | },
|
92 |
|
93 | 'Matrix, BigNumber, string': function (x, k, format) {
|
94 | return _diag(x, k.toNumber(), x.size(), format)
|
95 | }
|
96 | })
|
97 |
|
98 | diag.toTex = undefined
|
99 |
|
100 | return diag
|
101 |
|
102 | |
103 |
|
104 |
|
105 |
|
106 |
|
107 |
|
108 |
|
109 |
|
110 |
|
111 | function _diag (x, k, size, format) {
|
112 | if (!isInteger(k)) {
|
113 | throw new TypeError('Second parameter in function diag must be an integer')
|
114 | }
|
115 |
|
116 | const kSuper = k > 0 ? k : 0
|
117 | const kSub = k < 0 ? -k : 0
|
118 |
|
119 |
|
120 | switch (size.length) {
|
121 | case 1:
|
122 | return _createDiagonalMatrix(x, k, format, size[0], kSub, kSuper)
|
123 | case 2:
|
124 | return _getDiagonal(x, k, format, size, kSub, kSuper)
|
125 | }
|
126 | throw new RangeError('Matrix for function diag must be 2 dimensional')
|
127 | }
|
128 |
|
129 | function _createDiagonalMatrix (x, k, format, l, kSub, kSuper) {
|
130 |
|
131 | const ms = [l + kSub, l + kSuper]
|
132 |
|
133 | const F = type.Matrix.storage(format || 'dense')
|
134 |
|
135 | const m = F.diagonal(ms, x, k)
|
136 |
|
137 | return format !== null ? m : m.valueOf()
|
138 | }
|
139 |
|
140 | function _getDiagonal (x, k, format, s, kSub, kSuper) {
|
141 |
|
142 | if (type.isMatrix(x)) {
|
143 |
|
144 | const dm = x.diagonal(k)
|
145 |
|
146 | if (format !== null) {
|
147 |
|
148 | if (format !== dm.storage()) { return matrix(dm, format) }
|
149 | return dm
|
150 | }
|
151 | return dm.valueOf()
|
152 | }
|
153 |
|
154 | const n = Math.min(s[0] - kSub, s[1] - kSuper)
|
155 |
|
156 | const vector = []
|
157 |
|
158 | for (let i = 0; i < n; i++) {
|
159 | vector[i] = x[i + kSub][i + kSuper]
|
160 | }
|
161 |
|
162 | return format !== null ? matrix(vector) : vector
|
163 | }
|
164 | }
|
165 |
|
166 | exports.name = 'diag'
|
167 | exports.factory = factory
|