obs-StreamFX/source/ffmpeg/hwapi/d3d11.hpp

84 lines
3.2 KiB
C++
Raw Normal View History

2020-01-13 00:52:30 +00:00
// FFMPEG Video Encoder Integration for OBS Studio
// Copyright (c) 2019 Michael Fabian Dirks <info@xaymar.com>
//
// Permission is hereby granted, free of charge, to any person obtaining a 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
2020-01-13 21:40:15 +00:00
#include "base.hpp"
#ifdef _MSC_VER
2020-01-13 21:40:15 +00:00
#pragma warning(push)
#pragma warning(disable : 4191 4242 4244 4365 4777 4986 5039 5204)
#endif
2020-01-13 00:52:30 +00:00
#include <atlutil.h>
#include <d3d11.h>
#include <d3d11_1.h>
#include <dxgi.h>
#ifdef _MSC_VER
2020-01-13 21:40:15 +00:00
#pragma warning(pop)
#endif
2020-01-13 00:52:30 +00:00
2021-06-08 02:12:55 +00:00
namespace streamfx::ffmpeg::hwapi {
class d3d11 : public streamfx::ffmpeg::hwapi::base {
2020-01-13 21:40:15 +00:00
typedef HRESULT(__stdcall* CreateDXGIFactory_t)(REFIID, void**);
typedef HRESULT(__stdcall* CreateDXGIFactory1_t)(REFIID, void**);
typedef HRESULT(__stdcall* D3D11CreateDevice_t)(IDXGIAdapter*, D3D_DRIVER_TYPE, HMODULE, UINT,
CONST D3D_FEATURE_LEVEL*, UINT, UINT, ID3D11Device**,
D3D_FEATURE_LEVEL*, ID3D11DeviceContext**);
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
HMODULE _dxgi_module;
CreateDXGIFactory_t _CreateDXGIFactory;
CreateDXGIFactory1_t _CreateDXGIFactory1;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
HMODULE _d3d11_module;
D3D11CreateDevice_t _D3D11CreateDevice;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
ATL::CComPtr<IDXGIFactory1> _dxgifactory;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
public:
d3d11();
virtual ~d3d11();
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
virtual std::list<hwapi::device> enumerate_adapters() override;
2020-01-13 00:52:30 +00:00
virtual std::shared_ptr<hwapi::instance> create(const hwapi::device& target) override;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
virtual std::shared_ptr<hwapi::instance> create_from_obs() override;
};
2020-01-13 00:52:30 +00:00
2021-06-08 02:12:55 +00:00
class d3d11_instance : public streamfx::ffmpeg::hwapi::instance {
2020-01-13 21:40:15 +00:00
ATL::CComPtr<ID3D11Device> _device;
ATL::CComPtr<ID3D11DeviceContext> _context;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
public:
d3d11_instance(ATL::CComPtr<ID3D11Device> device, ATL::CComPtr<ID3D11DeviceContext> context);
virtual ~d3d11_instance();
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
virtual AVBufferRef* create_device_context() override;
2020-01-13 00:52:30 +00:00
2020-01-13 21:40:15 +00:00
virtual std::shared_ptr<AVFrame> allocate_frame(AVBufferRef* frames) override;
2020-01-13 00:52:30 +00:00
virtual void copy_from_obs(AVBufferRef* frames, uint32_t handle, uint64_t lock_key, uint64_t* next_lock_key,
std::shared_ptr<AVFrame> frame) override;
2020-01-13 00:52:30 +00:00
virtual std::shared_ptr<AVFrame> avframe_from_obs(AVBufferRef* frames, uint32_t handle, uint64_t lock_key,
2020-01-13 21:40:15 +00:00
uint64_t* next_lock_key) override;
};
2021-06-08 02:12:55 +00:00
} // namespace streamfx::ffmpeg::hwapi