sleepwithoutbz/cbh
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: 格式化ElaWidgetTools代码

Browse Source
This commit is contained in:
sleepwithoutbz
2025-09-20 01:39:00 +08:00
parent d2fded145d
commit 4ab391f8a1
394 changed files with 10913 additions and 15860 deletions
Download Patch File Download Diff File Expand all files Collapse all files

256
ElaWidgetTools/DeveloperComponents/ElaDxgi.cpp
View File

@@ -1,130 +1,107 @@
#include "ElaDxgi.h"
#ifdef Q_OS_WIN
#include <QDateTime>
#include <QDebug>
#include <QDateTime>
#include <QDebug>
ElaDxgi::ElaDxgi(QObject* parent)
: QObject(parent)
{
_pIsInitSuccess = false;
_pIsGrabActive = false;
_pGrabFrameRate = 120;
ElaDxgi::ElaDxgi(QObject *parent) : QObject(parent) {
_pIsInitSuccess = false;
_pIsGrabActive = false;
_pGrabFrameRate = 120;
_pTimeoutMsValue = 50;
_pIsGrabStoped = true;
_pIsGrabCenter = false;
_pIsGrabStoped = true;
_pIsGrabCenter = false;
}
ElaDxgi::~ElaDxgi()
{
releaseInterface();
}
ElaDxgi::~ElaDxgi() { releaseInterface(); }
bool ElaDxgi::initialize(int dxID, int outputID)
{
_pIsInitSuccess = false;
_pDxDeviceID = dxID;
bool ElaDxgi::initialize(int dxID, int outputID) {
_pIsInitSuccess = false;
_pDxDeviceID = dxID;
_pOutputDeviceID = outputID;
releaseInterface();
ID3D11Device* d3dDevice = nullptr;
ID3D11DeviceContext* d3dContext = nullptr;
D3D_FEATURE_LEVEL feat = D3D_FEATURE_LEVEL_11_0;
HRESULT hr = D3D11CreateDevice(nullptr, D3D_DRIVER_TYPE_HARDWARE, nullptr, 0, nullptr, 0,
D3D11_SDK_VERSION, &d3dDevice, &feat, &d3dContext);
if (FAILED(hr))
{
ID3D11Device *d3dDevice = nullptr;
ID3D11DeviceContext *d3dContext = nullptr;
D3D_FEATURE_LEVEL feat = D3D_FEATURE_LEVEL_11_0;
HRESULT hr = D3D11CreateDevice(nullptr, D3D_DRIVER_TYPE_HARDWARE, nullptr, 0, nullptr, 0, D3D11_SDK_VERSION, &d3dDevice, &feat, &d3dContext);
if (FAILED(hr)) {
_pLastError = "Failed to D3D11CreateDevice ErrorCode = " + QString::number(uint(hr), 16);
qDebug() << _pLastError;
if (d3dDevice)
{
if (d3dDevice) {
d3dDevice->Release();
}
if (d3dContext)
{
if (d3dContext) {
d3dContext->Release();
}
return false;
}
IDXGIFactory* dxgiFactory = nullptr;
CreateDXGIFactory(__uuidof(dxgiFactory), reinterpret_cast<void**>(&dxgiFactory));
IDXGIAdapter* dxgiAdapter = nullptr;
QVector<IDXGIAdapter*> dxgiAdapterVector;
for (uint i = 0; dxgiFactory->EnumAdapters(i, &dxgiAdapter) != DXGI_ERROR_NOT_FOUND; ++i)
{
IDXGIFactory *dxgiFactory = nullptr;
CreateDXGIFactory(__uuidof(dxgiFactory), reinterpret_cast<void **>(&dxgiFactory));
IDXGIAdapter *dxgiAdapter = nullptr;
QVector<IDXGIAdapter *> dxgiAdapterVector;
for (uint i = 0; dxgiFactory->EnumAdapters(i, &dxgiAdapter) != DXGI_ERROR_NOT_FOUND; ++i) {
dxgiAdapterVector.append(dxgiAdapter);
}
dxgiFactory->Release();
dxgiAdapter = nullptr;
_pDxDeviceList.clear();
for (int i = 0; i < dxgiAdapterVector.count(); i++)
{
IDXGIAdapter* adapt = dxgiAdapterVector.at(i);
for (int i = 0; i < dxgiAdapterVector.count(); i++) {
IDXGIAdapter *adapt = dxgiAdapterVector.at(i);
DXGI_ADAPTER_DESC desc;
adapt->GetDesc(&desc);
_pDxDeviceList.append(QString::fromWCharArray(desc.Description));
}
if (dxID >= 0 && dxID < dxgiAdapterVector.count())
{
if (dxID >= 0 && dxID < dxgiAdapterVector.count()) {
dxgiAdapter = dxgiAdapterVector.at(dxID);
}
if (!dxgiAdapter)
{
if (!dxgiAdapter) {
_pLastError = "Failed to found gpu";
qDebug() << _pLastError;
d3dDevice->Release();
d3dContext->Release();
for (auto adapter: dxgiAdapterVector)
{
for (auto adapter : dxgiAdapterVector) {
adapter->Release();
}
return false;
}
IDXGIOutput* dxgiOutput = nullptr;
QVector<IDXGIOutput*> dxgiOutputVector;
for (uint i = 0; dxgiAdapter->EnumOutputs(i, &dxgiOutput) != DXGI_ERROR_NOT_FOUND; ++i)
{
IDXGIOutput *dxgiOutput = nullptr;
QVector<IDXGIOutput *> dxgiOutputVector;
for (uint i = 0; dxgiAdapter->EnumOutputs(i, &dxgiOutput) != DXGI_ERROR_NOT_FOUND; ++i) {
dxgiOutputVector.append(dxgiOutput);
}
for (auto adapter: dxgiAdapterVector)
{
for (auto adapter : dxgiAdapterVector) {
adapter->Release();
}
dxgiOutput = nullptr;
_pOutputDeviceList.clear();
for (int i = 0; i < dxgiOutputVector.count(); i++)
{
const auto& output = dxgiOutputVector.at(i);
for (int i = 0; i < dxgiOutputVector.count(); i++) {
const auto &output = dxgiOutputVector.at(i);
DXGI_OUTPUT_DESC desc;
output->GetDesc(&desc);
_pOutputDeviceList.append(QString::fromWCharArray(desc.DeviceName));
}
if (outputID >= 0 && outputID < dxgiOutputVector.count())
{
if (outputID >= 0 && outputID < dxgiOutputVector.count()) {
dxgiOutput = dxgiOutputVector.at(outputID);
}
if (!dxgiOutput)
{
if (!dxgiOutput) {
_pLastError = "Failed to found screen!";
qDebug() << _pLastError;
d3dDevice->Release();
d3dContext->Release();
for (auto output: dxgiOutputVector)
{
for (auto output : dxgiOutputVector) {
output->Release();
}
return false;
}
IDXGIOutput6* dxgiOutput6 = nullptr;
hr = dxgiOutput->QueryInterface(__uuidof(IDXGIOutput6), reinterpret_cast<void**>(&dxgiOutput6));
for (auto output: dxgiOutputVector)
{
IDXGIOutput6 *dxgiOutput6 = nullptr;
hr = dxgiOutput->QueryInterface(__uuidof(IDXGIOutput6), reinterpret_cast<void **>(&dxgiOutput6));
for (auto output : dxgiOutputVector) {
output->Release();
}
if (FAILED(hr))
{
if (FAILED(hr)) {
_pLastError = "Failed to QueryInterface IDXGIOutput6 ErrorCode = " + QString::number(uint(hr), 16);
qDebug() << _pLastError;
d3dDevice->Release();
@@ -132,149 +109,124 @@ bool ElaDxgi::initialize(int dxID, int outputID)
dxgiOutput6->Release();
return false;
}
UINT supportedFormatsCount = 1; // 支持的格式数
DXGI_FORMAT supportedFormats[1] = {DXGI_FORMAT_B8G8R8A8_UNORM};
hr = dxgiOutput6->DuplicateOutput1(d3dDevice, 0, supportedFormatsCount, supportedFormats, &_duplication);
UINT supportedFormatsCount = 1; // 支持的格式数
DXGI_FORMAT supportedFormats[1] = {DXGI_FORMAT_B8G8R8A8_UNORM};
hr = dxgiOutput6->DuplicateOutput1(d3dDevice, 0, supportedFormatsCount, supportedFormats, &_duplication);
dxgiOutput6->Release();
if (FAILED(hr))
{
if (FAILED(hr)) {
_pLastError = "Failed to DuplicateOutput ErrorCode = " + QString::number(uint(hr), 16);
qDebug() << _pLastError;
d3dDevice->Release();
d3dContext->Release();
if (_duplication)
{
if (_duplication) {
_duplication->Release();
_duplication = nullptr;
}
return false;
}
_device = d3dDevice;
_context = d3dContext;
_device = d3dDevice;
_context = d3dContext;
_pIsInitSuccess = true;
return true;
}
QImage ElaDxgi::getGrabImage() const
{
QImage ElaDxgi::getGrabImage() const {
QImage grabImage(_imageBits, _descWidth, _descHeight, QImage::Format_ARGB32);
if (_pIsGrabCenter)
{
return grabImage.copy(QRect((_descWidth - _pGrabArea.width()) / 2, (_descHeight - _pGrabArea.height()) / 2, _pGrabArea.width(), _pGrabArea.height()));
}
else
{
if (_pIsGrabCenter) {
return grabImage.copy(
QRect((_descWidth - _pGrabArea.width()) / 2, (_descHeight - _pGrabArea.height()) / 2, _pGrabArea.width(), _pGrabArea.height()));
} else {
return grabImage.copy(_pGrabArea);
}
}
void ElaDxgi::onGrabScreen()
{
if (!_duplication || !_device || !_context)
{
void ElaDxgi::onGrabScreen() {
if (!_duplication || !_device || !_context) {
setIsGrabStoped(true);
return;
}
_pIsGrabStoped = false;
while (_pIsGrabActive)
{
IDXGIResource* desktopRes = nullptr;
while (_pIsGrabActive) {
IDXGIResource *desktopRes = nullptr;
DXGI_OUTDUPL_FRAME_INFO frameInfo;
while (true)
{
if (!_pIsGrabActive)
{
while (true) {
if (!_pIsGrabActive) {
setIsGrabStoped(true);
return;
}
_grabTimer.start();
_duplication->ReleaseFrame();
HRESULT hr = _duplication->AcquireNextFrame(_pTimeoutMsValue, &frameInfo, &desktopRes);
if (FAILED(hr))
{
if (hr != DXGI_ERROR_WAIT_TIMEOUT)
{
if (desktopRes)
{
if (FAILED(hr)) {
if (hr != DXGI_ERROR_WAIT_TIMEOUT) {
if (desktopRes) {
desktopRes->Release();
desktopRes = nullptr;
}
while (true)
{
if (!_duplication || !_device || !_context)
{
while (true) {
if (!_duplication || !_device || !_context) {
setIsGrabStoped(true);
return;
}
if (initialize(_pDxDeviceID, _pOutputDeviceID))
{
if (initialize(_pDxDeviceID, _pOutputDeviceID)) {
break;
}
}
}
continue;
}
if (frameInfo.LastPresentTime.QuadPart)
{
if (frameInfo.LastPresentTime.QuadPart) {
break;
}
}
D3D11_TEXTURE2D_DESC desc;
ID3D11Texture2D* textrueRes = nullptr;
HRESULT hr = desktopRes->QueryInterface(__uuidof(ID3D11Texture2D), reinterpret_cast<void**>(&textrueRes));
ID3D11Texture2D *textrueRes = nullptr;
HRESULT hr = desktopRes->QueryInterface(__uuidof(ID3D11Texture2D), reinterpret_cast<void **>(&textrueRes));
desktopRes->Release();
if (FAILED(hr))
{
if (FAILED(hr)) {
qDebug() << "Failed to ID3D11Texture2D result =" << QString::number(uint(hr), 16);
continue;
}
textrueRes->GetDesc(&desc);
D3D11_TEXTURE2D_DESC texDesc;
ZeroMemory(&texDesc, sizeof(texDesc));
texDesc.Width = desc.Width;
texDesc.Height = desc.Height;
texDesc.MipLevels = 1;
texDesc.ArraySize = 1;
texDesc.SampleDesc.Count = 1;
texDesc.Width = desc.Width;
texDesc.Height = desc.Height;
texDesc.MipLevels = 1;
texDesc.ArraySize = 1;
texDesc.SampleDesc.Count = 1;
texDesc.SampleDesc.Quality = 0;
texDesc.Usage = D3D11_USAGE_STAGING;
texDesc.Format = desc.Format;
texDesc.BindFlags = 0;
texDesc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
texDesc.MiscFlags = 0;
texDesc.Usage = D3D11_USAGE_STAGING;
texDesc.Format = desc.Format;
texDesc.BindFlags = 0;
texDesc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
texDesc.MiscFlags = 0;
_device->CreateTexture2D(&texDesc, nullptr, &_texture);
_context->CopyResource(_texture, textrueRes);
IDXGISurface1* surface = nullptr;
hr = _texture->QueryInterface(__uuidof(IDXGISurface1), reinterpret_cast<void**>(&surface));
if (FAILED(hr))
{
qDebug() << "Failed to QueryInterface IDXGISurface1 ErrorCode ="
<< QString::number(uint(hr), 16);
IDXGISurface1 *surface = nullptr;
hr = _texture->QueryInterface(__uuidof(IDXGISurface1), reinterpret_cast<void **>(&surface));
if (FAILED(hr)) {
qDebug() << "Failed to QueryInterface IDXGISurface1 ErrorCode =" << QString::number(uint(hr), 16);
continue;
}
DXGI_MAPPED_RECT map;
surface->Map(&map, DXGI_MAP_READ);
_imageBits = static_cast<uchar*>(map.pBits);
_descWidth = desc.Width;
_imageBits = static_cast<uchar *>(map.pBits);
_descWidth = desc.Width;
_descHeight = desc.Height;
surface->Unmap();
surface->Release();
_texture->Release();
QImage grabImage(_imageBits, _descWidth, _descHeight, QImage::Format_ARGB32);
if (_pIsGrabCenter)
{
Q_EMIT grabScreenOver(std::move(grabImage.copy(QRect((_descWidth - _pGrabArea.width()) / 2, (_descHeight - _pGrabArea.height()) / 2, _pGrabArea.width(), _pGrabArea.height()))));
}
else
{
if (_pIsGrabCenter) {
Q_EMIT grabScreenOver(std::move(grabImage.copy(
QRect((_descWidth - _pGrabArea.width()) / 2, (_descHeight - _pGrabArea.height()) / 2, _pGrabArea.width(), _pGrabArea.height()))));
} else {
Q_EMIT grabScreenOver(std::move(grabImage.copy(_pGrabArea)));
}
if (_lastGrabTime == 0)
{
if (_lastGrabTime == 0) {
_lastGrabTime = _grabTimer.elapsed(); // 毫秒
}
else
{
} else {
_lastGrabTime = (_grabTimer.elapsed() + _lastGrabTime) / 2;
}
_cpuSleepTime = (1000 - _lastGrabTime * _pGrabFrameRate) / _pGrabFrameRate;
@@ -284,29 +236,23 @@ void ElaDxgi::onGrabScreen()
setIsGrabStoped(true);
}
void ElaDxgi::releaseInterface()
{
if (_duplication)
{
void ElaDxgi::releaseInterface() {
if (_duplication) {
_duplication->Release();
_duplication = nullptr;
}
if (_device)
{
if (_device) {
_device->Release();
_device = nullptr;
}
if (_context)
{
if (_context) {
_context->Release();
_context = nullptr;
}
}
void ElaDxgi::cpuSleep(qint64 usec)
{
if (usec <= 0)
{
void ElaDxgi::cpuSleep(qint64 usec) {
if (usec <= 0) {
return;
}
LARGE_INTEGER cpuFerq;
@@ -314,11 +260,9 @@ void ElaDxgi::cpuSleep(qint64 usec)
LARGE_INTEGER endTime;
QueryPerformanceFrequency(&cpuFerq);
QueryPerformanceCounter(&startTime);
while (true)
{
while (true) {
QueryPerformanceCounter(&endTime);
if (((endTime.QuadPart - startTime.QuadPart) * 1000000) / cpuFerq.QuadPart > usec)
{
if (((endTime.QuadPart - startTime.QuadPart) * 1000000) / cpuFerq.QuadPart > usec) {
break;
}
}