-
Notifications
You must be signed in to change notification settings - Fork 0
/
TGA.cpp
145 lines (119 loc) · 3.3 KB
/
TGA.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
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
#include "UniHeader.h"
TGA::TGA()
{
}
TGA::~TGA()
{
}
bool TGA::readTGA(string filename, BitmapMin *bmp)
{
TGAFILE tgaFile;
FILE *filePtr;
unsigned char ucharBad;
short int sintBad;
long imageSize;
int colorMode;
unsigned char colorSwap;
//Open TGA file
filePtr = fopen(filename.c_str(), "rb");
if (filePtr == NULL)
{
return false;
}
fread(&ucharBad, sizeof(unsigned char), 1, filePtr);
fread(&ucharBad, sizeof(unsigned char), 1, filePtr);
fread(&tgaFile.typeCode, sizeof(unsigned char), 1, filePtr);
//Type code 2 (uncompressed RGB image)
//Type code 3 (uncompressed black-and-white images).
if (tgaFile.typeCode != 2 && tgaFile.typeCode != 3)
{
fclose(filePtr);
return false;
}
fread(&sintBad, sizeof(short int), 1, filePtr);
fread(&sintBad, sizeof(short int), 1, filePtr);
fread(&ucharBad, sizeof(unsigned char), 1, filePtr);
fread(&sintBad, sizeof(short int), 1, filePtr);
fread(&sintBad, sizeof(short int), 1, filePtr);
//Read the image's width and height.
fread(&tgaFile.width, sizeof(short int), 1, filePtr);
fread(&tgaFile.height, sizeof(short int), 1, filePtr);
//Read the bit depth.
fread(&tgaFile.bitCount, sizeof(unsigned char), 1, filePtr);
//Read one byte of data we don't need.
fread(&ucharBad, sizeof(unsigned char), 1, filePtr);
//Color mode -> 3 = BGR, 4 = BGRA.
colorMode = tgaFile.bitCount / 8;
imageSize = tgaFile.width * tgaFile.height * colorMode;
//Allocate memory for the image data.
tgaFile.data = (unsigned char*)malloc(sizeof(unsigned char)*imageSize);
//Read the image data.
fread(tgaFile.data, sizeof(unsigned char), imageSize, filePtr);
for (int imageIdx = 0; imageIdx < imageSize; imageIdx += colorMode)
{
colorSwap = tgaFile.data[imageIdx];
tgaFile.data[imageIdx] = tgaFile.data[imageIdx + 2];
tgaFile.data[imageIdx + 2] = colorSwap;
}
bmp->width = tgaFile.width;
bmp->height = tgaFile.height;
//Load filedata to bitmap object
int curX = -1;
int curY = 0;
int n1 = 0;
int n = 0;
for (int a = 0; a < tgaFile.width; a++)
{
for (int b = 0; b < tgaFile.height; b++)
{
bmp->map[tgaFile.height - curY - 1][curX].b = int(tgaFile.data[n1 + 2]);
bmp->map[tgaFile.height - curY - 1][curX].g = int(tgaFile.data[n1 + 1]);
bmp->map[tgaFile.height - curY - 1][curX].r = int(tgaFile.data[n1]);
curX++;
if (n >= tgaFile.width)
{
n = 0;
curX = 0;
curY++;
}
n++;
n1 += 3;
}
}
fclose(filePtr);
return true;
}
//--------------------------------------
// Desc: Write a BitmapMin object to a TGA image file
//--------------------------------------
bool TGA::writeTGA(string filename, BitmapMin *bmp)
{
unsigned width = bmp->width;
unsigned height = bmp->height;
ofstream tgafile(filename.c_str(), ios::binary);
if (!tgafile)
{
tgafile.close();
return false;
}
//The image header
char header[18] = { 0 };
header[2] = 2; // truecolor
header[12] = bmp->width & 0xFF;
header[13] = (bmp->width >> 8) & 0xFF;
header[14] = bmp->height & 0xFF;
header[15] = (bmp->height >> 8) & 0xFF;
header[16] = 24; // bits per pixel
tgafile.write((const char*)header, 18);
for (int y = bmp->height - 1; y >= 0; y--)
{
for (int x = 0; x < bmp->width; x++)
{
tgafile.put((char)bmp->map[y][x].b);//b
tgafile.put((char)bmp->map[y][x].g);//g
tgafile.put((char)bmp->map[y][x].r);//r
}
}
tgafile.close();
return true;
}