ui/shell_dialogs/base_shell_dialog_win.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "ui/shell_dialogs/base_shell_dialog_win.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/threading/thread.h"
  10 #include "base/win/scoped_com_initializer.h"
  11
  12 namespace ui {
  13
  14 // static
  15 BaseShellDialogImpl::Owners BaseShellDialogImpl::owners_;
  16 int BaseShellDialogImpl::instance_count_ = 0;
  17
  18 BaseShellDialogImpl::BaseShellDialogImpl() {
  19   ++instance_count_;
  20 }
  21
  22 BaseShellDialogImpl::~BaseShellDialogImpl() {
  23   // All runs should be complete by the time this is called!
  24   if (--instance_count_ == 0)
  25     DCHECK(owners_.empty());
  26 }
  27
  28 BaseShellDialogImpl::RunState BaseShellDialogImpl::BeginRun(HWND owner) {
  29   // Cannot run a modal shell dialog if one is already running for this owner.
  30   DCHECK(!IsRunningDialogForOwner(owner));
  31   // The owner must be a top level window, otherwise we could end up with two
  32   // entries in our map for the same top level window.
  33   DCHECK(!owner || owner == GetAncestor(owner, GA_ROOT));
  34   RunState run_state;
  35   run_state.dialog_thread = CreateDialogThread();
  36   run_state.owner = owner;
  37   if (owner) {
  38     owners_.insert(owner);
  39     DisableOwner(owner);
  40   }
  41   return run_state;
  42 }
  43
  44 void BaseShellDialogImpl::EndRun(RunState run_state) {
  45   if (run_state.owner) {
  46     DCHECK(IsRunningDialogForOwner(run_state.owner));
  47     EnableOwner(run_state.owner);
  48     DCHECK(owners_.find(run_state.owner) != owners_.end());
  49     owners_.erase(run_state.owner);
  50   }
  51   DCHECK(run_state.dialog_thread);
  52   delete run_state.dialog_thread;
  53 }
  54
  55 bool BaseShellDialogImpl::IsRunningDialogForOwner(HWND owner) const {
  56   return (owner && owners_.find(owner) != owners_.end());
  57 }
  58
  59 void BaseShellDialogImpl::DisableOwner(HWND owner) {
  60   if (IsWindow(owner))
  61     EnableWindow(owner, FALSE);
  62 }
  63
  64 // static
  65 base::Thread* BaseShellDialogImpl::CreateDialogThread() {
  66   base::Thread* thread = new base::Thread("Chrome_ShellDialogThread");
  67   thread->init_com_with_mta(false);
  68   bool started = thread->Start();
  69   DCHECK(started);
  70   return thread;
  71 }
  72
  73 void BaseShellDialogImpl::EnableOwner(HWND owner) {
  74   if (IsWindow(owner))
  75     EnableWindow(owner, TRUE);
  76 }
  77
  78 }  // namespace ui