unify log

houxiao

2016-12-29 633e76c1d533c3d9c257b92df7ebdfd36c9fd8a0

 RtspFace/main.cpp

@@ -1,40 +1,95 @@
#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 <iostream>

using namespace std;

#include "PL_SensetimeFaceDetect.h"



#include "logger.h"



int main(int argc, char** argv)

{

   initLogger(2);



   PipeLine pipeLine;

   

   pipeLine.register_elem_creator("PL_RTSPClient", create_PL_RTSPClient);

   pipeLine.register_elem_creator("PL_RTSPServer", create_PL_RTSPServer);

   pipeLine.register_elem_creator("PL_H264Decoder", create_PL_H264Decoder);

   pipeLine.register_elem_creator("PL_AVFrameYUV420", create_PL_H264Decoder);

   pipeLine.register_elem_creator("PL_AVFrameYUV420", create_PL_AVFrameYUV420);

   pipeLine.register_elem_creator("PL_H264Encoder", create_PL_H264Encoder);

   pipeLine.register_elem_creator("PL_Queue", create_PL_Queue);

   

   PL_RTSPClient* rtspClient = (PL_RTSPClient*)pipeLine.push_elem("PL_RTSPClient");

   RTSPConfig rtspConfig;

   rtspConfig.progName = argv[0];

   rtspConfig.rtspURL = argv[1];

   bool ret = rtspClient->init(&rtspConfig);

   if (!ret)

   pipeLine.register_elem_creator("PL_SensetimeFaceDetect", create_PL_SensetimeFaceDetect);

	
   {

      cout << "rtspClient.init error" << endl;

      exit(EXIT_FAILURE);

      PL_RTSPClient* rtspClient = (PL_RTSPClient*)pipeLine.push_elem("PL_RTSPClient");

      RTSPConfig rtspConfig;

      rtspConfig.progName = argv[0];

      rtspConfig.rtspURL = argv[1];

      rtspConfig.aux = true; // ffmpeg need aux, but live555 not

      rtspConfig.verbosityLevel = 1;

      rtspConfig.tunnelOverHTTPPortNum = 0;

      rtspConfig.args = nullptr;

      bool ret = rtspClient->init(&rtspConfig);

      if (!ret)

      {

         LOG(ERROR) << "rtspClient.init error";

         exit(EXIT_FAILURE);

      }

   }



   {

      PL_H264Decoder* h264Decoder = (PL_H264Decoder*)pipeLine.push_elem("PL_H264Decoder");

      h264Decoder->init(nullptr);

   }



   {

      PL_AVFrameYUV420* avFrameYUV420 = (PL_AVFrameYUV420*)pipeLine.push_elem("PL_AVFrameYUV420");

      avFrameYUV420->init(nullptr);

   }



   {

      SensetimeFaceDetectConfig config;

      PL_SensetimeFaceDetect* stFaceDetect = (PL_SensetimeFaceDetect*)pipeLine.push_elem("PL_SensetimeFaceDetect");

      stFaceDetect->init(&config);

   }



   //{//#todo queue should support deep copy

   //   PL_Queue_Config config;

   //   PL_Queue* queue1 = (PL_Queue*)pipeLine.push_elem("PL_Queue");

   //   bool ret = queue1->init(&config);

   //   if (!ret)

   //   {

   //      LOG(ERROR) << "queue1.init error";

   //      exit(EXIT_FAILURE);

   //   }

   //}



   {

      PL_H264Encoder* h264Encoder = (PL_H264Encoder*)pipeLine.push_elem("PL_H264Encoder");

      h264Encoder->init(nullptr);

   }

   

   PL_H264Decoder* h264Decoder = (PL_H264Decoder*)pipeLine.push_elem("PL_H264Decoder");

   h264Decoder->init(nullptr);

	
   PL_AVFrameYUV420* AVFrameYUV420 = (PL_AVFrameYUV420*)pipeLine.push_elem("PL_AVFrameYUV420");

   AVFrameYUV420->init(nullptr);

   {

      RTSPServerConfig config;

      PL_RTSPServer* rtspServer = (PL_RTSPServer*)pipeLine.push_elem("PL_RTSPServer");

      bool ret = rtspServer->init(&config);

      if (!ret)

      {

         LOG(ERROR) << "rtspServer.init error";

         exit(EXIT_FAILURE);

      }

   }

   

   while(true)

   {

      //cout << "begin pipe" << endl;

      //LOG(ERROR) << "begin pipe";

      pipeLine.pipe();

      //cout << "end pipe" << endl;

      //LOG(ERROR) << "end pipe";

   }

}