diff options
Diffstat (limited to 'samples/lut_generator/lut_generator.cpp')
| -rw-r--r-- | samples/lut_generator/lut_generator.cpp | 429 |
1 files changed, 429 insertions, 0 deletions
diff --git a/samples/lut_generator/lut_generator.cpp b/samples/lut_generator/lut_generator.cpp new file mode 100644 index 0000000..72804a4 --- /dev/null +++ b/samples/lut_generator/lut_generator.cpp @@ -0,0 +1,429 @@ +//---------------------------------------------------------------------------------- +// File: FaceWorks/samples/lut_generator/lut_generator.cpp +// SDK Version: v1.0 +// Email: [email protected] +// Site: http://developer.nvidia.com/ +// +// Copyright (c) 2014-2016, NVIDIA CORPORATION. All rights reserved. +// +// Redistribution and use in source and binary forms, with or without +// modification, are permitted provided that the following conditions +// are met: +// * Redistributions of source code must retain the above copyright +// notice, this list of conditions and the following disclaimer. +// * Redistributions in binary form must reproduce the above copyright +// notice, this list of conditions and the following disclaimer in the +// documentation and/or other materials provided with the distribution. +// * Neither the name of NVIDIA CORPORATION nor the names of its +// contributors may be used to endorse or promote products derived +// from this software without specific prior written permission. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY +// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR +// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR +// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, +// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR +// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY +// OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +// +//---------------------------------------------------------------------------------- + +#include <algorithm> +#include <cassert> +#include <cstdio> +#include <ctime> +#include <vector> + +#include <GFSDK_FaceWorks.h> + +#define NOMINMAX +#define WIN32_LEAN_AND_MEAN +#include <windows.h> + + + +// Command-line processing + +void PrintUsage() +{ + printf( + "Usage: lut_generator [options]\n" + "Options:\n" + " -curvatureLUT FILENAME Generate curvature LUT (.bmp format)\n" + " -shadowLUT FILENAME Generate shadow LUT (.bmp format)\n" + "\n" + "Curvature LUT options:\n" + " -diffusionRadius FLOAT Radius of diffusion profile in mm; default is 2.7\n" + " -minCurvatureRadius FLOAT Minimum radius of curvature in mm; default is 1.0\n" + " -maxCurvatureRadius FLOAT Maximum radius of curvature in mm; default is 100.0\n" + " -width INT LUT width in pixels; default is 512\n" + " -height INT LUT height in pixels; default is 512\n" + "\n" + "Shadow LUT options:\n" + " -diffusionRadius FLOAT Radius of diffusion profile in mm; default is 2.7\n" + " -minShadowWidth FLOAT Minimum penumbra width in mm; default is 8.0\n" + " -maxShadowWidth FLOAT Maximum penumbra width in mm; default is 100.0\n" + " -shadowSharpening FLOAT Shadow sharpening ratio; default is 10.0\n" + " -width INT LUT width in pixels; default is 512\n" + " -height INT LUT height in pixels; default is 512\n" + "\n" + ); +} + +bool ReadInt( + const char ** argNameAndValue, + int * pValue, + int minValid = INT_MIN, + int maxValid = INT_MAX) +{ + assert(minValid <= maxValid); + + const char * argName = argNameAndValue[0]; + const char * strValue = argNameAndValue[1]; + + if (!strValue) + { + fprintf(stderr, "%s: missing argument; ignoring\n", argName); + return false; + } + + int value; + if (!sscanf_s(strValue, "%d", &value)) + { + fprintf( + stderr, + "%s: could not parse \"%s\" as an int; ignoring\n", + argName, + strValue); + return false; + } + + if (value < minValid) + { + fprintf( + stderr, + "%s: value must be at least %d; clamping\n", + argName, + minValid); + value = minValid; + } + + if (value > maxValid) + { + fprintf( + stderr, + "%s: value must be at most %d; clamping\n", + argName, + maxValid); + value = maxValid; + } + + *pValue = value; + return true; +} + +bool ReadFloat( + const char ** argNameAndValue, + float * pValue, + float minValid = -FLT_MAX, + float maxValid = FLT_MAX) +{ + assert(minValid <= maxValid); + + const char * argName = argNameAndValue[0]; + const char * strValue = argNameAndValue[1]; + + if (!strValue) + { + fprintf(stderr, "%s: missing argument; ignoring\n", argName); + return false; + } + + float value; + if (!sscanf_s(strValue, "%f", &value)) + { + fprintf( + stderr, + "%s: could not parse \"%s\" as a float; ignoring\n", + argName, + strValue); + return false; + } + + if (value < minValid) + { + fprintf( + stderr, + "%s: value must be at least %.2g; clamping\n", + argName, + minValid); + value = minValid; + } + + if (value > maxValid) + { + fprintf( + stderr, + "%s: value must be at most %.2g; clamping\n", + argName, + maxValid); + value = maxValid; + } + + *pValue = value; + return true; +} + + + +int GenerateCurvatureLUT( + const GFSDK_FaceWorks_CurvatureLUTConfig * pConfig, + const char * strFilename); + +int GenerateShadowLUT( + const GFSDK_FaceWorks_ShadowLUTConfig * pConfig, + const char * strFilename); + + + +int main(int argc, const char ** argv) +{ + // Set default params + GFSDK_FaceWorks_CurvatureLUTConfig curvatureConfig = + { + 2.7f, // m_diffusionRadius + 512, 512, // m_texWidth, m_texHeight + 1.0f, 100.0f, // m_curvatureRadiusMin, m_curvatureRadiusMax + }; + GFSDK_FaceWorks_ShadowLUTConfig shadowConfig = + { + 2.7f, // m_diffusionRadius + 512, 512, // m_texWidth, m_texHeight + 8.0f, 100.0f, // m_shadowWidthMin, m_shadowWidthMax + 10.0f, // m_shadowSharpening + }; + const char * strCurvatureFilename = NULL; + const char * strShadowFilename = NULL; + + // Parse command-line params + for (int iArg = 1; iArg < argc; ++iArg) + { + if (_stricmp(argv[iArg], "-h") == 0 || + _stricmp(argv[iArg], "-help") == 0 || + _stricmp(argv[iArg], "--help") == 0 || + _stricmp(argv[iArg], "/?") == 0) + { + PrintUsage(); + return 0; + } + else if (_stricmp(argv[iArg], "-curvatureLUT") == 0) + { + strCurvatureFilename = argv[++iArg]; + if (!strCurvatureFilename) + fprintf(stderr, "-curvatureLUT: filename expected\n"); + } + else if (_stricmp(argv[iArg], "-shadowLUT") == 0) + { + strShadowFilename = argv[++iArg]; + if (!strShadowFilename) + fprintf(stderr, "-shadowLUT: filename expected\n"); + } + else if (_stricmp(argv[iArg], "-width") == 0) + { + int width; + if (ReadInt(&argv[iArg++], &width, 1, 16384)) + { + curvatureConfig.m_texWidth = width; + shadowConfig.m_texWidth = width; + } + } + else if (_stricmp(argv[iArg], "-height") == 0) + { + int height; + if (ReadInt(&argv[iArg++], &height, 1, 16384)) + { + curvatureConfig.m_texHeight = height; + shadowConfig.m_texHeight = height; + } + } + else if (_stricmp(argv[iArg], "-minCurvatureRadius") == 0) + { + ReadFloat(&argv[iArg++], &curvatureConfig.m_curvatureRadiusMin, FLT_MIN); + } + else if (_stricmp(argv[iArg], "-maxCurvatureRadius") == 0) + { + ReadFloat(&argv[iArg++], &curvatureConfig.m_curvatureRadiusMax, FLT_MIN); + } + else if (_stricmp(argv[iArg], "-minShadowWidth") == 0) + { + ReadFloat(&argv[iArg++], &shadowConfig.m_shadowWidthMin, FLT_MIN); + } + else if (_stricmp(argv[iArg], "-maxShadowWidth") == 0) + { + ReadFloat(&argv[iArg++], &shadowConfig.m_shadowWidthMax, FLT_MIN); + } + else if (_stricmp(argv[iArg], "-shadowSharpening") == 0) + { + ReadFloat(&argv[iArg++], &shadowConfig.m_shadowSharpening, 1.0f); + } + else + { + fprintf( + stderr, + "Warning: unrecognized command-line parameter \"%s\"; ignoring\n", + argv[iArg]); + } + } + + if (!strCurvatureFilename && !strShadowFilename) + { + PrintUsage(); + return 0; + } + + if (strCurvatureFilename) + { + if (curvatureConfig.m_curvatureRadiusMin > curvatureConfig.m_curvatureRadiusMax) + { + fprintf(stderr, "Warning: minCurvatureRadius > maxCurvatureRadius; swapping\n"); + std::swap(curvatureConfig.m_curvatureRadiusMin, curvatureConfig.m_curvatureRadiusMax); + } + + int res = GenerateCurvatureLUT(&curvatureConfig, strCurvatureFilename); + if (res != 0) + return 1; + } + + if (strShadowFilename) + { + if (shadowConfig.m_shadowWidthMin > shadowConfig.m_shadowWidthMax) + { + fprintf(stderr, "Warning: minShadowWidth > maxShadowWidth; swapping\n"); + std::swap(shadowConfig.m_shadowWidthMin, shadowConfig.m_shadowWidthMax); + } + + int res = GenerateShadowLUT(&shadowConfig, strShadowFilename); + if (res != 0) + return 1; + } + + return 0; +} + + + +int WriteBMP( + int width, int height, + void * pPixelsRGBA, + const char * strFilename) +{ + FILE * pFile; + if (fopen_s(&pFile, strFilename, "wb") != 0 || !pFile) + { + fprintf(stderr, "Error: couldn't open %s for writing\n", strFilename); + return 1; + } + + BITMAPFILEHEADER bmpFileHeader = + { + 0x4d42, + 0, 0, 0, + sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER), + }; + fwrite(&bmpFileHeader, sizeof(bmpFileHeader), 1, pFile); + + BITMAPINFOHEADER bmpInfoHeader = + { + sizeof(BITMAPINFOHEADER), + width, -height, // negative height means top-down BMP + 1, // planes + 24, // bits per pixel + }; + fwrite(&bmpInfoHeader, sizeof(bmpInfoHeader), 1, pFile); + + // Convert RGBA to BGR pixel order + unsigned char * pRGBA = static_cast<unsigned char *>(pPixelsRGBA); + int pixelCount = width * height; + for (int i = 0; i < pixelCount; ++i) + { + unsigned char r = *(pRGBA++); + unsigned char g = *(pRGBA++); + unsigned char b = *(pRGBA++); + pRGBA++; // skip alpha + + fputc(b, pFile); + fputc(g, pFile); + fputc(r, pFile); + } + + fclose(pFile); + printf("Wrote %s\n", strFilename); + + return 0; +} + + + +int GenerateCurvatureLUT( + const GFSDK_FaceWorks_CurvatureLUTConfig * pConfig, + const char * strFilename) +{ + printf("Generating curvature LUT...\n"); + clock_t start = clock(); + + std::vector<unsigned char> pixels; + pixels.resize(GFSDK_FaceWorks_CalculateCurvatureLUTSizeBytes(pConfig)); + + GFSDK_FaceWorks_ErrorBlob errorBlob = {}; + GFSDK_FaceWorks_Result res = GFSDK_FaceWorks_GenerateCurvatureLUT(pConfig, &pixels[0], &errorBlob); + if (res != GFSDK_FaceWorks_OK) + { + fprintf(stderr, "GFSDK_FaceWorks_GenerateCurvatureLUT() failed:\n%s", errorBlob.m_msg); + GFSDK_FaceWorks_FreeErrorBlob(&errorBlob); + return 1; + } + + clock_t clocks = clock() - start; + printf("Done in %0.3f seconds\n", float(clocks) / float(CLOCKS_PER_SEC)); + + return WriteBMP( + pConfig->m_texWidth, + pConfig->m_texHeight, + &pixels[0], + strFilename); +} + + + +int GenerateShadowLUT( + const GFSDK_FaceWorks_ShadowLUTConfig * pConfig, + const char * strFilename) +{ + printf("Generating shadow LUT...\n"); + clock_t start = clock(); + + std::vector<unsigned char> pixels; + pixels.resize(GFSDK_FaceWorks_CalculateShadowLUTSizeBytes(pConfig)); + + GFSDK_FaceWorks_ErrorBlob errorBlob = {}; + GFSDK_FaceWorks_Result res = GFSDK_FaceWorks_GenerateShadowLUT(pConfig, &pixels[0], &errorBlob); + if (res != GFSDK_FaceWorks_OK) + { + fprintf(stderr, "GFSDK_FaceWorks_GenerateShadowLUT() failed:\n%s", errorBlob.m_msg); + GFSDK_FaceWorks_FreeErrorBlob(&errorBlob); + return 1; + } + + clock_t clocks = clock() - start; + printf("Done in %0.3f seconds\n", float(clocks) / float(CLOCKS_PER_SEC)); + + return WriteBMP( + pConfig->m_texWidth, + pConfig->m_texHeight, + &pixels[0], + strFilename); +} |