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
|
#include "BMPImage.h"
#include "reader.h"
#include <cstdint>
#include <iostream>
#include <memory>
using std::cout, std::endl;
namespace TehImage
{
int BMPImage::readFromFile(std::string filename)
{
Reader reader(filename, FileEndianness::LITTLE);
char magic[2];
uint8_t expected[] = {0x42, 0x4D};
reader.readBytes(magic, 2);
uint32_t fileSize = reader.readData<uint32_t>();
reader.skipBytes(4);
uint32_t offset = reader.readData<uint32_t>();
if(strncmp(magic, (char*)expected, 2) != 0)
{
cout << "Not a BMP" << endl;
return 1;
}
uint32_t headerSize = reader.readData<uint32_t>();
// BITMAPCOREHEADER
if(headerSize == 12)
{
width = reader.readData<uint16_t>();
height = reader.readData<uint16_t>();
uint16_t colorPlanes = reader.readData<uint16_t>();
bpp = reader.readData<uint16_t>();
}
// BITMAPINFOHEADER
else if(headerSize == 40)
{
width = reader.readData<uint32_t>();
height = reader.readData<uint32_t>();
uint16_t colorPlanes = reader.readData<uint16_t>();
bpp = reader.readData<uint16_t>();
uint32_t compressionMethod = reader.readData<uint32_t>();
if(compressionMethod != 0)
{
cout << "Compression method not supported: " << compressionMethod << endl;
return 1;
}
uint32_t size = reader.readData<uint32_t>();
uint32_t horrRes = reader.readData<uint32_t>();
uint32_t vertRes = reader.readData<uint32_t>();
uint32_t paletteSize = reader.readData<uint32_t>();
uint32_t importantColors = reader.readData<uint32_t>();
// cout << compressionMethod << " " << size << endl;
}
else
{
cout << "Header type not supported: " << headerSize << endl;
return 1;
}
if(bpp != 24)
{
cout << "bpp not supported: " << bpp << endl;
return 1;
}
pixels = std::make_unique<Pixel[]>(width * height);
for(int y = height-1; y >= 0; y--)
{
for(int x = 0; x < width; x++)
{
Pixel& pixel = (*this)[x,y];
pixel.b = reader.readByte();
pixel.g = reader.readByte();
pixel.r = reader.readByte();
pixel.a = 255;
}
reader.skipBytes((width * 3)%4);
}
// cout << "Not implemented" << endl;
return 2;
};
int BMPImage::writeToFile(std::string filename)
{
FILE* fd = fopen(filename.c_str(), "w");
char magic[] = "BM";
uint32_t rowSize = (bpp * width + 31)/32;
uint32_t fileSize = 14 + 12 + rowSize*height*/*(bitDepth/8)*/8*3;
char zero[] = "\0\0\0\0";
uint32_t offset = 26;
uint32_t headerSize = 12;
uint16_t width = this->width;
uint16_t height = this->height;
uint16_t colorPlanes = 1;
uint16_t bitsPerPixel = /*bitDepth*/8*3;
fwrite(magic, sizeof(char), 2, fd);
fwrite(&fileSize, sizeof(uint32_t), 1, fd);
fwrite(zero, sizeof(char), 4, fd);
fwrite(&offset, sizeof(uint32_t), 1, fd);
fwrite(&headerSize, sizeof(uint32_t), 1, fd);
fwrite(&width, sizeof(uint16_t), 1, fd);
fwrite(&height, sizeof(uint16_t), 1, fd);
fwrite(&colorPlanes, sizeof(uint16_t), 1, fd);
fwrite(&bitsPerPixel, sizeof(uint16_t), 1, fd);
for(int y = height-1; y >= 0; y--)
{
for(int x = 0; x < width; x++)
{
Pixel& pixel = (*this)[x,y];
fwrite(&pixel.b, sizeof(uint8_t), 1, fd);
fwrite(&pixel.g, sizeof(uint8_t), 1, fd);
fwrite(&pixel.r, sizeof(uint8_t), 1, fd);
}
fwrite(zero, sizeof(char), (width * 3)%4, fd);
}
fclose(fd);
return 0;
};
}
|
