1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include "MonitorResources.hpp"
#include "./cm/ColorManagement.hpp"
#include "../render/Renderer.hpp"
#include <algorithm>
#include <format>
using namespace Monitor;
using namespace NColorManagement;
static const int MAX_WORK_BUFFERS = 8;
static const int MAX_UNUSED_SECONDS = 5;
CMonitorResources::CMonitorResources(WP<CMonitor> monitor, DRMFormat format, Vector2D size, NColorManagement::PImageDescription imageDescription) :
m_stencilTex(g_pHyprRenderer->createStencilTexture(monitor->m_pixelSize.x, monitor->m_pixelSize.y)),
m_blurFB(g_pHyprRenderer->createFB(std::format("Monitor {} blur FB", monitor->m_name))), m_monitor(monitor), m_drmFormat(format), m_size(size),
m_imageDescription(imageDescription) {
initFB(m_blurFB);
monitor->m_blurFBDirty = true;
}
void CMonitorResources::initFB(SP<Render::IFramebuffer> fb) {
fb->addStencil(m_stencilTex);
fb->alloc(m_size.x, m_size.y, m_drmFormat);
fb->setImageDescription(m_imageDescription);
}
void CMonitorResources::setImageDescription(NColorManagement::PImageDescription imageDescription) {
if (m_imageDescription == imageDescription)
return;
m_imageDescription = imageDescription;
m_blurFB->setImageDescription(imageDescription);
for (const auto& res : m_workBuffers)
res.buffer->setImageDescription(imageDescription);
if (m_monitorMirrorFB)
m_monitorMirrorFB->setImageDescription(getMirrorTexImageDescription());
if (m_mirrorTex)
m_mirrorTex->m_imageDescription = getMirrorTexImageDescription();
}
SP<Render::IFramebuffer> CMonitorResources::getUnusedWorkBuffer() {
std::erase_if(m_workBuffers, [](const auto& res) { return res.lastUsed.getSeconds() >= MAX_UNUSED_SECONDS; });
auto found = std::ranges::find_if(m_workBuffers, [](const auto& res) { return res.buffer.strongRef() < 2; });
if (found != m_workBuffers.end()) {
found->lastUsed.reset();
return found->buffer;
}
if (m_workBuffers.size() >= MAX_WORK_BUFFERS)
return nullptr;
auto& res = m_workBuffers.emplace_back(g_pHyprRenderer->createFB(std::format("Monitor {} workbuffer", m_monitor->m_name)));
initFB(res.buffer);
res.lastUsed.reset();
return res.buffer;
}
void CMonitorResources::forEachUnusedFB(std::function<void(SP<Render::IFramebuffer>)> callback, bool includeNamed) {
for (const auto& res : m_workBuffers) {
if (res.buffer.strongRef() > 1)
continue;
callback(res.buffer);
}
if (includeNamed) {
if (m_blurFB && m_blurFB->isAllocated() && m_blurFB.strongRef() < 2)
callback(m_blurFB);
if (hasMirrorFB() && m_monitorMirrorFB.strongRef() < 2)
callback(m_monitorMirrorFB);
}
}
bool CMonitorResources::hasMirrorFB() {
return m_monitorMirrorFB && m_monitorMirrorFB->isAllocated();
}
SP<Render::IFramebuffer> CMonitorResources::mirrorFB() {
if (!m_monitorMirrorFB)
m_monitorMirrorFB = g_pHyprRenderer->createFB(std::format("Monitor {} mirror FB", m_monitor->m_name));
if (!m_monitorMirrorFB->isAllocated()) {
m_monitorMirrorFB->alloc(m_size.x, m_size.y, m_monitor->m_activeMonitorRule.m_enable10bit ? DRM_FORMAT_XRGB2101010 : DRM_FORMAT_XRGB8888);
m_monitorMirrorFB->setImageDescription(getMirrorTexImageDescription());
}
return m_monitorMirrorFB;
}
SP<Render::ITexture> CMonitorResources::getMirrorTexture() {
return hasMirrorFB() ? mirrorFB()->getTexture() : nullptr;
}
NColorManagement::PImageDescription CMonitorResources::getMirrorTexImageDescription() {
const auto TF = m_imageDescription->value().transferFunction;
if (TF == CM_TRANSFER_FUNCTION_GAMMA22 || TF == CM_TRANSFER_FUNCTION_SRGB)
return m_imageDescription;
return DEFAULT_SRGB_IMAGE_DESCRIPTION;
}
void CMonitorResources::enableMirror() {
if (m_mirrorTex)
return;
m_mirrorTex = g_pHyprRenderer->createTexture();
m_mirrorTex->allocate({m_size.x, m_size.y}, m_monitor->m_activeMonitorRule.m_enable10bit ? DRM_FORMAT_XRGB2101010 : DRM_FORMAT_XRGB8888);
m_mirrorTex->m_imageDescription = getMirrorTexImageDescription();
m_monitor->m_blurFBDirty = true;
}
void CMonitorResources::disableMirror() {
if (m_mirrorTex)
m_monitor->m_blurFBDirty = true;
m_mirrorTex.reset();
}
|
