src/gallium/auxiliary/util/u_tile.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2007 Tungsten Graphics, Inc., Cedar Park, Texas.
   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 TUNGSTEN GRAPHICS 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  **************************************************************************/
  27
  28 /**
  29  * RGBA/float tile get/put functions.
  30  * Usable both by drivers and state trackers.
  31  */
  32
  33
  34 #include "pipe/p_defines.h"
  35 #include "util/u_inlines.h"
  36
  37 #include "util/u_format.h"
  38 #include "util/u_math.h"
  39 #include "util/u_memory.h"
  40 #include "util/u_rect.h"
  41 #include "util/u_tile.h"
  42
  43
  44 /**
  45  * Move raw block of pixels from transfer object to user memory.
  46  */
  47 void
  48 pipe_get_tile_raw(struct pipe_context *pipe,
  49                   struct pipe_transfer *pt,
  50                   uint x, uint y, uint w, uint h,
  51                   void *dst, int dst_stride)
  52 {
  53    const void *src;
  54
  55    if (dst_stride == 0)
  56       dst_stride = util_format_get_stride(pt->resource->format, w);
  57
  58    if (u_clip_tile(x, y, &w, &h, &pt->box))
  59       return;
  60
  61    src = pipe->transfer_map(pipe, pt);
  62    assert(src);
  63    if(!src)
  64       return;
  65
  66    util_copy_rect(dst, pt->resource->format, dst_stride, 0, 0, w, h, src, pt->stride, x, y);
  67
  68    pipe->transfer_unmap(pipe, pt);
  69 }
  70
  71
  72 /**
  73  * Move raw block of pixels from user memory to transfer object.
  74  */
  75 void
  76 pipe_put_tile_raw(struct pipe_context *pipe,
  77                   struct pipe_transfer *pt,
  78                   uint x, uint y, uint w, uint h,
  79                   const void *src, int src_stride)
  80 {
  81    void *dst;
  82    enum pipe_format format = pt->resource->format;
  83
  84    if (src_stride == 0)
  85       src_stride = util_format_get_stride(format, w);
  86
  87    if (u_clip_tile(x, y, &w, &h, &pt->box))
  88       return;
  89
  90    dst = pipe->transfer_map(pipe, pt);
  91    assert(dst);
  92    if(!dst)
  93       return;
  94
  95    util_copy_rect(dst, format, pt->stride, x, y, w, h, src, src_stride, 0, 0);
  96
  97    pipe->transfer_unmap(pipe, pt);
  98 }
  99
 100
 101
 102
 103 /** Convert short in [-32768,32767] to GLfloat in [-1.0,1.0] */
 104 #define SHORT_TO_FLOAT(S)   ((2.0F * (S) + 1.0F) * (1.0F/65535.0F))
 105
 106 #define UNCLAMPED_FLOAT_TO_SHORT(us, f)  \
 107    us = ( (short) ( CLAMP((f), -1.0, 1.0) * 32767.0F) )
 108
 109
 110
 111 /*** PIPE_FORMAT_Z16_UNORM ***/
 112
 113 /**
 114  * Return each Z value as four floats in [0,1].
 115  */
 116 static void
 117 z16_get_tile_rgba(const ushort *src,
 118                   unsigned w, unsigned h,
 119                   float *p,
 120                   unsigned dst_stride)
 121 {
 122    const float scale = 1.0f / 65535.0f;
 123    unsigned i, j;
 124
 125    for (i = 0; i < h; i++) {
 126       float *pRow = p;
 127       for (j = 0; j < w; j++, pRow += 4) {
 128          pRow[0] =
 129          pRow[1] =
 130          pRow[2] =
 131          pRow[3] = *src++ * scale;
 132       }
 133       p += dst_stride;
 134    }
 135 }
 136
 137
 138
 139
 140 /*** PIPE_FORMAT_Z32_UNORM ***/
 141
 142 /**
 143  * Return each Z value as four floats in [0,1].
 144  */
 145 static void
 146 z32_get_tile_rgba(const unsigned *src,
 147                   unsigned w, unsigned h,
 148                   float *p,
 149                   unsigned dst_stride)
 150 {
 151    const double scale = 1.0 / (double) 0xffffffff;
 152    unsigned i, j;
 153
 154    for (i = 0; i < h; i++) {
 155       float *pRow = p;
 156       for (j = 0; j < w; j++, pRow += 4) {
 157          pRow[0] =
 158          pRow[1] =
 159          pRow[2] =
 160          pRow[3] = (float) (*src++ * scale);
 161       }
 162       p += dst_stride;
 163    }
 164 }
 165
 166
 167 /*** PIPE_FORMAT_Z24_UNORM_S8_USCALED ***/
 168
 169 /**
 170  * Return Z component as four float in [0,1].  Stencil part ignored.
 171  */
 172 static void
 173 s8z24_get_tile_rgba(const unsigned *src,
 174                     unsigned w, unsigned h,
 175                     float *p,
 176                     unsigned dst_stride)
 177 {
 178    const double scale = 1.0 / ((1 << 24) - 1);
 179    unsigned i, j;
 180
 181    for (i = 0; i < h; i++) {
 182       float *pRow = p;
 183       for (j = 0; j < w; j++, pRow += 4) {
 184          pRow[0] =
 185          pRow[1] =
 186          pRow[2] =
 187          pRow[3] = (float) (scale * (*src++ & 0xffffff));
 188       }
 189       p += dst_stride;
 190    }
 191 }
 192
 193
 194 /*** PIPE_FORMAT_S8_USCALED_Z24_UNORM ***/
 195
 196 /**
 197  * Return Z component as four float in [0,1].  Stencil part ignored.
 198  */
 199 static void
 200 z24s8_get_tile_rgba(const unsigned *src,
 201                     unsigned w, unsigned h,
 202                     float *p,
 203                     unsigned dst_stride)
 204 {
 205    const double scale = 1.0 / ((1 << 24) - 1);
 206    unsigned i, j;
 207
 208    for (i = 0; i < h; i++) {
 209       float *pRow = p;
 210       for (j = 0; j < w; j++, pRow += 4) {
 211          pRow[0] =
 212          pRow[1] =
 213          pRow[2] =
 214          pRow[3] = (float) (scale * (*src++ >> 8));
 215       }
 216       p += dst_stride;
 217    }
 218 }
 219
 220 /*** PIPE_FORMAT_S8X24_USCALED ***/
 221
 222 /**
 223  * Return S component as four uint32_t in [0..255].  Z part ignored.
 224  */
 225 static void
 226 s8x24_get_tile_rgba(const unsigned *src,
 227                     unsigned w, unsigned h,
 228                     float *p,
 229                     unsigned dst_stride)
 230 {
 231    unsigned i, j;
 232
 233    for (i = 0; i < h; i++) {
 234       float *pRow = p;
 235
 236       for (j = 0; j < w; j++, pRow += 4) {
 237          pRow[0] =
 238          pRow[1] =
 239          pRow[2] =
 240          pRow[3] = (float)((*src++ >> 24) & 0xff);
 241       }
 242
 243       p += dst_stride;
 244    }
 245 }
 246
 247 /*** PIPE_FORMAT_X24S8_USCALED ***/
 248
 249 /**
 250  * Return S component as four uint32_t in [0..255].  Z part ignored.
 251  */
 252 static void
 253 x24s8_get_tile_rgba(const unsigned *src,
 254                     unsigned w, unsigned h,
 255                     float *p,
 256                     unsigned dst_stride)
 257 {
 258    unsigned i, j;
 259
 260    for (i = 0; i < h; i++) {
 261       float *pRow = p;
 262       for (j = 0; j < w; j++, pRow += 4) {
 263          pRow[0] =
 264          pRow[1] =
 265          pRow[2] =
 266          pRow[3] = (float)(*src++ & 0xff);
 267       }
 268       p += dst_stride;
 269    }
 270 }
 271
 272
 273 /**
 274  * Return S component as four uint32_t in [0..255].  Z part ignored.
 275  */
 276 static void
 277 s8_get_tile_rgba(const unsigned char *src,
 278                  unsigned w, unsigned h,
 279                  float *p,
 280                  unsigned dst_stride)
 281 {
 282    unsigned i, j;
 283
 284    for (i = 0; i < h; i++) {
 285       float *pRow = p;
 286       for (j = 0; j < w; j++, pRow += 4) {
 287          pRow[0] =
 288          pRow[1] =
 289          pRow[2] =
 290          pRow[3] = (float)(*src++ & 0xff);
 291       }
 292       p += dst_stride;
 293    }
 294 }
 295
 296 /*** PIPE_FORMAT_Z32_FLOAT ***/
 297
 298 /**
 299  * Return each Z value as four floats in [0,1].
 300  */
 301 static void
 302 z32f_get_tile_rgba(const float *src,
 303                    unsigned w, unsigned h,
 304                    float *p,
 305                    unsigned dst_stride)
 306 {
 307    unsigned i, j;
 308
 309    for (i = 0; i < h; i++) {
 310       float *pRow = p;
 311       for (j = 0; j < w; j++, pRow += 4) {
 312          pRow[0] =
 313          pRow[1] =
 314          pRow[2] =
 315          pRow[3] = *src++;
 316       }
 317       p += dst_stride;
 318    }
 319 }
 320
 321
 322 void
 323 pipe_tile_raw_to_rgba(enum pipe_format format,
 324                       void *src,
 325                       uint w, uint h,
 326                       float *dst, unsigned dst_stride)
 327 {
 328    switch (format) {
 329    case PIPE_FORMAT_Z16_UNORM:
 330       z16_get_tile_rgba((ushort *) src, w, h, dst, dst_stride);
 331       break;
 332    case PIPE_FORMAT_Z32_UNORM:
 333       z32_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 334       break;
 335    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 336    case PIPE_FORMAT_Z24X8_UNORM:
 337       s8z24_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 338       break;
 339    case PIPE_FORMAT_S8_USCALED:
 340       s8_get_tile_rgba((unsigned char *) src, w, h, dst, dst_stride);
 341       break;
 342    case PIPE_FORMAT_X24S8_USCALED:
 343       s8x24_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 344       break;
 345    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 346    case PIPE_FORMAT_X8Z24_UNORM:
 347       z24s8_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 348       break;
 349    case PIPE_FORMAT_S8X24_USCALED:
 350       x24s8_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 351       break;
 352    case PIPE_FORMAT_Z32_FLOAT:
 353       z32f_get_tile_rgba((float *) src, w, h, dst, dst_stride);
 354       break;
 355    default:
 356       util_format_read_4f(format,
 357                           dst, dst_stride * sizeof(float),
 358                           src, util_format_get_stride(format, w),
 359                           0, 0, w, h);
 360    }
 361 }
 362
 363
 364 void
 365 pipe_get_tile_rgba(struct pipe_context *pipe,
 366                    struct pipe_transfer *pt,
 367                    uint x, uint y, uint w, uint h,
 368                    float *p)
 369 {
 370    pipe_get_tile_rgba_format(pipe, pt, x, y, w, h, pt->resource->format, p);
 371 }
 372
 373
 374 void
 375 pipe_get_tile_rgba_format(struct pipe_context *pipe,
 376                           struct pipe_transfer *pt,
 377                           uint x, uint y, uint w, uint h,
 378                           enum pipe_format format,
 379                           float *p)
 380 {
 381    unsigned dst_stride = w * 4;
 382    void *packed;
 383
 384    if (u_clip_tile(x, y, &w, &h, &pt->box)) {
 385       return;
 386    }
 387
 388    packed = MALLOC(util_format_get_nblocks(format, w, h) * util_format_get_blocksize(format));
 389    if (!packed) {
 390       return;
 391    }
 392
 393    if (format == PIPE_FORMAT_UYVY || format == PIPE_FORMAT_YUYV) {
 394       assert((x & 1) == 0);
 395    }
 396
 397    pipe_get_tile_raw(pipe, pt, x, y, w, h, packed, 0);
 398
 399    pipe_tile_raw_to_rgba(format, packed, w, h, p, dst_stride);
 400
 401    FREE(packed);
 402 }
 403
 404
 405 void
 406 pipe_put_tile_rgba(struct pipe_context *pipe,
 407                    struct pipe_transfer *pt,
 408                    uint x, uint y, uint w, uint h,
 409                    const float *p)
 410 {
 411    pipe_put_tile_rgba_format(pipe, pt, x, y, w, h, pt->resource->format, p);
 412 }
 413
 414
 415 void
 416 pipe_put_tile_rgba_format(struct pipe_context *pipe,
 417                           struct pipe_transfer *pt,
 418                           uint x, uint y, uint w, uint h,
 419                           enum pipe_format format,
 420                           const float *p)
 421 {
 422    unsigned src_stride = w * 4;
 423    void *packed;
 424
 425    if (u_clip_tile(x, y, &w, &h, &pt->box))
 426       return;
 427
 428    packed = MALLOC(util_format_get_nblocks(format, w, h) * util_format_get_blocksize(format));
 429
 430    if (!packed)
 431       return;
 432
 433    switch (format) {
 434    case PIPE_FORMAT_Z16_UNORM:
 435       /*z16_put_tile_rgba((ushort *) packed, w, h, p, src_stride);*/
 436       break;
 437    case PIPE_FORMAT_Z32_UNORM:
 438       /*z32_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 439       break;
 440    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 441    case PIPE_FORMAT_Z24X8_UNORM:
 442       /*s8z24_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 443       break;
 444    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 445    case PIPE_FORMAT_X8Z24_UNORM:
 446       /*z24s8_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 447       break;
 448    default:
 449       util_format_write_4f(format,
 450                            p, src_stride * sizeof(float),
 451                            packed, util_format_get_stride(format, w),
 452                            0, 0, w, h);
 453    }
 454
 455    pipe_put_tile_raw(pipe, pt, x, y, w, h, packed, 0);
 456
 457    FREE(packed);
 458 }
 459
 460
 461 /**
 462  * Get a block of Z values, converted to 32-bit range.
 463  */
 464 void
 465 pipe_get_tile_z(struct pipe_context *pipe,
 466                 struct pipe_transfer *pt,
 467                 uint x, uint y, uint w, uint h,
 468                 uint *z)
 469 {
 470    const uint dstStride = w;
 471    ubyte *map;
 472    uint *pDest = z;
 473    uint i, j;
 474    enum pipe_format format = pt->resource->format;
 475
 476    if (u_clip_tile(x, y, &w, &h, &pt->box))
 477       return;
 478
 479    map = (ubyte *)pipe->transfer_map(pipe, pt);
 480    if (!map) {
 481       assert(0);
 482       return;
 483    }
 484
 485    switch (format) {
 486    case PIPE_FORMAT_Z32_UNORM:
 487       {
 488          const uint *ptrc
 489             = (const uint *)(map  + y * pt->stride + x*4);
 490          for (i = 0; i < h; i++) {
 491             memcpy(pDest, ptrc, 4 * w);
 492             pDest += dstStride;
 493             ptrc += pt->stride/4;
 494          }
 495       }
 496       break;
 497    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 498    case PIPE_FORMAT_Z24X8_UNORM:
 499       {
 500          const uint *ptrc
 501             = (const uint *)(map + y * pt->stride + x*4);
 502          for (i = 0; i < h; i++) {
 503             for (j = 0; j < w; j++) {
 504                /* convert 24-bit Z to 32-bit Z */
 505                pDest[j] = (ptrc[j] << 8) | ((ptrc[j] >> 16) & 0xff);
 506             }
 507             pDest += dstStride;
 508             ptrc += pt->stride/4;
 509          }
 510       }
 511       break;
 512    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 513    case PIPE_FORMAT_X8Z24_UNORM:
 514       {
 515          const uint *ptrc
 516             = (const uint *)(map + y * pt->stride + x*4);
 517          for (i = 0; i < h; i++) {
 518             for (j = 0; j < w; j++) {
 519                /* convert 24-bit Z to 32-bit Z */
 520                pDest[j] = (ptrc[j] & 0xffffff00) | ((ptrc[j] >> 24) & 0xff);
 521             }
 522             pDest += dstStride;
 523             ptrc += pt->stride/4;
 524          }
 525       }
 526       break;
 527    case PIPE_FORMAT_Z16_UNORM:
 528       {
 529          const ushort *ptrc
 530             = (const ushort *)(map + y * pt->stride + x*2);
 531          for (i = 0; i < h; i++) {
 532             for (j = 0; j < w; j++) {
 533                /* convert 16-bit Z to 32-bit Z */
 534                pDest[j] = (ptrc[j] << 16) | ptrc[j];
 535             }
 536             pDest += dstStride;
 537             ptrc += pt->stride/2;
 538          }
 539       }
 540       break;
 541    default:
 542       assert(0);
 543    }
 544
 545    pipe->transfer_unmap(pipe, pt);
 546 }
 547
 548
 549 void
 550 pipe_put_tile_z(struct pipe_context *pipe,
 551                 struct pipe_transfer *pt,
 552                 uint x, uint y, uint w, uint h,
 553                 const uint *zSrc)
 554 {
 555    const uint srcStride = w;
 556    const uint *ptrc = zSrc;
 557    ubyte *map;
 558    uint i, j;
 559    enum pipe_format format = pt->resource->format;
 560
 561    if (u_clip_tile(x, y, &w, &h, &pt->box))
 562       return;
 563
 564    map = (ubyte *)pipe->transfer_map(pipe, pt);
 565    if (!map) {
 566       assert(0);
 567       return;
 568    }
 569
 570    switch (format) {
 571    case PIPE_FORMAT_Z32_UNORM:
 572       {
 573          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 574          for (i = 0; i < h; i++) {
 575             memcpy(pDest, ptrc, 4 * w);
 576             pDest += pt->stride/4;
 577             ptrc += srcStride;
 578          }
 579       }
 580       break;
 581    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 582       {
 583          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 584          /*assert((pt->usage & PIPE_TRANSFER_READ_WRITE) == PIPE_TRANSFER_READ_WRITE);*/
 585          for (i = 0; i < h; i++) {
 586             for (j = 0; j < w; j++) {
 587                /* convert 32-bit Z to 24-bit Z, preserve stencil */
 588                pDest[j] = (pDest[j] & 0xff000000) | ptrc[j] >> 8;
 589             }
 590             pDest += pt->stride/4;
 591             ptrc += srcStride;
 592          }
 593       }
 594       break;
 595    case PIPE_FORMAT_Z24X8_UNORM:
 596       {
 597          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 598          for (i = 0; i < h; i++) {
 599             for (j = 0; j < w; j++) {
 600                /* convert 32-bit Z to 24-bit Z (0 stencil) */
 601                pDest[j] = ptrc[j] >> 8;
 602             }
 603             pDest += pt->stride/4;
 604             ptrc += srcStride;
 605          }
 606       }
 607       break;
 608    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 609       {
 610          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 611          /*assert((pt->usage & PIPE_TRANSFER_READ_WRITE) == PIPE_TRANSFER_READ_WRITE);*/
 612          for (i = 0; i < h; i++) {
 613             for (j = 0; j < w; j++) {
 614                /* convert 32-bit Z to 24-bit Z, preserve stencil */
 615                pDest[j] = (pDest[j] & 0xff) | (ptrc[j] & 0xffffff00);
 616             }
 617             pDest += pt->stride/4;
 618             ptrc += srcStride;
 619          }
 620       }
 621       break;
 622    case PIPE_FORMAT_X8Z24_UNORM:
 623       {
 624          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 625          for (i = 0; i < h; i++) {
 626             for (j = 0; j < w; j++) {
 627                /* convert 32-bit Z to 24-bit Z (0 stencil) */
 628                pDest[j] = ptrc[j] & 0xffffff00;
 629             }
 630             pDest += pt->stride/4;
 631             ptrc += srcStride;
 632          }
 633       }
 634       break;
 635    case PIPE_FORMAT_Z16_UNORM:
 636       {
 637          ushort *pDest = (ushort *) (map + y * pt->stride + x*2);
 638          for (i = 0; i < h; i++) {
 639             for (j = 0; j < w; j++) {
 640                /* convert 32-bit Z to 16-bit Z */
 641                pDest[j] = ptrc[j] >> 16;
 642             }
 643             pDest += pt->stride/2;
 644             ptrc += srcStride;
 645          }
 646       }
 647       break;
 648    default:
 649       assert(0);
 650    }
 651
 652    pipe->transfer_unmap(pipe, pt);
 653 }
 654
 655