谷歌开源项目Google Preview Image Extractor(PIEX) (附上完整demo代码)

前天偶然看到谷歌开源项目中有一个近乎无人问津的项目Google Preview Image Extractor(PIEX) 。

项目地址:

https://github.com/google/piex

官方的描述是这样的：

The Preview Image Extractor (PIEX) is designed to find and extract the largest
JPEG compressed preview image contained in a RAW file.

也就是说，这个项目是用来提取无损图片格式(RAW格式)中内嵌的JPG预览。阅读代码和实际编写demo后，发现不是所有的无损图片格式(RAW格式)都有内嵌jpg预览，理论上最新的摄像设备应该都支持内嵌预览图了。支持解析载入如下图像格式: ARW, CR2, DNG, NEF, NRW, ORF, PEF, RAF, RW2, SRW感觉支持的格式也挺全面的。

而该项目下面没有example相关的代码，由于感兴趣，故阅读其代码写了demo。

这个项目可以考虑用在特定情况下加速加载无损格式的预览图，提升用户体验。

关于本人采用的一些RAW格式素材的下载，见链接：

http://www.ed2kfile.com/126607注:这些素材有部分是没有内嵌预览jpg的。

贴上对应的提取效果图：

[无损格式图片集合].sony_a200_3[www.ed2kfile.com].ARW

电驴下载链接:ed2k://|file|[%E6%97%A0%E6%8D%9F%E6%A0%BC%E5%BC%8F%E5%9B%BE%E7%89%87%E9%9B%86%E5%90%88].sony_a200_3[www.ed2kfile.com].ARW|10689180|c0ed57fc9898e1f75d55cdca62cd8ee8|h=gsggltnbno7emg4l2ns5qyi6pedcnub3|/

预览图:

缩略图:

贴上完整demo代码:

#include <iostream>
#include <windows.h>
#include "piex/piex.h"
#include "piex/piex_types.h"
#include <fstream>
#include <vector>
#if defined(_MSC_VER) || defined(__ANDROID_API__)
#define USE_OMP
#endif
#ifndef USE_OMP
#include <chrono>
auto const epoch = std::chrono::steady_clock::now();
double now() {
	return std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now() - epoch).count() / 1000.0;
};
#else
#include <omp.h>
auto const epoch = omp_get_wtime();
double now() {
	return omp_get_wtime() - epoch;
};
#endif
template<typename FN>
double bench(const FN &fn) {
	auto took = -now();
	return (fn(), took + now());
}
#include <chrono>
#include <thread>
void sleep(double secs) {
	std::chrono::microseconds duration((int)(secs * 1000000));
	std::this_thread::sleep_for(duration);
}

class FileStream : public piex::StreamInterface
{
public:
	FileStream(const std::string & file)
	{
		if (file.empty()) {
			buffer.clear();
			bufferSize = 0;
			return;
		}
		std::ifstream ifs(file.c_str(), std::ios::binary);
		if (!ifs.good()) {
			buffer.clear();
			bufferSize = 0;
			return;
		}
		std::vector<unsigned char> curBuffer((std::istreambuf_iterator<char>(ifs)), (std::istreambuf_iterator<char>()));
		buffer.swap(curBuffer);
		bufferSize = buffer.size();

	}
	FileStream(std::vector<unsigned char>& abuffer)
	{
		int bufsize = abuffer.size();
		if (bufsize > 0)
		{
			buffer.resize(bufsize);
			memcpy(buffer.data(), abuffer.data(), bufsize*sizeof(unsigned char));
			bufferSize = bufsize;
		}
	}
	~FileStream(){ bufferSize = 0; buffer.clear(); };
	piex::Error GetData(const size_t offset, const size_t length,
		unsigned char* data)
	{
		if (bufferSize > 0)
		{
			unsigned char *readOffset = &buffer[offset];
			unsigned char *endBuffer = &buffer[bufferSize - 1];
			if ((readOffset + length) <= endBuffer)
			{
				memcpy(data, readOffset, length);
				return	piex::kOk;
			}
			else
			{
				return	piex::kFail;
			}
		}
		return	piex::kUnsupported;
	}

