Method byte_array_to_image

StegaPy/util/image_util.py:15–108

将字节数组转换为PIL图像

(data, filename=None)

    
    @staticmethod
    def byte_array_to_image(data, filename=None):
        """将字节数组转换为PIL图像"""
        try:
            # 检查数据是否为空
            if data is None:
                raise ValueError("图像数据为空")
            
            # 统一转换为字节数组
            data_bytes = None
            if isinstance(data, io.BytesIO):
                # 对于 BytesIO 对象，先读取所有内容
                try:
                    current_pos = data.tell()
                    data.seek(0)  # 重置指针到开头
                    data_bytes = data.read()
                    # 恢复原位置（如果可能）
                    try:
                        data.seek(current_pos)
                    except:
                        pass
                except Exception as e:
                    raise ValueError(f"无法从BytesIO读取数据: {str(e)}")
            elif isinstance(data, (bytes, bytearray)):
                data_bytes = bytes(data)
            else:
                # 尝试转换为字节数组
                try:
                    data_bytes = bytes(data)
                except (TypeError, ValueError) as e:
                    raise ValueError(f"不支持的图像数据类型: {type(data).__name__}, 错误: {str(e)}")
            
            # 验证数据不为空
            if data_bytes is None or len(data_bytes) == 0:
                raise ValueError("图像数据为空")
            
            # 验证数据大小（至少要有文件头）
            if len(data_bytes) < 8:
                raise ValueError(f"图像数据太小（仅{len(data_bytes)}字节），不是有效的图像文件")
            
            # 验证图像文件头（检查常见格式）
            header = data_bytes[:8]
            is_valid_image = False
            
            # PNG: 89 50 4E 47 0D 0A 1A 0A
            if header.startswith(b'\x89PNG\r\n\x1a\n'):
                is_valid_image = True
            # JPEG: FF D8 FF
            elif header.startswith(b'\xff\xd8\xff'):
                is_valid_image = True
            # BMP: 42 4D
            elif header.startswith(b'BM'):
                is_valid_image = True
            # GIF: 47 49 46 38
            elif header.startswith(b'GIF8'):
                is_valid_image = True
            # TIFF: 49 49 2A 00 或 4D 4D 00 2A
            elif header.startswith(b'II*\x00') or header.startswith(b'MM\x00*'):
                is_valid_image = True