1 /*
2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #include "webrtc/modules/desktop_capture/win/scoped_thread_desktop.h"
12 
13 #include "webrtc/system_wrappers/include/logging.h"
14 
15 #include "webrtc/modules/desktop_capture/win/desktop.h"
16 
17 namespace webrtc {
18 
ScopedThreadDesktop()19 ScopedThreadDesktop::ScopedThreadDesktop()
20     : initial_(Desktop::GetThreadDesktop()) {
21 }
22 
~ScopedThreadDesktop()23 ScopedThreadDesktop::~ScopedThreadDesktop() {
24   Revert();
25 }
26 
IsSame(const Desktop & desktop)27 bool ScopedThreadDesktop::IsSame(const Desktop& desktop) {
28   if (assigned_.get() != NULL) {
29     return assigned_->IsSame(desktop);
30   } else {
31     return initial_->IsSame(desktop);
32   }
33 }
34 
Revert()35 void ScopedThreadDesktop::Revert() {
36   if (assigned_.get() != NULL) {
37     initial_->SetThreadDesktop();
38     assigned_.reset();
39   }
40 }
41 
SetThreadDesktop(Desktop * desktop)42 bool ScopedThreadDesktop::SetThreadDesktop(Desktop* desktop) {
43   Revert();
44 
45   rtc::scoped_ptr<Desktop> scoped_desktop(desktop);
46 
47   if (initial_->IsSame(*desktop))
48     return true;
49 
50   if (!desktop->SetThreadDesktop())
51     return false;
52 
53   assigned_.reset(scoped_desktop.release());
54   return true;
55 }
56 
57 }  // namespace webrtc
58