-
Notifications
You must be signed in to change notification settings - Fork 112
/
solver.js
90 lines (78 loc) · 2.91 KB
/
solver.js
1
2
3
4
5
6
7
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
/** @license kiwi.js v1.1.2
* #------------------------------------------------------------------------------
* # Copyright (c) 2013, Nucleic Development Team & H. Rutjes.
* #
* # Distributed under the terms of the Modified BSD License.
* #
* # The full license is in the file COPYING.txt, distributed with this software.
* #------------------------------------------------------------------------------
**/
import { Solver, Variable } from 'kiwi.js';
/**
* Applies the given constraints to the objects in-place.
* A solution is approximated iteratively.
* @param {array} constraints The constraints
* @param {function} constraint.base.solve A function that solves the constraint in-place
* @param {number} iterations The number of iterations
* @param {?object} constants The constants used by constraints
*/
export const solveLoose = (constraints, iterations, constants) => {
for (let i = 0; i < iterations; i += 1) {
for (const co of constraints) {
co.base.solve(co, constants);
}
}
};
/**
* Applies the given constraints to the objects in-place.
* A solution is found exactly for the constraints that are solvable.
* Any unsolvable constraints will be skipped and a warning logged in the console.
* @param {array} constraints The constraints
* @param {string} constraint.base.property The property name on `a` and `b` to constrain
* @param {function} constraint.base.strict A function returns the constraint in strict form
* @param {object} constraint.a The first object to constrain
* @param {object} constraint.b The second object to constrain
* @param {object} constraint.a.id A unique id for the first object
* @param {object} constraint.b.id A unique id for the second object
* @param {?object} constants The constants used by constraints
*/
export const solveStrict = (constraints, constants) => {
const solver = new Solver();
const variables = {};
const variableId = (obj, property) => `${obj.id}_${property}`;
const addVariable = (obj, property) => {
const id = variableId(obj, property);
if (!variables[id]) {
const variable = (variables[id] = new Variable());
variable.property = property;
variable.obj = obj;
}
};
for (const co of constraints) {
addVariable(co.a, co.base.property);
addVariable(co.b, co.base.property);
}
let unsolvableCount = 0;
for (const co of constraints) {
try {
solver.addConstraint(
co.base.strict(
co,
constants,
variables[variableId(co.a, co.base.property)],
variables[variableId(co.b, co.base.property)]
)
);
} catch (err) {
unsolvableCount += 1;
}
}
if (unsolvableCount > 0) {
console.warn(`Skipped ${unsolvableCount} unsolvable constraints`);
}
solver.updateVariables();
const variablesList = Object.values(variables);
for (const variable of variablesList) {
variable.obj[variable.property] = variable.value();
}
};