ui/views/accessibility/ax_aura_obj_cache.cc

   1 // Copyright 2014 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/views/accessibility/ax_aura_obj_cache.h"
   6
   7 #include "base/memory/singleton.h"
   8 #include "base/stl_util.h"
   9 #include "ui/aura/window.h"
  10 #include "ui/views/accessibility/ax_aura_obj_wrapper.h"
  11 #include "ui/views/accessibility/ax_view_obj_wrapper.h"
  12 #include "ui/views/accessibility/ax_widget_obj_wrapper.h"
  13 #include "ui/views/accessibility/ax_window_obj_wrapper.h"
  14 #include "ui/views/view.h"
  15 #include "ui/views/widget/widget.h"
  16
  17 namespace views {
  18
  19 // static
  20 AXAuraObjCache* AXAuraObjCache::GetInstance() {
  21   return Singleton<AXAuraObjCache>::get();
  22 }
  23
  24 AXAuraObjWrapper* AXAuraObjCache::GetOrCreate(View* view) {
  25   return CreateInternal<AXViewObjWrapper>(view, view_to_id_map_);
  26 }
  27
  28 AXAuraObjWrapper* AXAuraObjCache::GetOrCreate(Widget* widget) {
  29   return CreateInternal<AXWidgetObjWrapper>(widget, widget_to_id_map_);
  30 }
  31
  32 AXAuraObjWrapper* AXAuraObjCache::GetOrCreate(aura::Window* window) {
  33   return CreateInternal<AXWindowObjWrapper>(window, window_to_id_map_);
  34 }
  35
  36 int32 AXAuraObjCache::GetID(View* view) {
  37   return GetIDInternal(view, view_to_id_map_);
  38 }
  39
  40 int32 AXAuraObjCache::GetID(Widget* widget) {
  41   return GetIDInternal(widget, widget_to_id_map_);
  42 }
  43
  44 int32 AXAuraObjCache::GetID(aura::Window* window) {
  45   return GetIDInternal(window, window_to_id_map_);
  46 }
  47
  48 void AXAuraObjCache::Remove(View* view) {
  49   RemoveInternal(view, view_to_id_map_);
  50 }
  51
  52 void AXAuraObjCache::Remove(Widget* widget) {
  53   RemoveInternal(widget, widget_to_id_map_);
  54 }
  55
  56 void AXAuraObjCache::Remove(aura::Window* window) {
  57   RemoveInternal(window, window_to_id_map_);
  58 }
  59
  60 AXAuraObjWrapper* AXAuraObjCache::Get(int32 id) {
  61   std::map<int32, AXAuraObjWrapper*>::iterator it = cache_.find(id);
  62
  63   if (it == cache_.end())
  64     return NULL;
  65
  66   return it->second;
  67 }
  68
  69 void AXAuraObjCache::Remove(int32 id) {
  70   AXAuraObjWrapper* obj = Get(id);
  71
  72   if (id == -1 || !obj)
  73     return;
  74
  75   cache_.erase(id);
  76   delete obj;
  77 }
  78
  79 void AXAuraObjCache::GetTopLevelWindows(
  80     std::vector<AXAuraObjWrapper*>* children) {
  81   for (std::map<aura::Window*, int32>::iterator it = window_to_id_map_.begin();
  82        it != window_to_id_map_.end(); ++it) {
  83     if (!it->first->parent())
  84       children->push_back(GetOrCreate(it->first));
  85   }
  86 }
  87
  88 AXAuraObjCache::AXAuraObjCache() : current_id_(1), is_destroying_(false) {
  89 }
  90
  91 AXAuraObjCache::~AXAuraObjCache() {
  92   is_destroying_ = true;
  93   STLDeleteContainerPairSecondPointers(cache_.begin(), cache_.end());
  94   cache_.clear();
  95 }
  96
  97 template <typename AuraViewWrapper, typename AuraView>
  98 AXAuraObjWrapper* AXAuraObjCache::CreateInternal(
  99     AuraView* aura_view, std::map<AuraView*, int32>& aura_view_to_id_map) {
 100   if (!aura_view)
 101     return NULL;
 102
 103   typename std::map<AuraView*, int32>::iterator it =
 104       aura_view_to_id_map.find(aura_view);
 105
 106   if (it != aura_view_to_id_map.end())
 107     return Get(it->second);
 108
 109   AXAuraObjWrapper* wrapper = new AuraViewWrapper(aura_view);
 110   aura_view_to_id_map[aura_view] = current_id_;
 111   cache_[current_id_] = wrapper;
 112   current_id_++;
 113   return wrapper;
 114 }
 115
 116 template<typename AuraView> int32 AXAuraObjCache::GetIDInternal(
 117     AuraView* aura_view, std::map<AuraView*, int32>& aura_view_to_id_map) {
 118   if (!aura_view)
 119     return -1;
 120
 121   typename std::map<AuraView*, int32>::iterator it =
 122       aura_view_to_id_map.find(aura_view);
 123
 124   if (it != aura_view_to_id_map.end())
 125     return it->second;
 126
 127   return -1;
 128 }
 129
 130 template<typename AuraView>
 131 void AXAuraObjCache::RemoveInternal(
 132     AuraView* aura_view, std::map<AuraView*, int32>& aura_view_to_id_map) {
 133   int32 id = GetID(aura_view);
 134   if (id == -1)
 135     return;
 136   aura_view_to_id_map.erase(aura_view);
 137   Remove(id);
 138 }
 139
 140 }  // namespace views