ref_counted_managed.h revision 68043e1e95eeb07d5cae7aca370b26518b0867d6 
1// Copyright 2013 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#ifndef CC_BASE_REF_COUNTED_MANAGED_H_
6#define CC_BASE_REF_COUNTED_MANAGED_H_
7
8#include "base/logging.h"
9#include "base/memory/ref_counted.h"
10#include "cc/base/cc_export.h"
11
12namespace cc {
13
14template <typename T> class RefCountedManaged;
15
16template <typename T>
17class CC_EXPORT RefCountedManager {
18 protected:
19  RefCountedManager() : live_object_count_(0) {}
20  ~RefCountedManager() {
21    CHECK_EQ(0, live_object_count_);
22  }
23
24  virtual void Release(T* object) = 0;
25
26 private:
27  friend class RefCountedManaged<T>;
28  int live_object_count_;
29};
30
31template <typename T>
32class CC_EXPORT RefCountedManaged : public base::subtle::RefCountedBase {
33 public:
34  explicit RefCountedManaged(RefCountedManager<T>* manager)
35      : manager_(manager) {
36    manager_->live_object_count_++;
37  }
38
39  void AddRef() const {
40    base::subtle::RefCountedBase::AddRef();
41  }
42
43  void Release() {
44    if (base::subtle::RefCountedBase::Release()) {
45      manager_->Release(static_cast<T*>(this));
46
47      DCHECK_GT(manager_->live_object_count_, 0);
48      manager_->live_object_count_--;
49    }
50  }
51
52 protected:
53  ~RefCountedManaged() {}
54
55 private:
56  RefCountedManager<T>* manager_;
57
58  DISALLOW_COPY_AND_ASSIGN(RefCountedManaged<T>);
59};
60
61}  // namespace cc
62
63#endif  // CC_BASE_REF_COUNTED_MANAGED_H_
64