From b0bdc18c270d34598d95b2e8ca917783204fb93f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Branimir=20Karad=C5=BEi=C4=87?= Date: Wed, 19 Apr 2017 21:13:03 -0700 Subject: [PATCH] texturec: Synced with bgfx code. --- tools/texturec/texturec.cpp | 649 ++++++++++++++++++++---------------- 1 file changed, 360 insertions(+), 289 deletions(-) diff --git a/tools/texturec/texturec.cpp b/tools/texturec/texturec.cpp index c938cf0..1898c9a 100644 --- a/tools/texturec/texturec.cpp +++ b/tools/texturec/texturec.cpp @@ -20,6 +20,320 @@ #include #include +struct Options +{ + Options() + : maxSize(UINT32_MAX) + , edge(0.0f) + , format(bimg::TextureFormat::Count) + , mips(false) + , normalMap(false) + , iqa(false) + , sdf(false) + { + } + + uint32_t maxSize; + float edge; + bimg::TextureFormat::Enum format; + bool mips; + bool normalMap; + bool iqa; + bool sdf; +}; + +bimg::ImageContainer* convert(bx::AllocatorI* _allocator, const void* _inputData, uint32_t _inputSize, const Options& _options) +{ + const uint8_t* inputData = (uint8_t*)_inputData; + + bimg::ImageContainer* output = NULL; + bimg::ImageContainer* input = bimg::imageParse(_allocator, inputData, _inputSize); + + if (NULL != input) + { + bimg::TextureFormat::Enum format = input->m_format; + + if (bimg::TextureFormat::Count != _options.format) + { + format = _options.format; + } + + if (input->m_width > _options.maxSize + || input->m_height > _options.maxSize) + { + bimg::ImageContainer* src = bimg::imageConvert(_allocator, bimg::TextureFormat::RGBA32F, *input); + + uint32_t width; + uint32_t height; + + if (input->m_width > input->m_height) + { + width = _options.maxSize; + height = input->m_height * width / input->m_width; + } + else + { + height = _options.maxSize; + width = input->m_width * height / input->m_height; + } + + bimg::ImageContainer* dst = bimg::imageAlloc(_allocator + , bimg::TextureFormat::RGBA32F + , uint16_t(width) + , uint16_t(height) + , 1, 1, false, false + ); + + bimg::imageResizeRgba32fLinear(dst, src); + + bimg::imageFree(src); + bimg::imageFree(input); + + input = bimg::imageConvert(_allocator, format, *dst); + bimg::imageFree(dst); + } + + bimg::ImageMip mip; + if (bimg::imageGetRawData(*input, 0, 0, input->m_data, input->m_size, mip) ) + { + uint8_t numMips = _options.mips + ? bimg::imageGetNumMips(format, uint16_t(mip.m_width), uint16_t(mip.m_height) ) + : 1 + ; + + void* temp = NULL; + + if (_options.normalMap) + { + output = bimg::imageAlloc(_allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, _options.mips); + + bimg::ImageMip dstMip; + bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); + + if (mip.m_width != dstMip.m_width + && mip.m_height != dstMip.m_height) + { + printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" + , mip.m_width + , mip.m_height + , dstMip.m_width + , dstMip.m_height + , getName(format) + ); + return NULL; + } + + uint32_t size = bimg::imageGetSize( + NULL + , uint16_t(dstMip.m_width) + , uint16_t(dstMip.m_height) + , 0 + , false + , false + , 1 + , bimg::TextureFormat::RGBA32F + ); + temp = BX_ALLOC(_allocator, size); + float* rgba = (float*)temp; + float* rgbaDst = (float*)BX_ALLOC(_allocator, size); + + bimg::imageDecodeToRgba32f(_allocator + , rgba + , mip.m_data + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , mip.m_format + ); + + if (bimg::TextureFormat::BC5 != mip.m_format) + { + for (uint32_t yy = 0; yy < mip.m_height; ++yy) + { + for (uint32_t xx = 0; xx < mip.m_width; ++xx) + { + const uint32_t offset = (yy*mip.m_width + xx) * 4; + float* inout = &rgba[offset]; + inout[0] = inout[0] * 2.0f - 1.0f; + inout[1] = inout[1] * 2.0f - 1.0f; + inout[2] = inout[2] * 2.0f - 1.0f; + inout[3] = inout[3] * 2.0f - 1.0f; + } + } + } + + bimg::imageRgba32f11to01(rgbaDst, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); + bimg::imageEncodeFromRgba32f(_allocator, output->m_data, rgbaDst, dstMip.m_width, dstMip.m_height, format); + + for (uint8_t lod = 1; lod < numMips; ++lod) + { + bimg::imageRgba32fDownsample2x2NormalMap(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); + bimg::imageRgba32f11to01(rgbaDst, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); + bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); + uint8_t* data = const_cast(dstMip.m_data); + bimg::imageEncodeFromRgba32f(_allocator, data, rgbaDst, dstMip.m_width, dstMip.m_height, format); + } + + BX_FREE(_allocator, rgbaDst); + } + else if (8 != bimg::getBlockInfo(input->m_format).rBits) + { + output = bimg::imageAlloc(_allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, _options.mips); + + bimg::ImageMip dstMip; + bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); + + if (mip.m_width != dstMip.m_width + && mip.m_height != dstMip.m_height) + { + printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" + , mip.m_width + , mip.m_height + , dstMip.m_width + , dstMip.m_height + , getName(format) + ); + return NULL; + } + + uint32_t size = bimg::imageGetSize( + NULL + , uint16_t(dstMip.m_width) + , uint16_t(dstMip.m_height) + , 0 + , false + , false + , 1 + , bimg::TextureFormat::RGBA32F + ); + temp = BX_ALLOC(_allocator, size); + float* rgba = (float*)temp; + float* rgbaDst = (float*)BX_ALLOC(_allocator, size); + + bimg::imageDecodeToRgba32f(_allocator + , rgba + , mip.m_data + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , mip.m_format + ); + bimg::imageEncodeFromRgba32f(_allocator, output->m_data, rgba, dstMip.m_width, dstMip.m_height, format); + + bimg::imageRgba32fToLinear(rgba + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , rgba + ); + + for (uint8_t lod = 1; lod < numMips; ++lod) + { + bimg::imageRgba32fLinearDownsample2x2(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); + bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); + uint8_t* data = const_cast(dstMip.m_data); + + bimg::imageRgba32fToGamma(rgbaDst + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , rgba + ); + + bimg::imageEncodeFromRgba32f(_allocator, data, rgbaDst, dstMip.m_width, dstMip.m_height, format); + } + + BX_FREE(_allocator, rgbaDst); + } + else + { + output = bimg::imageAlloc(_allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, _options.mips); + + bimg::ImageMip dstMip; + bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); + + if (mip.m_width != dstMip.m_width + && mip.m_height != dstMip.m_height) + { + printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" + , mip.m_width + , mip.m_height + , dstMip.m_width + , dstMip.m_height + , getName(format) + ); + return NULL; + } + + uint32_t size = bimg::imageGetSize( + NULL + , uint16_t(dstMip.m_width) + , uint16_t(dstMip.m_height) + , 0 + , false + , false + , 1 + , bimg::TextureFormat::RGBA8 + ); + temp = BX_ALLOC(_allocator, size); + bx::memSet(temp, 0, size); + uint8_t* rgba = (uint8_t*)temp; + + bimg::imageDecodeToRgba8(rgba + , mip.m_data + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , mip.m_format + ); + + void* ref = NULL; + if (_options.iqa) + { + ref = BX_ALLOC(_allocator, size); + bx::memCopy(ref, rgba, size); + } + + bimg::imageEncodeFromRgba8(output->m_data, rgba, dstMip.m_width, dstMip.m_height, format); + + for (uint8_t lod = 1; lod < numMips; ++lod) + { + bimg::imageRgba8Downsample2x2(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*4, rgba); + bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); + uint8_t* data = const_cast(dstMip.m_data); + bimg::imageEncodeFromRgba8(data, rgba, dstMip.m_width, dstMip.m_height, format); + } + + if (NULL != ref) + { + bimg::imageDecodeToRgba8(rgba + , output->m_data + , mip.m_width + , mip.m_height + , mip.m_width*mip.m_bpp/8 + , format + ); + + float result = bimg::imageQualityRgba8( + ref + , rgba + , uint16_t(mip.m_width) + , uint16_t(mip.m_height) + ); + + printf("%f\n", result); + + BX_FREE(_allocator, ref); + } + } + + BX_FREE(_allocator, temp); + } + } + + return output; +} + void help(const char* _error = NULL) { if (NULL != _error) @@ -53,6 +367,8 @@ void help(const char* _error = NULL) " -n, --normalmap Input texture is normal map.\n" " --sdf Compute SDF texture.\n" " --iqa Image Quality Assesment\n" + " --max Maximum width/height (image will be scaled down and\n" + " aspect ratio will be preserved.\n" "\n" "For additional information, see https://github.com/bkaradzic/bgfx\n" @@ -83,19 +399,24 @@ int main(int _argc, const char* _argv[]) return EXIT_FAILURE; } - bool sdf = false; - double edge = 16.0; + Options options; + const char* edgeOpt = cmdLine.findOption("sdf"); if (NULL != edgeOpt) { - sdf = true; - edge = atof(edgeOpt); + options.sdf = true; + options.edge = (float)atof(edgeOpt); } - BX_UNUSED(sdf, edge); - const bool mips = cmdLine.hasArg('m', "mips"); - const bool normalMap = cmdLine.hasArg('n', "normalmap"); - const bool iqa = cmdLine.hasArg('\0', "iqa"); + options.mips = cmdLine.hasArg('m', "mips"); + options.normalMap = cmdLine.hasArg('n', "normalmap"); + options.iqa = cmdLine.hasArg('\0', "iqa"); + + const char* maxSize = cmdLine.findOption("max"); + if (NULL != maxSize) + { + options.maxSize = atoi(maxSize); + } bx::CrtFileReader reader; if (!bx::open(&reader, inputFileName) ) @@ -104,6 +425,19 @@ int main(int _argc, const char* _argv[]) return EXIT_FAILURE; } + options.format = bimg::TextureFormat::Count; + const char* type = cmdLine.findOption('t'); + if (NULL != type) + { + options.format = bimg::getFormat(type); + + if (!bimg::isValid(options.format) ) + { + help("Invalid format specified."); + return EXIT_FAILURE; + } + } + bx::CrtAllocator allocator; uint32_t inputSize = (uint32_t)bx::getSize(&reader); @@ -112,297 +446,34 @@ int main(int _argc, const char* _argv[]) bx::read(&reader, inputData, inputSize); bx::close(&reader); + bimg::ImageContainer* output = convert(&allocator, inputData, inputSize, options); + + BX_FREE(&allocator, inputData); + + if (NULL != output) { - bimg::ImageContainer* input = bimg::imageParse(&allocator, inputData, inputSize); - - if (NULL != input) + bx::CrtFileWriter writer; + if (bx::open(&writer, outputFileName) ) { - BX_FREE(&allocator, inputData); - - const char* type = cmdLine.findOption('t'); - bimg::TextureFormat::Enum format = input->m_format; - - if (NULL != type) + if (NULL != bx::stristr(outputFileName, ".ktx") ) { - format = bimg::getFormat(type); - - if (!bimg::isValid(format) ) - { - help("Invalid format specified."); - return EXIT_FAILURE; - } + bimg::imageWriteKtx(&writer, *output, output->m_data, output->m_size); } - bimg::ImageContainer* output = NULL; - - bimg::ImageMip mip; - if (bimg::imageGetRawData(*input, 0, 0, input->m_data, input->m_size, mip) ) - { - uint8_t numMips = mips - ? bimg::imageGetNumMips(format, uint16_t(mip.m_width), uint16_t(mip.m_height) ) - : 1 - ; - - void* temp = NULL; - - if (normalMap) - { - output = bimg::imageAlloc(&allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, mips); - - bimg::ImageMip dstMip; - bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); - - if (mip.m_width != dstMip.m_width - && mip.m_height != dstMip.m_height) - { - printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" - , mip.m_width - , mip.m_height - , dstMip.m_width - , dstMip.m_height - , getName(format) - ); - return EXIT_FAILURE; - } - - uint32_t size = bimg::imageGetSize( - NULL - , uint16_t(dstMip.m_width) - , uint16_t(dstMip.m_height) - , 0 - , false - , false - , 1 - , bimg::TextureFormat::RGBA32F - ); - temp = BX_ALLOC(&allocator, size); - float* rgba = (float*)temp; - float* rgbaDst = (float*)BX_ALLOC(&allocator, size); - - bimg::imageDecodeToRgba32f(&allocator - , rgba - , mip.m_data - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , mip.m_format - ); - - if (bimg::TextureFormat::BC5 != mip.m_format) - { - for (uint32_t yy = 0; yy < mip.m_height; ++yy) - { - for (uint32_t xx = 0; xx < mip.m_width; ++xx) - { - const uint32_t offset = (yy*mip.m_width + xx) * 4; - float* inout = &rgba[offset]; - inout[0] = inout[0] * 2.0f - 1.0f; - inout[1] = inout[1] * 2.0f - 1.0f; - inout[2] = inout[2] * 2.0f - 1.0f; - inout[3] = inout[3] * 2.0f - 1.0f; - } - } - } - - bimg::imageRgba32f11to01(rgbaDst, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); - bimg::imageEncodeFromRgba32f(&allocator, output->m_data, rgbaDst, dstMip.m_width, dstMip.m_height, format); - - for (uint8_t lod = 1; lod < numMips; ++lod) - { - bimg::imageRgba32fDownsample2x2NormalMap(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); - bimg::imageRgba32f11to01(rgbaDst, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); - bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); - uint8_t* data = const_cast(dstMip.m_data); - bimg::imageEncodeFromRgba32f(&allocator, data, rgbaDst, dstMip.m_width, dstMip.m_height, format); - } - - BX_FREE(&allocator, rgbaDst); - } - else if (8 != getBlockInfo(input->m_format).rBits) - { - output = imageAlloc(&allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, mips); - - bimg::ImageMip dstMip; - bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); - - if (mip.m_width != dstMip.m_width - && mip.m_height != dstMip.m_height) - { - printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" - , mip.m_width - , mip.m_height - , dstMip.m_width - , dstMip.m_height - , getName(format) - ); - return EXIT_FAILURE; - } - - uint32_t size = bimg::imageGetSize( - NULL - , uint16_t(dstMip.m_width) - , uint16_t(dstMip.m_height) - , 0 - , false - , false - , 1 - , bimg::TextureFormat::RGBA32F - ); - temp = BX_ALLOC(&allocator, size); - float* rgba = (float*)temp; - float* rgbaDst = (float*)BX_ALLOC(&allocator, size); - - bimg::imageDecodeToRgba32f(&allocator - , rgba - , mip.m_data - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , mip.m_format - ); - bimg::imageEncodeFromRgba32f(&allocator, output->m_data, rgba, dstMip.m_width, dstMip.m_height, format); - - bimg::imageRgba32fToLinear(rgba - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , rgba - ); - - for (uint8_t lod = 1; lod < numMips; ++lod) - { - bimg::imageRgba32fLinearDownsample2x2(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*16, rgba); - bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); - uint8_t* data = const_cast(dstMip.m_data); - - bimg::imageRgba32fToGamma(rgbaDst - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , rgba - ); - - bimg::imageEncodeFromRgba32f(&allocator, data, rgbaDst, dstMip.m_width, dstMip.m_height, format); - } - - BX_FREE(&allocator, rgbaDst); - } - else - { - output = imageAlloc(&allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, mips); - - bimg::ImageMip dstMip; - bimg::imageGetRawData(*output, 0, 0, NULL, 0, dstMip); - - if (mip.m_width != dstMip.m_width - && mip.m_height != dstMip.m_height) - { - printf("Invalid input image size %dx%d, it must be at least %dx%d to be converted to %s format.\n" - , mip.m_width - , mip.m_height - , dstMip.m_width - , dstMip.m_height - , getName(format) - ); - return EXIT_FAILURE; - } - - uint32_t size = bimg::imageGetSize( - NULL - , uint16_t(dstMip.m_width) - , uint16_t(dstMip.m_height) - , 0 - , false - , false - , 1 - , bimg::TextureFormat::RGBA8 - ); - temp = BX_ALLOC(&allocator, size); - bx::memSet(temp, 0, size); - uint8_t* rgba = (uint8_t*)temp; - - bimg::imageDecodeToRgba8(rgba - , mip.m_data - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , mip.m_format - ); - - void* ref = NULL; - if (iqa) - { - ref = BX_ALLOC(&allocator, size); - bx::memCopy(ref, rgba, size); - } - - bimg::imageEncodeFromRgba8(output->m_data, rgba, dstMip.m_width, dstMip.m_height, format); - - for (uint8_t lod = 1; lod < numMips; ++lod) - { - bimg::imageRgba8Downsample2x2(rgba, dstMip.m_width, dstMip.m_height, dstMip.m_width*4, rgba); - bimg::imageGetRawData(*output, 0, lod, output->m_data, output->m_size, dstMip); - uint8_t* data = const_cast(dstMip.m_data); - bimg::imageEncodeFromRgba8(data, rgba, dstMip.m_width, dstMip.m_height, format); - } - - if (NULL != ref) - { - bimg::imageDecodeToRgba8(rgba - , output->m_data - , mip.m_width - , mip.m_height - , mip.m_width*mip.m_bpp/8 - , format - ); - - float result = bimg::imageQualityRgba8( - ref - , rgba - , mip.m_width - , mip.m_height - ); - - printf("%f\n", result); - - BX_FREE(&allocator, ref); - } - } - - BX_FREE(&allocator, temp); - } - - if (NULL != output) - { - bx::CrtFileWriter writer; - if (bx::open(&writer, outputFileName) ) - { - if (NULL != bx::stristr(outputFileName, ".ktx") ) - { - imageWriteKtx(&writer, *output, output->m_data, output->m_size); - } - - bx::close(&writer); - } - else - { - help("Failed to open output file."); - return EXIT_FAILURE; - } - - bimg::imageFree(output); - } - else - { - help("No output generated."); - return EXIT_FAILURE; - } + bx::close(&writer); } else { - help("Failed to load input file."); + help("Failed to open output file."); return EXIT_FAILURE; } + + bimg::imageFree(output); + } + else + { + help("No output generated."); + return EXIT_FAILURE; } return EXIT_SUCCESS;