1#include <winpr/sysinfo.h>
2#include <winpr/assert.h>
6#include <freerdp/settings.h>
7#include <freerdp/codec/region.h>
8#include <freerdp/primitives.h>
9#include <freerdp/log.h>
10#include <freerdp/codec/yuv.h>
12#define TAG FREERDP_TAG("codec")
19 const BYTE* pYUVData[3];
25} YUV_PROCESS_WORK_PARAM;
30 const BYTE* pYUVData[3];
36} YUV_COMBINE_WORK_PARAM;
48 BYTE* pYUVLumaData[3];
49 BYTE* pYUVChromaData[3];
51} YUV_ENCODE_WORK_PARAM;
64 UINT32 work_object_count;
65 PTP_WORK* work_objects;
66 YUV_ENCODE_WORK_PARAM* work_enc_params;
67 YUV_PROCESS_WORK_PARAM* work_dec_params;
68 YUV_COMBINE_WORK_PARAM* work_combined_params;
71static INLINE BOOL avc420_yuv_to_rgb(
const BYTE* WINPR_RESTRICT pYUVData[3],
72 const UINT32 iStride[3],
73 const RECTANGLE_16* WINPR_RESTRICT rect, UINT32 nDstStep,
74 BYTE* WINPR_RESTRICT pDstData, DWORD DstFormat)
78 const BYTE* pYUVPoint[3];
80 WINPR_ASSERT(pYUVData);
81 WINPR_ASSERT(iStride);
83 WINPR_ASSERT(pDstData);
85 const INT32 width = rect->right - rect->left;
86 const INT32 height = rect->bottom - rect->top;
87 BYTE* pDstPoint = pDstData + 1ULL * rect->top * nDstStep +
88 1ULL * rect->left * FreeRDPGetBytesPerPixel(DstFormat);
90 pYUVPoint[0] = pYUVData[0] + 1ULL * rect->top * iStride[0] + rect->left;
91 pYUVPoint[1] = pYUVData[1] + 1ULL * rect->top / 2 * iStride[1] + rect->left / 2;
92 pYUVPoint[2] = pYUVData[2] + 1ULL * rect->top / 2 * iStride[2] + rect->left / 2;
94 roi.width = WINPR_ASSERTING_INT_CAST(uint32_t, width);
95 roi.height = WINPR_ASSERTING_INT_CAST(uint32_t, height);
97 if (prims->YUV420ToRGB_8u_P3AC4R(pYUVPoint, iStride, pDstPoint, nDstStep, DstFormat, &roi) !=
104static INLINE BOOL avc444_yuv_to_rgb(
const BYTE* WINPR_RESTRICT pYUVData[3],
105 const UINT32 iStride[3],
106 const RECTANGLE_16* WINPR_RESTRICT rect, UINT32 nDstStep,
107 BYTE* WINPR_RESTRICT pDstData, DWORD DstFormat)
111 const BYTE* pYUVPoint[3];
113 WINPR_ASSERT(pYUVData);
114 WINPR_ASSERT(iStride);
116 WINPR_ASSERT(pDstData);
118 const INT32 width = rect->right - rect->left;
119 const INT32 height = rect->bottom - rect->top;
120 BYTE* pDstPoint = pDstData + 1ULL * rect->top * nDstStep +
121 1ULL * rect->left * FreeRDPGetBytesPerPixel(DstFormat);
123 pYUVPoint[0] = pYUVData[0] + 1ULL * rect->top * iStride[0] + rect->left;
124 pYUVPoint[1] = pYUVData[1] + 1ULL * rect->top * iStride[1] + rect->left;
125 pYUVPoint[2] = pYUVData[2] + 1ULL * rect->top * iStride[2] + rect->left;
127 roi.width = WINPR_ASSERTING_INT_CAST(uint32_t, width);
128 roi.height = WINPR_ASSERTING_INT_CAST(uint32_t, height);
130 if (prims->YUV444ToRGB_8u_P3AC4R(pYUVPoint, iStride, pDstPoint, nDstStep, DstFormat, &roi) !=
137static void CALLBACK yuv420_process_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
140 YUV_PROCESS_WORK_PARAM* param = (YUV_PROCESS_WORK_PARAM*)context;
141 WINPR_UNUSED(instance);
145 if (!avc420_yuv_to_rgb(param->pYUVData, param->iStride, ¶m->rect, param->nDstStep,
146 param->dest, param->DstFormat))
147 WLog_WARN(TAG,
"avc420_yuv_to_rgb failed");
150static void CALLBACK yuv444_process_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
153 YUV_PROCESS_WORK_PARAM* param = (YUV_PROCESS_WORK_PARAM*)context;
154 WINPR_UNUSED(instance);
158 if (!avc444_yuv_to_rgb(param->pYUVData, param->iStride, ¶m->rect, param->nDstStep,
159 param->dest, param->DstFormat))
160 WLog_WARN(TAG,
"avc444_yuv_to_rgb failed");
163BOOL yuv_context_reset(YUV_CONTEXT* WINPR_RESTRICT context, UINT32 width, UINT32 height)
166 WINPR_ASSERT(context);
168 context->width = width;
169 context->height = height;
170 context->heightStep = (height / context->nthreads);
172 if (context->useThreads)
179 const size_t pw = (width + TILE_SIZE - width % TILE_SIZE) / 16;
180 const size_t ph = (height + TILE_SIZE - height % TILE_SIZE) / 16;
182 const size_t count = pw * ph;
184 context->work_object_count = 0;
185 if (context->encoder)
187 void* tmp = winpr_aligned_recalloc(context->work_enc_params, count,
188 sizeof(YUV_ENCODE_WORK_PARAM), 32);
191 memset(tmp, 0, count *
sizeof(YUV_ENCODE_WORK_PARAM));
193 context->work_enc_params = tmp;
197 void* tmp = winpr_aligned_recalloc(context->work_dec_params, count,
198 sizeof(YUV_PROCESS_WORK_PARAM), 32);
201 memset(tmp, 0, count *
sizeof(YUV_PROCESS_WORK_PARAM));
203 context->work_dec_params = tmp;
205 void* ctmp = winpr_aligned_recalloc(context->work_combined_params, count,
206 sizeof(YUV_COMBINE_WORK_PARAM), 32);
209 memset(ctmp, 0, count *
sizeof(YUV_COMBINE_WORK_PARAM));
211 context->work_combined_params = ctmp;
215 winpr_aligned_recalloc((
void*)context->work_objects, count,
sizeof(PTP_WORK), 32);
218 memset(wtmp, 0, count *
sizeof(PTP_WORK));
220 context->work_objects = (PTP_WORK*)wtmp;
221 context->work_object_count = WINPR_ASSERTING_INT_CAST(uint32_t, count);
228YUV_CONTEXT* yuv_context_new(BOOL encoder, UINT32 ThreadingFlags)
231 YUV_CONTEXT* ret = winpr_aligned_calloc(1,
sizeof(*ret), 32);
238 ret->encoder = encoder;
240 if (!(ThreadingFlags & THREADING_FLAGS_DISABLE_THREADS))
242 GetNativeSystemInfo(&sysInfos);
243 ret->useThreads = (sysInfos.dwNumberOfProcessors > 1);
246 ret->nthreads = sysInfos.dwNumberOfProcessors;
247 ret->threadPool = CreateThreadpool(NULL);
248 if (!ret->threadPool)
250 goto error_threadpool;
253 InitializeThreadpoolEnvironment(&ret->ThreadPoolEnv);
254 SetThreadpoolCallbackPool(&ret->ThreadPoolEnv, ret->threadPool);
261 WINPR_PRAGMA_DIAG_PUSH
262 WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
263 yuv_context_free(ret);
264 WINPR_PRAGMA_DIAG_POP
268void yuv_context_free(YUV_CONTEXT* context)
272 if (context->useThreads)
274 if (context->threadPool)
275 CloseThreadpool(context->threadPool);
276 DestroyThreadpoolEnvironment(&context->ThreadPoolEnv);
277 winpr_aligned_free((
void*)context->work_objects);
278 winpr_aligned_free(context->work_combined_params);
279 winpr_aligned_free(context->work_enc_params);
280 winpr_aligned_free(context->work_dec_params);
282 winpr_aligned_free(context);
285static INLINE YUV_PROCESS_WORK_PARAM pool_decode_param(
const RECTANGLE_16* WINPR_RESTRICT rect,
286 YUV_CONTEXT* WINPR_RESTRICT context,
287 const BYTE* WINPR_RESTRICT pYUVData[3],
288 const UINT32 iStride[3], UINT32 DstFormat,
289 BYTE* WINPR_RESTRICT dest, UINT32 nDstStep)
291 YUV_PROCESS_WORK_PARAM current = { 0 };
294 WINPR_ASSERT(context);
295 WINPR_ASSERT(pYUVData);
296 WINPR_ASSERT(iStride);
299 current.context = context;
300 current.DstFormat = DstFormat;
301 current.pYUVData[0] = pYUVData[0];
302 current.pYUVData[1] = pYUVData[1];
303 current.pYUVData[2] = pYUVData[2];
304 current.iStride[0] = iStride[0];
305 current.iStride[1] = iStride[1];
306 current.iStride[2] = iStride[2];
307 current.nDstStep = nDstStep;
309 current.rect = *rect;
313static BOOL submit_object(PTP_WORK* WINPR_RESTRICT work_object, PTP_WORK_CALLBACK cb,
314 const void* WINPR_RESTRICT param, YUV_CONTEXT* WINPR_RESTRICT context)
329 if (!param || !context)
332 *work_object = CreateThreadpoolWork(cb, cnv.pv, &context->ThreadPoolEnv);
336 SubmitThreadpoolWork(*work_object);
340static void free_objects(PTP_WORK* work_objects, UINT32 waitCount)
342 WINPR_ASSERT(work_objects || (waitCount == 0));
344 for (UINT32 i = 0; i < waitCount; i++)
346 PTP_WORK cur = work_objects[i];
347 work_objects[i] = NULL;
352 WaitForThreadpoolWorkCallbacks(cur, FALSE);
353 CloseThreadpoolWork(cur);
357static BOOL intersects(UINT32 pos,
const RECTANGLE_16* WINPR_RESTRICT regionRects,
358 UINT32 numRegionRects)
360 WINPR_ASSERT(regionRects || (numRegionRects == 0));
362 for (UINT32 x = pos + 1; x < numRegionRects; x++)
367 if (rectangles_intersects(what, rect))
369 WLog_WARN(TAG,
"YUV decoder: intersecting rectangles, aborting");
377static RECTANGLE_16 clamp(YUV_CONTEXT* WINPR_RESTRICT context,
378 const RECTANGLE_16* WINPR_RESTRICT rect, UINT32 srcHeight)
380 WINPR_ASSERT(context);
384 const UINT32 height = MIN(context->height, srcHeight);
386 c.top = WINPR_ASSERTING_INT_CAST(UINT16, height);
387 if (c.bottom > height)
388 c.bottom = WINPR_ASSERTING_INT_CAST(UINT16, height);
392static BOOL pool_decode(YUV_CONTEXT* WINPR_RESTRICT context, PTP_WORK_CALLBACK cb,
393 const BYTE* WINPR_RESTRICT pYUVData[3],
const UINT32 iStride[3],
394 UINT32 yuvHeight, UINT32 DstFormat, BYTE* WINPR_RESTRICT dest,
395 UINT32 nDstStep,
const RECTANGLE_16* WINPR_RESTRICT regionRects,
396 UINT32 numRegionRects)
399 UINT32 waitCount = 0;
402 WINPR_ASSERT(context);
404 WINPR_ASSERT(pYUVData);
405 WINPR_ASSERT(iStride);
407 WINPR_ASSERT(regionRects || (numRegionRects == 0));
409 if (context->encoder)
411 WLog_ERR(TAG,
"YUV context set up for encoding, can not decode with it, aborting");
415 if (!context->useThreads || (primitives_flags(prims) & PRIM_FLAGS_HAVE_EXTGPU))
417 for (UINT32 y = 0; y < numRegionRects; y++)
419 const RECTANGLE_16 rect = clamp(context, ®ionRects[y], yuvHeight);
420 YUV_PROCESS_WORK_PARAM current =
421 pool_decode_param(&rect, context, pYUVData, iStride, DstFormat, dest, nDstStep);
422 cb(NULL, ¤t, NULL);
428 for (UINT32 x = 0; x < numRegionRects; x++)
430 RECTANGLE_16 r = clamp(context, ®ionRects[x], yuvHeight);
432 if (intersects(x, regionRects, numRegionRects))
435 while (r.left < r.right)
438 y.right = MIN(r.right, r.left + TILE_SIZE);
440 while (y.top < y.bottom)
444 if (context->work_object_count <= waitCount)
446 free_objects(context->work_objects, context->work_object_count);
450 YUV_PROCESS_WORK_PARAM* cur = &context->work_dec_params[waitCount];
451 z.bottom = MIN(z.bottom, z.top + TILE_SIZE);
452 if (rectangle_is_empty(&z))
454 *cur = pool_decode_param(&z, context, pYUVData, iStride, DstFormat, dest, nDstStep);
455 if (!submit_object(&context->work_objects[waitCount], cb, cur, context))
466 free_objects(context->work_objects, context->work_object_count);
470static INLINE BOOL check_rect(
const YUV_CONTEXT* WINPR_RESTRICT yuv,
471 const RECTANGLE_16* WINPR_RESTRICT rect, UINT32 nDstWidth,
478 if ((rect->right > yuv->width) || (rect->left > yuv->width))
481 if ((rect->top > yuv->height) || (rect->bottom > yuv->height))
485 if ((rect->right > nDstWidth) || (rect->left > nDstWidth))
488 if ((rect->bottom > nDstHeight) || (rect->top > nDstHeight))
494static void CALLBACK yuv444_combine_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
497 YUV_COMBINE_WORK_PARAM* param = (YUV_COMBINE_WORK_PARAM*)context;
501 YUV_CONTEXT* yuv = param->context;
507 const UINT32 alignedWidth = yuv->width + ((yuv->width % 16 != 0) ? 16 - yuv->width % 16 : 0);
508 const UINT32 alignedHeight =
509 yuv->height + ((yuv->height % 16 != 0) ? 16 - yuv->height % 16 : 0);
511 WINPR_UNUSED(instance);
514 if (!check_rect(param->context, rect, yuv->width, yuv->height))
517 if (prims->YUV420CombineToYUV444(param->type, param->pYUVData, param->iStride, alignedWidth,
518 alignedHeight, param->pYUVDstData, param->iDstStride,
519 rect) != PRIMITIVES_SUCCESS)
520 WLog_WARN(TAG,
"YUV420CombineToYUV444 failed");
523static INLINE YUV_COMBINE_WORK_PARAM
524pool_decode_rect_param(
const RECTANGLE_16* WINPR_RESTRICT rect, YUV_CONTEXT* WINPR_RESTRICT context,
525 BYTE type,
const BYTE* WINPR_RESTRICT pYUVData[3],
const UINT32 iStride[3],
526 BYTE* WINPR_RESTRICT pYUVDstData[3],
const UINT32 iDstStride[3])
528 YUV_COMBINE_WORK_PARAM current = { 0 };
531 WINPR_ASSERT(context);
532 WINPR_ASSERT(pYUVData);
533 WINPR_ASSERT(iStride);
534 WINPR_ASSERT(pYUVDstData);
535 WINPR_ASSERT(iDstStride);
537 current.context = context;
538 current.pYUVData[0] = pYUVData[0];
539 current.pYUVData[1] = pYUVData[1];
540 current.pYUVData[2] = pYUVData[2];
541 current.pYUVDstData[0] = pYUVDstData[0];
542 current.pYUVDstData[1] = pYUVDstData[1];
543 current.pYUVDstData[2] = pYUVDstData[2];
544 current.iStride[0] = iStride[0];
545 current.iStride[1] = iStride[1];
546 current.iStride[2] = iStride[2];
547 current.iDstStride[0] = iDstStride[0];
548 current.iDstStride[1] = iDstStride[1];
549 current.iDstStride[2] = iDstStride[2];
551 current.rect = *rect;
555static BOOL pool_decode_rect(YUV_CONTEXT* WINPR_RESTRICT context, BYTE type,
556 const BYTE* WINPR_RESTRICT pYUVData[3],
const UINT32 iStride[3],
557 BYTE* WINPR_RESTRICT pYUVDstData[3],
const UINT32 iDstStride[3],
558 const RECTANGLE_16* WINPR_RESTRICT regionRects, UINT32 numRegionRects)
561 UINT32 waitCount = 0;
562 PTP_WORK_CALLBACK cb = yuv444_combine_work_callback;
565 WINPR_ASSERT(context);
566 WINPR_ASSERT(pYUVData);
567 WINPR_ASSERT(iStride);
568 WINPR_ASSERT(pYUVDstData);
569 WINPR_ASSERT(iDstStride);
570 WINPR_ASSERT(regionRects || (numRegionRects == 0));
572 if (!context->useThreads || (primitives_flags(prims) & PRIM_FLAGS_HAVE_EXTGPU))
574 for (UINT32 y = 0; y < numRegionRects; y++)
576 YUV_COMBINE_WORK_PARAM current = pool_decode_rect_param(
577 ®ionRects[y], context, type, pYUVData, iStride, pYUVDstData, iDstStride);
578 cb(NULL, ¤t, NULL);
584 for (waitCount = 0; waitCount < numRegionRects; waitCount++)
586 YUV_COMBINE_WORK_PARAM* current = NULL;
588 if (context->work_object_count <= waitCount)
590 free_objects(context->work_objects, context->work_object_count);
593 current = &context->work_combined_params[waitCount];
594 *current = pool_decode_rect_param(®ionRects[waitCount], context, type, pYUVData, iStride,
595 pYUVDstData, iDstStride);
597 if (!submit_object(&context->work_objects[waitCount], cb, current, context))
603 free_objects(context->work_objects, context->work_object_count);
607BOOL yuv444_context_decode(YUV_CONTEXT* WINPR_RESTRICT context, BYTE type,
608 const BYTE* WINPR_RESTRICT pYUVData[3],
const UINT32 iStride[3],
609 UINT32 srcYuvHeight, BYTE* WINPR_RESTRICT pYUVDstData[3],
610 const UINT32 iDstStride[3], DWORD DstFormat, BYTE* WINPR_RESTRICT dest,
611 UINT32 nDstStep,
const RECTANGLE_16* WINPR_RESTRICT regionRects,
612 UINT32 numRegionRects)
614 const BYTE* pYUVCDstData[3];
616 WINPR_ASSERT(context);
617 WINPR_ASSERT(pYUVData);
618 WINPR_ASSERT(iStride);
619 WINPR_ASSERT(pYUVDstData);
620 WINPR_ASSERT(iDstStride);
622 WINPR_ASSERT(regionRects || (numRegionRects == 0));
624 if (context->encoder)
626 WLog_ERR(TAG,
"YUV context set up for encoding, can not decode with it, aborting");
629 if (!pool_decode_rect(context, type, pYUVData, iStride, pYUVDstData, iDstStride, regionRects,
633 pYUVCDstData[0] = pYUVDstData[0];
634 pYUVCDstData[1] = pYUVDstData[1];
635 pYUVCDstData[2] = pYUVDstData[2];
636 return pool_decode(context, yuv444_process_work_callback, pYUVCDstData, iDstStride,
637 srcYuvHeight, DstFormat, dest, nDstStep, regionRects, numRegionRects);
640BOOL yuv420_context_decode(YUV_CONTEXT* WINPR_RESTRICT context,
641 const BYTE* WINPR_RESTRICT pYUVData[3],
const UINT32 iStride[3],
642 UINT32 yuvHeight, DWORD DstFormat, BYTE* WINPR_RESTRICT dest,
643 UINT32 nDstStep,
const RECTANGLE_16* WINPR_RESTRICT regionRects,
644 UINT32 numRegionRects)
646 return pool_decode(context, yuv420_process_work_callback, pYUVData, iStride, yuvHeight,
647 DstFormat, dest, nDstStep, regionRects, numRegionRects);
650static void CALLBACK yuv420_encode_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
654 YUV_ENCODE_WORK_PARAM* param = (YUV_ENCODE_WORK_PARAM*)context;
657 const BYTE* src = NULL;
659 WINPR_UNUSED(instance);
663 roi.width = param->rect.right - param->rect.left;
664 roi.height = param->rect.bottom - param->rect.top;
665 src = param->pSrcData + 1ULL * param->nSrcStep * param->rect.top +
666 1ULL * param->rect.left * FreeRDPGetBytesPerPixel(param->SrcFormat);
668 param->pYUVLumaData[0] + 1ULL * param->rect.top * param->iStride[0] + param->rect.left;
669 pYUVData[1] = param->pYUVLumaData[1] + 1ULL * param->rect.top / 2 * param->iStride[1] +
670 param->rect.left / 2;
671 pYUVData[2] = param->pYUVLumaData[2] + 1ULL * param->rect.top / 2 * param->iStride[2] +
672 param->rect.left / 2;
674 if (prims->RGBToYUV420_8u_P3AC4R(src, param->SrcFormat, param->nSrcStep, pYUVData,
675 param->iStride, &roi) != PRIMITIVES_SUCCESS)
677 WLog_ERR(TAG,
"error when decoding lines");
681static void CALLBACK yuv444v1_encode_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
685 YUV_ENCODE_WORK_PARAM* param = (YUV_ENCODE_WORK_PARAM*)context;
687 BYTE* pYUVLumaData[3];
688 BYTE* pYUVChromaData[3];
689 const BYTE* src = NULL;
691 WINPR_UNUSED(instance);
695 roi.width = param->rect.right - param->rect.left;
696 roi.height = param->rect.bottom - param->rect.top;
697 src = param->pSrcData + 1ULL * param->nSrcStep * param->rect.top +
698 1ULL * param->rect.left * FreeRDPGetBytesPerPixel(param->SrcFormat);
700 param->pYUVLumaData[0] + 1ULL * param->rect.top * param->iStride[0] + param->rect.left;
701 pYUVLumaData[1] = param->pYUVLumaData[1] + 1ULL * param->rect.top / 2 * param->iStride[1] +
702 param->rect.left / 2;
703 pYUVLumaData[2] = param->pYUVLumaData[2] + 1ULL * param->rect.top / 2 * param->iStride[2] +
704 param->rect.left / 2;
706 param->pYUVChromaData[0] + 1ULL * param->rect.top * param->iStride[0] + param->rect.left;
707 pYUVChromaData[1] = param->pYUVChromaData[1] + 1ULL * param->rect.top / 2 * param->iStride[1] +
708 param->rect.left / 2;
709 pYUVChromaData[2] = param->pYUVChromaData[2] + 1ULL * param->rect.top / 2 * param->iStride[2] +
710 param->rect.left / 2;
711 if (prims->RGBToAVC444YUV(src, param->SrcFormat, param->nSrcStep, pYUVLumaData, param->iStride,
712 pYUVChromaData, param->iStride, &roi) != PRIMITIVES_SUCCESS)
714 WLog_ERR(TAG,
"error when decoding lines");
718static void CALLBACK yuv444v2_encode_work_callback(PTP_CALLBACK_INSTANCE instance,
void* context,
722 YUV_ENCODE_WORK_PARAM* param = (YUV_ENCODE_WORK_PARAM*)context;
724 BYTE* pYUVLumaData[3];
725 BYTE* pYUVChromaData[3];
726 const BYTE* src = NULL;
728 WINPR_UNUSED(instance);
732 roi.width = param->rect.right - param->rect.left;
733 roi.height = param->rect.bottom - param->rect.top;
734 src = param->pSrcData + 1ULL * param->nSrcStep * param->rect.top +
735 1ULL * param->rect.left * FreeRDPGetBytesPerPixel(param->SrcFormat);
737 param->pYUVLumaData[0] + 1ULL * param->rect.top * param->iStride[0] + param->rect.left;
738 pYUVLumaData[1] = param->pYUVLumaData[1] + 1ULL * param->rect.top / 2 * param->iStride[1] +
739 param->rect.left / 2;
740 pYUVLumaData[2] = param->pYUVLumaData[2] + 1ULL * param->rect.top / 2 * param->iStride[2] +
741 param->rect.left / 2;
743 param->pYUVChromaData[0] + 1ULL * param->rect.top * param->iStride[0] + param->rect.left;
744 pYUVChromaData[1] = param->pYUVChromaData[1] + 1ULL * param->rect.top / 2 * param->iStride[1] +
745 param->rect.left / 2;
746 pYUVChromaData[2] = param->pYUVChromaData[2] + 1ULL * param->rect.top / 2 * param->iStride[2] +
747 param->rect.left / 2;
748 if (prims->RGBToAVC444YUVv2(src, param->SrcFormat, param->nSrcStep, pYUVLumaData,
749 param->iStride, pYUVChromaData, param->iStride,
750 &roi) != PRIMITIVES_SUCCESS)
752 WLog_ERR(TAG,
"error when decoding lines");
756static INLINE YUV_ENCODE_WORK_PARAM pool_encode_fill(
757 const RECTANGLE_16* WINPR_RESTRICT rect, YUV_CONTEXT* WINPR_RESTRICT context,
758 const BYTE* WINPR_RESTRICT pSrcData, UINT32 nSrcStep, UINT32 SrcFormat,
const UINT32 iStride[],
759 BYTE* WINPR_RESTRICT pYUVLumaData[], BYTE* WINPR_RESTRICT pYUVChromaData[])
761 YUV_ENCODE_WORK_PARAM current = { 0 };
764 WINPR_ASSERT(context);
765 WINPR_ASSERT(pSrcData);
766 WINPR_ASSERT(iStride);
767 WINPR_ASSERT(pYUVLumaData);
769 current.context = context;
770 current.pSrcData = pSrcData;
771 current.SrcFormat = SrcFormat;
772 current.nSrcStep = nSrcStep;
773 current.pYUVLumaData[0] = pYUVLumaData[0];
774 current.pYUVLumaData[1] = pYUVLumaData[1];
775 current.pYUVLumaData[2] = pYUVLumaData[2];
778 current.pYUVChromaData[0] = pYUVChromaData[0];
779 current.pYUVChromaData[1] = pYUVChromaData[1];
780 current.pYUVChromaData[2] = pYUVChromaData[2];
782 current.iStride[0] = iStride[0];
783 current.iStride[1] = iStride[1];
784 current.iStride[2] = iStride[2];
786 current.rect = *rect;
791static BOOL pool_encode(YUV_CONTEXT* WINPR_RESTRICT context, PTP_WORK_CALLBACK cb,
792 const BYTE* WINPR_RESTRICT pSrcData, UINT32 nSrcStep, UINT32 SrcFormat,
793 const UINT32 iStride[], BYTE* WINPR_RESTRICT pYUVLumaData[],
794 BYTE* WINPR_RESTRICT pYUVChromaData[],
795 const RECTANGLE_16* WINPR_RESTRICT regionRects, UINT32 numRegionRects)
799 UINT32 waitCount = 0;
801 WINPR_ASSERT(context);
803 WINPR_ASSERT(pSrcData);
804 WINPR_ASSERT(iStride);
805 WINPR_ASSERT(regionRects || (numRegionRects == 0));
807 if (!context->encoder)
810 WLog_ERR(TAG,
"YUV context set up for decoding, can not encode with it, aborting");
814 if (!context->useThreads || (primitives_flags(prims) & PRIM_FLAGS_HAVE_EXTGPU))
816 for (UINT32 x = 0; x < numRegionRects; x++)
818 YUV_ENCODE_WORK_PARAM current =
819 pool_encode_fill(®ionRects[x], context, pSrcData, nSrcStep, SrcFormat, iStride,
820 pYUVLumaData, pYUVChromaData);
821 cb(NULL, ¤t, NULL);
827 for (UINT32 x = 0; x < numRegionRects; x++)
830 const UINT32 height = rect->bottom - rect->top;
831 const UINT32 steps = (height + context->heightStep / 2) / context->heightStep;
836 for (UINT32 x = 0; x < numRegionRects; x++)
839 const UINT32 height = rect->bottom - rect->top;
840 const UINT32 steps = (height + context->heightStep / 2) / context->heightStep;
842 for (UINT32 y = 0; y < steps; y++)
845 YUV_ENCODE_WORK_PARAM* current = NULL;
847 if (context->work_object_count <= waitCount)
849 free_objects(context->work_objects, context->work_object_count);
853 current = &context->work_enc_params[waitCount];
854 r.top += y * context->heightStep;
855 *current = pool_encode_fill(&r, context, pSrcData, nSrcStep, SrcFormat, iStride,
856 pYUVLumaData, pYUVChromaData);
857 if (!submit_object(&context->work_objects[waitCount], cb, current, context))
865 free_objects(context->work_objects, context->work_object_count);
869BOOL yuv420_context_encode(YUV_CONTEXT* WINPR_RESTRICT context,
const BYTE* WINPR_RESTRICT pSrcData,
870 UINT32 nSrcStep, UINT32 SrcFormat,
const UINT32 iStride[3],
871 BYTE* WINPR_RESTRICT pYUVData[3],
872 const RECTANGLE_16* WINPR_RESTRICT regionRects, UINT32 numRegionRects)
874 if (!context || !pSrcData || !iStride || !pYUVData || !regionRects)
877 return pool_encode(context, yuv420_encode_work_callback, pSrcData, nSrcStep, SrcFormat, iStride,
878 pYUVData, NULL, regionRects, numRegionRects);
881BOOL yuv444_context_encode(YUV_CONTEXT* WINPR_RESTRICT context, BYTE version,
882 const BYTE* WINPR_RESTRICT pSrcData, UINT32 nSrcStep, UINT32 SrcFormat,
883 const UINT32 iStride[3], BYTE* WINPR_RESTRICT pYUVLumaData[3],
884 BYTE* WINPR_RESTRICT pYUVChromaData[3],
885 const RECTANGLE_16* WINPR_RESTRICT regionRects, UINT32 numRegionRects)
887 PTP_WORK_CALLBACK cb = NULL;
891 cb = yuv444v1_encode_work_callback;
894 cb = yuv444v2_encode_work_callback;
900 return pool_encode(context, cb, pSrcData, nSrcStep, SrcFormat, iStride, pYUVLumaData,
901 pYUVChromaData, regionRects, numRegionRects);