Fixed PNG width.

This commit is contained in:
n-a-c-h
2004-05-06 19:03:17 +00:00
parent 90ce4873a2
commit b43b708893

View File

@@ -1,230 +1,232 @@
#include "zpng.h"
#include "zpng.h"
#ifdef __WIN32__
#include <windows.h> #ifdef __WIN32__
#include <sys/stat.h> #include <windows.h>
#include <sys/stat.h>
#ifdef __WIN32DBG__
#include <crtdbg.h> #ifdef __WIN32DBG__
#endif #include <crtdbg.h>
#endif #endif
#endif
#ifdef __MSDOS__
#include <sys/types.h> #ifdef __MSDOS__
#include <sys/stat.h> #include <sys/types.h>
#include <unistd.h> #include <sys/stat.h>
#include <stdlib.h> #include <unistd.h>
#endif #include <stdlib.h>
#endif
#ifdef __LINUX__
#include "../gblhdr.h" #ifdef __LINUX__
#endif #include "../gblhdr.h"
#endif
extern unsigned int vidbuffer;
extern unsigned int vidbuffer;
#ifdef __PNG__
#ifdef __PNG__
int Png_Dump(const char * filename, unsigned short width, unsigned short height, unsigned char * image_data, bool usebgr)
{ int Png_Dump(const char * filename, unsigned short width, unsigned short height, unsigned char * image_data, bool usebgr)
png_structp png_ptr; {
png_infop info_ptr; png_structp png_ptr;
png_bytep * row_pointers; png_infop info_ptr;
/*Set scanline width for 32-bit color data*/ png_bytep * row_pointers;
int scanline=width*4; /*Set scanline width for 32-bit color data*/
int i; /*counter.*/ int scanline=width*4;
int png_transforms=0; int i; /*counter.*/
png_color fake_pal; int png_transforms=0;
/*Try to open the file.*/ png_color fake_pal;
FILE *fp = fopen(filename, "wb"); /*Try to open the file.*/
if (!fp) FILE *fp = fopen(filename, "wb");
{ if (!fp)
return (-1); {
} return (-1);
}
fake_pal.red = 0;
fake_pal.green = 0; fake_pal.red = 0;
fake_pal.blue = 0; fake_pal.green = 0;
/*Try to create png write struct, fail if we cannot.*/ fake_pal.blue = 0;
png_ptr = png_create_write_struct /*Try to create png write struct, fail if we cannot.*/
(PNG_LIBPNG_VER_STRING, NULL,/*(png_voidp)user_error_ptr, png_ptr = png_create_write_struct
user_error_fn*/NULL, NULL/*user_warning_fn*/); (PNG_LIBPNG_VER_STRING, NULL,/*(png_voidp)user_error_ptr,
if (!png_ptr) user_error_fn*/NULL, NULL/*user_warning_fn*/);
return (-1); if (!png_ptr)
return (-1);
/*set png I/O source.*/
png_init_io(png_ptr, fp); /*set png I/O source.*/
png_init_io(png_ptr, fp);
/* set the zlib compression level */
png_set_compression_level(png_ptr, /* set the zlib compression level */
Z_BEST_COMPRESSION); png_set_compression_level(png_ptr,
Z_BEST_COMPRESSION);
/* set other zlib parameters */
png_set_compression_mem_level(png_ptr, 8); /* set other zlib parameters */
png_set_compression_strategy(png_ptr, png_set_compression_mem_level(png_ptr, 8);
Z_DEFAULT_STRATEGY); png_set_compression_strategy(png_ptr,
png_set_compression_window_bits(png_ptr, 15); Z_DEFAULT_STRATEGY);
png_set_compression_method(png_ptr, 8); png_set_compression_window_bits(png_ptr, 15);
png_set_compression_buffer_size(png_ptr, 8192); png_set_compression_method(png_ptr, 8);
png_set_compression_buffer_size(png_ptr, 8192);
/*try to create info struct. Fail and delete existing structs if info struct cannot be created.*/
info_ptr = png_create_info_struct(png_ptr); /*try to create info struct. Fail and delete existing structs if info struct cannot be created.*/
if (!info_ptr) info_ptr = png_create_info_struct(png_ptr);
{ if (!info_ptr)
png_destroy_write_struct(&png_ptr, {
(png_infopp)NULL); png_destroy_write_struct(&png_ptr,
return (-1); (png_infopp)NULL);
} return (-1);
}
/*set a lot of image info (code adapted from libpng documentation!)*/
png_set_IHDR(png_ptr, info_ptr, width, height, /*set a lot of image info (code adapted from libpng documentation!)*/
8, PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE, png_set_IHDR(png_ptr, info_ptr, width, height,
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT); 8, PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE,
PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
info_ptr->color_type=PNG_COLOR_TYPE_RGB_ALPHA;
info_ptr->color_type=PNG_COLOR_TYPE_RGB_ALPHA;
/*Allocate an array of scanline pointers*/
row_pointers=(png_bytep*)malloc(height*sizeof(png_bytep)); /*Allocate an array of scanline pointers*/
for (i=0;i<height;i++) row_pointers=(png_bytep*)malloc(height*sizeof(png_bytep));
{ for (i=0;i<height;i++)
#ifdef __UPSIDE_DOWN__ {
/*invert to normal image format.*/ #ifdef __UPSIDE_DOWN__
row_pointers[i]=&image_data[scanline*(height-i-1)]; /*invert to normal image format.*/
#else row_pointers[i]=&image_data[scanline*(height-i-1)];
row_pointers[i]=&image_data[scanline*i]; #else
#endif row_pointers[i]=&image_data[scanline*i];
} #endif
}
/*tell the png library what to encode.*/
png_set_rows(png_ptr, info_ptr, row_pointers); /*tell the png library what to encode.*/
png_set_rows(png_ptr, info_ptr, row_pointers);
if(usebgr)
png_transforms|=PNG_TRANSFORM_BGR; if(usebgr)
/*Write image to file*/ png_transforms|=PNG_TRANSFORM_BGR;
png_write_png(png_ptr, info_ptr, png_transforms, NULL); /*Write image to file*/
png_write_png(png_ptr, info_ptr, png_transforms, NULL);
/*close file*/
fclose(fp); /*close file*/
fclose(fp);
/*Destroy PNG structs*/
png_destroy_write_struct(&png_ptr, &info_ptr); /*Destroy PNG structs*/
png_destroy_write_struct(&png_ptr, &info_ptr);
/*clean up dynamically allocated RAM.*/
free(row_pointers); /*clean up dynamically allocated RAM.*/
free(row_pointers);
#ifdef __WIN32DBG__
_CrtDumpMemoryLeaks(); #ifdef __WIN32DBG__
#endif _CrtDumpMemoryLeaks();
return 0; #endif
} return 0;
}
char *generate_filename(void)
{ char *generate_filename(void)
extern char fnames; {
char *filename; extern char fnames;
char *tmp = &fnames; char *filename;
char *tmp2 = 0; char *tmp = &fnames;
short i=0; char *tmp2 = 0;
struct stat buf; short i=0;
struct stat buf;
#ifdef __MSDOS__
filename = (char *)malloc(14); #ifdef __MSDOS__
for(i=0;i<10000;i++) filename = (char *)malloc(14);
{ for(i=0;i<10000;i++)
if(i>1000) {
sprintf(filename, "Image%03d.png", i); if(i>1000)
else sprintf(filename, "Imag%04d.png", i); sprintf(filename, "Image%03d.png", i);
if(stat(filename, &buf)==-1) else sprintf(filename, "Imag%04d.png", i);
break; if(stat(filename, &buf)==-1)
} break;
return filename; }
#endif return filename;
#endif
tmp++; // the first char is the string length
// removes the path if there is one tmp++; // the first char is the string length
while (*tmp!=0) tmp++; // removes the path if there is one
while ((*tmp!='/') && (tmp!=&fnames)) tmp--; while (*tmp!=0) tmp++;
tmp++; while ((*tmp!='/') && (tmp!=&fnames)) tmp--;
// allocates enough memory to store the filename tmp++;
filename = (char *)malloc(strlen(tmp)+10); // allocates enough memory to store the filename
strcpy(filename, tmp); filename = (char *)malloc(strlen(tmp)+10);
strcpy(filename, tmp);
tmp = filename+strlen(filename);
while (*tmp!='.') tmp--; tmp = filename+strlen(filename);
while (*tmp!='.') tmp--;
#ifdef __LINUX__
tmp2 = filename; #ifdef __LINUX__
while (tmp2<tmp) { tmp2 = filename;
if (*tmp2 == ' ') *tmp2 = '_'; while (tmp2<tmp) {
tmp2++; if (*tmp2 == ' ') *tmp2 = '_';
} tmp2++;
#endif }
#endif
/*find first unused file*/
/*find first unused file*/
/*Note: this results in a 9999 image limit!*/
/*Note: this results in a 9999 image limit!*/
for(i=0;i<10000;i++)
{ for(i=0;i<10000;i++)
#ifdef __LINUX__ {
sprintf(tmp, "_%04d.png\0", i); #ifdef __LINUX__
#else sprintf(tmp, "_%04d.png\0", i);
sprintf(tmp, " %04d.png\0", i); #else
#endif sprintf(tmp, " %04d.png\0", i);
if(stat(filename, &buf)==-1) #endif
break; if(stat(filename, &buf)==-1)
} break;
}
return filename;
} return filename;
}
void Grab_PNG_Data(void)
{ #define SNAP_WIDTH 223
char *filename; void Grab_PNG_Data(void)
bool is_bgr_data=true; {
char *filename;
/*These are the variables used to perform the 32-bit conversion*/ bool is_bgr_data=true;
int i,j;
unsigned short* pixel; /*These are the variables used to perform the 32-bit conversion*/
unsigned short conv_pixel; int i,j;
/*Set scanline width for 32-bit color data: 4*256 = 1024*/ unsigned short* pixel;
int scanline=1024; unsigned short conv_pixel;
unsigned char *DIBits; /*Set scanline width for 32-bit color data: 4*256 = 1024*/
unsigned int * DBits; int scanline=1024;
unsigned char *DIBits;
filename = generate_filename(); unsigned int * DBits;
/*Allocate image buffer for DIB data*/ filename = generate_filename();
DIBits=(unsigned char*)malloc(scanline*224);
/*Allocate image buffer for DIB data*/
/*Cast pointer to 32-bit data type*/ DIBits=(unsigned char*)malloc(scanline*SNAP_WIDTH);
DBits=(unsigned int*) DIBits;
/*Cast pointer to 32-bit data type*/
/*Use zsKnight's 16 to 32 bit color conversion*/ DBits=(unsigned int*) DIBits;
pixel=(unsigned short*)(vidbuffer);
for(i=0;i<224;i++) /*Use zsKnight's 16 to 32 bit color conversion*/
{ pixel=(unsigned short*)(vidbuffer);
for(j=0;j<256;j++) for(i=0;i<SNAP_WIDTH;i++)
{ {
for(j=0;j<256;j++)
conv_pixel=pixel[(i*288)+j+16]; {
DBits[i*256+j]=((conv_pixel&0xF800)<<8)|
((conv_pixel&0x07E0)<<5)| conv_pixel=pixel[(i*288)+j+16];
((conv_pixel&0x001F)<<3)|0xFF000000; DBits[i*256+j]=((conv_pixel&0xF800)<<8)|
} ((conv_pixel&0x07E0)<<5)|
} ((conv_pixel&0x001F)<<3)|0xFF000000;
}
/*compress and write the PNG*/ }
Png_Dump(filename, 256, 224, DIBits, is_bgr_data);
/*compress and write the PNG*/
free(DIBits); Png_Dump(filename, 256, SNAP_WIDTH, DIBits, is_bgr_data);
free(filename);
free(DIBits);
#ifdef __WIN32DBG__ free(filename);
_CrtDumpMemoryLeaks();
#endif #ifdef __WIN32DBG__
_CrtDumpMemoryLeaks();
} #endif
#endif }
#endif