legacy-imlib2/src/modules/loaders/loader_ff.c

143 lines
3.3 KiB
C

/* Farbfeld (http://tools.suckless.org/farbfeld) */
#include "config.h"
#include "Imlib2_Loader.h"
#include <arpa/inet.h>
static const char *const _formats[] = { "ff" };
#define mm_check(p) ((const char *)(p) <= (const char *)im->fi->fdata + im->fi->fsize)
typedef struct {
unsigned char magic[8];
uint32_t w, h;
} ff_hdr_t;
static int
_load(ImlibImage *im, int load_data)
{
int rc;
int rowlen, i, j;
const ff_hdr_t *hdr;
const uint16_t *row;
uint8_t *imdata;
rc = LOAD_FAIL;
if (im->fi->fsize < (long)sizeof(ff_hdr_t))
return rc;
/* read and check the header */
hdr = im->fi->fdata;
if (memcmp("farbfeld", hdr->magic, sizeof(hdr->magic)))
goto quit;
rc = LOAD_BADIMAGE; /* Format accepted */
im->w = ntohl(hdr->w);
im->h = ntohl(hdr->h);
if (!IMAGE_DIMENSIONS_OK(im->w, im->h))
goto quit;
im->has_alpha = 1;
if (!load_data)
QUIT_WITH_RC(LOAD_SUCCESS);
/* Load data */
if (!__imlib_AllocateData(im))
QUIT_WITH_RC(LOAD_OOM);
rowlen = 4 * im->w; /* RGBA */
imdata = (uint8_t *) im->data;
row = (uint16_t *) (hdr + 1);
for (i = 0; i < im->h; i++, imdata += rowlen, row += rowlen)
{
if (!mm_check(row + rowlen))
goto quit;
for (j = 0; j < rowlen; j += 4)
{
/*
* 16-Bit to 8-Bit (RGBA -> BGRA)
* 255 * 257 = 65535 = 2^16-1 = UINT16_MAX
*/
imdata[j + 2] = ntohs(row[j + 0]) / 257;
imdata[j + 1] = ntohs(row[j + 1]) / 257;
imdata[j + 0] = ntohs(row[j + 2]) / 257;
imdata[j + 3] = ntohs(row[j + 3]) / 257;
}
if (im->lc && __imlib_LoadProgressRows(im, i, 1))
QUIT_WITH_RC(LOAD_BREAK);
}
rc = LOAD_SUCCESS;
quit:
return rc;
}
static int
_save(ImlibImage *im)
{
int rc;
FILE *f = im->fi->fp;
size_t rowlen, i, j;
uint32_t tmp32;
uint16_t *row;
const uint8_t *imdata;
rc = LOAD_BADFILE;
row = NULL;
/* write header */
fputs("farbfeld", f);
tmp32 = htonl(im->w);
if (fwrite(&tmp32, sizeof(uint32_t), 1, f) != 1)
goto quit;
tmp32 = htonl(im->h);
if (fwrite(&tmp32, sizeof(uint32_t), 1, f) != 1)
goto quit;
/* write data */
rowlen = im->w * (sizeof("RGBA") - 1);
row = malloc(rowlen * sizeof(uint16_t));
if (!row)
QUIT_WITH_RC(LOAD_OOM);
imdata = (uint8_t *) im->data;
for (i = 0; i < (uint32_t) im->h; ++i, imdata += rowlen)
{
for (j = 0; j < rowlen; j += 4)
{
/*
* 8-Bit to 16-Bit
* 255 * 257 = 65535 = 2^16-1 = UINT16_MAX
*/
row[j + 0] = htons(imdata[j + 2] * 257);
row[j + 1] = htons(imdata[j + 1] * 257);
row[j + 2] = htons(imdata[j + 0] * 257);
row[j + 3] = htons(imdata[j + 3] * 257);
}
if (fwrite(row, sizeof(uint16_t), rowlen, f) != rowlen)
goto quit;
if (im->lc && __imlib_LoadProgressRows(im, i, 1))
QUIT_WITH_RC(LOAD_BREAK);
}
rc = LOAD_SUCCESS;
quit:
free(row);
return rc;
}
IMLIB_LOADER(_formats, _load, _save);