Created
January 18, 2022 21:05
-
-
Save gouldingken/e91764e3b452b906a1c19c4518306cf9 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import { | |
| Group, | |
| Mesh, | |
| MeshNormalMaterial, | |
| Loader, | |
| SmoothShading | |
| } from 'three'; | |
| import ViewerObjectLoader from "../../speckle-objects/modules/ViewerObjectLoader"; | |
| class SpeckleLoader extends Loader { | |
| load(options, onLoad, onProgress, onError) { | |
| const {token, objectUrl} = JSON.parse(options); | |
| this.viewerObjectLoader = new ViewerObjectLoader(objectUrl, token); | |
| const container = new Group(); | |
| const material = new MeshNormalMaterial(); | |
| const addObject = (o) => { | |
| const mesh = new Mesh(o.bufferGeometry, material); | |
| // mesh.geometry.computeFaceNormals(); | |
| // mesh.geometry.computeVertexNormals(); | |
| mesh.material.shading = SmoothShading; | |
| container.add(mesh); | |
| } | |
| const onProgressInfo = (progress) => {//TODO not sure what structure the onProgress and onError for the threejs loaders expect | |
| console.log('onProgress', progress) | |
| } | |
| const onErrorInfo = (progress) => { | |
| console.log('onError', progress) | |
| } | |
| this.viewerObjectLoader.load(addObject, onProgressInfo, onErrorInfo).then(() => { | |
| onLoad(container); | |
| }); | |
| } | |
| } | |
| export {SpeckleLoader}; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import ObjectLoader from '@speckle/objectloader' | |
| import Converter from './Converter' | |
| /** | |
| * Helper wrapper around the ObjectLoader class, with some built in assumptions. | |
| */ | |
| export default class ViewerObjectLoader { | |
| constructor( objectUrl, authToken ) { | |
| // this.viewer = parent | |
| this.token = null | |
| try { | |
| this.token = authToken || localStorage.getItem( 'AuthToken' ) | |
| } catch ( error ) { | |
| // Accessing localStorage may throw when executing on sandboxed document, ignore. | |
| } | |
| if ( !this.token ) { | |
| console.warn( 'Viewer: no auth token present. Requests to non-public stream objects will fail.' ) | |
| } | |
| // example url: `https://staging.speckle.dev/streams/a75ab4f10f/objects/f33645dc9a702de8af0af16bd5f655b0` | |
| let url = new URL( objectUrl ) | |
| let segments = url.pathname.split( '/' ) | |
| if ( segments.length < 5 || url.pathname.indexOf( 'streams' ) === -1 || url.pathname.indexOf( 'objects' ) === -1 ) { | |
| throw new Error( 'Unexpected object url format.' ) | |
| } | |
| this.serverUrl = url.origin | |
| this.streamId = segments[2] | |
| this.objectId = segments[4] | |
| this.loader = new ObjectLoader( { | |
| serverUrl: this.serverUrl, | |
| token: this.token, | |
| streamId: this.streamId, | |
| objectId: this.objectId | |
| } ) | |
| this.converter = new Converter( this.loader ) | |
| } | |
| async load( addObject, onProgress, onError ) { | |
| let first = true | |
| let current = 0 | |
| let total = 0 | |
| let viewerLoads = 0 | |
| let firstObjectPromise = null | |
| for await ( let obj of this.loader.getObjectIterator() ) { | |
| if ( first ) { | |
| firstObjectPromise = this.converter.traverseAndConvert( obj, ( o ) => { | |
| console.log('ADD TO SCENEMANAGER:', o) | |
| addObject(o); | |
| viewerLoads++ | |
| } ) | |
| first = false | |
| total = obj.totalChildrenCount | |
| } | |
| current++ | |
| onProgress({progress: current / (total + 1), id: this.objectId}); | |
| // console.log( 'load-progress', { progress: current / ( total + 1 ), id: this.objectId } ) | |
| // this.viewer.emit( 'load-progress', { progress: current / ( total + 1 ), id: this.objectId } ) | |
| } | |
| if ( firstObjectPromise ) { | |
| await firstObjectPromise | |
| } | |
| if ( viewerLoads === 0 ) { | |
| onError({warning:`Viewer: no 3d objects found in object ${this.objectId}`}); | |
| // console.warn( `Viewer: no 3d objects found in object ${this.objectId}` ) | |
| // this.viewer.emit( 'load-warning', { message: `No displayable objects found in object ${this.objectId}.` } ) | |
| } | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment