数据压缩学习实验（三）PNG文件读取及转换至YUV色彩空间的C++实现

最新推荐文章于 2024-05-31 14:42:05 发布
最新推荐文章于 2024-05-31 14:42:05 发布
阅读量690 收藏 5
点赞数 1
分类专栏： 本科数据压缩课程 文章标签： c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/AaronXueNF/article/details/105443903
版权

实验目的

  1. 学习PNG文件格式，为随后学习图像的压缩做准备。
  2. 编程读取PNG文件中的外围解析，使用lodepng库解码被压缩的图像内容。
  3. 将文件内容转换至YUV色彩空间。

实验原理

  1. PNG文件格式初步了解：数据压缩学习笔记（二）以PNG为例探讨设计思想和理解
  2. YUV色彩空间的转换：数据压缩学习实验（一）RGB与YUV色彩空间转换的C++实现及误差分析
  3. 使用lodepng库对压缩图像进行解码：LodePNG
  4. 使用easyx库绘制解码后的图像：EasyX

实验结果

实验素材：
在这里插入图片描述
文件信息读取结果：
在这里插入图片描述
使用easyx库绘制的图像：
在这里插入图片描述
转换后YUV图像播放结果：
在这里插入图片描述

总结与误差分析

  1. 由于该程序执行的仍然是RGB色彩空间向YUV 4:2:0的转换，因此转换部分误差情况与之前RGB转YUV实验误差情况相同。
  2. 通过VScode软件查看二进制发现在PNG文件中，各项数据使用大端排序（big-endian），即高位字节在低地址处，这与之前学习的BMP文件使用的小端排序不同，因此读取方法需要做相应修改。

程序代码

YUV文件代码 “YUV_RAW.h” , “YUW_RAW.cpp” 见数据压缩学习实验（一）RGB与YUV色彩空间转换的C++实现及误差分析

converter.h

#ifndef tr
#define tr
#include "YUV_raw.h"
#include <vector>
using namespace std;

void init_LookupTable();
uint1 clamp_Y(int t);
uint1 clamp_UV(int t);
uint1 clamp_RGB(int t);
bool png2yuv(std::vector<unsigned char> data, uint4 w, uint4 h, YUV_raw_image& yuv, uint4 frame = 0);

#endif

converter.cpp

#include "converter.h"
#include <iostream>
#include <fstream>
using namespace std;

static uint1 RGBYUV02290[256];
static uint1 RGBYUV05870[256];
static uint1 RGBYUV01140[256];
static uint1 RGBYUV01684[256];
static uint1 RGBYUV03316[256];
static uint1 RGBYUV05000[256];
static uint1 RGBYUV04187[256];
static uint1 RGBYUV00813[256];
static int YUVRGB14020[256];
static int YUVRGB07141[256];
static int YUVRGB03144[256];
static int YUVRGB17720[256];
static bool init_flg = false;

void init_LookupTable()
{
	if (init_flg == false) {
		for (int i = 0; i < 256; i++) {
			RGBYUV02290[i] = (float)i * 0.299;
			RGBYUV05870[i] = (float)i * 0.587;
			RGBYUV01140[i] = (float)i * 0.114;
			RGBYUV01684[i] = (float)i * 0.1684;
			RGBYUV03316[i] = (float)i * 0.3316;
			RGBYUV05000[i] = (float)i * 0.5;
			RGBYUV04187[i] = (float)i * 0.4187;
			RGBYUV00813[i] = (float)i * 0.0813;
			YUVRGB14020[i] = (float)(i - 128) * 1.402;
			YUVRGB07141[i] = (float)(i - 128) * 0.71414;
			YUVRGB03144[i] = (float)(i - 128) * 0.314414;
			YUVRGB17720[i] = (float)(i - 128) * 1.772;
		}
		init_flg = true;
	}
}

uint1 clamp_Y(int t)
{
	if (t < 16) {
		return 16;
	}
	else if (t > 235) {
		return 235;
	}
	else {
		return t;
	}
}

uint1 clamp_UV(int t)
{
	if (t < 16) {
		return 16;
	}
	else if (t > 240) {
		return 240;
	}
	else {
		return t;
	}
}

uint1 clamp_RGB(int t)
{
	if (t < 0) {
		return 0;
	}
	else if (t > 255) {
		return 255;
	}
	else {
		return t;
	}
}

