| /************************************************************************** |
| * |
| * Copyright 2010 Luca Barbieri |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining |
| * a copy of this software and associated documentation files (the |
| * "Software"), to deal in the Software without restriction, including |
| * without limitation the rights to use, copy, modify, merge, publish, |
| * distribute, sublicense, and/or sell copies of the Software, and to |
| * permit persons to whom the Software is furnished to do so, subject to |
| * the following conditions: |
| * |
| * The above copyright notice and this permission notice (including the |
| * next paragraph) shall be included in all copies or substantial |
| * portions of the Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
| * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
| * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE |
| * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION |
| * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION |
| * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
| * |
| **************************************************************************/ |
| |
| #ifndef DXBC_H_ |
| #define DXBC_H_ |
| |
| #include <stdint.h> |
| #include <vector> |
| #include <map> |
| #include <iostream> |
| #include "le32.h" |
| |
| #define FOURCC(a, b, c, d) ((uint32_t)(uint8_t)(a) | ((uint32_t)(uint8_t)(b) << 8) | ((uint32_t)(uint8_t)(c) << 16) | ((uint32_t)(uint8_t)(d) << 24 )) |
| #define FOURCC_DXBC FOURCC('D', 'X', 'B', 'C') |
| #define FOURCC_RDEF FOURCC('R', 'D', 'E', 'F') |
| #define FOURCC_ISGN FOURCC('I', 'S', 'G', 'N') |
| #define FOURCC_OSGN FOURCC('O', 'S', 'G', 'N') |
| #define FOURCC_SHDR FOURCC('S', 'H', 'D', 'R') |
| #define FOURCC_SHEX FOURCC('S', 'H', 'E', 'X') |
| #define FOURCC_STAT FOURCC('S', 'T', 'A', 'T') |
| #define FOURCC_PCSG FOURCC('P', 'C', 'S', 'G') |
| |
| /* this is always little-endian! */ |
| struct dxbc_chunk_header |
| { |
| unsigned fourcc; |
| unsigned size; |
| }; |
| |
| /* this is always little-endian! */ |
| struct dxbc_chunk_signature : public dxbc_chunk_header |
| { |
| uint32_t count; |
| uint32_t unk; |
| struct |
| { |
| uint32_t name_offset; |
| uint32_t semantic_index; |
| uint32_t system_value_type; |
| uint32_t component_type; |
| uint32_t register_num; |
| uint8_t mask; |
| uint8_t read_write_mask; |
| uint8_t stream; /* TODO: guess! */ |
| uint8_t unused; |
| } elements[]; |
| }; |
| |
| struct dxbc_container |
| { |
| const void* data; |
| std::vector<dxbc_chunk_header*> chunks; |
| std::map<unsigned, unsigned> chunk_map; |
| }; |
| |
| struct dxbc_container_header |
| { |
| unsigned fourcc; |
| uint32_t unk[4]; |
| uint32_t one; |
| uint32_t total_size; |
| uint32_t chunk_count; |
| }; |
| |
| dxbc_container* dxbc_parse(const void* data, int size); |
| std::ostream& operator <<(std::ostream& out, const dxbc_container& container); |
| |
| dxbc_chunk_header* dxbc_find_chunk(const void* data, int size, unsigned fourcc); |
| |
| static inline dxbc_chunk_header* dxbc_find_shader_bytecode(const void* data, int size) |
| { |
| dxbc_chunk_header* chunk; |
| chunk = dxbc_find_chunk(data, size, FOURCC_SHDR); |
| if(!chunk) |
| chunk = dxbc_find_chunk(data, size, FOURCC_SHEX); |
| return chunk; |
| } |
| |
| #define DXBC_FIND_INPUT_SIGNATURE 0 |
| #define DXBC_FIND_OUTPUT_SIGNATURE 1 |
| #define DXBC_FIND_PATCH_SIGNATURE 2 |
| |
| static inline dxbc_chunk_signature* dxbc_find_signature(const void* data, int size, unsigned kind) |
| { |
| unsigned fourcc; |
| switch(kind) { |
| case DXBC_FIND_INPUT_SIGNATURE: fourcc = FOURCC_ISGN; break; |
| case DXBC_FIND_OUTPUT_SIGNATURE: fourcc = FOURCC_OSGN; break; |
| case DXBC_FIND_PATCH_SIGNATURE: fourcc = FOURCC_PCSG; break; |
| default: |
| return NULL; |
| } |
| return (dxbc_chunk_signature*)dxbc_find_chunk(data, size, fourcc); |
| } |
| |
| struct _D3D11_SIGNATURE_PARAMETER_DESC; |
| typedef struct _D3D11_SIGNATURE_PARAMETER_DESC D3D11_SIGNATURE_PARAMETER_DESC; |
| int dxbc_parse_signature(dxbc_chunk_signature* sig, D3D11_SIGNATURE_PARAMETER_DESC** params); |
| |
| std::pair<void*, size_t> dxbc_assemble(struct dxbc_chunk_header** chunks, unsigned num_chunks); |
| |
| #endif /* DXBC_H_ */ |