xuxiuxi

2017-07-27 f5948ed9f21e65288f8772a3a9df0ab68bf4f727

 FaceServer/main_face_daemon.cpp

@@ -4,8 +4,9 @@
#include "ev_server.h" 

#include "ev_proto.h"

#include "face_daemon_proto.h"

#include "sample_face_search.h"

#include "STFaceCache.h"

#include <PbFaceList.pb.h>

#include "PbFaceList.pb.h"



#include <signal.h>



@@ -24,25 +25,29 @@
   server_stop();

}



bool send_SensetimeFaceDetectResultJson(EVClientStub& client, const fdr_vec_t& result)

bool send_SensetimeFaceDetectResultJson(EVClientStub& client, const fdr_vec_t& result, int ret = 0)

{

   std::stringstream ss;

   ss << "{" << std::endl;

   ss << "\"ret\":" << WRAPPER_TEXT(0) << "," << std::endl;

   ss << "\"count\":" << WRAPPER_TEXT(result.size()) << "," << std::endl;

   ss << "{" << LOG_ENDL;

   ss << "\"ret\":" << WRAPPER_TEXT(ret) << "," << LOG_ENDL;

   ss << "\"count\":" << WRAPPER_TEXT(result.size()) << "," << LOG_ENDL;

   ss << "\"result\":[";

   for(fdr_vec_t::const_iterator iter = result.begin(); iter != result.end(); ++iter)

   {

      ss << "[" << WRAPPER_TEXT(iter->db_id) << "," << WRAPPER_TEXT(iter->st_id) << "]";

      ss << "[" << WRAPPER_TEXT(iter->db_id) << "," 
         << WRAPPER_TEXT(iter->st_id) <<  "," 
         << WRAPPER_TEXT(iter->confidence) <<  "," 
         << WRAPPER_TEXT(iter->camera_id) <<  "," 
         << WRAPPER_TEXT(iter->st_track_id) << "]";

      if (iter != std::prev(result.end()))

         ss << ",";

   }

   ss << "]" << std::endl;

   ss << "]" << LOG_ENDL;

   ss << "}";



   std::string output(ss.str());//#todo avoid copy

   client.sendBuffSize = output.size() + 1;

   client.sendBuff = new uint8_t[client.sendBuffSize];

   client.sendBuffSize = output.size();

   client.sendBuff = new uint8_t[output.size() + 1];

   strcpy((char*)client.sendBuff, output.c_str());

   client.deleteSendBuff = true;

   

@@ -55,11 +60,12 @@
{

   //#test send //00038300FE4B0000+pb

   //EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   //LOG_DEBUG << "cmd=" << evpHeader->cmd << ", size=" << evpHeader->size << ", \t" << (char*)(evpHeader + sizeof(EVPHeader)) << std::endl;

   //LOG_DEBUG << "cmd=" << evpHeader->cmd << ", size=" << evpHeader->size << ", \t" << (char*)(evpHeader + sizeof(EVPHeader)) << LOG_ENDL;

   //return true;

   

   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   FDP_FaceDetectPB* fdpFaceDetectPB = (FDP_FaceDetectPB*)(client.recvBuff + sizeof(EVPHeader));

   fdpFaceDetectPB->ntoh();

   

   PbFaceList pbFaceList;

   pbFaceList.ParseFromArray(client.recvBuff + sizeof(EVPHeader) + sizeof(FDP_FaceDetectPB), evpHeader->size - sizeof(EVPHeader) - sizeof(FDP_FaceDetectPB));

@@ -91,7 +97,21 @@
      stimg.height = pbFaceListImage.height();

      stimg.size = img->size();

      stimg.buff = (const uint8_t*)img->data();

      result.push_back(g_STFaceCache.detect(stimg));

		
      FDP_FaceDetectResult res(g_STFaceCache.detect(stimg));

		
#ifdef ENABLE_ADD_TO_TEMPDB_WHEN_NOT_DETECT

      if (res.db_id == STFS_DBID_INVALID)

      {

         // #todo optimize extract feature once

         stimg.db_id = STFS_DBID_TEMPDB_1;

         res = g_STFaceCache.add(stimg);

      }

#endif



      res.camera_id = fdpFaceDetectPB->camera_id;

      res.st_track_id = pbFaceListImage.st_track_id();

      result.push_back(res);

   }



   //#test

@@ -102,6 +122,125 @@
   return send_SensetimeFaceDetectResultJson(client, result);

}



bool ev_proc_SensetimeFaceDetectSave(EVClientStub& client)