bool png2yuv(std::vector<unsigned char> data, uint4 w, uint4 h, YUV_raw_image& yuv, uint4 frame)
{
	if (init_flg == false) {
		init_LookupTable();
	}
	int width = w;
	int height = h;
	chrominace_mode mode = yuv.get_mode();
	if (yuv.get_width() != width || yuv.get_height() != height) {
		cerr << __func__ << "\t The size of yuv file doesn\t match!" << endl;
		return false;
	}
	int y, u, v;
	int r, g, b;
	YUV* yuv_data = yuv.get_data();
	/*loop to traverse all pixels*/
	int loop_y(0), loop_uv(0);
	for (int loop_i = 0; loop_i < height; loop_i++) {
		for (int loop_j = 0; loop_j < width; loop_j++) {
			r = data[4 * loop_i * w + 4 * loop_j + 0];
			g = data[4 * loop_i * w + 4 * loop_j + 1];
			b = data[4 * loop_i * w + 4 * loop_j + 2];
			y = RGBYUV02290[r] + RGBYUV05870[g] + RGBYUV01140[b];
			u = RGBYUV05000[b] - RGBYUV01684[r] - RGBYUV03316[g] + 128;
			v = RGBYUV05000[r] - RGBYUV04187[g] - RGBYUV00813[b] + 128;
			y = clamp_Y(y);
			u = clamp_UV(u);
			v = clamp_UV(v);
			switch (mode)
			{
			case(C444): {
				*(yuv_data->Y_data + loop_y) = y;
				*(yuv_data->U_data + loop_y) = u;
				*(yuv_data->V_data + loop_y) = v;
			}break;
			case(C422): {
				*(yuv_data->Y_data + loop_y) = y;
				if (loop_y % 2 == 0) {
					*(yuv_data->U_data + loop_uv) = u;
					*(yuv_data->V_data + loop_uv) = v;
					loop_uv++;
				}
			}break;
			case(C420): {
				*(yuv_data->Y_data + loop_y) = y;
				if ((loop_y % 2 == 0) && (((loop_y - (loop_y % width)) / width) % 2 == 0)) {
					/*this condition means the position of chrominace sample*/
					*(yuv_data->U_data + loop_uv) = u;
					*(yuv_data->V_data + loop_uv) = v;
					loop_uv++;
				}
			}break;
			case(C411): {
				*(yuv_data->Y_data + loop_y) = y;
				if (loop_y % 4 == 0) {
					*(yuv_data->U_data + loop_uv) = u;
					*(yuv_data->V_data + loop_uv) = v;
					loop_uv++;
				}
			}break;
			}
			loop_y++;  //Important!!!!!!!!!!!!!!!!
		}
	}
	return true;
}

main.cpp

#include "lodepng.h"  // using lodepng to decode
#include "YUV_raw.h"
#include "converter.h"
#include <easyx.h>  // using easyx to show image
#include <conio.h>  // using easyx to show image
#include <math.h>
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;

typedef unsigned char uint1;
typedef unsigned short uint2;
typedef unsigned int uint4;
typedef int int4;

struct IHDR {
	uint4 length;
	uint4 type_code;
	uint4 width;
	uint4 height;
	uint1 depth;
	uint1 color_type;
	uint1 compression;
	uint1 filter;
	uint1 interlace;
};

uint4 B2D(uint1 input[4]);
bool find_IHDR(ifstream& file_in, IHDR& ihdr);
bool find_IDAT(ifstream& file_in);
bool find_IEND(ifstream& file_in);
void display(std::vector<unsigned char>data, uint4 w, uint4 h);

int main(void)
{
	//load file
	const char* path = "./down.png";
	std::ifstream file_in(path,ios::binary);

	//data pf png image
	IHDR ihdr;
	std::vector<unsigned char> data;

	//seek hdrl chunk and get information
	if (find_IHDR(file_in, ihdr))
	{
		cout << "Find IHDR chunk, details: " << endl;
		cout << "width:" << (int)ihdr.width << endl;
		cout << "height:" << (int)ihdr.height << endl;
		cout << "depth:" << (int)ihdr.depth << endl;
		cout << "color type:" << (int)ihdr.color_type << endl;
		cout << "compression method:" << (int)ihdr.compression << endl;
		cout << "filter method:" << (int)ihdr.filter << endl;
		cout << "interlace method:" << (int)ihdr.interlace << endl << endl;
	}
	else {
		cout << "Can't find IHDR chunk!" << endl;
		exit(1);
	}
	if (find_IDAT(file_in)) {
		cout << "Find IDAT chunk!" << endl;
	}
	else {
		cout << "Can't find IDAT chunk!" << endl;
	}

	if (find_IEND(file_in)) {
		cout << "Find IEND chunk!" << endl;
		cout << "Starting decode..." << endl;
		unsigned w(ihdr.width), h(ihdr.height);
		char err = lodepng::decode(data, w, h, path);
		if (err) {
			cout << "Can not decode png file, error code: " << err << endl;
		}
		else {
			cout << "Decoding finished!" << endl;
		}
	}
	else {
		cout << "Can't find IEND chunk!" << endl;
	}

	//close png file
	file_in.close();
	
	//conver to 4:2:0 YUV file
	YUV_raw_image yuv(ihdr.width, ihdr.height, 1, C420);
	png2yuv(data, ihdr.width, ihdr.height, yuv);
	yuv.save("./result.yuv");
	
	//show image
	display(data, ihdr.width, ihdr.height);
	system("Pause");
	closegraph();
	return 0;
}

