1 | import ndarray from 'ndarray'
|
2 |
|
3 | import { mapRange } from './transform.js'
|
4 | import { checkCoverage } from '../validate.js'
|
5 | import { ensureClockwisePolygon, getPointInPolygonsFn } from '../domain/polygon.js'
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 | export function maskByPolygon (cov, polygon, axes = ['x', 'y']) {
|
18 | checkCoverage(cov)
|
19 |
|
20 | if (polygon.type === 'Polygon') {
|
21 | polygon = {
|
22 | type: 'MultiPolygon',
|
23 | coordinates: [polygon.coordinates]
|
24 | }
|
25 | }
|
26 |
|
27 | let polygons = polygon.coordinates
|
28 | polygons.forEach(p => ensureClockwisePolygon(p))
|
29 |
|
30 | let pip = getPointInPolygonsFn(polygons)
|
31 |
|
32 | let [X, Y] = axes
|
33 |
|
34 | return cov.loadDomain().then(domain => {
|
35 | let x = domain.axes.get(X).values
|
36 | let y = domain.axes.get(Y).values
|
37 | let pnpolyCache = ndarray(new Uint8Array(x.length * y.length), [x.length, y.length])
|
38 |
|
39 | for (let i = 0; i < x.length; i++) {
|
40 | for (let j = 0; j < y.length; j++) {
|
41 | let inside = pip([x[i], y[j]]) >= 0
|
42 | pnpolyCache.set(i, j, inside)
|
43 | }
|
44 | }
|
45 |
|
46 | let fn = (obj, range) => {
|
47 | if (pnpolyCache.get(obj[X] || 0, obj[Y] || 0)) {
|
48 | return range.get(obj)
|
49 | } else {
|
50 | return null
|
51 | }
|
52 | }
|
53 |
|
54 | let newcov = cov
|
55 | for (let key of cov.parameters.keys()) {
|
56 | newcov = mapRange(newcov, key, fn)
|
57 | }
|
58 | return newcov
|
59 | })
|
60 | }
|