2.RTSP客户端接收存储数据并转化为H264文件格式

最新推荐文章于 2024-01-26 15:53:32 发布

赤云

最新推荐文章于 2024-01-26 15:53:32 发布

阅读量1.1k

点赞数 1

分类专栏： live555

live555 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种方法，通过修改testRTSPClient程序，将接收到的RTSP视频流保存为H264格式的文件。文中详细展示了如何处理每一帧数据，并插入起始码以符合H264标准。

有网友在testRTSPClient基础上，把接收的数据写成h264文件了

void DummySink::afterGettingFrame(void* clientData, unsigned frameSize, unsigned numTruncatedBytes,
                  struct timeval presentationTime, unsigned durationInMicroseconds) {
  DummySink* sink = (DummySink*)clientData;
  sink->afterGettingFrame(frameSize, numTruncatedBytes, presentationTime, durationInMicroseconds);
}

// If you don't want to see debugging output for each received frame, then comment out the following line:
#define DEBUG_PRINT_EACH_RECEIVED_FRAME 1

void DummySink::afterGettingFrame(unsigned frameSize, unsigned numTruncatedBytes,
 struct timeval presentationTime, unsigned /*durationInMicroseconds*/) {
  // We've just received a frame of data.  (Optionally) print out information about it:
#ifdef DEBUG_PRINT_EACH_RECEIVED_FRAME
  if (fStreamId != NULL) envir() << "Stream \"" << fStreamId << "\"; ";
  envir() << fSubsession.mediumName() << "/" << fSubsession.codecName() << ":\tReceived " << frameSize << " bytes";
  if (numTruncatedBytes > 0) envir() << " (with " << numTruncatedBytes << " bytes truncated)";
  char uSecsStr[6+1]; // used to output the 'microseconds' part of the presentation time
  sprintf(uSecsStr, "%06u", (unsigned)presentationTime.tv_usec);
  envir() << ".\tPresentation time: " << (unsigned)presentationTime.tv_sec << "." << uSecsStr;
  if (fSubsession.rtpSource() != NULL && !fSubsession.rtpSource()->hasBeenSynchronizedUsingRTCP()) {
    envir() << "!"; // mark the debugging output to indicate that this presentation time is not RTCP-synchronized
  }
  envir() << "\n";
#endif
  
  //todo one frame
  //save to file
  if(!strcmp(fSubsession.mediumName(), "video"))
  {
     if(firstFrame)
     {
         unsigned int num;
         SPropRecord *sps = parseSPropParameterSets(fSubsession.fmtp_spropparametersets(), num);
         // For H.264 video stream, we use a special sink that insert start_codes:
         struct timeval tv= {0,0};
         unsigned char start_code[4] = {0x00, 0x00, 0x00, 0x01};
         FILE *fp = fopen("test.264", "a+b");
         if(fp)
         {
             fwrite(start_code, 4, 1, fp);
             fwrite(sps[0].sPropBytes, sps[0].sPropLength, 1, fp);
             fwrite(start_code, 4, 1, fp);
             fwrite(sps[1].sPropBytes, sps[1].sPropLength, 1, fp);
             fclose(fp);
             fp = NULL;
         }
         delete [] sps;
         firstFrame = False;
     }

     char *pbuf = (char *)fReceiveBuffer;
     char head[4] = {0x00, 0x00, 0x00, 0x01};
     FILE *fp = fopen("test.264", "a+b");
     if(fp)
     {
         fwrite(head, 4, 1, fp);
         fwrite(fReceiveBuffer, frameSize, 1, fp);
         fclose(fp);
         fp = NULL;
     }
  }

  // Then continue, to request the next frame of data:
  continuePlaying();
}

Boolean DummySink::continuePlaying() {
  if (fSource == NULL) return False; // sanity check (should not happen)

  // Request the next frame of data from our input source.  "afterGettingFrame()" will get called later, when it arrives:
  fSource->getNextFrame(fReceiveBuffer, DUMMY_SINK_RECEIVE_BUFFER_SIZE,
                        afterGettingFrame, this,
                        onSourceClosure, this);
  return True;
}