← Back to branch summary

~ubuntu-branches/ubuntu/raring/mesa/raring-proposed

« back to all changes in this revision

Viewing changes to src/gallium/tests/graw/fragment-shader/fragment-shader.py

  • Committer: Bazaar Package Importer
  • Author(s): Cyril Brulebois
  • Date: 2011-06-19 21:26:00 UTC
  • mfrom: (1.6.1 upstream) (3.3.18 sid)
  • mto: (3.3.20 sid)
  • mto: This revision was merged to the branch mainline in revision 145.
  • Revision ID: james.westby@ubuntu.com-20110619212600-rleaapdmnbtstekb
Tags: 7.11~0-2
Thank you sbuild for giving a green light when that's not actually the
case. Fix missing Pre-Depends for the libegl1-mesa-drivers package
(multiarch-support).

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/gallium/tests/graw/fragment-shader/fragment-shader.py
1
 
#!/usr/bin/env python
2
 
##########################################################################
3
 
# 
4
 
# Copyright 2009 VMware, Inc.
5
 
# All Rights Reserved.
6
 
# 
7
 
# Permission is hereby granted, free of charge, to any person obtaining a
8
 
# copy of this software and associated documentation files (the
9
 
# "Software"), to deal in the Software without restriction, including
10
 
# without limitation the rights to use, copy, modify, merge, publish,
11
 
# distribute, sub license, and/or sell copies of the Software, and to
12
 
# permit persons to whom the Software is furnished to do so, subject to
13
 
# the following conditions:
14
 
# 
15
 
# The above copyright notice and this permission notice (including the
16
 
# next paragraph) shall be included in all copies or substantial portions
17
 
# of the Software.
18
 
# 
19
 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
20
 
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
21
 
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
22
 
# IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
23
 
# ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
24
 
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
25
 
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
26
 
# 
27
 
##########################################################################
28
 
 
29
 
import struct
30
 
 
31
 
from gallium import *
32
 
 
33
 
def make_image(surface):
34
 
    data = surface.get_tile_rgba8(0, 0, surface.width, surface.height)
35
 
 
36
 
    import Image
37
 
    outimage = Image.fromstring('RGBA', (surface.width, surface.height), data, "raw", 'RGBA', 0, 1)
38
 
    return outimage
39
 
 
40
 
def save_image(filename, surface):
41
 
    outimage = make_image(surface)
42
 
    outimage.save(filename, "PNG")
43
 
 
44
 
def test(dev, name):
45
 
    ctx = dev.context_create()
46
 
 
47
 
    width = 320
48
 
    height = 320
49
 
    minz = 0.0
50
 
    maxz = 1.0
51
 
 
52
 
    # disabled blending/masking
53
 
    blend = Blend()
54
 
    blend.rt[0].rgb_src_factor = PIPE_BLENDFACTOR_ONE
55
 
    blend.rt[0].alpha_src_factor = PIPE_BLENDFACTOR_ONE
56
 
    blend.rt[0].rgb_dst_factor = PIPE_BLENDFACTOR_ZERO
57
 
    blend.rt[0].alpha_dst_factor = PIPE_BLENDFACTOR_ZERO
58
 
    blend.rt[0].colormask = PIPE_MASK_RGBA
59
 
    ctx.set_blend(blend)
60
 
 
61
 
    # depth/stencil/alpha
62
 
    depth_stencil_alpha = DepthStencilAlpha()
63
 
    depth_stencil_alpha.depth.enabled = 0
64
 
    depth_stencil_alpha.depth.writemask = 1
65
 
    depth_stencil_alpha.depth.func = PIPE_FUNC_LESS
66
 
    ctx.set_depth_stencil_alpha(depth_stencil_alpha)
67
 
 
68
 
    # rasterizer
69
 
    rasterizer = Rasterizer()
70
 
    rasterizer.front_winding = PIPE_WINDING_CW
71
 
    rasterizer.cull_mode = PIPE_WINDING_NONE
72
 
    rasterizer.scissor = 1
73
 
    ctx.set_rasterizer(rasterizer)
74
 
 
75
 
    # viewport
