#include "PipeLine.h"
|
#include "PL_RTSPClient.h"
|
#include "PL_RTSPServer.h"
|
#include "PL_H264Decoder.h"
|
#include "PL_H264Encoder.h"
|
#include "PL_AVFrameYUV420.h"
|
#include "PL_AVFrameBGRA.h"
|
#include "PL_Queue.h"
|
#include "PL_Scale.h"
|
#include "PL_Fork.h"
|
#include "PL_SensetimeFaceTrack.h"
|
#include "PL_DlibFaceTrack.h"
|
|
#include "PipeLinePool.h"
|
|
#include "ev_server.h"
|
#include "ev_proto.h"
|
|
#include "logger.h"
|
|
PipeLinePool g_PipeLinePool;
|
|
evclient_proc_t evclient_proc;
|
|
bool ev_proc_SensetimeFaceDetect(EVClientStub& client)
|
{
|
}
|
|
bool ev_proc(EVClientStub& client)
|
{
|
EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;
|
//#todo check cmd and size
|
|
//#test send 01000B0000004142434445
|
//LOG_DEBUG << "cmd=" << evpHeader->cmd << ", size=" << evpHeader->size << ", \t" << (char*)(evpHeader + sizeof(EVPHeader));
|
//return true;
|
|
PipeLine* pipeLine = nullptr;
|
if (g_PipeLinePool.wait_free())
|
pipeLine = g_PipeLinePool.get_free();
|
|
if (pipeLine == nullptr)
|
{
|
LOG_WARN << "can't get free pipeline";//#todo send err packet
|
return false;
|
}
|
|
PipeMaterial pm;
|
// fill
|
|
PipeLineElem* plElem = pipeLine.pipe(&pm);
|
if (! pipeLine.check_pipe_complete(plElem))
|
{
|
LOG_WARN << "pipeline not complete";
|
g_PipeLinePool.release(pipeLine);//#todo send err packet
|
return false;
|
}
|
|
if (!plElem->gain(pm))
|
{
|
LOG_WARN << "pipeline gain error";
|
g_PipeLinePool.release(pipeLine);//#todo send err packet
|
return false;
|
}
|
|
if (pm.type == PipeMaterial::PMT_PM_LIST)
|
{
|
PipeMaterial& facePM = ((PipeMaterial*)(pm.buffer))[1];
|
st_ff_vect_t& faceFeatures = *((st_ff_vect_t*)facePM.buffer);
|
LOG_NOTICE << "faceFeatures " << faceFeatures.size();
|
//#todo send result packet
|
}
|
|
g_PipeLinePool.release(pipeLine);
|
return false;
|
}
|
|
int main(int argc, char** argv)
|
{
|
initLogger(LV_DEBUG);
|
|
PipeLine::register_global_elem_creator("PL_SensetimeFaceTrack", create_PL_SensetimeFaceTrack);
|
|
g_PipeLinePool = new PipeLinePool(true);
|
|
for (int i = 0; i < 5; i++)
|
{
|
PipeLine* pipeLine = new PipeLine;
|
|
{//payer//#todo
|
}
|
|
{
|
SensetimeFaceTrackConfig config;
|
config.generate_face_feature = true;
|
PL_SensetimeFaceTrack* sensetimeFaceTrack = (PL_SensetimeFaceTrack*)pipeLine->push_elem("PL_SensetimeFaceTrack");
|
bool ret = sensetimeFaceTrack->init(&config);
|
if (!ret)
|
{
|
LOG_ERROR << "sensetimeFaceTrack init error";
|
exit(EXIT_FAILURE);
|
}
|
}
|
|
g_PipeLinePool.manage(pipeLine);
|
}
|
|
evclient_proc = ev_proc;
|
return server_main(argc, argv);
|
|
while(true)
|
{
|
//LOG_ERROR << "begin pipe";
|
|
PipeMaterial pm;
|
if (pipeLine.pipe(&pm) == sensetimeFaceTrack);
|
sensetimeFaceTrack->gain(pm);
|
|
if (pm.type == PipeMaterial::PMT_PM_LIST)
|
{
|
PipeMaterial& facePM = ((PipeMaterial*)(pm.buffer))[1];
|
st_ff_vect_t& faceFeatures = *((st_ff_vect_t*)facePM.buffer);
|
LOG_NOTICE << "faceFeatures " << faceFeatures.size();
|
}
|
|
//LOG_ERROR << "end pipe";
|
}
|
}
|
