| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281 |
- 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() {
- AbstractSTRtree.constructor_.apply(this, arguments)
- }
- static constructor_() {
- this._root = null
- this._built = false
- this._itemBoundables = new ArrayList()
- this._nodeCapacity = null
- if (arguments.length === 0) {
- AbstractSTRtree.constructor_.call(this, AbstractSTRtree.DEFAULT_NODE_CAPACITY)
- } else if (arguments.length === 1) {
- const nodeCapacity = arguments[0]
- Assert.isTrue(nodeCapacity > 1, 'Node capacity must be greater than 1')
- this._nodeCapacity = nodeCapacity
- }
- }
- static compareDoubles(a, b) {
- return a > b ? 1 : a < b ? -1 : 0
- }
- queryInternal() {
- if (hasInterface(arguments[2], ItemVisitor) && (arguments[0] instanceof Object && arguments[1] instanceof AbstractNode)) {
- const searchBounds = arguments[0], node = arguments[1], 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.queryInternal(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], node = arguments[1], 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.queryInternal(searchBounds, childBoundable, matches)
- else if (childBoundable instanceof ItemBoundable)
- matches.add(childBoundable.getItem())
- else
- Assert.shouldNeverReachHere()
-
- }
- }
- }
- 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 (let 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 (let 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 (let 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], top = arguments[1], boundables = arguments[2]
- Assert.isTrue(level > -2)
- if (top.getLevel() === level) {
- boundables.add(top)
- return null
- }
- for (let 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.queryInternal(searchBounds, this._root, matches)
-
- return matches
- } else if (arguments.length === 2) {
- const searchBounds = arguments[0], visitor = arguments[1]
- this.build()
- if (this.isEmpty())
- return null
-
- if (this.getIntersectsOp().intersects(this._root.getBounds(), searchBounds))
- this.queryInternal(searchBounds, this._root, visitor)
-
- }
- }
- 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], 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], node = arguments[1], item = arguments[2]
- let found = this.removeItem(node, item)
- if (found) return true
- let childToPrune = null
- for (let 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 (let 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 (let 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()
- }
- get interfaces_() {
- return [Serializable]
- }
- }
- function IntersectsOp() {}
- AbstractSTRtree.IntersectsOp = IntersectsOp
- AbstractSTRtree.DEFAULT_NODE_CAPACITY = 10
|