816 lines
31 KiB
C++
816 lines
31 KiB
C++
/*
|
|
* Copyright (C) 2015 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
|
|
#include <iomanip>
|
|
|
|
#include <math/scalar.h>
|
|
#include <math/vec4.h>
|
|
|
|
#include <image/ImageDecoder.h>
|
|
#include <image/ImageEncoder.h>
|
|
|
|
#include <utils/Path.h>
|
|
|
|
#include <getopt/getopt.h>
|
|
|
|
#include "Cubemap.h"
|
|
#include "CubemapIBL.h"
|
|
#include "CubemapSH.h"
|
|
#include "CubemapUtils.h"
|
|
|
|
using namespace math;
|
|
using namespace image;
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
enum class ShFile {
|
|
SH_NONE, SH_CROSS, SH_TEXT
|
|
};
|
|
static image::ImageEncoder::Format g_format = image::ImageEncoder::Format::PNG;
|
|
static std::string g_compression = "";
|
|
static bool g_extract_faces = false;
|
|
static double g_extract_blur = 0.0;
|
|
static utils::Path g_extract_dir;
|
|
|
|
static size_t g_output_size = 0;
|
|
|
|
bool g_quiet = false; // needed outside of this file
|
|
static bool g_debug = false;
|
|
|
|
static size_t g_sh_compute = 0;
|
|
static bool g_sh_output = false;
|
|
static bool g_sh_shader = false;
|
|
static bool g_sh_irradiance = false;
|
|
static ShFile g_sh_file = ShFile::SH_NONE;
|
|
static utils::Path g_sh_filename;
|
|
|
|
static bool g_is_mipmap = false;
|
|
static utils::Path g_is_mipmap_dir;
|
|
static bool g_prefilter = false;
|
|
static utils::Path g_prefilter_dir;
|
|
static bool g_dfg = false;
|
|
static utils::Path g_dfg_filename;
|
|
static bool g_dfg_multiscatter = false;
|
|
|
|
static bool g_deploy = false;
|
|
static utils::Path g_deploy_dir;
|
|
|
|
static size_t g_num_samples = 1024;
|
|
|
|
static bool g_mirror = false;
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
static void generateMipmaps(std::vector<Cubemap>& levels, std::vector<Image>& images);
|
|
static void sphericalHarmonics(const utils::Path& iname, const Cubemap& inputCubemap);
|
|
static void iblRoughnessPrefilter(const utils::Path& iname, const std::vector<Cubemap>& levels,
|
|
const utils::Path& dir);
|
|
static void iblMipmapPrefilter(const utils::Path& iname, const std::vector<Image>& images,
|
|
const std::vector<Cubemap>& levels, const utils::Path& dir);
|
|
static void iblLutDfg(const utils::Path& filename, size_t size = 128, bool multiscatter = false);
|
|
static void extractCubemapFaces(const utils::Path& iname, const Cubemap& cm, const utils::Path& dir);
|
|
static void outputSh(std::ostream& out, const std::unique_ptr<math::double3[]>& sh, size_t numBands);
|
|
static void outputSpectrum(std::ostream& out, const std::unique_ptr<math::double3[]>& sh,
|
|
size_t numBands);
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
void generateUVGrid(Cubemap const& cml, size_t gridFrequency, size_t dim);
|
|
|
|
static void printUsage(char* name) {
|
|
std::string exec_name(utils::Path(name).getName());
|
|
std::string usage(
|
|
"CMGEN is a command-line tool for generating SH and mipmap levels from a cubemap\n"
|
|
"Usages:\n"
|
|
" CMGEN [options] <input-file>\n"
|
|
" CMGEN [options] <uv[N]>\n"
|
|
"\n"
|
|
"Supported input formats:\n"
|
|
" PNG, 8 and 16 bits\n"
|
|
" Radiance (.hdr)\n"
|
|
" Photoshop (.psd), 16 and 32 bits\n"
|
|
" OpenEXR (.exr)\n"
|
|
"\n"
|
|
"Options:\n"
|
|
" --help, -h\n"
|
|
" print this message\n\n"
|
|
" --license\n"
|
|
" Print copyright and license information\n\n"
|
|
" --quiet, -q\n"
|
|
" Quiet mode. Suppress all non-error output\n\n"
|
|
" --format=[exr|hdr|psd|rgbm|png|dds], -f [exr|hdr|psd|rgbm|png|dds]\n"
|
|
" specify output file format\n\n"
|
|
" --compression=COMPRESSION, -c COMPRESSION\n"
|
|
" format specific compression:\n"
|
|
" PNG: Ignored\n"
|
|
" PNG RGBM: Ignored\n"
|
|
" Radiance: Ignored\n"
|
|
" Photoshop: 16 (default), 32\n"
|
|
" OpenEXR: RAW, RLE, ZIPS, ZIP, PIZ (default)\n"
|
|
" DDS: 8, 16 (default), 32\n\n"
|
|
" --size=power-of-two, -s power-of-two\n"
|
|
" size of the output cubemaps (base level), 256 by default\n\n"
|
|
" --deploy=dir, -x dir\n"
|
|
" Generate everything needed for deployment into <dir>\n\n"
|
|
" --extract=dir\n"
|
|
" Extract faces of the cubemap into <dir>\n\n"
|
|
" --extract-blur=roughness\n"
|
|
" Blurs the cubemap before saving the faces using the roughness blur\n\n"
|
|
" --mirror\n"
|
|
" Mirrors generated cubemaps for reflections\n\n"
|
|
" --ibl-samples=numSamples\n"
|
|
" Number of samples tu use for IBL integrations (default 1024)\n\n"
|
|
"\n"
|
|
"Private use only:\n"
|
|
" --ibl-dfg=filename.[exr|hdr|psd|png|rgbm|dds|h|hpp|c|cpp|inc|txt]\n"
|
|
" Computes the IBL DFG LUT\n\n"
|
|
" --ibl-dfg-multiscatter\n"
|
|
" If --ibl-dfg is set, computes the DFG for multi-scattering GGX\n\n"
|
|
" --ibl-is-mipmap=dir\n"
|
|
" Generates mipmap for pre-filtered importance sampling\n\n"
|
|
" --ibl-ld=dir\n"
|
|
" Roughness prefilter into <dir>\n\n"
|
|
" --sh=bands\n"
|
|
" SH decomposition of input cubemap\n\n"
|
|
" --sh-output=filename.[exr|hdr|psd|rgbm|png|dds|txt]\n"
|
|
" SH output format. The filename extension determines the output format\n\n"
|
|
" --sh-irradiance, -i\n"
|
|
" Irradiance SH coefficients\n\n"
|
|
" --sh-shader\n"
|
|
" Generate irradiance SH for shader code\n\n"
|
|
" --debug, -d\n"
|
|
" Generate extra data for debugging\n\n"
|
|
);
|
|
const std::string from("CMGEN");
|
|
for (size_t pos = usage.find(from); pos != std::string::npos; pos = usage.find(from, pos)) {
|
|
usage.replace(pos, from.length(), exec_name);
|
|
}
|
|
printf("%s", usage.c_str());
|
|
}
|
|
|
|
static void license() {
|
|
std::cout <<
|
|
#include "licenses/licenses.inc"
|
|
;
|
|
}
|
|
|
|
static int handleCommandLineArgments(int argc, char* argv[]) {
|
|
static constexpr const char* OPTSTR = "hqidf:c:s:x:";
|
|
static const struct option OPTIONS[] = {
|
|
{ "help", no_argument, 0, 'h' },
|
|
{ "license", no_argument, 0, 'l' },
|
|
{ "quiet", no_argument, 0, 'q' },
|
|
{ "format", required_argument, 0, 'f' },
|
|
{ "compression", required_argument, 0, 'c' },
|
|
{ "size", required_argument, 0, 's' },
|
|
{ "extract", required_argument, 0, 'e' },
|
|
{ "extract-blur", required_argument, 0, 'r' },
|
|
{ "sh", optional_argument, 0, 'z' },
|
|
{ "sh-output", required_argument, 0, 'o' },
|
|
{ "sh-irradiance", no_argument, 0, 'i' },
|
|
{ "sh-shader", no_argument, 0, 'b' },
|
|
{ "ibl-is-mipmap", required_argument, 0, 'y' },
|
|
{ "ibl-ld", required_argument, 0, 'p' },
|
|
{ "ibl-dfg", required_argument, 0, 'a' },
|
|
{ "ibl-dfg-multiscatter", no_argument, 0, 'u' },
|
|
{ "ibl-samples", required_argument, 0, 'k' },
|
|
{ "deploy", required_argument, 0, 'x' },
|
|
{ "mirror", no_argument, 0, 'm' },
|
|
{ "debug", no_argument, 0, 'd' },
|
|
{ 0, 0, 0, 0 } // termination of the option list
|
|
};
|
|
int opt;
|
|
int option_index = 0;
|
|
int num_sh_bands = 3;
|
|
bool format_specified = false;
|
|
while ((opt = getopt_long(argc, argv, OPTSTR, OPTIONS, &option_index)) >= 0) {
|
|
std::string arg(optarg ? optarg : "");
|
|
switch (opt) {
|
|
default:
|
|
case 'h':
|
|
printUsage(argv[0]);
|
|
exit(0);
|
|
break;
|
|
case 'l':
|
|
license();
|
|
exit(0);
|
|
break;
|
|
case 'q':
|
|
g_quiet = true;
|
|
break;
|
|
case 'f':
|
|
if (arg == "png") {
|
|
g_format = ImageEncoder::Format::PNG;
|
|
format_specified = true;
|
|
}
|
|
if (arg == "hdr") {
|
|
g_format = ImageEncoder::Format::HDR;
|
|
format_specified = true;
|
|
}
|
|
if (arg == "rgbm") {
|
|
g_format = ImageEncoder::Format::RGBM;
|
|
format_specified = true;
|
|
}
|
|
if (arg == "exr") {
|
|
g_format = ImageEncoder::Format::EXR;
|
|
format_specified = true;
|
|
}
|
|
if (arg == "psd") {
|
|
g_format = ImageEncoder::Format::PSD;
|
|
format_specified = true;
|
|
}
|
|
if (arg == "dds") {
|
|
g_format = ImageEncoder::Format::DDS_LINEAR;
|
|
format_specified = true;
|
|
}
|
|
break;
|
|
case 'c':
|
|
g_compression = arg;
|
|
break;
|
|
case 's':
|
|
g_output_size = std::stoul(arg);
|
|
if (!isPOT(g_output_size)) {
|
|
std::cerr << "output size must be a power of two" << std::endl;
|
|
exit(0);
|
|
}
|
|
break;
|
|
case 'z':
|
|
g_sh_compute = 1;
|
|
g_sh_output = true;
|
|
try {
|
|
num_sh_bands = std::stoi(arg);
|
|
} catch (std::invalid_argument &e) {
|
|
// keep default value
|
|
}
|
|
break;
|
|
case 'o':
|
|
g_sh_compute = 1;
|
|
g_sh_output = true;
|
|
g_sh_file = ShFile::SH_CROSS;
|
|
g_sh_filename = arg;
|
|
if (g_sh_filename.getExtension() == "txt") {
|
|
g_sh_file = ShFile::SH_TEXT;
|
|
}
|
|
break;
|
|
case 'i':
|
|
g_sh_compute = 1;
|
|
g_sh_irradiance = true;
|
|
break;
|
|
case 'b':
|
|
g_sh_compute = 1;
|
|
g_sh_irradiance = true;
|
|
g_sh_shader = true;
|
|
break;
|
|
case 'e':
|
|
g_extract_dir = arg;
|
|
g_extract_faces = true;
|
|
break;
|
|
case 'r':
|
|
g_extract_blur = std::stod(arg);
|
|
if (g_extract_blur < 0 || g_extract_blur > 1) {
|
|
std::cerr << "roughness (blur) parameter must be between 0.0 and 1.0" <<
|
|
std::endl;
|
|
exit(0);
|
|
}
|
|
break;
|
|
case 'y':
|
|
g_is_mipmap = true;
|
|
g_is_mipmap_dir = arg;
|
|
break;
|
|
case 'p':
|
|
g_prefilter = true;
|
|
g_prefilter_dir = arg;
|
|
break;
|
|
case 'a':
|
|
g_dfg = true;
|
|
g_dfg_filename = arg;
|
|
break;
|
|
case 'u':
|
|
g_dfg_multiscatter = true;
|
|
break;
|
|
case 'k':
|
|
g_num_samples = (size_t)std::stoi(arg);
|
|
break;
|
|
case 'x':
|
|
g_deploy = true;
|
|
g_deploy_dir = arg;
|
|
break;
|
|
case 'd':
|
|
g_debug = true;
|
|
break;
|
|
case 'm':
|
|
g_mirror = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (g_deploy && !format_specified) {
|
|
g_format = ImageEncoder::Format::RGBM;
|
|
}
|
|
|
|
if (num_sh_bands && g_sh_compute) {
|
|
g_sh_compute = (size_t) num_sh_bands;
|
|
}
|
|
return optind;
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
int option_index = handleCommandLineArgments(argc, argv);
|
|
int num_args = argc - option_index;
|
|
if (!g_dfg && num_args < 1) {
|
|
printUsage(argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
if (g_dfg) {
|
|
if (!g_quiet) {
|
|
std::cout << "Generating IBL DFG LUT..." << std::endl;
|
|
}
|
|
size_t size = g_output_size ? g_output_size : 128;
|
|
iblLutDfg(g_dfg_filename, size, g_dfg_multiscatter);
|
|
if (num_args < 1) return 0;
|
|
}
|
|
|
|
std::string command(argv[option_index]);
|
|
utils::Path iname(command);
|
|
|
|
if (g_deploy) {
|
|
utils::Path out_dir = g_deploy_dir + iname.getNameWithoutExtension();
|
|
|
|
// generate pre-scaled irradiance sh to text file
|
|
g_sh_compute = 3;
|
|
g_sh_shader = true;
|
|
g_sh_irradiance = true;
|
|
g_sh_filename = out_dir + "sh.txt";
|
|
g_sh_file = ShFile::SH_TEXT;
|
|
g_sh_output = true;
|
|
|
|
// faces
|
|
g_extract_dir = g_deploy_dir;
|
|
g_extract_faces = true;
|
|
|
|
// prefilter
|
|
g_prefilter = true;
|
|
g_prefilter_dir = g_deploy_dir;
|
|
}
|
|
|
|
if (g_debug) {
|
|
if (g_prefilter && !g_is_mipmap) {
|
|
g_is_mipmap = true;
|
|
g_is_mipmap_dir = g_prefilter_dir;
|
|
}
|
|
}
|
|
|
|
// Images store the actual data
|
|
std::vector<Image> images;
|
|
|
|
// Cubemaps are just views on Images
|
|
std::vector<Cubemap> levels;
|
|
|
|
if (iname.exists()) {
|
|
if (!g_quiet) {
|
|
std::cout << "Decoding image..." << std::endl;
|
|
}
|
|
std::ifstream input_stream(iname.getPath(), std::ios::binary);
|
|
Image inputImage = ImageDecoder::decode(input_stream, iname.getPath());
|
|
if (!inputImage.isValid()) {
|
|
std::cerr << "Unsupported image format!" << std::endl;
|
|
exit(0);
|
|
}
|
|
|
|
CubemapUtils::clamp(inputImage);
|
|
|
|
size_t width = inputImage.getWidth();
|
|
size_t height = inputImage.getHeight();
|
|
|
|
if ((isPOT(width) && (width * 3 == height * 4)) ||
|
|
(isPOT(height) && (height * 3 == width * 4))) {
|
|
// This is cross cubemap
|
|
size_t dim = std::max(height, width) / 4;
|
|
levels.push_back(Cubemap(dim));
|
|
if (!g_quiet) {
|
|
std::cout << "Loading cross... " << std::endl;
|
|
}
|
|
Cubemap& base(levels[0]);
|
|
CubemapUtils::setAllFacesFromCross(base, inputImage);
|
|
base.makeSeamless();
|
|
images.push_back(std::move(inputImage));
|
|
} else if (width == 2 * height) {
|
|
// we assume a spherical (equirectangular) image, which we will convert to a cross image
|
|
size_t dim = g_output_size ? g_output_size : 256;
|
|
if (!g_quiet) {
|
|
std::cout << "Converting equirectangular image... " << std::endl;
|
|
}
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, dim);
|
|
CubemapUtils::equirectangularToCubemap(cml, inputImage);
|
|
cml.makeSeamless();
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(cml));
|
|
} else {
|
|
std::cerr << "Aspect ratio not supported: " << width << "x" << height << std::endl;
|
|
std::cerr << "Supported aspect ratios:" << std::endl;
|
|
std::cerr << " 2:1, lat/long or equirectangular" << std::endl;
|
|
std::cerr << " 3:4, vertical cross (width must be power of two)" << std::endl;
|
|
std::cerr << " 4:3, horizontal cross (height must be power of two)" << std::endl;
|
|
exit(0);
|
|
}
|
|
} else {
|
|
if (!g_quiet) {
|
|
std::cout << "Generating image..." << std::endl;
|
|
}
|
|
|
|
size_t dim = g_output_size ? g_output_size : 256;
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, dim);
|
|
|
|
unsigned int p = 0;
|
|
std::string name = iname.getNameWithoutExtension();
|
|
if (sscanf(name.c_str(), "uv%u", &p) == 1) {
|
|
CubemapUtils::generateUVGrid(cml, p);
|
|
} else if (sscanf(name.c_str(), "brdf%u", &p) == 1) {
|
|
double linear_roughness = sq(p / std::log2(dim));
|
|
CubemapIBL::brdf(cml, linear_roughness);
|
|
} else {
|
|
CubemapUtils::generateUVGrid(cml, 1);
|
|
}
|
|
|
|
cml.makeSeamless();
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(cml));
|
|
}
|
|
|
|
// Now generate all the mipmap levels
|
|
generateMipmaps(levels, images);
|
|
|
|
if (g_mirror) {
|
|
if (!g_quiet) {
|
|
std::cout << "Mirroring..." << std::endl;
|
|
}
|
|
|
|
std::vector<Cubemap> mirrorLevels;
|
|
std::vector<Image> mirrorImages;
|
|
|
|
for (auto& level : levels) {
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, level.getDimensions());
|
|
CubemapUtils::mirrorCubemap(cml, level);
|
|
cml.makeSeamless();
|
|
|
|
mirrorImages.push_back(std::move(temp));
|
|
mirrorLevels.push_back(std::move(cml));
|
|
}
|
|
|
|
std::swap(levels, mirrorLevels);
|
|
std::swap(images, mirrorImages);
|
|
}
|
|
|
|
if (g_sh_compute) {
|
|
if (!g_quiet) {
|
|
std::cout << "Spherical harmonics..." << std::endl;
|
|
}
|
|
Cubemap const& cm(levels[0]);
|
|
sphericalHarmonics(iname, cm);
|
|
}
|
|
|
|
if (g_is_mipmap) {
|
|
if (!g_quiet) {
|
|
std::cout << "IBL mipmaps for prefiltered importance sampling..." << std::endl;
|
|
}
|
|
iblMipmapPrefilter(iname, images, levels, g_is_mipmap_dir);
|
|
}
|
|
|
|
if (g_prefilter) {
|
|
if (!g_quiet) {
|
|
std::cout << "IBL prefiltering..." << std::endl;
|
|
}
|
|
iblRoughnessPrefilter(iname, levels, g_prefilter_dir);
|
|
}
|
|
|
|
if (g_extract_faces) {
|
|
Cubemap const& cm(levels[0]);
|
|
if (g_extract_blur != 0) {
|
|
if (!g_quiet) {
|
|
std::cout << "Blurring..." << std::endl;
|
|
}
|
|
const double linear_roughness = g_extract_blur*g_extract_blur;
|
|
const size_t dim = g_output_size ? g_output_size : cm.getDimensions();
|
|
Image image;
|
|
Cubemap blurred = CubemapUtils::create(image, dim);
|
|
CubemapIBL::roughnessFilter(blurred, levels, linear_roughness, g_num_samples);
|
|
if (!g_quiet) {
|
|
std::cout << "Extract faces..." << std::endl;
|
|
}
|
|
extractCubemapFaces(iname, blurred, g_extract_dir);
|
|
} else {
|
|
if (!g_quiet) {
|
|
std::cout << "Extract faces..." << std::endl;
|
|
}
|
|
extractCubemapFaces(iname, cm, g_extract_dir);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void generateMipmaps(std::vector<Cubemap>& levels, std::vector<Image>& images) {
|
|
Image temp;
|
|
const Cubemap& base(levels[0]);
|
|
size_t dim = base.getDimensions();
|
|
size_t mipLevel = 0;
|
|
while (dim > 1) {
|
|
dim >>= 1;
|
|
Cubemap dst = CubemapUtils::create(temp, dim);
|
|
const Cubemap& src(levels[mipLevel++]);
|
|
CubemapUtils::downsampleCubemapLevelBoxFilter(dst, src);
|
|
dst.makeSeamless();
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(dst));
|
|
}
|
|
}
|
|
|
|
void sphericalHarmonics(const utils::Path& iname, const Cubemap& inputCubemap) {
|
|
std::unique_ptr<math::double3[]> sh;
|
|
if (g_sh_shader) {
|
|
sh = CubemapSH::computeIrradianceSH3Bands(inputCubemap);
|
|
} else {
|
|
sh = CubemapSH::computeSH(inputCubemap, g_sh_compute, g_sh_irradiance);
|
|
}
|
|
|
|
if (g_sh_output) {
|
|
outputSh(std::cout, sh, g_sh_compute);
|
|
}
|
|
|
|
if (g_sh_file != ShFile::SH_NONE || g_debug) {
|
|
Image image;
|
|
const size_t dim = g_output_size ? g_output_size : inputCubemap.getDimensions();
|
|
Cubemap cm = CubemapUtils::create(image, dim);
|
|
|
|
if (g_sh_file != ShFile::SH_NONE) {
|
|
utils::Path outputDir(g_sh_filename.getAbsolutePath().getParent());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
if (g_sh_shader) {
|
|
CubemapSH::renderPreScaledSH3Bands(cm, sh);
|
|
} else {
|
|
CubemapSH::renderSH(cm, sh, g_sh_compute);
|
|
}
|
|
|
|
if (g_sh_file == ShFile::SH_CROSS) {
|
|
std::ofstream outputStream(g_sh_filename, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream,
|
|
ImageEncoder::chooseFormat(g_sh_filename.getName()),
|
|
image, g_compression, g_sh_filename);
|
|
}
|
|
if (g_sh_file == ShFile::SH_TEXT) {
|
|
std::ofstream outputStream(g_sh_filename, std::ios::trunc);
|
|
outputSh(outputStream, sh, g_sh_compute);
|
|
}
|
|
}
|
|
|
|
if (g_debug) {
|
|
utils::Path outputDir(g_sh_filename.getAbsolutePath().getParent());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
{ // save a file with what we just calculated (radiance or irradiance)
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath =
|
|
outputDir + (basename + "_sh" + (g_sh_irradiance ? "_i" : "_r") + ".png");
|
|
std::ofstream outputStream(filePath, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, ImageEncoder::Format::PNG, image, "",
|
|
filePath.getPath());
|
|
}
|
|
|
|
{ // save a file with the "other one" (irradiance or radiance)
|
|
sh = CubemapSH::computeSH(inputCubemap, g_sh_compute, !g_sh_irradiance);
|
|
CubemapSH::renderSH(cm, sh, g_sh_compute);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath =
|
|
outputDir + (basename + "_sh" + (!g_sh_irradiance ? "_i" : "_r") + ".png");
|
|
std::ofstream outputStream(filePath, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, ImageEncoder::Format::PNG, image, "",
|
|
filePath.getPath());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void outputSh(std::ostream& out,
|
|
const std::unique_ptr<math::double3[]>& sh, size_t numBands) {
|
|
for (ssize_t l=0 ; l<numBands ; l++) {
|
|
for (ssize_t m=-l ; m<=l ; m++) {
|
|
size_t i = CubemapSH::getShIndex(m, (size_t) l);
|
|
std::string name = "L" + std::to_string(l) + std::to_string(m);
|
|
if (g_sh_irradiance) {
|
|
name.append(", irradiance");
|
|
}
|
|
if (g_sh_shader) {
|
|
name.append(", pre-scaled base");
|
|
}
|
|
out << "("
|
|
<< std::fixed << std::setprecision(15) << std::setw(18) << sh[i].r << ", "
|
|
<< std::fixed << std::setprecision(15) << std::setw(18) << sh[i].g << ", "
|
|
<< std::fixed << std::setprecision(15) << std::setw(18) << sh[i].b
|
|
<< "); // " << name
|
|
<< std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
void outputSpectrum(std::ostream& out,
|
|
const std::unique_ptr<math::double3[]>& sh, size_t numBands) {
|
|
// We assume a symetrical function (i.e. m!=0 terms are zero)
|
|
for (ssize_t l=0 ; l<numBands ; l++) {
|
|
size_t i = CubemapSH::getShIndex(0, (size_t) l);
|
|
double L = dot(sh[i], double3{ 0.2126, 0.7152, 0.0722 });
|
|
out << std::fixed << std::setprecision(15) << std::setw(18) << sq(L) << std::endl;
|
|
}
|
|
}
|
|
|
|
void iblMipmapPrefilter(const utils::Path& iname,
|
|
const std::vector<Image>& images, const std::vector<Cubemap>& levels,
|
|
const utils::Path& dir) {
|
|
utils::Path outputDir(dir.getAbsolutePath() + iname.getNameWithoutExtension());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
const size_t numLevels = levels.size();
|
|
for (size_t level=0 ; level<numLevels ; level++) {
|
|
Cubemap const& dst(levels[level]);
|
|
Image const& img(images[level]);
|
|
if (g_debug) {
|
|
ImageEncoder::Format debug_format = ImageEncoder::Format::PNG;
|
|
std::string ext = ImageEncoder::chooseExtension(debug_format);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath = outputDir + (basename + "_is_m" + (std::to_string(level) + ext));
|
|
|
|
std::ofstream outputStream(filePath, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, debug_format, img,
|
|
g_compression, filePath.getPath());
|
|
}
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
for (size_t i = 0; i < 6; i++) {
|
|
Cubemap::Face face = (Cubemap::Face)i;
|
|
std::string filename = outputDir
|
|
+ ("is_m" + std::to_string(level) + "_" + CubemapUtils::getFaceName(face) + ext);
|
|
std::ofstream outputStream(filename, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, g_format, dst.getImageForFace(face),
|
|
g_compression, filename);
|
|
}
|
|
}
|
|
}
|
|
|
|
void iblRoughnessPrefilter(const utils::Path& iname,
|
|
const std::vector<Cubemap>& levels, const utils::Path& dir) {
|
|
utils::Path outputDir(dir.getAbsolutePath() + iname.getNameWithoutExtension());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
// DEBUG: enable this to generate prefilter mipmaps at full resolution
|
|
// (of course, they're not mimaps at this point)
|
|
// This is useful for debugging.
|
|
const bool DEBUG_FULL_RESOLUTION = false;
|
|
|
|
const size_t baseExp = __builtin_ctz(g_output_size ? g_output_size : 256);
|
|
size_t numSamples = g_num_samples;
|
|
const size_t numLevels = baseExp + 1;
|
|
for (ssize_t i=baseExp ; i>=0 ; --i) {
|
|
const size_t dim = 1U << (DEBUG_FULL_RESOLUTION ? baseExp : i);
|
|
const size_t level = baseExp - i;
|
|
if (level >= 2) {
|
|
// starting at level 2, we increase the number of samples per level
|
|
// this helps as the filter gets wider, and since there are 4x less work
|
|
// per level, this doesn't slow things down a lot.
|
|
if (!DEBUG_FULL_RESOLUTION) {
|
|
numSamples *= 2;
|
|
}
|
|
}
|
|
const double roughness = saturate(level / (numLevels - 1.0));
|
|
const double linear_roughness = roughness * roughness;
|
|
if (!g_quiet) {
|
|
std::cout << "Level " << level << ", roughness = " <<
|
|
std::setprecision(3) << roughness << ", roughness(lin) = " << linear_roughness
|
|
<< std::endl;
|
|
}
|
|
Image image;
|
|
Cubemap dst = CubemapUtils::create(image, dim);
|
|
CubemapIBL::roughnessFilter(dst, levels, linear_roughness, numSamples);
|
|
|
|
if (g_debug) {
|
|
ImageEncoder::Format debug_format = ImageEncoder::Format::PNG;
|
|
std::string ext = ImageEncoder::chooseExtension(debug_format);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath = outputDir + (basename + "_roughness_m" + (std::to_string(level) + ext));
|
|
|
|
std::ofstream outputStream(filePath, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, debug_format, image,
|
|
g_compression, filePath.getPath());
|
|
}
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
for (size_t j = 0; j < 6; j++) {
|
|
Cubemap::Face face = (Cubemap::Face) j;
|
|
std::string filename = outputDir
|
|
+ ("m" + std::to_string(level) + "_" + CubemapUtils::getFaceName(face) + ext);
|
|
std::ofstream outputStream(filename, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, g_format, dst.getImageForFace(face),
|
|
g_compression, filename);
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool isTextFile(const utils::Path& filename) {
|
|
std::string extension(filename.getExtension());
|
|
return extension == "h" || extension == "hpp" ||
|
|
extension == "c" || extension == "cpp" ||
|
|
extension == "inc" || extension == "txt";
|
|
}
|
|
|
|
static bool isIncludeFile(const utils::Path& filename) {
|
|
std::string extension(filename.getExtension());
|
|
return extension == "inc";
|
|
}
|
|
|
|
void iblLutDfg(const utils::Path& filename, size_t size, bool multiscatter) {
|
|
std::unique_ptr<uint8_t[]> buf(new uint8_t[size*size*sizeof(float3)]);
|
|
Image image(std::move(buf), size, size, size*sizeof(float3), sizeof(float3));
|
|
CubemapIBL::DFG(image, multiscatter);
|
|
|
|
utils::Path outputDir(filename.getAbsolutePath().getParent());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
if (isTextFile(filename)) {
|
|
const bool isInclude = isIncludeFile(filename);
|
|
std::ofstream outputStream(filename, std::ios::trunc);
|
|
|
|
outputStream << "// generated with: cmgen --ibl-dfg=" << filename.c_str() << std::endl;
|
|
outputStream << "// DFG LUT stored as an RG16F texture, in GL order" << std::endl;
|
|
if (!isInclude) {
|
|
outputStream << "const uint16_t DFG_LUT[] = {";
|
|
}
|
|
for (size_t y = 0; y < size; y++) {
|
|
for (size_t x = 0; x < size; x++) {
|
|
if (x % 4 == 0) outputStream << std::endl << " ";
|
|
const half2 d = half2(static_cast<float3*>(image.getPixelRef(x, size - 1 - y))->xy);
|
|
const uint16_t r = *reinterpret_cast<const uint16_t*>(&d.r);
|
|
const uint16_t g = *reinterpret_cast<const uint16_t*>(&d.g);
|
|
outputStream << "0x" << std::setfill('0') << std::setw(4) << std::hex << r << ", ";
|
|
outputStream << "0x" << std::setfill('0') << std::setw(4) << std::hex << g << ", ";
|
|
}
|
|
}
|
|
if (!isInclude) {
|
|
outputStream << std::endl << "};" << std::endl;
|
|
}
|
|
|
|
outputStream << std::endl;
|
|
outputStream.flush();
|
|
outputStream.close();
|
|
} else {
|
|
std::ofstream outputStream(filename, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::Format format = ImageEncoder::chooseFormat(filename.getName(), true);
|
|
ImageEncoder::encode(outputStream, format, image, g_compression, filename.getPath());
|
|
}
|
|
}
|
|
|
|
void extractCubemapFaces(const utils::Path& iname, const Cubemap& cm, const utils::Path& dir) {
|
|
utils::Path outputDir(dir.getAbsolutePath() + iname.getNameWithoutExtension());
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
for (size_t i=0 ; i<6 ; i++) {
|
|
Cubemap::Face face = (Cubemap::Face)i;
|
|
std::string filename(outputDir + (CubemapUtils::getFaceName(face) + ext));
|
|
std::ofstream outputStream(filename, std::ios::binary | std::ios::trunc);
|
|
ImageEncoder::encode(outputStream, g_format, cm.getImageForFace(face),
|
|
g_compression, filename);
|
|
}
|
|
}
|