src/gallium/auxiliary/vl/vl_compositor_cs.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2019 Advanced Micro Devices, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  * Authors: James Zhu <james.zhu<@amd.com>
  27  *
  28  **************************************************************************/
  29
  30 #include <assert.h>
  31
  32 #include "tgsi/tgsi_text.h"
  33 #include "vl_compositor_cs.h"
  34
  35 struct cs_viewport {
  36    float scale_x;
  37    float scale_y;
  38    struct u_rect area;
  39    int translate_x;
  40    int translate_y;
  41    float sampler0_w;
  42    float sampler0_h;
  43 };
  44
  45 const char *compute_shader_video_buffer =
  46       "COMP\n"
  47       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
  48       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
  49       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
  50
  51       "DCL SV[0], THREAD_ID\n"
  52       "DCL SV[1], BLOCK_ID\n"
  53
  54       "DCL CONST[0..5]\n"
  55       "DCL SVIEW[0..2], RECT, FLOAT\n"
  56       "DCL SAMP[0..2]\n"
  57
  58       "DCL IMAGE[0], 2D, WR\n"
  59       "DCL TEMP[0..7]\n"
  60
  61       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
  62       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
  63
  64       "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n"
  65
  66       /* Drawn area check */
  67       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
  68       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
  69       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
  70       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
  71       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
  72
  73       "UIF TEMP[1].xxxx\n"
  74          /* Translate */
  75          "UADD TEMP[2].xy, TEMP[0].xyyy, -CONST[5].xyxy\n"
  76          "U2F TEMP[2].xy, TEMP[2].xyyy\n"
  77          "DIV TEMP[3].xy, TEMP[2].xyyy, IMM[1].yyyy\n"
  78
  79          /* Scale */
  80          "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwww\n"
  81          "DIV TEMP[3].xy, TEMP[3].xyyy, CONST[3].zwww\n"
  82
  83          /* Fetch texels */
  84          "TEX_LZ TEMP[4].x, TEMP[2].xyyy, SAMP[0], RECT\n"
  85          "TEX_LZ TEMP[4].y, TEMP[3].xyyy, SAMP[1], RECT\n"
  86          "TEX_LZ TEMP[4].z, TEMP[3].xyyy, SAMP[2], RECT\n"
  87
  88          "MOV TEMP[4].w, IMM[1].xxxx\n"
  89
  90          /* Color Space Conversion */
  91          "DP4 TEMP[7].x, CONST[0], TEMP[4]\n"
  92          "DP4 TEMP[7].y, CONST[1], TEMP[4]\n"
  93          "DP4 TEMP[7].z, CONST[2], TEMP[4]\n"
  94
  95          "MOV TEMP[5].w, TEMP[4].zzzz\n"
  96          "SLE TEMP[6].w, TEMP[5].wwww, CONST[3].xxxx\n"
  97          "SGT TEMP[5].w, TEMP[5].wwww, CONST[3].yyyy\n"
  98
  99          "MAX TEMP[7].w, TEMP[5].wwww, TEMP[6].wwww\n"
 100
 101          "STORE IMAGE[0], TEMP[0].xyyy, TEMP[7], 2D\n"
 102       "ENDIF\n"
 103
 104       "END\n";
 105
 106 const char *compute_shader_weave =
 107       "COMP\n"
 108       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 109       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 110       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 111
 112       "DCL SV[0], THREAD_ID\n"
 113       "DCL SV[1], BLOCK_ID\n"
 114
 115       "DCL CONST[0..5]\n"
 116       "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n"
 117       "DCL SAMP[0..2]\n"
 118
 119       "DCL IMAGE[0], 2D, WR\n"
 120       "DCL TEMP[0..15]\n"
 121
 122       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 123       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 124       "IMM[2] UINT32 { 1, 2, 4, 0}\n"
 125       "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n"
 126
 127       "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n"
 128
 129       /* Drawn area check */
 130       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 131       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 132       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 133       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 134       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 135
 136       "UIF TEMP[1].xxxx\n"
 137          "MOV TEMP[2].xy, TEMP[0].xyyy\n"
 138          /* Translate */
 139          "UADD TEMP[2].xy, TEMP[2].xyyy, -CONST[5].xyxy\n"
 140
 141          /* Top Y */
 142          "U2F TEMP[2].xy, TEMP[2].xyyy\n"
 143          "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n"
 144          /* Down Y */
 145          "MOV TEMP[12].xy, TEMP[2].xyyy\n"
 146
 147          /* Top UV */
 148          "MOV TEMP[3].xy, TEMP[2].xyyy\n"
 149          "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n"
 150          /* Down UV */
 151          "MOV TEMP[13].xy, TEMP[3].xyyy\n"
 152
 153          /* Texture offset */
 154          "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n"
 155          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 156          "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n"
 157          "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n"
 158
 159          "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n"
 160          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n"
 161          "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n"
 162          "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n"
 163
 164          /* Scale */
 165          "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwzw\n"
 166          "DIV TEMP[12].xy, TEMP[12].xyyy, CONST[3].zwzw\n"
 167          "DIV TEMP[3].xy, TEMP[3].xyyy, CONST[3].zwzw\n"
 168          "DIV TEMP[13].xy, TEMP[13].xyyy, CONST[3].zwzw\n"
 169
 170          /* Weave offset */
 171          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 172          "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n"
 173          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n"
 174          "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n"
 175
 176          /* Texture layer */
 177          "MOV TEMP[14].x, TEMP[2].yyyy\n"
 178          "MOV TEMP[14].yz, TEMP[3].yyyy\n"
 179          "ROUND TEMP[15].xyz, TEMP[14].xyzz\n"
 180          "ADD TEMP[14].xyz, TEMP[14].xyzz, -TEMP[15].xyzz\n"
 181          "MOV TEMP[14].xyz, |TEMP[14].xyzz|\n"
 182          "MUL TEMP[14].xyz, TEMP[14].xyzz, IMM[1].yyyy\n"
 183
 184          /* Normalize */
 185          "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[5].zwzw\n"
 186          "DIV TEMP[12].xy, TEMP[12].xyyy, CONST[5].zwzw\n"
 187          "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n"
 188          "DIV TEMP[3].xy, TEMP[3].xyyy, TEMP[15].xyxy\n"
 189          "DIV TEMP[13].xy, TEMP[13].xyyy, TEMP[15].xyxy\n"
 190
 191          /* Fetch texels */
 192          "MOV TEMP[2].z, IMM[1].wwww\n"
 193          "MOV TEMP[3].z, IMM[1].wwww\n"
 194          "TEX_LZ TEMP[10].x, TEMP[2].xyzz, SAMP[0], 2D_ARRAY\n"
 195          "TEX_LZ TEMP[10].y, TEMP[3].xyzz, SAMP[1], 2D_ARRAY\n"
 196          "TEX_LZ TEMP[10].z, TEMP[3].xyzz, SAMP[2], 2D_ARRAY\n"
 197
 198          "MOV TEMP[12].z, IMM[1].xxxx\n"
 199          "MOV TEMP[13].z, IMM[1].xxxx\n"
 200          "TEX_LZ TEMP[11].x, TEMP[12].xyzz, SAMP[0], 2D_ARRAY\n"
 201          "TEX_LZ TEMP[11].y, TEMP[13].xyzz, SAMP[1], 2D_ARRAY\n"
 202          "TEX_LZ TEMP[11].z, TEMP[13].xyzz, SAMP[2], 2D_ARRAY\n"
 203
 204          "LRP TEMP[6].xyz, TEMP[14].xyzz, TEMP[10].xyzz, TEMP[11].xyzz\n"
 205          "MOV TEMP[6].w, IMM[1].xxxx\n"
 206
 207          /* Color Space Conversion */
 208          "DP4 TEMP[9].x, CONST[0], TEMP[6]\n"
 209          "DP4 TEMP[9].y, CONST[1], TEMP[6]\n"
 210          "DP4 TEMP[9].z, CONST[2], TEMP[6]\n"
 211
 212          "MOV TEMP[7].w, TEMP[6].zzzz\n"
 213          "SLE TEMP[8].w, TEMP[7].wwww, CONST[3].xxxx\n"
 214          "SGT TEMP[7].w, TEMP[7].wwww, CONST[3].yyyy\n"
 215
 216          "MAX TEMP[9].w, TEMP[7].wwww, TEMP[8].wwww\n"
 217
 218          "STORE IMAGE[0], TEMP[0].xyyy, TEMP[9], 2D\n"
 219       "ENDIF\n"
 220
 221       "END\n";
 222
 223 const char *compute_shader_rgba =
 224       "COMP\n"
 225       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 226       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 227       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 228
 229       "DCL SV[0], THREAD_ID\n"
 230       "DCL SV[1], BLOCK_ID\n"
 231
 232       "DCL CONST[0..5]\n"
 233       "DCL SVIEW[0], RECT, FLOAT\n"
 234       "DCL SAMP[0]\n"
 235
 236       "DCL IMAGE[0], 2D, WR\n"
 237       "DCL TEMP[0..3]\n"
 238
 239       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 240       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 241
 242       "UMAD TEMP[0].xy, SV[1].xyyy, IMM[0].xyyy, SV[0].xyyy\n"
 243
 244       /* Drawn area check */
 245       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 246       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 247       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 248       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 249       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 250
 251       "UIF TEMP[1].xxxx\n"
 252          /* Translate */
 253          "UADD TEMP[2].xy, TEMP[0].xyyy, -CONST[5].xyxy\n"
 254          "U2F TEMP[2].xy, TEMP[2].xyyy\n"
 255
 256          /* Scale */
 257          "DIV TEMP[2].xy, TEMP[2].xyyy, CONST[3].zwzw\n"
 258
 259          /* Fetch texels */
 260          "TEX_LZ TEMP[3], TEMP[2].xyyy, SAMP[0], RECT\n"
 261
 262          "STORE IMAGE[0], TEMP[0].xyyy, TEMP[3], 2D\n"
 263       "ENDIF\n"
 264
 265       "END\n";
 266
 267 static const char *compute_shader_yuv_weave_y =
 268       "COMP\n"
 269       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 270       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 271       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 272
 273       "DCL SV[0], THREAD_ID\n"
 274       "DCL SV[1], BLOCK_ID\n"
 275
 276       "DCL CONST[0..5]\n"
 277       "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n"
 278       "DCL SAMP[0..2]\n"
 279
 280       "DCL IMAGE[0], 2D, WR\n"
 281       "DCL TEMP[0..15]\n"
 282
 283       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 284       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 285       "IMM[2] UINT32 { 1, 2, 4, 0}\n"
 286       "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n"
 287
 288       "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n"
 289
 290       /* Drawn area check */
 291       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 292       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 293       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 294       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 295       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 296
 297       "UIF TEMP[1]\n"
 298          "MOV TEMP[2], TEMP[0]\n"
 299          /* Translate */
 300          "UADD TEMP[2].xy, TEMP[2], -CONST[5].xyxy\n"
 301
 302          /* Top Y */
 303          "U2F TEMP[2], TEMP[2]\n"
 304          "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n"
 305          /* Down Y */
 306          "MOV TEMP[12], TEMP[2]\n"
 307
 308          /* Top UV */
 309          "MOV TEMP[3], TEMP[2]\n"
 310          "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n"
 311          /* Down UV */
 312          "MOV TEMP[13], TEMP[3]\n"
 313
 314          /* Texture offset */
 315          "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n"
 316          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 317          "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n"
 318          "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n"
 319
 320          "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n"
 321          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n"
 322          "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n"
 323          "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n"
 324
 325          /* Scale */
 326          "DIV TEMP[2].xy, TEMP[2], CONST[3].zwzw\n"
 327          "DIV TEMP[12].xy, TEMP[12], CONST[3].zwzw\n"
 328          "DIV TEMP[3].xy, TEMP[3], CONST[3].zwzw\n"
 329          "DIV TEMP[13].xy, TEMP[13], CONST[3].zwzw\n"
 330
 331          /* Weave offset */
 332          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 333          "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n"
 334          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n"
 335          "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n"
 336
 337          /* Texture layer */
 338          "MOV TEMP[14].x, TEMP[2].yyyy\n"
 339          "MOV TEMP[14].yz, TEMP[3].yyyy\n"
 340          "ROUND TEMP[15], TEMP[14]\n"
 341          "ADD TEMP[14], TEMP[14], -TEMP[15]\n"
 342          "MOV TEMP[14], |TEMP[14]|\n"
 343          "MUL TEMP[14], TEMP[14], IMM[1].yyyy\n"
 344
 345          /* Normalize */
 346          "DIV TEMP[2].xy, TEMP[2], CONST[5].zwzw\n"
 347          "DIV TEMP[12].xy, TEMP[12], CONST[5].zwzw\n"
 348          "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n"
 349          "DIV TEMP[3].xy, TEMP[3], TEMP[15].xyxy\n"
 350          "DIV TEMP[13].xy, TEMP[13], TEMP[15].xyxy\n"
 351
 352          /* Fetch texels */
 353          "MOV TEMP[2].z, IMM[1].wwww\n"
 354          "MOV TEMP[3].z, IMM[1].wwww\n"
 355          "TEX_LZ TEMP[10].x, TEMP[2], SAMP[0], 2D_ARRAY\n"
 356          "TEX_LZ TEMP[10].y, TEMP[3], SAMP[1], 2D_ARRAY\n"
 357          "TEX_LZ TEMP[10].z, TEMP[3], SAMP[2], 2D_ARRAY\n"
 358
 359          "MOV TEMP[12].z, IMM[1].xxxx\n"
 360          "MOV TEMP[13].z, IMM[1].xxxx\n"
 361          "TEX_LZ TEMP[11].x, TEMP[12], SAMP[0], 2D_ARRAY\n"
 362          "TEX_LZ TEMP[11].y, TEMP[13], SAMP[1], 2D_ARRAY\n"
 363          "TEX_LZ TEMP[11].z, TEMP[13], SAMP[2], 2D_ARRAY\n"
 364
 365          "LRP TEMP[6], TEMP[14], TEMP[10], TEMP[11]\n"
 366          "MOV TEMP[6].w, IMM[1].xxxx\n"
 367
 368          "STORE IMAGE[0], TEMP[0], TEMP[6], 2D\n"
 369       "ENDIF\n"
 370
 371       "END\n";
 372
 373 static const char *compute_shader_yuv_weave_uv =
 374       "COMP\n"
 375       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 376       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 377       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 378
 379       "DCL SV[0], THREAD_ID\n"
 380       "DCL SV[1], BLOCK_ID\n"
 381
 382       "DCL CONST[0..5]\n"
 383       "DCL SVIEW[0..2], 2D_ARRAY, FLOAT\n"
 384       "DCL SAMP[0..2]\n"
 385
 386       "DCL IMAGE[0], 2D, WR\n"
 387       "DCL TEMP[0..15]\n"
 388
 389       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 390       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 391       "IMM[2] UINT32 { 1, 2, 4, 0}\n"
 392       "IMM[3] FLT32 { 0.25, 0.5, 0.125, 0.125}\n"
 393
 394       "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n"
 395
 396       /* Drawn area check */
 397       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 398       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 399       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 400       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 401       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 402
 403       "UIF TEMP[1]\n"
 404          "MOV TEMP[2], TEMP[0]\n"
 405          /* Translate */
 406          "UADD TEMP[2].xy, TEMP[2], -CONST[5].xyxy\n"
 407
 408          /* Top Y */
 409          "U2F TEMP[2], TEMP[2]\n"
 410          "DIV TEMP[2].y, TEMP[2].yyyy, IMM[1].yyyy\n"
 411          /* Down Y */
 412          "MOV TEMP[12], TEMP[2]\n"
 413
 414          /* Top UV */
 415          "MOV TEMP[3], TEMP[2]\n"
 416          "DIV TEMP[3].xy, TEMP[3], IMM[1].yyyy\n"
 417          /* Down UV */
 418          "MOV TEMP[13], TEMP[3]\n"
 419
 420          /* Texture offset */
 421          "ADD TEMP[2].x, TEMP[2].xxxx, IMM[3].yyyy\n"
 422          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 423          "ADD TEMP[12].x, TEMP[12].xxxx, IMM[3].yyyy\n"
 424          "ADD TEMP[12].y, TEMP[12].yyyy, IMM[3].xxxx\n"
 425
 426          "ADD TEMP[3].x, TEMP[3].xxxx, IMM[3].xxxx\n"
 427          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].wwww\n"
 428          "ADD TEMP[13].x, TEMP[13].xxxx, IMM[3].xxxx\n"
 429          "ADD TEMP[13].y, TEMP[13].yyyy, IMM[3].wwww\n"
 430
 431          /* Scale */
 432          "DIV TEMP[2].xy, TEMP[2], CONST[3].zwzw\n"
 433          "DIV TEMP[12].xy, TEMP[12], CONST[3].zwzw\n"
 434          "DIV TEMP[3].xy, TEMP[3], CONST[3].zwzw\n"
 435          "DIV TEMP[13].xy, TEMP[13], CONST[3].zwzw\n"
 436
 437          /* Weave offset */
 438          "ADD TEMP[2].y, TEMP[2].yyyy, IMM[3].xxxx\n"
 439          "ADD TEMP[12].y, TEMP[12].yyyy, -IMM[3].xxxx\n"
 440          "ADD TEMP[3].y, TEMP[3].yyyy, IMM[3].xxxx\n"
 441          "ADD TEMP[13].y, TEMP[13].yyyy, -IMM[3].xxxx\n"
 442
 443          /* Texture layer */
 444          "MOV TEMP[14].x, TEMP[2].yyyy\n"
 445          "MOV TEMP[14].yz, TEMP[3].yyyy\n"
 446          "ROUND TEMP[15], TEMP[14]\n"
 447          "ADD TEMP[14], TEMP[14], -TEMP[15]\n"
 448          "MOV TEMP[14], |TEMP[14]|\n"
 449          "MUL TEMP[14], TEMP[14], IMM[1].yyyy\n"
 450
 451          /* Normalize */
 452          "DIV TEMP[2].xy, TEMP[2], CONST[5].zwzw\n"
 453          "DIV TEMP[12].xy, TEMP[12], CONST[5].zwzw\n"
 454          "DIV TEMP[15].xy, CONST[5].zwzw, IMM[1].yyyy\n"
 455          "DIV TEMP[3].xy, TEMP[3], TEMP[15].xyxy\n"
 456          "DIV TEMP[13].xy, TEMP[13], TEMP[15].xyxy\n"
 457
 458          /* Fetch texels */
 459          "MOV TEMP[2].z, IMM[1].wwww\n"
 460          "MOV TEMP[3].z, IMM[1].wwww\n"
 461          "TEX_LZ TEMP[10].x, TEMP[2], SAMP[0], 2D_ARRAY\n"
 462          "TEX_LZ TEMP[10].y, TEMP[3], SAMP[1], 2D_ARRAY\n"
 463          "TEX_LZ TEMP[10].z, TEMP[3], SAMP[2], 2D_ARRAY\n"
 464
 465          "MOV TEMP[12].z, IMM[1].xxxx\n"
 466          "MOV TEMP[13].z, IMM[1].xxxx\n"
 467          "TEX_LZ TEMP[11].x, TEMP[12], SAMP[0], 2D_ARRAY\n"
 468          "TEX_LZ TEMP[11].y, TEMP[13], SAMP[1], 2D_ARRAY\n"
 469          "TEX_LZ TEMP[11].z, TEMP[13], SAMP[2], 2D_ARRAY\n"
 470
 471          "LRP TEMP[6], TEMP[14], TEMP[10], TEMP[11]\n"
 472          "MOV TEMP[6].w, IMM[1].xxxx\n"
 473
 474          "MOV TEMP[7].xy, TEMP[6].yzww\n"
 475
 476          "STORE IMAGE[0], TEMP[0], TEMP[7], 2D\n"
 477       "ENDIF\n"
 478
 479       "END\n";
 480
 481 static const char *compute_shader_yuv_bob_y =
 482       "COMP\n"
 483       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 484       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 485       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 486
 487       "DCL SV[0], THREAD_ID\n"
 488       "DCL SV[1], BLOCK_ID\n"
 489
 490       "DCL CONST[0..5]\n"
 491       "DCL SVIEW[0..2], RECT, FLOAT\n"
 492       "DCL SAMP[0..2]\n"
 493
 494       "DCL IMAGE[0], 2D, WR\n"
 495       "DCL TEMP[0..4]\n"
 496
 497       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 498       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 499
 500       "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n"
 501
 502       /* Drawn area check */
 503       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 504       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 505       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 506       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 507       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 508
 509       "UIF TEMP[1]\n"
 510          /* Translate */
 511          "UADD TEMP[2].xy, TEMP[0], -CONST[5].xyxy\n"
 512          "U2F TEMP[2], TEMP[2]\n"
 513          "DIV TEMP[3], TEMP[2], IMM[1].yyyy\n"
 514
 515          /* Scale */
 516          "DIV TEMP[2], TEMP[2], CONST[3].zwzw\n"
 517          "DIV TEMP[3], TEMP[3], CONST[3].zwzw\n"
 518
 519          /* Fetch texels */
 520          "TEX_LZ TEMP[4].x, TEMP[2], SAMP[0], RECT\n"
 521          "TEX_LZ TEMP[4].y, TEMP[3], SAMP[1], RECT\n"
 522          "TEX_LZ TEMP[4].z, TEMP[3], SAMP[2], RECT\n"
 523
 524          "MOV TEMP[4].w, IMM[1].xxxx\n"
 525
 526          "STORE IMAGE[0], TEMP[0], TEMP[4], 2D\n"
 527       "ENDIF\n"
 528
 529       "END\n";
 530
 531 static const char *compute_shader_yuv_bob_uv =
 532       "COMP\n"
 533       "PROPERTY CS_FIXED_BLOCK_WIDTH 8\n"
 534       "PROPERTY CS_FIXED_BLOCK_HEIGHT 8\n"
 535       "PROPERTY CS_FIXED_BLOCK_DEPTH 1\n"
 536
 537       "DCL SV[0], THREAD_ID\n"
 538       "DCL SV[1], BLOCK_ID\n"
 539
 540       "DCL CONST[0..5]\n"
 541       "DCL SVIEW[0..2], RECT, FLOAT\n"
 542       "DCL SAMP[0..2]\n"
 543
 544       "DCL IMAGE[0], 2D, WR\n"
 545       "DCL TEMP[0..5]\n"
 546
 547       "IMM[0] UINT32 { 8, 8, 1, 0}\n"
 548       "IMM[1] FLT32 { 1.0, 2.0, 0.0, 0.0}\n"
 549
 550       "UMAD TEMP[0], SV[1], IMM[0], SV[0]\n"
 551
 552       /* Drawn area check */
 553       "USGE TEMP[1].xy, TEMP[0].xyxy, CONST[4].xyxy\n"
 554       "USLT TEMP[1].zw, TEMP[0].xyxy, CONST[4].zwzw\n"
 555       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].yyyy\n"
 556       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].zzzz\n"
 557       "AND TEMP[1].x, TEMP[1].xxxx, TEMP[1].wwww\n"
 558
 559       "UIF TEMP[1]\n"
 560          /* Translate */
 561          "UADD TEMP[2].xy, TEMP[0], -CONST[5].xyxy\n"
 562          "U2F TEMP[2], TEMP[2]\n"
 563          "DIV TEMP[3], TEMP[2], IMM[1].yyyy\n"
 564
 565          /* Scale */
 566          "DIV TEMP[2], TEMP[2], CONST[3].zwzw\n"
 567          "DIV TEMP[3], TEMP[3], CONST[3].zwzw\n"
 568
 569          /* Fetch texels */
 570          "TEX_LZ TEMP[4].x, TEMP[2], SAMP[0], RECT\n"
 571          "TEX_LZ TEMP[4].y, TEMP[3], SAMP[1], RECT\n"
 572          "TEX_LZ TEMP[4].z, TEMP[3], SAMP[2], RECT\n"
 573
 574          "MOV TEMP[4].w, IMM[1].xxxx\n"
 575
 576          "MOV TEMP[5].xy, TEMP[4].yzww\n"
 577
 578          "STORE IMAGE[0], TEMP[0], TEMP[5], 2D\n"
 579       "ENDIF\n"
 580
 581       "END\n";
 582
 583 static void
 584 cs_launch(struct vl_compositor *c,
 585           void                 *cs,
 586           const struct u_rect  *draw_area)
 587 {
 588    struct pipe_context *ctx = c->pipe;
 589
 590    /* Bind the image */
 591    struct pipe_image_view image = {0};
 592    image.resource = c->fb_state.cbufs[0]->texture;
 593    image.shader_access = image.access = PIPE_IMAGE_ACCESS_READ_WRITE;
 594    image.format = c->fb_state.cbufs[0]->texture->format;
 595
 596    ctx->set_shader_images(c->pipe, PIPE_SHADER_COMPUTE, 0, 1, &image);
 597
 598    /* Bind compute shader */
 599    ctx->bind_compute_state(ctx, cs);
 600
 601    /* Dispatch compute */
 602    struct pipe_grid_info info = {0};
 603    info.block[0] = 8;
 604    info.block[1] = 8;
 605    info.block[2] = 1;
 606    info.grid[0] = DIV_ROUND_UP(draw_area->x1, info.block[0]);
 607    info.grid[1] = DIV_ROUND_UP(draw_area->y1, info.block[1]);
 608    info.grid[2] = 1;
 609
 610    ctx->launch_grid(ctx, &info);
 611
 612    /* Make the result visible to all clients. */
 613    ctx->memory_barrier(ctx, PIPE_BARRIER_ALL);
 614
 615 }
 616
 617 static inline struct u_rect
 618 calc_drawn_area(struct vl_compositor_state *s,
 619                 struct vl_compositor_layer *layer)
 620 {
 621    struct vertex2f tl, br;
 622    struct u_rect result;
 623
 624    assert(s && layer);
 625
 626    tl = layer->dst.tl;
 627    br = layer->dst.br;
 628
 629    /* Scale */
 630    result.x0 = tl.x * layer->viewport.scale[0] + layer->viewport.translate[0];
 631    result.y0 = tl.y * layer->viewport.scale[1] + layer->viewport.translate[1];
 632    result.x1 = br.x * layer->viewport.scale[0] + layer->viewport.translate[0];
 633    result.y1 = br.y * layer->viewport.scale[1] + layer->viewport.translate[1];
 634
 635    /* Clip */
 636    result.x0 = MAX2(result.x0, s->scissor.minx);
 637    result.y0 = MAX2(result.y0, s->scissor.miny);
 638    result.x1 = MIN2(result.x1, s->scissor.maxx);
 639    result.y1 = MIN2(result.y1, s->scissor.maxy);
 640    return result;
 641 }
 642
 643 static bool
 644 set_viewport(struct vl_compositor_state *s,
 645              struct cs_viewport         *drawn)
 646 {
 647    struct pipe_transfer *buf_transfer;
 648
 649    assert(s && drawn);
 650
 651    void *ptr = pipe_buffer_map(s->pipe, s->shader_params,
 652                                PIPE_TRANSFER_READ | PIPE_TRANSFER_WRITE,
 653                                &buf_transfer);
 654
 655    if (!ptr)
 656      return false;
 657
 658    float *ptr_float = (float *)ptr;
 659    ptr_float += sizeof(vl_csc_matrix)/sizeof(float) + 2;
 660    *ptr_float++ = drawn->scale_x;
 661    *ptr_float++ = drawn->scale_y;
 662
 663    int *ptr_int = (int *)ptr_float;
 664    *ptr_int++ = drawn->area.x0;
 665    *ptr_int++ = drawn->area.y0;
 666    *ptr_int++ = drawn->area.x1;
 667    *ptr_int++ = drawn->area.y1;
 668    *ptr_int++ = drawn->translate_x;
 669    *ptr_int++ = drawn->translate_y;
 670
 671    ptr_float = (float *)ptr_int;
 672    *ptr_float++ = drawn->sampler0_w;
 673    *ptr_float = drawn->sampler0_h;
 674    pipe_buffer_unmap(s->pipe, buf_transfer);
 675
 676    return true;
 677 }
 678
 679 static void
 680 draw_layers(struct vl_compositor       *c,
 681             struct vl_compositor_state *s,
 682             struct u_rect              *dirty)
 683 {
 684    unsigned i;
 685
 686    assert(c);
 687
 688    for (i = 0; i < VL_COMPOSITOR_MAX_LAYERS; ++i) {
 689       if (s->used_layers & (1 << i)) {
 690          struct vl_compositor_layer *layer = &s->layers[i];
 691          struct pipe_sampler_view **samplers = &layer->sampler_views[0];
 692          unsigned num_sampler_views = !samplers[1] ? 1 : !samplers[2] ? 2 : 3;
 693          struct cs_viewport drawn;
 694
 695          drawn.area = calc_drawn_area(s, layer);
 696          drawn.scale_x = layer->viewport.scale[0] /
 697                   (float)layer->sampler_views[0]->texture->width0;
 698          drawn.scale_y = drawn.scale_x;
 699          drawn.translate_x = (int)layer->viewport.translate[0];
 700          drawn.translate_y = (int)layer->viewport.translate[1];
 701          drawn.sampler0_w = (float)layer->sampler_views[0]->texture->width0;
 702          drawn.sampler0_h = (float)layer->sampler_views[0]->texture->height0;
 703          set_viewport(s, &drawn);
 704
 705          c->pipe->bind_sampler_states(c->pipe, PIPE_SHADER_COMPUTE, 0,
 706                         num_sampler_views, layer->samplers);
 707          c->pipe->set_sampler_views(c->pipe, PIPE_SHADER_COMPUTE, 0,
 708                         num_sampler_views, samplers);
 709
 710          cs_launch(c, layer->cs, &(drawn.area));
 711
 712          /* Unbind. */
 713          c->pipe->set_shader_images(c->pipe, PIPE_SHADER_COMPUTE, 0, 1, NULL);
 714          c->pipe->set_constant_buffer(c->pipe, PIPE_SHADER_COMPUTE, 0, NULL);
 715          c->pipe->set_sampler_views(c->pipe, PIPE_SHADER_FRAGMENT, 0,
 716                         num_sampler_views, NULL);
 717          c->pipe->bind_compute_state(c->pipe, NULL);
 718          c->pipe->bind_sampler_states(c->pipe, PIPE_SHADER_COMPUTE, 0,
 719                         num_sampler_views, NULL);
 720
 721          if (dirty) {
 722             struct u_rect drawn = calc_drawn_area(s, layer);
 723             dirty->x0 = MIN2(drawn.x0, dirty->x0);
 724             dirty->y0 = MIN2(drawn.y0, dirty->y0);
 725             dirty->x1 = MAX2(drawn.x1, dirty->x1);
 726             dirty->y1 = MAX2(drawn.y1, dirty->y1);
 727          }
 728       }
 729    }
 730 }
 731
 732 void *
 733 vl_compositor_cs_create_shader(struct vl_compositor *c,
 734                                const char           *compute_shader_text)
 735 {
 736    assert(c && compute_shader_text);
 737
 738    struct tgsi_token tokens[1024];
 739    if (!tgsi_text_translate(compute_shader_text, tokens, ARRAY_SIZE(tokens))) {
 740       assert(0);
 741       return NULL;
 742    }
 743
 744    struct pipe_compute_state state = {0};
 745    state.ir_type = PIPE_SHADER_IR_TGSI;
 746    state.prog = tokens;
 747
 748    /* create compute shader */
 749    return c->pipe->create_compute_state(c->pipe, &state);
 750 }
 751
 752 void
 753 vl_compositor_cs_render(struct vl_compositor_state *s,
 754                         struct vl_compositor       *c,
 755                         struct pipe_surface        *dst_surface,
 756                         struct u_rect              *dirty_area,
 757                         bool                        clear_dirty)
 758 {
 759    assert(c && s);
 760    assert(dst_surface);
 761
 762    c->fb_state.width = dst_surface->width;
 763    c->fb_state.height = dst_surface->height;
 764    c->fb_state.cbufs[0] = dst_surface;
 765
 766    if (!s->scissor_valid) {
 767       s->scissor.minx = 0;
 768       s->scissor.miny = 0;
 769       s->scissor.maxx = dst_surface->width;
 770       s->scissor.maxy = dst_surface->height;
 771    }
 772
 773    if (clear_dirty && dirty_area &&
 774        (dirty_area->x0 < dirty_area->x1 || dirty_area->y0 < dirty_area->y1)) {
 775
 776       c->pipe->clear_render_target(c->pipe, dst_surface, &s->clear_color,
 777                        0, 0, dst_surface->width, dst_surface->height, false);
 778       dirty_area->x0 = dirty_area->y0 = VL_COMPOSITOR_MAX_DIRTY;
 779       dirty_area->x1 = dirty_area->y1 = VL_COMPOSITOR_MIN_DIRTY;
 780    }
 781
 782    pipe_set_constant_buffer(c->pipe, PIPE_SHADER_COMPUTE, 0, s->shader_params);
 783
 784    draw_layers(c, s, dirty_area);
 785 }
 786
 787 bool vl_compositor_cs_init_shaders(struct vl_compositor *c)
 788 {
 789         assert(c);
 790
 791         c->cs_video_buffer = vl_compositor_cs_create_shader(c, compute_shader_video_buffer);
 792         if (!c->cs_video_buffer) {
 793                 debug_printf("Unable to create video_buffer compute shader.\n");
 794                 return false;
 795         }
 796
 797         c->cs_weave_rgb = vl_compositor_cs_create_shader(c, compute_shader_weave);
 798         if (!c->cs_weave_rgb) {
 799                 debug_printf("Unable to create weave_rgb compute shader.\n");
 800                 return false;
 801         }
 802
 803         c->cs_yuv.weave.y = vl_compositor_cs_create_shader(c, compute_shader_yuv_weave_y);
 804         c->cs_yuv.weave.uv = vl_compositor_cs_create_shader(c, compute_shader_yuv_weave_uv);
 805         c->cs_yuv.bob.y = vl_compositor_cs_create_shader(c, compute_shader_yuv_bob_y);
 806         c->cs_yuv.bob.uv = vl_compositor_cs_create_shader(c, compute_shader_yuv_bob_uv);
 807         if (!c->cs_yuv.weave.y || !c->cs_yuv.weave.uv ||
 808             !c->cs_yuv.bob.y || !c->cs_yuv.bob.uv) {
 809                 debug_printf("Unable to create YCbCr i-to-YCbCr p deint compute shader.\n");
 810                 return false;
 811         }
 812
 813         return true;
 814 }
 815
 816 void vl_compositor_cs_cleanup_shaders(struct vl_compositor *c)
 817 {
 818         assert(c);
 819
 820         if (c->cs_video_buffer)
 821                 c->pipe->delete_compute_state(c->pipe, c->cs_video_buffer);
 822         if (c->cs_weave_rgb)
 823                 c->pipe->delete_compute_state(c->pipe, c->cs_weave_rgb);
 824         if (c->cs_yuv.weave.y)
 825                 c->pipe->delete_compute_state(c->pipe, c->cs_yuv.weave.y);
 826         if (c->cs_yuv.weave.uv)
 827                 c->pipe->delete_compute_state(c->pipe, c->cs_yuv.weave.uv);
 828         if (c->cs_yuv.bob.y)
 829                 c->pipe->delete_compute_state(c->pipe, c->cs_yuv.bob.y);
 830         if (c->cs_yuv.bob.uv)
 831                 c->pipe->delete_compute_state(c->pipe, c->cs_yuv.bob.uv);
 832 }