46 lines
1.7 KiB
C++
46 lines
1.7 KiB
C++
|
#include "bmp.hpp"
|
|||
|
|
|
|||
|
|
#include <fstream>
|
|||
|
|
#include <iostream>
|
|||
|
|
|
|||
|
|
// 将RGB24格式像素数据封装为BMP图像
|
|||
|
|
bool write_bmp(const char* filename, uint8_t* data, int32_t width,
|
|||
|
|
int32_t height) {
|
|||
|
|
BMPFileHeader file_header = {0};
|
|||
|
|
BMPInfoHeader info_header = {0};
|
|||
|
|
std::ofstream ofs(filename, std::ios::binary);
|
|||
|
|
if (!ofs) {
|
|||
|
|
std::cerr << "Failed to create file: " << filename << std::endl;
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
// BMP文件头
|
|||
|
|
file_header.type = 0x4D42; // BM
|
|||
|
|
file_header.size =
|
|||
|
|
sizeof(BMPFileHeader) + sizeof(BMPInfoHeader) + width * height * 3;
|
|||
|
|
file_header.offset = sizeof(BMPFileHeader) + sizeof(BMPInfoHeader);
|
|||
|
|
ofs.write(reinterpret_cast<char*>(&file_header), sizeof(file_header));
|
|||
|
|
|
|||
|
|
// BMP位图信息头
|
|||
|
|
info_header.size = sizeof(BMPInfoHeader);
|
|||
|
|
info_header.width = width;
|
|||
|
|
info_header.height = height;
|
|||
|
|
info_header.planes = 1;
|
|||
|
|
info_header.bit_count = 24;
|
|||
|
|
info_header.size_image = width * height * 3;
|
|||
|
|
ofs.write(reinterpret_cast<char*>(&info_header), sizeof(info_header));
|
|||
|
|
|
|||
|
|
// 像素数据
|
|||
|
|
int32_t row_size = (((width + 1) * 3) / 4) * 4; // 行字节数，必须为4的倍数
|
|||
|
|
uint8_t* row_data = new uint8_t[row_size];
|
|||
|
|
for (int32_t y = height - 1; y >= 0; --y) { // BMP图像的行是从下往上存储的
|
|||
|
|
for (int32_t x = 0; x < width; ++x) {
|
|||
|
|
row_data[x * 3 + 2] = data[(y * width + x) * 3 + 0]; // B
|
|||
|
|
row_data[x * 3 + 1] = data[(y * width + x) * 3 + 1]; // G
|
|||
|
|
row_data[x * 3 + 0] = data[(y * width + x) * 3 + 2]; // R
|
|||
|
|
}
|
|||
|
|
ofs.write(reinterpret_cast<char*>(row_data), row_size);
|
|||
|
|
}
|
|||
|
|
delete[] row_data;
|
|||
|
|
ofs.close();
|
|||
|
|
return true;
|
|||
|
|
}
|