lee/renderdoc
1
0
Fork 0
mirror of https://github.com/baldurk/renderdoc.git synced 2026-05-06 01:50:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add variants to stb_image_write that take FILE* instead of filename

Browse Source
This commit is contained in:
baldurk
2014-09-14 13:44:02 +01:00
parent 1c77f6d27e
commit 8ba8041a2a
1 changed files with 70 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
renderdoc/3rdparty/stb/stb_image_write.h
Vendored
+70 -27
View File
@@ -70,6 +70,12 @@ extern int stbi_write_bmp(char const *filename, int w, int h, int comp, const vo
extern int stbi_write_tga(char const *filename, int w, int h, int comp, const void *data);
extern int stbi_write_hdr(char const *filename, int w, int h, int comp, const void *data);
// versions taking a FILE* instead of filename
extern int stbi_write_png_to_file(FILE *f, int w, int h, int comp, const void *data, int stride_in_bytes);
extern int stbi_write_bmp_to_file(FILE *f, int w, int h, int comp, const void *data);
extern int stbi_write_tga_to_file(FILE *f, int w, int h, int comp, const void *data);
extern int stbi_write_hdr_to_file(FILE *f, int w, int h, int comp, const void *data);
#ifdef __cplusplus
}
#endif
@@ -157,40 +163,59 @@ static void write_pixels(FILE *f, int rgb_dir, int vdir, int x, int y, int comp,
}
}
static int outfile(char const *filename, int rgb_dir, int vdir, int x, int y, int comp, void *data, int alpha, int pad, const char *fmt, ...)
static int outfile(FILE *f, int rgb_dir, int vdir, int x, int y, int comp, void *data, int alpha, int pad, const char *fmt, ...)
{
FILE *f;
if (y < 0 || x < 0) return 0;
f = fopen(filename, "wb");
if (f) {
va_list v;
va_start(v, fmt);
writefv(f, fmt, v);
va_end(v);
write_pixels(f,rgb_dir,vdir,x,y,comp,data,alpha,pad);
fclose(f);
}
return f != NULL;
}
int stbi_write_bmp(char const *filename, int x, int y, int comp, const void *data)
int stbi_write_bmp_to_file(FILE *f, int x, int y, int comp, const void *data)
{
int pad = (-x*3) & 3;
return outfile(filename,-1,-1,x,y,comp,(void *) data,0,pad,
return outfile(f,-1,-1,x,y,comp,(void *) data,0,pad,
"11 4 22 4" "4 44 22 444444",
'B', 'M', 14+40+(x*3+pad)*y, 0,0, 14+40, // file header
40, x,y, 1,24, 0,0,0,0,0,0); // bitmap header
}
int stbi_write_tga(char const *filename, int x, int y, int comp, const void *data)
int stbi_write_bmp(char const *filename, int x, int y, int comp, const void *data)
{
FILE *f;
int ret;
f = fopen(filename, "wb");
if(!f) return 0;
ret = stbi_write_bmp_to_file(f, x, y, comp, data);
fclose(f);
return ret;
}
int stbi_write_tga_to_file(FILE *f, int x, int y, int comp, const void *data)
{
int has_alpha = (comp == 2 || comp == 4);
int colorbytes = has_alpha ? comp-1 : comp;
int format = colorbytes < 2 ? 3 : 2; // 3 color channels (RGB/RGBA) = 2, 1 color channel (Y/YA) = 3
return outfile(filename, -1,-1, x, y, comp, (void *) data, has_alpha, 0,
return outfile(f, -1,-1, x, y, comp, (void *) data, has_alpha, 0,
"111 221 2222 11", 0,0,format, 0,0,0, 0,0,x,y, (colorbytes+has_alpha)*8, has_alpha*8);
}
int stbi_write_tga(char const *filename, int x, int y, int comp, const void *data)
{
FILE *f;
int ret;
f = fopen(filename, "wb");
if(!f) return 0;
ret = stbi_write_tga_to_file(f, x, y, comp, data);
fclose(f);
return ret;
}
// *************************************************************************************************
// Radiance RGBE HDR writer
// originally by Baldur Karlsson
@@ -324,24 +349,32 @@ void stbiw__write_hdr_scanline(FILE *f, int width, int comp, unsigned char *scra
}
}
int stbi_write_hdr(char const *filename, int x, int y, int comp, const void *data)
int stbi_write_hdr_to_file(FILE *f, int x, int y, int comp, const void *data)
{
int i;
FILE *f;
unsigned char *scratch;
float *scanline;
if (y <= 0 || x <= 0) return 0;
if (!f) return 0;
scanline = (float *)data;
/* Each component is stored separately. Allocate scratch space for full output scanline. */
scratch = (unsigned char *) malloc(x*4); if (!scanline) { fclose(f); return 0; }
fprintf(f, "#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n");
fprintf(f, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
for(i=0; i < y; i++) stbiw__write_hdr_scanline(f, x, comp, scratch, scanline + comp*i*x);
free(scratch);
return 1;
}
int stbi_write_hdr(char const *filename, int x, int y, int comp, const void *data)
{
FILE *f;
int ret;
f = fopen(filename, "wb");
if (f) {
float *scanline = (float *)data;
/* Each component is stored separately. Allocate scratch space for full output scanline. */
scratch = (unsigned char *) malloc(x*4); if (!scanline) { fclose(f); return 0; }
fprintf(f, "#?RADIANCE\n# Written by stb_image_write.h\nFORMAT=32-bit_rle_rgbe\n");
fprintf(f, "EXPOSURE= 1.0000000000000\n\n-Y %d +X %d\n", y, x);
for(i=0; i < y; i++) stbiw__write_hdr_scanline(f, x, comp, scratch, scanline + comp*i*x);
free(scratch);
fclose(f);
}
return f != NULL;
if(!f) return 0;
ret = stbi_write_hdr_to_file(f, x, y, comp, data);
fclose(f);
return ret;
}
// stretchy buffer; stbiw__sbpush() == vector<>::push_back() -- stbiw__sbcount() == vector<>::size()
@@ -649,19 +682,29 @@ unsigned char *stbi_write_png_to_mem(unsigned char *pixels, int stride_bytes, in
return out;
}
int stbi_write_png(char const *filename, int x, int y, int comp, const void *data, int stride_bytes)
int stbi_write_png_to_file(FILE *f, int x, int y, int comp, const void *data, int stride_bytes)
{
FILE *f;
int len;
unsigned char *png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
unsigned char *png;
if(!f) return 0;
png = stbi_write_png_to_mem((unsigned char *) data, stride_bytes, x, y, comp, &len);
if (!png) return 0;
f = fopen(filename, "wb");
if (!f) { free(png); return 0; }
fwrite(png, 1, len, f);
fclose(f);
free(png);
return 1;
}
int stbi_write_png(char const *filename, int x, int y, int comp, const void *data, int stride_bytes)
{
FILE *f;
int ret;
f = fopen(filename, "wb");
if(!f) return 0;
ret = stbi_write_png_to_file(f, x, y, comp, data, stride_bytes);
fclose(f);
return ret;
}
#endif // STB_IMAGE_WRITE_IMPLEMENTATION
/* Revision history