mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
75 lines
1.8 KiB
C++
75 lines
1.8 KiB
C++
// Copyright 2019 Microsoft Corporation. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
#ifndef WEAK_BUFFER_H
|
|
#define WEAK_BUFFER_H
|
|
|
|
#include <wrl.h>
|
|
#include <wrl/client.h>
|
|
|
|
#include <windows.storage.streams.h>
|
|
#include <robuffer.h>
|
|
|
|
namespace WinMLTest {
|
|
|
|
template <typename T>
|
|
struct WeakBuffer
|
|
: public Microsoft::WRL::RuntimeClass<
|
|
Microsoft::WRL::RuntimeClassFlags<Microsoft::WRL::WinRtClassicComMix | Microsoft::WRL::InhibitRoOriginateError>,
|
|
ABI::Windows::Storage::Streams::IBuffer,
|
|
Windows::Storage::Streams::IBufferByteAccess> {
|
|
InspectableClass(L"WinMLTest.WeakBuffer", BaseTrust)
|
|
|
|
private:
|
|
const T* m_p_begin;
|
|
const T* m_p_end;
|
|
|
|
public:
|
|
HRESULT RuntimeClassInitialize(_In_ const T* p_begin, _In_ const T* p_end) {
|
|
m_p_begin = p_begin;
|
|
m_p_end = p_end;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE get_Capacity(
|
|
UINT32 * value)
|
|
{
|
|
if (value == nullptr) {
|
|
return E_POINTER;
|
|
}
|
|
|
|
*value = static_cast<uint32_t>(m_p_end - m_p_begin) * sizeof(T);
|
|
return S_OK;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE get_Length(
|
|
UINT32 * value)
|
|
{
|
|
if (value == nullptr) {
|
|
return E_POINTER;
|
|
}
|
|
*value = static_cast<uint32_t>(m_p_end - m_p_begin) * sizeof(T);
|
|
return S_OK;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE put_Length(
|
|
UINT32 /*value*/)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHOD(Buffer)(uint8_t** value)
|
|
{
|
|
if (value == nullptr) {
|
|
return E_POINTER;
|
|
}
|
|
|
|
*value = reinterpret_cast<uint8_t*>(const_cast<T*>(m_p_begin));
|
|
return S_OK;
|
|
}
|
|
};
|
|
|
|
} // namespace WinMLTest
|
|
|
|
#endif // WEAK_BUFFER_H
|