jsadvent/lib/Vector.js

107 lines
2.5 KiB
JavaScript
Raw Normal View History

2020-12-10 08:33:00 +00:00
const Math = require('./bnmath');
const Glib = require('./Glib');
module.exports = class Vector {
static fromString(string) {
if (!string.startsWith('(') || !string.endsWith(')')) {
throw new Error('invalid format');
}
return new Vector(
...Glib.fromSplit(string.slice(1, -1), ',').map((i) => BigInt(i.trim())),
);
}
constructor(...v) {
this.v = v.map((i) => {
switch (typeof i) {
case 'number':
return BigInt(i);
case 'bigint':
return i;
default:
throw new Error('invalid input');
}
});
}
_check(v) {
if (typeof v !== 'object' || !(v instanceof Vector)) {
throw new Error('nope');
}
if (this.len !== v.len) {
throw new Error('mismatched length');
}
}
_resolve(v) {
if (typeof v === 'number' || typeof v === 'bigint') {
return new Vector(...this.v.map((e) => v));
}
if (typeof v !== 'object' || !(v instanceof Vector)) {
throw new Error('nope');
}
if (this.len !== v.len) {
throw new Error('mismatched length');
}
return v;
}
multiply(v) {
v = this._resolve(v);
return new Vector(...this.v.map((e, i) => e * v.v[i]));
}
add(v) {
v = this._resolve(v);
return new Vector(...this.v.map((e, i) => e + v.v[i]));
}
subtract(v) {
v = this._resolve(v);
return this.add(v.multiply(-1));
}
remainder(v) {
v = this._resolve(v);
return new Vector(...this.v.map((e, i) => e % v.v[i]));
}
modulo(v) {
v = this._resolve(v);
return new Vector(
...this.v.map((e, i) => ((e % v.v[i]) + v.v[i]) % v.v[i]),
);
}
distance(v) {
this._check(v);
return this.v
.map((e, i) => {
return Math.abs(e + v.v[i]);
})
.reduce((a, b) => a + b);
}
neighbors() {
return new Glib(
(function*(_this) {
let permutations = 3n ** _this.length;
const skipped = Math.parseInt('1'.repeat(_this.v.length), 3);
for (let i = 0n; i < permutations; i++) {
if (i === skipped) {
continue;
}
yield _this.add(
new Vector(
...Glib.fromIterable(
i.toString(3).padStart(_this.v.length, '0'),
).map((i) => BigInt(i) - 1n),
),
);
}
})(this),
);
}
toArray() {
return this.v.slice();
}
get length() {
return BigInt(this.v.length);
}
get string() {
return '(' + this.v.join(', ') + ')';
}
};