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/interface/logging.h"
14
15#include "webrtc/modules/desktop_capture/win/desktop.h"
16
17namespace webrtc {
18
19ScopedThreadDesktop::ScopedThreadDesktop()
20    : initial_(Desktop::GetThreadDesktop()) {
21}
22
23ScopedThreadDesktop::~ScopedThreadDesktop() {
24  Revert();
25}
26
27bool 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
35void ScopedThreadDesktop::Revert() {
36  if (assigned_.get() != NULL) {
37    initial_->SetThreadDesktop();
38    assigned_.reset();
39  }
40}
41
42bool ScopedThreadDesktop::SetThreadDesktop(Desktop* desktop) {
43  Revert();
44
45  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