>(new Set())
+
const outputUrl = currentJob?.outputUrl ?? ''
const modelUrl =
currentJob?.status === 'done' && currentJob.outputUrl
? `${apiUrl}${currentJob.outputUrl}`
: null
+ const autoCenter = !appliedUrls.current.has(outputUrl)
+
// A .ply/.splat reaching the viewer is always a Gaussian splat here: mesh
// plys are converted to GLB on import and workflow mesh outputs are .glb.
const isSplat = /\.(ply|splat)$/i.test(outputUrl)
@@ -446,6 +480,50 @@ export default function Viewer3D({ lightSettings = DEFAULT_LIGHT_SETTINGS }: { l
link.click()
}
+ const handleResetTransform = () => {
+ const original = currentJob?.originalOutputUrl
+ if (original && currentJob?.outputUrl !== original) {
+ // Was baked via Apply — reload the original (auto-centred on load).
+ updateCurrentJob({ outputUrl: original })
+ pushMeshUrl(original)
+ } else {
+ // Live transform only — re-centre the current scene in place.
+ setResetToken((t) => t + 1)
+ }
+ }
+
+ const handleApplyTransform = async () => {
+ if (!meshObject || !currentJob?.outputUrl) return
+ const url = currentJob.outputUrl
+ if (!url.startsWith('/workspace/')) return
+ const path = url.slice('/workspace/'.length)
+
+ // We bake the full world matrix, which includes the centering offset the
+ // viewer applies on load ("bake what you see"). The result URL is then
+ // flagged so it is NOT re-centered on reload, keeping the visible pose.
+ meshObject.updateWorldMatrix(true, false)
+ const e = meshObject.matrixWorld.elements
+ // THREE stores column-major; emit a row-major 4x4 for the backend.
+ const matrix = [
+ [e[0], e[4], e[8], e[12]],
+ [e[1], e[5], e[9], e[13]],
+ [e[2], e[6], e[10], e[14]],
+ [e[3], e[7], e[11], e[15]],
+ ]
+
+ setApplying(true)
+ try {
+ const result = await transformMesh(path, matrix)
+ appliedUrls.current.add(result.url)
+ updateCurrentJob({ outputUrl: result.url })
+ pushMeshUrl(result.url)
+ } catch (err) {
+ showError(err instanceof Error ? err.message : String(err))
+ } finally {
+ setApplying(false)
+ }
+ }
+
return (
}>
@@ -504,13 +582,20 @@ export default function Viewer3D({ lightSettings = DEFAULT_LIGHT_SETTINGS }: { l
jobId={currentJob.id}
viewMode={viewMode}
selected={selected}
+ autoCenter={autoCenter}
+ resetToken={resetToken}
onStats={setStoreMeshStats}
onSelect={() => setSelected(true)}
+ onObject={setMeshObject}
/>
) : null}
+ {selected && gizmoMode && meshObject && (
+
+ )}
+
)}
+ {/* Transform apply/reset — visible while a gizmo tool is active */}
+ {!isSplat && modelUrl && selected && gizmoMode && (
+
+
+
+
@@ -565,4 +677,4 @@ export default function Viewer3D({ lightSettings = DEFAULT_LIGHT_SETTINGS }: { l
)
-}
+}
\ No newline at end of file
diff --git a/src/shared/hooks/useApi.ts b/src/shared/hooks/useApi.ts
index 81f0c42..c98ac65 100644
--- a/src/shared/hooks/useApi.ts
+++ b/src/shared/hooks/useApi.ts
@@ -116,5 +116,18 @@ export function useApi() {
return { url: data.url }
}
- return { generateFromImage, pollJobStatus, cancelJob, getModelStatus, downloadModel, optimizeMesh, smoothMesh, importMesh }
-}
+ // Bakes a world-space 4x4 transform into the GLB geometry.
+ // `matrix` is row-major (4 rows of 4), matching the backend's reshape.
+ async function transformMesh(
+ path: string,
+ matrix: number[][],
+ ): Promise<{ url: string }> {
+ const { data } = await client.post<{ url: string }>('/optimize/transform', {
+ path,
+ matrix,
+ })
+ return { url: data.url }
+ }
+
+ return { generateFromImage, pollJobStatus, cancelJob, getModelStatus, downloadModel, optimizeMesh, smoothMesh, importMesh, transformMesh }
+}
\ No newline at end of file