	std::vector<unsigned char>& GetBuffer()
	{
		return	buffer;
	}
protected:
	std::vector<unsigned char> buffer;
	int  bufferSize = 0;
};

bool writefile(const std::string &filename, const std::vector< char>& buffer) {
	if (!buffer.empty()) {
		std::ofstream ofs(filename.c_str(), std::ios::binary);
		ofs.write(&buffer[0], buffer.size());
		return ofs.good();
	}
	return false;
}
int main(int argc, char **argv) {

	std::cout << "Google Preview Image Extractor(PIEX) Demo" << std::endl;
	std::cout << "项目地址:https://github.com/google/piex" << std::endl;
	std::cout << "-----------------------------------------------------" << std::endl;
	std::cout << "支持解析载入如下图像格式: " << std::endl;
	std::cout << "ARW, CR2, DNG, NEF, NRW, " << std::endl;
	std::cout << "ORF, PEF, RAF, RW2, SRW" << std::endl;
	std::cout << " 输出JPG图像格式." << std::endl;
	std::cout << "Demo By Gaozhihan (Build 2016-01-19)" << std::endl;
	std::cout << "本人博客: http://tntmonks.cnblogs.com/" << std::endl;
	std::cout << "-----------------------------------------------------" << std::endl;
	if (argc < 2) {
		std::cout << "用法: " << argv[0] << " rawImage [rawImage [...]]" << std::endl;
		return -1;
	}
	std::string szfile;
	std::string savefile;

	for (int i = 1; i < argc; ++i) {
		szfile = argv[i];
		std::cout << "加载文件: " << std::endl << szfile.c_str() << std::endl;
		char drive[_MAX_DRIVE];
		char dir[_MAX_DIR];
		char fname[_MAX_FNAME];
		char ext[_MAX_EXT];
		_splitpath_s(szfile.c_str(), drive, dir, fname, ext);
		savefile += drive;
		savefile += dir;
		savefile += fname;
		savefile += "_";
		std::vector< char> previewBuffer;
		std::vector< char> previewThumbnail;

		try {
			double costTime = bench([&]{
				FileStream fsfile(szfile.c_str());
				piex::StreamInterface* data_stream = &fsfile;
				piex::PreviewImageData rawPreviewImage;
				piex::Error piexErr = piex::GetPreviewImageData(data_stream, &rawPreviewImage);
				if (piexErr == piex::Error::kFail) {
					std::cout << "传入的无损格式文件数据有误." << std::endl;
				}
				else if (piexErr == piex::Error::kUnsupported) {
					std::cout << "传入的无损格式文件数据不支持." << std::endl;
				}
				else
				{
					std::cout << " 有效raw 格式,提取内嵌预览数据." << std::endl;
					previewBuffer.resize(rawPreviewImage.preview_length);
					memcpy(previewBuffer.data(), fsfile.GetBuffer().data() + rawPreviewImage.preview_offset, rawPreviewImage.preview_length);

					previewThumbnail.resize(rawPreviewImage.thumbnail_length);
					memcpy(previewThumbnail.data(), fsfile.GetBuffer().data() + rawPreviewImage.thumbnail_offset, rawPreviewImage.thumbnail_length);
				}
			});
			std::cout << " 处理耗时: " << int(costTime * 1000) << " ms" << std::endl;

			std::string previewFile = savefile;
			previewFile += "_preview.jpg";

			std::string thumbnailFile = savefile;
			thumbnailFile += "_thumbnail.jpg";
			costTime = bench([&]{
				writefile(previewFile, previewBuffer);
				writefile(thumbnailFile, previewThumbnail);
			});
			std::cout << " 保存耗时: " << int(costTime * 1000) << " ms" << std::endl;
			ShellExecuteA(NULL, "open", previewFile.c_str(), NULL, NULL, SW_SHOW);
		}
		catch (...) {
			std::cout << "\r出错！" << std::endl;
		}
	}
	std::cout << "处理完毕，按任意键退出." << std::endl;
	getchar();
	return 0;
}

　　代码下载

本文只是抛砖引玉一下，若有其他相关问题或者需求也可以邮件联系我探讨。

邮箱地址是:

gaozhihan@vip.qq.com