forked from Chlumsky/msdfgen
-
Notifications
You must be signed in to change notification settings - Fork 2
/
save-bmp.cpp
109 lines (89 loc) · 3.02 KB
/
save-bmp.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include "save-bmp.h"
#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
#ifdef MSDFGEN_USE_CPP11
#include <cstdint>
#else
typedef int int32_t;
typedef unsigned uint32_t;
typedef unsigned short uint16_t;
typedef unsigned char uint8_t;
#endif
#include "arithmetics.hpp"
namespace msdfgen {
template <typename T>
static bool writeValue(FILE *file, T value) {
#ifdef __BIG_ENDIAN__
T reverse = 0;
for (int i = 0; i < sizeof(T); ++i) {
reverse <<= 8;
reverse |= value&T(0xff);
value >>= 8;
}
return fwrite(&reverse, sizeof(T), 1, file) == 1;
#else
return fwrite(&value, sizeof(T), 1, file) == 1;
#endif
}
static bool writeBmpHeader(FILE *file, int width, int height, int &paddedWidth) {
paddedWidth = 3*width+3&~3;
const uint32_t bitmapStart = 54;
const uint32_t bitmapSize = paddedWidth*height;
const uint32_t fileSize = bitmapStart+bitmapSize;
writeValue<uint16_t>(file, 0x4d42u);
writeValue<uint32_t>(file, fileSize);
writeValue<uint16_t>(file, 0);
writeValue<uint16_t>(file, 0);
writeValue<uint32_t>(file, bitmapStart);
writeValue<uint32_t>(file, 40);
writeValue<int32_t>(file, width);
writeValue<int32_t>(file, height);
writeValue<uint16_t>(file, 1);
writeValue<uint16_t>(file, 24);
writeValue<uint32_t>(file, 0);
writeValue<uint32_t>(file, bitmapSize);
writeValue<uint32_t>(file, 2835);
writeValue<uint32_t>(file, 2835);
writeValue<uint32_t>(file, 0);
writeValue<uint32_t>(file, 0);
return true;
}
bool saveBmp(const Bitmap<float> &bitmap, const char *filename) {
FILE *file = fopen(filename, "wb");
if (!file)
return false;
int paddedWidth;
writeBmpHeader(file, bitmap.width(), bitmap.height(), paddedWidth);
const uint8_t padding[4] = { };
for (int y = 0; y < bitmap.height(); ++y) {
for (int x = 0; x < bitmap.width(); ++x) {
uint8_t px = (uint8_t) clamp(int(bitmap(x, y)*0x100), 0xff);
fwrite(&px, sizeof(uint8_t), 1, file);
fwrite(&px, sizeof(uint8_t), 1, file);
fwrite(&px, sizeof(uint8_t), 1, file);
}
fwrite(padding, 1, paddedWidth-3*bitmap.width(), file);
}
return !fclose(file);
}
bool saveBmp(const Bitmap<FloatRGB> &bitmap, const char *filename) {
FILE *file = fopen(filename, "wb");
if (!file)
return false;
int paddedWidth;
writeBmpHeader(file, bitmap.width(), bitmap.height(), paddedWidth);
const uint8_t padding[4] = { };
for (int y = 0; y < bitmap.height(); ++y) {
for (int x = 0; x < bitmap.width(); ++x) {
uint8_t bgr[3] = {
(uint8_t) clamp(int(bitmap(x, y).b*0x100), 0xff),
(uint8_t) clamp(int(bitmap(x, y).g*0x100), 0xff),
(uint8_t) clamp(int(bitmap(x, y).r*0x100), 0xff)
};
fwrite(bgr, sizeof(uint8_t), 3, file);
}
fwrite(padding, 1, paddedWidth-3*bitmap.width(), file);
}
return !fclose(file);
}
}