76
 
    viewport = Viewport()
77
 
    scale = FloatArray(4)
78
 
    scale[0] = width / 2.0
79
 
    scale[1] = -height / 2.0
80
 
    scale[2] = (maxz - minz) / 2.0
81
 
    scale[3] = 1.0
82
 
    viewport.scale = scale
83
 
    translate = FloatArray(4)
84
 
    translate[0] = width / 2.0
85
 
    translate[1] = height / 2.0
86
 
    translate[2] = (maxz - minz) / 2.0
87
 
    translate[3] = 0.0
88
 
    viewport.translate = translate
89
 
    ctx.set_viewport(viewport)
90
 
 
91
 
    # samplers
92
 
    sampler = Sampler()
93
 
    sampler.wrap_s = PIPE_TEX_WRAP_CLAMP_TO_EDGE
94
 
    sampler.wrap_t = PIPE_TEX_WRAP_CLAMP_TO_EDGE
95
 
    sampler.wrap_r = PIPE_TEX_WRAP_CLAMP_TO_EDGE
96
 
    sampler.min_mip_filter = PIPE_TEX_MIPFILTER_NONE
97
 
    sampler.min_img_filter = PIPE_TEX_MIPFILTER_NEAREST
98
 
    sampler.mag_img_filter = PIPE_TEX_MIPFILTER_NEAREST
99
 
    sampler.normalized_coords = 1
100
 
    ctx.set_fragment_sampler(0, sampler)
101
 
 
102
 
    # scissor
103
 
    scissor = Scissor()
104
 
    scissor.minx = 0
105
 
    scissor.miny = 0
106
 
    scissor.maxx = width
107
 
    scissor.maxy = height
108
 
    ctx.set_scissor(scissor)
109
 
 
110
 
    clip = Clip()
111
 
    clip.nr = 0
112
 
    ctx.set_clip(clip)
113
 
 
114
 
    # framebuffer
115
 
    cbuf = dev.resource_create(
116
 
        PIPE_FORMAT_B8G8R8X8_UNORM,
117
 
        width, height,
118
 
        bind=PIPE_BIND_RENDER_TARGET,
119
 
    ).get_surface()
120
 
    fb = Framebuffer()
121
 
    fb.width = width
122
 
    fb.height = height
123
 
    fb.nr_cbufs = 1
124
 
    fb.set_cbuf(0, cbuf)
125
 
    ctx.set_framebuffer(fb)
126
 
    rgba = FloatArray(4);
127
 
    rgba[0] = 0.5
128
 
    rgba[1] = 0.5
129
 
    rgba[2] = 0.5
130
 
    rgba[3] = 0.5
131
 
    ctx.clear(PIPE_CLEAR_COLOR, rgba, 0.0, 0)
132
 
 
133
 
    # vertex shader
134
 
    vs = Shader('''
135
 
        VERT
136
 
        DCL IN[0], POSITION
137
 
        DCL IN[1], COLOR
138
 
        DCL OUT[0], POSITION
139
 
        DCL OUT[1], COLOR
140
 
        MOV OUT[0], IN[0]
141
 
        MOV OUT[1], IN[1]
142
 
        END
143
 
    ''')
144
 
    ctx.set_vertex_shader(vs)
145
 
 
146
 
    # fragment shader
147
 
    fs = Shader(file('frag-' + name + '.sh', 'rt').read())
148
 
    ctx.set_fragment_shader(fs)
149
 
 
150
 
    constbuf0 = dev.buffer_create(64,
151
 
                                  (PIPE_BUFFER_USAGE_CONSTANT |
152
 
                                   PIPE_BUFFER_USAGE_GPU_READ |
153
 
                                   PIPE_BUFFER_USAGE_CPU_WRITE),
154
 
                                  4 * 4 * 4)
155
 
 
156
 
    cbdata = ''
157
 
    cbdata += struct.pack('4f', 0.4, 0.0, 0.0, 1.0)
158
 
    cbdata += struct.pack('4f', 1.0, 1.0, 1.0, 1.0)
