canvas3d.ts 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801
  1. /**
  2. * Copyright (c) 2018-2021 mol* contributors, licensed under MIT, See LICENSE file for more info.
  3. *
  4. * @author Alexander Rose <alexander.rose@weirdbyte.de>
  5. * @author David Sehnal <david.sehnal@gmail.com>
  6. */
  7. import { BehaviorSubject, Subscription } from 'rxjs';
  8. import { now } from '../mol-util/now';
  9. import { Vec3, Vec2 } from '../mol-math/linear-algebra';
  10. import { InputObserver, ModifiersKeys, ButtonsType } from '../mol-util/input/input-observer';
  11. import { Renderer, RendererStats, RendererParams } from '../mol-gl/renderer';
  12. import { GraphicsRenderObject } from '../mol-gl/render-object';
  13. import { TrackballControls, TrackballControlsParams } from './controls/trackball';
  14. import { Viewport } from './camera/util';
  15. import { createContext, WebGLContext, getGLContext } from '../mol-gl/webgl/context';
  16. import { Representation } from '../mol-repr/representation';
  17. import { Scene } from '../mol-gl/scene';
  18. import { PickingId } from '../mol-geo/geometry/picking';
  19. import { MarkerAction } from '../mol-util/marker-action';
  20. import { Loci, EmptyLoci, isEmptyLoci } from '../mol-model/loci';
  21. import { Camera } from './camera';
  22. import { ParamDefinition as PD } from '../mol-util/param-definition';
  23. import { DebugHelperParams } from './helper/bounding-sphere-helper';
  24. import { SetUtils } from '../mol-util/set';
  25. import { Canvas3dInteractionHelper } from './helper/interaction-events';
  26. import { PostprocessingParams } from './passes/postprocessing';
  27. import { MultiSampleHelper, MultiSampleParams, MultiSamplePass } from './passes/multi-sample';
  28. import { PickData } from './passes/pick';
  29. import { PickHelper } from './passes/pick';
  30. import { ImagePass, ImageProps } from './passes/image';
  31. import { Sphere3D } from '../mol-math/geometry';
  32. import { isDebugMode } from '../mol-util/debug';
  33. import { CameraHelperParams } from './helper/camera-helper';
  34. import { produce } from 'immer';
  35. import { HandleHelperParams } from './helper/handle-helper';
  36. import { StereoCamera, StereoCameraParams } from './camera/stereo';
  37. import { Helper } from './helper/helper';
  38. import { Passes } from './passes/passes';
  39. import { shallowEqual } from '../mol-util';
  40. export const Canvas3DParams = {
  41. camera: PD.Group({
  42. mode: PD.Select('perspective', PD.arrayToOptions(['perspective', 'orthographic'] as const), { label: 'Camera' }),
  43. helper: PD.Group(CameraHelperParams, { isFlat: true }),
  44. stereo: PD.MappedStatic('off', {
  45. on: PD.Group(StereoCameraParams),
  46. off: PD.Group({})
  47. }, { cycle: true, hideIf: p => p?.mode !== 'perspective' }),
  48. manualReset: PD.Boolean(false, { isHidden: true }),
  49. }, { pivot: 'mode' }),
  50. cameraFog: PD.MappedStatic('on', {
  51. on: PD.Group({
  52. intensity: PD.Numeric(15, { min: 1, max: 100, step: 1 }),
  53. }),
  54. off: PD.Group({})
  55. }, { cycle: true, description: 'Show fog in the distance' }),
  56. cameraClipping: PD.Group({
  57. radius: PD.Numeric(100, { min: 0, max: 99, step: 1 }, { label: 'Clipping', description: 'How much of the scene to show.' }),
  58. far: PD.Boolean(true, { description: 'Hide scene in the distance' }),
  59. }, { pivot: 'radius' }),
  60. viewport: PD.MappedStatic('canvas', {
  61. canvas: PD.Group({}),
  62. custom: PD.Group({
  63. x: PD.Numeric(0),
  64. y: PD.Numeric(0),
  65. width: PD.Numeric(128),
  66. height: PD.Numeric(128)
  67. })
  68. }),
  69. cameraResetDurationMs: PD.Numeric(250, { min: 0, max: 1000, step: 1 }, { description: 'The time it takes to reset the camera.' }),
  70. transparentBackground: PD.Boolean(false),
  71. multiSample: PD.Group(MultiSampleParams),
  72. postprocessing: PD.Group(PostprocessingParams),
  73. renderer: PD.Group(RendererParams),
  74. trackball: PD.Group(TrackballControlsParams),
  75. debug: PD.Group(DebugHelperParams),
  76. handle: PD.Group(HandleHelperParams),
  77. };
  78. export const DefaultCanvas3DParams = PD.getDefaultValues(Canvas3DParams);
  79. export type Canvas3DProps = PD.Values<typeof Canvas3DParams>
  80. export type PartialCanvas3DProps = {
  81. [K in keyof Canvas3DProps]?: Canvas3DProps[K] extends { name: string, params: any } ? Canvas3DProps[K] : Partial<Canvas3DProps[K]>
  82. }
  83. export { Canvas3DContext };
  84. /** Can be used to create multiple Canvas3D objects */
  85. interface Canvas3DContext {
  86. readonly canvas: HTMLCanvasElement
  87. readonly webgl: WebGLContext
  88. readonly input: InputObserver
  89. readonly passes: Passes
  90. readonly attribs: Readonly<Canvas3DContext.Attribs>
  91. readonly contextLost: BehaviorSubject<now.Timestamp>
  92. readonly contextRestored: BehaviorSubject<now.Timestamp>
  93. dispose: (options?: Partial<{ doNotForceWebGLContextLoss: boolean }>) => void
  94. }
  95. namespace Canvas3DContext {
  96. const DefaultAttribs = {
  97. /** true by default to avoid issues with Safari (Jan 2021) */
  98. antialias: true,
  99. /** true to support multiple Canvas3D objects with a single context */
  100. preserveDrawingBuffer: true,
  101. pixelScale: 1,
  102. pickScale: 0.25,
  103. enableWboit: true
  104. };
  105. export type Attribs = typeof DefaultAttribs
  106. export function fromCanvas(canvas: HTMLCanvasElement, attribs: Partial<Attribs> = {}): Canvas3DContext {
  107. const a = { ...DefaultAttribs, ...attribs };
  108. const { antialias, preserveDrawingBuffer, pixelScale } = a;
  109. const gl = getGLContext(canvas, {
  110. antialias,
  111. preserveDrawingBuffer,
  112. alpha: true, // the renderer requires an alpha channel
  113. depth: true, // the renderer requires a depth buffer
  114. premultipliedAlpha: true, // the renderer outputs PMA
  115. });
  116. if (gl === null) throw new Error('Could not create a WebGL rendering context');
  117. const input = InputObserver.fromElement(canvas, { pixelScale });
  118. const webgl = createContext(gl, { pixelScale });
  119. const passes = new Passes(webgl, attribs);
  120. if (isDebugMode) {
  121. const loseContextExt = gl.getExtension('WEBGL_lose_context');
  122. if (loseContextExt) {
  123. // Hold down shift+ctrl+alt and press any mouse button to call `loseContext`.
  124. // After 1 second `restoreContext` will be called.
  125. canvas.addEventListener('mousedown', e => {
  126. if (webgl.isContextLost) return;
  127. if (!e.shiftKey || !e.ctrlKey || !e.altKey) return;
  128. if (isDebugMode) console.log('lose context');
  129. loseContextExt.loseContext();
  130. setTimeout(() => {
  131. if (!webgl.isContextLost) return;
  132. if (isDebugMode) console.log('restore context');
  133. loseContextExt.restoreContext();
  134. }, 1000);
  135. }, false);
  136. }
  137. }
  138. // https://www.khronos.org/webgl/wiki/HandlingContextLost
  139. const contextLost = new BehaviorSubject<now.Timestamp>(0 as now.Timestamp);
  140. const handleWebglContextLost = (e: Event) => {
  141. webgl.setContextLost();
  142. e.preventDefault();
  143. if (isDebugMode) console.log('context lost');
  144. contextLost.next(now());
  145. };
  146. const handlewWebglContextRestored = () => {
  147. if (!webgl.isContextLost) return;
  148. webgl.handleContextRestored(() => {
  149. passes.draw.reset();
  150. });
  151. if (isDebugMode) console.log('context restored');
  152. };
  153. canvas.addEventListener('webglcontextlost', handleWebglContextLost, false);
  154. canvas.addEventListener('webglcontextrestored', handlewWebglContextRestored, false);
  155. return {
  156. canvas,
  157. webgl,
  158. input,
  159. passes,
  160. attribs: a,
  161. contextLost,
  162. contextRestored: webgl.contextRestored,
  163. dispose: (options?: Partial<{ doNotForceWebGLContextLoss: boolean }>) => {
  164. input.dispose();
  165. canvas.removeEventListener('webglcontextlost', handleWebglContextLost, false);
  166. canvas.removeEventListener('webglcontextrestored', handlewWebglContextRestored, false);
  167. webgl.destroy(options);
  168. }
  169. };
  170. }
  171. }
  172. export { Canvas3D };
  173. interface Canvas3D {
  174. readonly webgl: WebGLContext,
  175. add(repr: Representation.Any): void
  176. remove(repr: Representation.Any): void
  177. /**
  178. * This function must be called if animate() is not set up so that add/remove actions take place.
  179. */
  180. commit(isSynchronous?: boolean): void
  181. /**
  182. * Funcion for external "animation" control
  183. * Calls commit.
  184. */
  185. tick(t: now.Timestamp, options?: { isSynchronous?: boolean, manualDraw?: boolean }): void
  186. update(repr?: Representation.Any, keepBoundingSphere?: boolean): void
  187. clear(): void
  188. syncVisibility(): void
  189. requestDraw(force?: boolean): void
  190. /** Reset the timers, used by "animate" */
  191. resetTime(t: number): void
  192. animate(): void
  193. pause(): void
  194. identify(x: number, y: number): PickData | undefined
  195. mark(loci: Representation.Loci, action: MarkerAction): void
  196. getLoci(pickingId: PickingId | undefined): Representation.Loci
  197. notifyDidDraw: boolean,
  198. readonly didDraw: BehaviorSubject<now.Timestamp>
  199. readonly commited: BehaviorSubject<now.Timestamp>
  200. readonly reprCount: BehaviorSubject<number>
  201. readonly resized: BehaviorSubject<any>
  202. handleResize(): void
  203. /** performs handleResize on the next animation frame */
  204. requestResize(): void
  205. /** Focuses camera on scene's bounding sphere, centered and zoomed. */
  206. requestCameraReset(options?: { durationMs?: number, snapshot?: Partial<Camera.Snapshot> }): void
  207. readonly camera: Camera
  208. readonly boundingSphere: Readonly<Sphere3D>
  209. setProps(props: PartialCanvas3DProps | ((old: Canvas3DProps) => Partial<Canvas3DProps> | void), doNotRequestDraw?: boolean /* = false */): void
  210. getImagePass(props: Partial<ImageProps>): ImagePass
  211. /** Returns a copy of the current Canvas3D instance props */
  212. readonly props: Readonly<Canvas3DProps>
  213. readonly input: InputObserver
  214. readonly stats: RendererStats
  215. readonly interaction: Canvas3dInteractionHelper['events']
  216. dispose(): void
  217. }
  218. const requestAnimationFrame = typeof window !== 'undefined'
  219. ? window.requestAnimationFrame
  220. : (f: (time: number) => void) => setImmediate(() => f(Date.now())) as unknown as number;
  221. const cancelAnimationFrame = typeof window !== 'undefined'
  222. ? window.cancelAnimationFrame
  223. : (handle: number) => clearImmediate(handle as unknown as NodeJS.Immediate);
  224. namespace Canvas3D {
  225. export interface HoverEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, page?: Vec2, position?: Vec3 }
  226. export interface DragEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, pageStart: Vec2, pageEnd: Vec2 }
  227. export interface ClickEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, page?: Vec2, position?: Vec3 }
  228. export function create({ webgl, input, passes, attribs }: Canvas3DContext, props: Partial<Canvas3DProps> = {}): Canvas3D {
  229. const p: Canvas3DProps = { ...DefaultCanvas3DParams, ...props };
  230. const reprRenderObjects = new Map<Representation.Any, Set<GraphicsRenderObject>>();
  231. const reprUpdatedSubscriptions = new Map<Representation.Any, Subscription>();
  232. const reprCount = new BehaviorSubject(0);
  233. let startTime = now();
  234. const didDraw = new BehaviorSubject<now.Timestamp>(0 as now.Timestamp);
  235. const commited = new BehaviorSubject<now.Timestamp>(0 as now.Timestamp);
  236. const { gl, contextRestored } = webgl;
  237. let x = 0;
  238. let y = 0;
  239. let width = 128;
  240. let height = 128;
  241. updateViewport();
  242. const scene = Scene.create(webgl);
  243. const camera = new Camera({
  244. position: Vec3.create(0, 0, 100),
  245. mode: p.camera.mode,
  246. fog: p.cameraFog.name === 'on' ? p.cameraFog.params.intensity : 0,
  247. clipFar: p.cameraClipping.far
  248. }, { x, y, width, height }, { pixelScale: attribs.pixelScale });
  249. const stereoCamera = new StereoCamera(camera, p.camera.stereo.params);
  250. const controls = TrackballControls.create(input, camera, p.trackball);
  251. const renderer = Renderer.create(webgl, p.renderer);
  252. const helper = new Helper(webgl, scene, p);
  253. const pickHelper = new PickHelper(webgl, renderer, scene, helper, passes.pick, { x, y, width, height });
  254. const interactionHelper = new Canvas3dInteractionHelper(identify, getLoci, input, camera);
  255. const multiSampleHelper = new MultiSampleHelper(passes.multiSample);
  256. let drawPending = false;
  257. let cameraResetRequested = false;
  258. let nextCameraResetDuration: number | undefined = void 0;
  259. let nextCameraResetSnapshot: Partial<Camera.Snapshot> | undefined = void 0;
  260. let resizeRequested = false;
  261. let notifyDidDraw = true;
  262. function getLoci(pickingId: PickingId | undefined) {
  263. let loci: Loci = EmptyLoci;
  264. let repr: Representation.Any = Representation.Empty;
  265. if (pickingId) {
  266. loci = helper.handle.getLoci(pickingId);
  267. reprRenderObjects.forEach((_, _repr) => {
  268. const _loci = _repr.getLoci(pickingId);
  269. if (!isEmptyLoci(_loci)) {
  270. if (!isEmptyLoci(loci)) {
  271. console.warn('found another loci, this should not happen');
  272. }
  273. loci = _loci;
  274. repr = _repr;
  275. }
  276. });
  277. }
  278. return { loci, repr };
  279. }
  280. function mark(reprLoci: Representation.Loci, action: MarkerAction) {
  281. const { repr, loci } = reprLoci;
  282. let changed = false;
  283. if (repr) {
  284. changed = repr.mark(loci, action);
  285. } else {
  286. changed = helper.handle.mark(loci, action);
  287. reprRenderObjects.forEach((_, _repr) => { changed = _repr.mark(loci, action) || changed; });
  288. }
  289. if (changed) {
  290. scene.update(void 0, true);
  291. helper.handle.scene.update(void 0, true);
  292. const prevPickDirty = pickHelper.dirty;
  293. draw(true);
  294. pickHelper.dirty = prevPickDirty; // marking does not change picking buffers
  295. }
  296. }
  297. function render(force: boolean) {
  298. if (webgl.isContextLost) return false;
  299. let resized = false;
  300. if (resizeRequested) {
  301. handleResize(false);
  302. resizeRequested = false;
  303. resized = true;
  304. }
  305. if (x > gl.drawingBufferWidth || x + width < 0 ||
  306. y > gl.drawingBufferHeight || y + height < 0
  307. ) return false;
  308. let didRender = false;
  309. controls.update(currentTime);
  310. const cameraChanged = camera.update();
  311. const multiSampleChanged = multiSampleHelper.update(force || cameraChanged, p.multiSample);
  312. if (resized || force || cameraChanged || multiSampleChanged) {
  313. let cam: Camera | StereoCamera = camera;
  314. if (p.camera.stereo.name === 'on') {
  315. stereoCamera.update();
  316. cam = stereoCamera;
  317. }
  318. if (MultiSamplePass.isEnabled(p.multiSample)) {
  319. multiSampleHelper.render(renderer, cam, scene, helper, true, p.transparentBackground, p);
  320. } else {
  321. passes.draw.render(renderer, cam, scene, helper, true, p.transparentBackground, p.postprocessing);
  322. }
  323. pickHelper.dirty = true;
  324. didRender = true;
  325. }
  326. return didRender;
  327. }
  328. let forceNextDraw = false;
  329. let forceDrawAfterAllCommited = false;
  330. let currentTime = 0;
  331. function draw(force?: boolean) {
  332. if (render(!!force || forceNextDraw) && notifyDidDraw) {
  333. didDraw.next(now() - startTime as now.Timestamp);
  334. }
  335. forceNextDraw = false;
  336. drawPending = false;
  337. }
  338. function requestDraw(force?: boolean) {
  339. if (drawPending) return;
  340. drawPending = true;
  341. forceNextDraw = !!force;
  342. }
  343. let animationFrameHandle = 0;
  344. function tick(t: now.Timestamp, options?: { isSynchronous?: boolean, manualDraw?: boolean }) {
  345. currentTime = t;
  346. commit(options?.isSynchronous);
  347. camera.transition.tick(currentTime);
  348. if (options?.manualDraw) {
  349. return;
  350. }
  351. draw(false);
  352. if (!camera.transition.inTransition && !webgl.isContextLost) {
  353. interactionHelper.tick(currentTime);
  354. }
  355. }
  356. function _animate() {
  357. tick(now());
  358. animationFrameHandle = requestAnimationFrame(_animate);
  359. }
  360. function resetTime(t: now.Timestamp) {
  361. startTime = t;
  362. controls.start(t);
  363. }
  364. function animate() {
  365. controls.start(now());
  366. if (animationFrameHandle === 0) _animate();
  367. }
  368. function pause() {
  369. cancelAnimationFrame(animationFrameHandle);
  370. animationFrameHandle = 0;
  371. }
  372. function identify(x: number, y: number): PickData | undefined {
  373. const cam = p.camera.stereo.name === 'on' ? stereoCamera : camera;
  374. return webgl.isContextLost ? undefined : pickHelper.identify(x, y, cam);
  375. }
  376. function commit(isSynchronous: boolean = false) {
  377. const allCommited = commitScene(isSynchronous);
  378. // Only reset the camera after the full scene has been commited.
  379. if (allCommited) {
  380. resolveCameraReset();
  381. if (forceDrawAfterAllCommited) {
  382. if (helper.debug.isEnabled) helper.debug.update();
  383. draw(true);
  384. forceDrawAfterAllCommited = false;
  385. }
  386. commited.next(now());
  387. }
  388. }
  389. function resolveCameraReset() {
  390. if (!cameraResetRequested) return;
  391. const { center, radius } = scene.boundingSphereVisible;
  392. if (radius > 0) {
  393. const duration = nextCameraResetDuration === undefined ? p.cameraResetDurationMs : nextCameraResetDuration;
  394. const focus = camera.getFocus(center, radius);
  395. const snapshot = nextCameraResetSnapshot ? { ...focus, ...nextCameraResetSnapshot } : focus;
  396. camera.setState({ ...snapshot, radiusMax: scene.boundingSphere.radius }, duration);
  397. }
  398. nextCameraResetDuration = void 0;
  399. nextCameraResetSnapshot = void 0;
  400. cameraResetRequested = false;
  401. }
  402. const oldBoundingSphereVisible = Sphere3D();
  403. const cameraSphere = Sphere3D();
  404. function shouldResetCamera() {
  405. if (camera.state.radiusMax === 0) return true;
  406. if (camera.transition.inTransition || nextCameraResetSnapshot) return false;
  407. let cameraSphereOverlapsNone = true;
  408. Sphere3D.set(cameraSphere, camera.state.target, camera.state.radius);
  409. // check if any renderable has moved outside of the old bounding sphere
  410. // and if no renderable is overlapping with the camera sphere
  411. for (const r of scene.renderables) {
  412. if (!r.state.visible) continue;
  413. const b = r.values.boundingSphere.ref.value;
  414. if (!b.radius) continue;
  415. const cameraDist = Vec3.distance(cameraSphere.center, b.center);
  416. if ((cameraDist > cameraSphere.radius || cameraDist > b.radius || b.radius > camera.state.radiusMax) && !Sphere3D.includes(oldBoundingSphereVisible, b)) return true;
  417. if (Sphere3D.overlaps(cameraSphere, b)) cameraSphereOverlapsNone = false;
  418. }
  419. return cameraSphereOverlapsNone;
  420. }
  421. const sceneCommitTimeoutMs = 250;
  422. function commitScene(isSynchronous: boolean) {
  423. if (!scene.needsCommit) return true;
  424. // snapshot the current bounding sphere of visible objects
  425. Sphere3D.copy(oldBoundingSphereVisible, scene.boundingSphereVisible);
  426. if (!scene.commit(isSynchronous ? void 0 : sceneCommitTimeoutMs)) return false;
  427. if (helper.debug.isEnabled) helper.debug.update();
  428. if (!p.camera.manualReset && (reprCount.value === 0 || shouldResetCamera())) {
  429. cameraResetRequested = true;
  430. }
  431. if (oldBoundingSphereVisible.radius === 0) nextCameraResetDuration = 0;
  432. camera.setState({ radiusMax: scene.boundingSphere.radius }, 0);
  433. reprCount.next(reprRenderObjects.size);
  434. if (isDebugMode) consoleStats();
  435. return true;
  436. }
  437. function consoleStats() {
  438. console.table(scene.renderables.map(r => ({
  439. drawCount: r.values.drawCount.ref.value,
  440. instanceCount: r.values.instanceCount.ref.value,
  441. materialId: r.materialId,
  442. renderItemId: r.id,
  443. })));
  444. console.log(webgl.stats);
  445. const { texture, attribute, elements } = webgl.resources.getByteCounts();
  446. console.log({
  447. texture: `${(texture / 1024 / 1024).toFixed(3)} MiB`,
  448. attribute: `${(attribute / 1024 / 1024).toFixed(3)} MiB`,
  449. elements: `${(elements / 1024 / 1024).toFixed(3)} MiB`,
  450. });
  451. }
  452. function add(repr: Representation.Any) {
  453. registerAutoUpdate(repr);
  454. const oldRO = reprRenderObjects.get(repr);
  455. const newRO = new Set<GraphicsRenderObject>();
  456. repr.renderObjects.forEach(o => newRO.add(o));
  457. if (oldRO) {
  458. if (!SetUtils.areEqual(newRO, oldRO)) {
  459. newRO.forEach(o => { if (!oldRO.has(o)) scene.add(o); });
  460. oldRO.forEach(o => { if (!newRO.has(o)) scene.remove(o); });
  461. }
  462. } else {
  463. repr.renderObjects.forEach(o => scene.add(o));
  464. }
  465. reprRenderObjects.set(repr, newRO);
  466. scene.update(repr.renderObjects, false);
  467. forceDrawAfterAllCommited = true;
  468. if (isDebugMode) consoleStats();
  469. }
  470. function remove(repr: Representation.Any) {
  471. unregisterAutoUpdate(repr);
  472. const renderObjects = reprRenderObjects.get(repr);
  473. if (renderObjects) {
  474. renderObjects.forEach(o => scene.remove(o));
  475. reprRenderObjects.delete(repr);
  476. forceDrawAfterAllCommited = true;
  477. if (isDebugMode) consoleStats();
  478. }
  479. }
  480. function registerAutoUpdate(repr: Representation.Any) {
  481. if (reprUpdatedSubscriptions.has(repr)) return;
  482. reprUpdatedSubscriptions.set(repr, repr.updated.subscribe(_ => {
  483. if (!repr.state.syncManually) add(repr);
  484. }));
  485. }
  486. function unregisterAutoUpdate(repr: Representation.Any) {
  487. const updatedSubscription = reprUpdatedSubscriptions.get(repr);
  488. if (updatedSubscription) {
  489. updatedSubscription.unsubscribe();
  490. reprUpdatedSubscriptions.delete(repr);
  491. }
  492. }
  493. function getProps(): Canvas3DProps {
  494. const radius = scene.boundingSphere.radius > 0
  495. ? 100 - Math.round((camera.transition.target.radius / scene.boundingSphere.radius) * 100)
  496. : 0;
  497. return {
  498. camera: {
  499. mode: camera.state.mode,
  500. helper: { ...helper.camera.props },
  501. stereo: { ...p.camera.stereo },
  502. manualReset: !!p.camera.manualReset
  503. },
  504. cameraFog: camera.state.fog > 0
  505. ? { name: 'on' as const, params: { intensity: camera.state.fog } }
  506. : { name: 'off' as const, params: {} },
  507. cameraClipping: { far: camera.state.clipFar, radius },
  508. cameraResetDurationMs: p.cameraResetDurationMs,
  509. transparentBackground: p.transparentBackground,
  510. viewport: p.viewport,
  511. postprocessing: { ...p.postprocessing },
  512. multiSample: { ...p.multiSample },
  513. renderer: { ...renderer.props },
  514. trackball: { ...controls.props },
  515. debug: { ...helper.debug.props },
  516. handle: { ...helper.handle.props },
  517. };
  518. }
  519. const contextRestoredSub = contextRestored.subscribe(() => {
  520. pickHelper.dirty = true;
  521. draw(true);
  522. // Unclear why, but in Chrome with wboit enabled the first `draw` only clears
  523. // the drawingBuffer. Note that in Firefox the drawingBuffer is preserved after
  524. // context loss so it is unclear if it behaves the same.
  525. draw(true);
  526. });
  527. const resized = new BehaviorSubject<any>(0);
  528. function handleResize(draw = true) {
  529. passes.updateSize();
  530. updateViewport();
  531. syncViewport();
  532. if (draw) requestDraw(true);
  533. resized.next(+new Date());
  534. }
  535. return {
  536. webgl,
  537. add,
  538. remove,
  539. commit,
  540. update: (repr, keepSphere) => {
  541. if (repr) {
  542. if (!reprRenderObjects.has(repr)) return;
  543. scene.update(repr.renderObjects, !!keepSphere);
  544. } else {
  545. scene.update(void 0, !!keepSphere);
  546. }
  547. forceDrawAfterAllCommited = true;
  548. },
  549. clear: () => {
  550. reprUpdatedSubscriptions.forEach(v => v.unsubscribe());
  551. reprUpdatedSubscriptions.clear();
  552. reprRenderObjects.clear();
  553. scene.clear();
  554. helper.debug.clear();
  555. requestDraw(true);
  556. reprCount.next(reprRenderObjects.size);
  557. },
  558. syncVisibility: () => {
  559. if (camera.state.radiusMax === 0) {
  560. cameraResetRequested = true;
  561. nextCameraResetDuration = 0;
  562. }
  563. if (scene.syncVisibility()) {
  564. if (helper.debug.isEnabled) helper.debug.update();
  565. }
  566. requestDraw(true);
  567. },
  568. requestDraw,
  569. tick,
  570. animate,
  571. resetTime,
  572. pause,
  573. identify,
  574. mark,
  575. getLoci,
  576. handleResize,
  577. requestResize: () => {
  578. resizeRequested = true;
  579. },
  580. requestCameraReset: options => {
  581. nextCameraResetDuration = options?.durationMs;
  582. nextCameraResetSnapshot = options?.snapshot;
  583. cameraResetRequested = true;
  584. },
  585. camera,
  586. boundingSphere: scene.boundingSphere,
  587. get notifyDidDraw() { return notifyDidDraw; },
  588. set notifyDidDraw(v: boolean) { notifyDidDraw = v; },
  589. didDraw,
  590. commited,
  591. reprCount,
  592. resized,
  593. setProps: (properties, doNotRequestDraw = false) => {
  594. const props: PartialCanvas3DProps = typeof properties === 'function'
  595. ? produce(getProps(), properties)
  596. : properties;
  597. const cameraState: Partial<Camera.Snapshot> = Object.create(null);
  598. if (props.camera && props.camera.mode !== undefined && props.camera.mode !== camera.state.mode) {
  599. cameraState.mode = props.camera.mode;
  600. }
  601. if (props.cameraFog !== undefined && props.cameraFog.params) {
  602. const newFog = props.cameraFog.name === 'on' ? props.cameraFog.params.intensity : 0;
  603. if (newFog !== camera.state.fog) cameraState.fog = newFog;
  604. }
  605. if (props.cameraClipping !== undefined) {
  606. if (props.cameraClipping.far !== undefined && props.cameraClipping.far !== camera.state.clipFar) {
  607. cameraState.clipFar = props.cameraClipping.far;
  608. }
  609. if (props.cameraClipping.radius !== undefined) {
  610. const radius = (scene.boundingSphere.radius / 100) * (100 - props.cameraClipping.radius);
  611. if (radius > 0 && radius !== cameraState.radius) {
  612. // if radius = 0, NaNs happen
  613. cameraState.radius = Math.max(radius, 0.01);
  614. }
  615. }
  616. }
  617. if (Object.keys(cameraState).length > 0) camera.setState(cameraState);
  618. if (props.camera?.helper) helper.camera.setProps(props.camera.helper);
  619. if (props.camera?.manualReset !== undefined) p.camera.manualReset = props.camera.manualReset;
  620. if (props.camera?.stereo !== undefined) Object.assign(p.camera.stereo, props.camera.stereo);
  621. if (props.cameraResetDurationMs !== undefined) p.cameraResetDurationMs = props.cameraResetDurationMs;
  622. if (props.transparentBackground !== undefined) p.transparentBackground = props.transparentBackground;
  623. if (props.viewport !== undefined) {
  624. const doNotUpdate = p.viewport === props.viewport ||
  625. (p.viewport.name === props.viewport.name && shallowEqual(p.viewport.params, props.viewport.params));
  626. if (!doNotUpdate) {
  627. p.viewport = props.viewport;
  628. updateViewport();
  629. syncViewport();
  630. }
  631. }
  632. if (props.postprocessing) Object.assign(p.postprocessing, props.postprocessing);
  633. if (props.multiSample) Object.assign(p.multiSample, props.multiSample);
  634. if (props.renderer) renderer.setProps(props.renderer);
  635. if (props.trackball) controls.setProps(props.trackball);
  636. if (props.debug) helper.debug.setProps(props.debug);
  637. if (props.handle) helper.handle.setProps(props.handle);
  638. if (cameraState.mode === 'orthographic') {
  639. p.camera.stereo.name = 'off';
  640. }
  641. if (!doNotRequestDraw) {
  642. requestDraw(true);
  643. }
  644. },
  645. getImagePass: (props: Partial<ImageProps> = {}) => {
  646. return new ImagePass(webgl, renderer, scene, camera, helper, passes.draw.wboitEnabled, props);
  647. },
  648. get props() {
  649. return getProps();
  650. },
  651. get input() {
  652. return input;
  653. },
  654. get stats() {
  655. return renderer.stats;
  656. },
  657. get interaction() {
  658. return interactionHelper.events;
  659. },
  660. dispose: () => {
  661. contextRestoredSub.unsubscribe();
  662. scene.clear();
  663. helper.debug.clear();
  664. controls.dispose();
  665. renderer.dispose();
  666. interactionHelper.dispose();
  667. }
  668. };
  669. function updateViewport() {
  670. if (p.viewport.name === 'canvas') {
  671. x = 0;
  672. y = 0;
  673. width = gl.drawingBufferWidth;
  674. height = gl.drawingBufferHeight;
  675. } else {
  676. x = p.viewport.params.x * webgl.pixelRatio;
  677. y = p.viewport.params.y * webgl.pixelRatio;
  678. width = p.viewport.params.width * webgl.pixelRatio;
  679. height = p.viewport.params.height * webgl.pixelRatio;
  680. }
  681. }
  682. function syncViewport() {
  683. pickHelper.setViewport(x, y, width, height);
  684. renderer.setViewport(x, y, width, height);
  685. Viewport.set(camera.viewport, x, y, width, height);
  686. Viewport.set(controls.viewport, x, y, width, height);
  687. }
  688. }
  689. }