id-scanner-lib 1.0.0

package/src/id-recognition/data-extractor.ts

@@ -0,0 +1,165 @@
+/**
+ * @file 数据提取工具类
+ * @description 提供身份证信息的验证和格式化功能
+ * @module DataExtractor
+ */
+
+import { IDCardInfo } from '../utils/types';
+
+/**
+ * 数据提取工具类
+ * 
+ * 提供身份证信息的验证、提取和增强功能，可以从身份证号码中提取出生日期、性别等信息，
+ * 并对OCR识别结果进行补充和验证
+ * 
+ * @example
+ * ```typescript
+ * // 验证身份证号码
+ * const isValid = DataExtractor.validateIDNumber('110101199001011234');
+ * 
+ * // 从身份证号码提取出生日期
+ * const birthDate = DataExtractor.extractBirthDateFromID('110101199001011234');
+ * // 结果: '1990-01-01'
+ * 
+ * // 增强OCR识别结果
+ * const enhancedInfo = DataExtractor.enhanceIDCardInfo({
+ *   name: '张三',
+ *   idNumber: '110101199001011234'
+ * });
+ * // 结果会自动补充性别和出生日期
+ * ```
+ */
+export class DataExtractor {
+  /**
+   * 验证身份证号码格式
+   * 
+   * 检查身份证号码的长度、格式和出生日期部分是否有效
+   * 
+   * @param {string} idNumber - 要验证的身份证号码
+   * @returns {boolean} 是否是有效的身份证号码
+   */
+  static validateIDNumber(idNumber: string): boolean {
+    // 简单校验长度
+    if (!idNumber || idNumber.length !== 18) {
+      return false;
+    }
+    
+    // 校验格式 (前17位必须是数字，最后一位可以是数字或X)
+    const pattern = /^\d{17}[\dX]$/;
+    if (!pattern.test(idNumber)) {
+      return false;
+    }
+    
+    // 校验出生日期
+    const year = parseInt(idNumber.substr(6, 4));
+    const month = parseInt(idNumber.substr(10, 2));
+    const day = parseInt(idNumber.substr(12, 2));
+    
+    const date = new Date(year, month - 1, day);
+    if (
+      date.getFullYear() !== year || 
+      date.getMonth() + 1 !== month || 
+      date.getDate() !== day
+    ) {
+      return false;
+    }
+    
+    // 简单的校验规则，实际项目中可以加入更完善的验证
+    return true;
+  }
+  
+  /**
+   * 从身份证号码提取出生日期
+   * 
+   * @param {string} idNumber - 身份证号码
+   * @returns {string|null} 格式化的出生日期（YYYY-MM-DD），如果身份证号码无效则返回null
+   */
+  static extractBirthDateFromID(idNumber: string): string | null {
+    if (!this.validateIDNumber(idNumber)) {
+      return null;
+    }
+    
+    const year = idNumber.substr(6, 4);
+    const month = idNumber.substr(10, 2);
+    const day = idNumber.substr(12, 2);
+    
+    return `${year}-${month}-${day}`;
+  }
+  
+  /**
+   * 从身份证号码提取性别
+   * 
+   * 根据身份证号码第17位判断性别，奇数为男，偶数为女
+   * 
+   * @param {string} idNumber - 身份证号码
+   * @returns {string|null} '男'或'女'，如果身份证号码无效则返回null
+   */
+  static extractGenderFromID(idNumber: string): string | null {
+    if (!this.validateIDNumber(idNumber)) {
+      return null;
+    }
+    
+    // 第17位，奇数为男，偶数为女
+    const genderCode = parseInt(idNumber.charAt(16));
+    return genderCode % 2 === 1 ? '男' : '女';
+  }
+  
+  /**
+   * 从身份证号码提取地区编码
+   * 
+   * @param {string} idNumber - 身份证号码
+   * @returns {string|null} 地区编码（前6位），如果身份证号码无效则返回null
+   */
+  static extractRegionFromID(idNumber: string): string | null {
+    if (!this.validateIDNumber(idNumber)) {
+      return null;
+    }
+    
+    return idNumber.substr(0, 6);
+  }
+  
+  /**
+   * 合并并优化身份证信息
+   * 
+   * 使用多个来源的数据进行交叉验证和补充，如果OCR识别结果缺少某些信息，
+   * 但有身份证号码，则可以从号码中提取出生日期和性别等信息
+   * 
+   * @param {IDCardInfo} ocrInfo - OCR识别到的身份证信息
+   * @param {string} [idNumber] - 可选的外部提供的身份证号码，优先级高于OCR识别结果
+   * @returns {IDCardInfo} 增强后的身份证信息
+   */
+  static enhanceIDCardInfo(ocrInfo: IDCardInfo, idNumber?: string): IDCardInfo {
+    const result = { ...ocrInfo };
+    
+    // 如果OCR识别出身份证号，但没有识别出生日期或性别，则从身份证号码提取
+    if (result.idNumber && this.validateIDNumber(result.idNumber)) {
+      // 从身份证号提取出生日期
+      if (!result.birthDate) {
+        result.birthDate = this.extractBirthDateFromID(result.idNumber) || undefined;
+      }
+      
+      // 从身份证号提取性别
+      if (!result.gender) {
+        result.gender = this.extractGenderFromID(result.idNumber) || undefined;
+      }
+    }
+    
+    // 如果外部传入了身份证号，则优先使用它并提取信息
+    if (idNumber && this.validateIDNumber(idNumber)) {
+      result.idNumber = idNumber;
+      
+      // 使用身份证号码再次验证或补充信息
+      const birthDate = this.extractBirthDateFromID(idNumber);
+      if (birthDate) {
+        result.birthDate = birthDate;
+      }
+      
+      const gender = this.extractGenderFromID(idNumber);
+      if (gender) {
+        result.gender = gender;
+      }
+    }
+    
+    return result;
+  }
+} 

