usulpro
/
math-tasks
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: update remove-bg

This commit is contained in:
Oleg Proskurin 2026-03-01 21:36:42 +07:00
parent eb7f1ed0c7
commit 2bdee2076e
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/scripts/remove-bg.mjs
View File

@ -7,7 +7,7 @@ const PROJECT_ROOT = resolve(fileURLToPath(import.meta.url), '../../..');
const DEFAULT_INPUT = resolve(PROJECT_ROOT, 'assets/icons/pack2'); const DEFAULT_INPUT = resolve(PROJECT_ROOT, 'assets/icons/pack2');
const DEFAULT_THRESHOLD = 230; const DEFAULT_THRESHOLD = 230;
const DEFAULT_FUZZ = 0; const DEFAULT_FUZZ = 0;
const PNG_EXTENSION = /\.png$/i; const IMAGE_EXTENSION = /\.(png|jpe?g|webp)$/i;
function parseArgs(args) { function parseArgs(args) {
const result = { input: DEFAULT_INPUT, threshold: DEFAULT_THRESHOLD, fuzz: DEFAULT_FUZZ }; const result = { input: DEFAULT_INPUT, threshold: DEFAULT_THRESHOLD, fuzz: DEFAULT_FUZZ };
@ -22,7 +22,7 @@ function parseArgs(args) {
function collectFiles(input) { function collectFiles(input) {
if (statSync(input).isFile()) return [input]; if (statSync(input).isFile()) return [input];
return readdirSync(input) return readdirSync(input)
.filter(f => PNG_EXTENSION.test(f)) .filter(f => IMAGE_EXTENSION.test(f))
.sort() .sort()
.map(f => join(input, f)); .map(f => join(input, f));
} }
@ -135,6 +135,12 @@ function applyAntiAlias(data, width, height, transparent, fuzz) {
} }
} }
function outputPath(filePath) {
const ext = extname(filePath).toLowerCase();
if (ext === '.png') return filePath;
return join(resolve(filePath, '..'), basename(filePath, extname(filePath)) + '.png');
}
async function removeBackground(filePath, threshold, fuzz) { async function removeBackground(filePath, threshold, fuzz) {
const { data, info } = await sharp(filePath) const { data, info } = await sharp(filePath)
.ensureAlpha() .ensureAlpha()
@ -145,11 +151,12 @@ async function removeBackground(filePath, threshold, fuzz) {
const { transparent, count } = floodFillFromEdges(data, width, height, threshold); const { transparent, count } = floodFillFromEdges(data, width, height, threshold);
applyAntiAlias(data, width, height, transparent, fuzz); applyAntiAlias(data, width, height, transparent, fuzz);
const outFile = outputPath(filePath);
await sharp(data, { raw: { width, height, channels: 4 } }) await sharp(data, { raw: { width, height, channels: 4 } })
.png() .png()
.toFile(filePath); .toFile(outFile);
return count; return { count, outFile };
} }
async function main() { async function main() {
@ -163,7 +170,7 @@ async function main() {
const files = collectFiles(input); const files = collectFiles(input);
if (files.length === 0) { if (files.length === 0) {
console.error(`No PNG files found in: ${input}`); console.error(`No image files found in: ${input}`);
process.exit(1); process.exit(1);
} }
@ -171,9 +178,10 @@ async function main() {
const results = await Promise.all( const results = await Promise.all(
files.map(async (filePath) => { files.map(async (filePath) => {
const removed = await removeBackground(filePath, threshold, fuzz); const { count, outFile } = await removeBackground(filePath, threshold, fuzz);
console.log(` ${basename(filePath)} -> ${removed} pixels made transparent`); const suffix = outFile !== filePath ? ` (saved as ${basename(outFile)})` : '';
return removed; console.log(` ${basename(filePath)} -> ${count} pixels made transparent${suffix}`);
return count;
}) })
); );