Spaces:
Sleeping
Sleeping
File size: 4,946 Bytes
fcf3b03 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 | // @flow
import {RGBAImage} from '../util/image';
import {register} from '../util/web_worker_transfer';
import potpack from 'potpack';
import type {StyleImage} from '../style/style_image';
import type ImageManager from './image_manager';
import type Texture from './texture';
const IMAGE_PADDING: number = 1;
export {IMAGE_PADDING};
type Rect = {
x: number,
y: number,
w: number,
h: number
};
export class ImagePosition {
paddedRect: Rect;
pixelRatio: number;
version: number;
stretchY: ?Array<[number, number]>;
stretchX: ?Array<[number, number]>;
content: ?[number, number, number, number];
constructor(paddedRect: Rect, {pixelRatio, version, stretchX, stretchY, content}: StyleImage) {
this.paddedRect = paddedRect;
this.pixelRatio = pixelRatio;
this.stretchX = stretchX;
this.stretchY = stretchY;
this.content = content;
this.version = version;
}
get tl(): [number, number] {
return [
this.paddedRect.x + IMAGE_PADDING,
this.paddedRect.y + IMAGE_PADDING
];
}
get br(): [number, number] {
return [
this.paddedRect.x + this.paddedRect.w - IMAGE_PADDING,
this.paddedRect.y + this.paddedRect.h - IMAGE_PADDING
];
}
get tlbr(): Array<number> {
return this.tl.concat(this.br);
}
get displaySize(): [number, number] {
return [
(this.paddedRect.w - IMAGE_PADDING * 2) / this.pixelRatio,
(this.paddedRect.h - IMAGE_PADDING * 2) / this.pixelRatio
];
}
}
export default class ImageAtlas {
image: RGBAImage;
iconPositions: {[_: string]: ImagePosition};
patternPositions: {[_: string]: ImagePosition};
haveRenderCallbacks: Array<string>;
uploaded: ?boolean;
constructor(icons: {[_: string]: StyleImage}, patterns: {[_: string]: StyleImage}) {
const iconPositions = {}, patternPositions = {};
this.haveRenderCallbacks = [];
const bins = [];
this.addImages(icons, iconPositions, bins);
this.addImages(patterns, patternPositions, bins);
const {w, h} = potpack(bins);
const image = new RGBAImage({width: w || 1, height: h || 1});
for (const id in icons) {
const src = icons[id];
const bin = iconPositions[id].paddedRect;
RGBAImage.copy(src.data, image, {x: 0, y: 0}, {x: bin.x + IMAGE_PADDING, y: bin.y + IMAGE_PADDING}, src.data);
}
for (const id in patterns) {
const src = patterns[id];
const bin = patternPositions[id].paddedRect;
const x = bin.x + IMAGE_PADDING,
y = bin.y + IMAGE_PADDING,
w = src.data.width,
h = src.data.height;
RGBAImage.copy(src.data, image, {x: 0, y: 0}, {x, y}, src.data);
// Add 1 pixel wrapped padding on each side of the image.
RGBAImage.copy(src.data, image, {x: 0, y: h - 1}, {x, y: y - 1}, {width: w, height: 1}); // T
RGBAImage.copy(src.data, image, {x: 0, y: 0}, {x, y: y + h}, {width: w, height: 1}); // B
RGBAImage.copy(src.data, image, {x: w - 1, y: 0}, {x: x - 1, y}, {width: 1, height: h}); // L
RGBAImage.copy(src.data, image, {x: 0, y: 0}, {x: x + w, y}, {width: 1, height: h}); // R
}
this.image = image;
this.iconPositions = iconPositions;
this.patternPositions = patternPositions;
}
addImages(images: {[_: string]: StyleImage}, positions: {[_: string]: ImagePosition}, bins: Array<Rect>) {
for (const id in images) {
const src = images[id];
const bin = {
x: 0,
y: 0,
w: src.data.width + 2 * IMAGE_PADDING,
h: src.data.height + 2 * IMAGE_PADDING,
};
bins.push(bin);
positions[id] = new ImagePosition(bin, src);
if (src.hasRenderCallback) {
this.haveRenderCallbacks.push(id);
}
}
}
patchUpdatedImages(imageManager: ImageManager, texture: Texture) {
imageManager.dispatchRenderCallbacks(this.haveRenderCallbacks);
for (const name in imageManager.updatedImages) {
this.patchUpdatedImage(this.iconPositions[name], imageManager.getImage(name), texture);
this.patchUpdatedImage(this.patternPositions[name], imageManager.getImage(name), texture);
}
}
patchUpdatedImage(position: ?ImagePosition, image: ?StyleImage, texture: Texture) {
if (!position || !image) return;
if (position.version === image.version) return;
position.version = image.version;
const [x, y] = position.tl;
texture.update(image.data, undefined, {x, y});
}
}
register('ImagePosition', ImagePosition);
register('ImageAtlas', ImageAtlas);
|