Camera Hal3实现水印或者弹幕的功能

//#define LOG_NDEBUG 0 #define LOG_TAG "WaterMark" #include <fstream> #include <iostream> #include <vector> #include <ctime> #include <sys/mman.h> #include <libyuv.h> #include <utils/Errors.h> #include <log/log.h> #include <sched.h> #include <stdio.h> #include <stdlib.h> #include <fcntl.h> #include "watermark.h" namespace android { namespace IspCamera { WaterMark::WaterMark() { for(int i = 0; i < MAX_ICON_NUM; i++) { icon_arry[i].id = i; icon_arry[i].width = 0; icon_arry[i].height = 0; icon_arry[i].y = NULL; icon_arry[i].alph = NULL; icon_arry[i].c = NULL; } msinglewm = {200, 200, "0123456789ABC"}; ALOGE("construct WaterMark exit\n"); } WaterMark::~WaterMark() { for(int i = 0; i < MAX_ICON_NUM; i++){ if(icon_arry[i].y != NULL) free(icon_arry[i].y); } ALOGE("delete WaterMark\n"); } void WaterMark::ARGB8888ToYUV420sp(unsigned char *src_p, int width, int height, unsigned char *dest_y, unsigned char *dest_c, unsigned char *dest_alph) { int i,j; for(i = 0; i < (int)height; i++) { if((i&1) == 0) { for(j= 0; j< (int)width; j++) { *dest_y = (299*src_p[2]+587*src_p[1]+114*src_p[0])/1000; if((j&1) == 0) *dest_c++ = 128+(564*(src_p[0]-*dest_y)/1000); //cb else *dest_c++ = 128+(713*(src_p[2]-*dest_y)/1000); //cr *dest_alph++ = src_p[3]; src_p +=4; dest_y++; } } else { for(j= 0; j< (int)width; j++) { *dest_y = (299*src_p[2]+587*src_p[1]+114*src_p[0])/1000; *dest_alph++ = src_p[3]; src_p +=4; dest_y++; } } } return; } int WaterMark::InitwWaterMark() { FILE *fp = NULL; int i; int pic_width = 0, pic_height = 0; char filename[256]; unsigned char *argb_buf = NULL; for(i = 0; i < MAX_ICON_NUM; i++) { sprintf(filename, "%s/argb8888_24x32_%d.bmp", WATERMARK_BMP_DIR, i); ALOGE("open file %s\n", filename); fp = fopen(filename, "r"); if(NULL == fp) { ALOGE("Fail to open file %s(%s)!", filename, strerror(errno)); goto OPEN_FILE_ERR; } icon_arry[i].id = i; fseek(fp, 18, SEEK_SET); fread(&pic_width, 1, 4, fp); fread(&pic_height, 1, 4, fp); icon_arry[i].width = abs(pic_width); icon_arry[i].height = abs(pic_height); ALOGE("real width=%d(%d), height=%d(%d)\n", icon_arry[i].width, pic_width, icon_arry[i].height, pic_height); icon_arry[i].y = (unsigned char*)malloc(icon_arry[i].width * icon_arry[i].height*5/2); if(NULL == icon_arry[i].y) { ALOGE("malloc picture yuv fail !\n"); fclose(fp); goto ALLOC_Y_ERR; } memset(icon_arry[i].y, 0xff, (icon_arry[i].width * icon_arry[i].height*5/2)); icon_arry[i].alph = icon_arry[i].y + icon_arry[i].width * icon_arry[i].height; icon_arry[i].c = icon_arry[i].alph + icon_arry[i].width * icon_arry[i].height; argb_buf = (unsigned char *)malloc(icon_arry[i].width * icon_arry[i].height * 4); if (NULL == argb_buf) { ALOGE("malloc bmp buf fail !\n"); fclose(fp); goto ALLOC_BUF_ERR; } fseek(fp, 54, SEEK_SET); fread(argb_buf, icon_arry[i].width * icon_arry[i].height * 4, 1, fp); ARGB8888ToYUV420sp(argb_buf, icon_arry[i].width, icon_arry[i].height, icon_arry[i].y, icon_arry[i].c, icon_arry[i].alph); fclose(fp); free(argb_buf); } return 0; ALLOC_BUF_ERR: ALLOC_Y_ERR: OPEN_FILE_ERR: for(i = 0; i < MAX_ICON_NUM; ++i) { if(icon_arry[i].y != NULL) { free(icon_arry[i].y); icon_arry[i].y = NULL; } } return -1; } int WaterMark::GetWordNum(char val) { int i; int cnt = 0; for(i=7; i>=0; i--) { if(!(val & (1<<i))) break; else cnt++; } if (cnt == 0) return 1; return cnt; } #define DEFAULT_NUM 36 int WaterMark::GetWordIndex(char *data, int wordlen) { int index = DEFAULT_NUM; if(wordlen == 1) { if (*data >= '0' && *data <= '9') { ALOGE("choose numben 0 ~ 9\n"); index = (*data - 48); //落在数组下标0开始 } else if (*data >= 'A' && *data <= 'Z') { ALOGE("choose A~Z\n"); index = (*data - 55); //落在数组下标10开始 } else if (*data == ' ') { ALOGE("choose kong\n"); index = 36; } else if (*data == '-') { ALOGE("choose -\n"); index = 37; } else if (*data == '.') { ALOGE("choose .\n"); index = DEFAULT_NUM; } else if (*data == ':') { ALOGE("choose eng :\n"); index = 38; } else if (*data == '/') { ALOGE("choose /\n"); index = 39; } else { ALOGE("not match %s , use DEFAULT_NUM\n", data); index = DEFAULT_NUM; } } else if (wordlen == 3) { if(!memcmp(data, "京", 3)) { ALOGE("choose %s\n", "京"); index = 40; } else if (!memcmp(data, "沪", 3)) { ALOGE("choose %s\n", "沪"); index = 41; } else if (!memcmp(data, "：", 3)) { ALOGE("choose chn :\n"); index = 38; } else { ALOGE("choose DEFAULT_NUM\n"); index = DEFAULT_NUM; } } ALOGE("\twordlen -> %d, index -> %d\n", wordlen, index); return index; } void WaterMark::Yuv420spBlending(unsigned char *bg_y, unsigned char *bg_c, int bg_width, int bg_height, int left, unsigned int top, int fg_width, int fg_height, unsigned char *fg_y, unsigned char *fg_c, unsigned char *alph) { unsigned char *bg_y_p = NULL; unsigned char *bg_c_p = NULL; int i = 0; int j = 0; bg_y_p = bg_y + top * bg_width + left; bg_c_p = bg_c + (top >>1)*bg_width + left; for(i = 0; i<(int)fg_height; i++) { if((i&1) == 0) { for(j=0; j< (int)fg_width; j++) { *bg_y_p = ((256 - *alph)*(*bg_y_p) + (*fg_y++)*(*alph))>>8; *bg_c_p = ((256 - *alph)*(*bg_c_p) + (*fg_c++)*(*alph))>>8; alph++; bg_y_p++; bg_c_p++; } bg_c_p = bg_c_p + bg_width - fg_width; } else { for(j=0; j< (int)fg_width; j++) { *bg_y_p = ((256 - *alph)*(*bg_y_p) + (*fg_y++)*(*alph))>>8; alph++; bg_y_p++; } } bg_y_p = bg_y_p + bg_width - fg_width; } } int WaterMark::PushWm2bg(unsigned char *bg_y_buf, unsigned char *bg_c_buf, int bg_width, int bg_height, struct disp_content *single_wm) { int pos_x; struct backgroud_info bg_info; int buflen, wordlen; char *p_content; int pic_index; int i; bg_info.width = bg_width; bg_info.height = bg_height; bg_info.y = bg_y_buf; bg_info.c = bg_c_buf; p_content = single_wm->content; buflen = strlen(single_wm->content); pos_x = single_wm->x; ALOGE("\nshow: %s(%d Bytes):\n", p_content, buflen); for (i = 0; i < strlen(single_wm->content); i++) { wordlen = GetWordNum(p_content[0]); buflen -= wordlen; if(buflen < 0) break; pic_index = GetWordIndex(p_content, wordlen); Yuv420spBlending(bg_info.y, bg_info.c, bg_info.width, bg_info.height, pos_x,