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,