cione-comp-lib
Version:
`$ yarn add cione-comp-lib` 或者 `$ npm i cione-comp-lib -S`
102 lines (101 loc) • 3.96 kB
JavaScript
import { MeshBasicMaterial, PlaneGeometry, Scene, Mesh, OrthographicCamera, WebGLRenderTarget } from "../../../../../_three@0.150.1@three/build/three.module.mjs";
import { blobCanvas } from "../../model-viewer-base.mjs";
import { Renderer } from "../../three-components/Renderer.mjs";
import { ThreeDOMElement, $correlatedObjects, $sourceObject, $onUpdate } from "./three-dom-element.mjs";
/* @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.
*/
const quadMaterial = new MeshBasicMaterial();
const quad = new PlaneGeometry(2, 2);
let adhocNum = 0;
const $threeTexture = Symbol("threeTexture");
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 element() {
const texture = this[$threeTexture];
if (texture && (texture.isCanvasTexture || texture.isVideoTexture)) {
return texture.image;
}
return;
}
get animation() {
const texture = this[$threeTexture];
if (texture && texture.isCanvasTexture && texture.animation) {
return texture.animation;
}
return;
}
get type() {
return this.uri != null ? "external" : "embedded";
}
set name(name) {
this[$sourceObject].name = name;
}
update() {
const texture = this[$threeTexture];
if (texture && texture.isCanvasTexture && !texture.animation) {
this[$threeTexture].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");
});
}
}
export { $threeTexture, Image };