uint4 B2D(uint1 input[4])
{
	/* Bytes to DWORD*/
	uint4 value = 0;
	for (int i = 0; i < 4; i++) {
		value += input[i] * (pow(256, 3 - i));
	}
	return value;
}

bool find_IHDR(ifstream& file_in, IHDR& ihdr)
{
	istream::pos_type start_pos = file_in.tellg();
	file_in.seekg(0, ios_base::end);
	istream::pos_type end_pos = file_in.tellg();
	file_in.seekg(start_pos);
	uint4 size = end_pos - start_pos - 4;

	for (int i = 0; i < size ; i++) {
		uint1 temp[4];
		file_in.read(reinterpret_cast<char*>(temp), 4);
		file_in.seekg(-3, ios::cur);
		if (B2D(temp) == 0x49484452)
		{
			file_in.seekg(-5, ios::cur);
			break;
		}
		if (i == size - 1) {
			return false;
		}
	}

	uint1 temp4[4];
	uint1 temp1[1];
	file_in.read(reinterpret_cast<char*>(temp4), 4);
	ihdr.length = B2D(temp4);
	file_in.read(reinterpret_cast<char*>(temp4), 4);
	ihdr.type_code = B2D(temp4);
	file_in.read(reinterpret_cast<char*>(temp4), 4);
	ihdr.width = B2D(temp4);
	file_in.read(reinterpret_cast<char*>(temp4), 4);
	ihdr.height = B2D(temp4);
	file_in.read(reinterpret_cast<char*>(temp1), 1);
	ihdr.depth = *temp1;
	file_in.read(reinterpret_cast<char*>(temp1), 1);
	ihdr.color_type = *temp1;
	file_in.read(reinterpret_cast<char*>(temp1), 1);
	ihdr.compression = *temp1;
	file_in.read(reinterpret_cast<char*>(temp1), 1);
	ihdr.filter = *temp1;
	file_in.read(reinterpret_cast<char*>(temp1), 1);
	ihdr.interlace = *temp1;
	file_in.seekg(start_pos);
	return true;
}

bool find_IDAT(ifstream& file_in)
{
	istream::pos_type start_pos = file_in.tellg();
	file_in.seekg(0, ios_base::end);
	istream::pos_type end_pos = file_in.tellg();
	file_in.seekg(start_pos);
	uint4 size = end_pos - start_pos - 4;

	for (int i = 0; i < size; i++) {
		uint1 temp[4];
		file_in.read(reinterpret_cast<char*>(temp), 4);
		file_in.seekg(-3, ios::cur);
		if (B2D(temp) == 0x49444154)
		{
			file_in.seekg(-5, ios::cur);
			break;
		}
		if (i == size - 1) {
			return false;
		}
	}
	file_in.seekg(start_pos);
	return true;
}

bool find_IEND(ifstream& file_in)
{
	istream::pos_type start_pos = file_in.tellg();
	file_in.seekg(0, ios_base::end);
	istream::pos_type end_pos = file_in.tellg();
	file_in.seekg(start_pos);
	uint4 size = end_pos - start_pos - 4;

	for (int i = 0; i < size; i++) {
		uint1 temp[4];
		file_in.read(reinterpret_cast<char*>(temp), 4);
		file_in.seekg(-3, ios::cur);
		if (B2D(temp) == 0x49454e44)
		{
			file_in.seekg(-5, ios::cur);
			break;
		}
		if (i == size - 1) {
			return false;
		}
	}
	file_in.seekg(start_pos);
	return true;
}

