304 lines
9.4 KiB
JavaScript
304 lines
9.4 KiB
JavaScript
const _ = require('lodash');
|
|
const got = require('got');
|
|
const Canvas = require('canvas');
|
|
const { GifReader } = require('omggif');
|
|
const GifEncoder = require('gifencoder');
|
|
|
|
const Image = { Canvas };
|
|
|
|
function loadFromUrl(url) {
|
|
return got(url, { encoding: null }).then(res => ({
|
|
type: res.headers['content-type'],
|
|
data: res.body
|
|
}));
|
|
}
|
|
|
|
function createCanvas(width, height) {
|
|
const canvas = new Canvas(width, height);
|
|
return canvas;
|
|
}
|
|
|
|
class ImageEx {
|
|
constructor(url) {
|
|
this.url = url;
|
|
this.loaded = loadFromUrl(url).then(result => {
|
|
this.type = result.type;
|
|
this.data = result.data;
|
|
if (this.type === 'image/gif') {
|
|
this.initGif();
|
|
} else {
|
|
this.initStatic();
|
|
}
|
|
return this;
|
|
});
|
|
}
|
|
|
|
|
|
initGif() {
|
|
const reader = new GifReader(new Uint8Array(this.data));
|
|
this.width = reader.width;
|
|
this.height = reader.height;
|
|
this.frames = this.decodeFrames(reader);
|
|
|
|
this.renderAllFrames();
|
|
}
|
|
|
|
initStatic() {
|
|
const img = new Image();
|
|
img.src = this.data;
|
|
return new Promise(resolve => {
|
|
img.onload = () => {
|
|
this.width = img.width;
|
|
this.height = img.height;
|
|
this.frames = [{
|
|
actualOffset: 0,
|
|
actualDelay: Infinity,
|
|
delay: Infinity
|
|
}];
|
|
this.spriteSheet = createCanvas(this.width, this.height);
|
|
const spriteSheetCtx = this.spriteSheet.getContext('2d');
|
|
spriteSheetCtx.drawImage(img, 0, 0);
|
|
resolve();
|
|
};
|
|
});
|
|
}
|
|
|
|
decodeFrames(reader) {
|
|
const frames = [];
|
|
let offset = 0;
|
|
for (let i = 0; i < reader.numFrames(); ++i) {
|
|
const frameInfo = reader.frameInfo(i);
|
|
frameInfo.pixels = new Uint8ClampedArray(reader.width * reader.height * 4);
|
|
reader.decodeAndBlitFrameRGBA(i, frameInfo.pixels);
|
|
frameInfo.buffer = this.createBufferCanvas(frameInfo, this.width, this.height);
|
|
frameInfo.actualOffset = offset;
|
|
frameInfo.actualDelay = Math.max(frameInfo.delay * 10, 20);
|
|
offset += frameInfo.actualDelay;
|
|
frames.push(frameInfo);
|
|
}
|
|
this.totalDuration = offset;
|
|
return frames;
|
|
}
|
|
|
|
renderAllFrames() {
|
|
let disposeFrame = null;
|
|
const canvas = createCanvas(this.width, this.height);
|
|
const ctx = canvas.getContext('2d');
|
|
let saved;
|
|
this.spriteSheet = createCanvas(this.width * this.frames.length, this.height);
|
|
const spriteSheetCtx = this.spriteSheet.getContext('2d');
|
|
for (let i = 0; i < this.frames.length; ++i) {
|
|
const frame = this.frames[i];
|
|
if (typeof disposeFrame === 'function') disposeFrame();
|
|
|
|
switch (frame.disposal) {
|
|
case 2:
|
|
disposeFrame = () => ctx.clearRect(0, 0, canvas.width, canvas.height);
|
|
break;
|
|
case 3:
|
|
saved = ctx.getImageData(0, 0, canvas.width, canvas.height);
|
|
disposeFrame = () => ctx.putImageData(saved, 0, 0); // eslint-disable-line no-loop-func
|
|
break;
|
|
default:
|
|
this.disposeFrame = null;
|
|
}
|
|
|
|
// draw current frame
|
|
ctx.drawImage(frame.buffer, frame.x, frame.y);
|
|
// draw the frame onto the sprite sheet
|
|
spriteSheetCtx.drawImage(canvas, this.width * i, 0);
|
|
}
|
|
}
|
|
|
|
createBufferCanvas(frame, width, height) {
|
|
const canvas = createCanvas(frame.width, frame.height);
|
|
const ctx = canvas.getContext('2d');
|
|
|
|
const imageData = ctx.createImageData(width, height);
|
|
imageData.data.set(frame.pixels);
|
|
|
|
ctx.putImageData(imageData, -frame.x, -frame.y);
|
|
return canvas;
|
|
}
|
|
|
|
drawFrame(ctx, frameNum, x, y, args = {}) {
|
|
const sx = frameNum * this.width + (args.sx || 0);
|
|
const sy = args.sy || 0;
|
|
const swidth = Math.min(args.swidth || this.width, this.width) - (args.sx || 0);
|
|
const sheight = args.sheight || this.height;
|
|
|
|
console.log(`Drawing frame ${frameNum} at`);
|
|
console.log('sx', sx);
|
|
console.log('sy', sy);
|
|
console.log('sw', swidth);
|
|
console.log('sh', sheight);
|
|
console.log('x', x);
|
|
console.log('y', y);
|
|
console.log('w', args.width);
|
|
console.log('h', args.height);
|
|
|
|
ctx.drawImage(this.spriteSheet, sx, sy, swidth, sheight, x, y, args.width || swidth, args.height || sheight);
|
|
// ctx.drawImage(this.spriteSheet, 0, 0, 112, 112, 0, 0, 112, 112)
|
|
}
|
|
}
|
|
class CanvasEx {
|
|
constructor(width, height) {
|
|
this.width = width;
|
|
this.height = height;
|
|
this.frames = [];
|
|
this.totalDuration = Infinity;
|
|
}
|
|
|
|
addFrame(actualDelay, delay) {
|
|
if ((actualDelay === undefined || actualDelay === null)
|
|
&& (delay === undefined || delay === null)) throw new Error('Delay has to be set!');
|
|
const canvas = createCanvas(this.width, this.height);
|
|
const frame = {
|
|
actualOffset: this.totalDuration,
|
|
delay: delay || Math.max(Math.round(actualDelay / 10), 2),
|
|
actualDelay: actualDelay || Math.max(delay * 10, 20),
|
|
canvas,
|
|
ctx: canvas.getContext('2d')
|
|
};
|
|
this.totalDuration += delay;
|
|
this.frames.push(frame);
|
|
}
|
|
|
|
drawImage(img, x, y, args = {}) {
|
|
console.log('Drawing image ', img);
|
|
console.log('At ', x, y, args);
|
|
if (img.frames && img.frames.length > 1) {
|
|
if (this.frames.length > 1) throw new Error('Cannot render animations onto animated canvases!');
|
|
this.totalDuration = img.totalDuration;
|
|
// we are drawing an animated image onto a static one.
|
|
// for each frame in the image, create a frame on this one, cloning the original picture (if any),
|
|
// render the original on each frame, and draw the frame on top.
|
|
for (let i = this.frames.length; i < img.frames.length; ++i) {
|
|
const frame = img.frames[i];
|
|
// console.log(`Adding frame ${i}:`, frame);
|
|
this.addFrame(null, frame.delay);
|
|
if (this.frames.length > 0) {
|
|
this.frames[i].ctx.drawImage(this.frames[0].canvas, 0, 0);
|
|
}
|
|
}
|
|
for (let i = 0; i < img.frames.length; ++i) {
|
|
// console.log(`Drawing frame ${i}:`, img.frames[i]);
|
|
// draw the i-th source frame to the i-th target frame
|
|
img.drawFrame(this.frames[i].ctx, i, x, y, args);
|
|
}
|
|
console.log('Done drawing animated image, CanvasEx is now', this);
|
|
} else {
|
|
// we are drawing a static image on top of a (possibly animated) image.
|
|
// for each frame, just draw, nothing fancy.
|
|
if (img.frames) {
|
|
// the image cant have more than one frame, and if it has 0, we dont need to do anything at all
|
|
if (img.frames.length === 1) {
|
|
for (let i = 0; i < this.frames.length; ++i) {
|
|
img.drawFrame(this.frames[i].ctx, 0, x, y, args);
|
|
}
|
|
}
|
|
} else {
|
|
for (let i = 0; i < this.frames.length; ++i) {
|
|
this.drawImage(this.frames[i].ctx, img, x, y, args);
|
|
}
|
|
}
|
|
console.log('Done drawing static image, CanvasEx is now', this);
|
|
}
|
|
}
|
|
|
|
drawFrame(ctx, frameNum, x, y, args = {}) {
|
|
// console.log(`Drawing frame ${frameNum} of`, this)
|
|
// ctx.drawImage(this.frames[frameNum].canvas, x, y);
|
|
this.drawImage(ctx, this.frames[frameNum].canvas, x, y, args);
|
|
}
|
|
|
|
_drawImage(ctx, img, x, y, args = {}) {
|
|
if (args.transform) {
|
|
ctx.save();
|
|
_.each(args.transform, (value, property) => {
|
|
ctx[property](...value);
|
|
});
|
|
}
|
|
if (args.sx !== undefined || args.sy !== undefined || args.swidth !== undefined || args.sheight !== undefined) {
|
|
ctx.drawImage(img, args.sx, args.sy, args.swidth, args.sheight, x, y, args.width, args.height);
|
|
} else {
|
|
ctx.drawImage(img, x, y, args.width, args.height);
|
|
}
|
|
if (args.transform) {
|
|
ctx.restore();
|
|
}
|
|
}
|
|
|
|
export(outStream) {
|
|
const gif = new GifEncoder(this.width, this.height);
|
|
gif.createReadStream().pipe(outStream);
|
|
gif.setTransparent(0);
|
|
gif.setRepeat(0);
|
|
gif.start();
|
|
for (let i = 0; i < this.frames.length; ++i) {
|
|
const frame = this.frames[i];
|
|
gif.setDelay(frame.delay);
|
|
gif.addFrame(frame.ctx);
|
|
}
|
|
gif.finish();
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
CanvasEx,
|
|
ImageEx
|
|
};
|
|
/*
|
|
// const img = new ImageEx("https://cors-anywhere.herokuapp.com/https://cdn.betterttv.net/emote/554da1a289d53f2d12781907/3x");
|
|
|
|
const img = new ImageEx('https://cors-anywhere.herokuapp.com/https://cdn.discordapp.com/emojis/393563453824040983.gif');
|
|
const _canvas = document.getElementById('c');
|
|
const _ctx = _canvas.getContext('2d');
|
|
|
|
const img2 = new ImageEx('https://rawgit.com/CBenni/beebot/master/resources/SnowGlobe.png');
|
|
|
|
const _cnv = new CanvasEx(128, 128);
|
|
img.loaded.then(() => {
|
|
_cnv.drawImage(img, 0, 0);
|
|
// img.drawFrame(_ctx, 0, 0, 0)
|
|
return img2.loaded;
|
|
}).then(() => {
|
|
_cnv.drawImage(img2, 0, 0, { width: 128, height: 128 });
|
|
}).then(() => {
|
|
console.log('Drawing CanvasEx to screen', _cnv);
|
|
_cnv.drawFrame(_ctx, 0, 0, 0);
|
|
|
|
console.log('Starting gif render');
|
|
const GifEncoder = new GIF({
|
|
workers: 2,
|
|
quality: 10,
|
|
transparent: 'rgba(0,0,0,0)'
|
|
});
|
|
|
|
for (let i = 0; i < _cnv.frames.length; ++i) {
|
|
const frame = _cnv.frames[i];
|
|
console.log(`Rendering frame ${i}`);
|
|
GifEncoder.addFrame(frame.canvas, { delay: frame.delay });
|
|
}
|
|
|
|
GifEncoder.on('finished', blob => {
|
|
console.log('Done rendering!', URL.createObjectURL(blob));
|
|
document.getElementById('i').src = URL.createObjectURL(blob);
|
|
});
|
|
|
|
GifEncoder.render();
|
|
});
|
|
|
|
let curFrm = 0;
|
|
setInterval(() => {
|
|
_ctx.clearRect(0, 0, 200, 200);
|
|
_cnv.drawFrame(_ctx, curFrm++ % _cnv.frames.length, 0, 0);
|
|
}, 100);
|
|
|
|
|
|
/* img2.loaded.then(() => {
|
|
_ctx.drawImage(img2.spriteSheet, 0, 100)
|
|
})
|
|
|
|
img.loaded.then(()=>_ctx.drawImage(img.spriteSheet,0,0)) */
|