170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri/************************************************************************** 270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * 370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * Copyright 2010 Luca Barbieri 470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * 570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * Permission is hereby granted, free of charge, to any person obtaining 670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * a copy of this software and associated documentation files (the 770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * "Software"), to deal in the Software without restriction, including 870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * without limitation the rights to use, copy, modify, merge, publish, 970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * distribute, sublicense, and/or sell copies of the Software, and to 1070fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * permit persons to whom the Software is furnished to do so, subject to 1170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * the following conditions: 1270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * 1370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * The above copyright notice and this permission notice (including the 1470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * next paragraph) shall be included in all copies or substantial 1570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * portions of the Software. 1670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * 1770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 1870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 1970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 2070fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * IN NO EVENT SHALL THE COPYRIGHT OWNER(S) AND/OR ITS SUPPLIERS BE 2170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 2270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 2370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 2470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri * 2570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri **************************************************************************/ 2670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 2770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri#include <stdlib.h> 2870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri#include <string.h> 2970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri#include "dxbc.h" 3070fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 3170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieristd::pair<void*, size_t> dxbc_assemble(struct dxbc_chunk_header** chunks, unsigned num_chunks) 3270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri{ 3370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri size_t data_size = 0; 3470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri for(unsigned i = 0; i < num_chunks; ++i) 3570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri data_size += sizeof(uint32_t) + sizeof(dxbc_chunk_header) + bswap_le32(chunks[i]->size); 3670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 3770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri size_t total_size = sizeof(dxbc_container_header) + data_size; 3870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri dxbc_container_header* header = (dxbc_container_header*)malloc(total_size); 3970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri if(!header) 4070fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri return std::make_pair((void*)0, 0); 4170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 4270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri header->fourcc = bswap_le32(FOURCC_DXBC); 4370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri memset(header->unk, 0, sizeof(header->unk)); 4470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri header->one = bswap_le32(1); 4570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri header->total_size = bswap_le32(total_size); 4670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri header->chunk_count = num_chunks; 4770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 4870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri uint32_t* chunk_offsets = (uint32_t*)(header + 1); 4970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri uint32_t off = sizeof(struct dxbc_container_header) + num_chunks * sizeof(uint32_t); 5070fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri for(unsigned i = 0; i < num_chunks; ++i) 5170fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri { 5270fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri chunk_offsets[i] = bswap_le32(off); 5370fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri unsigned chunk_full_size = sizeof(dxbc_chunk_header) + bswap_le32(chunks[i]->size); 5470fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri memcpy((char*)header + off, chunks[i], chunk_full_size); 5570fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri off += chunk_full_size; 5670fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri } 5770fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri 5870fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri return std::make_pair((void*)header, total_size); 5970fed0b0ec8a3ec4f6b9b47f1fe98cc54c6037f0Luca Barbieri} 60