extern/KasmVNC
1
0
Fork 0
mirror of https://github.com/kasmtech/KasmVNC.git synced 2025-06-27 21:21:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

VNC-127 Introduce TBB task arena for optimized parallel execution

Browse Source
This commit is contained in:
El 2025-05-08 16:59:22 +05:00
parent c89bf7b119
commit b51a09c477
No known key found for this signature in database
GPG Key ID: EB3F4C9EA29CDE59
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
common/rfb/EncodeManager.cxx
View File

@ -44,6 +44,7 @@
#include <rfb/TightWEBPEncoder.h> #include <rfb/TightWEBPEncoder.h>
#include <rfb/TightQOIEncoder.h> #include <rfb/TightQOIEncoder.h>
#include <execution> #include <execution>
#include <tbb/parallel_for.h>
using namespace rfb; using namespace rfb;
@ -210,6 +211,9 @@ EncodeManager::EncodeManager(SConnection* conn_, EncCache *encCache_) : conn(con
dynamicQualityMin = Server::dynamicQualityMin; dynamicQualityMin = Server::dynamicQualityMin;
dynamicQualityOff = Server::dynamicQualityMax - Server::dynamicQualityMin; dynamicQualityOff = Server::dynamicQualityMax - Server::dynamicQualityMin;
} }
const auto num_cores = tbb::this_task_arena::max_concurrency() / 2;
arena.initialize(num_cores);
} }
EncodeManager::~EncodeManager() EncodeManager::~EncodeManager()
@ -1237,13 +1241,14 @@ void EncodeManager::writeRects(const Region& changed, const PixelBuffer* pb,
} }
scalingTime = msSince(&scalestart); scalingTime = msSince(&scalestart);
std::for_each(std::execution::par_unseq, std::begin(indices), std::end(indices), [&](size_t i) arena.execute([&] {
{ tbb::parallel_for(static_cast<size_t>(0), subrects_size, [&](size_t i) {
encoderTypes[i] = getEncoderType(subrects[i], pb, &palettes[i], compresseds[i], encoderTypes[i] = getEncoderType(subrects[i], pb, &palettes[i], compresseds[i],
&isWebp[i], &fromCache[i], &isWebp[i], &fromCache[i],
scaledpb, scaledrects[i], ms[i]); scaledpb, scaledrects[i], ms[i]);
checkWebpFallback(start); checkWebpFallback(start);
}); });
});
for (uint32_t i = 0; i < subrects_size; ++i) { for (uint32_t i = 0; i < subrects_size; ++i) {
if (encoderTypes[i] == encoderFullColour) { if (encoderTypes[i] == encoderFullColour) {

2
common/rfb/EncodeManager.h
View File

@ -32,6 +32,7 @@
#include <stdint.h> #include <stdint.h>
#include <atomic> #include <atomic>
#include <tbb/task_arena.h>
#include <sys/time.h> #include <sys/time.h>
namespace rfb { namespace rfb {
@ -167,6 +168,7 @@ namespace rfb {
protected: protected:
SConnection *conn; SConnection *conn;
tbb::task_arena arena;
std::vector<Encoder*> encoders; std::vector<Encoder*> encoders;
std::vector<int> activeEncoders; std::vector<int> activeEncoders;