// Copyright (c) 2016-2018 Easy2D - Nomango
// 
// Permission is hereby granted, free of charge, to any person obtaining lhs copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#pragma once
#include "../macros.h"
#include <utility>

namespace easy2d
{
	namespace intrusive
	{
		template <typename T>
		class SmartPointer
		{
			T* ptr_{ nullptr };

		public:
			using Type = T;

			SmartPointer() E2D_NOEXCEPT {}

			SmartPointer(nullptr_t) E2D_NOEXCEPT {}

			SmartPointer(Type* p) E2D_NOEXCEPT : ptr_(p)
			{
				IntrusivePtrAddRef(ptr_);
			}

			SmartPointer(const SmartPointer& other) E2D_NOEXCEPT : ptr_(other.ptr_)
			{
				IntrusivePtrAddRef(ptr_);
			}

			template <typename U>
			SmartPointer(const SmartPointer<U>& other) E2D_NOEXCEPT : ptr_(other.Get())
			{
				IntrusivePtrAddRef(ptr_);
			}

			SmartPointer(SmartPointer&& other) E2D_NOEXCEPT
			{
				ptr_ = other.ptr_;
				other.ptr_ = nullptr;
			}

			~SmartPointer() E2D_NOEXCEPT
			{
				IntrusivePtrRelease(ptr_);
			}

			inline Type* Get() const E2D_NOEXCEPT { return ptr_; }

			inline void Swap(SmartPointer& other) E2D_NOEXCEPT
			{
				std::swap(ptr_, other.ptr_);
			}

			inline Type* operator ->() const
			{
				E2D_ASSERT(ptr_ != nullptr && "Invalid pointer");
				return ptr_;
			}

			inline Type& operator *() const
			{
				E2D_ASSERT(ptr_ != nullptr && "Invalid pointer");
				return *ptr_;
			}

			inline Type** operator &()
			{
				E2D_ASSERT(ptr_ == nullptr && "Memory leak");
				return &ptr_;
			}

			inline operator bool() const E2D_NOEXCEPT { return ptr_ != nullptr; }

			inline bool operator !() const E2D_NOEXCEPT { return ptr_ == 0; }

			inline SmartPointer& operator =(const SmartPointer& other) E2D_NOEXCEPT
			{
				if (other.ptr_ != ptr_)
					SmartPointer(other).Swap(*this);
				return *this;
			}

			inline SmartPointer& operator =(SmartPointer&& other) E2D_NOEXCEPT
			{
				IntrusivePtrRelease(ptr_);
				ptr_ = other.ptr_;
				other.ptr_ = nullptr;
				return *this;
			}

			inline SmartPointer& operator =(Type* p) E2D_NOEXCEPT
			{
				if (p != ptr_)
					SmartPointer(p).Swap(*this);
				return *this;
			}

			inline SmartPointer& operator =(nullptr_t) E2D_NOEXCEPT
			{
				if (nullptr != ptr_)
					SmartPointer{}.Swap(*this);
				return *this;
			}
		};

		template<class T, class U>
		inline bool operator==(SmartPointer<T> const& lhs, SmartPointer<U> const& rhs) E2D_NOEXCEPT
		{
			return lhs.Get() == rhs.Get();
		}

		template<class T, class U>
		inline bool operator!=(SmartPointer<T> const& lhs, SmartPointer<U> const& rhs) E2D_NOEXCEPT
		{
			return lhs.Get() != rhs.Get();
		}

		template<class T, class U>
		inline bool operator<(SmartPointer<T> const& lhs, SmartPointer<U> const& rhs) E2D_NOEXCEPT
		{
			return lhs.Get() < rhs.Get();
		}

		template<class T>
		inline bool operator==(SmartPointer<T> const& lhs, T* rhs) E2D_NOEXCEPT
		{
			return lhs.Get() == rhs;
		}

		template<class T>
		inline bool operator!=(SmartPointer<T> const& lhs, T* rhs) E2D_NOEXCEPT
		{
			return lhs.Get() != rhs;
		}

		template<class T>
		inline bool operator==(T* lhs, SmartPointer<T> const& rhs) E2D_NOEXCEPT
		{
			return lhs == rhs.Get();
		}

		template<class T>
		inline bool operator!=(T* lhs, SmartPointer<T> const& rhs) E2D_NOEXCEPT
		{
			return lhs != rhs.Get();
		}

		template<class T>
		inline bool operator==(SmartPointer<T> const& lhs, nullptr_t) E2D_NOEXCEPT
		{
			return !static_cast<bool>(lhs);
		}

		template<class T>
		inline bool operator!=(SmartPointer<T> const& lhs, nullptr_t) E2D_NOEXCEPT
		{
			return static_cast<bool>(lhs);
		}

		template<class T>
		inline bool operator==(nullptr_t, SmartPointer<T> const& rhs) E2D_NOEXCEPT
		{
			return !static_cast<bool>(rhs);
		}

		template<class T>
		inline bool operator!=(nullptr_t, SmartPointer<T> const& rhs) E2D_NOEXCEPT
		{
			return static_cast<bool>(rhs);
		}
	}

	// template class cannot specialize std::swap,
	// so implement a swap function in easy2d namespace
	template<class T>
	inline void swap(intrusive::SmartPointer<T>& lhs, intrusive::SmartPointer<T>& rhs) E2D_NOEXCEPT
	{
		lhs.Swap(rhs);
	}

}