123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
import type {  AdjacencyStructure,  AdjacencyStructureConstructor,  IndexedCollection,  TypedArrayConstructors,} from "./utility-types.ts";
/** * Creates an Adjacency Matrix class extending a given TypedArray class. * * @param Base a TypedArray class to extend */export function AdjacencyMatrixWeightedDirectedMixin<  U extends TypedArrayConstructors,>(Base: U): AdjacencyStructureConstructor<U> {  // deno-lint-ignore no-empty-interface  interface AdjacencyMatrixWeightedDirected extends IndexedCollection {}  /**   * Implements Adjacency Matrix for weighted directed graphs.   */  class AdjacencyMatrixWeightedDirected extends Base    implements AdjacencyStructure {    static directed = true;    static weighted = true;    empty = 0;
    static get [Symbol.species](): U {      return Base;    }
    _vertices = 0;
    get vertices() {      return (        this._vertices ||        (this._vertices = (this          .constructor as typeof AdjacencyMatrixWeightedDirected).getVertices(            this.length,          )), this._vertices      );    }
    get edges() {      return this.vertices ** 2;    }
    static create<      T extends AdjacencyStructureConstructor<TypedArrayConstructors>,    >(vertices: number): InstanceType<T> {      const length = this.getLength(vertices);      return new this(length) as InstanceType<T>;    }
    static getLength(vertices: number): number {      return vertices * vertices;    }
    protected static getVertices(length: number): number {      return Math.sqrt(length);    }
    addEdge(x: number, y: number, weight: number): this {      this[this.getIndex(x, y)] = weight;      return this;    }
    getEdge(x: number, y: number) {      return this[this.getIndex(x, y)];    }
    getIndex(x: number, y: number): number {      return x * this.vertices + y;    }
    hasEdge(x: number, y: number) {      const edge = this.getEdge(x, y);      return edge !== undefined && edge !== this.empty;    }
    *inEdges(vertex: number) {      const { vertices } = this;      for (let i = 0; i < vertices; i++) {        if (this.hasEdge(i, vertex)) yield i;      }    }
    isFull(): boolean {      return false;    }
    *outEdges(vertex: number) {      const { vertices } = this;      const offset = vertex * vertices;      for (let i = 0; i < vertices; i++) {        const edge = this[offset + i];        if (edge !== undefined && edge !== this.empty) yield i;      }    }
    removeEdge(x: number, y: number): this {      this[this.getIndex(x, y)] = this.empty;      return this;    }  }
  return AdjacencyMatrixWeightedDirected;}