// Copyright 2014 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.
|
|
#ifndef MINI_CHROMIUM_BASE_MAC_SCOPED_TYPEREF_H_
|
#define MINI_CHROMIUM_BASE_MAC_SCOPED_TYPEREF_H_
|
|
#include "base/compiler_specific.h"
|
#include "base/logging.h"
|
#include "base/memory/scoped_policy.h"
|
|
namespace base {
|
|
template <typename T>
|
struct ScopedTypeRefTraits;
|
|
template <typename T, typename Traits = ScopedTypeRefTraits<T>>
|
class ScopedTypeRef {
|
public:
|
typedef T element_type;
|
|
ScopedTypeRef(
|
T object = Traits::InvalidValue(),
|
base::scoped_policy::OwnershipPolicy policy = base::scoped_policy::ASSUME)
|
: object_(object) {
|
if (object_ && policy == base::scoped_policy::RETAIN)
|
object_ = Traits::Retain(object_);
|
}
|
|
ScopedTypeRef(const ScopedTypeRef<T, Traits>& that) : object_(that.object_) {
|
if (object_)
|
object_ = Traits::Retain(object_);
|
}
|
|
~ScopedTypeRef() {
|
if (object_)
|
Traits::Release(object_);
|
}
|
|
ScopedTypeRef& operator=(const ScopedTypeRef<T, Traits>& that) {
|
reset(that.get(), base::scoped_policy::RETAIN);
|
return *this;
|
}
|
|
T* InitializeInto() WARN_UNUSED_RESULT {
|
DCHECK(!object_);
|
return &object_;
|
}
|
|
void reset(T object = Traits::InvalidValue(),
|
base::scoped_policy::OwnershipPolicy policy =
|
base::scoped_policy::ASSUME) {
|
if (object && policy == base::scoped_policy::RETAIN)
|
object = Traits::Retain(object);
|
if (object_)
|
Traits::Release(object_);
|
object_ = object;
|
}
|
|
bool operator==(T that) const { return object_ == that; }
|
|
bool operator!=(T that) const { return object_ != that; }
|
|
operator T() const { return object_; }
|
|
T get() const { return object_; }
|
|
void swap(ScopedTypeRef& that) {
|
T temp = that.object_;
|
that.object_ = object_;
|
object_ = temp;
|
}
|
|
T release() WARN_UNUSED_RESULT {
|
T temp = object_;
|
object_ = Traits::InvalidValue();
|
return temp;
|
}
|
|
private:
|
T object_;
|
};
|
|
} // namespace base
|
|
#endif // MINI_CHROMIUM_BASE_MAC_SCOPED_TYPEREF_H_
|