summaryrefslogtreecommitdiff
path: root/thirdparty/msdfgen/core/save-tiff.cpp
blob: 71405e00e53c0bdcfb94de2c5e73620cd172008b (plain)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190

#define _CRT_SECURE_NO_WARNINGS

#include "save-tiff.h"

#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

namespace msdfgen {

template <typename T>
static bool writeValue(FILE *file, T value) {
    return fwrite(&value, sizeof(T), 1, file) == 1;
}
template <typename T>
static void writeValueRepeated(FILE *file, T value, int times) {
    for (int i = 0; i < times; ++i)
        writeValue(file, value);
}

static bool writeTiffHeader(FILE *file, int width, int height, int channels) {
    #ifdef __BIG_ENDIAN__
        writeValue<uint16_t>(file, 0x4d4du);
    #else
        writeValue<uint16_t>(file, 0x4949u);
    #endif
    writeValue<uint16_t>(file, 42);
    writeValue<uint32_t>(file, 0x0008u); // Offset of first IFD
    // Offset = 0x0008

    writeValue<uint16_t>(file, 15); // Number of IFD entries

    // ImageWidth
    writeValue<uint16_t>(file, 0x0100u);
    writeValue<uint16_t>(file, 0x0004u);
    writeValue<uint32_t>(file, 1);
    writeValue<int32_t>(file, width);
    // ImageLength
    writeValue<uint16_t>(file, 0x0101u);
    writeValue<uint16_t>(file, 0x0004u);
    writeValue<uint32_t>(file, 1);
    writeValue<int32_t>(file, height);
    // BitsPerSample
    writeValue<uint16_t>(file, 0x0102u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, channels);
    if (channels > 1)
        writeValue<uint32_t>(file, 0x00c2u); // Offset of 32, 32, ...
    else {
        writeValue<uint16_t>(file, 32);
        writeValue<uint16_t>(file, 0);
    }
    // Compression
    writeValue<uint16_t>(file, 0x0103u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint16_t>(file, 1);
    writeValue<uint16_t>(file, 0);
    // PhotometricInterpretation
    writeValue<uint16_t>(file, 0x0106u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint16_t>(file, channels >= 3 ? 2 : 1);
    writeValue<uint16_t>(file, 0);
    // StripOffsets
    writeValue<uint16_t>(file, 0x0111u);
    writeValue<uint16_t>(file, 0x0004u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint32_t>(file, 0x00d2u+(channels > 1)*channels*12); // Offset of pixel data
    // SamplesPerPixel
    writeValue<uint16_t>(file, 0x0115u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint16_t>(file, channels);
    writeValue<uint16_t>(file, 0);
    // RowsPerStrip
    writeValue<uint16_t>(file, 0x0116u);
    writeValue<uint16_t>(file, 0x0004u);
    writeValue<uint32_t>(file, 1);
    writeValue<int32_t>(file, height);
    // StripByteCounts
    writeValue<uint16_t>(file, 0x0117u);
    writeValue<uint16_t>(file, 0x0004u);
    writeValue<uint32_t>(file, 1);
    writeValue<int32_t>(file, sizeof(float)*channels*width*height);
    // XResolution
    writeValue<uint16_t>(file, 0x011au);
    writeValue<uint16_t>(file, 0x0005u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint32_t>(file, 0x00c2u+(channels > 1)*channels*2); // Offset of 300, 1
    // YResolution
    writeValue<uint16_t>(file, 0x011bu);
    writeValue<uint16_t>(file, 0x0005u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint32_t>(file, 0x00cau+(channels > 1)*channels*2); // Offset of 300, 1
    // ResolutionUnit
    writeValue<uint16_t>(file, 0x0128u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, 1);
    writeValue<uint16_t>(file, 2);
    writeValue<uint16_t>(file, 0);
    // SampleFormat
    writeValue<uint16_t>(file, 0x0153u);
    writeValue<uint16_t>(file, 0x0003u);
    writeValue<uint32_t>(file, channels);
    if (channels > 1)
        writeValue<uint32_t>(file, 0x00d2u+channels*2); // Offset of 3, 3, ...
    else {
        writeValue<uint16_t>(file, 3);
        writeValue<uint16_t>(file, 0);
    }
    // SMinSampleValue
    writeValue<uint16_t>(file, 0x0154u);
    writeValue<uint16_t>(file, 0x000bu);
    writeValue<uint32_t>(file, channels);
    if (channels > 1)
        writeValue<uint32_t>(file, 0x00d2u+channels*4); // Offset of 0.f, 0.f, ...
    else
        writeValue<float>(file, 0.f);
    // SMaxSampleValue
    writeValue<uint16_t>(file, 0x0155u);
    writeValue<uint16_t>(file, 0x000bu);
    writeValue<uint32_t>(file, channels);
    if (channels > 1)
        writeValue<uint32_t>(file, 0x00d2u+channels*8); // Offset of 1.f, 1.f, ...
    else
        writeValue<float>(file, 1.f);
    // Offset = 0x00be

    writeValue<uint32_t>(file, 0);

    if (channels > 1) {
        // 0x00c2 BitsPerSample data
        writeValueRepeated<uint16_t>(file, 32, channels);
        // 0x00c2 + 2*N XResolution data
        writeValue<uint32_t>(file, 300);
        writeValue<uint32_t>(file, 1);
        // 0x00ca + 2*N YResolution data
        writeValue<uint32_t>(file, 300);
        writeValue<uint32_t>(file, 1);
        // 0x00d2 + 2*N SampleFormat data
        writeValueRepeated<uint16_t>(file, 3, channels);
        // 0x00d2 + 4*N SMinSampleValue data
        writeValueRepeated<float>(file, 0.f, channels);
        // 0x00d2 + 8*N SMaxSampleValue data
        writeValueRepeated<float>(file, 1.f, channels);
        // Offset = 0x00d2 + 12*N
    } else {
        // 0x00c2 XResolution data
        writeValue<uint32_t>(file, 300);
        writeValue<uint32_t>(file, 1);
        // 0x00ca YResolution data
        writeValue<uint32_t>(file, 300);
        writeValue<uint32_t>(file, 1);
        // Offset = 0x00d2
    }

    return true;
}

template <int N>
bool saveTiffFloat(const BitmapConstRef<float, N> &bitmap, const char *filename) {
    FILE *file = fopen(filename, "wb");
    if (!file)
        return false;
    writeTiffHeader(file, bitmap.width, bitmap.height, N);
    for (int y = bitmap.height-1; y >= 0; --y)
        fwrite(bitmap(0, y), sizeof(float), N*bitmap.width, file);
    return !fclose(file);
}

bool saveTiff(const BitmapConstRef<float, 1> &bitmap, const char *filename) {
    return saveTiffFloat(bitmap, filename);
}
bool saveTiff(const BitmapConstRef<float, 3> &bitmap, const char *filename) {
    return saveTiffFloat(bitmap, filename);
}
bool saveTiff(const BitmapConstRef<float, 4> &bitmap, const char *filename) {
    return saveTiffFloat(bitmap, filename);
}

}