1 | import { factory } from '../../utils/factory.js';
|
2 | import { isInteger } from '../../utils/number.js';
|
3 | import { arraySize as size } from '../../utils/array.js';
|
4 | import { powNumber } from '../../plain/number/index.js';
|
5 | var name = 'pow';
|
6 | var dependencies = ['typed', 'config', 'identity', 'multiply', 'matrix', 'fraction', 'number', 'Complex'];
|
7 | export var createPow = factory(name, dependencies, (_ref) => {
|
8 | var {
|
9 | typed,
|
10 | config,
|
11 | identity,
|
12 | multiply,
|
13 | matrix,
|
14 | number,
|
15 | fraction,
|
16 | Complex
|
17 | } = _ref;
|
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 | return typed(name, {
|
52 | 'number, number': _pow,
|
53 | 'Complex, Complex': function ComplexComplex(x, y) {
|
54 | return x.pow(y);
|
55 | },
|
56 | 'BigNumber, BigNumber': function BigNumberBigNumber(x, y) {
|
57 | if (y.isInteger() || x >= 0 || config.predictable) {
|
58 | return x.pow(y);
|
59 | } else {
|
60 | return new Complex(x.toNumber(), 0).pow(y.toNumber(), 0);
|
61 | }
|
62 | },
|
63 | 'Fraction, Fraction': function FractionFraction(x, y) {
|
64 | if (y.d !== 1) {
|
65 | if (config.predictable) {
|
66 | throw new Error('Function pow does not support non-integer exponents for fractions.');
|
67 | } else {
|
68 | return _pow(x.valueOf(), y.valueOf());
|
69 | }
|
70 | } else {
|
71 | return x.pow(y);
|
72 | }
|
73 | },
|
74 | 'Array, number': _powArray,
|
75 | 'Array, BigNumber': function ArrayBigNumber(x, y) {
|
76 | return _powArray(x, y.toNumber());
|
77 | },
|
78 | 'Matrix, number': _powMatrix,
|
79 | 'Matrix, BigNumber': function MatrixBigNumber(x, y) {
|
80 | return _powMatrix(x, y.toNumber());
|
81 | },
|
82 | 'Unit, number | BigNumber': function UnitNumberBigNumber(x, y) {
|
83 | return x.pow(y);
|
84 | }
|
85 | });
|
86 | |
87 |
|
88 |
|
89 |
|
90 |
|
91 |
|
92 |
|
93 |
|
94 | function _pow(x, y) {
|
95 |
|
96 |
|
97 | if (config.predictable && !isInteger(y) && x < 0) {
|
98 |
|
99 | try {
|
100 | var yFrac = fraction(y);
|
101 | var yNum = number(yFrac);
|
102 |
|
103 | if (y === yNum || Math.abs((y - yNum) / y) < 1e-14) {
|
104 | if (yFrac.d % 2 === 1) {
|
105 | return (yFrac.n % 2 === 0 ? 1 : -1) * Math.pow(-x, y);
|
106 | }
|
107 | }
|
108 | } catch (ex) {
|
109 | }
|
110 |
|
111 | }
|
112 |
|
113 |
|
114 |
|
115 |
|
116 | if (config.predictable && (x < -1 && y === Infinity || x > -1 && x < 0 && y === -Infinity)) {
|
117 | return NaN;
|
118 | }
|
119 |
|
120 | if (isInteger(y) || x >= 0 || config.predictable) {
|
121 | return powNumber(x, y);
|
122 | } else {
|
123 |
|
124 |
|
125 |
|
126 | if (x * x < 1 && y === Infinity || x * x > 1 && y === -Infinity) {
|
127 | return 0;
|
128 | }
|
129 |
|
130 | return new Complex(x, 0).pow(y, 0);
|
131 | }
|
132 | }
|
133 | |
134 |
|
135 |
|
136 |
|
137 |
|
138 |
|
139 |
|
140 |
|
141 |
|
142 | function _powArray(x, y) {
|
143 | if (!isInteger(y) || y < 0) {
|
144 | throw new TypeError('For A^b, b must be a positive integer (value is ' + y + ')');
|
145 | }
|
146 |
|
147 |
|
148 | var s = size(x);
|
149 |
|
150 | if (s.length !== 2) {
|
151 | throw new Error('For A^b, A must be 2 dimensional (A has ' + s.length + ' dimensions)');
|
152 | }
|
153 |
|
154 | if (s[0] !== s[1]) {
|
155 | throw new Error('For A^b, A must be square (size is ' + s[0] + 'x' + s[1] + ')');
|
156 | }
|
157 |
|
158 | var res = identity(s[0]).valueOf();
|
159 | var px = x;
|
160 |
|
161 | while (y >= 1) {
|
162 | if ((y & 1) === 1) {
|
163 | res = multiply(px, res);
|
164 | }
|
165 |
|
166 | y >>= 1;
|
167 | px = multiply(px, px);
|
168 | }
|
169 |
|
170 | return res;
|
171 | }
|
172 | |
173 |
|
174 |
|
175 |
|
176 |
|
177 |
|
178 |
|
179 |
|
180 |
|
181 | function _powMatrix(x, y) {
|
182 | return matrix(_powArray(x.valueOf(), y));
|
183 | }
|
184 | }); |
\ | No newline at end of file |