[WebRTC] Two DirectX capturers cannot work concurrently

webrtc/modules/desktop_capture/win/dxgi_output_duplicator.cc

 /*
  *  Copyright (c) 2016 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 100 matching lines @@
   if (error.Error() != S_OK) {
     LOG(LS_ERROR) << "Failed to release frame from IDXGIOutputDuplication, "
                      "error"
                   << error.ErrorMessage() << ", code " << error.Error();
     return false;
   }
   return true;
 }
 
 bool DxgiOutputDuplicator::Duplicate(Context* context,
-                                     const DesktopFrame* last_frame,
-                                     const DesktopVector offset,
-                                     DesktopFrame* target) {
+                                     const DesktopVector& offset,
+                                     SharedDesktopFrame* target) {
   RTC_DCHECK(duplication_);
   RTC_DCHECK(texture_);
   RTC_DCHECK(target);
+  if (!DesktopRect::MakeSize(target->size())
+           .ContainsRect(TranslatedDesktopRect(offset))) {
+    // target size is not large enough to cover current output region.
+    return false;
+  }
+
   DXGI_OUTDUPL_FRAME_INFO frame_info;
   memset(&frame_info, 0, sizeof(frame_info));
   ComPtr<IDXGIResource> resource;
   _com_error error = duplication_->AcquireNextFrame(
       kAcquireTimeoutMs, &frame_info, resource.GetAddressOf());
   if (error.Error() != S_OK && error.Error() != DXGI_ERROR_WAIT_TIMEOUT) {
     LOG(LS_ERROR) << "Failed to capture frame, error " << error.ErrorMessage()
                   << ", code " << error.Error();
     return false;
   }
 
   // We need to merge updated region with the one from last frame, since current
   // frame contains the content one frame before. Note, this is for double
   // buffering implementation, as what we have in ScreenCapturerWinDirectx. If
   // a consumer uses single buffering, we should clear context->updated_region
   // after it has been merged to updated_region.
-  DesktopRegion updated_region = context->updated_region;
+  DesktopRegion updated_region;
+  updated_region.Swap(&context->updated_region);
   if (error.Error() == S_OK && frame_info.AccumulatedFrames > 0) {
     DetectUpdatedRegion(frame_info, offset, &context->updated_region);
+    if (!texture_->CopyFrom(frame_info, resource.Get(),
+                            context->updated_region)) {
+      return false;
+    }
     SpreadContextChange(context);
     updated_region.AddRegion(context->updated_region);
-    if (!texture_->CopyFrom(frame_info, resource.Get(), updated_region)) {
-      return false;
-    }
 
     const DesktopFrame& source = texture_->AsDesktopFrame();
-    DesktopRect target_rect(DesktopRect::MakeSize(target->size()));
     for (DesktopRegion::Iterator it(updated_region); !it.IsAtEnd();
          it.Advance()) {
-      if (!target_rect.ContainsRect(it.rect())) {
-        // target size is not large enough to copy the pixel from texture.
-        return false;
-      }
-      target->CopyPixelsFrom(source, it.rect().top_left().subtract(offset),
-                             it.rect());
+      target->CopyPixelsFrom(source, SourceRect(it.rect()).top_left(),
+                             TargetRect(it.rect(), offset));
     }
+    last_frame_ = target->Share();
+    last_frame_offset_ = offset;
     target->mutable_updated_region()->AddRegion(updated_region);
     return texture_->Release() && ReleaseFrame();
   }
 
-  if (last_frame != nullptr) {
-    // DxgiOutputDuplicatorContainer::Duplicate() makes sure target size and
-    // last frame size are consistent.
-    RTC_DCHECK(target->size().equals(last_frame->size()));
+  if (last_frame_) {
     // No change since last frame or AcquireNextFrame() timed out, we will
     // export last frame to the target.
     context->updated_region.Clear();

[Sergey Ulanov, 2016/09/02 23:57:35]

Don't need this anymore

[Hzj_jie, 2016/09/05 21:24:22]

Done.

     for (DesktopRegion::Iterator it(updated_region); !it.IsAtEnd();

[Sergey Ulanov, 2016/09/02 23:57:35]

This duplicates the code that you have above in lines 160-168. You always want
to copy from last_frame_. Remove return from line 168 and always run this code?

[Hzj_jie, 2016/09/05 21:24:22]

The code between 160 - 168 copies from texture_->AsDesktopFrame(). This is do a
little bit confusing.

          it.Advance()) {
-      target->CopyPixelsFrom(*last_frame, it.rect().top_left(), it.rect());
+      target->CopyPixelsFrom(*last_frame_, SourceRect(it.rect()).top_left(),
+                             TargetRect(it.rect(), offset));
     }
     target->mutable_updated_region()->AddRegion(updated_region);
   }
   // If AcquireNextFrame() failed with timeout error, we do not need to release
   // the frame.
   return error.Error() == DXGI_ERROR_WAIT_TIMEOUT || ReleaseFrame();
 }
 
 DesktopRect DxgiOutputDuplicator::TranslatedDesktopRect(
-    const DesktopVector offset) {
+    const DesktopVector& offset) {
   DesktopRect result(DesktopRect::MakeSize(desktop_rect_.size()));
   result.Translate(offset);
   return result;
 }
 
 void DxgiOutputDuplicator::DetectUpdatedRegion(
     const DXGI_OUTDUPL_FRAME_INFO& frame_info,
-    const DesktopVector offset,
+    const DesktopVector& offset,
     DesktopRegion* updated_region) {
   if (DoDetectUpdatedRegion(frame_info, updated_region)) {
     updated_region->Translate(offset.x(), offset.y());
     // Make sure even a region returned by Windows API is out of the scope of
     // desktop_rect_, we still won't export it to the target DesktopFrame.
     updated_region->IntersectWith(TranslatedDesktopRect(offset));
   } else {
     updated_region->SetRect(TranslatedDesktopRect(offset));
   }
 }
@@ skipping 89 matching lines @@
 }
 
 void DxgiOutputDuplicator::SpreadContextChange(const Context* const source) {
   for (Context* dest : contexts_) {
     if (dest != source) {
       dest->updated_region.AddRegion(source->updated_region);
     }
   }
 }
 
+DesktopRect DxgiOutputDuplicator::SourceRect(DesktopRect rect) {
+  // |texture_|->AsDesktopFrame() starts from (0, 0).
+  rect.Translate(-desktop_rect_.left(), -desktop_rect_.top());
+  return rect;
+}
+
+DesktopRect DxgiOutputDuplicator::TargetRect(DesktopRect rect,
+                                             DesktopVector offset) {
+  rect = SourceRect(rect);
+  rect.Translate(offset);
+  return rect;
+}
+
 }  // namespace webrtc