components/mus/public/cpp/view_tracker.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 "components/mus/public/cpp/view_tracker.h"
   6
   7 namespace mus {
   8
   9 ViewTracker::ViewTracker() {}
  10
  11 ViewTracker::~ViewTracker() {
  12   for (Views::iterator i = views_.begin(); i != views_.end(); ++i)
  13     (*i)->RemoveObserver(this);
  14 }
  15
  16 void ViewTracker::Add(View* view) {
  17   if (views_.count(view))
  18     return;
  19
  20   view->AddObserver(this);
  21   views_.insert(view);
  22 }
  23
  24 void ViewTracker::Remove(View* view) {
  25   if (views_.count(view)) {
  26     views_.erase(view);
  27     view->RemoveObserver(this);
  28   }
  29 }
  30
  31 bool ViewTracker::Contains(View* view) {
  32   return views_.count(view) > 0;
  33 }
  34
  35 void ViewTracker::OnViewDestroying(View* view) {
  36   DCHECK_GT(views_.count(view), 0u);
  37   Remove(view);
  38 }
  39
  40 }  // namespace mus