src/gallium/state_trackers/python/p_context.i

   1  /**************************************************************************
   2  *
   3  * Copyright 2008 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  * @file
  30  * SWIG interface definion for Gallium types.
  31  *
  32  * @author Jose Fonseca <jrfonseca@tungstengraphics.com>
  33  */
  34
  35 %nodefaultctor st_context;
  36 %nodefaultdtor st_context;
  37
  38 struct st_context {
  39 };
  40
  41 %extend st_context {
  42
  43    ~st_context() {
  44       st_context_destroy($self);
  45    }
  46
  47    /*
  48     * State functions (create/bind/destroy state objects)
  49     */
  50
  51    void set_blend( const struct pipe_blend_state *state ) {
  52       cso_set_blend($self->cso, state);
  53    }
  54
  55    void set_fragment_sampler( unsigned index, const struct pipe_sampler_state *state ) {
  56       cso_single_sampler($self->cso, index, state);
  57       cso_single_sampler_done($self->cso);
  58    }
  59
  60    void set_vertex_sampler( unsigned index, const struct pipe_sampler_state *state ) {
  61       cso_single_vertex_sampler($self->cso, index, state);
  62       cso_single_vertex_sampler_done($self->cso);
  63    }
  64
  65    void set_rasterizer( const struct pipe_rasterizer_state *state ) {
  66       cso_set_rasterizer($self->cso, state);
  67    }
  68
  69    void set_depth_stencil_alpha(const struct pipe_depth_stencil_alpha_state *state) {
  70       cso_set_depth_stencil_alpha($self->cso, state);
  71    }
  72
  73    void set_fragment_shader( const struct pipe_shader_state *state ) {
  74       void *fs;
  75
  76       if(!state) {
  77          cso_set_fragment_shader_handle($self->cso, NULL);
  78          return;
  79       }
  80
  81       fs = $self->pipe->create_fs_state($self->pipe, state);
  82       if(!fs)
  83          return;
  84
  85       if(cso_set_fragment_shader_handle($self->cso, fs) != PIPE_OK)
  86          return;
  87
  88       cso_delete_fragment_shader($self->cso, $self->fs);
  89       $self->fs = fs;
  90    }
  91
  92    void set_vertex_shader( const struct pipe_shader_state *state ) {
  93       void *vs;
  94
  95       if(!state) {
  96          cso_set_vertex_shader_handle($self->cso, NULL);
  97          return;
  98       }
  99
 100       vs = $self->pipe->create_vs_state($self->pipe, state);
 101       if(!vs)
 102          return;
 103
 104       if(cso_set_vertex_shader_handle($self->cso, vs) != PIPE_OK)
 105          return;
 106
 107       cso_delete_vertex_shader($self->cso, $self->vs);
 108       $self->vs = vs;
 109    }
 110
 111    void set_geometry_shader( const struct pipe_shader_state *state ) {
 112       void *gs;
 113
 114       if(!state) {
 115          cso_set_geometry_shader_handle($self->cso, NULL);
 116          return;
 117       }
 118
 119       gs = $self->pipe->create_gs_state($self->pipe, state);
 120       if(!gs)
 121          return;
 122
 123       if(cso_set_geometry_shader_handle($self->cso, gs) != PIPE_OK)
 124          return;
 125
 126       cso_delete_geometry_shader($self->cso, $self->gs);
 127       $self->gs = gs;
 128    }
 129
 130    /*
 131     * Parameter-like state (or properties)
 132     */
 133
 134    void set_blend_color(const struct pipe_blend_color *state ) {
 135       cso_set_blend_color($self->cso, state);
 136    }
 137
 138    void set_stencil_ref(const struct pipe_stencil_ref *state ) {
 139       cso_set_stencil_ref($self->cso, state);
 140    }
 141
 142    void set_clip(const struct pipe_clip_state *state ) {
 143       $self->pipe->set_clip_state($self->pipe, state);
 144    }
 145
 146    void set_constant_buffer(unsigned shader, unsigned index,
 147                             struct pipe_buffer *buffer )
 148    {
 149       $self->pipe->set_constant_buffer($self->pipe, shader, index, buffer);
 150    }
 151
 152    void set_framebuffer(const struct pipe_framebuffer_state *state )
 153    {
 154       memcpy(&$self->framebuffer, state, sizeof *state);
 155       cso_set_framebuffer($self->cso, state);
 156    }
 157
 158    void set_polygon_stipple(const struct pipe_poly_stipple *state ) {
 159       $self->pipe->set_polygon_stipple($self->pipe, state);
 160    }
 161
 162    void set_scissor(const struct pipe_scissor_state *state ) {
 163       $self->pipe->set_scissor_state($self->pipe, state);
 164    }
 165
 166    void set_viewport(const struct pipe_viewport_state *state) {
 167       cso_set_viewport($self->cso, state);
 168    }
 169
 170    void set_fragment_sampler_texture(unsigned index,
 171                                      struct pipe_texture *texture) {
 172       struct pipe_sampler_view templ;
 173
 174       if(!texture)
 175          texture = $self->default_texture;
 176       pipe_sampler_view_reference(&$self->fragment_sampler_views[index], NULL);
 177       u_sampler_view_default_template(&templ,
 178                                       texture,
 179                                       texture->format);
 180       $self->fragment_sampler_views[index] = $self->pipe->create_sampler_view($self->pipe,
 181                                                                               texture,
 182                                                                               &templ);
 183       $self->pipe->set_fragment_sampler_views($self->pipe,
 184                                               PIPE_MAX_SAMPLERS,
 185                                               $self->fragment_sampler_views);
 186    }
 187
 188    void set_vertex_sampler_texture(unsigned index,
 189                                    struct pipe_texture *texture) {
 190       struct pipe_sampler_view templ;
 191
 192       if(!texture)
 193          texture = $self->default_texture;
 194       pipe_sampler_view_reference(&$self->vertex_sampler_views[index], NULL);
 195       u_sampler_view_default_template(&templ,
 196                                       texture,
 197                                       texture->format);
 198       $self->vertex_sampler_views[index] = $self->pipe->create_sampler_view($self->pipe,
 199                                                                             texture,
 200                                                                             &templ);
 201
 202       $self->pipe->set_vertex_sampler_views($self->pipe,
 203                                             PIPE_MAX_VERTEX_SAMPLERS,
 204                                             $self->vertex_sampler_views);
 205    }
 206
 207    void set_vertex_buffer(unsigned index,
 208                           unsigned stride,
 209                           unsigned max_index,
 210                           unsigned buffer_offset,
 211                           struct pipe_buffer *buffer)
 212    {
 213       unsigned i;
 214       struct pipe_vertex_buffer state;
 215
 216       memset(&state, 0, sizeof(state));
 217       state.stride = stride;
 218       state.max_index = max_index;
 219       state.buffer_offset = buffer_offset;
 220       state.buffer = buffer;
 221
 222       memcpy(&$self->vertex_buffers[index], &state, sizeof(state));
 223
 224       for(i = 0; i < PIPE_MAX_ATTRIBS; ++i)
 225          if(self->vertex_buffers[i].buffer)
 226             $self->num_vertex_buffers = i + 1;
 227
 228       $self->pipe->set_vertex_buffers($self->pipe,
 229                                       $self->num_vertex_buffers,
 230                                       $self->vertex_buffers);
 231    }
 232
 233    void set_vertex_element(unsigned index,
 234                            const struct pipe_vertex_element *element)
 235    {
 236       memcpy(&$self->vertex_elements[index], element, sizeof(*element));
 237    }
 238
 239    void set_vertex_elements(unsigned num)
 240    {
 241       $self->num_vertex_elements = num;
 242       cso_set_vertex_elements($self->cso,
 243                               $self->num_vertex_elements,
 244                               $self->vertex_elements);
 245    }
 246
 247    /*
 248     * Draw functions
 249     */
 250
 251    void draw_arrays(unsigned mode, unsigned start, unsigned count) {
 252       $self->pipe->draw_arrays($self->pipe, mode, start, count);
 253    }
 254
 255    void draw_elements( struct pipe_buffer *indexBuffer,
 256                        unsigned indexSize,
 257                        unsigned mode, unsigned start, unsigned count)
 258    {
 259       $self->pipe->draw_elements($self->pipe,
 260                                  indexBuffer,
 261                                  indexSize,
 262                                  mode, start, count);
 263    }
 264
 265    void draw_range_elements( struct pipe_buffer *indexBuffer,
 266                              unsigned indexSize, unsigned minIndex, unsigned maxIndex,
 267                              unsigned mode, unsigned start, unsigned count)
 268    {
 269       $self->pipe->draw_range_elements($self->pipe,
 270                                        indexBuffer,
 271                                        indexSize, minIndex, maxIndex,
 272                                        mode, start, count);
 273    }
 274
 275    void draw_vertices(unsigned prim,
 276                       unsigned num_verts,
 277                       unsigned num_attribs,
 278                       const float *vertices)
 279    {
 280       struct pipe_context *pipe = $self->pipe;
 281       struct pipe_screen *screen = pipe->screen;
 282       struct pipe_buffer *vbuf;
 283       struct pipe_vertex_element velements[PIPE_MAX_ATTRIBS];
 284       struct pipe_vertex_buffer vbuffer;
 285       float *map;
 286       unsigned size;
 287       unsigned i;
 288
 289       size = num_verts * num_attribs * 4 * sizeof(float);
 290
 291       vbuf = pipe_buffer_create(screen,
 292                                 32,
 293                                 PIPE_BUFFER_USAGE_VERTEX,
 294                                 size);
 295       if(!vbuf)
 296          goto error1;
 297
 298       map = pipe_buffer_map(screen, vbuf, PIPE_BUFFER_USAGE_CPU_WRITE);
 299       if (!map)
 300          goto error2;
 301       memcpy(map, vertices, size);
 302       pipe_buffer_unmap(screen, vbuf);
 303
 304       cso_save_vertex_elements($self->cso);
 305
 306       /* tell pipe about the vertex attributes */
 307       for (i = 0; i < num_attribs; i++) {
 308          velements[i].src_offset = i * 4 * sizeof(float);
 309          velements[i].instance_divisor = 0;
 310          velements[i].vertex_buffer_index = 0;
 311          velements[i].src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
 312       }
 313       cso_set_vertex_elements($self->cso, num_attribs, velements);
 314
 315       /* tell pipe about the vertex buffer */
 316       memset(&vbuffer, 0, sizeof(vbuffer));
 317       vbuffer.buffer = vbuf;
 318       vbuffer.stride = num_attribs * 4 * sizeof(float);  /* vertex size */
 319       vbuffer.buffer_offset = 0;
 320       vbuffer.max_index = num_verts - 1;
 321       pipe->set_vertex_buffers(pipe, 1, &vbuffer);
 322
 323       /* draw */
 324       pipe->draw_arrays(pipe, prim, 0, num_verts);
 325
 326       cso_restore_vertex_elements($self->cso);
 327
 328 error2:
 329       pipe_buffer_reference(&vbuf, NULL);
 330 error1:
 331       ;
 332    }
 333
 334    void
 335    clear(unsigned buffers, const float *rgba, double depth = 0.0f,
 336          unsigned stencil = 0)
 337    {
 338       $self->pipe->clear($self->pipe, buffers, rgba, depth, stencil);
 339    }
 340
 341    void
 342    flush(unsigned flags = 0) {
 343       struct pipe_fence_handle *fence = NULL;
 344       $self->pipe->flush($self->pipe, flags | PIPE_FLUSH_RENDER_CACHE, &fence);
 345       if(fence) {
 346          /* TODO: allow asynchronous operation */
 347          $self->pipe->screen->fence_finish( $self->pipe->screen, fence, 0 );
 348          $self->pipe->screen->fence_reference( $self->pipe->screen, &fence, NULL );
 349       }
 350    }
 351
 352    /*
 353     * Surface functions
 354     */
 355
 356    void surface_copy(struct st_surface *dst,
 357                      unsigned destx, unsigned desty,
 358                      struct st_surface *src,
 359                      unsigned srcx, unsigned srcy,
 360                      unsigned width, unsigned height)
 361    {
 362       struct pipe_surface *_dst = NULL;
 363       struct pipe_surface *_src = NULL;
 364
 365       _dst = st_pipe_surface(dst, PIPE_BUFFER_USAGE_GPU_WRITE);
 366       if(!_dst)
 367          SWIG_exception(SWIG_ValueError, "couldn't acquire destination surface for writing");
 368
 369       _src = st_pipe_surface(src, PIPE_BUFFER_USAGE_GPU_READ);
 370       if(!_src)
 371          SWIG_exception(SWIG_ValueError, "couldn't acquire source surface for reading");
 372
 373       $self->pipe->surface_copy($self->pipe, _dst, destx, desty, _src, srcx, srcy, width, height);
 374
 375    fail:
 376       pipe_surface_reference(&_src, NULL);
 377       pipe_surface_reference(&_dst, NULL);
 378    }
 379
 380    void surface_fill(struct st_surface *dst,
 381                      unsigned x, unsigned y,
 382                      unsigned width, unsigned height,
 383                      unsigned value)
 384    {
 385       struct pipe_surface *_dst = NULL;
 386
 387       _dst = st_pipe_surface(dst, PIPE_BUFFER_USAGE_GPU_WRITE);
 388       if(!_dst)
 389          SWIG_exception(SWIG_ValueError, "couldn't acquire destination surface for writing");
 390
 391       $self->pipe->surface_fill($self->pipe, _dst, x, y, width, height, value);
 392
 393    fail:
 394       pipe_surface_reference(&_dst, NULL);
 395    }
 396
 397    %cstring_output_allocate_size(char **STRING, int *LENGTH, free(*$1));
 398    void
 399    surface_read_raw(struct st_surface *surface,
 400                     unsigned x, unsigned y, unsigned w, unsigned h,
 401                     char **STRING, int *LENGTH)
 402    {
 403       struct pipe_texture *texture = surface->texture;
 404       struct pipe_context *pipe = $self->pipe;
 405       struct pipe_transfer *transfer;
 406       unsigned stride;
 407
 408       stride = util_format_get_stride(texture->format, w);
 409       *LENGTH = util_format_get_nblocksy(texture->format, h) * stride;
 410       *STRING = (char *) malloc(*LENGTH);
 411       if(!*STRING)
 412          return;
 413
 414       transfer = pipe->get_tex_transfer(pipe,
 415                                           surface->texture,
 416                                           surface->face,
 417                                           surface->level,
 418                                           surface->zslice,
 419                                           PIPE_TRANSFER_READ,
 420                                           x, y, w, h);
 421       if(transfer) {
 422          pipe_get_tile_raw(pipe, transfer, 0, 0, w, h, *STRING, stride);
 423          pipe->tex_transfer_destroy(pipe, transfer);
 424       }
 425    }
 426
 427    %cstring_input_binary(const char *STRING, unsigned LENGTH);
 428    void
 429    surface_write_raw(struct st_surface *surface,
 430                      unsigned x, unsigned y, unsigned w, unsigned h,
 431                      const char *STRING, unsigned LENGTH, unsigned stride = 0)
 432    {
 433       struct pipe_texture *texture = surface->texture;
 434       struct pipe_context *pipe = $self->pipe;
 435       struct pipe_transfer *transfer;
 436
 437       if(stride == 0)
 438          stride = util_format_get_stride(texture->format, w);
 439
 440       if(LENGTH < util_format_get_nblocksy(texture->format, h) * stride)
 441          SWIG_exception(SWIG_ValueError, "offset must be smaller than buffer size");
 442
 443       transfer = pipe->get_tex_transfer(pipe,
 444                                           surface->texture,
 445                                           surface->face,
 446                                           surface->level,
 447                                           surface->zslice,
 448                                           PIPE_TRANSFER_WRITE,
 449                                           x, y, w, h);
 450       if(!transfer)
 451          SWIG_exception(SWIG_MemoryError, "couldn't initiate transfer");
 452
 453       pipe_put_tile_raw(pipe, transfer, 0, 0, w, h, STRING, stride);
 454       pipe->tex_transfer_destroy(pipe, transfer);
 455
 456    fail:
 457       return;
 458    }
 459
 460    void
 461    surface_read_rgba(struct st_surface *surface,
 462                      unsigned x, unsigned y, unsigned w, unsigned h,
 463                      float *rgba)
 464    {
 465       struct pipe_context *pipe = $self->pipe;
 466       struct pipe_transfer *transfer;
 467       transfer = pipe->get_tex_transfer(pipe,
 468                                           surface->texture,
 469                                           surface->face,
 470                                           surface->level,
 471                                           surface->zslice,
 472                                           PIPE_TRANSFER_READ,
 473                                           x, y, w, h);
 474       if(transfer) {
 475          pipe_get_tile_rgba(pipe, transfer, 0, 0, w, h, rgba);
 476          pipe->tex_transfer_destroy(pipe, transfer);
 477       }
 478    }
 479
 480    void
 481    surface_write_rgba(struct st_surface *surface,
 482                       unsigned x, unsigned y, unsigned w, unsigned h,
 483                       const float *rgba)
 484    {
 485       struct pipe_context *pipe = $self->pipe;
 486       struct pipe_transfer *transfer;
 487       transfer = pipe->get_tex_transfer(pipe,
 488                                           surface->texture,
 489                                           surface->face,
 490                                           surface->level,
 491                                           surface->zslice,
 492                                           PIPE_TRANSFER_WRITE,
 493                                           x, y, w, h);
 494       if(transfer) {
 495          pipe_put_tile_rgba(pipe, transfer, 0, 0, w, h, rgba);
 496          pipe->tex_transfer_destroy(pipe, transfer);
 497       }
 498    }
 499
 500    %cstring_output_allocate_size(char **STRING, int *LENGTH, free(*$1));
 501    void
 502    surface_read_rgba8(struct st_surface *surface,
 503                       unsigned x, unsigned y, unsigned w, unsigned h,
 504                       char **STRING, int *LENGTH)
 505    {
 506       struct pipe_context *pipe = $self->pipe;
 507       struct pipe_transfer *transfer;
 508       float *rgba;
 509       unsigned char *rgba8;
 510       unsigned i, j, k;
 511
 512       *LENGTH = 0;
 513       *STRING = NULL;
 514
 515       if (!surface)
 516          return;
 517
 518       *LENGTH = h*w*4;
 519       *STRING = (char *) malloc(*LENGTH);
 520       if(!*STRING)
 521          return;
 522
 523       rgba = malloc(h*w*4*sizeof(float));
 524       if(!rgba)
 525          return;
 526
 527       rgba8 = (unsigned char *) *STRING;
 528
 529       transfer = pipe->get_tex_transfer(pipe,
 530                                           surface->texture,
 531                                           surface->face,
 532                                           surface->level,
 533                                           surface->zslice,
 534                                           PIPE_TRANSFER_READ,
 535                                           x, y,
 536                                           w, h);
 537       if(transfer) {
 538          pipe_get_tile_rgba(pipe, transfer, 0, 0, w, h, rgba);
 539          for(j = 0; j < h; ++j) {
 540             for(i = 0; i < w; ++i)
 541                for(k = 0; k <4; ++k)
 542                   rgba8[j*w*4 + i*4 + k] = float_to_ubyte(rgba[j*w*4 + i*4 + k]);
 543          }
 544          pipe->tex_transfer_destroy(pipe, transfer);
 545       }
 546
 547       free(rgba);
 548    }
 549
 550    void
 551    surface_read_z(struct st_surface *surface,
 552                   unsigned x, unsigned y, unsigned w, unsigned h,
 553                   unsigned *z)
 554    {
 555       struct pipe_context *pipe = $self->pipe;
 556       struct pipe_transfer *transfer;
 557       transfer = pipe->get_tex_transfer(pipe,
 558                                           surface->texture,
 559                                           surface->face,
 560                                           surface->level,
 561                                           surface->zslice,
 562                                           PIPE_TRANSFER_READ,
 563                                           x, y, w, h);
 564       if(transfer) {
 565          pipe_get_tile_z(pipe, transfer, 0, 0, w, h, z);
 566          pipe->tex_transfer_destroy(pipe, transfer);
 567       }
 568    }
 569
 570    void
 571    surface_write_z(struct st_surface *surface,
 572                    unsigned x, unsigned y, unsigned w, unsigned h,
 573                    const unsigned *z)
 574    {
 575       struct pipe_context *pipe = $self->pipe;
 576       struct pipe_transfer *transfer;
 577       transfer = pipe->get_tex_transfer(pipe,
 578                                           surface->texture,
 579                                           surface->face,
 580                                           surface->level,
 581                                           surface->zslice,
 582                                           PIPE_TRANSFER_WRITE,
 583                                           x, y, w, h);
 584       if(transfer) {
 585          pipe_put_tile_z(pipe, transfer, 0, 0, w, h, z);
 586          pipe->tex_transfer_destroy(pipe, transfer);
 587       }
 588    }
 589
 590    void
 591    surface_sample_rgba(struct st_surface *surface,
 592                        float *rgba,
 593                        int norm = 0)
 594    {
 595       st_sample_surface($self->pipe, surface, rgba, norm != 0);
 596    }
 597
 598    unsigned
 599    surface_compare_rgba(struct st_surface *surface,
 600                         unsigned x, unsigned y, unsigned w, unsigned h,
 601                         const float *rgba, float tol = 0.0)
 602    {
 603       struct pipe_context *pipe = $self->pipe;
 604       struct pipe_transfer *transfer;
 605       float *rgba2;
 606       const float *p1;
 607       const float *p2;
 608       unsigned i, j, n;
 609
 610       rgba2 = MALLOC(h*w*4*sizeof(float));
 611       if(!rgba2)
 612          return ~0;
 613
 614       transfer = pipe->get_tex_transfer(pipe,
 615                                           surface->texture,
 616                                           surface->face,
 617                                           surface->level,
 618                                           surface->zslice,
 619                                           PIPE_TRANSFER_READ,
 620                                           x, y, w, h);
 621       if(!transfer) {
 622          FREE(rgba2);
 623          return ~0;
 624       }
 625
 626       pipe_get_tile_rgba(pipe, transfer, 0, 0, w, h, rgba2);
 627       pipe->tex_transfer_destroy(pipe, transfer);
 628
 629       p1 = rgba;
 630       p2 = rgba2;
 631       n = 0;
 632       for(i = h*w; i; --i) {
 633          unsigned differs = 0;
 634          for(j = 4; j; --j) {
 635             float delta = *p2++ - *p1++;
 636             if (delta < -tol || delta > tol)
 637                 differs = 1;
 638          }
 639          n += differs;
 640       }
 641
 642       FREE(rgba2);
 643
 644       return n;
 645    }
 646
 647 };