jsketcher/web/app/sketcher/selectionMatcher.js

export function matchSelection(definition, matchIndex, fast) {
  const selection = matchIndex.selection;
  if (definition.selector === 'function') {
    return definition.match(selection, fast)
  } else if (definition.selector === 'matchAll') {

    const {minQuantity: min, types} = definition;
    if (min !== undefined && selection.length < min) {
      return false;
    }
    for (let obj of selection) {

      let hit = false;
      for (let constructor of types) {
        if (constructor.prototype._class === obj._class) {
          hit = true;
          break;
        }
      }
      if (!hit) {
        return false;
      }
    }
    return fast ? true : selection;

  } else if (definition.selector === 'matchSequence') {

    matchIndex.reset(fast);


    for (let item of definition.sequence) {
      if (!matchIndex.mark(item.types, item.quantity)) {
        return false;
      }
    }

    return matchIndex.allHit() ? (fast ? true : matchIndex.result) : false;

  } else {
    throw 'unsupported'
  }

}

export function getDescription(definition) {

  if (definition.selector === 'function') {
    return definition.description;
  } else if (definition.selector === 'matchAll') {
    return `at least ${definition.minQuantity} of ${stringifyTypes(definition.types, definition.minQuantity)}`;
  } else if (definition.selector === 'matchSequence') {

    let out = '';


    for (let i = 0; i< definition.sequence.length; ++i) {
      const item = definition.sequence[i];
      if (i !== 0) {
        out += i === definition.sequence.length - 1 ? ' and ': ', ';
      }

      out += item.quantity + ' ' + stringifyTypes(item.types, item.quantity);
    }

    return out;

  } else {
    throw 'unsupported'
  }
}

function stringifyTypes(types, minQuantity) {
  return types.map(t => t.prototype._class.replace('TCAD.TWO.', '') + (minQuantity > 1 ? 's' : '')).join(' or ');
}

export class MatchIndex {

  typeMap = new Map();

  overallHits = 0;

  constructor(selection) {
    this.selection = selection;
    selection.forEach(obj => {
      let info = this.typeMap.get(obj._class);
      if (!info) {
        info = {
          hits: 0,
          objects: []
        };
        this.typeMap.set(obj._class, info);
      }
      info.objects.push(obj);
    })
  }

  reset(fast) {
    this.overallHits = 0;
    this.typeMap.forEach(i => i.hits = 0);
    this.result = fast ? null : [];
  }

  mark(types, quantity) {
    for (let type of types) {
      const info = this.typeMap.get(type.prototype._class);
      if (!info) {
        continue;
      }
      const toAdd = Math.min(quantity, info.objects.length - info.hits);
      if (this.result) {
        for (let i = 0; i < toAdd; ++i) {
          this.result.push(info.objects[info.hits + i]);
        }
      }
      quantity -= toAdd;
      info.hits += toAdd;
      this.overallHits += toAdd;
    }
    return quantity === 0;
  }

  allHit() {
    return this.selection.length === this.overallHits;
  }

}