package/src/id-recognition/id-detector.ts

@@ -0,0 +1,171 @@
+/**
+ * @file 身份证检测模块
+ * @description 提供自动检测和定位图像中的身份证功能
+ * @module IDCardDetector
+ */
+
+import { Camera } from '../utils/camera';
+import { ImageProcessor } from '../utils/image-processing';
+import { DetectionResult } from '../utils/types';
+
+/**
+ * 身份证检测器类
+ * 
+ * 通过图像处理和计算机视觉技术，实时检测视频流中的身份证，并提取身份证区域
+ * 注意：当前实现是简化版，实际项目中建议使用OpenCV.js进行更精确的检测
+ * 
+ * @example
+ * ```typescript
+ * // 创建身份证检测器
+ * const detector = new IDCardDetector((result) => {
+ *   if (result.success && result.croppedImage) {
+ *     console.log('检测到身份证!');
+ *     // 对裁剪出的身份证图像进行处理
+ *     processIDCardImage(result.croppedImage);
+ *   }
+ * });
+ * 
+ * // 启动检测
+ * const videoElement = document.getElementById('video') as HTMLVideoElement;
+ * await detector.start(videoElement);
+ * 
+ * // 停止检测
+ * detector.stop();
+ * ```
+ */
+export class IDCardDetector {
+  private camera: Camera;
+  private detecting = false;
+  private detectTimer: number | null = null;
+  
+  /**
+   * 创建身份证检测器实例
+   * 
+   * @param {Function} [onDetected] - 身份证检测成功回调函数，接收检测结果对象
+   */
+  constructor(private onDetected?: (result: DetectionResult) => void) {
+    this.camera = new Camera();
+  }
+  
+  /**
+   * 启动身份证检测
+   * 
+   * 初始化相机并开始连续检测视频帧中的身份证
+   * 
+   * @param {HTMLVideoElement} videoElement - 用于显示相机画面的video元素
+   * @returns {Promise<void>} 启动完成的Promise
+   */
+  async start(videoElement: HTMLVideoElement): Promise<void> {
+    await this.camera.initialize(videoElement);
+    this.detecting = true;
+    this.detect();
+  }
+  
+  /**
+   * 执行一次身份证检测
+   * 
+   * 内部方法，捕获当前视频帧并尝试检测其中的身份证
+   * 
+   * @private
+   */
+  private async detect(): Promise<void> {
+    if (!this.detecting) return;
+    
+    const imageData = this.camera.captureFrame();
+    
+    if (imageData) {
+      try {
+        // 简单实现，因为没有完整的OpenCV.js
+        // 实际项目中应该使用OpenCV.js做更精确的边缘检测
+        const result = await this.detectIDCard(imageData);
+        
+        if (this.onDetected) {
+          this.onDetected(result);
+        }
+      } catch (error) {
+        console.error('身份证检测错误:', error);
+      }
+    }
+    
+    this.detectTimer = window.setTimeout(() => this.detect(), 200);
+  }
+  
+  /**
+   * 身份证检测核心算法
+   * 
+   * 通过图像处理技术检测和提取图像中的身份证区域
+   * 
+   * @private
+   * @param {ImageData} imageData - 需要检测身份证的图像数据
+   * @returns {Promise<DetectionResult>} 检测结果，包含成功标志和裁剪后的身份证图像
+   */
+  private async detectIDCard(imageData: ImageData): Promise<DetectionResult> {
+    // 图像预处理
+    const grayscale = ImageProcessor.toGrayscale(imageData);
+    const enhanced = ImageProcessor.adjustBrightnessContrast(grayscale, 10, 30);
+    
+    // 简化的身份证检测算法
+    // 在真实项目中，这里应该使用OpenCV.js进行轮廓检测和矩形检测
+    
+    // 模拟检测过程
+    const success = Math.random() > 0.7; // 模拟70%的概率检测成功
+    
+    if (success) {
+      // 模拟一个身份证区域，实际项目中应该是根据检测结果
+      const cardWidth = Math.floor(imageData.width * 0.8);
+      const cardHeight = Math.floor(cardWidth * 0.63); // 身份证比例大约是8:5
+      const x = Math.floor((imageData.width - cardWidth) / 2);
+      const y = Math.floor((imageData.height - cardHeight) / 2);
+      
+      // 模拟四个角点
+      const corners = [
+        { x, y }, // 左上
+        { x: x + cardWidth, y }, // 右上
+        { x: x + cardWidth, y: y + cardHeight }, // 右下
+        { x, y: y + cardHeight } // 左下
+      ];
+      
+      // 模拟裁剪图像（实际项目中应该做透视变换）
+      const canvas = document.createElement('canvas');
+      canvas.width = cardWidth;
+      canvas.height = cardHeight;
+      const ctx = canvas.getContext('2d');
+      
+      if (ctx) {
+        // 从原图中裁剪身份证区域
+        const sourceCanvas = ImageProcessor.imageDataToCanvas(imageData);
+        ctx.drawImage(
+          sourceCanvas, 
+          x, y, cardWidth, cardHeight,
+          0, 0, cardWidth, cardHeight
+        );
+        
+        const croppedImage = ctx.getImageData(0, 0, cardWidth, cardHeight);
+        
+        return {
+          success: true,
+          corners,
+          croppedImage
+        };
+      }
+    }
+    
+    return { success: false };
+  }
+  
+  /**
+   * 停止身份证检测
+   * 
+   * 停止检测循环并释放相机资源
+   */
+  stop(): void {
+    this.detecting = false;
+    
+    if (this.detectTimer) {
+      clearTimeout(this.detectTimer);
+      this.detectTimer = null;
+    }
+    
+    this.camera.release();
+  }
+} 

