升级XR插件版本

Packages/com.unity.ide.trae@0231d4b833/Editor/COMIntegration/COMIntegration~/ComPtr.h

@@ -0,0 +1,186 @@
+#pragma once
+
+namespace win
+{
+	template<typename T>
+	class ComPtr;
+
+	template<typename T>
+	class ComPtrRef
+	{
+	private:
+		ComPtr<T>& m_ComPtr;
+
+		ComPtrRef(ComPtr<T>& comPtr) :
+			m_ComPtr(comPtr)
+		{
+		}
+
+		friend class ComPtr<T>;
+
+	public:
+		inline operator T**()
+		{
+			return m_ComPtr.ReleaseAndGetAddressOf();
+		}
+
+		inline operator void**()
+		{
+			return reinterpret_cast<void**>(m_ComPtr.ReleaseAndGetAddressOf());
+		}
+
+		inline T* operator*() throw ()
+		{
+			return m_ComPtr;
+		}
+
+	};
+
+	template<typename T>
+	class ComPtr
+	{
+	private:
+		T *ptr;
+
+	public:
+		inline ComPtr(void) : ptr(NULL) {}
+		inline ~ComPtr(void) { this->Free(); }
+
+		ComPtr(T *ptr)
+		{
+			if (NULL != (this->ptr = ptr))
+			{
+				this->ptr->AddRef();
+			}
+		}
+
+		ComPtr(const ComPtr &ptr)
+		{
+			if (NULL != (this->ptr = ptr.ptr))
+			{
+				this->ptr->AddRef();
+			}
+		}
+
+		inline bool operator!() const
+		{
+			return (NULL == this->ptr);
+		}
+
+		inline operator T*() const { return this->ptr; }
+
+		inline T *operator->() const
+		{
+			//_assert(NULL != this->ptr);
+			return this->ptr;
+		}
+
+		inline T &operator*()
+		{
+			//_assert(NULL != this->ptr);
+			return *this->ptr;
+		}
+
+		inline ComPtrRef<T> operator&()
+		{
+			return ComPtrRef<T>(*this);
+		}
+
+		const ComPtr &operator=(T *ptr)
+		{
+			if (this->ptr != ptr)
+			{
+				this->Free();
+
+				if (NULL != (this->ptr = ptr))
+				{
+					this->ptr->AddRef();
+				}
+			}
+
+			return *this;
+		}
+
+		const ComPtr &operator=(const ComPtr &ptr)
+		{
+			if (this->ptr != ptr.ptr)
+			{
+				this->Free();
+
+				if (NULL != (this->ptr = ptr.ptr))
+				{
+					this->ptr->AddRef();
+				}
+			}
+
+			return *this;
+		}
+
+		void Free(void)
+		{
+			if (NULL != this->ptr)
+			{
+				this->ptr->Release();
+				this->ptr = NULL;
+			}
+		}
+
+		inline T** ReleaseAndGetAddressOf()
+		{
+			Free();
+			return &ptr;
+		}
+
+		template<typename U>
+		inline HRESULT As(ComPtrRef<U> p) const throw ()
+		{
+			return ptr->QueryInterface(__uuidof(U), p);
+		}
+
+		inline bool operator==(std::nullptr_t) const
+		{
+			return this->ptr == nullptr;
+		}
+
+		template<typename U>
+		inline bool operator==(U* other)
+		{
+			if (ptr == nullptr || other == nullptr)
+				return ptr == other;
+
+			ComPtr<IUnknown> meUnknown;
+			ComPtr<IUnknown> otherUnknown;
+
+			if (FAILED(this->ptr->QueryInterface(__uuidof(IUnknown), &meUnknown)))
+				return false;
+
+			if (FAILED(other->QueryInterface(__uuidof(IUnknown), &otherUnknown)))
+				return false;
+
+			return static_cast<IUnknown*>(meUnknown) == static_cast<IUnknown*>(otherUnknown);
+		}
+
+		template<typename U>
+		inline bool operator==(ComPtr<U>& other)
+		{
+			return *this == static_cast<U*>(other);
+		}
+
+		inline bool operator!=(std::nullptr_t) const
+		{
+			return this->ptr != nullptr;
+		}
+
+		template<typename U>
+		inline bool operator!=(U* other)
+		{
+			return !(*this == other);
+		}
+
+		template<typename U>
+		inline bool operator!=(ComPtr<U>& other)
+		{
+			return *this != static_cast<U*>(other);
+		}
+	};
+}