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
|
'use strict';
const DEFAULT_WEIGHT_MAX = 4;
const graph_proto = {
connect: function(source, sink, weight) {
return network_connect(this, source, sink, weight);
},
};
export function network(input_count, internal_count, output_count, weight_max = 4) {
const count = input_count + internal_count + output_count;
const n = Object.create(graph_proto);
n.input_count = input_count;
n.output_count = output_count;
n.adjacency = new Array(count).fill([]);
n.weight = [];
return Object.freeze(n);
}
function is_input(n, index) {
return index < n.input_count;
}
function is_output(n, index) {
return index >= (n.adjacency.length - n.output_count);
}
function network_connect(n, source, sink, weight) {
if (is_input(n, sink)) {
// inputs cannot be sinks
throw new Error("attempt to use input as sink");
}
if (is_output(n, source)) {
// outputs cannot be sources
throw new Error("attempt to use output as source");
}
const nn = Object.create(graph_proto);
nn.input_count = n.input_count;
nn.output_count = n.output_count;
nn.adjacency = n.adjacency.map((row, i) => {
if (i === source && i === sink) {
// self-loop
return [...row, 2];
} else if (i === source) {
return [...row, 1];
} else if (i === sink) {
return [...row, -1];
} else {
return [...row, 0];
}
});
nn.weight = [...n.weight, weight];
return Object.freeze(nn);
}
|
