js-draw 1.0.0 → 1.0.1

package/src/EditorImage.ts

@@ -0,0 +1,609 @@
+import Editor from './Editor';
+import AbstractRenderer from './rendering/renderers/AbstractRenderer';
+import Viewport from './Viewport';
+import AbstractComponent from './components/AbstractComponent';
+import { Rect2, Vec2, Mat33, Mat33Array } from '@js-draw/math';
+import { EditorLocalization } from './localization';
+import RenderingCache from './rendering/caching/RenderingCache';
+import SerializableCommand from './commands/SerializableCommand';
+import EventDispatcher from './EventDispatcher';
+import { assertIsNumber, assertIsNumberArray } from './util/assertions';
+
+// @internal Sort by z-index, low to high
+export const sortLeavesByZIndex = (leaves: Array<ImageNode>) => {
+	leaves.sort((a, b) => a.getContent()!.getZIndex() - b.getContent()!.getZIndex());
+};
+
+export enum EditorImageEventType {
+	ExportViewportChanged
+}
+
+export type EditorImageNotifier = EventDispatcher<EditorImageEventType, { image: EditorImage }>;
+
+// Handles lookup/storage of elements in the image
+export default class EditorImage {
+	private root: ImageNode;
+	private background: ImageNode;
+	private componentsById: Record<string, AbstractComponent>;
+	private componentCount: number = 0;
+
+	/** Viewport for the exported/imported image. */
+	private importExportViewport: Viewport;
+
+	// @internal
+	public readonly notifier: EditorImageNotifier;
+
+	// @internal
+	public constructor() {
+		this.root = new ImageNode();
+		this.background = new ImageNode();
+		this.componentsById = {};
+
+		this.notifier = new EventDispatcher();
+		this.importExportViewport = new Viewport(() => {
+			this.notifier.dispatch(EditorImageEventType.ExportViewportChanged, {
+				image: this,
+			});
+		});
+
+		// Default to a 500x500 image
+		this.importExportViewport.updateScreenSize(Vec2.of(500, 500));
+	}
+
+	// Returns all components that make up the background of this image. These
+	// components are rendered below all other components.
+	public getBackgroundComponents(): AbstractComponent[] {
+		const result = [];
+
+		const leaves = this.background.getLeaves();
+		sortLeavesByZIndex(leaves);
+
+		for (const leaf of leaves) {
+			const content = leaf.getContent();
+
+			if (content) {
+				result.push(content);
+			}
+		}
+		return result;
+	}
+
+	// Returns the parent of the given element, if it exists.
+	public findParent(elem: AbstractComponent): ImageNode|null {
+		return this.background.getChildWithContent(elem) ?? this.root.getChildWithContent(elem);
+	}
+
+	// Forces a re-render of `elem` when the image is next re-rendered as a whole.
+	// Does nothing if `elem` is not in this.
+	public queueRerenderOf(elem: AbstractComponent) {
+		// TODO: Make more efficient (e.g. increase IDs of all parents,
+		// make cache take into account last modified time instead of IDs, etc.)
+		const parent = this.findParent(elem);
+
+		if (parent) {
+			parent.remove();
+			this.addElementDirectly(elem);
+		}
+	}
+
+	/** @internal */
+	public renderWithCache(screenRenderer: AbstractRenderer, cache: RenderingCache, viewport: Viewport) {
+		this.background.render(screenRenderer, viewport.visibleRect);
+		cache.render(screenRenderer, this.root, viewport);
+	}
+
+	/**
+	 * Renders all nodes visible from `viewport` (or all nodes if `viewport = null`).
+	 *
+	 * `viewport` is used to improve rendering performance. If given, it must match
+	 * the viewport used by the `renderer` (if any).
+	 */
+	public render(renderer: AbstractRenderer, viewport: Viewport|null) {
+		this.background.render(renderer, viewport?.visibleRect);
+		this.root.render(renderer, viewport?.visibleRect);
+	}
+
+	/** Renders all nodes, even ones not within the viewport. @internal */
+	public renderAll(renderer: AbstractRenderer) {
+		this.render(renderer, null);
+	}
+
+	/** @returns all elements in the image, sorted by z-index. This can be slow for large images. */
+	public getAllElements() {
+		const leaves = this.root.getLeaves();
+		sortLeavesByZIndex(leaves);
+
+		return leaves.map(leaf => leaf.getContent()!);
+	}
+
+	/** Returns the number of elements added to this image. @internal */
+	public estimateNumElements() {
+		return this.componentCount;
+	}
+
+	/** @returns a list of `AbstractComponent`s intersecting `region`, sorted by z-index. */
+	public getElementsIntersectingRegion(region: Rect2): AbstractComponent[] {
+		const leaves = this.root.getLeavesIntersectingRegion(region);
+		sortLeavesByZIndex(leaves);
+
+		return leaves.map(leaf => leaf.getContent()!);
+	}
+
+	/** Called whenever an element is completely removed. @internal */
+	public onDestroyElement(elem: AbstractComponent) {
+		this.componentCount --;
+		delete this.componentsById[elem.getId()];
+	}
+
+	/**
+	 * @returns the AbstractComponent with `id`, if it exists.
+	 *
+	 * @see {@link AbstractComponent.getId}
+	 */
+	public lookupElement(id: string): AbstractComponent|null {
+		return this.componentsById[id] ?? null;
+	}
+
+	private addElementDirectly(elem: AbstractComponent): ImageNode {
+		elem.onAddToImage(this);
+
+		this.componentCount ++;
+		this.componentsById[elem.getId()] = elem;
+
+		// If a background component, add to the background. Else,
+		// add to the normal component tree.
+		const parentTree = elem.isBackground() ? this.background : this.root;
+		return parentTree.addLeaf(elem);
+	}
+
+	private removeElementDirectly(element: AbstractComponent) {
+		const container = this.findParent(element);
+		container?.remove();
+
+		if (container) {
+			this.onDestroyElement(element);
+			return true;
+		}
+
+		return false;
+	}
+
+	/**
+	 * Returns a command that adds the given element to the `EditorImage`.
+	 * If `applyByFlattening` is true, the content of the wet ink renderer is
+	 * rendered onto the main rendering canvas instead of doing a full re-render.
+	 *
+	 * @see {@link Display.flatten}
+	 */
+	public static addElement(elem: AbstractComponent, applyByFlattening: boolean = false): SerializableCommand {
+		return new EditorImage.AddElementCommand(elem, applyByFlattening);
+	}
+
+	/** @see EditorImage.addElement */
+	public addElement(elem: AbstractComponent, applyByFlattening?: boolean) {
+		return EditorImage.addElement(elem, applyByFlattening);
+	}
+
+	// A Command that can access private [EditorImage] functionality
+	private static AddElementCommand = class extends SerializableCommand {
+		private serializedElem: any|null = null;
+
+		// If [applyByFlattening], then the rendered content of this element
+		// is present on the display's wet ink canvas. As such, no re-render is necessary
+		// the first time this command is applied (the surfaces are joined instead).
+		public constructor(
+			private element: AbstractComponent,
+			private applyByFlattening: boolean = false
+		) {
+			super('add-element');
+
+			// FIXME: The serialized version of this command may be inaccurate if this is
+			//        serialized when this command is not on the top of the undo stack.
+			//
+			// Caching the element's serialized data leads to additional memory usage *and*
+			// sometimes incorrect behavior in collaborative editing.
+			this.serializedElem = null;
+
+			if (isNaN(element.getBBox().area)) {
+				throw new Error('Elements in the image cannot have NaN bounding boxes');
+			}
+		}
+
+		public apply(editor: Editor) {
+			editor.image.addElementDirectly(this.element);
+
+			if (!this.applyByFlattening) {
+				editor.queueRerender();
+			} else {
+				this.applyByFlattening = false;
+				editor.display.flatten();
+			}
+		}
+
+		public unapply(editor: Editor) {
+			editor.image.removeElementDirectly(this.element);
+			editor.queueRerender();
+		}
+
+		public description(_editor: Editor, localization: EditorLocalization) {
+			return localization.addElementAction(this.element.description(localization));
+		}
+
+		protected serializeToJSON() {
+			return {
+				elemData: this.serializedElem ?? this.element.serialize(),
+			};
+		}
+
+		static {
+			SerializableCommand.register('add-element', (json: any, editor: Editor) => {
+				const id = json.elemData.id;
+				const foundElem = editor.image.lookupElement(id);
+				const elem = foundElem ?? AbstractComponent.deserialize(json.elemData);
+				const result = new EditorImage.AddElementCommand(elem);
+				result.serializedElem = json.elemData;
+				return result;
+			});
+		}
+	};
+
+
+
+	/**
+	 * @returns a `Viewport` for rendering the image when importing/exporting.
+	 */
+	public getImportExportViewport() {
+		return this.importExportViewport;
+	}
+
+	public setImportExportRect(imageRect: Rect2): SerializableCommand {
+		const importExportViewport = this.getImportExportViewport();
+		const origSize = importExportViewport.visibleRect.size;
+		const origTransform = importExportViewport.canvasToScreenTransform;
+
+		return new EditorImage.SetImportExportRectCommand(origSize, origTransform, imageRect);
+	}
+
+	// Handles resizing the background import/export region of the image.
+	private static SetImportExportRectCommand = class extends SerializableCommand {
+		private static commandId = 'set-import-export-rect';
+
+		public constructor(
+			private originalSize: Vec2,
+			private originalTransform: Mat33,
+			private finalRect: Rect2,
+		) {
+			super(EditorImage.SetImportExportRectCommand.commandId);
+		}
+
+		public apply(editor: Editor) {
+			const viewport = editor.image.getImportExportViewport();
+			viewport.updateScreenSize(this.finalRect.size);
+			viewport.resetTransform(Mat33.translation(this.finalRect.topLeft.times(-1)));
+			editor.queueRerender();
+		}
+
+		public unapply(editor: Editor) {
+			const viewport = editor.image.getImportExportViewport();
+			viewport.updateScreenSize(this.originalSize);
+			viewport.resetTransform(this.originalTransform);
+			editor.queueRerender();
+		}
+
+		public description(_editor: Editor, localization: EditorLocalization) {
+			return localization.resizeOutputCommand(this.finalRect);
+		}
+
+		protected serializeToJSON() {
+			return {
+				originalSize: this.originalSize.xy,
+				originalTransform: this.originalTransform.toArray(),
+				newRegion: {
+					x: this.finalRect.x,
+					y: this.finalRect.y,
+					w: this.finalRect.w,
+					h: this.finalRect.h,
+				},
+			};
+		}
+
+		static {
+			const commandId = this.commandId;
+			SerializableCommand.register(commandId, (json: any, _editor: Editor) => {
+				assertIsNumber(json.originalSize.x);
+				assertIsNumber(json.originalSize.y);
+				assertIsNumberArray(json.originalTransform);
+				assertIsNumberArray([
+					json.newRegion.x,
+					json.newRegion.y,
+					json.newRegion.w,
+					json.newRegion.h,
+				]);
+
+				const originalSize = Vec2.ofXY(json.originalSize);
+				const originalTransform = new Mat33(...(json.originalTransform as Mat33Array));
+				const finalRect = new Rect2(
+					json.newRegion.x, json.newRegion.y, json.newRegion.w, json.newRegion.h
+				);
+				return new EditorImage.SetImportExportRectCommand(
+					originalSize, originalTransform, finalRect
+				);
+			});
+		}
+	};
+}
+
+type TooSmallToRenderCheck = (rect: Rect2)=> boolean;
+
+/** Part of the Editor's image. @internal */
+export class ImageNode {
+	private content: AbstractComponent|null;
+	private bbox: Rect2;
+	private children: ImageNode[];
+	private targetChildCount: number = 30;
+
+	private id: number;
+	private static idCounter: number = 0;
+
+	public constructor(
+		private parent: ImageNode|null = null
+	) {
+		this.children = [];
+		this.bbox = Rect2.empty;
+		this.content = null;
+
+		this.id = ImageNode.idCounter++;
+	}
+
+	public getId() {
+		return this.id;
+	}
+
+	public onContentChange() {
+		this.id = ImageNode.idCounter++;
+	}
+
+	public getContent(): AbstractComponent|null {
+		return this.content;
+	}
+
+	public getParent(): ImageNode|null {
+		return this.parent;
+	}
+
+	private getChildrenIntersectingRegion(region: Rect2): ImageNode[] {
+		return this.children.filter(child => {
+			return child.getBBox().intersects(region);
+		});
+	}
+
+	public getChildrenOrSelfIntersectingRegion(region: Rect2): ImageNode[] {
+		if (this.content) {
+			return [this];
+		}
+		return this.getChildrenIntersectingRegion(region);
+	}
+
+	// Returns a list of `ImageNode`s with content (and thus no children).
+	public getLeavesIntersectingRegion(region: Rect2, isTooSmall?: TooSmallToRenderCheck): ImageNode[] {
+		const result: ImageNode[] = [];
+		let current: ImageNode|undefined;
+		const workList: ImageNode[] = [];
+
+		workList.push(this);
+		const toNext = () => {
+			current = undefined;
+
+			const next = workList.pop();
+			if (next && !isTooSmall?.(next.bbox)) {
+				current = next;
+
+				if (current.content !== null && current.getBBox().intersection(region)) {
+					result.push(current);
+				}
+
+				workList.push(
+					...current.getChildrenIntersectingRegion(region)
+				);
+			}
+		};
+
+		while (workList.length > 0) {
+			toNext();
+		}
+
+		return result;
+	}
+
+	// Returns the child of this with the target content or `null` if no
+	// such child exists.
+	public getChildWithContent(target: AbstractComponent): ImageNode|null {
+		const candidates = this.getLeavesIntersectingRegion(target.getBBox());
+		for (const candidate of candidates) {
+			if (candidate.getContent() === target) {
+				return candidate;
+			}
+		}
+
+		return null;
+	}
+
+	// Returns a list of leaves with this as an ancestor.
+	// Like getLeavesInRegion, but does not check whether ancestors are in a given rectangle
+	public getLeaves(): ImageNode[] {
+		if (this.content) {
+			return [this];
+		}
+
+		const result: ImageNode[] = [];
+		for (const child of this.children) {
+			result.push(...child.getLeaves());
+		}
+
+		return result;
+	}
+
+	public addLeaf(leaf: AbstractComponent): ImageNode {
+		this.onContentChange();
+
+		if (this.content === null && this.children.length === 0) {
+			this.content = leaf;
+			this.recomputeBBox(true);
+
+			return this;
+		}
+
+		if (this.content !== null) {
+			console.assert(this.children.length === 0);
+
+			const contentNode = new ImageNode(this);
+			contentNode.content = this.content;
+			this.content = null;
+			this.children.push(contentNode);
+			contentNode.recomputeBBox(false);
+		}
+
+		// If this node is contained within the leaf, make this and the leaf
+		// share a parent.
+		const leafBBox = leaf.getBBox();
+		if (leafBBox.containsRect(this.getBBox())) {
+			const nodeForNewLeaf = new ImageNode(this);
+
+			if (this.children.length < this.targetChildCount) {
+				this.children.push(nodeForNewLeaf);
+			} else {
+				const nodeForChildren = new ImageNode(this);
+
+				nodeForChildren.children = this.children;
+				this.children = [nodeForNewLeaf, nodeForChildren];
+				nodeForChildren.recomputeBBox(true);
+				nodeForChildren.updateParents();
+			}
+			return nodeForNewLeaf.addLeaf(leaf);
+		}
+
+		const containingNodes = this.children.filter(
+			child => child.getBBox().containsRect(leafBBox)
+		);
+
+		// Does the leaf already fit within one of the children?
+		if (containingNodes.length > 0 && this.children.length >= this.targetChildCount) {
+			// Sort the containers in ascending order by area
+			containingNodes.sort((a, b) => a.getBBox().area - b.getBBox().area);
+
+			// Choose the smallest child that contains the new element.
+			const result = containingNodes[0].addLeaf(leaf);
+			result.rebalance();
+			return result;
+		}
+
+
+		const newNode = new ImageNode(this);
+		this.children.push(newNode);
+		newNode.content = leaf;
+		newNode.recomputeBBox(true);
+
+		return newNode;
+	}
+
+	public getBBox(): Rect2 {
+		return this.bbox;
+	}
+
+	// Recomputes this' bounding box. If [bubbleUp], also recompute
+	// this' ancestors bounding boxes. This also re-computes this' bounding box
+	// in the z-direction (z-indicies).
+	public recomputeBBox(bubbleUp: boolean) {
+		const oldBBox = this.bbox;
+		if (this.content !== null) {
+			this.bbox = this.content.getBBox();
+		} else {
+			this.bbox = Rect2.union(...this.children.map(child => child.getBBox()));
+		}
+
+		if (bubbleUp && !oldBBox.eq(this.bbox)) {
+			this.parent?.recomputeBBox(true);
+		}
+	}
+
+	private updateParents(recursive: boolean = false) {
+		for (const child of this.children) {
+			child.parent = this;
+
+			if (recursive) {
+				child.updateParents(recursive);
+			}
+		}
+	}
+
+	private rebalance() {
+		// If the current node is its parent's only child,
+		if (this.parent && this.parent.children.length === 1) {
+			console.assert(this.parent.content === null);
+			console.assert(this.parent.children[0] === this);
+
+			// Remove this' parent, if this' parent isn't the root.
+			const oldParent = this.parent;
+			if (oldParent.parent !== null) {
+				oldParent.children = [];
+				this.parent = oldParent.parent;
+				this.parent.children.push(this);
+				oldParent.parent = null;
+				this.parent.recomputeBBox(false);
+			} else if (this.content === null) {
+				// Remove this and transfer this' children to the parent.
+				this.parent.children = this.children;
+				this.parent.updateParents();
+				this.parent = null;
+			}
+		}
+	}
+
+	// Remove this node and all of its children
+	public remove() {
+		this.content?.onRemoveFromImage();
+
+		if (!this.parent) {
+			this.content = null;
+			this.children = [];
+
+			return;
+		}
+
+		const oldChildCount = this.parent.children.length;
+		this.parent.children = this.parent.children.filter(node => {
+			return node !== this;
+		});
+
+		console.assert(
+			this.parent.children.length === oldChildCount - 1,
+			`${oldChildCount - 1} ≠ ${this.parent.children.length} after removing all nodes equal to ${this}. Nodes should only be removed once.`
+		);
+
+		this.parent.children.forEach(child => {
+			child.rebalance();
+		});
+
+		this.parent.recomputeBBox(true);
+
+		// Invalidate/disconnect this.
+		this.content = null;
+		this.parent = null;
+		this.children = [];
+	}
+
+	public render(renderer: AbstractRenderer, visibleRect?: Rect2) {
+		let leaves;
+		if (visibleRect) {
+			leaves = this.getLeavesIntersectingRegion(visibleRect, rect => renderer.isTooSmallToRender(rect));
+		} else {
+			leaves = this.getLeaves();
+		}
+		sortLeavesByZIndex(leaves);
+
+		for (const leaf of leaves) {
+			// Leaves by definition have content
+			leaf.getContent()!.render(renderer, visibleRect);
+		}
+	}
+}