dlls/d3dx9_36/d3dx9_private.h

   1 /*
   2  * Copyright (C) 2002 Raphael Junqueira
   3  * Copyright (C) 2008 David Adam
   4  * Copyright (C) 2008 Tony Wasserka
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  *
  20  */
  21
  22 #ifndef __WINE_D3DX9_PRIVATE_H
  23 #define __WINE_D3DX9_PRIVATE_H
  24
  25 #include <stdint.h>
  26 #include "wine/debug.h"
  27 #include "wine/heap.h"
  28 #include "wine/rbtree.h"
  29
  30 #define COBJMACROS
  31 #include "d3dx9.h"
  32
  33 #define ULONG64_MAX (~(ULONG64)0)
  34
  35 struct vec4
  36 {
  37     float x, y, z, w;
  38 };
  39
  40 struct volume
  41 {
  42     UINT width;
  43     UINT height;
  44     UINT depth;
  45 };
  46
  47 /* for internal use */
  48 enum format_type {
  49     FORMAT_ARGB,   /* unsigned */
  50     FORMAT_ARGBF16,/* float 16 */
  51     FORMAT_ARGBF,  /* float */
  52     FORMAT_DXT,
  53     FORMAT_INDEX,
  54     FORMAT_UNKNOWN
  55 };
  56
  57 struct pixel_format_desc {
  58     D3DFORMAT format;
  59     BYTE bits[4];
  60     BYTE shift[4];
  61     UINT bytes_per_pixel;
  62     UINT block_width;
  63     UINT block_height;
  64     UINT block_byte_count;
  65     enum format_type type;
  66     void (*from_rgba)(const struct vec4 *src, struct vec4 *dst);
  67     void (*to_rgba)(const struct vec4 *src, struct vec4 *dst, const PALETTEENTRY *palette);
  68 };
  69
  70 struct d3dx_include_from_file
  71 {
  72     ID3DXInclude ID3DXInclude_iface;
  73 };
  74
  75 extern CRITICAL_SECTION from_file_mutex;
  76 extern const struct ID3DXIncludeVtbl d3dx_include_from_file_vtbl;
  77
  78 static inline BOOL is_conversion_from_supported(const struct pixel_format_desc *format)
  79 {
  80     if (format->type == FORMAT_ARGB || format->type == FORMAT_ARGBF16
  81             || format->type == FORMAT_ARGBF || format->type == FORMAT_DXT)
  82         return TRUE;
  83     return !!format->to_rgba;
  84 }
  85
  86 static inline BOOL is_conversion_to_supported(const struct pixel_format_desc *format)
  87 {
  88     if (format->type == FORMAT_ARGB || format->type == FORMAT_ARGBF16
  89             || format->type == FORMAT_ARGBF || format->type == FORMAT_DXT)
  90         return TRUE;
  91     return !!format->from_rgba;
  92 }
  93
  94 HRESULT map_view_of_file(const WCHAR *filename, void **buffer, DWORD *length);
  95 HRESULT load_resource_into_memory(HMODULE module, HRSRC resinfo, void **buffer, DWORD *length);
  96
  97 HRESULT write_buffer_to_file(const WCHAR *filename, ID3DXBuffer *buffer);
  98
  99 const struct pixel_format_desc *get_format_info(D3DFORMAT format);
 100 const struct pixel_format_desc *get_format_info_idx(int idx);
 101
 102 void format_to_vec4(const struct pixel_format_desc *format, const BYTE *src, struct vec4 *dst);
 103
 104 void copy_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch,
 105     BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch, const struct volume *size,
 106     const struct pixel_format_desc *format);
 107 void convert_argb_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch,
 108     const struct volume *src_size, const struct pixel_format_desc *src_format,
 109     BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch, const struct volume *dst_size,
 110     const struct pixel_format_desc *dst_format, D3DCOLOR color_key, const PALETTEENTRY *palette);
 111 void point_filter_argb_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch,
 112     const struct volume *src_size, const struct pixel_format_desc *src_format,
 113     BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch, const struct volume *dst_size,
 114     const struct pixel_format_desc *dst_format, D3DCOLOR color_key, const PALETTEENTRY *palette);
 115
 116 HRESULT load_texture_from_dds(IDirect3DTexture9 *texture, const void *src_data, const PALETTEENTRY *palette,
 117         DWORD filter, D3DCOLOR color_key, const D3DXIMAGE_INFO *src_info, unsigned int skip_levels,
 118         unsigned int *loaded_miplevels);
 119 HRESULT load_cube_texture_from_dds(IDirect3DCubeTexture9 *cube_texture, const void *src_data,
 120     const PALETTEENTRY *palette, DWORD filter, D3DCOLOR color_key, const D3DXIMAGE_INFO *src_info);
 121 HRESULT load_volume_from_dds(IDirect3DVolume9 *dst_volume, const PALETTEENTRY *dst_palette,
 122     const D3DBOX *dst_box, const void *src_data, const D3DBOX *src_box, DWORD filter, D3DCOLOR color_key,
 123     const D3DXIMAGE_INFO *src_info);
 124 HRESULT load_volume_texture_from_dds(IDirect3DVolumeTexture9 *volume_texture, const void *src_data,
 125     const PALETTEENTRY *palette, DWORD filter, DWORD color_key, const D3DXIMAGE_INFO *src_info);
 126 HRESULT lock_surface(IDirect3DSurface9 *surface, const RECT *surface_rect, D3DLOCKED_RECT *lock,
 127         IDirect3DSurface9 **temp_surface, BOOL write);
 128 HRESULT unlock_surface(IDirect3DSurface9 *surface, const RECT *surface_rect,
 129         IDirect3DSurface9 *temp_surface, BOOL update);
 130
 131 unsigned short float_32_to_16(const float in);
 132 float float_16_to_32(const unsigned short in);
 133
 134 /* debug helpers */
 135 const char *debug_d3dxparameter_class(D3DXPARAMETER_CLASS c);
 136 const char *debug_d3dxparameter_type(D3DXPARAMETER_TYPE t);
 137 const char *debug_d3dxparameter_registerset(D3DXREGISTER_SET r);
 138
 139 /* parameter type conversion helpers */
 140 static inline BOOL get_bool(D3DXPARAMETER_TYPE type, const void *data)
 141 {
 142     switch (type)
 143     {
 144         case D3DXPT_FLOAT:
 145         case D3DXPT_INT:
 146         case D3DXPT_BOOL:
 147             return !!*(DWORD *)data;
 148
 149         case D3DXPT_VOID:
 150             return *(BOOL *)data;
 151
 152         default:
 153             return FALSE;
 154     }
 155 }
 156
 157 static inline int get_int(D3DXPARAMETER_TYPE type, const void *data)
 158 {
 159     switch (type)
 160     {
 161         case D3DXPT_FLOAT:
 162             return (int)(*(float *)data);
 163
 164         case D3DXPT_INT:
 165         case D3DXPT_VOID:
 166             return *(int *)data;
 167
 168         case D3DXPT_BOOL:
 169             return get_bool(type, data);
 170
 171         default:
 172             return 0;
 173     }
 174 }
 175
 176 static inline float get_float(D3DXPARAMETER_TYPE type, const void *data)
 177 {
 178     switch (type)
 179     {
 180         case D3DXPT_FLOAT:
 181         case D3DXPT_VOID:
 182             return *(float *)data;
 183
 184         case D3DXPT_INT:
 185             return (float)(*(int *)data);
 186
 187         case D3DXPT_BOOL:
 188             return (float)get_bool(type, data);
 189
 190         default:
 191             return 0.0f;
 192     }
 193 }
 194
 195 static inline void set_number(void *outdata, D3DXPARAMETER_TYPE outtype, const void *indata, D3DXPARAMETER_TYPE intype)
 196 {
 197     if (outtype == intype)
 198     {
 199         *(DWORD *)outdata = *(DWORD *)indata;
 200         return;
 201     }
 202
 203     switch (outtype)
 204     {
 205         case D3DXPT_FLOAT:
 206             *(float *)outdata = get_float(intype, indata);
 207             break;
 208
 209         case D3DXPT_BOOL:
 210             *(BOOL *)outdata = get_bool(intype, indata);
 211             break;
 212
 213         case D3DXPT_INT:
 214             *(int *)outdata = get_int(intype, indata);
 215             break;
 216
 217         default:
 218             *(DWORD *)outdata = 0;
 219             break;
 220     }
 221 }
 222
 223 static inline BOOL is_param_type_sampler(D3DXPARAMETER_TYPE type)
 224 {
 225     return type == D3DXPT_SAMPLER
 226             || type == D3DXPT_SAMPLER1D || type == D3DXPT_SAMPLER2D
 227             || type == D3DXPT_SAMPLER3D || type == D3DXPT_SAMPLERCUBE;
 228 }
 229
 230 /* Returns the smallest power of 2 which is greater than or equal to num */
 231 static inline uint32_t make_pow2(uint32_t num)
 232 {
 233     DWORD index;
 234
 235     return BitScanReverse(&index, num - 1) ? 1u << (index + 1) : 1;
 236 }
 237
 238 struct d3dx_parameter;
 239
 240 enum pres_reg_tables
 241 {
 242     PRES_REGTAB_IMMED,
 243     PRES_REGTAB_CONST,
 244     PRES_REGTAB_INPUT,
 245     PRES_REGTAB_OCONST,
 246     PRES_REGTAB_OBCONST,
 247     PRES_REGTAB_OICONST,
 248     PRES_REGTAB_TEMP,
 249     PRES_REGTAB_COUNT,
 250     PRES_REGTAB_FIRST_SHADER = PRES_REGTAB_CONST,
 251 };
 252
 253 struct d3dx_const_param_eval_output
 254 {
 255     struct d3dx_parameter *param;
 256     enum pres_reg_tables table;
 257     enum D3DXPARAMETER_CLASS constant_class;
 258     unsigned int register_index;
 259     unsigned int register_count;
 260     BOOL direct_copy;
 261     unsigned int element_count;
 262 };
 263
 264 struct d3dx_const_tab
 265 {
 266     unsigned int input_count;
 267     D3DXCONSTANT_DESC *inputs;
 268     struct d3dx_parameter **inputs_param;
 269     unsigned int const_set_count;
 270     unsigned int const_set_size;
 271     struct d3dx_const_param_eval_output *const_set;
 272     const enum pres_reg_tables *regset2table;
 273     ULONG64 update_version;
 274 };
 275
 276 struct d3dx_regstore
 277 {
 278     void *tables[PRES_REGTAB_COUNT];
 279     unsigned int table_sizes[PRES_REGTAB_COUNT]; /* registers count */
 280 };
 281
 282 struct d3dx_pres_ins;
 283
 284 struct d3dx_preshader
 285 {
 286     struct d3dx_regstore regs;
 287
 288     unsigned int ins_count;
 289     struct d3dx_pres_ins *ins;
 290
 291     struct d3dx_const_tab inputs;
 292 };
 293
 294 struct d3dx_param_eval
 295 {
 296     D3DXPARAMETER_TYPE param_type;
 297
 298     struct d3dx_preshader pres;
 299     struct d3dx_const_tab shader_inputs;
 300
 301     ULONG64 *version_counter;
 302 };
 303
 304 struct param_rb_entry
 305 {
 306     struct wine_rb_entry entry;
 307     char *full_name;
 308     struct d3dx_parameter *param;
 309 };
 310
 311 struct d3dx_shared_data;
 312 struct d3dx_top_level_parameter;
 313
 314 struct d3dx_parameter
 315 {
 316     char magic_string[4];
 317     struct d3dx_top_level_parameter *top_level_param;
 318     struct d3dx_param_eval *param_eval;
 319     char *name;
 320     void *data;
 321     D3DXPARAMETER_CLASS class;
 322     D3DXPARAMETER_TYPE  type;
 323     unsigned int rows;
 324     unsigned int columns;
 325     unsigned int element_count;
 326     unsigned int member_count;
 327     uint32_t flags;
 328     unsigned int bytes;
 329     unsigned int object_id;
 330
 331     struct d3dx_parameter *members;
 332     char *semantic;
 333
 334     char *full_name;
 335     struct wine_rb_entry rb_entry;
 336 };
 337
 338 struct d3dx_top_level_parameter
 339 {
 340     struct d3dx_parameter param;
 341     unsigned int annotation_count;
 342     struct d3dx_parameter *annotations;
 343     ULONG64 update_version;
 344     ULONG64 *version_counter;
 345     struct d3dx_shared_data *shared_data;
 346 };
 347
 348 struct d3dx_shared_data
 349 {
 350     void *data;
 351     struct d3dx_top_level_parameter **parameters;
 352     unsigned int size, count;
 353     ULONG64 update_version;
 354 };
 355
 356 struct d3dx_effect;
 357
 358 static inline BOOL is_top_level_parameter(struct d3dx_parameter *param)
 359 {
 360     return &param->top_level_param->param == param;
 361 }
 362
 363 static inline struct d3dx_top_level_parameter
 364         *top_level_parameter_from_parameter(struct d3dx_parameter *param)
 365 {
 366     return CONTAINING_RECORD(param, struct d3dx_top_level_parameter, param);
 367 }
 368
 369 static inline ULONG64 next_update_version(ULONG64 *version_counter)
 370 {
 371     return ++*version_counter;
 372 }
 373
 374 static inline BOOL is_top_level_param_dirty(struct d3dx_top_level_parameter *param, ULONG64 update_version)
 375 {
 376     struct d3dx_shared_data *shared_data;
 377
 378     if ((shared_data = param->shared_data))
 379         return update_version < shared_data->update_version;
 380     else
 381         return update_version < param->update_version;
 382 }
 383
 384 static inline BOOL is_param_dirty(struct d3dx_parameter *param, ULONG64 update_version)
 385 {
 386     return is_top_level_param_dirty(param->top_level_param, update_version);
 387 }
 388
 389 struct d3dx_parameters_store
 390 {
 391     struct wine_rb_tree tree;
 392     struct d3dx_top_level_parameter *parameters;
 393     unsigned int count;
 394
 395     char *full_name_tmp;
 396     unsigned int full_name_tmp_size;
 397 };
 398
 399 HRESULT d3dx_init_parameters_store(struct d3dx_parameters_store *store, unsigned int count);
 400 void d3dx_parameters_store_cleanup(struct d3dx_parameters_store *store);
 401 struct d3dx_parameter *get_parameter_by_name(struct d3dx_parameters_store *store,
 402         struct d3dx_parameter *parameter, const char *name);
 403
 404 #define SET_D3D_STATE_(manager, device, method, ...) (manager ? manager->lpVtbl->method(manager, __VA_ARGS__) \
 405         : device->lpVtbl->method(device, __VA_ARGS__))
 406 #define SET_D3D_STATE(base_effect, ...) SET_D3D_STATE_(base_effect->manager, base_effect->device, __VA_ARGS__)
 407
 408 HRESULT d3dx_create_param_eval(struct d3dx_parameters_store *parameters, void *byte_code,
 409         unsigned int byte_code_size, D3DXPARAMETER_TYPE type,
 410         struct d3dx_param_eval **peval, ULONG64 *version_counter,
 411         const char **skip_constants, unsigned int skip_constants_count);
 412 void d3dx_free_param_eval(struct d3dx_param_eval *peval);
 413 HRESULT d3dx_evaluate_parameter(struct d3dx_param_eval *peval,
 414         const struct d3dx_parameter *param, void *param_value);
 415 HRESULT d3dx_param_eval_set_shader_constants(ID3DXEffectStateManager *manager, struct IDirect3DDevice9 *device,
 416         struct d3dx_param_eval *peval, BOOL update_all);
 417 BOOL is_param_eval_input_dirty(struct d3dx_param_eval *peval, ULONG64 update_version);
 418
 419 struct ctab_constant {
 420     D3DXCONSTANT_DESC desc;
 421     WORD constantinfo_reserved;
 422     struct ctab_constant *constants;
 423 };
 424
 425 const struct ctab_constant *d3dx_shader_get_ctab_constant(ID3DXConstantTable *iface,
 426         D3DXHANDLE constant);
 427
 428 #endif /* __WINE_D3DX9_PRIVATE_H */