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

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'));