| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "media/cast/sender/h264_vt_encoder.h" | 5 #include "media/base/mac/videotoolbox_helpers.h" |
| 6 | 6 |
| 7 #include <stddef.h> | |
| 8 | |
| 9 #include <string> | |
| 10 #include <vector> | 7 #include <vector> |
| 11 | 8 |
| 12 #include "base/big_endian.h" | 9 #include "base/big_endian.h" |
| 13 #include "base/bind.h" | |
| 14 #include "base/bind_helpers.h" | |
| 15 #include "base/location.h" | |
| 16 #include "base/logging.h" | |
| 17 #include "base/macros.h" | |
| 18 #include "base/power_monitor/power_monitor.h" | |
| 19 #include "base/synchronization/lock.h" | |
| 20 #include "build/build_config.h" | |
| 21 #include "media/base/mac/corevideo_glue.h" | |
| 22 #include "media/base/mac/video_frame_mac.h" | |
| 23 #include "media/cast/common/rtp_time.h" | |
| 24 #include "media/cast/constants.h" | |
| 25 #include "media/cast/sender/video_frame_factory.h" | |
| 26 | 10 |
| 27 namespace media { | 11 namespace media { |
| 28 namespace cast { | |
| 29 | |
| 30 namespace { | |
| 31 | |
| 32 // Container for the associated data of a video frame being processed. | |
| 33 struct InProgressFrameEncode { | |
| 34 const RtpTimeTicks rtp_timestamp; | |
| 35 const base::TimeTicks reference_time; | |
| 36 const VideoEncoder::FrameEncodedCallback frame_encoded_callback; | |
| 37 | |
| 38 InProgressFrameEncode(RtpTimeTicks rtp, | |
| 39 base::TimeTicks r_time, | |
| 40 VideoEncoder::FrameEncodedCallback callback) | |
| 41 : rtp_timestamp(rtp), | |
| 42 reference_time(r_time), | |
| 43 frame_encoded_callback(callback) {} | |
| 44 }; | |
| 45 | 12 |
| 46 base::ScopedCFTypeRef<CFDictionaryRef> | 13 base::ScopedCFTypeRef<CFDictionaryRef> |
| 47 DictionaryWithKeysAndValues(CFTypeRef* keys, CFTypeRef* values, size_t size) { | 14 DictionaryWithKeysAndValues(CFTypeRef* keys, CFTypeRef* values, size_t size) { |
| 48 return base::ScopedCFTypeRef<CFDictionaryRef>(CFDictionaryCreate( | 15 return base::ScopedCFTypeRef<CFDictionaryRef>(CFDictionaryCreate( |
| 49 kCFAllocatorDefault, keys, values, size, &kCFTypeDictionaryKeyCallBacks, | 16 kCFAllocatorDefault, keys, values, size, &kCFTypeDictionaryKeyCallBacks, |
| 50 &kCFTypeDictionaryValueCallBacks)); | 17 &kCFTypeDictionaryValueCallBacks)); |
| 51 } | 18 } |
| 52 | 19 |
| 53 base::ScopedCFTypeRef<CFDictionaryRef> DictionaryWithKeyValue(CFTypeRef key, | 20 base::ScopedCFTypeRef<CFDictionaryRef> DictionaryWithKeyValue(CFTypeRef key, |
| 54 CFTypeRef value) { | 21 CFTypeRef value) { |
| (...skipping 147 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 202 CopyNalsToAnnexB<uint8_t>(bb_data, bb_size, annexb_buffer); | 169 CopyNalsToAnnexB<uint8_t>(bb_data, bb_size, annexb_buffer); |
| 203 } else if (nal_size_field_bytes == 2) { | 170 } else if (nal_size_field_bytes == 2) { |
| 204 CopyNalsToAnnexB<uint16_t>(bb_data, bb_size, annexb_buffer); | 171 CopyNalsToAnnexB<uint16_t>(bb_data, bb_size, annexb_buffer); |
| 205 } else if (nal_size_field_bytes == 4) { | 172 } else if (nal_size_field_bytes == 4) { |
| 206 CopyNalsToAnnexB<uint32_t>(bb_data, bb_size, annexb_buffer); | 173 CopyNalsToAnnexB<uint32_t>(bb_data, bb_size, annexb_buffer); |
| 207 } else { | 174 } else { |
| 208 NOTREACHED(); | 175 NOTREACHED(); |
| 209 } | 176 } |
| 210 } | 177 } |
| 211 | 178 |
| 212 } // namespace | 179 bool SetSessionProperty(base::ScopedCFTypeRef<VTCompressionSessionRef> session, |
| 213 | 180 const VideoToolboxGlue* const glue, |
| 214 class H264VideoToolboxEncoder::VideoFrameFactoryImpl | 181 CFStringRef key, |
| 215 : public base::RefCountedThreadSafe<VideoFrameFactoryImpl>, | 182 int32_t value) { |
| 216 public VideoFrameFactory { | 183 base::ScopedCFTypeRef<CFNumberRef> cfvalue( |
| 217 public: | 184 CFNumberCreate(nullptr, kCFNumberSInt32Type, &value)); |
| 218 // Type that proxies the VideoFrameFactory interface to this class. | 185 return glue->VTSessionSetProperty(session, key, cfvalue) == noErr; |
| 219 class Proxy; | |
| 220 | |
| 221 VideoFrameFactoryImpl(const base::WeakPtr<H264VideoToolboxEncoder>& encoder, | |
| 222 const scoped_refptr<CastEnvironment>& cast_environment) | |
| 223 : encoder_(encoder), cast_environment_(cast_environment) {} | |
| 224 | |
| 225 scoped_refptr<VideoFrame> MaybeCreateFrame( | |
| 226 const gfx::Size& frame_size, | |
| 227 base::TimeDelta timestamp) final { | |
| 228 if (frame_size.IsEmpty()) { | |
| 229 DVLOG(1) << "Rejecting empty video frame."; | |
| 230 return nullptr; | |
| 231 } | |
| 232 | |
| 233 base::AutoLock auto_lock(lock_); | |
| 234 | |
| 235 // If the pool size does not match, speculatively reset the encoder to use | |
| 236 // the new size and return null. Cache the new frame size right away and | |
| 237 // toss away the pixel buffer pool to avoid spurious tasks until the encoder | |
| 238 // is done resetting. | |
| 239 if (frame_size != pool_frame_size_) { | |
| 240 DVLOG(1) << "MaybeCreateFrame: Detected frame size change."; | |
| 241 cast_environment_->PostTask( | |
| 242 CastEnvironment::MAIN, FROM_HERE, | |
| 243 base::Bind(&H264VideoToolboxEncoder::UpdateFrameSize, encoder_, | |
| 244 frame_size)); | |
| 245 pool_frame_size_ = frame_size; | |
| 246 pool_.reset(); | |
| 247 return nullptr; | |
| 248 } | |
| 249 | |
| 250 if (!pool_) { | |
| 251 DVLOG(1) << "MaybeCreateFrame: No pixel buffer pool."; | |
| 252 return nullptr; | |
| 253 } | |
| 254 | |
| 255 // Allocate a pixel buffer from the pool and return a wrapper VideoFrame. | |
| 256 base::ScopedCFTypeRef<CVPixelBufferRef> buffer; | |
| 257 auto status = CVPixelBufferPoolCreatePixelBuffer(kCFAllocatorDefault, pool_, | |
| 258 buffer.InitializeInto()); | |
| 259 if (status != kCVReturnSuccess) { | |
| 260 DLOG(ERROR) << "CVPixelBufferPoolCreatePixelBuffer failed: " << status; | |
| 261 return nullptr; | |
| 262 } | |
| 263 | |
| 264 DCHECK(buffer); | |
| 265 return VideoFrame::WrapCVPixelBuffer(buffer, timestamp); | |
| 266 } | |
| 267 | |
| 268 void Update(const base::ScopedCFTypeRef<CVPixelBufferPoolRef>& pool, | |
| 269 const gfx::Size& frame_size) { | |
| 270 base::AutoLock auto_lock(lock_); | |
| 271 pool_ = pool; | |
| 272 pool_frame_size_ = frame_size; | |
| 273 } | |
| 274 | |
| 275 private: | |
| 276 friend class base::RefCountedThreadSafe<VideoFrameFactoryImpl>; | |
| 277 ~VideoFrameFactoryImpl() final {} | |
| 278 | |
| 279 base::Lock lock_; | |
| 280 base::ScopedCFTypeRef<CVPixelBufferPoolRef> pool_; | |
| 281 gfx::Size pool_frame_size_; | |
| 282 | |
| 283 // Weak back reference to the encoder and the cast envrionment so we can | |
| 284 // message the encoder when the frame size changes. | |
| 285 const base::WeakPtr<H264VideoToolboxEncoder> encoder_; | |
| 286 const scoped_refptr<CastEnvironment> cast_environment_; | |
| 287 | |
| 288 DISALLOW_COPY_AND_ASSIGN(VideoFrameFactoryImpl); | |
| 289 }; | |
| 290 | |
| 291 class H264VideoToolboxEncoder::VideoFrameFactoryImpl::Proxy | |
| 292 : public VideoFrameFactory { | |
| 293 public: | |
| 294 explicit Proxy( | |
| 295 const scoped_refptr<VideoFrameFactoryImpl>& video_frame_factory) | |
| 296 : video_frame_factory_(video_frame_factory) { | |
| 297 DCHECK(video_frame_factory_); | |
| 298 } | |
| 299 | |
| 300 scoped_refptr<VideoFrame> MaybeCreateFrame( | |
| 301 const gfx::Size& frame_size, | |
| 302 base::TimeDelta timestamp) final { | |
| 303 return video_frame_factory_->MaybeCreateFrame(frame_size, timestamp); | |
| 304 } | |
| 305 | |
| 306 private: | |
| 307 ~Proxy() final {} | |
| 308 | |
| 309 const scoped_refptr<VideoFrameFactoryImpl> video_frame_factory_; | |
| 310 | |
| 311 DISALLOW_COPY_AND_ASSIGN(Proxy); | |
| 312 }; | |
| 313 | |
| 314 // static | |
| 315 bool H264VideoToolboxEncoder::IsSupported( | |
| 316 const VideoSenderConfig& video_config) { | |
| 317 return video_config.codec == CODEC_VIDEO_H264 && VideoToolboxGlue::Get(); | |
| 318 } | 186 } |
| 319 | 187 |
| 320 H264VideoToolboxEncoder::H264VideoToolboxEncoder( | 188 bool SetSessionProperty(base::ScopedCFTypeRef<VTCompressionSessionRef> session, |
| 321 const scoped_refptr<CastEnvironment>& cast_environment, | 189 const VideoToolboxGlue* const glue, |
| 322 const VideoSenderConfig& video_config, | 190 CFStringRef key, |
| 323 const StatusChangeCallback& status_change_cb) | 191 bool value) { |
| 324 : cast_environment_(cast_environment), | 192 CFBooleanRef cfvalue = (value) ? kCFBooleanTrue : kCFBooleanFalse; |
| 325 videotoolbox_glue_(VideoToolboxGlue::Get()), | 193 return glue->VTSessionSetProperty(session, key, cfvalue) == noErr; |
| 326 video_config_(video_config), | |
| 327 status_change_cb_(status_change_cb), | |
| 328 last_frame_id_(kFirstFrameId - 1), | |
| 329 encode_next_frame_as_keyframe_(false), | |
| 330 power_suspended_(false), | |
| 331 weak_factory_(this) { | |
| 332 DCHECK(cast_environment_->CurrentlyOn(CastEnvironment::MAIN)); | |
| 333 DCHECK(!status_change_cb_.is_null()); | |
| 334 | |
| 335 OperationalStatus operational_status = | |
| 336 H264VideoToolboxEncoder::IsSupported(video_config) | |
| 337 ? STATUS_INITIALIZED | |
| 338 : STATUS_UNSUPPORTED_CODEC; | |
| 339 cast_environment_->PostTask( | |
| 340 CastEnvironment::MAIN, FROM_HERE, | |
| 341 base::Bind(status_change_cb_, operational_status)); | |
| 342 | |
| 343 if (operational_status == STATUS_INITIALIZED) { | |
| 344 // Create the shared video frame factory. It persists for the combined | |
| 345 // lifetime of the encoder and all video frame factory proxies created by | |
| 346 // |CreateVideoFrameFactory| that reference it. | |
| 347 video_frame_factory_ = | |
| 348 scoped_refptr<VideoFrameFactoryImpl>(new VideoFrameFactoryImpl( | |
| 349 weak_factory_.GetWeakPtr(), cast_environment_)); | |
| 350 | |
| 351 // Register for power state changes. | |
| 352 auto power_monitor = base::PowerMonitor::Get(); | |
| 353 if (power_monitor) { | |
| 354 power_monitor->AddObserver(this); | |
| 355 VLOG(1) << "Registered for power state changes."; | |
| 356 } else { | |
| 357 DLOG(WARNING) << "No power monitor. Process suspension will invalidate " | |
| 358 "the encoder."; | |
| 359 } | |
| 360 } | |
| 361 } | 194 } |
| 362 | 195 |
| 363 H264VideoToolboxEncoder::~H264VideoToolboxEncoder() { | 196 bool SetSessionProperty(base::ScopedCFTypeRef<VTCompressionSessionRef> session, |
| 364 DestroyCompressionSession(); | 197 const VideoToolboxGlue* const glue, |
| 365 | 198 CFStringRef key, |
| 366 // If video_frame_factory_ is not null, the encoder registered for power state | 199 CFStringRef value) { |
| 367 // changes in the ctor and it must now unregister. | 200 return glue->VTSessionSetProperty(session, key, value) == noErr; |
| 368 if (video_frame_factory_) { | |
| 369 auto power_monitor = base::PowerMonitor::Get(); | |
| 370 if (power_monitor) | |
| 371 power_monitor->RemoveObserver(this); | |
| 372 } | |
| 373 } | 201 } |
| 374 | 202 |
| 375 void H264VideoToolboxEncoder::ResetCompressionSession() { | |
| 376 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 377 | |
| 378 // Ignore reset requests while power suspended. | |
| 379 if (power_suspended_) | |
| 380 return; | |
| 381 | |
| 382 // Notify that we're resetting the encoder. | |
| 383 cast_environment_->PostTask( | |
| 384 CastEnvironment::MAIN, FROM_HERE, | |
| 385 base::Bind(status_change_cb_, STATUS_CODEC_REINIT_PENDING)); | |
| 386 | |
| 387 // Destroy the current session, if any. | |
| 388 DestroyCompressionSession(); | |
| 389 | |
| 390 // On OS X, allow the hardware encoder. Don't require it, it does not support | |
| 391 // all configurations (some of which are used for testing). | |
| 392 base::ScopedCFTypeRef<CFDictionaryRef> encoder_spec; | |
| 393 #if !defined(OS_IOS) | |
| 394 encoder_spec = DictionaryWithKeyValue( | |
| 395 videotoolbox_glue_ | |
| 396 ->kVTVideoEncoderSpecification_EnableHardwareAcceleratedVideoEncoder()
, | |
| 397 kCFBooleanTrue); | |
| 398 #endif | |
| 399 | |
| 400 // Force 420v so that clients can easily use these buffers as GPU textures. | |
| 401 const int format[] = { | |
| 402 CoreVideoGlue::kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange}; | |
| 403 | |
| 404 // Keep these attachment settings in-sync with those in ConfigureSession(). | |
| 405 CFTypeRef attachments_keys[] = {kCVImageBufferColorPrimariesKey, | |
| 406 kCVImageBufferTransferFunctionKey, | |
| 407 kCVImageBufferYCbCrMatrixKey}; | |
| 408 CFTypeRef attachments_values[] = {kCVImageBufferColorPrimaries_ITU_R_709_2, | |
| 409 kCVImageBufferTransferFunction_ITU_R_709_2, | |
| 410 kCVImageBufferYCbCrMatrix_ITU_R_709_2}; | |
| 411 CFTypeRef buffer_attributes_keys[] = {kCVPixelBufferPixelFormatTypeKey, | |
| 412 kCVBufferPropagatedAttachmentsKey}; | |
| 413 CFTypeRef buffer_attributes_values[] = { | |
| 414 ArrayWithIntegers(format, arraysize(format)).release(), | |
| 415 DictionaryWithKeysAndValues(attachments_keys, attachments_values, | |
| 416 arraysize(attachments_keys)).release()}; | |
| 417 const base::ScopedCFTypeRef<CFDictionaryRef> buffer_attributes = | |
| 418 DictionaryWithKeysAndValues(buffer_attributes_keys, | |
| 419 buffer_attributes_values, | |
| 420 arraysize(buffer_attributes_keys)); | |
| 421 for (auto& v : buffer_attributes_values) | |
| 422 CFRelease(v); | |
| 423 | |
| 424 // Create the compression session. | |
| 425 | |
| 426 // Note that the encoder object is given to the compression session as the | |
| 427 // callback context using a raw pointer. The C API does not allow us to use a | |
| 428 // smart pointer, nor is this encoder ref counted. However, this is still | |
| 429 // safe, because we 1) we own the compression session and 2) we tear it down | |
| 430 // safely. When destructing the encoder, the compression session is flushed | |
| 431 // and invalidated. Internally, VideoToolbox will join all of its threads | |
| 432 // before returning to the client. Therefore, when control returns to us, we | |
| 433 // are guaranteed that the output callback will not execute again. | |
| 434 OSStatus status = videotoolbox_glue_->VTCompressionSessionCreate( | |
| 435 kCFAllocatorDefault, frame_size_.width(), frame_size_.height(), | |
| 436 CoreMediaGlue::kCMVideoCodecType_H264, encoder_spec, buffer_attributes, | |
| 437 nullptr /* compressedDataAllocator */, | |
| 438 &H264VideoToolboxEncoder::CompressionCallback, | |
| 439 reinterpret_cast<void*>(this), compression_session_.InitializeInto()); | |
| 440 if (status != noErr) { | |
| 441 DLOG(ERROR) << " VTCompressionSessionCreate failed: " << status; | |
| 442 // Notify that reinitialization has failed. | |
| 443 cast_environment_->PostTask( | |
| 444 CastEnvironment::MAIN, FROM_HERE, | |
| 445 base::Bind(status_change_cb_, STATUS_CODEC_INIT_FAILED)); | |
| 446 return; | |
| 447 } | |
| 448 | |
| 449 // Configure the session (apply session properties based on the current state | |
| 450 // of the encoder, experimental tuning and requirements). | |
| 451 ConfigureCompressionSession(); | |
| 452 | |
| 453 // Update the video frame factory. | |
| 454 base::ScopedCFTypeRef<CVPixelBufferPoolRef> pool( | |
| 455 videotoolbox_glue_->VTCompressionSessionGetPixelBufferPool( | |
| 456 compression_session_), | |
| 457 base::scoped_policy::RETAIN); | |
| 458 video_frame_factory_->Update(pool, frame_size_); | |
| 459 | |
| 460 // Notify that reinitialization is done. | |
| 461 cast_environment_->PostTask( | |
| 462 CastEnvironment::MAIN, FROM_HERE, | |
| 463 base::Bind(status_change_cb_, STATUS_INITIALIZED)); | |
| 464 } | |
| 465 | |
| 466 void H264VideoToolboxEncoder::ConfigureCompressionSession() { | |
| 467 SetSessionProperty( | |
| 468 videotoolbox_glue_->kVTCompressionPropertyKey_ProfileLevel(), | |
| 469 videotoolbox_glue_->kVTProfileLevel_H264_Main_AutoLevel()); | |
| 470 SetSessionProperty(videotoolbox_glue_->kVTCompressionPropertyKey_RealTime(), | |
| 471 true); | |
| 472 SetSessionProperty( | |
| 473 videotoolbox_glue_->kVTCompressionPropertyKey_AllowFrameReordering(), | |
| 474 false); | |
| 475 SetSessionProperty( | |
| 476 videotoolbox_glue_->kVTCompressionPropertyKey_MaxKeyFrameInterval(), 240); | |
| 477 SetSessionProperty( | |
| 478 videotoolbox_glue_ | |
| 479 ->kVTCompressionPropertyKey_MaxKeyFrameIntervalDuration(), | |
| 480 240); | |
| 481 // TODO(jfroy): implement better bitrate control | |
| 482 // https://crbug.com/425352 | |
| 483 SetSessionProperty( | |
| 484 videotoolbox_glue_->kVTCompressionPropertyKey_AverageBitRate(), | |
| 485 (video_config_.min_bitrate + video_config_.max_bitrate) / 2); | |
| 486 SetSessionProperty( | |
| 487 videotoolbox_glue_->kVTCompressionPropertyKey_ExpectedFrameRate(), | |
| 488 video_config_.max_frame_rate); | |
| 489 // Keep these attachment settings in-sync with those in Initialize(). | |
| 490 SetSessionProperty( | |
| 491 videotoolbox_glue_->kVTCompressionPropertyKey_ColorPrimaries(), | |
| 492 kCVImageBufferColorPrimaries_ITU_R_709_2); | |
| 493 SetSessionProperty( | |
| 494 videotoolbox_glue_->kVTCompressionPropertyKey_TransferFunction(), | |
| 495 kCVImageBufferTransferFunction_ITU_R_709_2); | |
| 496 SetSessionProperty( | |
| 497 videotoolbox_glue_->kVTCompressionPropertyKey_YCbCrMatrix(), | |
| 498 kCVImageBufferYCbCrMatrix_ITU_R_709_2); | |
| 499 if (video_config_.max_number_of_video_buffers_used > 0) { | |
| 500 SetSessionProperty( | |
| 501 videotoolbox_glue_->kVTCompressionPropertyKey_MaxFrameDelayCount(), | |
| 502 video_config_.max_number_of_video_buffers_used); | |
| 503 } | |
| 504 } | |
| 505 | |
| 506 void H264VideoToolboxEncoder::DestroyCompressionSession() { | |
| 507 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 508 | |
| 509 // If the compression session exists, invalidate it. This blocks until all | |
| 510 // pending output callbacks have returned and any internal threads have | |
| 511 // joined, ensuring no output callback ever sees a dangling encoder pointer. | |
| 512 // | |
| 513 // Before destroying the compression session, the video frame factory's pool | |
| 514 // is updated to null so that no thread will produce new video frames via the | |
| 515 // factory until a new compression session is created. The current frame size | |
| 516 // is passed to prevent the video frame factory from posting |UpdateFrameSize| | |
| 517 // tasks. Indeed, |DestroyCompressionSession| is either called from | |
| 518 // |ResetCompressionSession|, in which case a new pool and frame size will be | |
| 519 // set, or from callsites that require that there be no compression session | |
| 520 // (ex: the dtor). | |
| 521 if (compression_session_) { | |
| 522 video_frame_factory_->Update( | |
| 523 base::ScopedCFTypeRef<CVPixelBufferPoolRef>(nullptr), frame_size_); | |
| 524 videotoolbox_glue_->VTCompressionSessionInvalidate(compression_session_); | |
| 525 compression_session_.reset(); | |
| 526 } | |
| 527 } | |
| 528 | |
| 529 bool H264VideoToolboxEncoder::EncodeVideoFrame( | |
| 530 const scoped_refptr<media::VideoFrame>& video_frame, | |
| 531 const base::TimeTicks& reference_time, | |
| 532 const FrameEncodedCallback& frame_encoded_callback) { | |
| 533 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 534 DCHECK(!frame_encoded_callback.is_null()); | |
| 535 | |
| 536 // Reject empty video frames. | |
| 537 const gfx::Size frame_size = video_frame->visible_rect().size(); | |
| 538 if (frame_size.IsEmpty()) { | |
| 539 DVLOG(1) << "Rejecting empty video frame."; | |
| 540 return false; | |
| 541 } | |
| 542 | |
| 543 // Handle frame size changes. This will reset the compression session. | |
| 544 if (frame_size != frame_size_) { | |
| 545 DVLOG(1) << "EncodeVideoFrame: Detected frame size change."; | |
| 546 UpdateFrameSize(frame_size); | |
| 547 } | |
| 548 | |
| 549 // Need a compression session to continue. | |
| 550 if (!compression_session_) { | |
| 551 DLOG(ERROR) << "No compression session."; | |
| 552 return false; | |
| 553 } | |
| 554 | |
| 555 // Wrap the VideoFrame in a CVPixelBuffer. In all cases, no data will be | |
| 556 // copied. If the VideoFrame was created by this encoder's video frame | |
| 557 // factory, then the returned CVPixelBuffer will have been obtained from the | |
| 558 // compression session's pixel buffer pool. This will eliminate a copy of the | |
| 559 // frame into memory visible by the hardware encoder. The VideoFrame's | |
| 560 // lifetime is extended for the lifetime of the returned CVPixelBuffer. | |
| 561 auto pixel_buffer = media::WrapVideoFrameInCVPixelBuffer(*video_frame); | |
| 562 if (!pixel_buffer) { | |
| 563 DLOG(ERROR) << "WrapVideoFrameInCVPixelBuffer failed."; | |
| 564 return false; | |
| 565 } | |
| 566 | |
| 567 // Convert the frame timestamp to CMTime. | |
| 568 auto timestamp_cm = CoreMediaGlue::CMTimeMake( | |
| 569 (reference_time - base::TimeTicks()).InMicroseconds(), USEC_PER_SEC); | |
| 570 | |
| 571 // Wrap information we'll need after the frame is encoded in a heap object. | |
| 572 // We'll get the pointer back from the VideoToolbox completion callback. | |
| 573 scoped_ptr<InProgressFrameEncode> request(new InProgressFrameEncode( | |
| 574 RtpTimeTicks::FromTimeDelta(video_frame->timestamp(), kVideoFrequency), | |
| 575 reference_time, frame_encoded_callback)); | |
| 576 | |
| 577 // Build a suitable frame properties dictionary for keyframes. | |
| 578 base::ScopedCFTypeRef<CFDictionaryRef> frame_props; | |
| 579 if (encode_next_frame_as_keyframe_) { | |
| 580 frame_props = DictionaryWithKeyValue( | |
| 581 videotoolbox_glue_->kVTEncodeFrameOptionKey_ForceKeyFrame(), | |
| 582 kCFBooleanTrue); | |
| 583 encode_next_frame_as_keyframe_ = false; | |
| 584 } | |
| 585 | |
| 586 // Submit the frame to the compression session. The function returns as soon | |
| 587 // as the frame has been enqueued. | |
| 588 OSStatus status = videotoolbox_glue_->VTCompressionSessionEncodeFrame( | |
| 589 compression_session_, pixel_buffer, timestamp_cm, | |
| 590 CoreMediaGlue::CMTime{0, 0, 0, 0}, frame_props, | |
| 591 reinterpret_cast<void*>(request.release()), nullptr); | |
| 592 if (status != noErr) { | |
| 593 DLOG(ERROR) << " VTCompressionSessionEncodeFrame failed: " << status; | |
| 594 return false; | |
| 595 } | |
| 596 | |
| 597 return true; | |
| 598 } | |
| 599 | |
| 600 void H264VideoToolboxEncoder::UpdateFrameSize(const gfx::Size& size_needed) { | |
| 601 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 602 | |
| 603 // Our video frame factory posts a task to update the frame size when its | |
| 604 // cache of the frame size differs from what the client requested. To avoid | |
| 605 // spurious encoder resets, check again here. | |
| 606 if (size_needed == frame_size_) { | |
| 607 DCHECK(compression_session_); | |
| 608 return; | |
| 609 } | |
| 610 | |
| 611 VLOG(1) << "Resetting compression session (for frame size change from " | |
| 612 << frame_size_.ToString() << " to " << size_needed.ToString() << ")."; | |
| 613 | |
| 614 // If there is an existing session, finish every pending frame. | |
| 615 if (compression_session_) { | |
| 616 EmitFrames(); | |
| 617 } | |
| 618 | |
| 619 // Store the new frame size. | |
| 620 frame_size_ = size_needed; | |
| 621 | |
| 622 // Reset the compression session. | |
| 623 ResetCompressionSession(); | |
| 624 } | |
| 625 | |
| 626 void H264VideoToolboxEncoder::SetBitRate(int /*new_bit_rate*/) { | |
| 627 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 628 // VideoToolbox does not seem to support bitrate reconfiguration. | |
| 629 } | |
| 630 | |
| 631 void H264VideoToolboxEncoder::GenerateKeyFrame() { | |
| 632 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 633 encode_next_frame_as_keyframe_ = true; | |
| 634 } | |
| 635 | |
| 636 scoped_ptr<VideoFrameFactory> | |
| 637 H264VideoToolboxEncoder::CreateVideoFrameFactory() { | |
| 638 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 639 return scoped_ptr<VideoFrameFactory>( | |
| 640 new VideoFrameFactoryImpl::Proxy(video_frame_factory_)); | |
| 641 } | |
| 642 | |
| 643 void H264VideoToolboxEncoder::EmitFrames() { | |
| 644 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 645 if (!compression_session_) | |
| 646 return; | |
| 647 | |
| 648 OSStatus status = videotoolbox_glue_->VTCompressionSessionCompleteFrames( | |
| 649 compression_session_, CoreMediaGlue::CMTime{0, 0, 0, 0}); | |
| 650 if (status != noErr) { | |
| 651 DLOG(ERROR) << " VTCompressionSessionCompleteFrames failed: " << status; | |
| 652 } | |
| 653 } | |
| 654 | |
| 655 void H264VideoToolboxEncoder::OnSuspend() { | |
| 656 VLOG(1) | |
| 657 << "OnSuspend: Emitting all frames and destroying compression session."; | |
| 658 EmitFrames(); | |
| 659 DestroyCompressionSession(); | |
| 660 power_suspended_ = true; | |
| 661 } | |
| 662 | |
| 663 void H264VideoToolboxEncoder::OnResume() { | |
| 664 power_suspended_ = false; | |
| 665 | |
| 666 // Reset the compression session only if the frame size is not zero (which | |
| 667 // will obviously fail). It is possible for the frame size to be zero if no | |
| 668 // frame was submitted for encoding or requested from the video frame factory | |
| 669 // before suspension. | |
| 670 if (!frame_size_.IsEmpty()) { | |
| 671 VLOG(1) << "OnResume: Resetting compression session."; | |
| 672 ResetCompressionSession(); | |
| 673 } | |
| 674 } | |
| 675 | |
| 676 bool H264VideoToolboxEncoder::SetSessionProperty(CFStringRef key, | |
| 677 int32_t value) { | |
| 678 base::ScopedCFTypeRef<CFNumberRef> cfvalue( | |
| 679 CFNumberCreate(nullptr, kCFNumberSInt32Type, &value)); | |
| 680 return videotoolbox_glue_->VTSessionSetProperty(compression_session_, key, | |
| 681 cfvalue) == noErr; | |
| 682 } | |
| 683 | |
| 684 bool H264VideoToolboxEncoder::SetSessionProperty(CFStringRef key, bool value) { | |
| 685 CFBooleanRef cfvalue = (value) ? kCFBooleanTrue : kCFBooleanFalse; | |
| 686 return videotoolbox_glue_->VTSessionSetProperty(compression_session_, key, | |
| 687 cfvalue) == noErr; | |
| 688 } | |
| 689 | |
| 690 bool H264VideoToolboxEncoder::SetSessionProperty(CFStringRef key, | |
| 691 CFStringRef value) { | |
| 692 return videotoolbox_glue_->VTSessionSetProperty(compression_session_, key, | |
| 693 value) == noErr; | |
| 694 } | |
| 695 | |
| 696 void H264VideoToolboxEncoder::CompressionCallback(void* encoder_opaque, | |
| 697 void* request_opaque, | |
| 698 OSStatus status, | |
| 699 VTEncodeInfoFlags info, | |
| 700 CMSampleBufferRef sbuf) { | |
| 701 auto encoder = reinterpret_cast<H264VideoToolboxEncoder*>(encoder_opaque); | |
| 702 const scoped_ptr<InProgressFrameEncode> request( | |
| 703 reinterpret_cast<InProgressFrameEncode*>(request_opaque)); | |
| 704 bool keyframe = false; | |
| 705 bool has_frame_data = false; | |
| 706 | |
| 707 if (status != noErr) { | |
| 708 DLOG(ERROR) << " encoding failed: " << status; | |
| 709 encoder->cast_environment_->PostTask( | |
| 710 CastEnvironment::MAIN, FROM_HERE, | |
| 711 base::Bind(encoder->status_change_cb_, STATUS_CODEC_RUNTIME_ERROR)); | |
| 712 } else if ((info & VideoToolboxGlue::kVTEncodeInfo_FrameDropped)) { | |
| 713 DVLOG(2) << " frame dropped"; | |
| 714 } else { | |
| 715 auto sample_attachments = | |
| 716 static_cast<CFDictionaryRef>(CFArrayGetValueAtIndex( | |
| 717 CoreMediaGlue::CMSampleBufferGetSampleAttachmentsArray(sbuf, true), | |
| 718 0)); | |
| 719 | |
| 720 // If the NotSync key is not present, it implies Sync, which indicates a | |
| 721 // keyframe (at least I think, VT documentation is, erm, sparse). Could | |
| 722 // alternatively use kCMSampleAttachmentKey_DependsOnOthers == false. | |
| 723 keyframe = !CFDictionaryContainsKey( | |
| 724 sample_attachments, | |
| 725 CoreMediaGlue::kCMSampleAttachmentKey_NotSync()); | |
| 726 has_frame_data = true; | |
| 727 } | |
| 728 | |
| 729 // Increment the encoder-scoped frame id and assign the new value to this | |
| 730 // frame. VideoToolbox calls the output callback serially, so this is safe. | |
| 731 const uint32_t frame_id = ++encoder->last_frame_id_; | |
| 732 | |
| 733 scoped_ptr<SenderEncodedFrame> encoded_frame(new SenderEncodedFrame()); | |
| 734 encoded_frame->frame_id = frame_id; | |
| 735 encoded_frame->reference_time = request->reference_time; | |
| 736 encoded_frame->rtp_timestamp = request->rtp_timestamp; | |
| 737 if (keyframe) { | |
| 738 encoded_frame->dependency = EncodedFrame::KEY; | |
| 739 encoded_frame->referenced_frame_id = frame_id; | |
| 740 } else { | |
| 741 encoded_frame->dependency = EncodedFrame::DEPENDENT; | |
| 742 // H.264 supports complex frame reference schemes (multiple reference | |
| 743 // frames, slice references, backward and forward references, etc). Cast | |
| 744 // doesn't support the concept of forward-referencing frame dependencies or | |
| 745 // multiple frame dependencies; so pretend that all frames are only | |
| 746 // decodable after their immediately preceding frame is decoded. This will | |
| 747 // ensure a Cast receiver only attempts to decode the frames sequentially | |
| 748 // and in order. Furthermore, the encoder is configured to never use forward | |
| 749 // references (see |kVTCompressionPropertyKey_AllowFrameReordering|). There | |
| 750 // is no way to prevent multiple reference frames. | |
| 751 encoded_frame->referenced_frame_id = frame_id - 1; | |
| 752 } | |
| 753 | |
| 754 if (has_frame_data) | |
| 755 CopySampleBufferToAnnexBBuffer(sbuf, &encoded_frame->data, keyframe); | |
| 756 | |
| 757 // TODO(miu): Compute and populate the |deadline_utilization| and | |
| 758 // |lossy_utilization| performance metrics in |encoded_frame|. | |
| 759 | |
| 760 encoded_frame->encode_completion_time = | |
| 761 encoder->cast_environment_->Clock()->NowTicks(); | |
| 762 encoder->cast_environment_->PostTask( | |
| 763 CastEnvironment::MAIN, FROM_HERE, | |
| 764 base::Bind(request->frame_encoded_callback, | |
| 765 base::Passed(&encoded_frame))); | |
| 766 } | |
| 767 | |
| 768 } // namespace cast | |
| 769 } // namespace media | 203 } // namespace media |
| OLD | NEW |