159
 
    cbdata += struct.pack('4f', 2.0, 2.0, 2.0, 2.0)
160
 
    cbdata += struct.pack('4f', 4.0, 8.0, 16.0, 32.0)
161
 
 
162
 
    constbuf0.write(cbdata, 0)
163
 
 
164
 
    ctx.set_constant_buffer(PIPE_SHADER_FRAGMENT,
165
 
                            0,
166
 
                            constbuf0)
167
 
 
168
 
    constbuf1 = dev.buffer_create(64,
169
 
                                  (PIPE_BUFFER_USAGE_CONSTANT |
170
 
                                   PIPE_BUFFER_USAGE_GPU_READ |
171
 
                                   PIPE_BUFFER_USAGE_CPU_WRITE),
172
 
                                  4 * 4 * 4)
173
 
 
174
 
    cbdata = ''
175
 
    cbdata += struct.pack('4f', 0.1, 0.1, 0.1, 0.1)
176
 
    cbdata += struct.pack('4f', 0.25, 0.25, 0.25, 0.25)
177
 
    cbdata += struct.pack('4f', 0.5, 0.5, 0.5, 0.5)
178
 
    cbdata += struct.pack('4f', 0.75, 0.75, 0.75, 0.75)
179
 
 
180
 
    constbuf1.write(cbdata, 0)
181
 
 
182
 
    ctx.set_constant_buffer(PIPE_SHADER_FRAGMENT,
183
 
                            1,
184
 
                            constbuf1)
185
 
 
186
 
    xy = [
187
 
        -0.8, -0.8,
188
 
         0.8, -0.8,
189
 
         0.0,  0.8,
190
 
    ]
191
 
    color = [
192
 
        1.0, 0.0, 0.0,
193
 
        0.0, 1.0, 0.0,
194
 
        0.0, 0.0, 1.0,
195
 
    ]
196
 
 
197
 
    nverts = 3
198
 
    nattrs = 2
199
 
    verts = FloatArray(nverts * nattrs * 4)
200
 
 
201
 
    for i in range(0, nverts):
202
 
        verts[i * nattrs * 4 + 0] = xy[i * 2 + 0] # x
203
 
        verts[i * nattrs * 4 + 1] = xy[i * 2 + 1] # y
204
 
        verts[i * nattrs * 4 + 2] = 0.5 # z
205
 
        verts[i * nattrs * 4 + 3] = 1.0 # w
206
 
        verts[i * nattrs * 4 + 4] = color[i * 3 + 0] # r
207
 
        verts[i * nattrs * 4 + 5] = color[i * 3 + 1] # g
208
 
        verts[i * nattrs * 4 + 6] = color[i * 3 + 2] # b
209
 
        verts[i * nattrs * 4 + 7] = 1.0 # a
210
 
 
211
 
    ctx.draw_vertices(PIPE_PRIM_TRIANGLES,
212
 
                      nverts,
213
 
                      nattrs,
214
 
                      verts)
215
 
 
216
 
    ctx.flush()
217
 
 
218
 
    save_image('frag-' + name + '.png', cbuf)
219
 
 
220
 
def main():
221
 
    tests = [
222
 
        'abs',
223
 
        'add',
224
 
        'cb-1d',
225
 
        'cb-2d',
226
 
        'dp3',
227
 
        'dp4',
228
 
        'dst',
229
 
        'ex2',
230
 
        'flr',
231
 
        'frc',
232
 
        'lg2',
233
 
        'lit',
234
 
        'lrp',
235
 
        'mad',
236
 
        'max',
237
 
        'min',
238
 
        'mov',
239
 
        'mul',
240
 
        'rcp',
241
 
        'rsq',
242
 
        'sge',
243
 
        'slt',
244
 
        'srcmod-abs',
245
 
        'srcmod-absneg',
246
 
        'srcmod-neg',
247
 
        'srcmod-swz',
248
 
        'sub',
249
 
        'xpd',
250
 
    ]
251
 
 
252
 
    dev = Device()
253
 
    for t in tests:
254
 
        test(dev, t)
255
 
 
256
 
if __name__ == '__main__':
257
 
    main()