canvas3d.ts 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726
  1. /**
  2. * Copyright (c) 2018-2020 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(50, { 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 { Canvas3D };
  84. interface Canvas3D {
  85. readonly webgl: WebGLContext,
  86. add(repr: Representation.Any): void
  87. remove(repr: Representation.Any): void
  88. /**
  89. * This function must be called if animate() is not set up so that add/remove actions take place.
  90. */
  91. commit(isSynchronous?: boolean): void
  92. /**
  93. * Funcion for external "animation" control
  94. * Calls commit.
  95. */
  96. tick(t: now.Timestamp, options?: { isSynchronous?: boolean, manualDraw?: boolean }): void
  97. update(repr?: Representation.Any, keepBoundingSphere?: boolean): void
  98. clear(): void
  99. syncVisibility(): void
  100. requestDraw(force?: boolean): void
  101. /** Reset the timers, used by "animate" */
  102. resetTime(t: number): void
  103. animate(): void
  104. pause(): void
  105. identify(x: number, y: number): PickData | undefined
  106. mark(loci: Representation.Loci, action: MarkerAction): void
  107. getLoci(pickingId: PickingId | undefined): Representation.Loci
  108. notifyDidDraw: boolean,
  109. readonly didDraw: BehaviorSubject<now.Timestamp>
  110. readonly reprCount: BehaviorSubject<number>
  111. readonly resized: BehaviorSubject<any>
  112. handleResize(): void
  113. /** Focuses camera on scene's bounding sphere, centered and zoomed. */
  114. requestCameraReset(options?: { durationMs?: number, snapshot?: Partial<Camera.Snapshot> }): void
  115. readonly camera: Camera
  116. readonly boundingSphere: Readonly<Sphere3D>
  117. setProps(props: PartialCanvas3DProps | ((old: Canvas3DProps) => Partial<Canvas3DProps> | void), doNotRequestDraw?: boolean /* = false */): void
  118. getImagePass(props: Partial<ImageProps>): ImagePass
  119. /** Returns a copy of the current Canvas3D instance props */
  120. readonly props: Readonly<Canvas3DProps>
  121. readonly input: InputObserver
  122. readonly stats: RendererStats
  123. readonly interaction: Canvas3dInteractionHelper['events']
  124. dispose(): void
  125. }
  126. const requestAnimationFrame = typeof window !== 'undefined'
  127. ? window.requestAnimationFrame
  128. : (f: (time: number) => void) => setImmediate(() => f(Date.now())) as unknown as number;
  129. const cancelAnimationFrame = typeof window !== 'undefined'
  130. ? window.cancelAnimationFrame
  131. : (handle: number) => clearImmediate(handle as unknown as NodeJS.Immediate);
  132. namespace Canvas3D {
  133. export interface HoverEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, page?: Vec2, position?: Vec3 }
  134. export interface DragEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, pageStart: Vec2, pageEnd: Vec2 }
  135. export interface ClickEvent { current: Representation.Loci, buttons: ButtonsType, button: ButtonsType.Flag, modifiers: ModifiersKeys, page?: Vec2, position?: Vec3 }
  136. export function fromCanvas(canvas: HTMLCanvasElement, props: Partial<Canvas3DProps> = {}, attribs: Partial<{ antialias: boolean, pixelScale: number, pickScale: number, enableWboit: boolean }> = {}) {
  137. const antialias = (attribs.antialias ?? true) && !attribs.enableWboit;
  138. const gl = getGLContext(canvas, {
  139. alpha: true,
  140. antialias,
  141. depth: true,
  142. preserveDrawingBuffer: true,
  143. premultipliedAlpha: true,
  144. });
  145. if (gl === null) throw new Error('Could not create a WebGL rendering context');
  146. const { pixelScale } = attribs;
  147. const input = InputObserver.fromElement(canvas, { pixelScale });
  148. const webgl = createContext(gl, { pixelScale });
  149. const passes = new Passes(webgl, attribs);
  150. if (isDebugMode) {
  151. const loseContextExt = gl.getExtension('WEBGL_lose_context');
  152. if (loseContextExt) {
  153. canvas.addEventListener('mousedown', e => {
  154. if (webgl.isContextLost) return;
  155. if (!e.shiftKey || !e.ctrlKey || !e.altKey) return;
  156. if (isDebugMode) console.log('lose context');
  157. loseContextExt.loseContext();
  158. setTimeout(() => {
  159. if (!webgl.isContextLost) return;
  160. if (isDebugMode) console.log('restore context');
  161. loseContextExt.restoreContext();
  162. }, 1000);
  163. }, false);
  164. }
  165. }
  166. // https://www.khronos.org/webgl/wiki/HandlingContextLost
  167. canvas.addEventListener('webglcontextlost', e => {
  168. webgl.setContextLost();
  169. e.preventDefault();
  170. if (isDebugMode) console.log('context lost');
  171. }, false);
  172. canvas.addEventListener('webglcontextrestored', () => {
  173. if (!webgl.isContextLost) return;
  174. webgl.handleContextRestored();
  175. if (isDebugMode) console.log('context restored');
  176. }, false);
  177. // disable postprocessing anti-aliasing if canvas anti-aliasing is enabled
  178. if (antialias && !props.postprocessing?.antialiasing) {
  179. props.postprocessing = {
  180. ...DefaultCanvas3DParams.postprocessing,
  181. antialiasing: { name: 'off', params: {} }
  182. };
  183. }
  184. return create(webgl, input, passes, props, { pixelScale });
  185. }
  186. export function create(webgl: WebGLContext, input: InputObserver, passes: Passes, props: Partial<Canvas3DProps> = {}, attribs: Partial<{ pixelScale: number }>): Canvas3D {
  187. const p: Canvas3DProps = { ...DefaultCanvas3DParams, ...props };
  188. const reprRenderObjects = new Map<Representation.Any, Set<GraphicsRenderObject>>();
  189. const reprUpdatedSubscriptions = new Map<Representation.Any, Subscription>();
  190. const reprCount = new BehaviorSubject(0);
  191. let startTime = now();
  192. const didDraw = new BehaviorSubject<now.Timestamp>(0 as now.Timestamp);
  193. const { gl, contextRestored } = webgl;
  194. let x = 0;
  195. let y = 0;
  196. let width = 128;
  197. let height = 128;
  198. updateViewport();
  199. const scene = Scene.create(webgl);
  200. const camera = new Camera({
  201. position: Vec3.create(0, 0, 100),
  202. mode: p.camera.mode,
  203. fog: p.cameraFog.name === 'on' ? p.cameraFog.params.intensity : 0,
  204. clipFar: p.cameraClipping.far
  205. }, { x, y, width, height }, { pixelScale: attribs.pixelScale });
  206. const stereoCamera = new StereoCamera(camera, p.camera.stereo.params);
  207. const controls = TrackballControls.create(input, camera, p.trackball);
  208. const renderer = Renderer.create(webgl, p.renderer);
  209. const helper = new Helper(webgl, scene, p);
  210. const pickHelper = new PickHelper(webgl, renderer, scene, helper, passes.pick, { x, y, width, height });
  211. const interactionHelper = new Canvas3dInteractionHelper(identify, getLoci, input, camera);
  212. const multiSampleHelper = new MultiSampleHelper(passes.multiSample);
  213. let drawPending = false;
  214. let cameraResetRequested = false;
  215. let nextCameraResetDuration: number | undefined = void 0;
  216. let nextCameraResetSnapshot: Partial<Camera.Snapshot> | undefined = void 0;
  217. let notifyDidDraw = true;
  218. function getLoci(pickingId: PickingId | undefined) {
  219. let loci: Loci = EmptyLoci;
  220. let repr: Representation.Any = Representation.Empty;
  221. if (pickingId) {
  222. loci = helper.handle.getLoci(pickingId);
  223. reprRenderObjects.forEach((_, _repr) => {
  224. const _loci = _repr.getLoci(pickingId);
  225. if (!isEmptyLoci(_loci)) {
  226. if (!isEmptyLoci(loci)) {
  227. console.warn('found another loci, this should not happen');
  228. }
  229. loci = _loci;
  230. repr = _repr;
  231. }
  232. });
  233. }
  234. return { loci, repr };
  235. }
  236. function mark(reprLoci: Representation.Loci, action: MarkerAction) {
  237. const { repr, loci } = reprLoci;
  238. let changed = false;
  239. if (repr) {
  240. changed = repr.mark(loci, action);
  241. } else {
  242. changed = helper.handle.mark(loci, action);
  243. reprRenderObjects.forEach((_, _repr) => { changed = _repr.mark(loci, action) || changed; });
  244. }
  245. if (changed) {
  246. scene.update(void 0, true);
  247. helper.handle.scene.update(void 0, true);
  248. const prevPickDirty = pickHelper.dirty;
  249. draw(true);
  250. pickHelper.dirty = prevPickDirty; // marking does not change picking buffers
  251. }
  252. }
  253. function render(force: boolean) {
  254. if (webgl.isContextLost) return false;
  255. if (x > gl.drawingBufferWidth || x + width < 0 ||
  256. y > gl.drawingBufferHeight || y + height < 0
  257. ) return false;
  258. let didRender = false;
  259. controls.update(currentTime);
  260. const cameraChanged = camera.update();
  261. const multiSampleChanged = multiSampleHelper.update(force || cameraChanged, p.multiSample);
  262. if (force || cameraChanged || multiSampleChanged) {
  263. let cam: Camera | StereoCamera = camera;
  264. if (p.camera.stereo.name === 'on') {
  265. stereoCamera.update();
  266. cam = stereoCamera;
  267. }
  268. if (MultiSamplePass.isEnabled(p.multiSample)) {
  269. multiSampleHelper.render(renderer, cam, scene, helper, true, p.transparentBackground, p);
  270. } else {
  271. passes.draw.render(renderer, cam, scene, helper, true, p.transparentBackground, p.postprocessing);
  272. }
  273. pickHelper.dirty = true;
  274. didRender = true;
  275. }
  276. return didRender;
  277. }
  278. let forceNextDraw = false;
  279. let forceDrawAfterAllCommited = false;
  280. let currentTime = 0;
  281. function draw(force?: boolean) {
  282. if (render(!!force || forceNextDraw) && notifyDidDraw) {
  283. didDraw.next(now() - startTime as now.Timestamp);
  284. }
  285. forceNextDraw = false;
  286. drawPending = false;
  287. }
  288. function requestDraw(force?: boolean) {
  289. if (drawPending) return;
  290. drawPending = true;
  291. forceNextDraw = !!force;
  292. }
  293. let animationFrameHandle = 0;
  294. function tick(t: now.Timestamp, options?: { isSynchronous?: boolean, manualDraw?: boolean }) {
  295. currentTime = t;
  296. commit(options?.isSynchronous);
  297. camera.transition.tick(currentTime);
  298. if (options?.manualDraw) {
  299. return;
  300. }
  301. draw(false);
  302. if (!camera.transition.inTransition && !webgl.isContextLost) {
  303. interactionHelper.tick(currentTime);
  304. }
  305. }
  306. function _animate() {
  307. tick(now());
  308. animationFrameHandle = requestAnimationFrame(_animate);
  309. }
  310. function resetTime(t: now.Timestamp) {
  311. startTime = t;
  312. controls.start(t);
  313. }
  314. function animate() {
  315. controls.start(now());
  316. if (animationFrameHandle === 0) _animate();
  317. }
  318. function pause() {
  319. cancelAnimationFrame(animationFrameHandle);
  320. animationFrameHandle = 0;
  321. }
  322. function identify(x: number, y: number): PickData | undefined {
  323. const cam = p.camera.stereo.name === 'on' ? stereoCamera : camera;
  324. return webgl.isContextLost ? undefined : pickHelper.identify(x, y, cam);
  325. }
  326. function commit(isSynchronous: boolean = false) {
  327. const allCommited = commitScene(isSynchronous);
  328. // Only reset the camera after the full scene has been commited.
  329. if (allCommited) {
  330. resolveCameraReset();
  331. if (forceDrawAfterAllCommited) {
  332. if (helper.debug.isEnabled) helper.debug.update();
  333. draw(true);
  334. forceDrawAfterAllCommited = false;
  335. }
  336. }
  337. }
  338. function resolveCameraReset() {
  339. if (!cameraResetRequested) return;
  340. const { center, radius } = scene.boundingSphereVisible;
  341. if (radius > 0) {
  342. const duration = nextCameraResetDuration === undefined ? p.cameraResetDurationMs : nextCameraResetDuration;
  343. const focus = camera.getFocus(center, radius);
  344. const snapshot = nextCameraResetSnapshot ? { ...focus, ...nextCameraResetSnapshot } : focus;
  345. camera.setState({ ...snapshot, radiusMax: scene.boundingSphere.radius }, duration);
  346. }
  347. nextCameraResetDuration = void 0;
  348. nextCameraResetSnapshot = void 0;
  349. cameraResetRequested = false;
  350. }
  351. const oldBoundingSphereVisible = Sphere3D();
  352. const cameraSphere = Sphere3D();
  353. function shouldResetCamera() {
  354. if (camera.state.radiusMax === 0) return true;
  355. if (camera.transition.inTransition || nextCameraResetSnapshot) return false;
  356. let cameraSphereOverlapsNone = true;
  357. Sphere3D.set(cameraSphere, camera.state.target, camera.state.radius);
  358. // check if any renderable has moved outside of the old bounding sphere
  359. // and if no renderable is overlapping with the camera sphere
  360. for (const r of scene.renderables) {
  361. if (!r.state.visible) continue;
  362. const b = r.values.boundingSphere.ref.value;
  363. if (!b.radius) continue;
  364. const cameraDist = Vec3.distance(cameraSphere.center, b.center);
  365. if ((cameraDist > cameraSphere.radius || cameraDist > b.radius || b.radius > camera.state.radiusMax) && !Sphere3D.includes(oldBoundingSphereVisible, b)) return true;
  366. if (Sphere3D.overlaps(cameraSphere, b)) cameraSphereOverlapsNone = false;
  367. }
  368. return cameraSphereOverlapsNone;
  369. }
  370. const sceneCommitTimeoutMs = 250;
  371. function commitScene(isSynchronous: boolean) {
  372. if (!scene.needsCommit) return true;
  373. // snapshot the current bounding sphere of visible objects
  374. Sphere3D.copy(oldBoundingSphereVisible, scene.boundingSphereVisible);
  375. if (!scene.commit(isSynchronous ? void 0 : sceneCommitTimeoutMs)) return false;
  376. if (helper.debug.isEnabled) helper.debug.update();
  377. if (!p.camera.manualReset && (reprCount.value === 0 || shouldResetCamera())) {
  378. cameraResetRequested = true;
  379. }
  380. if (oldBoundingSphereVisible.radius === 0) nextCameraResetDuration = 0;
  381. camera.setState({ radiusMax: scene.boundingSphere.radius }, 0);
  382. reprCount.next(reprRenderObjects.size);
  383. if (isDebugMode) consoleStats();
  384. return true;
  385. }
  386. function consoleStats() {
  387. console.table(scene.renderables.map(r => ({
  388. drawCount: r.values.drawCount.ref.value,
  389. instanceCount: r.values.instanceCount.ref.value,
  390. materialId: r.materialId,
  391. })));
  392. console.log(webgl.stats);
  393. }
  394. function add(repr: Representation.Any) {
  395. registerAutoUpdate(repr);
  396. const oldRO = reprRenderObjects.get(repr);
  397. const newRO = new Set<GraphicsRenderObject>();
  398. repr.renderObjects.forEach(o => newRO.add(o));
  399. if (oldRO) {
  400. if (!SetUtils.areEqual(newRO, oldRO)) {
  401. newRO.forEach(o => { if (!oldRO.has(o)) scene.add(o); });
  402. oldRO.forEach(o => { if (!newRO.has(o)) scene.remove(o); });
  403. }
  404. } else {
  405. repr.renderObjects.forEach(o => scene.add(o));
  406. }
  407. reprRenderObjects.set(repr, newRO);
  408. scene.update(repr.renderObjects, false);
  409. forceDrawAfterAllCommited = true;
  410. if (isDebugMode) consoleStats();
  411. }
  412. function remove(repr: Representation.Any) {
  413. unregisterAutoUpdate(repr);
  414. const renderObjects = reprRenderObjects.get(repr);
  415. if (renderObjects) {
  416. renderObjects.forEach(o => scene.remove(o));
  417. reprRenderObjects.delete(repr);
  418. forceDrawAfterAllCommited = true;
  419. if (isDebugMode) consoleStats();
  420. }
  421. }
  422. function registerAutoUpdate(repr: Representation.Any) {
  423. if (reprUpdatedSubscriptions.has(repr)) return;
  424. reprUpdatedSubscriptions.set(repr, repr.updated.subscribe(_ => {
  425. if (!repr.state.syncManually) add(repr);
  426. }));
  427. }
  428. function unregisterAutoUpdate(repr: Representation.Any) {
  429. const updatedSubscription = reprUpdatedSubscriptions.get(repr);
  430. if (updatedSubscription) {
  431. updatedSubscription.unsubscribe();
  432. reprUpdatedSubscriptions.delete(repr);
  433. }
  434. }
  435. function getProps(): Canvas3DProps {
  436. const radius = scene.boundingSphere.radius > 0
  437. ? 100 - Math.round((camera.transition.target.radius / scene.boundingSphere.radius) * 100)
  438. : 0;
  439. return {
  440. camera: {
  441. mode: camera.state.mode,
  442. helper: { ...helper.camera.props },
  443. stereo: { ...p.camera.stereo },
  444. manualReset: !!p.camera.manualReset
  445. },
  446. cameraFog: camera.state.fog > 0
  447. ? { name: 'on' as const, params: { intensity: camera.state.fog } }
  448. : { name: 'off' as const, params: {} },
  449. cameraClipping: { far: camera.state.clipFar, radius },
  450. cameraResetDurationMs: p.cameraResetDurationMs,
  451. transparentBackground: p.transparentBackground,
  452. viewport: p.viewport,
  453. postprocessing: { ...p.postprocessing },
  454. multiSample: { ...p.multiSample },
  455. renderer: { ...renderer.props },
  456. trackball: { ...controls.props },
  457. debug: { ...helper.debug.props },
  458. handle: { ...helper.handle.props },
  459. };
  460. }
  461. const contextRestoredSub = contextRestored.subscribe(() => {
  462. pickHelper.dirty = true;
  463. draw(true);
  464. });
  465. const resized = new BehaviorSubject<any>(0);
  466. return {
  467. webgl,
  468. add,
  469. remove,
  470. commit,
  471. update: (repr, keepSphere) => {
  472. if (repr) {
  473. if (!reprRenderObjects.has(repr)) return;
  474. scene.update(repr.renderObjects, !!keepSphere);
  475. } else {
  476. scene.update(void 0, !!keepSphere);
  477. }
  478. forceDrawAfterAllCommited = true;
  479. },
  480. clear: () => {
  481. reprUpdatedSubscriptions.forEach(v => v.unsubscribe());
  482. reprUpdatedSubscriptions.clear();
  483. reprRenderObjects.clear();
  484. scene.clear();
  485. helper.debug.clear();
  486. requestDraw(true);
  487. reprCount.next(reprRenderObjects.size);
  488. },
  489. syncVisibility: () => {
  490. if (camera.state.radiusMax === 0) {
  491. cameraResetRequested = true;
  492. nextCameraResetDuration = 0;
  493. }
  494. if (scene.syncVisibility()) {
  495. if (helper.debug.isEnabled) helper.debug.update();
  496. }
  497. requestDraw(true);
  498. },
  499. requestDraw,
  500. tick,
  501. animate,
  502. resetTime,
  503. pause,
  504. identify,
  505. mark,
  506. getLoci,
  507. handleResize: () => {
  508. passes.updateSize();
  509. updateViewport();
  510. syncViewport();
  511. requestDraw(true);
  512. resized.next(+new Date());
  513. },
  514. requestCameraReset: options => {
  515. nextCameraResetDuration = options?.durationMs;
  516. nextCameraResetSnapshot = options?.snapshot;
  517. cameraResetRequested = true;
  518. },
  519. camera,
  520. boundingSphere: scene.boundingSphere,
  521. get notifyDidDraw() { return notifyDidDraw; },
  522. set notifyDidDraw(v: boolean) { notifyDidDraw = v; },
  523. didDraw,
  524. reprCount,
  525. resized,
  526. setProps: (properties, doNotRequestDraw = false) => {
  527. const props: PartialCanvas3DProps = typeof properties === 'function'
  528. ? produce(getProps(), properties)
  529. : properties;
  530. const cameraState: Partial<Camera.Snapshot> = Object.create(null);
  531. if (props.camera && props.camera.mode !== undefined && props.camera.mode !== camera.state.mode) {
  532. cameraState.mode = props.camera.mode;
  533. }
  534. if (props.cameraFog !== undefined && props.cameraFog.params) {
  535. const newFog = props.cameraFog.name === 'on' ? props.cameraFog.params.intensity : 0;
  536. if (newFog !== camera.state.fog) cameraState.fog = newFog;
  537. }
  538. if (props.cameraClipping !== undefined) {
  539. if (props.cameraClipping.far !== undefined && props.cameraClipping.far !== camera.state.clipFar) {
  540. cameraState.clipFar = props.cameraClipping.far;
  541. }
  542. if (props.cameraClipping.radius !== undefined) {
  543. const radius = (scene.boundingSphere.radius / 100) * (100 - props.cameraClipping.radius);
  544. if (radius > 0 && radius !== cameraState.radius) {
  545. // if radius = 0, NaNs happen
  546. cameraState.radius = Math.max(radius, 0.01);
  547. }
  548. }
  549. }
  550. if (Object.keys(cameraState).length > 0) camera.setState(cameraState);
  551. if (props.camera?.helper) helper.camera.setProps(props.camera.helper);
  552. if (props.camera?.manualReset !== undefined) p.camera.manualReset = props.camera.manualReset;
  553. if (props.camera?.stereo !== undefined) Object.assign(p.camera.stereo, props.camera.stereo);
  554. if (props.cameraResetDurationMs !== undefined) p.cameraResetDurationMs = props.cameraResetDurationMs;
  555. if (props.transparentBackground !== undefined) p.transparentBackground = props.transparentBackground;
  556. if (props.viewport !== undefined) {
  557. const doNotUpdate = p.viewport === props.viewport ||
  558. (p.viewport.name === props.viewport.name && shallowEqual(p.viewport.params, props.viewport.params));
  559. if (!doNotUpdate) {
  560. p.viewport = props.viewport;
  561. updateViewport();
  562. syncViewport();
  563. }
  564. }
  565. if (props.postprocessing) Object.assign(p.postprocessing, props.postprocessing);
  566. if (props.multiSample) Object.assign(p.multiSample, props.multiSample);
  567. if (props.renderer) renderer.setProps(props.renderer);
  568. if (props.trackball) controls.setProps(props.trackball);
  569. if (props.debug) helper.debug.setProps(props.debug);
  570. if (props.handle) helper.handle.setProps(props.handle);
  571. if (cameraState.mode === 'orthographic') {
  572. p.camera.stereo.name = 'off';
  573. }
  574. if (!doNotRequestDraw) {
  575. requestDraw(true);
  576. }
  577. },
  578. getImagePass: (props: Partial<ImageProps> = {}) => {
  579. return new ImagePass(webgl, renderer, scene, camera, helper, passes.draw.wboitEnabled, props);
  580. },
  581. get props() {
  582. return getProps();
  583. },
  584. get input() {
  585. return input;
  586. },
  587. get stats() {
  588. return renderer.stats;
  589. },
  590. get interaction() {
  591. return interactionHelper.events;
  592. },
  593. dispose: () => {
  594. contextRestoredSub.unsubscribe();
  595. scene.clear();
  596. helper.debug.clear();
  597. input.dispose();
  598. controls.dispose();
  599. renderer.dispose();
  600. interactionHelper.dispose();
  601. }
  602. };
  603. function updateViewport() {
  604. if (p.viewport.name === 'canvas') {
  605. x = 0;
  606. y = 0;
  607. width = gl.drawingBufferWidth;
  608. height = gl.drawingBufferHeight;
  609. } else {
  610. x = p.viewport.params.x * webgl.pixelRatio;
  611. y = p.viewport.params.y * webgl.pixelRatio;
  612. width = p.viewport.params.width * webgl.pixelRatio;
  613. height = p.viewport.params.height * webgl.pixelRatio;
  614. }
  615. }
  616. function syncViewport() {
  617. pickHelper.setViewport(x, y, width, height);
  618. renderer.setViewport(x, y, width, height);
  619. Viewport.set(camera.viewport, x, y, width, height);
  620. Viewport.set(controls.viewport, x, y, width, height);
  621. }
  622. }
  623. }