index.js

var BKD = require('bkd-tree')
var once = require('once')

module.exports = MBKD

function MBKD (opts) {
  if (!(this instanceof MBKD)) return new MBKD(opts)
  this.storage = opts.storage
  this._getPoint = opts.getPoint
  this._isLinked = opts.isLinked
  this.bkd = new BKD(this.storage, {
    type: opts.type,
    branchFactor: opts.branchFactor,
    levels: opts.levels,
    compare: opts.compare
  })
}

MBKD.prototype.batch = function (rows, cb) {
  var self = this
  cb = once(cb || noop)
  var ops = []
  var pending = 1
  rows.forEach(function (row) {
    var links = row.links || []
    links.forEach(function (link) {
      pending++
      self._getPoint(link, function (err, pt) {
        if (err) return cb(err)
        if (pt) {
          ops.push({
            type: 'delete',
            point: pt.point,
            value: pt.value
          })
        }
        if (--pending === 0) done()
      })
    })
    pending++
    self._isLinked(row.id, function (err, ex) {
      if (err) return cb(err)
      if (!ex) {
        ops.push({
          type: row.type === 'delete' ? 'delete' : 'insert',
          point: row.point,
          value: Array.isArray(row.id) ? row.id : [row.id]
        })
      }
      if (--pending === 0) done()
    })
  })
  if (--pending === 0) done()

  function done () {
    self.bkd.batch(ops, cb)
  }
}

MBKD.prototype.query = function (bbox, cb) {
  return this.bkd.query(bbox, cb)
}

function noop () {}