nkkj
/
yss-project


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
							const fs = require('fs');
const path = require('path');
const { createCanvas, loadImage } = require('canvas');

// 配置
const CONFIG = {
  pixelSize: 100, // 马赛克块大小
  inputImage: path.join(__dirname, '参考图.png'),
  outputMosaic: path.join(__dirname, 'mosaic.png'),
  outputReport: path.join(__dirname, 'report.html')
};

// 主函数
async function main() {
  try {
    console.log('开始处理图片...');
    
    // 读取原始图片
    const image = await loadImage(CONFIG.inputImage);
    const width = image.width;
    const height = image.height;
    
    // 创建马赛克图片
    console.log('创建马赛克图片...');
    const { mosaicCanvas, colorData } = createMosaic(image, width, height, CONFIG.pixelSize);
    
    // 保存马赛克图片
    const mosaicBuffer = mosaicCanvas.toBuffer('image/png');
    fs.writeFileSync(CONFIG.outputMosaic, mosaicBuffer);
    
    // 生成报告
    console.log('生成HTML报告...');
    generateReport(colorData);
    
    console.log('处理完成! 报告已生成: ' + CONFIG.outputReport);
  } catch (error) {
    console.error('处理过程中出错:', error);
  }
}

// 创建马赛克图片并提取颜色
function createMosaic(image, width, height, pixelSize) {
  // 创建画布
  const canvas = createCanvas(width, height);
  const ctx = canvas.getContext('2d');
  
  // 绘制原始图片
  ctx.drawImage(image, 0, 0, width, height);
  
  // 颜色映射: 颜色值 -> 出现次数
  const colorMap = new Map();
  
  // 遍历图像，按块处理
  for (let y = 0; y < height; y += pixelSize) {
    for (let x = 0; x < width; x += pixelSize) {
      // 计算当前块的大小
      const blockWidth = Math.min(pixelSize, width - x);
      const blockHeight = Math.min(pixelSize, height - y);
      
      // 获取块的像素数据
      const imageData = ctx.getImageData(x, y, blockWidth, blockHeight);
      const data = imageData.data;
      
      // 计算平均颜色
      let r = 0, g = 0, b = 0;
      const pixelCount = blockWidth * blockHeight;
      
      for (let i = 0; i < data.length; i += 4) {
        r += data[i];
        g += data[i + 1];
        b += data[i + 2];
      }
      
      r = Math.floor(r / pixelCount);
      g = Math.floor(g / pixelCount);
      b = Math.floor(b / pixelCount);
      
      // 创建颜色键
      const colorKey = `rgb(${r},${g},${b})`;
      
      // 更新颜色映射
      if (colorMap.has(colorKey)) {
        colorMap.set(colorKey, colorMap.get(colorKey) + 1);
      } else {
        colorMap.set(colorKey, 1);
      }
      
      // 填充块
      ctx.fillStyle = colorKey;
      ctx.fillRect(x, y, blockWidth, blockHeight);
    }
  }
  
  // 转换为数组并排序
  const sortedColors = Array.from(colorMap.entries())
    .map(([color, count]) => ({ color, count }))
    .sort((a, b) => b.count - a.count);
  
  return { mosaicCanvas: canvas, colorData: sortedColors };
}

// 生成HTML报告
function generateReport(colorData) {
  // 创建HTML内容
  const htmlContent = `
<!DOCTYPE html>
<html lang="zh-CN">
<head>
  <meta charset="UTF-8">
  <meta name="viewport" content="width=device-width, initial-scale=1.0">
  <title>图片颜色分析报告</title>
  <style>
    body {
      font-family: 'Arial', sans-serif;
      line-height: 1.6;
      color: #333;
      max-width: 1200px;
      margin: 0 auto;
      padding: 20px;
    }
    h1, h2 {
      color: #2c3e50;
    }
    .image-container {
      display: flex;
      justify-content: space-between;
      margin-bottom: 30px;
      flex-wrap: wrap;
    }
    .image-box {
      width: 48%;
      margin-bottom: 20px;
    }
    .image-box img {
      max-width: 100%;
      height: auto;
      border: 1px solid #ddd;
    }
    .color-cards {
      display: flex;
      flex-wrap: wrap;
      gap: 10px;
      margin-top: 20px;
    }
    .color-card {
      width: 100px;
      height: 100px;
      display: flex;
      flex-direction: column;
      align-items: center;
      justify-content: center;
      border-radius: 5px;
      color: white;
      text-shadow: 0 0 2px #000;
      font-size: 12px;
      position: relative;
    }
    .color-info {
      background: rgba(0,0,0,0.7);
      padding: 3px 6px;
      border-radius: 3px;
    }
    .color-distribution {
      margin-top: 30px;
    }
    .color-bar {
      height: 30px;
      margin: 5px 0;
      display: flex;
    }
    .color-bar-segment {
      height: 100%;
      display: flex;
      align-items: center;
      justify-content: center;
      color: white;
      text-shadow: 0 0 2px #000;
      font-size: 12px;
    }
  </style>
</head>
<body>
  <h1>图片颜色分析报告</h1>
  
  <div class="image-container">
    <div class="image-box">
      <h2>原始图片</h2>
      <img src="参考图.png" alt="原始图片">
    </div>
    <div class="image-box">
      <h2>马赛克图片</h2>
      <img src="mosaic.png" alt="马赛克图片">
    </div>
  </div>
  
  <h2>颜色分布</h2>
  <p>从马赛克图中提取的主要颜色（按出现频率排序）：</p>
  
  <div class="color-cards">
    ${colorData.slice(0, 20).map((item, index) => `
      <div class="color-card" style="background-color: ${item.color}">
        <div class="color-info">
          ${item.color}<br>
          频率: ${((item.count / colorData.reduce((sum, c) => sum + c.count, 0)) * 100).toFixed(1)}%
        </div>
      </div>
    `).join('')}
  </div>
  
  <div class="color-distribution">
    <h2>颜色频率分布图</h2>
    <div class="color-bar">
      ${colorData.slice(0, 10).map(item => {
        const percentage = (item.count / colorData.reduce((sum, c) => sum + c.count, 0)) * 100;
        return `<div class="color-bar-segment" style="background-color: ${item.color}; width: ${percentage}%;">${percentage.toFixed(1)}%</div>`;
      }).join('')}
    </div>
  </div>
</body>
</html>
  `;
  
  // 写入HTML文件
  fs.writeFileSync(CONFIG.outputReport, htmlContent);
}

// 运行主函数
main();