Index: webrtc/modules/video_coding/codecs/test/videoprocessor.cc |
diff --git a/webrtc/modules/video_coding/codecs/test/videoprocessor.cc b/webrtc/modules/video_coding/codecs/test/videoprocessor.cc |
index 35bf5593664a8b2cc6cbefebad5e3791f75ccb94..3df8a042f014c06f531cd1f75a245e3157a70a97 100644 |
--- a/webrtc/modules/video_coding/codecs/test/videoprocessor.cc |
+++ b/webrtc/modules/video_coding/codecs/test/videoprocessor.cc |
@@ -60,9 +60,6 @@ VideoProcessorImpl::VideoProcessorImpl(webrtc::VideoEncoder* encoder, |
packet_manipulator_(packet_manipulator), |
config_(config), |
stats_(stats), |
- encode_callback_(nullptr), |
- decode_callback_(nullptr), |
- last_successful_frame_buffer_(nullptr), |
first_key_frame_has_been_excluded_(false), |
last_frame_missing_(false), |
initialized_(false), |
@@ -97,51 +94,36 @@ bool VideoProcessorImpl::Init() { |
// Initialize data structures used by the encoder/decoder APIs |
size_t frame_length_in_bytes = frame_reader_->FrameLength(); |
- last_successful_frame_buffer_ = new uint8_t[frame_length_in_bytes]; |
+ last_successful_frame_buffer_.reset(new uint8_t[frame_length_in_bytes]); |
// Set fixed properties common for all frames. |
// To keep track of spatial resize actions by encoder. |
last_encoder_frame_width_ = config_.codec_settings->width; |
last_encoder_frame_height_ = config_.codec_settings->height; |
// Setup required callbacks for the encoder/decoder: |
- encode_callback_ = new VideoProcessorEncodeCompleteCallback(this); |
- decode_callback_ = new VideoProcessorDecodeCompleteCallback(this); |
- int32_t register_result = |
- encoder_->RegisterEncodeCompleteCallback(encode_callback_); |
- if (register_result != WEBRTC_VIDEO_CODEC_OK) { |
- fprintf(stderr, |
- "Failed to register encode complete callback, return code: " |
- "%d\n", |
- register_result); |
- return false; |
- } |
- register_result = decoder_->RegisterDecodeCompleteCallback(decode_callback_); |
- if (register_result != WEBRTC_VIDEO_CODEC_OK) { |
- fprintf(stderr, |
- "Failed to register decode complete callback, return code: " |
- "%d\n", |
- register_result); |
- return false; |
- } |
+ encode_callback_.reset(new VideoProcessorEncodeCompleteCallback(this)); |
+ decode_callback_.reset(new VideoProcessorDecodeCompleteCallback(this)); |
+ RTC_CHECK_EQ(encoder_->RegisterEncodeCompleteCallback(encode_callback_.get()), |
+ WEBRTC_VIDEO_CODEC_OK) |
+ << "Failed to register encode complete callback"; |
+ RTC_CHECK_EQ(decoder_->RegisterDecodeCompleteCallback(decode_callback_.get()), |
+ WEBRTC_VIDEO_CODEC_OK) |
+ << "Failed to register decode complete callback"; |
+ |
// Init the encoder and decoder |
uint32_t nbr_of_cores = 1; |
if (!config_.use_single_core) { |
nbr_of_cores = CpuInfo::DetectNumberOfCores(); |
} |
- int32_t init_result = |
+ RTC_CHECK_EQ( |
encoder_->InitEncode(config_.codec_settings, nbr_of_cores, |
- config_.networking_config.max_payload_size_in_bytes); |
- if (init_result != WEBRTC_VIDEO_CODEC_OK) { |
- fprintf(stderr, "Failed to initialize VideoEncoder, return code: %d\n", |
- init_result); |
- return false; |
- } |
- init_result = decoder_->InitDecode(config_.codec_settings, nbr_of_cores); |
- if (init_result != WEBRTC_VIDEO_CODEC_OK) { |
- fprintf(stderr, "Failed to initialize VideoDecoder, return code: %d\n", |
- init_result); |
- return false; |
- } |
+ config_.networking_config.max_payload_size_in_bytes), |
+ WEBRTC_VIDEO_CODEC_OK) |
+ << "Failed to initialize VideoEncoder"; |
+ |
+ RTC_CHECK_EQ(decoder_->InitDecode(config_.codec_settings, nbr_of_cores), |
+ WEBRTC_VIDEO_CODEC_OK) |
+ << "Failed to initialize VideoDecoder"; |
if (config_.verbose) { |
printf("Video Processor:\n"); |
@@ -178,24 +160,16 @@ bool VideoProcessorImpl::Init() { |
} |
VideoProcessorImpl::~VideoProcessorImpl() { |
- delete[] last_successful_frame_buffer_; |
- encoder_->RegisterEncodeCompleteCallback(NULL); |
- delete encode_callback_; |
- decoder_->RegisterDecodeCompleteCallback(NULL); |
- delete decode_callback_; |
+ encoder_->RegisterEncodeCompleteCallback(nullptr); |
+ decoder_->RegisterDecodeCompleteCallback(nullptr); |
} |
void VideoProcessorImpl::SetRates(int bit_rate, int frame_rate) { |
int set_rates_result = encoder_->SetRateAllocation( |
bitrate_allocator_->GetAllocation(bit_rate * 1000, frame_rate), |
frame_rate); |
- RTC_CHECK_GE(set_rates_result, 0); |
- if (set_rates_result < 0) { |
- fprintf(stderr, |
- "Failed to update encoder with new rate %d, " |
- "return code: %d\n", |
- bit_rate, set_rates_result); |
- } |
+ RTC_CHECK_GE(set_rates_result, 0) << "Failed to update encoder with new rate " |
+ << bit_rate; |
num_dropped_frames_ = 0; |
num_spatial_resizes_ = 0; |
} |
@@ -218,10 +192,8 @@ int VideoProcessorImpl::NumberSpatialResizes() { |
bool VideoProcessorImpl::ProcessFrame(int frame_number) { |
RTC_DCHECK_GE(frame_number, 0); |
- if (!initialized_) { |
- fprintf(stderr, "Attempting to use uninitialized VideoProcessor!\n"); |
- return false; |
- } |
+ RTC_CHECK(initialized_) << "Attempting to use uninitialized VideoProcessor"; |
+ |
// |prev_time_stamp_| is used for getting number of dropped frames. |
if (frame_number == 0) { |
prev_time_stamp_ = -1; |
@@ -247,7 +219,8 @@ bool VideoProcessorImpl::ProcessFrame(int frame_number) { |
encoded_frame_size_ = 0; |
encoded_frame_type_ = kVideoFrameDelta; |
- int32_t encode_result = encoder_->Encode(source_frame, NULL, &frame_types); |
+ int32_t encode_result = |
+ encoder_->Encode(source_frame, nullptr, &frame_types); |
if (encode_result != WEBRTC_VIDEO_CODEC_OK) { |
fprintf(stderr, "Failed to encode frame %d, return code: %d\n", |
@@ -273,7 +246,7 @@ void VideoProcessorImpl::FrameEncoded( |
// For dropped frames, we write out the last decoded frame to avoid getting |
// out of sync for the computation of PSNR and SSIM. |
for (int i = 0; i < num_dropped_from_prev_encode; i++) { |
- frame_writer_->WriteFrame(last_successful_frame_buffer_); |
+ frame_writer_->WriteFrame(last_successful_frame_buffer_.get()); |
} |
} |
// Frame is not dropped, so update the encoded frame size |
@@ -337,12 +310,12 @@ void VideoProcessorImpl::FrameEncoded( |
// TODO(kjellander): Pass fragmentation header to the decoder when |
// CL 172001 has been submitted and PacketManipulator supports this. |
int32_t decode_result = |
- decoder_->Decode(copied_image, last_frame_missing_, NULL); |
+ decoder_->Decode(copied_image, last_frame_missing_, nullptr); |
stat.decode_return_code = decode_result; |
if (decode_result != WEBRTC_VIDEO_CODEC_OK) { |
// Write the last successful frame the output file to avoid getting it out |
// of sync with the source file for SSIM and PSNR comparisons: |
- frame_writer_->WriteFrame(last_successful_frame_buffer_); |
+ frame_writer_->WriteFrame(last_successful_frame_buffer_.get()); |
} |
// save status for losses so we can inform the decoder for the next frame: |
last_frame_missing_ = copied_image._length == 0; |
@@ -382,7 +355,8 @@ void VideoProcessorImpl::FrameDecoded(const VideoFrame& image) { |
int extracted_length = ExtractBuffer(up_image, length, image_buffer.get()); |
RTC_DCHECK_GT(extracted_length, 0); |
// Update our copy of the last successful frame: |
- memcpy(last_successful_frame_buffer_, image_buffer.get(), extracted_length); |
+ memcpy(last_successful_frame_buffer_.get(), image_buffer.get(), |
+ extracted_length); |
bool write_success = frame_writer_->WriteFrame(image_buffer.get()); |
RTC_DCHECK(write_success); |
if (!write_success) { |
@@ -395,7 +369,8 @@ void VideoProcessorImpl::FrameDecoded(const VideoFrame& image) { |
std::unique_ptr<uint8_t[]> image_buffer(new uint8_t[length]); |
int extracted_length = ExtractBuffer(image, length, image_buffer.get()); |
RTC_DCHECK_GT(extracted_length, 0); |
- memcpy(last_successful_frame_buffer_, image_buffer.get(), extracted_length); |
+ memcpy(last_successful_frame_buffer_.get(), image_buffer.get(), |
+ extracted_length); |
bool write_success = frame_writer_->WriteFrame(image_buffer.get()); |
RTC_DCHECK(write_success); |