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
 221 /*** PIPE_FORMAT_Z32_FLOAT ***/
 222
 223 /**
 224  * Return each Z value as four floats in [0,1].
 225  */
 226 static void
 227 z32f_get_tile_rgba(const float *src,
 228                    unsigned w, unsigned h,
 229                    float *p,
 230                    unsigned dst_stride)
 231 {
 232    unsigned i, j;
 233
 234    for (i = 0; i < h; i++) {
 235       float *pRow = p;
 236       for (j = 0; j < w; j++, pRow += 4) {
 237          pRow[0] =
 238          pRow[1] =
 239          pRow[2] =
 240          pRow[3] = *src++;
 241       }
 242       p += dst_stride;
 243    }
 244 }
 245
 246
 247 void
 248 pipe_tile_raw_to_rgba(enum pipe_format format,
 249                       void *src,
 250                       uint w, uint h,
 251                       float *dst, unsigned dst_stride)
 252 {
 253    switch (format) {
 254    case PIPE_FORMAT_Z16_UNORM:
 255       z16_get_tile_rgba((ushort *) src, w, h, dst, dst_stride);
 256       break;
 257    case PIPE_FORMAT_Z32_UNORM:
 258       z32_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 259       break;
 260    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 261    case PIPE_FORMAT_Z24X8_UNORM:
 262       s8z24_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 263       break;
 264    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 265    case PIPE_FORMAT_X8Z24_UNORM:
 266       z24s8_get_tile_rgba((unsigned *) src, w, h, dst, dst_stride);
 267       break;
 268    case PIPE_FORMAT_Z32_FLOAT:
 269       z32f_get_tile_rgba((float *) src, w, h, dst, dst_stride);
 270       break;
 271    default:
 272       util_format_read_4f(format,
 273                           dst, dst_stride * sizeof(float),
 274                           src, util_format_get_stride(format, w),
 275                           0, 0, w, h);
 276    }
 277 }
 278
 279
 280 void
 281 pipe_get_tile_rgba(struct pipe_context *pipe,
 282                    struct pipe_transfer *pt,
 283                    uint x, uint y, uint w, uint h,
 284                    float *p)
 285 {
 286    unsigned dst_stride = w * 4;
 287    void *packed;
 288    enum pipe_format format = pt->resource->format;
 289
 290    if (u_clip_tile(x, y, &w, &h, &pt->box))
 291       return;
 292
 293    packed = MALLOC(util_format_get_nblocks(format, w, h) * util_format_get_blocksize(format));
 294
 295    if (!packed)
 296       return;
 297
 298    if(format == PIPE_FORMAT_UYVY || format == PIPE_FORMAT_YUYV)
 299       assert((x & 1) == 0);
 300
 301    pipe_get_tile_raw(pipe, pt, x, y, w, h, packed, 0);
 302
 303    pipe_tile_raw_to_rgba(format, packed, w, h, p, dst_stride);
 304
 305    FREE(packed);
 306 }
 307
 308
 309 void
 310 pipe_get_tile_swizzle(struct pipe_context *pipe,
 311                       struct pipe_transfer *pt,
 312                       uint x,
 313                       uint y,
 314                       uint w,
 315                       uint h,
 316                       uint swizzle_r,
 317                       uint swizzle_g,
 318                       uint swizzle_b,
 319                       uint swizzle_a,
 320                       enum pipe_format format,
 321                       float *p)
 322 {
 323    unsigned dst_stride = w * 4;
 324    void *packed;
 325    uint iy;
 326    float rgba01[6];
 327
 328    if (u_clip_tile(x, y, &w, &h, &pt->box)) {
 329       return;
 330    }
 331
 332    packed = MALLOC(util_format_get_nblocks(format, w, h) * util_format_get_blocksize(format));
 333    if (!packed) {
 334       return;
 335    }
 336
 337    if (format == PIPE_FORMAT_UYVY || format == PIPE_FORMAT_YUYV) {
 338       assert((x & 1) == 0);
 339    }
 340
 341    pipe_get_tile_raw(pipe, pt, x, y, w, h, packed, 0);
 342
 343    pipe_tile_raw_to_rgba(format, packed, w, h, p, dst_stride);
 344
 345    FREE(packed);
 346
 347    if (swizzle_r == PIPE_SWIZZLE_RED &&
 348        swizzle_g == PIPE_SWIZZLE_GREEN &&
 349        swizzle_b == PIPE_SWIZZLE_BLUE &&
 350        swizzle_a == PIPE_SWIZZLE_ALPHA) {
 351       /* no-op, skip */
 352       return;
 353    }
 354
 355    rgba01[PIPE_SWIZZLE_ZERO] = 0.0f;
 356    rgba01[PIPE_SWIZZLE_ONE] = 1.0f;
 357
 358    for (iy = 0; iy < h; iy++) {
 359       float *row = p;
 360       uint ix;
 361
 362       for (ix = 0; ix < w; ix++) {
 363          rgba01[PIPE_SWIZZLE_RED] = row[0];
 364          rgba01[PIPE_SWIZZLE_GREEN] = row[1];
 365          rgba01[PIPE_SWIZZLE_BLUE] = row[2];
 366          rgba01[PIPE_SWIZZLE_ALPHA] = row[3];
 367
 368          *row++ = rgba01[swizzle_r];
 369          *row++ = rgba01[swizzle_g];
 370          *row++ = rgba01[swizzle_b];
 371          *row++ = rgba01[swizzle_a];
 372       }
 373       p += dst_stride;
 374    }
 375 }
 376
 377
 378 void
 379 pipe_put_tile_rgba(struct pipe_context *pipe,
 380                    struct pipe_transfer *pt,
 381                    uint x, uint y, uint w, uint h,
 382                    const float *p)
 383 {
 384    unsigned src_stride = w * 4;
 385    void *packed;
 386    enum pipe_format format = pt->resource->format;
 387
 388    if (u_clip_tile(x, y, &w, &h, &pt->box))
 389       return;
 390
 391    packed = MALLOC(util_format_get_nblocks(format, w, h) * util_format_get_blocksize(format));
 392
 393    if (!packed)
 394       return;
 395
 396    switch (format) {
 397    case PIPE_FORMAT_Z16_UNORM:
 398       /*z16_put_tile_rgba((ushort *) packed, w, h, p, src_stride);*/
 399       break;
 400    case PIPE_FORMAT_Z32_UNORM:
 401       /*z32_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 402       break;
 403    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 404    case PIPE_FORMAT_Z24X8_UNORM:
 405       /*s8z24_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 406       break;
 407    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 408    case PIPE_FORMAT_X8Z24_UNORM:
 409       /*z24s8_put_tile_rgba((unsigned *) packed, w, h, p, src_stride);*/
 410       break;
 411    default:
 412       util_format_write_4f(format,
 413                            p, src_stride * sizeof(float),
 414                            packed, util_format_get_stride(format, w),
 415                            0, 0, w, h);
 416    }
 417
 418    pipe_put_tile_raw(pipe, pt, x, y, w, h, packed, 0);
 419
 420    FREE(packed);
 421 }
 422
 423
 424 /**
 425  * Get a block of Z values, converted to 32-bit range.
 426  */
 427 void
 428 pipe_get_tile_z(struct pipe_context *pipe,
 429                 struct pipe_transfer *pt,
 430                 uint x, uint y, uint w, uint h,
 431                 uint *z)
 432 {
 433    const uint dstStride = w;
 434    ubyte *map;
 435    uint *pDest = z;
 436    uint i, j;
 437    enum pipe_format format = pt->resource->format;
 438
 439    if (u_clip_tile(x, y, &w, &h, &pt->box))
 440       return;
 441
 442    map = (ubyte *)pipe->transfer_map(pipe, pt);
 443    if (!map) {
 444       assert(0);
 445       return;
 446    }
 447
 448    switch (format) {
 449    case PIPE_FORMAT_Z32_UNORM:
 450       {
 451          const uint *ptrc
 452             = (const uint *)(map  + y * pt->stride + x*4);
 453          for (i = 0; i < h; i++) {
 454             memcpy(pDest, ptrc, 4 * w);
 455             pDest += dstStride;
 456             ptrc += pt->stride/4;
 457          }
 458       }
 459       break;
 460    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 461    case PIPE_FORMAT_Z24X8_UNORM:
 462       {
 463          const uint *ptrc
 464             = (const uint *)(map + y * pt->stride + x*4);
 465          for (i = 0; i < h; i++) {
 466             for (j = 0; j < w; j++) {
 467                /* convert 24-bit Z to 32-bit Z */
 468                pDest[j] = (ptrc[j] << 8) | ((ptrc[j] >> 16) & 0xff);
 469             }
 470             pDest += dstStride;
 471             ptrc += pt->stride/4;
 472          }
 473       }
 474       break;
 475    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 476    case PIPE_FORMAT_X8Z24_UNORM:
 477       {
 478          const uint *ptrc
 479             = (const uint *)(map + y * pt->stride + x*4);
 480          for (i = 0; i < h; i++) {
 481             for (j = 0; j < w; j++) {
 482                /* convert 24-bit Z to 32-bit Z */
 483                pDest[j] = (ptrc[j] & 0xffffff00) | ((ptrc[j] >> 24) & 0xff);
 484             }
 485             pDest += dstStride;
 486             ptrc += pt->stride/4;
 487          }
 488       }
 489       break;
 490    case PIPE_FORMAT_Z16_UNORM:
 491       {
 492          const ushort *ptrc
 493             = (const ushort *)(map + y * pt->stride + x*2);
 494          for (i = 0; i < h; i++) {
 495             for (j = 0; j < w; j++) {
 496                /* convert 16-bit Z to 32-bit Z */
 497                pDest[j] = (ptrc[j] << 16) | ptrc[j];
 498             }
 499             pDest += dstStride;
 500             ptrc += pt->stride/2;
 501          }
 502       }
 503       break;
 504    default:
 505       assert(0);
 506    }
 507
 508    pipe->transfer_unmap(pipe, pt);
 509 }
 510
 511
 512 void
 513 pipe_put_tile_z(struct pipe_context *pipe,
 514                 struct pipe_transfer *pt,
 515                 uint x, uint y, uint w, uint h,
 516                 const uint *zSrc)
 517 {
 518    const uint srcStride = w;
 519    const uint *ptrc = zSrc;
 520    ubyte *map;
 521    uint i, j;
 522    enum pipe_format format = pt->resource->format;
 523
 524    if (u_clip_tile(x, y, &w, &h, &pt->box))
 525       return;
 526
 527    map = (ubyte *)pipe->transfer_map(pipe, pt);
 528    if (!map) {
 529       assert(0);
 530       return;
 531    }
 532
 533    switch (format) {
 534    case PIPE_FORMAT_Z32_UNORM:
 535       {
 536          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 537          for (i = 0; i < h; i++) {
 538             memcpy(pDest, ptrc, 4 * w);
 539             pDest += pt->stride/4;
 540             ptrc += srcStride;
 541          }
 542       }
 543       break;
 544    case PIPE_FORMAT_Z24_UNORM_S8_USCALED:
 545       {
 546          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 547          //assert((pt->usage & PIPE_TRANSFER_READ_WRITE) == PIPE_TRANSFER_READ_WRITE);
 548          for (i = 0; i < h; i++) {
 549             for (j = 0; j < w; j++) {
 550                /* convert 32-bit Z to 24-bit Z, preserve stencil */
 551                pDest[j] = (pDest[j] & 0xff000000) | ptrc[j] >> 8;
 552             }
 553             pDest += pt->stride/4;
 554             ptrc += srcStride;
 555          }
 556       }
 557       break;
 558    case PIPE_FORMAT_Z24X8_UNORM:
 559       {
 560          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 561          for (i = 0; i < h; i++) {
 562             for (j = 0; j < w; j++) {
 563                /* convert 32-bit Z to 24-bit Z (0 stencil) */
 564                pDest[j] = ptrc[j] >> 8;
 565             }
 566             pDest += pt->stride/4;
 567             ptrc += srcStride;
 568          }
 569       }
 570       break;
 571    case PIPE_FORMAT_S8_USCALED_Z24_UNORM:
 572       {
 573          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 574          //assert((pt->usage & PIPE_TRANSFER_READ_WRITE) == PIPE_TRANSFER_READ_WRITE);
 575          for (i = 0; i < h; i++) {
 576             for (j = 0; j < w; j++) {
 577                /* convert 32-bit Z to 24-bit Z, preserve stencil */
 578                pDest[j] = (pDest[j] & 0xff) | (ptrc[j] & 0xffffff00);
 579             }
 580             pDest += pt->stride/4;
 581             ptrc += srcStride;
 582          }
 583       }
 584       break;
 585    case PIPE_FORMAT_X8Z24_UNORM:
 586       {
 587          uint *pDest = (uint *) (map + y * pt->stride + x*4);
 588          for (i = 0; i < h; i++) {
 589             for (j = 0; j < w; j++) {
 590                /* convert 32-bit Z to 24-bit Z (0 stencil) */
 591                pDest[j] = ptrc[j] & 0xffffff00;
 592             }
 593             pDest += pt->stride/4;
 594             ptrc += srcStride;
 595          }
 596       }
 597       break;
 598    case PIPE_FORMAT_Z16_UNORM:
 599       {
 600          ushort *pDest = (ushort *) (map + y * pt->stride + x*2);
 601          for (i = 0; i < h; i++) {
 602             for (j = 0; j < w; j++) {
 603                /* convert 32-bit Z to 16-bit Z */
 604                pDest[j] = ptrc[j] >> 16;
 605             }
 606             pDest += pt->stride/2;
 607             ptrc += srcStride;
 608          }
 609       }
 610       break;
 611    default:
 612       assert(0);
 613    }
 614
 615    pipe->transfer_unmap(pipe, pt);
 616 }
 617
 618