Add support for writing raw encoder output to .ivf files.

webrtc/modules/video_coding/generic_encoder.cc

 /*
  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
@@ skipping 73 matching lines @@
     case kVideoCodecGeneric:
       rtp->codec = kRtpVideoGeneric;
       rtp->simulcastIdx = info->codecSpecific.generic.simulcast_idx;
       return;
     default:
       return;
   }
 }
 }  // namespace
 
-// #define DEBUG_ENCODER_BIT_STREAM
-
 VCMGenericEncoder::VCMGenericEncoder(
     VideoEncoder* encoder,
     VideoEncoderRateObserver* rate_observer,
     VCMEncodedFrameCallback* encoded_frame_callback,
     bool internalSource)
     : encoder_(encoder),
       rate_observer_(rate_observer),
       vcm_encoded_frame_callback_(encoded_frame_callback),
       internal_source_(internalSource),
       encoder_params_({0, 0, 0, 0}),
@@ skipping 18 matching lines @@
   }
 
   is_screenshare_ = settings->mode == VideoCodecMode::kScreensharing;
   if (encoder_->InitEncode(settings, numberOfCores, maxPayloadSize) != 0) {
     LOG(LS_ERROR) << "Failed to initialize the encoder associated with "
                      "payload name: "
                   << settings->plName;
     return -1;
   }
   encoder_->RegisterEncodeCompleteCallback(vcm_encoded_frame_callback_);
+  vcm_encoded_frame_callback_->codec_type_ = settings->codecType;
   return 0;
 }
 
 int32_t VCMGenericEncoder::Encode(const VideoFrame& inputFrame,
                                   const CodecSpecificInfo* codecSpecificInfo,
                                   const std::vector<FrameType>& frameTypes) {
   TRACE_EVENT1("webrtc", "VCMGenericEncoder::Encode", "timestamp",
                inputFrame.timestamp());
 
   for (FrameType frame_type : frameTypes)
@@ skipping 83 matching lines @@
 /***************************
  * Callback Implementation
  ***************************/
 VCMEncodedFrameCallback::VCMEncodedFrameCallback(
     EncodedImageCallback* post_encode_callback)
     : send_callback_(),
       _mediaOpt(NULL),
       _payloadType(0),
       _internalSource(false),
       _rotation(kVideoRotation_0),
-      post_encode_callback_(post_encode_callback)
-#ifdef DEBUG_ENCODER_BIT_STREAM
-      ,
-      _bitStreamAfterEncoder(NULL)
-#endif
-{
-#ifdef DEBUG_ENCODER_BIT_STREAM
-  _bitStreamAfterEncoder = fopen("encoderBitStream.bit", "wb");
-#endif
-}
+      post_encode_callback_(post_encode_callback) {}
 
-VCMEncodedFrameCallback::~VCMEncodedFrameCallback() {
-#ifdef DEBUG_ENCODER_BIT_STREAM
-  fclose(_bitStreamAfterEncoder);
-#endif
-}
+VCMEncodedFrameCallback::~VCMEncodedFrameCallback() {}
 
 int32_t VCMEncodedFrameCallback::SetTransportCallback(
     VCMPacketizationCallback* transport) {
   send_callback_ = transport;
   return VCM_OK;
 }
 
 int32_t VCMEncodedFrameCallback::Encoded(
     const EncodedImage& encoded_image,
     const CodecSpecificInfo* codecSpecificInfo,
     const RTPFragmentationHeader* fragmentationHeader) {
   TRACE_EVENT_INSTANT1("webrtc", "VCMEncodedFrameCallback::Encoded",
                        "timestamp", encoded_image._timeStamp);
   post_encode_callback_->Encoded(encoded_image, NULL, NULL);
 
   if (send_callback_ == NULL) {
     return VCM_UNINITIALIZED;
   }
 
-#ifdef DEBUG_ENCODER_BIT_STREAM
-  if (_bitStreamAfterEncoder != NULL) {
-    fwrite(encoded_image._buffer, 1, encoded_image._length,
-           _bitStreamAfterEncoder);
+  if (kEnableFrameRecording) {
+    int layer = GetLayer(*codecSpecificInfo);
+    if (file_writers_[layer] == nullptr) {
+      uint64_t instance_id = reinterpret_cast<uint64_t>(this);
+      std::ostringstream oss;
+      oss << "bitstream_" << instance_id << "_" << layer << ".ivf";
+      file_writers_[layer] =
+          IvfFileWriter::Open(oss.str().c_str(), codec_type_);
+    }
+    if (file_writers_[layer].get() != nullptr) {
+      bool ok = file_writers_[layer]->WriteFrame(encoded_image);
+      RTC_DCHECK(ok);
+    }
   }
-#endif
 
   RTPVideoHeader rtpVideoHeader;
   memset(&rtpVideoHeader, 0, sizeof(RTPVideoHeader));
   RTPVideoHeader* rtpVideoHeaderPtr = &rtpVideoHeader;
   if (codecSpecificInfo) {
     CopyCodecSpecific(codecSpecificInfo, rtpVideoHeaderPtr);
   }
   rtpVideoHeader.rotation = _rotation;
 
   int32_t callbackReturn = send_callback_->SendData(
@@ skipping 14 matching lines @@
     media_optimization::MediaOptimization* mediaOpt) {
   _mediaOpt = mediaOpt;
 }
 
 void VCMEncodedFrameCallback::SignalLastEncoderImplementationUsed(
     const char* implementation_name) {
   if (send_callback_)
     send_callback_->OnEncoderImplementationName(implementation_name);
 }
 
+int VCMEncodedFrameCallback::GetLayer(const CodecSpecificInfo& codec_specific) {
+  switch (codec_type_) {
+    case kVideoCodecVP8:
+      return codec_specific.codecSpecific.VP8.simulcastIdx;
+    case kVideoCodecVP9:
+    case kVideoCodecH264:
+      return 1;
+    default:
+      RTC_NOTREACHED();
+      return -1;
+  }
+}
+
 }  // namespace webrtc