如何使用Node.js判断png图片是否存在透明像素

背景

png格式的图片存储空间会比jpg格式的图片大，但是png图片的质量明显更好。有时候并不需要图片的质量非常的好，但是为了减少包体，需要做一些优化，比如压缩图片，把没有带透明像素的png图片转化成jpg格式的图片。这片文章主要来讲讲怎么利用Node.js来检测没有带透明像素的png图片，以及如何把它转化成jpg图片。

代码

直接上代码

import canvas from 'canvas';
import fs from 'fs';

/**
 * 判断png图片是否存在透明像素
 *
 * @param {*} pngPath png图片路径
 * @param {number} [limit=255] 透明像素点限度，默认小与255视为透明像素
 * @param {boolean} [isToJpg=false] 如果没有透明像素是否转化为jpg图片
 * @returns
 */
function hasOpacity(pngPath, limit = 255, isToJpg = false) {
  // 获取图片的buffer
  const buffer = fs.readFileSync(pngPath);

  // 图片的宽度存在buffer的第17到20个字节
  const width = buffer.readUInt32BE(16);
  // 图片的宽度存在buffer的第21到24个字节
  const height = buffer.readUInt32BE(20);

  // 创建一个画布
  const pngCanvas = canvas.createCanvas(width, height);
  // 拿到画笔
  const context = pngCanvas.getContext('2d');
  // 创建一张图片
  const img = new canvas.Image();
  // 记载图片
  img.src = buffer;
  // 使用画笔把图片画在画布上
  context.drawImage(img, 0, 0, width, height);

  // 获取png图片的数据的像素数据
  let res = context.getImageData(0, 0, width, height);
  let imgData = res.data;
  // 每个像素占用4个字节，计算出一共有多少像素
  // [r, g, b, a]
  let piexCount = imgData.length / 4;

  // 是否已经找到透明像素
  let isOpacity = false;

  for (let i = 0; i < piexCount; i++) {
    // 遍历每个像素点，找透明通道
    let opacity = imgData[i * 4 + 3];
    if (opacity < limit) {
        // 如果小于limit，则存在透明像素，退出
      isOpacity = true;
      break;
    }
  }

  // 如果不存在透明像素且isToJpg为true，则转化为jpg格式的图拍呢
  if (!isOpacity && isToJpg) {
    const out = fs.createWriteStream(pngPath.split('.')[0] + '.jpg');
    pngCanvas.createJPEGStream().pipe(out);
    out.on('finish', () => console.log(pngPath, '转成jpg成功'));
  }

  // 返回
  return isOpacity;
}

console.log(hasOpacity('hh.png', 254, true));
console.log(hasOpacity('jj.png'));

原理：

canvas采用四个字节存放像素，[r, g, b, a]，分别代表红色通道，绿色通道，蓝色通道，透明通道。每个字节是8位，所以每个通道的数据是0～255之间，就透明通道而言，255表示完全不透明，0表示完全透明。我们利用Node.js的buffer和canvas把png图片转化成一个个像素点的数据，然后通过遍历每个透明通道，就可以找到png图片是否带有透明像素。

疑问？为什么有个limit参数？

这是因为在实际应用中，有可能存在少量254、253的透明通过，这些看起来就跟完全不透明的像素一样，所以把他们也当作不是透明像素处理。这个根据自己能够接受的限度去传参。