H264 NALU解析 并在RTSPServer判断

xingzilong

2017-08-18 9e5babf9db52e64bdae60137be7696e56241fca6

 RtspFace/PL_AndroidMediaCodecEncoder.cpp

@@ -82,6 +82,7 @@
   PL_AndroidMediaCodecEncoder_Config* config = (PL_AndroidMediaCodecEncoder_Config*)args;

   in->config = *config;



   // see: developer.android.com/reference/android/media/MediaFormat.html#KEY_PROFILE

    AMediaFormat* format = AMediaFormat_new();



    AMediaFormat_setString(format, AMEDIAFORMAT_KEY_MIME, config->ak_mime.c_str());

@@ -90,12 +91,20 @@


   AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_BIT_RATE, config->ak_bit_rate);

   AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_FRAME_RATE, config->ak_frame_rate);

   AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_I_FRAME_INTERVAL, config->ak_i_frame_interval);

   AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_I_FRAME_INTERVAL, config->ak_i_frame_interval); // android multipled 10

   AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_CHANNEL_COUNT, 1);

   //AMediaFormat_setInt32(format, "profile", 0x00000100);



// see: https://developer.android.com/reference/android/media/MediaCodecInfo.CodecCapabilities.html#COLOR_FormatYUV420Flexible

#define AMEDIA_COLOR_FormatYUV420Flexible 0x7f420888

    AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_COLOR_FORMAT, config->ak_color_format);

    //AMediaFormat_setInt32(format, AMEDIAFORMAT_KEY_STRIDE, config->ak_width * 2);



   if (config->codecProfileLevel.profile != 0)

      AMediaFormat_setInt32(format, "profile", config->codecProfileLevel.profile);

   if (config->codecProfileLevel.level != 0)

      AMediaFormat_setInt32(format, "level", config->codecProfileLevel.level);





   //uint8_t sps[] = {0x0,0x0,0x0,0x1, 0x67, 0x42, 0x00, 0x2A, 0x95, 0xA8, 0x1E, 0x00, 0x89, 0xF9, 0x61, 0x00, 0x00, 0x07, 0x08, 0x00, 0x01, 0x5F, 0x90, 0x04};

   //uint8_t pps[] = {0x0,0x0,0x0,0x1, 0x68, 0xCE, 0x3C, 0x80};

@@ -231,7 +240,7 @@
         pm.buffSize = 0;



         //static size_t f = 0;

         //static FILE *pFile = fopen("/sdcard/aa.264", "wb");

         //static FILE *pFile = fopen("/data/aa.264", "wb");

         //fwrite(in->buffer, sizeof(char), in->buffSize, pFile);

         //if (++f > 400){

         //   fclose(pFile);

@@ -251,6 +260,28 @@
   {

      auto format = AMediaCodec_getOutputFormat(in->codec);

      LOGP(INFO, "format changed to: %s", AMediaFormat_toString(format));



      uint8_t* sps = nullptr;

      size_t spsSize = 0;

      uint8_t* pps = nullptr;

      size_t ppsSize = 0;



      AMediaFormat_getBuffer(format, "csd-0", (void**)&sps, &spsSize); // sps

      AMediaFormat_getBuffer(format, "csd-1", (void**)&pps, &ppsSize); // pps



      if (spsSize != 0)

      {

         char* tmp = nullptr;

         std::string spsStr = tmp = base64_encode(((const char*)sps) + 4, spsSize - 4);//#todo aux

         delete[] tmp;

         std::string ppsStr = tmp = base64_encode(((const char*)pps) + 4, ppsSize - 4);

         delete[] tmp;

         tmp = nullptr;



         this->manager->set_param(PLGP_ENC_SPS_B64, spsStr);

         this->manager->set_param(PLGP_ENC_PPS_B64, ppsStr);

      }



      AMediaFormat_delete(format);

   }

   else if (outputBuffIdx == AMEDIACODEC_INFO_TRY_AGAIN_LATER)