mirror of
https://github.com/ZDoom/Raze.git
synced 2024-12-12 21:52:01 +00:00
ccdba037b5
This was necessary because everything is already allocated with the Xmalloc() functions, but a future commit will make blocks allocated with those functions no longer compatible with the system implementation of free(), which Bfree() wraps. git-svn-id: https://svn.eduke32.com/eduke32@7705 1a8010ca-5511-0410-912e-c29ae57300e0 # Conflicts: # source/build/src/build.cpp # source/build/src/mdsprite.cpp # source/build/src/polymer.cpp # source/build/src/polymost.cpp # source/build/src/texcache.cpp # source/build/src/voxmodel.cpp
97 lines
2.7 KiB
C++
97 lines
2.7 KiB
C++
#include "compat.h"
|
|
#include "pngwrite.h"
|
|
#include "crc32.h"
|
|
|
|
#include "vfs.h"
|
|
|
|
pngwrite_t png;
|
|
|
|
#define png_write_buf(p, size) buildvfs_fwrite(p, size, 1, png.file)
|
|
|
|
static FORCE_INLINE void png_write_uint32(uint32_t const in)
|
|
{
|
|
uint32_t const buf = B_BIG32(in);
|
|
png_write_buf(&buf, sizeof(uint32_t));
|
|
}
|
|
|
|
static void png_write_chunk(uint32_t const size, char const *const type,
|
|
uint8_t const *const data, uint32_t flags)
|
|
{
|
|
mz_ulong chunk_size = (flags & CHUNK_COMPRESSED) ? compressBound(size) : size;
|
|
uint8_t * const chunk = (uint8_t *) Xcalloc(1, 4 + chunk_size);
|
|
|
|
Bmemcpy(chunk, type, 4);
|
|
|
|
if (flags & CHUNK_COMPRESSED)
|
|
compress(chunk + 4, (mz_ulong *) &chunk_size, data, size);
|
|
else
|
|
Bmemcpy(chunk + 4, data, size);
|
|
|
|
png_write_uint32(chunk_size);
|
|
png_write_buf(chunk, chunk_size + 4);
|
|
|
|
uint32_t crc = Bcrc32(NULL, 0, 0L);
|
|
crc = Bcrc32(chunk, chunk_size + 4, crc);
|
|
png_write_uint32(crc);
|
|
|
|
Xfree(chunk);
|
|
}
|
|
|
|
void png_set_pal(uint8_t const * const data, int numentries)
|
|
{
|
|
png.pal_entries = numentries;
|
|
png.pal_data = (uint8_t *)Xmalloc(numentries * 3);
|
|
|
|
Bmemcpy(png.pal_data, data, numentries * 3);
|
|
}
|
|
|
|
void png_set_text(char const * const keyword, char const * const text)
|
|
{
|
|
unsigned const keylen = Bstrlen(keyword);
|
|
Bassert(keylen < 79);
|
|
unsigned const textlen = Bstrlen(text);
|
|
|
|
png.textlen = keylen + textlen + 1;
|
|
png.text = (uint8_t *) Xrealloc(png.text, png.textlen);
|
|
|
|
Bmemcpy(png.text, keyword, keylen);
|
|
*(png.text + keylen) = 0;
|
|
Bmemcpy(png.text + keylen + 1, text, textlen);
|
|
}
|
|
|
|
void png_write(buildvfs_FILE const file, int const width, int const height,
|
|
uint8_t const type, uint8_t const * const data)
|
|
{
|
|
png.file = file;
|
|
|
|
png_write_buf("\x89\x50\x4E\x47\x0D\x0A\x1A\x0A", 8);
|
|
|
|
png_ihdr_t const png_header = { B_BIG32((unsigned)width), B_BIG32((unsigned)height), 8, type, 0 };
|
|
png_write_chunk(sizeof(png_ihdr_t), "IHDR", (uint8_t const *)&png_header, 0);
|
|
|
|
if (png.text)
|
|
{
|
|
png_write_chunk(png.textlen, "tEXt", png.text, 0);
|
|
DO_FREE_AND_NULL(png.text);
|
|
}
|
|
|
|
int const bytesPerPixel = (type == PNG_TRUECOLOR ? 3 : 1);
|
|
int const bytesPerLine = width * bytesPerPixel;
|
|
|
|
if (png.pal_data)
|
|
{
|
|
png_write_chunk(png.pal_entries * 3, "PLTE", png.pal_data, 0);
|
|
DO_FREE_AND_NULL(png.pal_data);
|
|
}
|
|
|
|
int const linesiz = height * bytesPerLine + height;
|
|
uint8_t *lines = (uint8_t *) Xcalloc(1, linesiz);
|
|
|
|
for (int i = 0; i < height; i++)
|
|
Bmemcpy(lines + i * bytesPerLine + i + 1, data + i * bytesPerLine, bytesPerLine);
|
|
|
|
png_write_chunk(linesiz, "IDAT", lines, CHUNK_COMPRESSED);
|
|
png_write_chunk(0, "IEND", NULL, 0);
|
|
|
|
Xfree(lines);
|
|
}
|