d3dx9: Add swizzle and writemask support to the shader assembler.
[wine.git] / dlls / d3dx9_36 / d3dx9_36_private.h
blob7146f993f11244cd3c82d4b5b794c46fbc78d787
1 /*
2 * Copyright (C) 2002 Raphael Junqueira
3 * Copyright (C) 2008 David Adam
4 * Copyright (C) 2008 Tony Wasserka
5 * Copyright (C) 2008 Stefan Dösinger
6 * Copyright (C) 2009 Matteo Bruni
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #ifndef __WINE_D3DX9_36_PRIVATE_H
25 #define __WINE_D3DX9_36_PRIVATE_H
27 #include <stdarg.h>
29 #define COBJMACROS
30 #include "winbase.h"
31 #include "wingdi.h"
32 #include "winuser.h"
33 #include "d3dx9.h"
35 /* for internal use */
36 typedef enum _FormatType {
37 FORMAT_ARGB, /* unsigned */
38 FORMAT_UNKNOWN
39 } FormatType;
41 typedef struct _PixelFormatDesc {
42 D3DFORMAT format;
43 BYTE bits[4];
44 BYTE shift[4];
45 UINT bytes_per_pixel;
46 FormatType type;
47 } PixelFormatDesc;
49 HRESULT map_view_of_file(LPCWSTR filename, LPVOID *buffer, DWORD *length);
50 HRESULT load_resource_into_memory(HMODULE module, HRSRC resinfo, LPVOID *buffer, DWORD *length);
52 const PixelFormatDesc *get_format_info(D3DFORMAT format);
55 extern const ID3DXBufferVtbl D3DXBuffer_Vtbl;
57 /* ID3DXBUFFER */
58 typedef struct ID3DXBufferImpl
60 /* IUnknown fields */
61 const ID3DXBufferVtbl *lpVtbl;
62 LONG ref;
64 /* ID3DXBuffer fields */
65 DWORD *buffer;
66 DWORD bufferSize;
67 } ID3DXBufferImpl;
70 /* ID3DXFont */
71 typedef struct ID3DXFontImpl
73 /* IUnknown fields */
74 const ID3DXFontVtbl *lpVtbl;
75 LONG ref;
77 /* ID3DXFont fields */
78 IDirect3DDevice9 *device;
79 D3DXFONT_DESCW desc;
81 HDC hdc;
82 HFONT hfont;
83 } ID3DXFontImpl;
85 /* ID3DXMatrixStack */
86 typedef struct ID3DXMatrixStackImpl
88 /* IUnknown fields */
89 const ID3DXMatrixStackVtbl *lpVtbl;
90 LONG ref;
92 /* ID3DXMatrixStack fields */
93 unsigned int current;
94 unsigned int stack_size;
95 D3DXMATRIX *stack;
96 } ID3DXMatrixStackImpl;
98 /*ID3DXSprite */
99 typedef struct _SPRITE {
100 LPDIRECT3DTEXTURE9 texture;
101 UINT texw, texh;
102 RECT rect;
103 D3DXVECTOR3 center;
104 D3DXVECTOR3 pos;
105 D3DCOLOR color;
106 } SPRITE;
108 typedef struct ID3DXSpriteImpl
110 /* IUnknown fields */
111 const ID3DXSpriteVtbl *lpVtbl;
112 LONG ref;
114 /* ID3DXSprite fields */
115 IDirect3DDevice9 *device;
116 IDirect3DVertexDeclaration9 *vdecl;
117 IDirect3DStateBlock9 *stateblock;
118 D3DXMATRIX transform;
119 D3DXMATRIX view;
120 DWORD flags;
121 BOOL ready;
123 /* Store the relevant caps to prevent multiple GetDeviceCaps calls */
124 DWORD texfilter_caps;
125 DWORD maxanisotropy;
126 DWORD alphacmp_caps;
128 SPRITE *sprites;
129 int sprite_count; /* number of sprites to be drawn */
130 int allocated_sprites; /* number of (pre-)allocated sprites */
131 } ID3DXSpriteImpl;
133 /* Shader assembler definitions */
134 typedef enum _shader_type {
135 ST_VERTEX,
136 ST_PIXEL,
137 } shader_type;
139 typedef enum BWRITER_COMPARISON_TYPE {
140 BWRITER_COMPARISON_NONE = 0,
141 } BWRITER_COMPARISON_TYPE;
143 struct shader_reg {
144 DWORD type;
145 DWORD regnum;
146 struct shader_reg *rel_reg;
147 DWORD srcmod;
148 union {
149 DWORD swizzle;
150 DWORD writemask;
154 struct instruction {
155 DWORD opcode;
156 DWORD dstmod;
157 DWORD shift;
158 BWRITER_COMPARISON_TYPE comptype;
159 BOOL has_dst;
160 struct shader_reg dst;
161 struct shader_reg *src;
162 unsigned int num_srcs; /* For freeing the rel_regs */
163 BOOL has_predicate;
164 struct shader_reg predicate;
167 struct declaration {
168 DWORD usage, usage_idx;
169 DWORD regnum;
170 DWORD writemask;
173 struct samplerdecl {
174 DWORD type;
175 DWORD regnum;
176 unsigned int line_no; /* for error messages */
179 #define INSTRARRAY_INITIAL_SIZE 8
180 struct bwriter_shader {
181 shader_type type;
183 /* Shader version selected */
184 DWORD version;
186 /* Local constants. Every constant that is not defined below is loaded from
187 * the global constant set at shader runtime
189 struct constant **constF;
190 struct constant **constI;
191 struct constant **constB;
192 unsigned int num_cf, num_ci, num_cb;
194 /* Declared input and output varyings */
195 struct declaration *inputs, *outputs;
196 unsigned int num_inputs, num_outputs;
197 struct samplerdecl *samplers;
198 unsigned int num_samplers;
200 /* Are special pixel shader 3.0 registers declared? */
201 BOOL vPos, vFace;
203 /* Array of shader instructions - The shader code itself */
204 struct instruction **instr;
205 unsigned int num_instrs, instr_alloc_size;
208 static inline LPVOID asm_alloc(SIZE_T size) {
209 return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, size);
212 static inline LPVOID asm_realloc(LPVOID ptr, SIZE_T size) {
213 return HeapReAlloc(GetProcessHeap(), 0, ptr, size);
216 static inline BOOL asm_free(LPVOID ptr) {
217 return HeapFree(GetProcessHeap(), 0, ptr);
220 struct asm_parser;
222 /* This structure is only used in asmshader.y, but since the .l file accesses the semantic types
223 * too it has to know it as well
225 struct rel_reg {
226 BOOL has_rel_reg;
227 DWORD type;
228 DWORD additional_offset;
229 DWORD rel_regnum;
230 DWORD swizzle;
233 #define MAX_SRC_REGS 4
235 struct src_regs {
236 struct shader_reg reg[MAX_SRC_REGS];
237 unsigned int count;
240 struct asmparser_backend {
241 void (*dstreg)(struct asm_parser *This, struct instruction *instr,
242 const struct shader_reg *dst);
243 void (*srcreg)(struct asm_parser *This, struct instruction *instr, int num,
244 const struct shader_reg *src);
246 void (*predicate)(struct asm_parser *This,
247 const struct shader_reg *predicate);
248 void (*coissue)(struct asm_parser *This);
250 void (*end)(struct asm_parser *This);
252 void (*instr)(struct asm_parser *This, DWORD opcode, DWORD mod, DWORD shift,
253 BWRITER_COMPARISON_TYPE comp, const struct shader_reg *dst,
254 const struct src_regs *srcs, int expectednsrcs);
257 struct instruction *alloc_instr(unsigned int srcs);
258 BOOL add_instruction(struct bwriter_shader *shader, struct instruction *instr);
260 #define MESSAGEBUFFER_INITIAL_SIZE 256
262 struct asm_parser {
263 /* The function table of the parser implementation */
264 const struct asmparser_backend *funcs;
266 /* Private data follows */
267 struct bwriter_shader *shader;
268 unsigned int m3x3pad_count;
270 enum parse_status {
271 PARSE_SUCCESS = 0,
272 PARSE_WARN = 1,
273 PARSE_ERR = 2
274 } status;
275 char *messages;
276 unsigned int messagesize;
277 unsigned int messagecapacity;
278 unsigned int line_no;
281 extern struct asm_parser asm_ctx;
283 void create_vs30_parser(struct asm_parser *ret);
285 struct bwriter_shader *parse_asm_shader(char **messages);
287 #ifdef __GNUC__
288 #define PRINTF_ATTR(fmt,args) __attribute__((format (printf,fmt,args)))
289 #else
290 #define PRINTF_ATTR(fmt,args)
291 #endif
293 void asmparser_message(struct asm_parser *ctx, const char *fmt, ...) PRINTF_ATTR(2,3);
294 void set_parse_status(struct asm_parser *ctx, enum parse_status status);
296 /* A reasonable value as initial size */
297 #define BYTECODEBUFFER_INITIAL_SIZE 32
298 struct bytecode_buffer {
299 DWORD *data;
300 DWORD size;
301 DWORD alloc_size;
302 /* For tracking rare out of memory situations without passing
303 * return values around everywhere
305 HRESULT state;
308 struct bc_writer; /* Predeclaration for use in vtable parameters */
310 typedef void (*instr_writer)(struct bc_writer *This,
311 const struct instruction *instr,
312 struct bytecode_buffer *buffer);
314 struct bytecode_backend {
315 void (*header)(struct bc_writer *This, const struct bwriter_shader *shader,
316 struct bytecode_buffer *buffer);
317 void (*end)(struct bc_writer *This, const struct bwriter_shader *shader,
318 struct bytecode_buffer *buffer);
319 void (*srcreg)(struct bc_writer *This, const struct shader_reg *reg,
320 struct bytecode_buffer *buffer);
321 void (*dstreg)(struct bc_writer *This, const struct shader_reg *reg,
322 struct bytecode_buffer *buffer, DWORD shift, DWORD mod);
323 void (*opcode)(struct bc_writer *This, const struct instruction *instr,
324 DWORD token, struct bytecode_buffer *buffer);
326 const struct instr_handler_table {
327 DWORD opcode;
328 instr_writer func;
329 } *instructions;
332 /* Bytecode writing stuff */
333 struct bc_writer {
334 const struct bytecode_backend *funcs;
336 /* Avoid result checking */
337 HRESULT state;
339 DWORD version;
342 /* Debug utility routines */
343 const char *debug_print_dstreg(const struct shader_reg *reg, shader_type st);
344 const char *debug_print_srcreg(const struct shader_reg *reg, shader_type st);
345 const char *debug_print_swizzle(DWORD swizzle);
346 const char *debug_print_writemask(DWORD mask);
347 const char *debug_print_opcode(DWORD opcode);
349 /* Utilities for internal->d3d constant mapping */
350 DWORD d3d9_swizzle(DWORD bwriter_swizzle);
351 DWORD d3d9_writemask(DWORD bwriter_writemask);
352 DWORD d3d9_register(DWORD bwriter_register);
353 DWORD d3d9_opcode(DWORD bwriter_opcode);
355 /* Used to signal an incorrect swizzle/writemask */
356 #define SWIZZLE_ERR ~0U
359 Enumerations and defines used in the bytecode writer
360 intermediate representation
362 typedef enum _BWRITERSHADER_INSTRUCTION_OPCODE_TYPE {
363 BWRITERSIO_MOV = 1,
365 BWRITERSIO_COMMENT = 0xfffe,
366 BWRITERSIO_END = 0Xffff,
367 } BWRITERSHADER_INSTRUCTION_OPCODE_TYPE;
369 typedef enum _BWRITERSHADER_PARAM_REGISTER_TYPE {
370 BWRITERSPR_TEMP = 0,
371 BWRITERSPR_CONST = 2,
372 } BWRITERSHADER_PARAM_REGISTER_TYPE;
374 #define BWRITERSP_WRITEMASK_0 0x1 /* .x r */
375 #define BWRITERSP_WRITEMASK_1 0x2 /* .y g */
376 #define BWRITERSP_WRITEMASK_2 0x4 /* .z b */
377 #define BWRITERSP_WRITEMASK_3 0x8 /* .w a */
378 #define BWRITERSP_WRITEMASK_ALL 0xf /* all */
380 typedef enum _BWRITERSHADER_PARAM_SRCMOD_TYPE {
381 BWRITERSPSM_NONE = 0,
382 } BWRITERSHADER_PARAM_SRCMOD_TYPE;
384 #define BWRITER_SM1_VS 0xfffe
385 #define BWRITER_SM1_PS 0xffff
387 #define BWRITERPS_VERSION(major, minor) ((BWRITER_SM1_PS << 16) | ((major) << 8) | (minor))
388 #define BWRITERVS_VERSION(major, minor) ((BWRITER_SM1_VS << 16) | ((major) << 8) | (minor))
390 #define BWRITERVS_SWIZZLE_SHIFT 16
391 #define BWRITERVS_SWIZZLE_MASK (0xFF << BWRITERVS_SWIZZLE_SHIFT)
393 #define BWRITERVS_X_X (0 << BWRITERVS_SWIZZLE_SHIFT)
394 #define BWRITERVS_X_Y (1 << BWRITERVS_SWIZZLE_SHIFT)
395 #define BWRITERVS_X_Z (2 << BWRITERVS_SWIZZLE_SHIFT)
396 #define BWRITERVS_X_W (3 << BWRITERVS_SWIZZLE_SHIFT)
398 #define BWRITERVS_Y_X (0 << (BWRITERVS_SWIZZLE_SHIFT + 2))
399 #define BWRITERVS_Y_Y (1 << (BWRITERVS_SWIZZLE_SHIFT + 2))
400 #define BWRITERVS_Y_Z (2 << (BWRITERVS_SWIZZLE_SHIFT + 2))
401 #define BWRITERVS_Y_W (3 << (BWRITERVS_SWIZZLE_SHIFT + 2))
403 #define BWRITERVS_Z_X (0 << (BWRITERVS_SWIZZLE_SHIFT + 4))
404 #define BWRITERVS_Z_Y (1 << (BWRITERVS_SWIZZLE_SHIFT + 4))
405 #define BWRITERVS_Z_Z (2 << (BWRITERVS_SWIZZLE_SHIFT + 4))
406 #define BWRITERVS_Z_W (3 << (BWRITERVS_SWIZZLE_SHIFT + 4))
408 #define BWRITERVS_W_X (0 << (BWRITERVS_SWIZZLE_SHIFT + 6))
409 #define BWRITERVS_W_Y (1 << (BWRITERVS_SWIZZLE_SHIFT + 6))
410 #define BWRITERVS_W_Z (2 << (BWRITERVS_SWIZZLE_SHIFT + 6))
411 #define BWRITERVS_W_W (3 << (BWRITERVS_SWIZZLE_SHIFT + 6))
413 #define BWRITERVS_NOSWIZZLE (BWRITERVS_X_X | BWRITERVS_Y_Y | BWRITERVS_Z_Z | BWRITERVS_W_W)
415 #define BWRITERVS_SWIZZLE_X (BWRITERVS_X_X | BWRITERVS_Y_X | BWRITERVS_Z_X | BWRITERVS_W_X)
416 #define BWRITERVS_SWIZZLE_Y (BWRITERVS_X_Y | BWRITERVS_Y_Y | BWRITERVS_Z_Y | BWRITERVS_W_Y)
417 #define BWRITERVS_SWIZZLE_Z (BWRITERVS_X_Z | BWRITERVS_Y_Z | BWRITERVS_Z_Z | BWRITERVS_W_Z)
418 #define BWRITERVS_SWIZZLE_W (BWRITERVS_X_W | BWRITERVS_Y_W | BWRITERVS_Z_W | BWRITERVS_W_W)
420 struct bwriter_shader *SlAssembleShader(const char *text, char **messages);
421 DWORD SlWriteBytecode(const struct bwriter_shader *shader, int dxversion, DWORD **result);
422 void SlDeleteShader(struct bwriter_shader *shader);
424 #endif /* __WINE_D3DX9_36_PRIVATE_H */