mirror of https://github.com/Xaymar/obs-StreamFX
81 lines
3.4 KiB
C++
81 lines
3.4 KiB
C++
|
// 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.
|
||
|
|
||
|
#include <atlutil.h>
|
||
|
#include <d3d11.h>
|
||
|
#include <d3d11_1.h>
|
||
|
#include <dxgi.h>
|
||
|
#include "base.hpp"
|
||
|
|
||
|
namespace obsffmpeg {
|
||
|
namespace hwapi {
|
||
|
class d3d11 : public ::obsffmpeg::hwapi::base {
|
||
|
typedef HRESULT(__stdcall* CreateDXGIFactory_t)(REFIID, void**);
|
||
|
typedef HRESULT(__stdcall* CreateDXGIFactory1_t)(REFIID, void**);
|
||
|
typedef HRESULT(__stdcall* D3D11CreateDevice_t)(_In_opt_ IDXGIAdapter*, D3D_DRIVER_TYPE,
|
||
|
HMODULE, UINT, CONST D3D_FEATURE_LEVEL*, UINT,
|
||
|
UINT, _Out_opt_ ID3D11Device**,
|
||
|
_Out_opt_ D3D_FEATURE_LEVEL*,
|
||
|
_Out_opt_ ID3D11DeviceContext**);
|
||
|
|
||
|
HMODULE _dxgi_module;
|
||
|
CreateDXGIFactory_t _CreateDXGIFactory;
|
||
|
CreateDXGIFactory1_t _CreateDXGIFactory1;
|
||
|
|
||
|
HMODULE _d3d11_module;
|
||
|
D3D11CreateDevice_t _D3D11CreateDevice;
|
||
|
|
||
|
ATL::CComPtr<IDXGIFactory1> _dxgifactory;
|
||
|
|
||
|
public:
|
||
|
d3d11();
|
||
|
virtual ~d3d11();
|
||
|
|
||
|
virtual std::list<obsffmpeg::hwapi::device> enumerate_adapters() override;
|
||
|
|
||
|
virtual std::shared_ptr<obsffmpeg::hwapi::instance>
|
||
|
create(obsffmpeg::hwapi::device target) override;
|
||
|
|
||
|
virtual std::shared_ptr<obsffmpeg::hwapi::instance> create_from_obs() override;
|
||
|
};
|
||
|
|
||
|
class d3d11_instance : public ::obsffmpeg::hwapi::instance {
|
||
|
ATL::CComPtr<ID3D11Device> _device;
|
||
|
ATL::CComPtr<ID3D11DeviceContext> _context;
|
||
|
|
||
|
public:
|
||
|
d3d11_instance(ATL::CComPtr<ID3D11Device> device, ATL::CComPtr<ID3D11DeviceContext> context);
|
||
|
virtual ~d3d11_instance();
|
||
|
|
||
|
virtual AVBufferRef* create_device_context() override;
|
||
|
|
||
|
virtual std::shared_ptr<AVFrame> allocate_frame(AVBufferRef* frames) override;
|
||
|
|
||
|
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;
|
||
|
|
||
|
virtual std::shared_ptr<AVFrame> avframe_from_obs(AVBufferRef* frames, uint32_t handle,
|
||
|
uint64_t lock_key,
|
||
|
uint64_t* next_lock_key) override;
|
||
|
};
|
||
|
} // namespace hwapi
|
||
|
} // namespace obsffmpeg
|