1 | "use strict";
|
2 |
|
3 | Object.defineProperty(exports, "__esModule", {
|
4 | value: true
|
5 | });
|
6 | exports.createDiag = void 0;
|
7 |
|
8 | var _is = require("../../utils/is.js");
|
9 |
|
10 | var _array = require("../../utils/array.js");
|
11 |
|
12 | var _number = require("../../utils/number.js");
|
13 |
|
14 | var _factory = require("../../utils/factory.js");
|
15 |
|
16 | var name = 'diag';
|
17 | var dependencies = ['typed', 'matrix', 'DenseMatrix', 'SparseMatrix'];
|
18 | var createDiag = (0, _factory.factory)(name, dependencies, function (_ref) {
|
19 | var typed = _ref.typed,
|
20 | matrix = _ref.matrix,
|
21 | DenseMatrix = _ref.DenseMatrix,
|
22 | SparseMatrix = _ref.SparseMatrix;
|
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 | return typed(name, {
|
62 |
|
63 | Array: function Array(x) {
|
64 | return _diag(x, 0, (0, _array.arraySize)(x), null);
|
65 | },
|
66 | 'Array, number': function ArrayNumber(x, k) {
|
67 | return _diag(x, k, (0, _array.arraySize)(x), null);
|
68 | },
|
69 | 'Array, BigNumber': function ArrayBigNumber(x, k) {
|
70 | return _diag(x, k.toNumber(), (0, _array.arraySize)(x), null);
|
71 | },
|
72 | 'Array, string': function ArrayString(x, format) {
|
73 | return _diag(x, 0, (0, _array.arraySize)(x), format);
|
74 | },
|
75 | 'Array, number, string': function ArrayNumberString(x, k, format) {
|
76 | return _diag(x, k, (0, _array.arraySize)(x), format);
|
77 | },
|
78 | 'Array, BigNumber, string': function ArrayBigNumberString(x, k, format) {
|
79 | return _diag(x, k.toNumber(), (0, _array.arraySize)(x), format);
|
80 | },
|
81 | Matrix: function Matrix(x) {
|
82 | return _diag(x, 0, x.size(), x.storage());
|
83 | },
|
84 | 'Matrix, number': function MatrixNumber(x, k) {
|
85 | return _diag(x, k, x.size(), x.storage());
|
86 | },
|
87 | 'Matrix, BigNumber': function MatrixBigNumber(x, k) {
|
88 | return _diag(x, k.toNumber(), x.size(), x.storage());
|
89 | },
|
90 | 'Matrix, string': function MatrixString(x, format) {
|
91 | return _diag(x, 0, x.size(), format);
|
92 | },
|
93 | 'Matrix, number, string': function MatrixNumberString(x, k, format) {
|
94 | return _diag(x, k, x.size(), format);
|
95 | },
|
96 | 'Matrix, BigNumber, string': function MatrixBigNumberString(x, k, format) {
|
97 | return _diag(x, k.toNumber(), x.size(), format);
|
98 | }
|
99 | });
|
100 | |
101 |
|
102 |
|
103 |
|
104 |
|
105 |
|
106 |
|
107 |
|
108 |
|
109 |
|
110 | function _diag(x, k, size, format) {
|
111 | if (!(0, _number.isInteger)(k)) {
|
112 | throw new TypeError('Second parameter in function diag must be an integer');
|
113 | }
|
114 |
|
115 | var kSuper = k > 0 ? k : 0;
|
116 | var kSub = k < 0 ? -k : 0;
|
117 |
|
118 | switch (size.length) {
|
119 | case 1:
|
120 | return _createDiagonalMatrix(x, k, format, size[0], kSub, kSuper);
|
121 |
|
122 | case 2:
|
123 | return _getDiagonal(x, k, format, size, kSub, kSuper);
|
124 | }
|
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 | var ms = [l + kSub, l + kSuper];
|
132 |
|
133 | if (format && format !== 'sparse' && format !== 'dense') {
|
134 | throw new TypeError("Unknown matrix type ".concat(format, "\""));
|
135 | }
|
136 |
|
137 |
|
138 | var m = format === 'sparse' ? SparseMatrix.diagonal(ms, x, k) : DenseMatrix.diagonal(ms, x, k);
|
139 |
|
140 | return format !== null ? m : m.valueOf();
|
141 | }
|
142 |
|
143 | function _getDiagonal(x, k, format, s, kSub, kSuper) {
|
144 |
|
145 | if ((0, _is.isMatrix)(x)) {
|
146 |
|
147 | var dm = x.diagonal(k);
|
148 |
|
149 | if (format !== null) {
|
150 |
|
151 | if (format !== dm.storage()) {
|
152 | return matrix(dm, format);
|
153 | }
|
154 |
|
155 | return dm;
|
156 | }
|
157 |
|
158 | return dm.valueOf();
|
159 | }
|
160 |
|
161 |
|
162 | var n = Math.min(s[0] - kSub, s[1] - kSuper);
|
163 |
|
164 | var vector = [];
|
165 |
|
166 | for (var i = 0; i < n; i++) {
|
167 | vector[i] = x[i + kSub][i + kSuper];
|
168 | }
|
169 |
|
170 |
|
171 | return format !== null ? matrix(vector) : vector;
|
172 | }
|
173 | });
|
174 | exports.createDiag = createDiag; |
\ | No newline at end of file |