xref: /foundation/ability/idl_tool/util/autoptr.h

/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef OHOS_IDL_AUTOPTR_H
#define OHOS_IDL_AUTOPTR_H

namespace OHOS {
namespace Idl {
template<class T>
class AutoPtr {
public:
    AutoPtr()
        : mPtr(nullptr)
    {}

    AutoPtr(T* other);

    AutoPtr(const AutoPtr<T>& other);

    AutoPtr(AutoPtr<T>&& other);

    ~AutoPtr();

    AutoPtr& operator=(T* other);

    AutoPtr& operator=(const AutoPtr<T>& other);

    AutoPtr& operator=(AutoPtr<T>&& other);

    void MoveTo(T** other);

    inline operator T*() const;

    inline T** operator&();

    inline T* operator->() const;

    inline T& operator*() const;

    inline T* Get() const;

    inline bool operator==(T* other) const;

    inline bool operator==(const AutoPtr<T>& other) const;

    inline bool operator!=(T* other) const;

    inline bool operator!=(const AutoPtr<T>& other) const;

    inline bool operator>(T* other) const;

    inline bool operator>(const AutoPtr<T>& other) const;

    inline bool operator<(T* other) const;

    inline bool operator<(const AutoPtr<T>& other) const;

    inline bool operator<=(T* other) const;

    inline bool operator<=(const AutoPtr<T>& other) const;

    inline bool operator>=(T* other) const;

    inline bool operator>=(const AutoPtr<T>& other) const;

private:
    T* mPtr;
};

template<class T>
AutoPtr<T>::AutoPtr(T* other)
    : mPtr(other)
{
    if (mPtr != nullptr) {
        mPtr->AddRef();
    }
}

template<class T>
AutoPtr<T>::AutoPtr(const AutoPtr<T>& other)
    : mPtr(other.mPtr)
{
    if (mPtr != nullptr) {
        mPtr->AddRef();
    }
}

template<class T>
AutoPtr<T>::AutoPtr(AutoPtr<T>&& other)
    : mPtr(other.mPtr)
{
    other.mPtr = nullptr;
}

template<class T>
AutoPtr<T>::~AutoPtr()
{
    if (mPtr != nullptr) {
        mPtr->Release();
    }
}

template<class T>
AutoPtr<T>& AutoPtr<T>::operator=(T* other)
{
    if (mPtr == other) return *this;

    if (other != nullptr) {
        other->AddRef();
    }
    if (mPtr != nullptr) {
        mPtr->Release();
    }
    mPtr = other;
    return *this;
}

template<class T>
AutoPtr<T>& AutoPtr<T>::operator=(const AutoPtr<T>& other)
{
    if (mPtr == other.mPtr) return *this;

    if (other.mPtr != nullptr) {
        other.mPtr->AddRef();
    }
    if (mPtr != nullptr) {
        mPtr->Release();
    }
    mPtr = other.mPtr;
    return *this;
}

template<class T>
AutoPtr<T>& AutoPtr<T>::operator=(AutoPtr<T>&& other)
{
    if (mPtr != nullptr) {
        mPtr->Release();
    }
    mPtr = other.mPtr;
    other.mPtr = nullptr;
    return *this;
}

template<class T>
void AutoPtr<T>::MoveTo(T** other)
{
    if (other != nullptr) {
        *other = mPtr;
        mPtr = nullptr;
    }
}

template<class T>
AutoPtr<T>::operator T*() const
{
    return mPtr;
}

template<class T>
T** AutoPtr<T>::operator&()
{
    return &mPtr;
}

template<class T>
T* AutoPtr<T>::operator->() const
{
    return mPtr;
}

template<class T>
T& AutoPtr<T>::operator*() const
{
    return *mPtr;
}

template<class T>
T* AutoPtr<T>::Get() const
{
    return mPtr;
}

template<class T>
bool AutoPtr<T>::operator==(T* other) const
{
    return mPtr == other;
}

template<class T>
bool AutoPtr<T>::operator==(const AutoPtr<T>& other) const
{
    return mPtr == other.mPtr;
}

template<class T>
bool AutoPtr<T>::operator!=(T* other) const
{
    return mPtr != other;
}

template<class T>
bool AutoPtr<T>::operator!=(const AutoPtr<T>& other) const
{
    return mPtr != other.mPtr;
}

template<class T>
bool AutoPtr<T>::operator>(T* other) const
{
    return mPtr > other;
}

template<class T>
bool AutoPtr<T>::operator>(const AutoPtr<T>& other) const
{
    return mPtr > other.mPtr;
}

template<class T>
bool AutoPtr<T>::operator<(T* other) const
{
    return mPtr < other;
}

template<class T>
bool AutoPtr<T>::operator<(const AutoPtr<T>& other) const
{
    return mPtr < other.mPtr;
}

template<class T>
bool AutoPtr<T>::operator<=(T* other) const
{
    return mPtr <= other;
}

template<class T>
bool AutoPtr<T>::operator<=(const AutoPtr<T>& other) const
{
    return mPtr <= other.mPtr;
}

template<class T>
bool AutoPtr<T>::operator>=(T* other) const
{
    return mPtr >= other;
}

template<class T>
bool AutoPtr<T>::operator>=(const AutoPtr<T>& other) const
{
    return mPtr >= other.mPtr;
}
}
}
#endif // OHOS_IDL_AUTOPTR_H