1 | import { isBigNumber } from '../../utils/is'
|
2 | import { isInteger } from '../../utils/number'
|
3 | import { resize } from '../../utils/array'
|
4 | import { factory } from '../../utils/factory'
|
5 |
|
6 | const name = 'ones'
|
7 | const dependencies = ['typed', 'config', 'matrix', 'BigNumber']
|
8 |
|
9 | export const createOnes = factory(name, dependencies, ({ typed, config, matrix, BigNumber }) => {
|
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 | return typed('ones', {
|
44 | '': function () {
|
45 | return (config.matrix === 'Array')
|
46 | ? _ones([])
|
47 | : _ones([], 'default')
|
48 | },
|
49 |
|
50 |
|
51 |
|
52 | '...number | BigNumber | string': function (size) {
|
53 | const last = size[size.length - 1]
|
54 | if (typeof last === 'string') {
|
55 | const format = size.pop()
|
56 | return _ones(size, format)
|
57 | } else if (config.matrix === 'Array') {
|
58 | return _ones(size)
|
59 | } else {
|
60 | return _ones(size, 'default')
|
61 | }
|
62 | },
|
63 |
|
64 | Array: _ones,
|
65 |
|
66 | Matrix: function (size) {
|
67 | const format = size.storage()
|
68 | return _ones(size.valueOf(), format)
|
69 | },
|
70 |
|
71 | 'Array | Matrix, string': function (size, format) {
|
72 | return _ones(size.valueOf(), format)
|
73 | }
|
74 | })
|
75 |
|
76 | |
77 |
|
78 |
|
79 |
|
80 |
|
81 |
|
82 |
|
83 | function _ones (size, format) {
|
84 | const hasBigNumbers = _normalize(size)
|
85 | const defaultValue = hasBigNumbers ? new BigNumber(1) : 1
|
86 | _validate(size)
|
87 |
|
88 | if (format) {
|
89 |
|
90 | const m = matrix(format)
|
91 | if (size.length > 0) {
|
92 | return m.resize(size, defaultValue)
|
93 | }
|
94 | return m
|
95 | } else {
|
96 |
|
97 | const arr = []
|
98 | if (size.length > 0) {
|
99 | return resize(arr, size, defaultValue)
|
100 | }
|
101 | return arr
|
102 | }
|
103 | }
|
104 |
|
105 |
|
106 | function _normalize (size) {
|
107 | let hasBigNumbers = false
|
108 | size.forEach(function (value, index, arr) {
|
109 | if (isBigNumber(value)) {
|
110 | hasBigNumbers = true
|
111 | arr[index] = value.toNumber()
|
112 | }
|
113 | })
|
114 | return hasBigNumbers
|
115 | }
|
116 |
|
117 |
|
118 | function _validate (size) {
|
119 | size.forEach(function (value) {
|
120 | if (typeof value !== 'number' || !isInteger(value) || value < 0) {
|
121 | throw new Error('Parameters in function ones must be positive integers')
|
122 | }
|
123 | })
|
124 | }
|
125 | })
|