Question

我正在尝试使用标准C ++库智能指针和一个使用MS COM来实现其大部分功能的库（我必须说我不熟悉COM）。所以，我的unique_ptr

有以下自定义删除功能

struct COMDeleter {
    template<typename T> void operator()(T* ptr) {
        if (ptr) ptr->Release();
    }
};

在示例代码中，我们有类似的内容：

class MyClass
{
public:
    MyClass(IDeckLink * device) 
      : m_deckLink(device)
    {            
    }  

    MyClass::~MyClass()
    {
        if (m_deckLink != NULL)
        {
            m_deckLink->Release();
            m_deckLink = NULL;
        }
    } 

    IDeckLink * m_deckLink;
};

这可以替换为：

class MyClass
{
public:
    MyClass(IDeckLink * device) 
    {  
        m_deckLink.reset(device);
    }   
    std::unique_ptr<IDeckLink, COMDeleter>  m_deckLink;
};

现在，我有另一个名为IDeckLinkInput的接口，我希望以类似的方式进行换行，但初始化的方式不同如下：

IDeckLinkInput* m_deckLinkInput = NULL;
if (m_deckLink->QueryInterface(IID_IDeckLinkInput, (void**) &m_deckLinkInput) != S_OK)
        return false;

所以，如果我有一个智能指针，如：

std::unique_ptr<IDeckLinkInput, COMDeleter> m_deckLinkInput(nullptr);

我不知道如何将其与初始化功能一起使用，如上所述？它甚至可以完成，还是应该坚持使用旧式C ++？

Answer 1

这样的事情：

template<class U, class T>
std::unique_ptr<U, COMDeleter>
upComInterface( GUID guid, T const& src ) {
  if (!src) return {};
  T* r = nullptr;
  if (src->QueryInterface( guid, (void**)&r) != S_OK)
    return {};
  return {r, {}};
}

然后我们：

auto deckLink = upComInterface<IDeckLinkInput>( IID_IDeckLinkInput, deckLink );

此处存在轻微的DRY违规行为 - 每次执行此操作时都必须重复IDeckLinkInput和IID_IDeckLinkInput之间的链接，并且错误导致未定义的行为。

我们可以通过多种机制解决这个问题。就个人而言，我会使用标签发送类型：

namespace MyComHelpers {
  template<class T> struct com_tag_t {using type=T; constexpr com_tag_t(){};};
  template<class T> constexpr com_tag_t<T> com_tag{};

  template<class T>
  constexpr void get_com_guid( com_tag_t<T> ) = delete; // overload this for your particular types

  template<class T>
  constexpr GUID interface_guid = get_com_guid( com_tag<T> );
}

现在我们可以将类型与guid相关联。在IDeckLinkInput的命名空间中执行此操作：

constexpr GUID get_com_guid( MyComHelpers::com_tag_t<IDeckLinkInput> ) {
  // constexpr code that returns the GUID
}

然后我们重写get接口函数：

std::unique_ptr<U, COMDeleter>
com_cast( T const& src ) {
  if (!src) return {};
  T* r = nullptr;
  if (src->QueryInterface( MyComHelpers::interface_guid<T>, (void**)&r) != S_OK)
    return {};
  return {r, {}};
}

并使用成为：

auto declLink = com_cast<IDeckLinkInput>(m_deckLinkInput);

有很多方法可以将类型与guid相关联，包括traits类。 constexpr基于ADL的查找函数和变量模板只是一种方式。

代码未经测试。

使用带有COM接口的STL智能指针

1 个答案: