summaryrefslogtreecommitdiffstats
path: root/media/base/callback.h
blob: c21312e738539200a43ebd478e33ed6826b9dd8d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Some basic utilities for aiding in the management of Tasks and Callbacks.
//
// AutoTaskRunner, and its brother AutoCallbackRunner are the scoped_ptr
// equivalents for callbacks.  They are useful for ensuring a callback is
// executed and delete in the face of multiple return points in a function.
//
// TaskToCallbackAdapter converts a Task to a Callback0::Type since the two type
// hierarchies are strangely separate.
//
// CleanupCallback wraps another Callback and provides the ability to register
// objects for deletion as well as cleanup tasks that will be run on the
// callback's destruction.  The deletion and cleanup tasks will be run on
// whatever thread the CleanupCallback is destroyed in.

#ifndef MEDIA_BASE_CALLBACK_
#define MEDIA_BASE_CALLBACK_

#include <vector>

#include "base/callback.h"
#include "base/scoped_ptr.h"
#include "base/task.h"

namespace media {

class AutoTaskRunner {
 public:
  // Takes ownership of the task.
  explicit AutoTaskRunner(Task* task)
      : task_(task) {
  }

  ~AutoTaskRunner();

  Task* release() { return task_.release(); }

 private:
  scoped_ptr<Task> task_;

  DISALLOW_COPY_AND_ASSIGN(AutoTaskRunner);
};

class AutoCallbackRunner {
 public:
  // Takes ownership of the callback.
  explicit AutoCallbackRunner(Callback0::Type* callback)
      : callback_(callback) {
  }

  ~AutoCallbackRunner();

  Callback0::Type* release() { return callback_.release(); }

 private:
  scoped_ptr<Callback0::Type> callback_;

  DISALLOW_COPY_AND_ASSIGN(AutoCallbackRunner);
};

class TaskToCallbackAdapter : public Callback0::Type {
 public:
  static Callback0::Type* NewCallback(Task* task);

  virtual ~TaskToCallbackAdapter();

  virtual void RunWithParams(const Tuple0& params);

 private:
  TaskToCallbackAdapter(Task* task);

  scoped_ptr<Task> task_;

  DISALLOW_COPY_AND_ASSIGN(TaskToCallbackAdapter);
};

template <typename CallbackType>
class CleanupCallback : public CallbackType {
 public:
  explicit CleanupCallback(CallbackType* callback) : callback_(callback) {}

  virtual ~CleanupCallback() {
    for (size_t i = 0; i < run_when_done_.size(); i++) {
      run_when_done_[i]->Run();
      delete run_when_done_[i];
    }
  }

  virtual void RunWithParams(const typename CallbackType::TupleType& params) {
    callback_->RunWithParams(params);
  }

  template <typename T>
  void DeleteWhenDone(T* ptr) {
    RunWhenDone(new DeleteTask<T>(ptr));
  }

  void RunWhenDone(Task* ptr) {
    run_when_done_.push_back(ptr);
  }

 private:
  scoped_ptr<CallbackType> callback_;
  std::vector<Task*> run_when_done_;

  DISALLOW_COPY_AND_ASSIGN(CleanupCallback);
};

}  // namespace media

#endif  // MEDIA_BASE_CALLBACK_