{

   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   FDP_Image* fdpImage = (FDP_Image*)(client.recvBuff + sizeof(EVPHeader));

   fdpImage->ntoh();

	
   STFaceImage stfaceImg;

   stfaceImg.db_id = fdpImage->db_id;

   stfaceImg.mb_type = fdpImage->mb_type;

   stfaceImg.width = fdpImage->width;

   stfaceImg.height = fdpImage->height;

   stfaceImg.size = evpHeader->size - sizeof(EVPHeader) - sizeof(FDP_Image);

   stfaceImg.buff = fdpImage->buff;

	
   LOGP(DEBUG, "stfaceImg db_id=%d, mb_type=%d, width=%d, height=%d, size=%d", 
      (int)stfaceImg.db_id, (int)stfaceImg.mb_type, (int)stfaceImg.width, (int)stfaceImg.height, (int)stfaceImg.size);

	
   //char imgfn[100 * 1024];

   //static int i = 0;

   //sprintf(imgfn, "IMG_%d_%d_w%d_h%d.rgb565", stfaceImg.db_id, ++i, stfaceImg.width, stfaceImg.height);

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

   //fwrite(stfaceImg.buff, sizeof(char), stfaceImg.size, pFile);

   //fclose(pFile);

   //pFile = nullptr;

	
   fdr_vec_t result;

   FDP_FaceDetectResult fdrResult = g_STFaceCache.add(stfaceImg);

   result.push_back(fdrResult);

	
   int ret = (fdrResult.db_id == 0 ? -1 : 0);

   return send_SensetimeFaceDetectResultJson(client, result, ret);

}



bool ev_proc_SensetimeFaceDetectCompare(EVClientStub& client)

{

   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

	
   FDP_Image* fdpImage1 = (FDP_Image*)(client.recvBuff + sizeof(EVPHeader));

   fdpImage1->ntoh();

	
   STFaceImage stfaceImg1;

   stfaceImg1.db_id = fdpImage1->db_id;

   stfaceImg1.mb_type = fdpImage1->mb_type;

   stfaceImg1.width = fdpImage1->width;

   stfaceImg1.height = fdpImage1->height;

   stfaceImg1.size = fdpImage1->size;

   stfaceImg1.buff = fdpImage1->buff;

	
   LOGP(DEBUG, "stfaceImg1 db_id=%d, mb_type=%d, width=%d, height=%d, size=%d", 
      (int)stfaceImg1.db_id, (int)stfaceImg1.mb_type, (int)stfaceImg1.width, (int)stfaceImg1.height, (int)stfaceImg1.size);

	
   FDP_Image* fdpImage2 = (FDP_Image*)(client.recvBuff + sizeof(EVPHeader) + sizeof(FDP_Image) + fdpImage1->size);

   fdpImage2->ntoh();

	
   STFaceImage stfaceImg2;

   stfaceImg2.db_id = fdpImage2->db_id;

   stfaceImg2.mb_type = fdpImage2->mb_type;

   stfaceImg2.width = fdpImage2->width;

   stfaceImg2.height = fdpImage2->height;

   stfaceImg2.size = fdpImage2->size;

   stfaceImg2.buff = fdpImage2->buff;

	
   LOGP(DEBUG, "stfaceImg2 db_id=%d, mb_type=%d, width=%d, height=%d, size=%d", 
      (int)stfaceImg2.db_id, (int)stfaceImg2.mb_type, (int)stfaceImg2.width, (int)stfaceImg2.height, (int)stfaceImg2.size);



   //char imgfn[100 * 1024];

   //static int i = 0;

   //sprintf(imgfn, "IMG_%d_%d_w%d_h%d.img", stfaceImg1.db_id, ++i, stfaceImg1.width, stfaceImg1.height);

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

   //fwrite(stfaceImg1.buff, sizeof(char), stfaceImg1.size, pFile);

   //fclose(pFile);

   //pFile = nullptr;

    //

   //sprintf(imgfn, "IMG_%d_%d_w%d_h%d.img", stfaceImg2.db_id, ++i, stfaceImg2.width, stfaceImg2.height);

   //pFile = fopen(imgfn, "wb");

   //fwrite(stfaceImg2.buff, sizeof(char), stfaceImg2.size, pFile);

   //fclose(pFile);

   //pFile = nullptr;

	
   fdr_vec_t result;

   FDP_FaceDetectResult fdrResult = g_STFaceCache.compare(stfaceImg1, stfaceImg2);

   result.push_back(fdrResult);



   return send_SensetimeFaceDetectResultJson(client, result, 0);

}



bool ev_proc_SensetimeFaceDetectSearch(EVClientStub& client)

