UNPKG

node-red-contrib-tak-registration

Version:

A Node-RED node to register to TAK and to help wrap files as datapackages to send to TAK

297 lines (293 loc) 10.3 kB
import ItemBoundable from './ItemBoundable' import hasInterface from '../../../../../hasInterface' import ItemVisitor from '../ItemVisitor' import AbstractNode from './AbstractNode' import Collections from '../../../../../java/util/Collections' import ArrayList from '../../../../../java/util/ArrayList' import Serializable from '../../../../../java/io/Serializable' import Assert from '../../util/Assert' import List from '../../../../../java/util/List' export default class AbstractSTRtree { constructor () { this._root = null this._built = false this._itemBoundables = new ArrayList() this._nodeCapacity = null if (arguments.length === 0) { const nodeCapacity = AbstractSTRtree.DEFAULT_NODE_CAPACITY this._nodeCapacity = nodeCapacity } else if (arguments.length === 1) { const nodeCapacity = arguments[0] Assert.isTrue(nodeCapacity > 1, 'Node capacity must be greater than 1') this._nodeCapacity = nodeCapacity } } getNodeCapacity () { return this._nodeCapacity } lastNode (nodes) { return nodes.get(nodes.size() - 1) } size () { if (arguments.length === 0) { if (this.isEmpty()) { return 0 } this.build() return this.size(this._root) } else if (arguments.length === 1) { const node = arguments[0] let size = 0 for (const i = node.getChildBoundables().iterator(); i.hasNext();) { const childBoundable = i.next() if (childBoundable instanceof AbstractNode) { size += this.size(childBoundable) } else if (childBoundable instanceof ItemBoundable) { size += 1 } } return size } } removeItem (node, item) { let childToRemove = null for (const i = node.getChildBoundables().iterator(); i.hasNext();) { const childBoundable = i.next() if (childBoundable instanceof ItemBoundable) { if (childBoundable.getItem() === item) childToRemove = childBoundable } } if (childToRemove !== null) { node.getChildBoundables().remove(childToRemove) return true } return false } itemsTree () { if (arguments.length === 0) { this.build() const valuesTree = this.itemsTree(this._root) if (valuesTree === null) return new ArrayList() return valuesTree } else if (arguments.length === 1) { const node = arguments[0] const valuesTreeForNode = new ArrayList() for (const i = node.getChildBoundables().iterator(); i.hasNext();) { const childBoundable = i.next() if (childBoundable instanceof AbstractNode) { const valuesTreeForChild = this.itemsTree(childBoundable) if (valuesTreeForChild !== null) valuesTreeForNode.add(valuesTreeForChild) } else if (childBoundable instanceof ItemBoundable) { valuesTreeForNode.add(childBoundable.getItem()) } else { Assert.shouldNeverReachHere() } } if (valuesTreeForNode.size() <= 0) return null return valuesTreeForNode } } insert (bounds, item) { Assert.isTrue(!this._built, 'Cannot insert items into an STR packed R-tree after it has been built.') this._itemBoundables.add(new ItemBoundable(bounds, item)) } boundablesAtLevel () { if (arguments.length === 1) { const level = arguments[0] const boundables = new ArrayList() this.boundablesAtLevel(level, this._root, boundables) return boundables } else if (arguments.length === 3) { const level = arguments[0] const top = arguments[1] const boundables = arguments[2] Assert.isTrue(level > -2) if (top.getLevel() === level) { boundables.add(top) return null } for (const i = top.getChildBoundables().iterator(); i.hasNext();) { const boundable = i.next() if (boundable instanceof AbstractNode) { this.boundablesAtLevel(level, boundable, boundables) } else { Assert.isTrue(boundable instanceof ItemBoundable) if (level === -1) { boundables.add(boundable) } } } return null } } query () { if (arguments.length === 1) { const searchBounds = arguments[0] this.build() const matches = new ArrayList() if (this.isEmpty()) { return matches } if (this.getIntersectsOp().intersects(this._root.getBounds(), searchBounds)) { this.query(searchBounds, this._root, matches) } return matches } else if (arguments.length === 2) { const searchBounds = arguments[0] const visitor = arguments[1] this.build() if (this.isEmpty()) { return null } if (this.getIntersectsOp().intersects(this._root.getBounds(), searchBounds)) { this.query(searchBounds, this._root, visitor) } } else if (arguments.length === 3) { if (hasInterface(arguments[2], ItemVisitor) && (arguments[0] instanceof Object && arguments[1] instanceof AbstractNode)) { const searchBounds = arguments[0] const node = arguments[1] const visitor = arguments[2] const childBoundables = node.getChildBoundables() for (let i = 0; i < childBoundables.size(); i++) { const childBoundable = childBoundables.get(i) if (!this.getIntersectsOp().intersects(childBoundable.getBounds(), searchBounds)) { continue } if (childBoundable instanceof AbstractNode) { this.query(searchBounds, childBoundable, visitor) } else if (childBoundable instanceof ItemBoundable) { visitor.visitItem(childBoundable.getItem()) } else { Assert.shouldNeverReachHere() } } } else if (hasInterface(arguments[2], List) && (arguments[0] instanceof Object && arguments[1] instanceof AbstractNode)) { const searchBounds = arguments[0] const node = arguments[1] const matches = arguments[2] const childBoundables = node.getChildBoundables() for (let i = 0; i < childBoundables.size(); i++) { const childBoundable = childBoundables.get(i) if (!this.getIntersectsOp().intersects(childBoundable.getBounds(), searchBounds)) { continue } if (childBoundable instanceof AbstractNode) { this.query(searchBounds, childBoundable, matches) } else if (childBoundable instanceof ItemBoundable) { matches.add(childBoundable.getItem()) } else { Assert.shouldNeverReachHere() } } } } } build () { if (this._built) return null this._root = this._itemBoundables.isEmpty() ? this.createNode(0) : this.createHigherLevels(this._itemBoundables, -1) this._itemBoundables = null this._built = true } getRoot () { this.build() return this._root } remove () { if (arguments.length === 2) { const searchBounds = arguments[0] const item = arguments[1] this.build() if (this.getIntersectsOp().intersects(this._root.getBounds(), searchBounds)) { return this.remove(searchBounds, this._root, item) } return false } else if (arguments.length === 3) { const searchBounds = arguments[0] const node = arguments[1] const item = arguments[2] let found = this.removeItem(node, item) if (found) return true let childToPrune = null for (const i = node.getChildBoundables().iterator(); i.hasNext();) { const childBoundable = i.next() if (!this.getIntersectsOp().intersects(childBoundable.getBounds(), searchBounds)) { continue } if (childBoundable instanceof AbstractNode) { found = this.remove(searchBounds, childBoundable, item) if (found) { childToPrune = childBoundable break } } } if (childToPrune !== null) { if (childToPrune.getChildBoundables().isEmpty()) { node.getChildBoundables().remove(childToPrune) } } return found } } createHigherLevels (boundablesOfALevel, level) { Assert.isTrue(!boundablesOfALevel.isEmpty()) const parentBoundables = this.createParentBoundables(boundablesOfALevel, level + 1) if (parentBoundables.size() === 1) { return parentBoundables.get(0) } return this.createHigherLevels(parentBoundables, level + 1) } depth () { if (arguments.length === 0) { if (this.isEmpty()) { return 0 } this.build() return this.depth(this._root) } else if (arguments.length === 1) { const node = arguments[0] let maxChildDepth = 0 for (const i = node.getChildBoundables().iterator(); i.hasNext();) { const childBoundable = i.next() if (childBoundable instanceof AbstractNode) { const childDepth = this.depth(childBoundable) if (childDepth > maxChildDepth) maxChildDepth = childDepth } } return maxChildDepth + 1 } } createParentBoundables (childBoundables, newLevel) { Assert.isTrue(!childBoundables.isEmpty()) const parentBoundables = new ArrayList() parentBoundables.add(this.createNode(newLevel)) const sortedChildBoundables = new ArrayList(childBoundables) Collections.sort(sortedChildBoundables, this.getComparator()) for (const i = sortedChildBoundables.iterator(); i.hasNext();) { const childBoundable = i.next() if (this.lastNode(parentBoundables).getChildBoundables().size() === this.getNodeCapacity()) { parentBoundables.add(this.createNode(newLevel)) } this.lastNode(parentBoundables).addChildBoundable(childBoundable) } return parentBoundables } isEmpty () { if (!this._built) return this._itemBoundables.isEmpty() return this._root.isEmpty() } interfaces_ () { return [Serializable] } getClass () { return AbstractSTRtree } static compareDoubles (a, b) { return a > b ? 1 : a < b ? -1 : 0 } static get IntersectsOp () { return IntersectsOp } static get serialVersionUID () { return -3886435814360241337 } static get DEFAULT_NODE_CAPACITY () { return 10 } } class IntersectsOp { }