package/src/id-recognition/ocr-processor.ts

@@ -0,0 +1,169 @@
+/**
+ * @file OCR处理模块
+ * @description 提供身份证文字识别和信息提取功能
+ * @module OCRProcessor
+ */
+
+import { createWorker } from 'tesseract.js';
+import { IDCardInfo } from '../utils/types';
+import { ImageProcessor } from '../utils/image-processing';
+
+/**
+ * OCR处理器类
+ * 
+ * 使用Tesseract.js实现对身份证图像的OCR文字识别和信息提取功能
+ * 
+ * @example
+ * ```typescript
+ * // 创建OCR处理器
+ * const ocrProcessor = new OCRProcessor();
+ * 
+ * // 初始化OCR引擎
+ * await ocrProcessor.initialize();
+ * 
+ * // 处理身份证图像
+ * const idInfo = await ocrProcessor.processIDCard(idCardImageData);
+ * console.log('识别到的身份证信息:', idInfo);
+ * 
+ * // 使用结束后释放资源
+ * await ocrProcessor.terminate();
+ * ```
+ */
+export class OCRProcessor {
+  private worker: any = null;
+  
+  constructor() {}
+  
+  /**
+   * 初始化OCR引擎
+   * 
+   * 加载Tesseract OCR引擎和中文简体语言包，并设置适合身份证识别的参数
+   * 
+   * @returns {Promise<void>} 初始化完成的Promise
+   */
+  async initialize(): Promise<void> {
+    this.worker = createWorker({
+      logger: (m: any) => console.log(m)
+    } as any);
+    
+    await this.worker.load();
+    await this.worker.loadLanguage('chi_sim');
+    await this.worker.initialize('chi_sim');
+    await this.worker.setParameters({
+      tessedit_char_whitelist: '0123456789X-年月日一二三四五六七八九十零壹贰叁肆伍陆柒捌玖拾ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz民族汉族满族回族维吾尔族藏族苗族彝族壮族朝鲜族侗族瑶族白族土家族哈尼族哈萨克族傣族黎族傈僳族佤族高山族拉祜族水族东乡族钠西族景颇族柯尔克孜族士族达斡尔族仫佬族羌族布朗族撒拉族毛南族仡佬族锡伯族阿昌族普米族塔吉克族怒族乌孜别克族俄罗斯族鄂温克族德昂族保安族裕固族京族塔塔尔族独龙族鄂伦春族赫哲族门巴族珞巴族基诺族男女性别住址出生公民身份号码签发机关有效期'
+    });
+  }
+  
+  /**
+   * 处理身份证图像并提取信息
+   * 
+   * 对身份证图像进行OCR识别，并从识别结果中提取结构化信息
+   * 
+   * @param {ImageData} imageData - 身份证图像数据
+   * @returns {Promise<IDCardInfo>} 提取到的身份证信息
+   */
+  async processIDCard(imageData: ImageData): Promise<IDCardInfo> {
+    if (!this.worker) {
+      await this.initialize();
+    }
+    
+    // 图像预处理，提高OCR识别率
+    const enhancedImage = ImageProcessor.adjustBrightnessContrast(imageData, 15, 25);
+    
+    // 转换ImageData为Canvas
+    const canvas = ImageProcessor.imageDataToCanvas(enhancedImage);
+    
+    // OCR识别
+    try {
+      const { data } = await this.worker.recognize(canvas);
+      
+      // 解析身份证信息
+      return this.parseIDCardText(data.text);
+    } catch (error) {
+      console.error('OCR识别错误:', error);
+      return {}; // 返回空对象
+    }
+  }
+  
+  /**
+   * 解析身份证文本信息
+   * 
+   * 从OCR识别到的文本中提取结构化的身份证信息
+   * 
+   * @private
+   * @param {string} text - OCR识别到的文本
+   * @returns {IDCardInfo} 提取到的身份证信息对象
+   */
+  private parseIDCardText(text: string): IDCardInfo {
+    const info: IDCardInfo = {};
+    
+    // 拆分为行
+    const lines = text.split('\n').filter(line => line.trim());
+    
+    // 解析身份证号码（最容易识别的部分）
+    const idNumberRegex = /(\d{17}[\dX])/;
+    const idNumberMatch = text.match(idNumberRegex);
+    if (idNumberMatch) {
+      info.idNumber = idNumberMatch[1];
+    }
+    
+    // 解析姓名
+    for (const line of lines) {
+      if (line.includes('姓名') || line.length < 10 && line.length > 1 && !/\d/.test(line)) {
+        info.name = line.replace('姓名', '').trim();
+        break;
+      }
+    }
+    
+    // 解析性别和民族
+    const genderNationalityRegex = /(男|女).*(族)/;
+    const genderMatch = text.match(genderNationalityRegex);
+    if (genderMatch) {
+      info.gender = genderMatch[1];
+      const nationalityText = genderMatch[0];
+      info.nationality = nationalityText.substring(nationalityText.indexOf(genderMatch[1]) + 1).trim();
+    }
+    
+    // 解析出生日期
+    const birthDateRegex = /(\d{4})年(\d{1,2})月(\d{1,2})日/;
+    const birthDateMatch = text.match(birthDateRegex);
+    if (birthDateMatch) {
+      info.birthDate = `${birthDateMatch[1]}-${birthDateMatch[2]}-${birthDateMatch[3]}`;
+    }
+    
+    // 解析地址
+    const addressRegex = /住址([\s\S]*?)公民身份号码/;
+    const addressMatch = text.match(addressRegex);
+    if (addressMatch) {
+      info.address = addressMatch[1].replace(/\n/g, '').trim();
+    }
+    
+    // 解析签发机关
+    const authorityRegex = /签发机关([\s\S]*?)有效期/;
+    const authorityMatch = text.match(authorityRegex);
+    if (authorityMatch) {
+      info.issuingAuthority = authorityMatch[1].replace(/\n/g, '').trim();
+    }
+    
+    // 解析有效期限
+    const validPeriodRegex = /有效期限([\s\S]*?)(-|至)/;
+    const validPeriodMatch = text.match(validPeriodRegex);
+    if (validPeriodMatch) {
+      info.validPeriod = validPeriodMatch[0].replace('有效期限', '').trim();
+    }
+    
+    return info;
+  }
+  
+  /**
+   * 终止OCR引擎并释放资源
+   * 
+   * @returns {Promise<void>} 终止完成的Promise
+   */
+  async terminate(): Promise<void> {
+    if (this.worker) {
+      await this.worker.terminate();
+      this.worker = null;
+    }
+  }
+}