xuxiuxi
2017-05-11 109ffe9a777658936a38d0c146579a67c60a0d17
RtspFace/main.cpp
@@ -14,8 +14,9 @@
#include "PL_DlibFaceTrack.h"
#include "logger.h"
#include <iostream>
Logger g_logger(stdout);
Logger g_logger(std::cout);
int main(int argc, char** argv)
{
@@ -27,6 +28,7 @@
   PipeLine::register_global_elem_creator("PL_RTSPServer", create_PL_RTSPServer);
   PipeLine::register_global_elem_creator("PL_H264Decoder", create_PL_H264Decoder);
   PipeLine::register_global_elem_creator("PL_AVFrameYUV420", create_PL_AVFrameYUV420);
   PipeLine::register_global_elem_creator("PL_AVFrameBGRA", create_PL_AVFrameBGRA);
   PipeLine::register_global_elem_creator("PL_H264Encoder", create_PL_H264Encoder);
   PipeLine::register_global_elem_creator("PL_Queue", create_PL_Queue);
   PipeLine::register_global_elem_creator("PL_Scale", create_PL_Scale);
@@ -75,8 +77,8 @@
   
   {
      PL_Scale_Config config;
      config.toWidth = 800;
      config.toHeight = 600;
      config.toWidth = 480;
      config.toHeight = 360;
      PL_Scale* ple = (PL_Scale*)pipeLine.push_elem("PL_Scale");
      bool ret = ple->init(&config);
      if (!ret)
@@ -85,13 +87,25 @@
         exit(EXIT_FAILURE);
      }
   }
   {
      SensetimeFaceTrackConfig config;
      //config.generate_face_feature = true;
      PL_SensetimeFaceTrack* ple = (PL_SensetimeFaceTrack*)pipeLine.push_elem("PL_SensetimeFaceTrack");
      ple->init(&config);
      PL_AVFrameBGRA_Config config;
      config.convertTo = PL_AVFrameBGRA_Config::I420_TO_RGBA8888;
      PL_AVFrameBGRA* ple = (PL_AVFrameBGRA*)pipeLine.push_elem("PL_AVFrameBGRA");
      bool ret = ple->init(&config);
      if (!ret)
      {
         LOG_ERROR << "PL_AVFrameBGRA.init error" << std::endl;
         exit(EXIT_FAILURE);
      }
   }
   //{
   //   SensetimeFaceTrackConfig config;
   //   //config.generate_face_feature = true;
   //   PL_SensetimeFaceTrack* ple = (PL_SensetimeFaceTrack*)pipeLine.push_elem("PL_SensetimeFaceTrack");
   //   ple->init(&config);
   //}
   
   //PipeLine pipeLine2;
   //{
@@ -126,26 +140,29 @@
   //   }
   //}
   {
      PL_H264Encoder_Config config;
      PL_H264Encoder* h264Encoder = (PL_H264Encoder*)pipeLine.push_elem("PL_H264Encoder");
      bool ret = h264Encoder->init(&config);
      if (!ret)
      {
         LOG_ERROR << "PL_H264Encoder.init error" << std::endl;
         exit(EXIT_FAILURE);
      }
   }
   {
      PL_RTSPServer* rtspServer = (PL_RTSPServer*)pipeLine.push_elem("PL_RTSPServer");
      bool ret = rtspServer->init(nullptr);
      if (!ret)
      {
         LOG_ERROR << "rtspServer.init error" << std::endl;
         exit(EXIT_FAILURE);
      }
   }
   //{
   //   PL_H264Encoder_Config config;
   //   config.av_opt_preset = "superfast";
   //   config.av_opt_tune = "zerolatency";
   //   config.avc_profile_str = "baseline";
   //   PL_H264Encoder* h264Encoder = (PL_H264Encoder*)pipeLine.push_elem("PL_H264Encoder");
   //   bool ret = h264Encoder->init(&config);
   //   if (!ret)
   //   {
   //      LOG_ERROR << "PL_H264Encoder.init error" << std::endl;
   //      exit(EXIT_FAILURE);
   //   }
   //}
    //
   //{
   //   PL_RTSPServer* rtspServer = (PL_RTSPServer*)pipeLine.push_elem("PL_RTSPServer");
   //   bool ret = rtspServer->init(nullptr);
   //   if (!ret)
   //   {
   //      LOG_ERROR << "rtspServer.init error" << std::endl;
   //      exit(EXIT_FAILURE);
   //   }
   //}
   
   while(true)
   {