COMUtil.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
 
#if defined(_WIN32) && !defined(SWIG)
#include <Windows.h>
#endif
 
AST_NAMESPACE_BEGIN
 
#if defined(_WIN32) && !defined(SWIG)
 
class CoInitializeGuard
{
public:
    explicit CoInitializeGuard(COINIT initFlags) : hr_(CoInitializeEx(nullptr, initFlags)) {}
    ~CoInitializeGuard() { if (SUCCEEDED(hr_)) CoUninitialize(); }
    CoInitializeGuard(const CoInitializeGuard&) = delete;
    CoInitializeGuard& operator=(const CoInitializeGuard&) = delete;
    HRESULT hr_{};
};
 
AST_UTIL_API HRESULT aEnsureCoInitialized();
 
 
 
template <typename T>
class ComScopedPtr
{
public:
    ComScopedPtr() = default;
    ComScopedPtr(const ComScopedPtr&) = delete;
    ComScopedPtr& operator=(const ComScopedPtr&) = delete;
    ComScopedPtr(ComScopedPtr&& other) noexcept : p_(other.p_) { other.p_ = nullptr; }
    ComScopedPtr& operator=(ComScopedPtr&& other) noexcept{std::swap(p_, other.p_); return *this;}
    ~ComScopedPtr() { reset(); }
 
    T* get() const { return p_; }
    T* operator->() const { return p_; }
    T& operator*() const { return *p_; }
 
    T** operator&() { return &p_; }
 
    void reset() { if (p_) p_->Release(); p_ = nullptr; }
 
    operator bool() const { return p_ != nullptr; }
private:
    T* p_{nullptr};
};
 
#else
 
// 非 Windows 平台：空实现
class CoInitializeGuard
{
public:
    explicit CoInitializeGuard(int) {}
    int hr_ = 0;
};
 
A_ALWAYS_INLINE int aEnsureCoInitialized() { return 0; }
 
#endif
 
AST_NAMESPACE_END