Files
workinf_Blender_Wasm/web/protocol/grease-pencil-editor.ts
2026-08-12 19:31:41 -04:00

189 lines
10 KiB
TypeScript

export const GREASE_PENCIL_EDITOR_SCHEMA = 1 as const;
export const GREASE_PENCIL_EDITOR_BUDGET = { maxSelection: 1_000_000, maxIdLength: 256 } as const;
export interface GreasePencilPointSelectionIR { strokeIndex: number; pointIndex: number }
export interface GreasePencilEditorIR {
schemaVersion: typeof GREASE_PENCIL_EDITOR_SCHEMA;
revision: number;
dataId: string;
layerId: string;
frame: number;
onionSkinning: boolean;
selectedStrokeIndices: number[];
selectedPoints: GreasePencilPointSelectionIR[];
}
export type GreasePencilEditorEditIR =
| { type: "SET_FRAME"; revision: number; frame: number }
| { type: "SET_LAYER"; revision: number; layerId: string }
| { type: "SET_ONION"; revision: number; enabled: boolean }
| { type: "SET_SELECTION"; revision: number; strokeIndices: number[]; points: GreasePencilPointSelectionIR[] }
| { type: "TRANSLATE_POINTS"; revision: number; translation: [number, number, number] };
export interface GreasePencilEditablePointIR {
position: [number, number, number];
radius?: number;
opacity?: number;
vertexColor?: [number, number, number, number];
}
export interface GreasePencilEditableStrokeIR {
cyclic?: boolean;
materialIndex?: number;
points: GreasePencilEditablePointIR[];
}
export interface GreasePencilPointEditResultIR {
editor: GreasePencilEditorIR;
strokes: GreasePencilEditableStrokeIR[];
}
function fail(path: string, message: string): never { throw new Error(`GREASE_PENCIL_EDITOR_INVALID: ${path} ${message}`); }
function record(value: unknown, path: string): Record<string, unknown> {
if (typeof value !== "object" || value === null || Array.isArray(value)) fail(path, "must be an object");
return value as Record<string, unknown>;
}
function id(value: unknown, path: string): string {
if (typeof value !== "string" || value.length === 0 || value.length > GREASE_PENCIL_EDITOR_BUDGET.maxIdLength) fail(path, "is invalid");
return value;
}
function integer(value: unknown, path: string): number {
if (typeof value !== "number" || !Number.isSafeInteger(value) || value < -1_000_000 || value > 1_000_000) fail(path, "is outside the supported frame/index range");
return value;
}
function finite(value: unknown, path: string): number {
if (typeof value !== "number" || !Number.isFinite(value) || Math.abs(value) > 1_000_000) fail(path, "must be finite and bounded");
return value;
}
function tuple(value: unknown, length: number, path: string): number[] {
if (!Array.isArray(value) || value.length !== length) fail(path, `must contain ${length} numbers`);
return value.map((item, index) => finite(item, `${path}[${index}]`));
}
function nonnegativeIndex(value: unknown, path: string): number {
const result = integer(value, path);
if (result < 0) fail(path, "must be non-negative");
return result;
}
function uniqueIndices(value: unknown, path: string): number[] {
if (!Array.isArray(value) || value.length > GREASE_PENCIL_EDITOR_BUDGET.maxSelection) fail(path, "exceeds the selection budget");
const result = value.map((item, itemIndex) => nonnegativeIndex(item, `${path}[${itemIndex}]`));
if (new Set(result).size !== result.length) fail(path, "contains duplicates");
return result.sort((left, right) => left - right);
}
function points(value: unknown, path: string): GreasePencilPointSelectionIR[] {
if (!Array.isArray(value) || value.length > GREASE_PENCIL_EDITOR_BUDGET.maxSelection) fail(path, "exceeds the selection budget");
const seen = new Set<string>();
const result = value.map((item, index) => {
const point = record(item, `${path}[${index}]`);
const parsed = { strokeIndex: nonnegativeIndex(point.strokeIndex, `${path}[${index}].strokeIndex`), pointIndex: nonnegativeIndex(point.pointIndex, `${path}[${index}].pointIndex`) };
const key = `${parsed.strokeIndex}:${parsed.pointIndex}`;
if (seen.has(key)) fail(`${path}[${index}]`, "contains duplicates");
seen.add(key);
return parsed;
});
return result.sort((left, right) => left.strokeIndex - right.strokeIndex || left.pointIndex - right.pointIndex);
}
export function parseGreasePencilEditor(value: unknown): GreasePencilEditorIR {
const editor = record(value, "editor");
if (editor.schemaVersion !== GREASE_PENCIL_EDITOR_SCHEMA) fail("schemaVersion", "is unsupported");
if (typeof editor.revision !== "number" || !Number.isSafeInteger(editor.revision) || editor.revision < 0) fail("revision", "is invalid");
if (typeof editor.frame !== "number" || !Number.isSafeInteger(editor.frame) || editor.frame < -1_000_000 || editor.frame > 1_000_000) fail("frame", "is invalid");
if (typeof editor.onionSkinning !== "boolean") fail("onionSkinning", "must be boolean");
return {
schemaVersion: GREASE_PENCIL_EDITOR_SCHEMA,
revision: editor.revision,
dataId: id(editor.dataId, "dataId"),
layerId: id(editor.layerId, "layerId"),
frame: editor.frame,
onionSkinning: editor.onionSkinning,
selectedStrokeIndices: uniqueIndices(editor.selectedStrokeIndices, "selectedStrokeIndices"),
selectedPoints: points(editor.selectedPoints, "selectedPoints"),
};
}
export function applyGreasePencilEditorEdit(value: unknown, editValue: unknown): GreasePencilEditorIR {
const editor = parseGreasePencilEditor(value);
const edit = record(editValue, "edit");
if (edit.revision !== editor.revision) throw new Error("REVISION_CONFLICT: Grease Pencil editor state is stale");
const revision = editor.revision + 1;
switch (edit.type) {
case "SET_FRAME": return parseGreasePencilEditor({ ...editor, revision, frame: integer(edit.frame, "edit.frame") });
case "SET_LAYER": return parseGreasePencilEditor({ ...editor, revision, layerId: id(edit.layerId, "edit.layerId") });
case "SET_ONION":
if (typeof edit.enabled !== "boolean") fail("edit.enabled", "must be boolean");
return parseGreasePencilEditor({ ...editor, revision, onionSkinning: edit.enabled });
case "SET_SELECTION": return parseGreasePencilEditor({ ...editor, revision, selectedStrokeIndices: uniqueIndices(edit.strokeIndices, "edit.strokeIndices"), selectedPoints: points(edit.points, "edit.points") });
case "TRANSLATE_POINTS":
tuple(edit.translation, 3, "edit.translation");
if (editor.selectedPoints.length === 0) fail("editor.selectedPoints", "must contain at least one point for translation");
return parseGreasePencilEditor({ ...editor, revision });
default: fail("edit.type", "is unsupported");
}
}
function parseEditablePoint(value: unknown, path: string): GreasePencilEditablePointIR {
const point = record(value, path);
const result: GreasePencilEditablePointIR = { position: tuple(point.position, 3, `${path}.position`) as [number, number, number] };
if (point.radius !== undefined) {
result.radius = finite(point.radius, `${path}.radius`);
if (result.radius < 0) fail(`${path}.radius`, "must be non-negative");
}
if (point.opacity !== undefined) {
result.opacity = finite(point.opacity, `${path}.opacity`);
if (result.opacity < 0 || result.opacity > 1) fail(`${path}.opacity`, "must be in [0,1]");
}
if (point.vertexColor !== undefined) {
result.vertexColor = tuple(point.vertexColor, 4, `${path}.vertexColor`) as [number, number, number, number];
if (result.vertexColor.some((component) => component < 0 || component > 1)) fail(`${path}.vertexColor`, "must be in [0,1]");
}
return result;
}
function parseEditableStrokes(value: unknown): GreasePencilEditableStrokeIR[] {
if (!Array.isArray(value) || value.length > GREASE_PENCIL_EDITOR_BUDGET.maxSelection) fail("strokes", "exceeds the stroke budget");
let pointCount = 0;
return value.map((item, strokeIndex) => {
const stroke = record(item, `strokes[${strokeIndex}]`);
if (!Array.isArray(stroke.points) || stroke.points.length === 0) fail(`strokes[${strokeIndex}].points`, "must be a non-empty array");
pointCount += stroke.points.length;
if (pointCount > GREASE_PENCIL_EDITOR_BUDGET.maxSelection) fail("strokes", "exceeds the point budget");
if (stroke.cyclic !== undefined && typeof stroke.cyclic !== "boolean") fail(`strokes[${strokeIndex}].cyclic`, "must be boolean");
if (stroke.materialIndex !== undefined) nonnegativeIndex(stroke.materialIndex, `strokes[${strokeIndex}].materialIndex`);
return {
...(stroke.cyclic !== undefined ? { cyclic: stroke.cyclic } : {}),
...(stroke.materialIndex !== undefined ? { materialIndex: stroke.materialIndex as number } : {}),
points: stroke.points.map((point, pointIndex) => parseEditablePoint(point, `strokes[${strokeIndex}].points[${pointIndex}]`)),
};
});
}
export function applyGreasePencilPointTranslation(
editorValue: unknown,
strokesValue: unknown,
editValue: unknown,
): GreasePencilPointEditResultIR {
const editor = parseGreasePencilEditor(editorValue);
const strokes = parseEditableStrokes(strokesValue);
const edit = record(editValue, "edit");
if (edit.type !== "TRANSLATE_POINTS") fail("edit.type", "must be TRANSLATE_POINTS");
if (edit.revision !== editor.revision) throw new Error("REVISION_CONFLICT: Grease Pencil editor state is stale");
const translation = tuple(edit.translation, 3, "edit.translation") as [number, number, number];
if (editor.selectedPoints.length === 0) fail("editor.selectedPoints", "must contain at least one point for translation");
const selected = new Set(editor.selectedPoints.map((point) => `${point.strokeIndex}:${point.pointIndex}`));
for (const point of editor.selectedPoints) {
if (!strokes[point.strokeIndex]?.points[point.pointIndex]) fail("editor.selectedPoints", `references missing point ${point.strokeIndex}:${point.pointIndex}`);
}
const translated = strokes.map((stroke, strokeIndex) => ({
...stroke,
points: stroke.points.map((point, pointIndex) => selected.has(`${strokeIndex}:${pointIndex}`) ? {
...point,
position: point.position.map((component, axis) => finite(component + translation[axis], `translated[${strokeIndex}][${pointIndex}][${axis}]`)) as [number, number, number],
} : { ...point, position: [...point.position] as [number, number, number] }),
}));
return {
editor: parseGreasePencilEditor({ ...editor, revision: editor.revision + 1 }),
strokes: translated,
};
}