mirror of https://github.com/mpv-player/mpv
127 lines
3.7 KiB
C
127 lines
3.7 KiB
C
#include <stdlib.h>
|
|
|
|
#include "config.h"
|
|
#include "bswap.h"
|
|
#include "postproc/rgb2rgb.h"
|
|
#include "libvo/fastmemcpy.h"
|
|
#include "mp_msg.h"
|
|
#include "png.h"
|
|
|
|
static int pngPointer;
|
|
|
|
static void pngReadFN( png_structp pngstr,png_bytep buffer,png_size_t size )
|
|
{
|
|
char * p = pngstr->io_ptr;
|
|
memcpy( buffer,(char *)&p[pngPointer],size );
|
|
pngPointer+=size;
|
|
}
|
|
|
|
void decode_mpng(
|
|
unsigned char *encoded,
|
|
int encoded_size,
|
|
unsigned char *decoded,
|
|
int width,
|
|
int height,
|
|
int bytes_per_pixel)
|
|
{
|
|
png_structp png;
|
|
png_infop info;
|
|
png_infop endinfo;
|
|
png_bytep data;
|
|
png_bytep * row_p;
|
|
png_uint_32 png_width,png_height;
|
|
char * palette = NULL;
|
|
int depth,color;
|
|
png_uint_32 i;
|
|
|
|
/* currently supporting only 24 and 32bpp */
|
|
if ((bytes_per_pixel != 3) && (bytes_per_pixel != 4))
|
|
{
|
|
/* is this memset really needed? */
|
|
memset(decoded, 0, width*height*bytes_per_pixel);
|
|
return;
|
|
}
|
|
|
|
png=png_create_read_struct( PNG_LIBPNG_VER_STRING,NULL,NULL,NULL );
|
|
info=png_create_info_struct( png );
|
|
endinfo=png_create_info_struct( png );
|
|
|
|
pngPointer=8;
|
|
png_set_read_fn( png,encoded,pngReadFN );
|
|
png_set_sig_bytes( png,8 );
|
|
png_read_info( png,info );
|
|
png_get_IHDR( png,info,&png_width,&png_height,&depth,&color,NULL,NULL,NULL );
|
|
|
|
png_set_bgr( png );
|
|
|
|
#if 0
|
|
switch( info->color_type )
|
|
{
|
|
case PNG_COLOR_TYPE_GRAY_ALPHA: printf( "[png] used GrayA -> stripping alpha channel\n" ); break;
|
|
case PNG_COLOR_TYPE_GRAY: printf( "[png] used Gray -> rgb\n" ); break;
|
|
case PNG_COLOR_TYPE_PALETTE: printf( "[png] used palette -> rgb\n" ); break;
|
|
case PNG_COLOR_TYPE_RGB_ALPHA: printf( "[png] used RGBA -> stripping alpha channel\n" ); break;
|
|
case PNG_COLOR_TYPE_RGB: printf( "[png] read rgb datas.\n" ); break;
|
|
}
|
|
#endif
|
|
|
|
if ( info->color_type == PNG_COLOR_TYPE_RGB ) data=decoded;
|
|
else data=(png_bytep)malloc( png_get_rowbytes( png,info ) * height );
|
|
|
|
row_p=(png_bytep*)malloc( sizeof( png_bytep ) * png_height );
|
|
for ( i=0; i < png_height; i++ ) row_p[i]=&data[png_get_rowbytes( png,info ) * i];
|
|
png_read_image( png,row_p );
|
|
free( row_p );
|
|
|
|
switch( info->color_type )
|
|
{
|
|
case PNG_COLOR_TYPE_GRAY_ALPHA:
|
|
mp_msg( MSGT_DECVIDEO,MSGL_INFO,"Sorry gray scaled png with alpha channel not supported at moment.\n" );
|
|
free( data );
|
|
break;
|
|
case PNG_COLOR_TYPE_GRAY:
|
|
/* constant 256 colors */
|
|
palette=malloc( 1024 );
|
|
for ( i=0;i < 256;i++ ) palette[(i*4)]=palette[(i*4)+1]=palette[(i*4)+2]=(char)i;
|
|
if (bytes_per_pixel == 4)
|
|
palette8torgb32( data,decoded,png_width * png_height,palette );
|
|
else
|
|
palette8torgb24( data,decoded,png_width * png_height,palette );
|
|
free( data );
|
|
break;
|
|
case PNG_COLOR_TYPE_PALETTE:
|
|
{
|
|
int cols;
|
|
unsigned char * pal;
|
|
png_get_PLTE( png,info,(png_colorp*)&pal,&cols );
|
|
palette=calloc( 1,cols*4 );
|
|
mp_dbg(MSGT_DECVIDEO, MSGL_DBG2, "[mPNG] palette. used colors: %d\n", cols);
|
|
for ( i=0;i < cols;i++ )
|
|
{
|
|
palette[(i*4) ]=pal[(i*3)+2];
|
|
palette[(i*4)+1]=pal[(i*3)+1];
|
|
palette[(i*4)+2]=pal[(i*3) ];
|
|
}
|
|
}
|
|
if (bytes_per_pixel == 4)
|
|
palette8torgb32( data,decoded,png_width * png_height,palette );
|
|
else
|
|
palette8torgb24( data,decoded,png_width * png_height,palette );
|
|
free( data );
|
|
break;
|
|
case PNG_COLOR_TYPE_RGB_ALPHA:
|
|
if (bytes_per_pixel == 4)
|
|
memcpy(decoded, data, png_width * png_height * 4);
|
|
else
|
|
rgb32to24( data,decoded,png_width * png_height * 4 );
|
|
free( data );
|
|
break;
|
|
}
|
|
|
|
if ( palette ) free( palette );
|
|
|
|
png_read_end( png,endinfo );
|
|
png_destroy_read_struct( &png,&info,&endinfo );
|
|
}
|
|
|