mpv/gui/bitmap.c

168 lines
4.3 KiB
C
Raw Normal View History

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "mp_msg.h"
#include "help_mp.h"
#include "bitmap.h"
#include "libavcodec/avcodec.h"
#include "libavutil/intreadwrite.h"
#include "libvo/fastmemcpy.h"
static int pngRead( unsigned char * fname,txSample * bf )
{
int decode_ok;
void *data;
int len;
AVCodecContext *avctx;
AVFrame *frame;
FILE *fp=fopen( fname,"rb" );
if ( !fp )
{
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[png] file read error ( %s )\n",fname );
return 1;
}
fseek(fp, 0, SEEK_END);
len = ftell(fp);
if (len > 50 * 1024 * 1024) return 2;
data = malloc(len + FF_INPUT_BUFFER_PADDING_SIZE);
fseek(fp, 0, SEEK_SET);
fread(data, len, 1, fp);
fclose(fp);
avctx = avcodec_alloc_context();
frame = avcodec_alloc_frame();
avcodec_register_all();
avcodec_open(avctx, avcodec_find_decoder(CODEC_ID_PNG));
avcodec_decode_video(avctx, frame, &decode_ok, data, len);
memset(bf, 0, sizeof(*bf));
switch (avctx->pix_fmt) {
case PIX_FMT_GRAY8: bf->BPP = 8; break;
case PIX_FMT_GRAY16BE: bf->BPP = 16; break;
case PIX_FMT_RGB24: bf->BPP = 24; break;
case PIX_FMT_RGB32: bf->BPP = 32; break;
default: bf->BPP = 0; break;
}
if (decode_ok && bf->BPP) {
int bpl;
bf->Width = avctx->width; bf->Height = avctx->height;
bpl = bf->Width * (bf->BPP / 8);
bf->ImageSize = bpl * bf->Height;
bf->Image = malloc(bf->ImageSize);
memcpy_pic(bf->Image, frame->data[0], bpl, bf->Height, bpl, frame->linesize[0]);
}
avcodec_close(avctx);
av_freep(&frame);
av_freep(&avctx);
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[png] filename: %s.\n",fname );
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[png] size: %dx%d bits: %d\n",bf->Width,bf->Height,bf->BPP );
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[png] imagesize: %lu\n",bf->ImageSize );
return !(decode_ok && bf->BPP);
}
static int conv24to32( txSample * bf )
{
unsigned char * tmpImage;
int i,c;
if ( bf->BPP == 24 )
{
tmpImage=bf->Image;
bf->ImageSize=bf->Width * bf->Height * 4;
bf->BPP=32;
if ( ( bf->Image=calloc( 1, bf->ImageSize ) ) == NULL )
{
free( tmpImage );
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[bitmap] not enough memory for image\n" );
return 1;
}
for ( c=0,i=0; c < bf->ImageSize; c += 4, i += 3)
{
*(uint32_t *)&bf->Image[c] = AV_RB24(&tmpImage[i]);
}
free( tmpImage );
}
return 0;
}
static void Normalize( txSample * bf )
{
int i;
#ifndef WORDS_BIGENDIAN
for ( i=0;i < (int)bf->ImageSize;i+=4 ) bf->Image[i+3]=0;
#else
for ( i=0;i < (int)bf->ImageSize;i+=4 ) bf->Image[i]=0;
#endif
}
static unsigned char tmp[512];
static unsigned char * fExist( unsigned char * fname )
{
FILE * fl;
unsigned char ext[][6] = { ".png\0",".PNG\0" };
int i;
fl=fopen( fname,"rb" );
if ( fl != NULL )
{
fclose( fl );
return fname;
}
for ( i=0;i<2;i++ )
{
snprintf( tmp,511,"%s%s",fname,ext[i] );
fl=fopen( tmp,"rb" );
if ( fl != NULL )
{
fclose( fl );
return tmp;
}
}
return NULL;
}
int bpRead( char * fname, txSample * bf )
{
fname=fExist( fname );
if ( fname == NULL ) return -2;
if ( pngRead( fname,bf ) )
{
mp_dbg( MSGT_GPLAYER,MSGL_FATAL,"[bitmap] unknown file type ( %s )\n",fname );
return -5;
}
if ( bf->BPP < 24 )
{
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[bitmap] Sorry, only 24 and 32 bpp bitmaps are supported.\n" );
return -1;
}
if ( conv24to32( bf ) ) return -8;
Normalize( bf );
return 0;
}
void Convert32to1( txSample * in,txSample * out,int adaptivlimit )
{
out->Width=in->Width;
out->Height=in->Height;
out->BPP=1;
out->ImageSize=(out->Width * out->Height + 7) / 8;
mp_dbg( MSGT_GPLAYER,MSGL_DBG2,"[c32to1] imagesize: %d\n",out->ImageSize );
out->Image=calloc( 1,out->ImageSize );
if ( out->Image == NULL ) mp_msg( MSGT_GPLAYER,MSGL_WARN,MSGTR_NotEnoughMemoryC32To1 );
{
int i,b,c=0; unsigned int * buf = NULL; unsigned char tmp = 0; int nothaveshape = 1;
buf=(unsigned int *)in->Image;
for ( b=0,i=0;i < (int)(out->Width * out->Height);i++ )
{
if ( (int)buf[i] != adaptivlimit ) tmp=( tmp >> 1 )|128;
else { tmp=tmp >> 1; buf[i]=nothaveshape=0; }
if ( b++ == 7 ) { out->Image[c++]=tmp; tmp=b=0; }
}
if ( b ) out->Image[c]=tmp;
if ( nothaveshape ) { free( out->Image ); out->Image=NULL; }
}
}