void display(std::vector<unsigned char> data, uint4 w, uint4 h)
{
	initgraph(w, h, SHOWCONSOLE);
	int r, g, b;
	for (int i = 0; i < h; i++) {
		for (int j = 0; j < w; j++) {
			r = data[4 * i * w + 4 * j + 0];
			g = data[4 * i * w + 4 * j + 1];
			b = data[4 * i * w + 4 * j + 2];
			putpixel(j, i, RGB(r, g, b));
		}
	}
}
AaronXueNF
关注
  • 1
    点赞
  • 踩
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
java png生成webp图片_jpg、png格式的图片转换成webp后颜色失真的问题
weixin_39602737的博客
02-28 740
今天简单的试用了一下 cweb.exe 将 jpg, png 格式的图片转换成 webp 格式。我今天下载的是当前最新版：1.0.0cwebp 3.jpg -q 85 -o 3.webp发现图片中的女模特原来粉红色的指甲颜色有失真的情况：变暗了。这个情况与使用 ffmpeg 将多张jpg图片生成 mp4 视频的时候出现的颜色失真有点类似。jpg格式：转换成webp格式后：百度了一下 webp 颜...
Jpg图片和Bmp图片转为YUV数据
04-21
可以将BMP文件和JPG文件转换为YUV数据，可以将YUV数据写成文件，也可以直接在内存中进行操作
抓屏转换成PNG，BMP与PNG互转 C++源码
10-23
抓屏，保存成高质量的PNG图片，完整工程，VC++2008编译通过，测试没有问题
YUV文件读取.rar_C++读取YUV_YUV_yuv；bmp_图片YUV_读取yuv
09-23
很好的一个操作yuv的例子，包含从yuv文件读取数据，复制数据，显示图片等等内容。
将png格式图片转换成YUV格式，C++
09-11
能将png格式图片转换成YUV格式。 C++实现，并且自带可运行实例。
PNG文件转YUV格式并实现播放
weixin_45671402的博客
04-08 1514
文章目录PNG文件转YUVPNG文件格式YUV文件格式什么是YUV格式文件？YUV采样YUV 4:2:0RGB转YUV实验过程获取PNG文件信息获取IHDR信息获取PLTE调色板信息获取IDAT信息对IDAT数据块进行解压缩将解压缩的IDAT数据进行转换多个YUV合成 PNG文件转YUV 前言：本次实验选用了PNG转YUV，做的过程十分煎熬…主要原因是PNG的IDAT数据块的解码部分没有自己手写轮子，直接调用的Python的zlib库，所以每个像素解压缩出来的数据分布完全没有概念，而且第一次选的图片还是带α
PNG文件转为YUV文件的C实现
pzp49666的博客
04-12 761
代码步骤核心思路具体实现首部处理数据块拆解子函数调用数据块分析IHDR文件头数据块 核心思路 核心思路是找到图像数据部分，根据调色板，找到数据部分后解码，处理为rgb后利用先前的rgb2yuv处理。 具体实现 首部处理 png文件开头8个字节为固定字节。目的是区分于其他文件，表明本文件是，其中第一个字节0x89超出了ASCII字符的范围，这是为了避免某些软件将PNG文件当做文本文件来处理。 ...
PNG转YUV(自造轮子版)
Cross_Entropy的博客
04-07 1077
文章目录前置知识PNG格式zlib文件格式deflate文件格式LZ77压缩算法霍夫曼编码字典树综述过程获取文件信息获取调色板信息对IDAT数据块进行解码编写位处理工具解码zlib和Defalte格式的文件头构建霍夫曼树获取霍夫曼码表进行解码格式转换实验结果完整代码 前置知识 PNG格式 zlib文件格式 deflate文件格式 LZ77压缩算法 霍夫曼编码 字典树 综...
FFmpeg批处理PNG转YUV
一直在笑的博客
04-21 1352
编写.bat文件，文件内容： :PNG to YUV for /L %%a in (1,1,5) do ( ffmpeg -i ParkScene_1920x1080_240_%%a.png -s 1920x1080 -pix_fmt yuv420p ParkScene_1920x1080_240_%%a.yuv) cmd
（一个失败的程序）C++实现将一张彩色bmp格式图片转换成yuv4:2:0格式
Tracytang97的博客
07-14 568
声明：以下内容均属于自己理解，不保证正确性 一、bmp图片的获取 从网上找到一张任意格式的图片，再用Windows系统自带的画图软件打开（在命令行输入mspaint即可快速打开画图），将图片保存成bmp24位图格式即可。注意在本程序中bmp图片的行数和列数都必须为偶数，否则可能会导致最终生成的Cb和Cr分量的个数不一样。 二、流程 代码过程大致分为以下几步： 1.以只读文件...
matlab实现读取YUV视频文件 程序源码.zip
02-17
【达摩老生出品，必属精品，亲测校正，质量保证】 ...源码说明： 基于matlab读取YUV视频文件，并提取视频的每一帧数据，包含完整源码和注释，非常适合借鉴学习 适合人群：新手及有一定经验的开发人员
YUV.rar_C 读取YUV文件_YUV数据读取_c yuv_output yuv file_读取YUV文件
09-24
这个程序用来读取yuv文件，并把数据输出到一个文本文件，代码很短，里面说明很详细，这对理解和使用yuv文件很方便
基于C++的RGB与YUV之间的相互转换设计与实现
05-12
基于C++的RGB与YUV之间的相互转换设计与实现
c++之yuv格式处理
qq_33619359的博客
02-07 670
一些奇怪的原因，要处理yuv格式的数据，代码在下面，自己看注释吧。
openCV实现图像的YUV转换打印输出矩阵类型的txt保存C++代码
dongbao520的博客
05-25 442
代码如下： int main() //hsv++yuv++rgb show { Mat img_h, img_s, img_v, imghsv; Mat image; string filepath01 = "E:\\xunleixx11\\pictures\\cio07.jpg"; //原图 string savepath001 = "E:\\xunleixx11\\txtxtxt\\txt525\\save_data001.txt";//路径1 string savepath002 = "
png文件转为yuv文件
Cross_Entropy的博客
03-31 1999
前言 由于png文件解码较为复杂，所以本次仅针对部分形式的图像，鲁棒性较差，仅作实验使用。 关于本次实验使用的文件： 实验过程 获取文件信息 #include <iostream> #include <fstream> #include <vector> #include <map> #include <set> using names...
【数据压缩（五）】基于C++实现BMP序列转YUV文件
zyzcuczyu的博客
03-30 404
多个BMP图片转YUV文件
使用ffmpeg进行图像格式转换以及图像缩放/sws_scale/linux/c++/c/rgb-yuv420
niansiqiao的专栏
12-15 1417
转自：http://blog.csdn.net/skys_broyal/article/details/10337147  利用ffmpeg进行图像数据格式的转换以及图片的缩放应用中，主要用到了swscale.h文件中的三个函数，分别是：       struct SwsContext *sws_getContext(int srcW, int srcH, enum AVPixelFo
C++中的智能指针类别
最新发布
GOLOJO的博客
05-31 916
C++中的智能指针是用来管理动态内存的对象，避免我们手动管理内存时可能会带来的复杂性和错误。C++标准库提供了几种不同类型的智能指针，包括``、``和`- 优先使用``和``来创建智能指针，避免手动使用`new`，可以提高代码的安全性和性能。- 在使用``时要特别小心循环引用的问题。可以使用``来打破循环引用。- 确保智能指针的生命周期管理正确，避免悬空指针和内存泄漏。- 虽然智能指针极大地简化了内存管理，但也带来了额外的开销，如引用计数的维护。因此，在性能敏感的代码中，应慎重使用`
matlab实现rgb空间转换到yuv空间
05-05
好的，很高兴为您解答问题。您可以使用MATLAB内置的rgb2yuv函数来实现RGB到YUV空间的转换，具体代码如下： ```matlab rgbImage = imread('your_image.png'); yuvImage = rgb2yuv(rgbImage); ``` 这样，您就可以将一个RGB图像转换为YUV图像了。 同时，您也可以手动实现该转换，代码如下： ```matlab rgbImage = imread('your_image.png'); yuvImage = zeros(size(rgbImage)); for i = 1:size(rgbImage, 1) for j = 1:size(rgbImage, 2) R = rgbImage(i, j, 1); G = rgbImage(i, j, 2); B = rgbImage(i, j, 3); Y = 0.299 * R + 0.587 * G + 0.114 * B; U = -0.14713 * R - 0.28886 * G + 0.436 * B; V = 0.615 * R - 0.51498 * G - 0.10001 * B; yuvImage(i, j, 1) = Y; yuvImage(i, j, 2) = U; yuvImage(i, j, 3) = V; end end ``` 以上代码中，我们手动实现了RGB到YUV的转换方式。其中，Y表示亮度，U和V则表示色度。这里需要注意的是，该转换中需要进行颜色空间的归一化。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值