aaaaa

houxiao

2017-06-29 9ee90782bf14b91442cd524bc033bbf6d1778bb2

 FaceServer/main_face_daemon.cpp

@@ -168,21 +168,59 @@
   

   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.img", stfaceImg1.db_id, ++i, stfaceImg1.width, stfaceImg1.height);

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

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

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

   fwrite(stfaceImg.buff, sizeof(char), stfaceImg.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;

   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)

@@ -232,6 +270,9 @@
   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);