From 4a2c2f6fa2dc3a6b1dd7fd57fff2511059d8d8a3 Mon Sep 17 00:00:00 2001
From: xuxiuxi <xuxiuxi@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期三, 12 四月 2017 12:33:51 +0800
Subject: [PATCH]
---
VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp | 37 +++++++++++++++++++++++++++++++------
1 files changed, 31 insertions(+), 6 deletions(-)
diff --git a/VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp b/VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp
index 160ed07..969383c 100644
--- a/VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp
+++ b/VisitFace/RtspNativeCodec/app/src/main/cpp/FaceCache.cpp
@@ -13,6 +13,8 @@
#include "PbFaceList.pb.h"
+#include <libyuv.h>
+
struct FcPmBreackerContext
{
uint8_t frameYUV[1920*1080*4];
@@ -46,6 +48,21 @@
bool convertYUV420ToRGB()
{
+#define SUBSAMPLE(v, a) ((((v) + (a) - 1)) / (a))
+
+ 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::I420ToRGB24(src_y, src_width,
+ src_u, SUBSAMPLE(src_width, 2),
+ src_v, SUBSAMPLE(src_width, 2),
+ frameRGB, 3 * src_width,
+ src_width, src_height);
+ frameRGBSize = src_height * src_width * 3;
+ return true;
}
};
@@ -58,7 +75,7 @@
ctx->faceFeatures = faceFeatures;
ctx->dataAvailable &= true;
- return false;
+ return true;
}
bool fc_pm_breaker_frame(const PipeMaterial* pm, void* args)
@@ -84,7 +101,7 @@
memcpy(ctx->frameYUV, lastFrame->buffer, ctx->frameYUVSize);
ctx->dataAvailable &= true;
- return false;
+ return true;
}
#ifdef USE_ST_SDK
@@ -119,9 +136,12 @@
bool FaceCache::getFaceListPb(uint8_t* buffer, size_t& buffMaxSize)
{
FcPmBreackerContext& ctx(*(FcPmBreackerContext*)_ctx);
+ if (!ctx.dataAvailable)
+ return false;
PbFaceList pbFaceList;
- pbFaceList.set_image_count(ctx.faceFeatures.size());
+ pbFaceList.set_magic(pbFaceList.magic());
+ pbFaceList.set_image_count(ctx.faceFeatures.size());
pbFaceList.set_src_width(ctx.width);
pbFaceList.set_src_height(ctx.height);
@@ -141,11 +161,14 @@
pbFaceListImage.set_top_left_x(faceRect.leftTop.x);
pbFaceListImage.set_top_left_y(faceRect.leftTop.y);
- pbFaceListImage.set_img(0, roiMat.ptr(), roiMat.elemSize());
+ pbFaceListImage.add_img(roiMat.ptr(), roiMat.total() * roiMat.elemSize());
}
+ size_t s = pbFaceList.ByteSize();
+ buffMaxSize = std::min(s, buffMaxSize);
+ pbFaceList.SerializeToArray(buffer, buffMaxSize);
- return false;
+ return buffMaxSize > 0;
}
bool FaceCache::getFaceListImage(int* buffIdx, size_t& count, uint8_t* buffImg, size_t& buffImgMaxSize)
@@ -154,6 +177,7 @@
if (ctx.frameRGBSize == 0)
{
+ //#todo should optimize not convert the whole image
if (! ctx.convertYUV420ToRGB())
return false;
}
@@ -169,7 +193,7 @@
cv::Mat roiMat(rgbMat, cv::Rect(faceRect.leftTop.x, faceRect.leftTop.y, faceRect.rightBottom.x - faceRect.leftTop.x, faceRect.rightBottom.y - faceRect.leftTop.y));
//isContinuous
- size_t s = roiMat.elemSize();
+ size_t s = roiMat.total() * roiMat.elemSize();
if (totalSize + s <= buffImgMaxSize)
{
memcpy(pBuff, roiMat.ptr(), s);
@@ -182,5 +206,6 @@
}
}
+ buffImgMaxSize = totalSize;
return (count != 0);
}
--
Gitblit v1.8.0