{

   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   FDP_Image* fdpImage = (FDP_Image*)(client.recvBuff + sizeof(EVPHeader));

   fdpImage->ntoh();

	
   STFaceImage stfaceImg;

   stfaceImg.db_id = fdpImage->db_id;

   stfaceImg.mb_type = fdpImage->mb_type;

   stfaceImg.width = fdpImage->width;

   stfaceImg.height = fdpImage->height;

   stfaceImg.size = evpHeader->size - sizeof(EVPHeader) - sizeof(FDP_Image);

   stfaceImg.buff = fdpImage->buff;

	
   LOGP(DEBUG, "stfaceImg db_id=%d, mb_type=%d, width=%d, height=%d, size=%d", 
      (int)stfaceImg.db_id, (int)stfaceImg.mb_type, (int)stfaceImg.width, (int)stfaceImg.height, (int)stfaceImg.size);

	
	
   char imgfn[100 * 1024];

   static int i = 0;

   sprintf(imgfn, "IMG_%d_%d_w%d_h%d.rgb565", stfaceImg.db_id, ++i, stfaceImg.width, stfaceImg.height);

   FILE * pFile = fopen(imgfn, "wb");

   fwrite(stfaceImg.buff, sizeof(char), stfaceImg.size, pFile);

   fclose(pFile);

   pFile = nullptr;

	
   fdr_vec_t result;

   g_STFaceCache.search(stfaceImg, result);

	
   //int ret = (fdrResult.db_id == 0 ? -1 : 0);

   return send_SensetimeFaceDetectResultJson(client, result, 0);

}



bool ev_dispatcher_proto_pb(EVClientStub& client)

{

   LOG_DEBUG << "ev_dispatcher_proto_pb" << LOG_ENDL;

@@ -109,7 +248,7 @@
   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   if (evpHeader->size != client.recvBuffSize)

   {

      LOG_WARN << "Truncated buffer " << (evpHeader->size - client.recvBuffSize) << " bytes" << std::endl;

      LOG_WARN << "Truncated buffer " << (evpHeader->size - client.recvBuffSize) << " bytes" << LOG_ENDL;

      return false;

   }



@@ -120,7 +259,40 @@
      return ev_proc_SensetimeFaceDetectPB(client);

   break;

   default:

      LOG_WARN << "Unknown command" << std::endl;

      LOG_WARN << "Unknown command" << LOG_ENDL;

      ev_send_status_packet(client, EVPStatus::EVPS_COMMAND_ERROR);

      return false;

   break;

   }

	
   // return false to disconnect

   return false;

}



bool ev_dispatcher_proto_rawbin(EVClientStub& client)

{

   LOG_DEBUG << "ev_dispatcher_proto_rawbin" << LOG_ENDL;

	
   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   if (evpHeader->size != client.recvBuffSize)

   {

      LOG_WARN << "Truncated buffer " << (evpHeader->size - client.recvBuffSize) << " bytes" << LOG_ENDL;

      return false;

   }



   switch(evpHeader->cmd)

   {

   case FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_SAVE:

      return ev_proc_SensetimeFaceDetectSave(client);

   break;

   case FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_COMPARE:

      return ev_proc_SensetimeFaceDetectCompare(client);

   break;

   case FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_SEARCH:

      return ev_proc_SensetimeFaceDetectSearch(client);

   break;

   default:

      LOG_WARN << "Unknown command" << LOG_ENDL;

      ev_send_status_packet(client, EVPStatus::EVPS_COMMAND_ERROR);

      return false;

   break;

@@ -137,7 +309,7 @@
   EVPHeader* evpHeader = (EVPHeader*)client.recvBuff;

   if (evpHeader->size != client.recvBuffSize)

   {

      LOG_WARN << "Truncated buffer " << (evpHeader->size - client.recvBuffSize) << " bytes" << std::endl;

      LOG_WARN << "Truncated buffer " << (evpHeader->size - client.recvBuffSize) << " bytes" << LOG_ENDL;

      return false;

   }



@@ -146,8 +318,11 @@
   case EVPProto::EVPP_PROTOBUF:

      return ev_dispatcher_proto_pb(client);

   break;

   case EVPProto::EVPP_RAW_BIN:

      return ev_dispatcher_proto_rawbin(client);

   break;

   default:

      LOG_WARN << "Unknown proto" << std::endl;

      LOG_WARN << "Unknown proto" << LOG_ENDL;

      ev_send_status_packet(client, EVPStatus::EVPS_PROTO_ERROR);

      return false;

   break;

@@ -163,6 +338,23 @@
   

   signal(SIGINT, _sigint);



   if (!g_STFaceCache.init())

   {

      LOG_WARN << "g_STFaceCache.init return false" << LOG_ENDL;

      return EXIT_FAILURE;

   }

	
   if (!g_STFaceCache.load_dbs())

   {

      LOG_WARN << "g_STFaceCache.load_dbs return false" << LOG_ENDL;

      return EXIT_FAILURE;

   }

	
   evclient_proc = ev_dispatcher_proto;

   return server_main(argc, argv);

   int ec = server_main(argc, argv);

	
   g_STFaceCache.close_dbs();

   g_STFaceCache.finit();

	
   return ec;

}