aboutsummaryrefslogtreecommitdiff
path: root/src/BMPImage.cpp
blob: b45ff3f1ca6335af0dad27c7166eed9595e8b8f5 (plain) (blame)
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
#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>();

		// BITMAPINFOHEADER
		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>();
			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>();

			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;
				}
			}
			
			// cout << compressionMethod << " " << size << endl;
		}
		
		// cout << "Not implemented" << endl;
		return 2;
	};

	int BMPImage::writeToFile(std::string filename)
	{
		FILE* fd = fopen(filename.c_str(), "w");
		char magic[] = "BM";
		uint32_t fileSize = 14 + 12 + width*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);
			}
		}

		fclose(fd);
	
		return 0;
	};

}