@immugio/three-math-extensions 0.0.1

package/src/Line3D.ts

@@ -0,0 +1,448 @@
+import { Line3, Vector3 } from "three";
+import { Vec3 } from "./Vec3";
+import { Point3 } from "./Point3";
+import { Line2D } from "./Line2D";
+
+export class Line3D extends Line3 {
+
+    public declare start: Vec3;
+    public declare end: Vec3;
+
+    #target = new Vec3();
+
+    constructor(start: Vec3, end: Vec3) {
+        super(start, end);
+    }
+
+    public static fromPoints(start: Point3, end: Point3): Line3D {
+        return new Line3D(new Vec3(start.x, start.y, start.z), new Vec3(end.x, end.y, end.z));
+    }
+
+    /**
+     * Returns lines that are the result of clipping this line by the @other line.
+     * Clips must be parallel to this line.
+     * Clones the line, does not modify this.
+     * @param other
+     * @param parallelTolerance
+     */
+    public clipLine(other: Line3D, parallelTolerance: number = Number.EPSILON): Line3D[] {
+        other = this.getParallelLineInTheSameDirection(other, parallelTolerance);
+
+        // 1) Lines aren't parallel
+        if (!other) {
+            return [this.clone()];
+        }
+
+        const left = this.clone();
+        left.end.copy(other.start);
+
+        const right = this.clone();
+        right.start.copy(other.end);
+
+        return [left, right].filter(x => x.direction.manhattanDistanceTo(this.direction) <= parallelTolerance);
+    }
+
+    /**
+     * Returns lines that are the result of clipping this line by the @clips.
+     * Clips must be parallel to this line.
+     * Clones the line, does not modify this.
+     * @param clips
+     * @param distanceTolerance
+     * @param parallelTolerance
+     */
+    public clipLines(clips: Line3D[], distanceTolerance: number = 0, parallelTolerance: number = Number.EPSILON): Line3D[] {
+        const free: Line3D[] = [];
+        const sources: Line3D[] = [this.clone()];
+
+        while (sources.length > 0) {
+
+            let isFree = true;
+
+            const tested = sources.pop();
+
+            for (const clip of clips) {
+
+                if (tested.overlaps(clip, distanceTolerance, parallelTolerance)) {
+                    isFree = false;
+                    const subtracted = tested.clipLine(clip, parallelTolerance);
+                    sources.push(...subtracted);
+                    break;
+                }
+            }
+
+            if (isFree) free.push(tested);
+        }
+
+        return free;
+    }
+
+    /**
+     * Joins a copy of this line with the @other line.
+     * Other must be parallel to this line.
+     * Returns null if there is no overlap
+     * Clones the line, does not modify this.
+     * @param other
+     * @param distanceTolerance
+     * @param parallelTolerance
+     */
+    public joinLine(other: Line3D, distanceTolerance: number = 0, parallelTolerance: number = Number.EPSILON): Line3D {
+        // 6 possible cases:
+        const otherParallel = this.getParallelLineInTheSameDirection(other, parallelTolerance);
+
+        // 1) Lines aren't parallel
+        if (!otherParallel) {
+            return null;
+        }
+
+        const thisContainsOtherStartPoint = this.containsPoint(otherParallel.start, distanceTolerance);
+        const thisContainsOtherEndPoint = this.containsPoint(otherParallel.end, distanceTolerance);
+        const otherContainsThisStartPoint = otherParallel.containsPoint(this.start, distanceTolerance);
+        const otherContainsThisEndPoint = otherParallel.containsPoint(this.end, distanceTolerance);
+
+        // 2) Lines don't overlap at all
+        if (
+            !thisContainsOtherStartPoint &&
+            !thisContainsOtherEndPoint &&
+            !otherContainsThisStartPoint &&
+            !otherContainsThisEndPoint
+        ) {
+            return null;
+        }
+
+        // 3) This line entirely covers the other line
+        if (thisContainsOtherStartPoint && thisContainsOtherEndPoint) {
+            return this.clone();
+        }
+
+        // 4) The other line entirely covers this line
+        if (otherContainsThisStartPoint && otherContainsThisEndPoint) {
+            return otherParallel.clone();
+        }
+
+        // 5) This line is overlapped by the start of the other line
+        if (thisContainsOtherStartPoint && !thisContainsOtherEndPoint) {
+            return new Line3D(this.start, otherParallel.end);
+        }
+
+        // 6) This line is overlapped by the end of the other line
+        if (thisContainsOtherEndPoint && !thisContainsOtherStartPoint) {
+            return new Line3D(otherParallel.start, this.end);
+        }
+
+        return null;
+    }
+
+    /**
+     * Joins provided lines into several joined lines.
+     * Lines must be parallel for joining.
+     * @param lines
+     * @param distanceTolerance
+     * @param parallelTolerance
+     */
+    public static joinLines(lines: Line3D[], distanceTolerance: number = 0, parallelTolerance: number = Number.EPSILON): Line3D[] {
+        if (lines.length < 2) {
+            return lines.map(x => x.clone());
+        }
+
+        const toProcess = lines.slice();
+        const result: Line3D[] = [];
+
+        while (toProcess.length > 0) {
+            const current = toProcess.pop();
+            let joinedLine: Line3D;
+
+            for (let i = 0; i < result.length; i++) {
+                const other = result[i];
+                joinedLine = current.joinLine(other, distanceTolerance, parallelTolerance);
+                if (joinedLine) {
+                    result.splice(i, 1);
+                    toProcess.push(joinedLine);
+                    break;
+                }
+            }
+
+            if (!joinedLine) {
+                result.push(current.clone());
+            }
+        }
+
+        return result;
+    }
+
+    /**
+     * Returns true if this line section completely overlaps the @other line section.
+     * @param other
+     * @param tolerance
+     */
+    public covers(other: Line3D, tolerance: number = 0): boolean {
+        return this.containsPoint(other.start, tolerance) && this.containsPoint(other.end, tolerance);
+    }
+
+    /**
+     * Returns true if there is any overlap between this line and the @other line section.
+     * @param other
+     * @param distanceTolerance
+     * @param parallelTolerance
+     */
+    public overlaps(other: Line3D, distanceTolerance: number = 0, parallelTolerance: number = Number.EPSILON): boolean {
+        // Special case
+        if (this.equals(other, distanceTolerance)) {
+            return true;
+        }
+
+        // Always have to be parallel
+        if (this.isParallelTo(other, parallelTolerance)) {
+            // To pass as overlapping, at least one point has to be within the other line, but they mush not be equal - "touching" is not considered overlapping
+
+            const otherStartEqualsToAnyOfThisPoint = other.start.distanceTo(this.start) <= distanceTolerance || other.start.distanceTo(this.end) <= distanceTolerance;
+            if (this.containsPoint(other.start, distanceTolerance) && !otherStartEqualsToAnyOfThisPoint) {
+                return true;
+            }
+
+            const otherEndEqualsToAnyOfThisPoint = other.end.distanceTo(this.start) <= distanceTolerance || other.end.distanceTo(this.end) <= distanceTolerance;
+            if (this.containsPoint(other.end, distanceTolerance) && !otherEndEqualsToAnyOfThisPoint) {
+                return true;
+            }
+
+            const thisStartEqualsToAnyOfOtherPoint = this.start.distanceTo(other.start) <= distanceTolerance || this.start.distanceTo(other.end) <= distanceTolerance;
+            if (other.containsPoint(this.start, distanceTolerance) && !thisStartEqualsToAnyOfOtherPoint) {
+                return true;
+            }
+
+            const thisEndEqualsToAnyOfOtherPoint = this.end.distanceTo(other.start) <= distanceTolerance || this.end.distanceTo(other.end) <= distanceTolerance;
+            if (other.containsPoint(this.end, distanceTolerance) && !thisEndEqualsToAnyOfOtherPoint) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
+    /**
+     * Returns this line's length.
+     */
+    public get length(): number {
+        return this.start.distanceTo(this.end);
+    }
+
+    /**
+     * Set the length of this line. Center and direction remain unchanged.
+     * @param length
+     */
+    public setLength(length: number): this {
+        const diff = length - this.length;
+        return this.resize(diff);
+    }
+
+    /**
+     * Returns the direction of this line.
+     */
+    public get direction(): Vec3 {
+        return this.end.clone().sub(this.start).normalize();
+    }
+
+    /**
+     * Returns the center of this line
+     */
+    public get center(): Vec3 {
+        return this.getCenter(new Vec3()) as Vec3;
+    }
+
+    /**
+     * Set the center of the line to the provided point. Length and direction remain unchanged.
+     * @param value
+     */
+    public setCenter(value: Vector3): this {
+        const current = this.center;
+        const diffX = current.x - value.x;
+        const diffY = current.y - value.y;
+        const diffZ = current.z - value.z;
+        this.start.x -= diffX;
+        this.start.y -= diffY;
+        this.start.z -= diffZ;
+        this.end.x -= diffX;
+        this.end.y -= diffY;
+        this.end.z -= diffZ;
+        return this;
+    }
+
+    /** Returns the start and end points of the line as an array. */
+    public get endpoints(): Vec3[] {
+        return [this.start, this.end];
+    }
+
+    /**
+     * Check that this line section contains provided point.
+     * @param p
+     * @param tolerance
+     */
+    public containsPoint(p: Vector3, tolerance: number = 0): boolean {
+        const closestPointToPoint = this.closestPointToPoint(p, true, this.#target);
+        return closestPointToPoint.distanceTo(p) <= tolerance;
+    }
+
+    /**
+     * Distance from this line to provided point.
+     * @param p
+     * @param clampToLine
+     */
+    public distanceToPoint(p: Vector3, clampToLine: boolean = true): number {
+        const closestPointToPoint = this.closestPointToPoint(p, clampToLine, this.#target);
+        return closestPointToPoint.distanceTo(p);
+    }
+
+    /**
+     * Returns a copy of @other line, the direction of @other is reversed if needed.
+     * Returns null if lines are not parallel.
+     * @param other
+     * @param tolerance
+     */
+    public getParallelLineInTheSameDirection(other: Line3D, tolerance: number = Number.EPSILON): Line3D {
+        const direction = this.direction;
+
+        const areTheSameDirection = direction.manhattanDistanceTo(other.direction) < tolerance;
+        if (areTheSameDirection) {
+            return other.clone();
+        }
+
+        const otherLineOppositeDirection = new Line3D(other.end, other.start);
+        if (otherLineOppositeDirection.direction.manhattanDistanceTo(direction) < tolerance) {
+            return otherLineOppositeDirection;
+        }
+
+        return null;
+    }
+
+    /**
+     * Check if @other is parallel to this line.
+     * @param other
+     * @param tolerance
+     */
+    public isParallelTo(other: Line3D, tolerance: number = Number.EPSILON): boolean {
+        const direction = this.direction;
+        const otherDirection = other.direction;
+
+        const areTheSameDirection = direction.manhattanDistanceTo(otherDirection) <= tolerance;
+        if (areTheSameDirection) {
+            return true;
+        }
+
+        return direction.negate().manhattanDistanceTo(otherDirection) < tolerance;
+    }
+
+    /*
+    * Extends or reduces the line to the given length while keeping the center of the line constant.
+    */
+    public resize(amount: number): this {
+        this.moveStartPoint(amount / 2);
+        this.moveEndPoint(amount / 2);
+        return this;
+    }
+
+    /*
+    * Moves start on the line by the given amount. Plus values move the point further away from the center.
+    */
+    public moveStartPoint(amount: number): Line3D {
+        const start = this.movePointOnThisLine(this.start, amount);
+        this.start.x = start.x;
+        this.start.y = start.y;
+        this.start.z = start.z;
+
+        return this;
+    }
+
+    /*
+    * Moves end on the line by the given amount in the current direction. Plus values move the point further away from the center.
+    */
+    public moveEndPoint(amount: number): Line3D {
+        const end = this.movePointOnThisLine(this.end, amount);
+        this.end.x = end.x;
+        this.end.y = end.y;
+        this.end.z = end.z;
+
+        return this;
+    }
+
+    /**
+     * Returns a new line that is the projection of this line onto @other. Uses `closestPointToPoint` to find the projection.
+     * @param other
+     * @param clampToLine
+     */
+    public projectOn(other: Line3D, clampToLine: boolean): Line3D {
+        const p1 = other.closestPointToPoint(this.start, clampToLine, new Vec3()) as Vec3;
+        const p2 = other.closestPointToPoint(this.end, clampToLine, new Vec3()) as Vec3;
+
+        return p1.distanceTo(this.start) < p2.distanceTo(this.start) ? new Line3D(p1, p2) : new Line3D(p2, p1);
+    }
+
+    /**
+     * Divides the Line3D into a number of segments of the given length.
+     * @param maxSegmentLength number
+     */
+    public chunk(maxSegmentLength: number): Line3D[] {
+        const source = this.clone();
+        const result: Line3D[] = [];
+        while (source.length > maxSegmentLength) {
+            const chunk = source.clone();
+            chunk.moveEndPoint(-(chunk.length - maxSegmentLength));
+            result.push(chunk);
+            source.start.copy(chunk.end);
+        }
+        if (source.length > 0) {
+            result.push(source);
+        }
+        return result;
+    }
+
+    /**
+     * Note that this works well for moving the endpoints as it's currently used
+     * If it were to be made public, it would need to handle the situation where the point to move is in the center of the line which would require a different approach
+     */
+    private movePointOnThisLine(point: Vec3, amount: number): Vec3 {
+        const center = this.getCenter(this.#target);
+        const vec = new Vec3(center.x - point.x, center.y - point.y, center.z - point.z);
+        const length = vec.length();
+        vec.normalize().multiplyScalar(length + amount);
+
+        return new Vec3(
+            center.x - vec.x,
+            center.y - vec.y,
+            center.z - vec.z
+        );
+    }
+
+    /**
+     * Move this line by the given vector.
+     * @param p
+     */
+    public translate(p: Vector3): this {
+        this.start.add(p);
+        this.end.add(p);
+        return this;
+    }
+
+    /**
+     * Project the line to 2D space, Y value is dropped
+     */
+    public onPlan(): Line2D {
+        return new Line2D(this.start.onPlan(), this.end.onPlan());
+    }
+
+    /**
+     * Equals with tolerance
+     */
+    public equals(other: Line3D, tolerance: number = 0): boolean {
+        return !!other && this.start.distanceTo(other.start) <= tolerance && this.end.distanceTo(other.end) <= tolerance;
+    }
+
+    /**
+     * Deep clone of this line
+     */
+    public clone(): this {
+        return new Line3D(this.start.clone(), this.end.clone()) as this;
+    }
+
+    public toString(): string {
+        return `Line3D { start: ${this.start.x}, ${this.start.y}, ${this.start.z}, end: ${this.end.x}, ${this.end.y}, ${this.end.z}}`;
+    }
+}

package/src/Point2.ts

@@ -0,0 +1,4 @@
+export interface Point2 {
+    x: number,
+    y: number,
+}

package/src/Point3.ts

@@ -0,0 +1,6 @@
+export interface Point3 {
+    x: number,
+    y: number,
+    z: number
+}
+

package/src/Vec2.ts

@@ -0,0 +1,18 @@
+import { Vector2 } from "three";
+
+export class Vec2 extends Vector2 {
+
+    public static fromPoint(point: { x: number, y: number }): Vec2 {
+        return new Vec2(point.x, point.y);
+    }
+
+    public roundIfCloseToInteger(max: number = 0.000000000001): this {
+        if (Math.abs(this.x - Math.round(this.x)) < max) {
+            this.x = Math.round(this.x);
+        }
+        if (Math.abs(this.y - Math.round(this.y)) < max) {
+            this.y = Math.round(this.y);
+        }
+        return this;
+    }
+}

package/src/Vec3.ts

@@ -0,0 +1,70 @@
+import { Vector3 } from "three";
+import { Vec2 } from "./Vec2";
+import { Point3 } from "./Point3";
+
+export class Vec3 extends Vector3 {
+
+    #target: Vector3;
+
+    public static fromPoint(point: Point3): Vec3 {
+        return new Vec3(point.x, point.y, point.z);
+    }
+
+    public moveTowards(target: Vector3, amount: number): Vec3 {
+        if (this.#target === undefined) {
+            this.#target = new Vector3();
+        }
+
+        this.#target.copy(target);
+        this.add(this.#target.sub(this).normalize().multiplyScalar(amount));
+
+        return this;
+    }
+
+    public centerTowards(target: Vector3): Vec3 {
+        if (this.#target === undefined) {
+            this.#target = new Vector3();
+        }
+
+        return this.moveTowards(target, this.distanceTo(target) / 2);
+    }
+
+    public addY(y: number): Vec3 {
+        this.y += y;
+        return this;
+    }
+
+    public addX(x: number): Vec3 {
+        this.x += x;
+        return this;
+    }
+
+    public scale(p: Vector3): Vec3 {
+        this.x *= p.x;
+        this.y *= p.y;
+        this.z *= p.z;
+        return this;
+    }
+
+    public closest(...points: Vector3[]): Vec3 {
+        const withDistances = points.map(p => ({ point: p, distance: this.distanceTo(p) }));
+        const closest = withDistances.reduce((a, b) => a.distance < b.distance ? a : b);
+        return Vec3.fromPoint(closest.point);
+    }
+
+    public toPointWithFlippedYZ(): Vec3 {
+        return new Vec3(this.x, this.z, this.y);
+    }
+
+    public onPlan(): Vec2 {
+        return new Vec2(this.x, this.z);
+    }
+
+    public horizontalDistanceTo(point: Vector3): number {
+        return new Vector3(this.x, 0, this.z).distanceTo(new Vector3(point.x, 0, point.z));
+    }
+
+    public clone(): this {
+        return super.clone();
+    }
+}