react-image-palette-hook
Version:
A React hook to extract and generate color palettes from images
168 lines (155 loc) • 7.78 kB
JavaScript
;
var react = require('react');
/******************************************************************************
Copyright (c) Microsoft Corporation.
Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
***************************************************************************** */
/* global Reflect, Promise, SuppressedError, Symbol, Iterator */
var __assign = function() {
__assign = Object.assign || function __assign(t) {
for (var s, i = 1, n = arguments.length; i < n; i++) {
s = arguments[i];
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p];
}
return t;
};
return __assign.apply(this, arguments);
};
function __awaiter(thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
}
function __generator(thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (g && (g = 0, op[0] && (_ = 0)), _) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
}
typeof SuppressedError === "function" ? SuppressedError : function (error, suppressed, message) {
var e = new Error(message);
return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
};
var getColorFromImageData = function (imageData, startX, startY, width, height) {
var r = 0;
var g = 0;
var b = 0;
var count = 0;
for (var x = startX; x < startX + width; x += 2) {
for (var y = startY; y < startY + height; y += 2) {
var i = (y * imageData.width + x) * 4;
r += imageData.data[i];
g += imageData.data[i + 1];
b += imageData.data[i + 2];
count += 1;
}
}
return {
r: Math.round(r / count),
g: Math.round(g / count),
b: Math.round(b / count)
};
};
var rgbToHex = function (r, g, b) {
return "#".concat([r, g, b]
.map(function (x) {
var hex = x.toString(16);
return hex.length === 1 ? "0".concat(hex) : hex;
})
.join(""));
};
var adjustColor = function (color, factor) {
return {
r: Math.min(255, Math.round(color.r * factor)),
g: Math.min(255, Math.round(color.g * factor)),
b: Math.min(255, Math.round(color.b * factor))
};
};
var usePalette = function (imageUrl) {
var _a = react.useState({
vibrant: null,
darkVibrant: null,
darkMuted: null,
loading: true
}), data = _a[0], setData = _a[1];
react.useEffect(function () {
if (!imageUrl) {
setData(function (prev) { return (__assign(__assign({}, prev), { loading: false })); });
return;
}
var extractColors = function () { return __awaiter(void 0, void 0, void 0, function () {
var img_1;
return __generator(this, function (_a) {
try {
img_1 = new Image();
img_1.crossOrigin = 'Anonymous';
img_1.onload = function () {
var canvas = document.createElement('canvas');
var ctx = canvas.getContext('2d');
if (!ctx) {
throw new Error('Could not get canvas context');
}
canvas.width = img_1.width;
canvas.height = img_1.height;
ctx.drawImage(img_1, 0, 0);
var imageData = ctx.getImageData(0, 0, canvas.width, canvas.height);
var centerColor = getColorFromImageData(imageData, Math.floor(canvas.width * 0.25), Math.floor(canvas.height * 0.25), Math.floor(canvas.width * 0.5), Math.floor(canvas.height * 0.5));
var darkVibrantColor = adjustColor(centerColor, 0.7);
var darkMutedColor = adjustColor(centerColor, 0.5);
setData({
vibrant: rgbToHex(centerColor.r, centerColor.g, centerColor.b),
darkVibrant: rgbToHex(darkVibrantColor.r, darkVibrantColor.g, darkVibrantColor.b),
darkMuted: rgbToHex(darkMutedColor.r, darkMutedColor.g, darkMutedColor.b),
loading: false
});
};
img_1.onerror = function () {
setData(function (prev) { return (__assign(__assign({}, prev), { loading: false })); });
};
img_1.src = imageUrl;
}
catch (error) {
setData(function (prev) { return (__assign(__assign({}, prev), { loading: false })); });
}
return [2 /*return*/];
});
}); };
extractColors();
}, [imageUrl]);
return { data: data };
};
exports.usePalette = usePalette;
//# sourceMappingURL=index.js.map