@google/model-viewer
Version:
Easily display interactive 3D models on the web and in AR!
100 lines • 4.19 kB
JavaScript
/* @license
* Copyright 2020 Google LLC. All Rights Reserved.
* Licensed under the Apache License, Version 2.0 (the 'License');
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an 'AS IS' BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { ImageLoader, Mesh, MeshBasicMaterial, OrthographicCamera, PlaneGeometry, Scene, WebGLRenderTarget } from 'three';
import { blobCanvas } from '../../model-viewer-base.js';
import { Renderer } from '../../three-components/Renderer.js';
import { $correlatedObjects, $onUpdate, $sourceObject, ThreeDOMElement } from './three-dom-element.js';
const loader = new ImageLoader();
const quadMaterial = new MeshBasicMaterial();
const quad = new PlaneGeometry(2, 2);
let adhocNum = 0;
export const $threeTexture = Symbol('threeTexture');
export const $applyTexture = Symbol('applyTexture');
/**
* Image facade implementation for Three.js textures
*/
export class Image extends ThreeDOMElement {
get [$threeTexture]() {
var _a;
console.assert(this[$correlatedObjects] != null && this[$correlatedObjects].size > 0, 'Image correlated object is undefined');
return (_a = this[$correlatedObjects]) === null || _a === void 0 ? void 0 : _a.values().next().value;
}
constructor(onUpdate, texture, gltfImage) {
gltfImage = gltfImage !== null && gltfImage !== void 0 ? gltfImage : {
name: (texture && texture.image && texture.image.src) ?
texture.image.src.split('/').pop() :
'adhoc_image',
uri: (texture && texture.image && texture.image.src) ?
texture.image.src :
'adhoc_image' + adhocNum++
};
super(onUpdate, gltfImage, new Set(texture ? [texture] : []));
}
get name() {
return this[$sourceObject].name || '';
}
get uri() {
return this[$sourceObject].uri;
}
get bufferView() {
return this[$sourceObject].bufferView;
}
get type() {
return this.uri != null ? 'external' : 'embedded';
}
set name(name) {
this[$sourceObject].name = name;
}
async setURI(uri) {
this[$sourceObject].uri = uri;
this[$sourceObject].name = uri.split('/').pop();
const image = await new Promise((resolve, reject) => {
loader.load(uri, resolve, undefined, reject);
});
const texture = this[$threeTexture];
texture.image = image;
texture.needsUpdate = true;
this[$onUpdate]();
}
async createThumbnail(width, height) {
const scene = new Scene();
quadMaterial.map = this[$threeTexture];
const mesh = new Mesh(quad, quadMaterial);
scene.add(mesh);
const camera = new OrthographicCamera(-1, 1, 1, -1, 0, 1);
const { threeRenderer } = Renderer.singleton;
const renderTarget = new WebGLRenderTarget(width, height);
threeRenderer.setRenderTarget(renderTarget);
threeRenderer.render(scene, camera);
threeRenderer.setRenderTarget(null);
const buffer = new Uint8Array(width * height * 4);
threeRenderer.readRenderTargetPixels(renderTarget, 0, 0, width, height, buffer);
blobCanvas.width = width;
blobCanvas.height = height;
const blobContext = blobCanvas.getContext('2d');
const imageData = blobContext.createImageData(width, height);
imageData.data.set(buffer);
blobContext.putImageData(imageData, 0, 0);
return new Promise(async (resolve, reject) => {
blobCanvas.toBlob(blob => {
if (!blob) {
return reject('Failed to capture thumbnail.');
}
resolve(URL.createObjectURL(blob));
}, 'image/png');
});
}
}
//# sourceMappingURL=image.js.map