| /************************************************************************** |
| * |
| * Copyright 2019 Advanced Micro Devices, Inc. |
| * All Rights Reserved. |
| * |
| * 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, sub license, 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 (including the |
| * next paragraph) 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 NON-INFRINGEMENT. |
| * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS 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. |
| * |
| * Authors: James Zhu <james.zhu<@amd.com> |
| * |
| **************************************************************************/ |
| |
| #include <assert.h> |
| |
| #include "tgsi/tgsi_text.h" |
| #include "vl_compositor_cs.h" |
| |
| struct cs_viewport { |
| float scale_x; |
| float scale_y; |
| struct u_rect area; |
| int translate_x; |
| int translate_y; |
| float sampler0_w; |
| float sampler0_h; |
| }; |
| |
| const char *compute_shader_video_buffer = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..6]\n" |
| "DCL SVIEW[0..2], RECT, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..7]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 0.0, 0.0, 0.0}\n" |
| |
| "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1].xxxx\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[0].xyyy, -CONST[5].xyxy\n" |
| "U2F TEMP[2].xy, TEMP[2].xyyy\n" |
| "MUL TEMP[3].xy, TEMP[2].xyyy, CONST[6].xyyy\n" |
| |
| /* Scale */ |
| "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwww\n" |
| "DIV TEMP[3].xy, TEMP[3].xyyy, CONST[3].zwww\n" |
| |
| /* Fetch texels */ |
| "TEX_LZ TEMP[4].x, TEMP[2].xyyy, SAMP[0], RECT\n" |
| "TEX_LZ TEMP[4].y, TEMP[3].xyyy, SAMP[1], RECT\n" |
| "TEX_LZ TEMP[4].z, TEMP[3].xyyy, SAMP[2], RECT\n" |
| |
| "MOV TEMP[4].w, IMM[1].xxxx\n" |
| |
| /* Color Space Conversion */ |
| "DP4 TEMP[7].x, CONST[0], TEMP[4]\n" |
| "DP4 TEMP[7].y, CONST[1], TEMP[4]\n" |
| "DP4 TEMP[7].z, CONST[2], TEMP[4]\n" |
| |
| "MOV TEMP[5].w, TEMP[4].zzzz\n" |
| "SLE TEMP[6].w, TEMP[5].wwww, CONST[3].xxxx\n" |
| "SGT TEMP[5].w, TEMP[5].wwww, CONST[3].yyyy\n" |
| |
| "MAX TEMP[7].w, TEMP[5].wwww, TEMP[6].wwww\n" |
| |
| "STORE IMAGE[0], TEMP[0].xyyy, TEMP[7], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| const char *compute_shader_weave = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..15]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| "IMM[2] UINT32 { 1, 2, 4, 0}\n" |
| "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n" |
| |
| "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1].xxxx\n" |
| "MOV TEMP[2].xy, TEMP[0].xyyy\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[2].xyyy, -CONST[5].xyxy\n" |
| |
| /* Top Y */ |
| "U2F TEMP[2].xy, TEMP[2].xyyy\n" |
| "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n" |
| /* Down Y */ |
| "MOV TEMP[12].xy, TEMP[2].xyyy\n" |
| |
| /* Top UV */ |
| "MOV TEMP[3].xy, TEMP[2].xyyy\n" |
| "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n" |
| /* Down UV */ |
| "MOV TEMP[13].xy, TEMP[3].xyyy\n" |
| |
| /* Texture offset */ |
| "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n" |
| |
| "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n" |
| "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n" |
| |
| /* Scale */ |
| "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12].xyyy, CONST[3].zwzw\n" |
| "DIV TEMP[3].xy, TEMP[3].xyyy, CONST[3].zwzw\n" |
| "DIV TEMP[13].xy, TEMP[13].xyyy, CONST[3].zwzw\n" |
| |
| /* Weave offset */ |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n" |
| |
| /* Texture layer */ |
| "MOV TEMP[14].x, TEMP[2].yyyy\n" |
| "MOV TEMP[14].yz, TEMP[3].yyyy\n" |
| "ROUND TEMP[15].xyz, TEMP[14].xyzz\n" |
| "ADD TEMP[14].xyz, TEMP[14].xyzz, -TEMP[15].xyzz\n" |
| "MOV TEMP[14].xyz, |TEMP[14].xyzz|\n" |
| "MUL TEMP[14].xyz, TEMP[14].xyzz, IMM[1].yyyy\n" |
| |
| /* Normalize */ |
| "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[5].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12].xyyy, CONST[5].zwzw\n" |
| "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n" |
| "DIV TEMP[3].xy, TEMP[3].xyyy, TEMP[15].xyxy\n" |
| "DIV TEMP[13].xy, TEMP[13].xyyy, TEMP[15].xyxy\n" |
| |
| /* Fetch texels */ |
| "MOV TEMP[2].z, IMM[1].wwww\n" |
| "MOV TEMP[3].z, IMM[1].wwww\n" |
| "TEX_LZ TEMP[10].x, TEMP[2].xyzz, SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].y, TEMP[3].xyzz, SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].z, TEMP[3].xyzz, SAMP[2], 2D_ARRAY\n" |
| |
| "MOV TEMP[12].z, IMM[1].xxxx\n" |
| "MOV TEMP[13].z, IMM[1].xxxx\n" |
| "TEX_LZ TEMP[11].x, TEMP[12].xyzz, SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].y, TEMP[13].xyzz, SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].z, TEMP[13].xyzz, SAMP[2], 2D_ARRAY\n" |
| |
| "LRP TEMP[6].xyz, TEMP[14].xyzz, TEMP[10].xyzz, TEMP[11].xyzz\n" |
| "MOV TEMP[6].w, IMM[1].xxxx\n" |
| |
| /* Color Space Conversion */ |
| "DP4 TEMP[9].x, CONST[0], TEMP[6]\n" |
| "DP4 TEMP[9].y, CONST[1], TEMP[6]\n" |
| "DP4 TEMP[9].z, CONST[2], TEMP[6]\n" |
| |
| "MOV TEMP[7].w, TEMP[6].zzzz\n" |
| "SLE TEMP[8].w, TEMP[7].wwww, CONST[3].xxxx\n" |
| "SGT TEMP[7].w, TEMP[7].wwww, CONST[3].yyyy\n" |
| |
| "MAX TEMP[9].w, TEMP[7].wwww, TEMP[8].wwww\n" |
| |
| "STORE IMAGE[0], TEMP[0].xyyy, TEMP[9], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| const char *compute_shader_rgba = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0], RECT, FLOAT\n" |
| "DCL SAMP[0]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..3]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| |
| "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1].xxxx\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[0].xyyy, -CONST[5].xyxy\n" |
| "U2F TEMP[2].xy, TEMP[2].xyyy\n" |
| |
| /* Scale */ |
| "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwzw\n" |
| |
| /* Fetch texels */ |
| "TEX_LZ TEMP[3], TEMP[2].xyyy, SAMP[0], RECT\n" |
| |
| "STORE IMAGE[0], TEMP[0].xyyy, TEMP[3], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| static const char *compute_shader_yuv_weave_y = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..15]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| "IMM[2] UINT32 { 1, 2, 4, 0}\n" |
| "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n" |
| |
| "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1]\n" |
| "MOV TEMP[2], TEMP[0]\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[2], -CONST[5].xyxy\n" |
| |
| /* Top Y */ |
| "U2F TEMP[2], TEMP[2]\n" |
| "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n" |
| /* Down Y */ |
| "MOV TEMP[12], TEMP[2]\n" |
| |
| /* Top UV */ |
| "MOV TEMP[3], TEMP[2]\n" |
| "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n" |
| /* Down UV */ |
| "MOV TEMP[13], TEMP[3]\n" |
| |
| /* Texture offset */ |
| "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n" |
| |
| "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n" |
| "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n" |
| |
| /* Scale */ |
| "DIV TEMP[2].xy, TEMP[2], CONST[3].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12], CONST[3].zwzw\n" |
| "DIV TEMP[3].xy, TEMP[3], CONST[3].zwzw\n" |
| "DIV TEMP[13].xy, TEMP[13], CONST[3].zwzw\n" |
| |
| /* Weave offset */ |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n" |
| |
| /* Texture layer */ |
| "MOV TEMP[14].x, TEMP[2].yyyy\n" |
| "MOV TEMP[14].yz, TEMP[3].yyyy\n" |
| "ROUND TEMP[15], TEMP[14]\n" |
| "ADD TEMP[14], TEMP[14], -TEMP[15]\n" |
| "MOV TEMP[14], |TEMP[14]|\n" |
| "MUL TEMP[14], TEMP[14], IMM[1].yyyy\n" |
| |
| /* Normalize */ |
| "DIV TEMP[2].xy, TEMP[2], CONST[5].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12], CONST[5].zwzw\n" |
| "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n" |
| "DIV TEMP[3].xy, TEMP[3], TEMP[15].xyxy\n" |
| "DIV TEMP[13].xy, TEMP[13], TEMP[15].xyxy\n" |
| |
| /* Fetch texels */ |
| "MOV TEMP[2].z, IMM[1].wwww\n" |
| "MOV TEMP[3].z, IMM[1].wwww\n" |
| "TEX_LZ TEMP[10].x, TEMP[2], SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].y, TEMP[3], SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].z, TEMP[3], SAMP[2], 2D_ARRAY\n" |
| |
| "MOV TEMP[12].z, IMM[1].xxxx\n" |
| "MOV TEMP[13].z, IMM[1].xxxx\n" |
| "TEX_LZ TEMP[11].x, TEMP[12], SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].y, TEMP[13], SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].z, TEMP[13], SAMP[2], 2D_ARRAY\n" |
| |
| "LRP TEMP[6], TEMP[14], TEMP[10], TEMP[11]\n" |
| "MOV TEMP[6].w, IMM[1].xxxx\n" |
| |
| "STORE IMAGE[0], TEMP[0], TEMP[6], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| static const char *compute_shader_yuv_weave_uv = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..15]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| "IMM[2] UINT32 { 1, 2, 4, 0}\n" |
| "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n" |
| |
| "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1]\n" |
| "MOV TEMP[2], TEMP[0]\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[2], -CONST[5].xyxy\n" |
| |
| /* Top Y */ |
| "U2F TEMP[2], TEMP[2]\n" |
| "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n" |
| /* Down Y */ |
| "MOV TEMP[12], TEMP[2]\n" |
| |
| /* Top UV */ |
| "MOV TEMP[3], TEMP[2]\n" |
| "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n" |
| /* Down UV */ |
| "MOV TEMP[13], TEMP[3]\n" |
| |
| /* Texture offset */ |
| "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n" |
| |
| "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n" |
| "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n" |
| |
| /* Scale */ |
| "DIV TEMP[2].xy, TEMP[2], CONST[3].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12], CONST[3].zwzw\n" |
| "DIV TEMP[3].xy, TEMP[3], CONST[3].zwzw\n" |
| "DIV TEMP[13].xy, TEMP[13], CONST[3].zwzw\n" |
| |
| /* Weave offset */ |
| "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n" |
| "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n" |
| "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n" |
| |
| /* Texture layer */ |
| "MOV TEMP[14].x, TEMP[2].yyyy\n" |
| "MOV TEMP[14].yz, TEMP[3].yyyy\n" |
| "ROUND TEMP[15], TEMP[14]\n" |
| "ADD TEMP[14], TEMP[14], -TEMP[15]\n" |
| "MOV TEMP[14], |TEMP[14]|\n" |
| "MUL TEMP[14], TEMP[14], IMM[1].yyyy\n" |
| |
| /* Normalize */ |
| "DIV TEMP[2].xy, TEMP[2], CONST[5].zwzw\n" |
| "DIV TEMP[12].xy, TEMP[12], CONST[5].zwzw\n" |
| "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n" |
| "DIV TEMP[3].xy, TEMP[3], TEMP[15].xyxy\n" |
| "DIV TEMP[13].xy, TEMP[13], TEMP[15].xyxy\n" |
| |
| /* Fetch texels */ |
| "MOV TEMP[2].z, IMM[1].wwww\n" |
| "MOV TEMP[3].z, IMM[1].wwww\n" |
| "TEX_LZ TEMP[10].x, TEMP[2], SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].y, TEMP[3], SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[10].z, TEMP[3], SAMP[2], 2D_ARRAY\n" |
| |
| "MOV TEMP[12].z, IMM[1].xxxx\n" |
| "MOV TEMP[13].z, IMM[1].xxxx\n" |
| "TEX_LZ TEMP[11].x, TEMP[12], SAMP[0], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].y, TEMP[13], SAMP[1], 2D_ARRAY\n" |
| "TEX_LZ TEMP[11].z, TEMP[13], SAMP[2], 2D_ARRAY\n" |
| |
| "LRP TEMP[6], TEMP[14], TEMP[10], TEMP[11]\n" |
| "MOV TEMP[6].w, IMM[1].xxxx\n" |
| |
| "MOV TEMP[7].xy, TEMP[6].yzww\n" |
| |
| "STORE IMAGE[0], TEMP[0], TEMP[7], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| static const char *compute_shader_yuv_bob_y = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0..2], RECT, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..4]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| |
| "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1]\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[0], -CONST[5].xyxy\n" |
| "U2F TEMP[2], TEMP[2]\n" |
| "DIV TEMP[3], TEMP[2], IMM[1].yyyy\n" |
| |
| /* Scale */ |
| "DIV TEMP[2], TEMP[2], CONST[3].zwzw\n" |
| "DIV TEMP[2], TEMP[2], IMM[1].xyxy\n" |
| "DIV TEMP[3], TEMP[3], CONST[3].zwzw\n" |
| "DIV TEMP[3], TEMP[3], IMM[1].xyxy\n" |
| |
| /* Fetch texels */ |
| "TEX_LZ TEMP[4].x, TEMP[2], SAMP[0], RECT\n" |
| "TEX_LZ TEMP[4].y, TEMP[3], SAMP[1], RECT\n" |
| "TEX_LZ TEMP[4].z, TEMP[3], SAMP[2], RECT\n" |
| |
| "MOV TEMP[4].w, IMM[1].xxxx\n" |
| |
| "STORE IMAGE[0], TEMP[0], TEMP[4], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| static const char *compute_shader_yuv_bob_uv = |
| "COMP\n" |
| "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n" |
| "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n" |
| "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n" |
| |
| "DCL SV[0], THREAD_ID\n" |
| "DCL SV[1], BLOCK_ID\n" |
| |
| "DCL CONST[0..5]\n" |
| "DCL SVIEW[0..2], RECT, FLOAT\n" |
| "DCL SAMP[0..2]\n" |
| |
| "DCL IMAGE[0], 2D, WR\n" |
| "DCL TEMP[0..5]\n" |
| |
| "IMM[0] UINT32 { 8, 8, 1, 0}\n" |
| "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n" |
| |
| "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n" |
| |
| /* Drawn area check */ |
| "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n" |
| "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n" |
| "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n" |
| |
| "UIF TEMP[1]\n" |
| /* Translate */ |
| "UADD TEMP[2].xy, TEMP[0], -CONST[5].xyxy\n" |
| "U2F TEMP[2], TEMP[2]\n" |
| "DIV TEMP[3], TEMP[2], IMM[1].yyyy\n" |
| |
| /* Scale */ |
| "DIV TEMP[2], TEMP[2], CONST[3].zwzw\n" |
| "DIV TEMP[2], TEMP[2], IMM[1].xyxy\n" |
| "DIV TEMP[3], TEMP[3], CONST[3].zwzw\n" |
| "DIV TEMP[3], TEMP[3], IMM[1].xyxy\n" |
| |
| /* Fetch texels */ |
| "TEX_LZ TEMP[4].x, TEMP[2], SAMP[0], RECT\n" |
| "TEX_LZ TEMP[4].y, TEMP[3], SAMP[1], RECT\n" |
| "TEX_LZ TEMP[4].z, TEMP[3], SAMP[2], RECT\n" |
| |
| "MOV TEMP[4].w, IMM[1].xxxx\n" |
| |
| "MOV TEMP[5].xy, TEMP[4].yzww\n" |
| |
| "STORE IMAGE[0], TEMP[0], TEMP[5], 2D\n" |
| "ENDIF\n" |
| |
| "END\n"; |
| |
| static void |
| cs_launch(struct vl_compositor *c, |
| void *cs, |
| const struct u_rect *draw_area) |
| { |
| struct pipe_context *ctx = c->pipe; |
| |
| /* Bind the image */ |
| struct pipe_image_view image = {0}; |
| image.resource = c->fb_state.cbufs[0]->texture; |
| image.shader_access = image.access = PIPE_IMAGE_ACCESS_READ_WRITE; |
| image.format = c->fb_state.cbufs[0]->texture->format; |
| |
| ctx->set_shader_images(c->pipe, PIPE_SHADER_COMPUTE, 0, 1, &image); |
| |
| /* Bind compute shader */ |
| ctx->bind_compute_state(ctx, cs); |
| |
| /* Dispatch compute */ |
| struct pipe_grid_info info = {0}; |
| info.block[0] = 8; |
| info.block[1] = 8; |
| info.block[2] = 1; |
| info.grid[0] = DIV_ROUND_UP(draw_area->x1, info.block[0]); |
| info.grid[1] = DIV_ROUND_UP(draw_area->y1, info.block[1]); |
| info.grid[2] = 1; |
| |
| ctx->launch_grid(ctx, &info); |
| |
| /* Make the result visible to all clients. */ |
| ctx->memory_barrier(ctx, PIPE_BARRIER_ALL); |
| |
| } |
| |
| static inline struct u_rect |
| calc_drawn_area(struct vl_compositor_state *s, |
| struct vl_compositor_layer *layer) |
| { |
| struct vertex2f tl, br; |
| struct u_rect result; |
| |
| assert(s && layer); |
| |
| tl = layer->dst.tl; |
| br = layer->dst.br; |
| |
| /* Scale */ |
| result.x0 = tl.x * layer->viewport.scale[0] + layer->viewport.translate[0]; |
| result.y0 = tl.y * layer->viewport.scale[1] + layer->viewport.translate[1]; |
| result.x1 = br.x * layer->viewport.scale[0] + layer->viewport.translate[0]; |
| result.y1 = br.y * layer->viewport.scale[1] + layer->viewport.translate[1]; |
| |
| /* Clip */ |
| result.x0 = MAX2(result.x0, s->scissor.minx); |
| result.y0 = MAX2(result.y0, s->scissor.miny); |
| result.x1 = MIN2(result.x1, s->scissor.maxx); |
| result.y1 = MIN2(result.y1, s->scissor.maxy); |
| return result; |
| } |
| |
| static bool |
| set_viewport(struct vl_compositor_state *s, |
| struct cs_viewport *drawn, |
| struct pipe_sampler_view **samplers) |
| { |
| struct pipe_transfer *buf_transfer; |
| |
| assert(s && drawn); |
| |
| void *ptr = pipe_buffer_map(s->pipe, s->shader_params, |
| PIPE_MAP_READ | PIPE_MAP_WRITE, |
| &buf_transfer); |
| |
| if (!ptr) |
| return false; |
| |
| float *ptr_float = (float *)ptr; |
| ptr_float += sizeof(vl_csc_matrix)/sizeof(float) + 2; |
| *ptr_float++ = drawn->scale_x; |
| *ptr_float++ = drawn->scale_y; |
| |
| int *ptr_int = (int *)ptr_float; |
| *ptr_int++ = drawn->area.x0; |
| *ptr_int++ = drawn->area.y0; |
| *ptr_int++ = drawn->area.x1; |
| *ptr_int++ = drawn->area.y1; |
| *ptr_int++ = drawn->translate_x; |
| *ptr_int++ = drawn->translate_y; |
| |
| ptr_float = (float *)ptr_int; |
| *ptr_float++ = drawn->sampler0_w; |
| *ptr_float++ = drawn->sampler0_h; |
| |
| /* compute_shader_video_buffer uses pixel coordinates based on the |
| * Y sampler dimensions. If U/V are using separate planes and are |
| * subsampled, we need to scale the coordinates */ |
| if (samplers[1]) { |
| float h_ratio = samplers[1]->texture->width0 / |
| (float) samplers[0]->texture->width0; |
| *ptr_float++ = h_ratio; |
| float v_ratio = samplers[1]->texture->height0 / |
| (float) samplers[0]->texture->height0; |
| *ptr_float++ = v_ratio; |
| } |
| pipe_buffer_unmap(s->pipe, buf_transfer); |
| |
| return true; |
| } |
| |
| static void |
| draw_layers(struct vl_compositor *c, |
| struct vl_compositor_state *s, |
| struct u_rect *dirty) |
| { |
| unsigned i; |
| |
| assert(c); |
| |
| for (i = 0; i < VL_COMPOSITOR_MAX_LAYERS; ++i) { |
| if (s->used_layers & (1 << i)) { |
| struct vl_compositor_layer *layer = &s->layers[i]; |
| struct pipe_sampler_view **samplers = &layer->sampler_views[0]; |
| unsigned num_sampler_views = !samplers[1] ? 1 : !samplers[2] ? 2 : 3; |
| struct cs_viewport drawn; |
| |
| drawn.area = calc_drawn_area(s, layer); |
| drawn.scale_x = layer->viewport.scale[0] / |
| (float)layer->sampler_views[0]->texture->width0 * |
| (layer->src.br.x - layer->src.tl.x); |
| drawn.scale_y = layer->viewport.scale[1] / |
| ((float)layer->sampler_views[0]->texture->height0 * |
| (s->interlaced ? 2.0 : 1.0) * |
| (layer->src.br.y - layer->src.tl.y)); |
| |
| drawn.translate_x = (int)layer->viewport.translate[0]; |
| drawn.translate_y = (int)layer->viewport.translate[1]; |
| drawn.sampler0_w = (float)layer->sampler_views[0]->texture->width0; |
| drawn.sampler0_h = (float)layer->sampler_views[0]->texture->height0; |
| set_viewport(s, &drawn, samplers); |
| |
| c->pipe->bind_sampler_states(c->pipe, PIPE_SHADER_COMPUTE, 0, |
| num_sampler_views, layer->samplers); |
| c->pipe->set_sampler_views(c->pipe, PIPE_SHADER_COMPUTE, 0, |
| num_sampler_views, samplers); |
| |
| cs_launch(c, layer->cs, &(drawn.area)); |
| |
| /* Unbind. */ |
| c->pipe->set_shader_images(c->pipe, PIPE_SHADER_COMPUTE, 0, 1, NULL); |
| c->pipe->set_constant_buffer(c->pipe, PIPE_SHADER_COMPUTE, 0, NULL); |
| c->pipe->set_sampler_views(c->pipe, PIPE_SHADER_FRAGMENT, 0, |
| num_sampler_views, NULL); |
| c->pipe->bind_compute_state(c->pipe, NULL); |
| c->pipe->bind_sampler_states(c->pipe, PIPE_SHADER_COMPUTE, 0, |
| num_sampler_views, NULL); |
| |
| if (dirty) { |
| struct u_rect drawn = calc_drawn_area(s, layer); |
| dirty->x0 = MIN2(drawn.x0, dirty->x0); |
| dirty->y0 = MIN2(drawn.y0, dirty->y0); |
| dirty->x1 = MAX2(drawn.x1, dirty->x1); |
| dirty->y1 = MAX2(drawn.y1, dirty->y1); |
| } |
| } |
| } |
| } |
| |
| void * |
| vl_compositor_cs_create_shader(struct vl_compositor *c, |
| const char *compute_shader_text) |
| { |
| assert(c && compute_shader_text); |
| |
| struct tgsi_token tokens[1024]; |
| if (!tgsi_text_translate(compute_shader_text, tokens, ARRAY_SIZE(tokens))) { |
| assert(0); |
| return NULL; |
| } |
| |
| struct pipe_compute_state state = {0}; |
| state.ir_type = PIPE_SHADER_IR_TGSI; |
| state.prog = tokens; |
| |
| /* create compute shader */ |
| return c->pipe->create_compute_state(c->pipe, &state); |
| } |
| |
| void |
| vl_compositor_cs_render(struct vl_compositor_state *s, |
| struct vl_compositor *c, |
| struct pipe_surface *dst_surface, |
| struct u_rect *dirty_area, |
| bool clear_dirty) |
| { |
| assert(c && s); |
| assert(dst_surface); |
| |
| c->fb_state.width = dst_surface->width; |
| c->fb_state.height = dst_surface->height; |
| c->fb_state.cbufs[0] = dst_surface; |
| |
| if (!s->scissor_valid) { |
| s->scissor.minx = 0; |
| s->scissor.miny = 0; |
| s->scissor.maxx = dst_surface->width; |
| s->scissor.maxy = dst_surface->height; |
| } |
| |
| if (clear_dirty && dirty_area && |
| (dirty_area->x0 < dirty_area->x1 || dirty_area->y0 < dirty_area->y1)) { |
| |
| c->pipe->clear_render_target(c->pipe, dst_surface, &s->clear_color, |
| 0, 0, dst_surface->width, dst_surface->height, false); |
| dirty_area->x0 = dirty_area->y0 = VL_COMPOSITOR_MAX_DIRTY; |
| dirty_area->x1 = dirty_area->y1 = VL_COMPOSITOR_MIN_DIRTY; |
| } |
| |
| pipe_set_constant_buffer(c->pipe, PIPE_SHADER_COMPUTE, 0, s->shader_params); |
| |
| draw_layers(c, s, dirty_area); |
| } |
| |
| bool vl_compositor_cs_init_shaders(struct vl_compositor *c) |
| { |
| assert(c); |
| |
| c->cs_video_buffer = vl_compositor_cs_create_shader(c, compute_shader_video_buffer); |
| if (!c->cs_video_buffer) { |
| debug_printf("Unable to create video_buffer compute shader.\n"); |
| return false; |
| } |
| |
| c->cs_weave_rgb = vl_compositor_cs_create_shader(c, compute_shader_weave); |
| if (!c->cs_weave_rgb) { |
| debug_printf("Unable to create weave_rgb compute shader.\n"); |
| return false; |
| } |
| |
| c->cs_yuv.weave.y = vl_compositor_cs_create_shader(c, compute_shader_yuv_weave_y); |
| c->cs_yuv.weave.uv = vl_compositor_cs_create_shader(c, compute_shader_yuv_weave_uv); |
| c->cs_yuv.bob.y = vl_compositor_cs_create_shader(c, compute_shader_yuv_bob_y); |
| c->cs_yuv.bob.uv = vl_compositor_cs_create_shader(c, compute_shader_yuv_bob_uv); |
| if (!c->cs_yuv.weave.y || !c->cs_yuv.weave.uv || |
| !c->cs_yuv.bob.y || !c->cs_yuv.bob.uv) { |
| debug_printf("Unable to create YCbCr i-to-YCbCr p deint compute shader.\n"); |
| return false; |
| } |
| |
| return true; |
| } |
| |
| void vl_compositor_cs_cleanup_shaders(struct vl_compositor *c) |
| { |
| assert(c); |
| |
| if (c->cs_video_buffer) |
| c->pipe->delete_compute_state(c->pipe, c->cs_video_buffer); |
| if (c->cs_weave_rgb) |
| c->pipe->delete_compute_state(c->pipe, c->cs_weave_rgb); |
| if (c->cs_yuv.weave.y) |
| c->pipe->delete_compute_state(c->pipe, c->cs_yuv.weave.y); |
| if (c->cs_yuv.weave.uv) |
| c->pipe->delete_compute_state(c->pipe, c->cs_yuv.weave.uv); |
| if (c->cs_yuv.bob.y) |
| c->pipe->delete_compute_state(c->pipe, c->cs_yuv.bob.y); |
| if (c->cs_yuv.bob.uv) |
| c->pipe->delete_compute_state(c->pipe, c->cs_yuv.bob.uv); |
| } |
