Mercurial > mplayer.hg
view libvo/vo_aa.c @ 18676:e813d4014acd
fix image adjustment (brightness etc.) for yuv=6
author | reimar |
---|---|
date | Sun, 11 Jun 2006 10:08:40 +0000 |
parents | f580a7755ac5 |
children | 8579acff875e |
line wrap: on
line source
/* * MPlayer * * Video driver for AAlib - 1.0 * * by Folke Ashberg <folke@ashberg.de> * * Code started: Sun Aug 12 2001 * Version 1.0 : Thu Aug 16 2001 * */ #include <stdio.h> #include <stdlib.h> #include <sys/stat.h> #include <unistd.h> #include <limits.h> #include <math.h> #include <stdarg.h> #include <time.h> #include <string.h> #include <errno.h> #include "config.h" #include "video_out.h" #include "video_out_internal.h" #include "aspect.h" #include "postproc/swscale.h" #include "libmpcodecs/vf_scale.h" #include "font_load.h" #include "sub.h" #include "osdep/keycodes.h" #include <aalib.h> #include "subopt-helper.h" #include "help_mp.h" #include "mp_msg.h" #define MESSAGE_DURATION 3 #define MESSAGE_SIZE 512 #define MESSAGE_DEKO " +++ %s +++ " static vo_info_t info = { "AAlib", "aa", "Alban Bedel <albeu@free.fr> and Folke Ashberg <folke@ashberg.de>", "" }; LIBVO_EXTERN(aa) /* aa's main context we use */ aa_context *c; aa_renderparams *p; static int fast =0; /* used for the sws */ static uint8_t * image[3]; static int image_stride[3]; /* image infos */ static int image_format; static int image_width; static int image_height; static int image_x, image_y; static int screen_x, screen_y; static int screen_w, screen_h; static int src_width; static int src_height; /* osd stuff */ time_t stoposd = 0; static int showosdmessage = 0; char osdmessagetext[MESSAGE_SIZE]; char posbar[MESSAGE_SIZE]; static int osdx, osdy; static int osd_text_length = 0; int aaconfigmode=1; #ifdef USE_OSD font_desc_t* vo_font_save = NULL; #endif static struct SwsContext *sws=NULL; /* our version of the playmodes :) */ extern void mplayer_put_key(int code); /* to disable stdout outputs when curses/linux mode */ extern int quiet; /* configuration */ int aaopt_osdcolor = AA_SPECIAL; int aaopt_subcolor = AA_SPECIAL; extern struct aa_hardware_params aa_defparams; extern struct aa_renderparams aa_defrenderparams; void resize(void){ /* * this function is called by aa lib if windows resizes * further during init, because here we have to calculate * a little bit */ aa_resize(c); aspect_save_screenres(aa_imgwidth(c),aa_imgheight(c)); image_height = aa_imgheight(c); //src_height; image_width = aa_imgwidth(c); //src_width; aspect(&image_width,&image_height,A_ZOOM); image_x = (aa_imgwidth(c) - image_width) / 2; image_y = (aa_imgheight(c) - image_height) / 2; screen_w = image_width * aa_scrwidth(c) / aa_imgwidth(c); screen_h = image_height * aa_scrheight(c) / aa_imgheight(c); screen_x = (aa_scrwidth(c) - screen_w) / 2; screen_y = (aa_scrheight(c) - screen_h) / 2; if(sws) sws_freeContext(sws); sws = sws_getContextFromCmdLine(src_width,src_height,image_format, image_width,image_height,IMGFMT_Y8); image[0] = aa_image(c) + image_y * aa_imgwidth(c) + image_x; image[1] = NULL; image[2] = NULL; image_stride[0] = aa_imgwidth(c); image_stride[1] = 0; image_stride[2] = 0; showosdmessage=0; } void osdmessage(int duration, int deko, char *fmt, ...) { /* * for outputting a centered string at the bottom * of our window for a while */ va_list ar; char m[MESSAGE_SIZE]; unsigned int old_len = strlen(osdmessagetext); va_start(ar, fmt); vsprintf(m, fmt, ar); va_end(ar); if (deko==1) sprintf(osdmessagetext, MESSAGE_DEKO , m); else strcpy(osdmessagetext, m); if(old_len > strlen(osdmessagetext)) { memset(c->textbuffer + osdy * aa_scrwidth(c) + osdx,' ',old_len); memset(c->attrbuffer + osdy * aa_scrwidth(c) + osdx,0,old_len); } showosdmessage=1; stoposd = time(NULL) + duration; osdx=(aa_scrwidth(c) / 2) - (strlen(osdmessagetext) / 2 ) ; posbar[0]='\0'; } void osdpercent(int duration, int deko, int min, int max, int val, char * desc, char * unit) { /* * prints a bar for setting values */ float step; int where; int i; step=(float)aa_scrwidth(c) /(float)(max-min); where=(val-min)*step; osdmessage(duration,deko,"%s: %i%s",desc, val, unit); posbar[0]='|'; posbar[aa_scrwidth(c)-1]='|'; for (i=0;i<aa_scrwidth(c);i++){ if (i==where) posbar[i]='#'; else posbar[i]='-'; } if (where!=0) posbar[0]='|'; if (where!=(aa_scrwidth(c)-1) ) posbar[aa_scrwidth(c)-1]='|'; posbar[aa_scrwidth(c)]='\0'; } void printosdtext(void) { if(osd_text_length > 0 && !vo_osd_text) { memset(c->textbuffer,' ',osd_text_length); memset(c->attrbuffer,0,osd_text_length); osd_text_length = 0; } /* * places the mplayer status osd */ if (vo_osd_text && vo_osd_text[0] != 0) { int len; if(vo_osd_text[0] < 32) { len = strlen(__sub_osd_names_short[vo_osd_text[0]]) + strlen(vo_osd_text+1) + 2; aa_printf(c, 0, 0 , aaopt_osdcolor, "%s %s ", __sub_osd_names_short[vo_osd_text[0]], vo_osd_text+1); } else { len = strlen(vo_osd_text) + 1; aa_printf(c, 0, 0 , aaopt_osdcolor, "%s ",vo_osd_text); } if(len < osd_text_length) { memset(c->textbuffer + len,' ',osd_text_length - len); memset(c->attrbuffer + len,0,osd_text_length - len); } osd_text_length = len; } } void printosdprogbar(void){ /* print mplayer osd-progbar */ if (vo_osd_progbar_type!=-1){ osdpercent(1,1,0,255,vo_osd_progbar_value, __sub_osd_names[vo_osd_progbar_type], ""); } } static int config(uint32_t width, uint32_t height, uint32_t d_width, uint32_t d_height, uint32_t flags, char *title, uint32_t format) { /* * main init * called by mplayer */ int i; aspect_save_orig(width,height); aspect_save_prescale(d_width,d_height); src_height = height; src_width = width; image_format = format; /* nothing will change its size, be we need some values initialized */ resize(); #ifdef USE_OSD /* now init out own 'font' (to use vo_draw_text_sub without edit them) */ if(!vo_font_save) vo_font_save = vo_font; if(vo_font == vo_font_save) { vo_font=malloc(sizeof(font_desc_t));//if(!desc) return NULL; memset(vo_font,0,sizeof(font_desc_t)); vo_font->pic_a[0]=malloc(sizeof(raw_file)); memset(vo_font->pic_a[0],0,sizeof(raw_file)); vo_font->pic_b[0]=malloc(sizeof(raw_file)); memset(vo_font->pic_b[0],0,sizeof(raw_file)); #ifdef HAVE_FREETYPE vo_font->dynamic = 0; #endif vo_font->spacewidth=1; vo_font->charspace=0; vo_font->height=1; vo_font->pic_a[0]->bmp=malloc(255); vo_font->pic_a[0]->pal=NULL; vo_font->pic_b[0]->bmp=malloc(255); vo_font->pic_b[0]->pal=NULL; vo_font->pic_a[0]->w=1; vo_font->pic_a[0]->h=1; for (i=0; i<255; i++){ vo_font->width[i]=1; vo_font->font[i]=0; vo_font->start[i]=i; vo_font->pic_a[0]->bmp[i]=i; vo_font->pic_b[0]->bmp[i]=i; } } #endif /* say hello */ osdmessage(5, 1, "Welcome to ASCII ART MPlayer"); mp_msg(MSGT_VO,MSGL_V,"VO: [aa] screendriver: %s\n", c->driver->name); mp_msg(MSGT_VO,MSGL_V,"VO: [aa] keyboarddriver: %s\n", c->kbddriver->name); mp_msg(MSGT_VO,MSGL_INFO, "\n" "Important suboptions\n" "\textended use use all 256 characters\n" "\teight use eight bit ascii\n" "\tdriver set recommended aalib driver (X11,curses,linux)\n" "\thelp to see all options provided by aalib\n" "\n" "AA-MPlayer Keys\n" "\t1 : contrast -\n" "\t2 : contrast +\n" "\t3 : brightness -\n" "\t4 : brightness +\n" "\t5 : fast rendering\n" "\t6 : dithering\n" "\t7 : invert image\n" "\ta : toggles between aa and mplayer control\n" "\n" "All other keys are MPlayer defaults.\n" ); return 0; } static int query_format(uint32_t format) { /* * ...are we able to... ? * called by mplayer * All input format supported by the sws */ switch(format){ case IMGFMT_YV12: case IMGFMT_I420: case IMGFMT_IYUV: case IMGFMT_IYU2: case IMGFMT_BGR32: case IMGFMT_BGR24: case IMGFMT_BGR16: case IMGFMT_BGR15: case IMGFMT_RGB32: case IMGFMT_RGB24: case IMGFMT_Y8: case IMGFMT_Y800: return VFCAP_CSP_SUPPORTED|VFCAP_SWSCALE #ifdef USE_OSD | VFCAP_OSD #endif ; } return 0; } static int draw_frame(uint8_t *src[]) { int stride[3] = { 0 , 0 , 0 }; switch(image_format) { case IMGFMT_BGR15: case IMGFMT_BGR16: stride[0] = src_width*2; break; case IMGFMT_IYU2: case IMGFMT_BGR24: stride[0] = src_width*3; break; case IMGFMT_BGR32: stride[0] = src_width*4; break; } sws_scale_ordered(sws,src,stride,0,src_height,image,image_stride); /* Now 'ASCIInate' the image */ if (fast) aa_fastrender(c, screen_x, screen_y, screen_w + screen_x, screen_h + screen_y ); else aa_render(c, p,screen_x, screen_y, screen_w + screen_x, screen_h + screen_y ); return 0; } static int draw_slice(uint8_t *src[], int stride[], int w, int h, int x, int y) { int dx1 = screen_x + (x * screen_w / src_width); int dy1 = screen_y + (y * screen_h / src_height); int dx2 = screen_x + ((x+w) * screen_w / src_width); int dy2 = screen_y + ((y+h) * screen_h / src_height); sws_scale_ordered(sws,src,stride,y,h,image,image_stride); /* Now 'ASCIInate' the image */ if (fast) aa_fastrender(c, dx1, dy1, dx2, dy2 ); else aa_render(c, p,dx1, dy1, dx2, dy2 ); return 0; } static void flip_page(void) { /* do we have to put *our* (messages, progbar) osd to aa's txtbuf ? */ if (showosdmessage) { if (time(NULL)>=stoposd ) { showosdmessage=0; if(osdmessagetext) { memset(c->textbuffer + osdy * aa_scrwidth(c) + osdx,' ',strlen(osdmessagetext)); memset(c->attrbuffer + osdy * aa_scrwidth(c) + osdx ,0,strlen(osdmessagetext)); osdmessagetext[0] = '\0'; } if(posbar) { memset(c->textbuffer + (osdy+1) * aa_scrwidth(c),' ',strlen(posbar)); memset(c->attrbuffer + (osdy+1) * aa_scrwidth(c),0,strlen(posbar)); } } else { /* update osd */ aa_puts(c, osdx, osdy, AA_SPECIAL, osdmessagetext); /* posbar? */ if (posbar[0]!='\0') aa_puts(c, 0, osdy + 1, AA_SPECIAL, posbar); } } /* OSD time & playmode , subtitles */ #ifdef USE_OSD printosdtext(); #endif /* print out */ aa_flush(c); } static void check_events(void) { /* * any events? * called by show_image and mplayer */ int key; while ((key=aa_getevent(c,0))!=AA_NONE ){ if (key>255){ /* some conversations */ switch (key) { case AA_UP: mplayer_put_key(KEY_UP); break; case AA_DOWN: mplayer_put_key(KEY_DOWN); break; case AA_LEFT: mplayer_put_key(KEY_LEFT); break; case AA_RIGHT: mplayer_put_key(KEY_RIGHT); break; case AA_ESC: mplayer_put_key(KEY_ESC); break; case 65765: mplayer_put_key(KEY_PAGE_UP); break; case 65766: mplayer_put_key(KEY_PAGE_DOWN); break; default: continue; /* aa lib special key */ break; } } if (key=='a' || key=='A'){ aaconfigmode=!aaconfigmode; osdmessage(MESSAGE_DURATION, 1, "aa config mode is now %s", aaconfigmode==1 ? "on. use keys 5-7" : "off"); } if (aaconfigmode==1) { switch (key) { /* AA image controls */ case '5': fast=!fast; osdmessage(MESSAGE_DURATION, 1, "Fast mode is now %s", fast==1 ? "on" : "off"); break; case '6': if (p->dither==AA_FLOYD_S){ p->dither=AA_NONE; osdmessage(MESSAGE_DURATION, 1, "Dithering: Off"); }else if (p->dither==AA_NONE){ p->dither=AA_ERRORDISTRIB; osdmessage(MESSAGE_DURATION, 1, "Dithering: Error Distribution"); }else if (p->dither==AA_ERRORDISTRIB){ p->dither=AA_FLOYD_S; osdmessage(MESSAGE_DURATION, 1, "Dithering: Floyd Steinberg"); } break; case '7': p->inversion=!p->inversion; osdmessage(MESSAGE_DURATION, 1, "Invert mode is now %s", p->inversion==1 ? "on" : "off"); break; default : /* nothing if we're interested in? * the mplayer should handle it! */ mplayer_put_key(key); break; } }// aaconfigmode else mplayer_put_key(key); } } static void uninit(void) { /* * THE END */ if (strstr(c->driver->name,"Curses") || strstr(c->driver->name,"Linux")){ freopen("/dev/tty", "w", stderr); } #ifdef USE_OSD if(vo_font_save) { free(vo_font->pic_a[0]->bmp); free(vo_font->pic_a[0]); free(vo_font->pic_b[0]->bmp); free(vo_font->pic_b[0]); free(vo_font); vo_font = vo_font_save; vo_font_save = NULL; } #endif aa_close(c); } #ifdef USE_OSD static void draw_alpha(int x,int y, int w,int h, unsigned char* src, unsigned char *srca, int stride){ int i,j; for (i = 0; i < h; i++) { for (j = 0; j < w; j++) { if (src[i*stride+j] > 0) { c->textbuffer[x + j + (y+i)*aa_scrwidth(c)] = src[i*stride+j]; c->attrbuffer[x + j + (y+i)*aa_scrwidth(c)] = aaopt_subcolor; } } } } static void clear_alpha(int x0,int y0, int w,int h) { int l; for(l = 0 ; l < h ; l++) { memset(c->textbuffer + (y0 + l) * aa_scrwidth(c) + x0,' ',w); memset(c->attrbuffer + (y0 + l) * aa_scrwidth(c) + x0,0,w); } } #endif static void draw_osd(void){ #ifdef USE_OSD char * vo_osd_text_save; int vo_osd_progbar_type_save; printosdprogbar(); /* let vo_draw_text only write subtitle */ vo_osd_text_save=vo_osd_text; /* we have to save the osd_text */ vo_osd_text=NULL; vo_osd_progbar_type_save=vo_osd_progbar_type; vo_osd_progbar_type=-1; vo_remove_text(aa_scrwidth(c), aa_scrheight(c),clear_alpha); vo_draw_text(aa_scrwidth(c), aa_scrheight(c), draw_alpha); vo_osd_text=vo_osd_text_save; vo_osd_progbar_type=vo_osd_progbar_type_save; #endif } int getcolor(char * s){ int i; char * rest; if (s==NULL) return -1; i=strtol(s, &rest, 10); if ((rest==NULL || strlen(rest)==0) && i>=0 && i<=5) return i; if (!strcasecmp(s, "normal")) return AA_NORMAL; else if (!strcasecmp(s, "dim")) return AA_DIM; else if (!strcasecmp(s, "bold")) return AA_BOLD; else if (!strcasecmp(s, "boldfont")) return AA_BOLDFONT; else if (!strcasecmp(s, "special")) return AA_SPECIAL; else return -1; } static int parse_suboptions(const char *arg) { char *pseudoargv[4], *osdcolor = NULL, *subcolor = NULL, **strings, *helpmsg = NULL; int pseudoargc, displayhelp = 0, *booleans; opt_t extra_opts[] = { {"osdcolor", OPT_ARG_MSTRZ, &osdcolor, NULL, 0}, {"subcolor", OPT_ARG_MSTRZ, &subcolor, NULL, 0}, {"help", OPT_ARG_BOOL, &displayhelp, NULL, 0} }; opt_t *subopts = NULL, *p; char *strings_list[] = {"-driver", "-kbddriver", "-mousedriver", "-font", "-width", "-height", "-minwidth", "-minheight", "-maxwidth", "-maxheight", "-recwidth", "-recheight", "-bright", "-contrast", "-gamma", "-dimmul", "-boldmul", "-random" }; char *booleans_list[] = {"-dim", "-bold", "-reverse", "-normal", "-boldfont", "-inverse", "-extended", "-eight", "-dither", "-floyd_steinberg", "-error_distribution"}; char *nobooleans_list[] = {"-nodim", "-nobold", "-noreverse", "-nonormal", "-noboldfont", "-noinverse", "-noextended", "-noeight", "-nodither", "-nofloyd_steinberg", "-noerror_distribution"}; const int nstrings = sizeof(strings_list) / sizeof(char*); const int nbooleans = sizeof(booleans_list) / sizeof(int); const int nextra_opts = sizeof(extra_opts) / sizeof(opt_t); const int nsubopts = nstrings + nbooleans + nextra_opts; int i, retval = 0; subopts = calloc(nsubopts + 1, sizeof(opt_t)); strings = calloc(nstrings, sizeof(char*)); booleans = calloc(nbooleans, sizeof(int)); p = subopts; for (i=0; i<nstrings; i++, p++) { p->name = strings_list[i] + 1; // skip '-' p->type = OPT_ARG_MSTRZ; p->valp = &strings[i]; } for (i=0; i<nbooleans; i++, p++) { p->name = booleans_list[i] + 1; p->type = OPT_ARG_BOOL; p->valp = &booleans[i]; } memcpy(p, extra_opts, sizeof(extra_opts)); retval = subopt_parse(arg, subopts); if (retval == 0 && displayhelp) { helpmsg = strdup(aa_help); for (i=0; i<(signed)strlen(helpmsg); i++) if (helpmsg[i] == '-') helpmsg[i] = ' '; mp_msg(MSGT_VO, MSGL_INFO, MSGTR_VO_AA_HelpHeader); mp_msg(MSGT_VO, MSGL_INFO, "%s\n\n", helpmsg); mp_msg(MSGT_VO, MSGL_INFO, MSGTR_VO_AA_AdditionalOptions); retval = -1; } if (retval == 0) { pseudoargv[3] = NULL; for (i=0; i<nstrings; i++) { pseudoargc = 3; // inside loop because aalib changes it if (strings[i] != NULL) { pseudoargv[1] = strings_list[i]; pseudoargv[2] = strings[i]; aa_parseoptions(&aa_defparams, &aa_defrenderparams, &pseudoargc, pseudoargv) != 1; } } pseudoargv[2] = NULL; for (i=0; i<nbooleans; i++) { pseudoargc = 2; if (booleans[i]) pseudoargv[1] = booleans_list[i]; else pseudoargv[1] = nobooleans_list[i]; aa_parseoptions(&aa_defparams, &aa_defrenderparams, &pseudoargc, pseudoargv) != 1; } if (osdcolor) aaopt_osdcolor = getcolor(osdcolor); if (subcolor) aaopt_subcolor = getcolor(subcolor); } if (subopts) free(subopts); if (booleans) free(booleans); if (strings) { for (i=0; i<nstrings; i++) if (strings[i]) free(strings[i]); free(strings); } if (osdcolor) free(osdcolor); if (subcolor) free(subcolor); if (helpmsg) free(helpmsg); return retval; } static int preinit(const char *arg) { char * hidis = NULL; struct stat sbuf; int fd, vt, major, minor; FILE * fp; char fname[12]; extern aa_linkedlist *aa_displayrecommended; if(arg) { if (parse_suboptions(arg) != 0) return ENOSYS; } /* initializing of aalib */ hidis=aa_getfirst(&aa_displayrecommended); if ( hidis==NULL ){ /* check /dev/vcsa<vt> */ /* check only, if no driver is explicit set */ fd = dup (fileno (stderr)); fstat (fd, &sbuf); major = sbuf.st_rdev >> 8; vt = minor = sbuf.st_rdev & 0xff; close (fd); sprintf (fname, "/dev/vcsa%2.2i", vt); fp = fopen (fname, "w+"); if (fp==NULL){ fprintf(stderr,"VO: [aa] cannot open %s for writing," "so we'll not use linux driver\n", fname); aa_recommendlowdisplay("linux"); aa_recommendhidisplay("curses"); aa_recommendhidisplay("X11"); }else fclose(fp); } else aa_recommendhidisplay(hidis); c = aa_autoinit(&aa_defparams); if (c == NULL) { mp_msg(MSGT_VO,MSGL_ERR,"Cannot initialize aalib\n"); return VO_ERROR; } if (!aa_autoinitkbd(c,0)) { mp_msg(MSGT_VO,MSGL_ERR,"Cannot initialize keyboard\n"); aa_close(c); return VO_ERROR; } aa_resizehandler(c, (void *)resize); aa_hidecursor(c); p = aa_getrenderparams(); if ((strstr(c->driver->name,"Curses")) || (strstr(c->driver->name,"Linux"))){ freopen("/dev/null", "w", stderr); /* disable console blanking */ printf("\033[9;0]"); } memset(image,0,3*sizeof(uint8_t)); osdmessagetext[0] = '\0'; osdx = osdy = 0; return 0; } static int control(uint32_t request, void *data, ...) { switch (request) { case VOCTRL_QUERY_FORMAT: return query_format(*((uint32_t*)data)); case VOCTRL_SET_EQUALIZER: { va_list ap; int val; va_start(ap, data); val = va_arg(ap, int); va_end(ap); if(strcmp((char*)data,"contrast") == 0) p->contrast = ( val + 100 ) * 64 / 100; else if(strcmp((char*)data,"brightness") == 0) p->bright = ( val + 100) * 128 / 100; return VO_TRUE; } case VOCTRL_GET_EQUALIZER: { va_list ap; int* val; va_start(ap, data); val = va_arg(ap, int*); va_end(ap); if(strcmp((char*)data,"contrast") == 0) *val = (p->contrast - 64) * 100 / 64; else if(strcmp((char*)data,"brightness") == 0) *val = (p->bright - 128) * 100 / 128; return VO_TRUE; } } return VO_NOTIMPL; }