optimize

houxiao

2017-04-27 071d3b01a6793ee8d0eb747ce965c33653cbc237

 VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp

@@ -11,13 +11,16 @@


#include <vector>



#include "PbFaceList.pb.h"

#include <PbFaceList.pb.h>



#include <libyuv.h>

#include <MediaHelper.h>



#define SUBSAMPLE(v, a) ((((v) + (a) - 1)) / (a))



//#define YUV420_TO_RGB888 1

#define YUV420_TO_RGB565 1

//#define YUV420_TO_RGB565 1

#define YUV420_TO_ARGB8888 1



struct FcPmBreackerContext

{

@@ -92,6 +95,23 @@
                            frameRGB, 2 * src_width,

                            src_width, src_height);

        frameRGBSize = src_height * src_width * 2;

        return true;

    }



    bool convertYUV420ToARGB8888()

    {

        int src_height = height;

        int src_width = width;

        const uint8* src_y = (const uint8*)(frameYUV);

        const uint8* src_u = (const uint8*)(src_y + (src_height * src_width));

        const uint8* src_v = (const uint8*)(src_u + (src_height * src_width / 4));



        libyuv::I420ToARGB(src_y, src_width,

                             src_u, SUBSAMPLE(src_width, 2),

                             src_v, SUBSAMPLE(src_width, 2),

                             frameRGB, 4 * src_width,

                             src_width, src_height);

        frameRGBSize = src_height * src_width * 4;

        return true;

    }

};

@@ -194,7 +214,7 @@
      PbFaceList_FaceListImage& pbFaceListImage(*(pbFaceList.add_images()));

      pbFaceListImage.set_idx(i);

      pbFaceListImage.set_size(s);

      pbFaceListImage.set_type(PbFaceList_FaceListImage_ImageType_IT_Y);

      pbFaceListImage.set_type(PbFaceList_FaceListImage_ImageType_MBFT_Y8);

      pbFaceListImage.set_width(roiMat.cols);

      pbFaceListImage.set_height(roiMat.rows);

      pbFaceListImage.set_top_left_x(faceRect.leftTop.x);

@@ -391,4 +411,67 @@
    return true;

}



#elif YUV420_TO_ARGB8888



bool FaceCache::getFaceListImage(std::vector<NativeImgIdx>& imgIdxes, uint8_t* buffImg, size_t& buffImgMaxSize)

{

    FcPmBreackerContext& ctx(*(FcPmBreackerContext*)_ctx);

    if (!ctx.dataAvailable)

        return false;



    if (ctx.frameRGBSize == 0)

    {

        //#todo should optimize not convert the whole image

        if (! ctx.convertYUV420ToARGB8888())

            return false;

    }



    imgIdxes.clear();

    size_t offset = 0;



    cv::Mat rgbMat(cv::Size(ctx.width, ctx.height), CV_8UC4, ctx.frameRGB);

    for (st_ff_vect_t::const_iterator ffiter = ctx.faceFeatures.begin(); ffiter != ctx.faceFeatures.end(); ++ffiter)

    {

        const FaceRect& faceRect(ffiter->rect);



        cv::Mat roiMat1(rgbMat, cv::Rect(faceRect.leftTop.x, faceRect.leftTop.y, faceRect.rightBottom.x - faceRect.leftTop.x, faceRect.rightBottom.y - faceRect.leftTop.y));

        cv::Mat roiMat(roiMat1.clone()); // #todo copy data should be avoid!!!!



        NativeImgIdx imgidx;

        imgidx.offset = offset;

        imgidx.size = roiMat.total() * roiMat.elemSize();

        imgidx.type = MB_Frame::MBFT_ARGB8888;

        imgidx.width = roiMat.cols;

        imgidx.height = roiMat.rows;



        const size_t newSize = imgidx.offset + imgidx.size;



        //isContinuous

        if (newSize > buffImgMaxSize)

        {

            LOG_ERROR << "FaceCache::getFaceListImage buffImgMaxSize truncated" << LOG_ENDL;

            return false;

        }



        uint8_t* pbuf = buffImg + imgidx.offset;

        memcpy(pbuf, roiMat.ptr(), imgidx.size);



        //{

        //    static size_t f = 0;

        //    char fname[50];

        //    sprintf(fname, "/sdcard/face-%u-w%d-h%d.argb", ++f, imgidx.width, imgidx.height);

        //    FILE *pFile = fopen(fname, "wb");

        //    fwrite(pbuf, 1, imgidx.size, pFile);

        //    fclose(pFile);

        //    if (f > 20)exit(0);

        //}



        imgIdxes.push_back(imgidx);

        offset = newSize;

    }



    buffImgMaxSize = offset;

    return true;

}



#endif