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
JavaScript
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 {
}