Alexandru Popovici
02b97bcb86
* chore(preview-frontend): Disabled outlines to avoid the extra complexity and weird artifacts * chore(preview-frontend): Pinned the viewer version * chore(preview-frontend): Added lockfile * chore(preview-frontend): Bumped viewer version * chore(preview-frontend): lockfile
95 lines
2.5 KiB
TypeScript
95 lines
2.5 KiB
TypeScript
import { TIME_MS } from '@speckle/shared'
|
|
import {
|
|
Load,
|
|
LoadArgs,
|
|
PreviewGenerator,
|
|
PreviewPageResult,
|
|
TakeScreenshot
|
|
} from '@speckle/shared/workers/previews'
|
|
import {
|
|
Viewer,
|
|
DefaultViewerParams,
|
|
SpeckleLoader,
|
|
UrlHelper,
|
|
UpdateFlags,
|
|
DefaultPipeline
|
|
} from '@speckle/viewer'
|
|
import { CameraController } from '@speckle/viewer'
|
|
|
|
declare global {
|
|
// eslint-disable-next-line @typescript-eslint/no-empty-object-type
|
|
interface Window extends PreviewGenerator {}
|
|
}
|
|
|
|
let viewer: Viewer | undefined = undefined
|
|
|
|
const init = async (): Promise<Viewer> => {
|
|
/** Get the HTML container */
|
|
const container = document.getElementById('renderer') as HTMLElement
|
|
|
|
/** Configure the viewer params */
|
|
const params = DefaultViewerParams
|
|
params.showStats = false
|
|
params.verbose = false
|
|
|
|
/** Create Viewer instance */
|
|
const viewer = new Viewer(container, params)
|
|
/** Initialise the viewer */
|
|
await viewer.init()
|
|
|
|
/** Add the stock camera controller extension */
|
|
viewer.createExtension(CameraController)
|
|
return viewer
|
|
}
|
|
|
|
const load: Load = async ({ url, token }: LoadArgs) => {
|
|
if (!viewer) {
|
|
viewer = await init()
|
|
viewer.resize()
|
|
}
|
|
/** Create a loader for the speckle stream */
|
|
const resourceUrls = await UrlHelper.getResourceUrls(url, token)
|
|
for (const resourceUrl of resourceUrls) {
|
|
const loader = new SpeckleLoader(viewer.getWorldTree(), resourceUrl, token)
|
|
/** Load the speckle data */
|
|
await viewer.loadObject(loader, true)
|
|
}
|
|
}
|
|
|
|
window.load = load
|
|
|
|
// TODO: replace with sleep from speckle/shared
|
|
const waitForAnimation = async (ms = 70) =>
|
|
await new Promise((resolve) => {
|
|
setTimeout(resolve, ms)
|
|
})
|
|
|
|
const takeScreenshot: TakeScreenshot = async () => {
|
|
if (!viewer) viewer = await init()
|
|
const ret: PreviewPageResult = {
|
|
durationSeconds: 0,
|
|
screenshots: {}
|
|
}
|
|
|
|
const t0 = Date.now()
|
|
|
|
viewer.resize()
|
|
const cameraController = viewer.getExtension(CameraController)
|
|
cameraController.setCameraView([], false)
|
|
viewer.getRenderer().pipeline = new DefaultPipeline(viewer.getRenderer(), {
|
|
edges: false
|
|
})
|
|
await waitForAnimation(100)
|
|
|
|
for (let i = 0; i < 24; i++) {
|
|
cameraController.setCameraView({ azimuth: Math.PI / 12, polar: 0 }, false)
|
|
viewer.requestRender(UpdateFlags.RENDER_RESET)
|
|
await waitForAnimation(10)
|
|
ret.screenshots[i + ''] = await viewer.screenshot()
|
|
console.log(`Screenshot taken at ${i}`)
|
|
}
|
|
ret.durationSeconds = (Date.now() - t0) / TIME_MS.second
|
|
return ret
|
|
}
|
|
window.takeScreenshot = takeScreenshot
|