9aeec3a759
mipgen can now emit basis-encoded KTX2 files. Both the desktop and
web "suzanne" samples use this as a test for compressed textures.
This PR does not add KTX2 support to glTF, but it's on the way.
`BasisEncoder` has a builder style API that calls the basis encoder to
create KTX2 files. This hides some low-level BasisU features that we are
not using, like file I/O and mipmap generation.
`Ktx2Reader` is an easy-to-use API for creating Filament textures from
KTX2 files. Its API primarily consists of these two methods:
bool requestFormat(Texture::InternalFormat format);
Filament::Texture* load(const uint8_t* data, size_t size);
The first method is used to build an ordered list of formats that are
supported by your hardware. The second method consumes the contents of a
basis-encoded KTX2 file and attempts to produce a Filament texture with
a preferred format.
IMPORTANT: Our tools still let you use KTX1 for non-compressed images
because it is useful for HDR, but you can no longer use KTX1 for
block-compressed data.
Partial fix for #4771.
1280 lines
49 KiB
C++
1280 lines
49 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 "ProgressUpdater.h"
|
|
|
|
#include <ibl/Cubemap.h>
|
|
#include <ibl/CubemapIBL.h>
|
|
#include <ibl/CubemapSH.h>
|
|
#include <ibl/CubemapUtils.h>
|
|
#include <ibl/Image.h>
|
|
#include <ibl/utilities.h>
|
|
|
|
#include <imageio/ImageDecoder.h>
|
|
#include <imageio/ImageEncoder.h>
|
|
|
|
#include <image/Ktx1Bundle.h>
|
|
#include <image/ColorTransform.h>
|
|
|
|
#include <utils/JobSystem.h>
|
|
#include <utils/Path.h>
|
|
#include <utils/algorithm.h>
|
|
|
|
#include <math/scalar.h>
|
|
#include <math/vec4.h>
|
|
|
|
#include <cmath>
|
|
#include <fstream>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
#include <string.h>
|
|
|
|
#include <getopt/getopt.h>
|
|
|
|
|
|
using namespace filament::math;
|
|
using namespace filament::ibl;
|
|
using namespace image;
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
enum class ShFile {
|
|
SH_NONE, SH_FILE, SH_TEXT
|
|
};
|
|
|
|
static const size_t DFG_LUT_DEFAULT_SIZE = 128;
|
|
static const size_t IBL_DEFAULT_SIZE = 256;
|
|
static const size_t IBL_DEFAULT_MIN_LOD_SIZE = 16;
|
|
|
|
enum class OutputType {
|
|
FACES, KTX, EQUIRECT, OCTAHEDRON
|
|
};
|
|
|
|
static image::ImageEncoder::Format g_format = image::ImageEncoder::Format::PNG;
|
|
static OutputType g_type = OutputType::FACES;
|
|
static std::string g_compression;
|
|
static bool g_extract_faces = false;
|
|
static float g_extract_blur = 0.0;
|
|
static utils::Path g_extract_dir;
|
|
|
|
static size_t g_output_size = 0;
|
|
static size_t g_min_lod_size = 0;
|
|
|
|
static bool g_quiet = false;
|
|
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 float g_sh_window = 0.0f; // <0 none, 0=auto, or cutoff
|
|
static bool g_noclamp = true;
|
|
static ShFile g_sh_file = ShFile::SH_NONE;
|
|
static utils::Path g_sh_filename;
|
|
static std::unique_ptr<filament::math::float3[]> g_sh_coefficients;
|
|
|
|
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_dfg_cloth = false;
|
|
|
|
static bool g_ibl_irradiance = false;
|
|
static bool g_ibl_no_prefilter = false;
|
|
static utils::Path g_ibl_irradiance_dir;
|
|
|
|
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(utils::JobSystem& js, std::vector<Cubemap>& levels,
|
|
std::vector<Image>& images);
|
|
static void sphericalHarmonics(utils::JobSystem& js, const utils::Path& iname,
|
|
const Cubemap& inputCubemap);
|
|
static void iblRoughnessPrefilter(
|
|
utils::JobSystem& js, const utils::Path& iname, const std::vector<Cubemap>& levels,
|
|
bool prefilter, const utils::Path& dir);
|
|
static void iblDiffuseIrradiance(utils::JobSystem& js, const utils::Path& iname,
|
|
const std::vector<Cubemap>& levels, const utils::Path& dir);
|
|
static void iblMipmapPrefilter(utils::JobSystem& js, const utils::Path& iname,
|
|
const std::vector<Image>& images, const std::vector<Cubemap>& levels,
|
|
const utils::Path& dir);
|
|
static void iblLutDfg(utils::JobSystem& js, const utils::Path& filename, size_t size,
|
|
bool multiscatter,
|
|
bool cloth);
|
|
static void extractCubemapFaces(utils::JobSystem& js, const utils::Path& iname, const Cubemap& cm,
|
|
const utils::Path& dir);
|
|
static void outputSh(std::ostream& out, const std::unique_ptr<filament::math::float3[]>& sh, size_t numBands);
|
|
static void UTILS_UNUSED outputSpectrum(std::ostream& out,
|
|
const std::unique_ptr<filament::math::float3[]>& sh, size_t numBands);
|
|
static void saveImage(const std::string& path, ImageEncoder::Format format, const Image& image,
|
|
const std::string& compression);
|
|
static LinearImage toLinearImage(const Image& image);
|
|
static void exportKtxFaces(Ktx1Bundle& container, uint32_t miplevel, const Cubemap& cm);
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
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 an env map.\n"
|
|
"Cubemaps and equirectangular formats are both supported, automatically detected \n"
|
|
"according to the aspect ratio of the source image.\n"
|
|
"\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"
|
|
" --type=[cubemap|equirect|octahedron|ktx], -t [cubemap|equirect|octahedron|ktx]\n"
|
|
" Specify output type (default: cubemap)\n\n"
|
|
" --format=[exr|hdr|psd|rgbm|rgb32f|png|dds|ktx], -f [format]\n"
|
|
" Specify output file format. ktx implies -type=ktx.\n"
|
|
" KTX files are always KTX1 files, not KTX2.\n"
|
|
" They are encoded with 3-channel RGB_10_11_11_REV data\n\n"
|
|
" --compression=COMPRESSION, -c COMPRESSION\n"
|
|
" Format specific compression:\n"
|
|
" KTX: ignored\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"
|
|
" Also applies to DFG LUT\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"
|
|
" --clamp\n"
|
|
" Clamp environment before processing\n\n"
|
|
" --no-mirror\n"
|
|
" Skip mirroring of generated cubemaps (for assets with mirroring already backed in)\n\n"
|
|
" --ibl-samples=numSamples\n"
|
|
" Number of samples to use for IBL integrations (default 1024)\n\n"
|
|
" --ibl-ld=dir\n"
|
|
" Roughness pre-filter into <dir>\n\n"
|
|
" --sh-shader\n"
|
|
" Generate irradiance SH for shader code\n\n"
|
|
"\n"
|
|
"Private use only:\n"
|
|
" --ibl-dfg=filename.[exr|hdr|psd|png|rgbm|rgb32f|dds|h|hpp|c|cpp|inc|txt]\n"
|
|
" Compute 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-dfg-cloth\n"
|
|
" If --ibl-dfg is set, adds a 3rd channel to the DFG for cloth shading\n\n"
|
|
" --ibl-is-mipmap=dir\n"
|
|
" Generate mipmap for pre-filtered importance sampling\n\n"
|
|
" --ibl-irradiance=dir\n"
|
|
" Diffuse irradiance into <dir>\n\n"
|
|
" --ibl-no-prefilter\n"
|
|
" Use importance sampling instead of prefiltered importance sampling\n\n"
|
|
" --ibl-min-lod-size\n"
|
|
" Minimum LOD size [default: 16]\n\n"
|
|
" --sh=bands\n"
|
|
" SH decomposition of input cubemap\n\n"
|
|
" --sh-output=filename.[exr|hdr|psd|rgbm|rgb32f|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-window=cutoff|no|auto (default), -w cutoff|no|auto (default)\n"
|
|
" SH windowing to reduce ringing\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() {
|
|
static const char *license[] = {
|
|
#include "licenses/licenses.inc"
|
|
nullptr
|
|
};
|
|
|
|
const char **p = &license[0];
|
|
while (*p)
|
|
std::cout << *p++ << std::endl;
|
|
}
|
|
|
|
static int handleCommandLineArgments(int argc, char* argv[]) {
|
|
static constexpr const char* OPTSTR = "hqidt:f:c:s:x:w:S:";
|
|
static const struct option OPTIONS[] = {
|
|
{ "help", no_argument, nullptr, 'h' },
|
|
{ "license", no_argument, nullptr, 'l' },
|
|
{ "quiet", no_argument, nullptr, 'q' },
|
|
{ "type", required_argument, nullptr, 't' },
|
|
{ "format", required_argument, nullptr, 'f' },
|
|
{ "compression", required_argument, nullptr, 'c' },
|
|
{ "size", required_argument, nullptr, 's' },
|
|
{ "extract", required_argument, nullptr, 'e' },
|
|
{ "extract-blur", required_argument, nullptr, 'r' },
|
|
{ "sh", optional_argument, nullptr, 'z' },
|
|
{ "sh-output", required_argument, nullptr, 'o' },
|
|
{ "sh-irradiance", no_argument, nullptr, 'i' },
|
|
{ "sh-shader", no_argument, nullptr, 'b' },
|
|
{ "sh-window", required_argument, nullptr, 'w' },
|
|
{ "clamp", no_argument, nullptr, 'K' },
|
|
{ "ibl-is-mipmap", required_argument, nullptr, 'y' },
|
|
{ "ibl-ld", required_argument, nullptr, 'p' },
|
|
{ "ibl-irradiance", required_argument, nullptr, 'P' },
|
|
{ "ibl-dfg", required_argument, nullptr, 'a' },
|
|
{ "ibl-dfg-multiscatter", no_argument, nullptr, 'u' },
|
|
{ "ibl-dfg-cloth", no_argument, nullptr, 'C' },
|
|
{ "ibl-no-prefilter", no_argument, nullptr, 'n' },
|
|
{ "ibl-min-lod-size", required_argument, nullptr, 'S' },
|
|
{ "ibl-samples", required_argument, nullptr, 'k' },
|
|
{ "deploy", required_argument, nullptr, 'x' },
|
|
{ "no-mirror", no_argument, nullptr, 'm' },
|
|
{ "debug", no_argument, nullptr, 'd' },
|
|
{ nullptr, 0, nullptr, 0 } // termination of the option list
|
|
};
|
|
int opt;
|
|
int option_index = 0;
|
|
int num_sh_bands = 3;
|
|
bool format_specified = false;
|
|
bool type_specified = false;
|
|
bool ktx_format_requested = 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; // NOLINT
|
|
case 'l':
|
|
license();
|
|
exit(0);
|
|
break; // NOLINT
|
|
case 'q':
|
|
g_quiet = true;
|
|
break;
|
|
case 't':
|
|
if (arg == "cubemap") {
|
|
g_type = OutputType::FACES;
|
|
type_specified = true;
|
|
}
|
|
if (arg == "ktx") {
|
|
g_type = OutputType::KTX;
|
|
type_specified = true;
|
|
}
|
|
if ((arg == "equirect") || (arg == "equirectangular")) {
|
|
g_type = OutputType::EQUIRECT;
|
|
type_specified = true;
|
|
}
|
|
if (arg == "octahedron") {
|
|
g_type = OutputType::OCTAHEDRON;
|
|
type_specified = 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 == "rgb32f") {
|
|
g_format = ImageEncoder::Format::RGB_10_11_11_REV;
|
|
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;
|
|
}
|
|
if (arg == "ktx") {
|
|
ktx_format_requested = true;
|
|
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 'S':
|
|
g_min_lod_size = std::stoul(arg);
|
|
if (!isPOT(g_min_lod_size)) {
|
|
std::cerr << "min LOD 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_FILE;
|
|
g_sh_filename = arg;
|
|
if (g_sh_filename.getExtension() == "txt") {
|
|
g_sh_file = ShFile::SH_TEXT;
|
|
}
|
|
break;
|
|
case 'w':
|
|
if (arg == "auto") { g_sh_window = 0.0f; }
|
|
else if (arg == "no") { g_sh_window = -1.0f; }
|
|
else { g_sh_window = std::stof(arg); }
|
|
break;
|
|
case 'K':
|
|
g_noclamp = false;
|
|
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 'P':
|
|
g_ibl_irradiance = true;
|
|
g_ibl_irradiance_dir = arg;
|
|
break;
|
|
case 'n':
|
|
g_ibl_no_prefilter = true;
|
|
break;
|
|
case 'a':
|
|
g_dfg = true;
|
|
g_dfg_filename = arg;
|
|
break;
|
|
case 'u':
|
|
g_dfg_multiscatter = true;
|
|
break;
|
|
case 'C':
|
|
g_dfg_cloth = 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 (ktx_format_requested) {
|
|
g_type = OutputType::KTX;
|
|
type_specified = true;
|
|
}
|
|
|
|
if (g_deploy && !type_specified) {
|
|
g_type = OutputType::FACES;
|
|
}
|
|
|
|
if (g_deploy && !format_specified) {
|
|
g_format = ImageEncoder::Format::RGB_10_11_11_REV;
|
|
}
|
|
|
|
if (num_sh_bands && g_sh_compute) {
|
|
g_sh_compute = (size_t) num_sh_bands;
|
|
}
|
|
return optind;
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
utils::JobSystem js;
|
|
js.adopt();
|
|
|
|
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 : DFG_LUT_DEFAULT_SIZE;
|
|
iblLutDfg(js, g_dfg_filename, size, g_dfg_multiscatter, g_dfg_cloth);
|
|
if (num_args < 1) return 0;
|
|
}
|
|
|
|
std::string command(argv[option_index]);
|
|
utils::Path iname(command);
|
|
|
|
if (g_deploy) {
|
|
utils::Path sh_dir = g_deploy_dir;
|
|
|
|
// KTX files are self-contained and do not need to live in a subfolder.
|
|
if (g_type != OutputType::KTX) {
|
|
sh_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 = sh_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;
|
|
}
|
|
|
|
// 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);
|
|
LinearImage linputImage = ImageDecoder::decode(input_stream, iname.getPath());
|
|
if (!linputImage.isValid()) {
|
|
std::cerr << "Unable to open image: " << iname.getPath() << std::endl;
|
|
exit(1);
|
|
}
|
|
if (linputImage.getChannels() != 3) {
|
|
std::cerr << "Input image must be RGB (3 channels)! This image has "
|
|
<< linputImage.getChannels() << " channels." << std::endl;
|
|
exit(1);
|
|
}
|
|
|
|
// Convert from LinearImage to the deprecated Image object which is used throughout cmgen.
|
|
const size_t width = linputImage.getWidth(), height = linputImage.getHeight();
|
|
Image inputImage(width, height);
|
|
memcpy(inputImage.getData(), linputImage.getPixelRef(), height * inputImage.getBytesPerRow());
|
|
|
|
if (!g_noclamp) {
|
|
CubemapUtils::clamp(inputImage);
|
|
}
|
|
|
|
if ((isPOT(width) && (width * 3 == height * 4)) ||
|
|
(isPOT(height) && (height * 3 == width * 4))) {
|
|
// This is cross cubemap
|
|
size_t dim = g_output_size ? g_output_size : IBL_DEFAULT_SIZE;
|
|
if (!g_quiet) {
|
|
std::cout << "Loading cross... " << std::endl;
|
|
}
|
|
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, dim);
|
|
CubemapUtils::crossToCubemap(js, cml, inputImage);
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(cml));
|
|
} 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 : IBL_DEFAULT_SIZE;
|
|
if (!g_quiet) {
|
|
std::cout << "Converting equirectangular image... " << std::endl;
|
|
}
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, dim);
|
|
CubemapUtils::equirectangularToCubemap(js, cml, inputImage);
|
|
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 (height must be power of two)" << std::endl;
|
|
std::cerr << " 4:3, horizontal cross (width must be power of two)" << std::endl;
|
|
exit(0);
|
|
}
|
|
} else {
|
|
if (!g_quiet) {
|
|
std::cout << iname << " does not exist; generating UV grid..." << std::endl;
|
|
}
|
|
|
|
size_t dim = g_output_size ? g_output_size : IBL_DEFAULT_SIZE;
|
|
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) { // NOLINT
|
|
CubemapUtils::generateUVGrid(js, cml, p, p);
|
|
} else if (sscanf(name.c_str(), "u%u", &p) == 1) { // NOLINT
|
|
CubemapUtils::generateUVGrid(js, cml, p, 1);
|
|
} else if (sscanf(name.c_str(), "v%u", &p) == 1) { // NOLINT
|
|
CubemapUtils::generateUVGrid(js, cml, 1, p);
|
|
} else if (sscanf(name.c_str(), "brdf%u", &p) == 1) { // NOLINT
|
|
float linear_roughness = sq(p / std::log2(dim));
|
|
CubemapIBL::brdf(js, cml, linear_roughness);
|
|
} else {
|
|
CubemapUtils::generateUVGrid(js, cml, 1, 1);
|
|
}
|
|
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(cml));
|
|
}
|
|
|
|
// we mirror by default -- the mirror option in fact un-mirrors.
|
|
g_mirror = !g_mirror;
|
|
if (g_mirror) {
|
|
if (!g_quiet) {
|
|
std::cout << "Mirroring..." << std::endl;
|
|
}
|
|
Image temp;
|
|
Cubemap cml = CubemapUtils::create(temp, levels[0].getDimensions());
|
|
CubemapUtils::mirrorCubemap(js, cml, levels[0]);
|
|
std::swap(levels[0], cml);
|
|
std::swap(images[0], temp);
|
|
} else {
|
|
if (!g_quiet) {
|
|
std::cout << "Skipped mirroring." << std::endl;
|
|
}
|
|
}
|
|
|
|
// make the cubemap seamless
|
|
levels[0].makeSeamless();
|
|
|
|
// Now generate all the mipmap levels
|
|
generateMipmaps(js, levels, images);
|
|
|
|
if (g_sh_compute) {
|
|
if (!g_quiet) {
|
|
std::cout << "Spherical harmonics..." << std::endl;
|
|
}
|
|
Cubemap const& cm(levels[0]);
|
|
sphericalHarmonics(js, iname, cm);
|
|
}
|
|
|
|
if (g_is_mipmap) {
|
|
if (!g_quiet) {
|
|
std::cout << "IBL mipmaps for prefiltered importance sampling..." << std::endl;
|
|
}
|
|
iblMipmapPrefilter(js, iname, images, levels, g_is_mipmap_dir);
|
|
}
|
|
|
|
if (g_prefilter) {
|
|
if (!g_quiet) {
|
|
std::cout << "IBL prefiltering..." << std::endl;
|
|
}
|
|
iblRoughnessPrefilter(js, iname, levels, !g_ibl_no_prefilter, g_prefilter_dir);
|
|
}
|
|
|
|
if (g_ibl_irradiance) {
|
|
if (!g_quiet) {
|
|
std::cout << "IBL diffuse irradiance..." << std::endl;
|
|
}
|
|
iblDiffuseIrradiance(js, iname, levels, g_ibl_irradiance_dir);
|
|
}
|
|
|
|
if (g_extract_faces) {
|
|
Cubemap const& cm(levels[0]);
|
|
if (g_extract_blur != 0) {
|
|
ProgressUpdater updater(1);
|
|
if (!g_quiet) {
|
|
std::cout << "Blurring..." << std::endl;
|
|
updater.start();
|
|
}
|
|
const float 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(js, blurred, levels, linear_roughness, g_num_samples,
|
|
float3{ 1, 1, 1 }, !g_ibl_no_prefilter,
|
|
[](size_t index, float v, void* userdata) {
|
|
if (!g_quiet) {
|
|
((ProgressUpdater*) userdata)->update(index, v);
|
|
}
|
|
}, &updater);
|
|
if (!g_quiet) {
|
|
updater.stop();
|
|
std::cout << "Extract faces..." << std::endl;
|
|
}
|
|
extractCubemapFaces(js, iname, blurred, g_extract_dir);
|
|
} else {
|
|
if (!g_quiet) {
|
|
std::cout << "Extract faces..." << std::endl;
|
|
}
|
|
extractCubemapFaces(js, iname, cm, g_extract_dir);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void generateMipmaps(utils::JobSystem& js, 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 >>= 1u;
|
|
Cubemap dst = CubemapUtils::create(temp, dim);
|
|
const Cubemap& src(levels[mipLevel++]);
|
|
CubemapUtils::downsampleCubemapLevelBoxFilter(js, dst, src);
|
|
dst.makeSeamless();
|
|
images.push_back(std::move(temp));
|
|
levels.push_back(std::move(dst));
|
|
}
|
|
}
|
|
|
|
void sphericalHarmonics(utils::JobSystem& js, const utils::Path& iname, const Cubemap& inputCubemap) {
|
|
std::unique_ptr<filament::math::float3[]> sh;
|
|
if (g_sh_shader) {
|
|
sh = CubemapSH::computeSH(js, inputCubemap, 3, true);
|
|
} else {
|
|
sh = CubemapSH::computeSH(js, inputCubemap, g_sh_compute, g_sh_irradiance);
|
|
}
|
|
|
|
if (g_sh_window >= 0) {
|
|
CubemapSH::windowSH(sh, g_sh_compute, g_sh_window);
|
|
}
|
|
|
|
if (g_sh_shader) {
|
|
CubemapSH::preprocessSHForShader(sh);
|
|
}
|
|
|
|
if (!g_quiet && 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(js, cm, sh);
|
|
} else {
|
|
CubemapSH::renderSH(js, cm, sh, g_sh_compute);
|
|
}
|
|
|
|
cm.makeSeamless();
|
|
|
|
if (g_sh_file == ShFile::SH_FILE) {
|
|
Image image;
|
|
if (g_type == OutputType::EQUIRECT) {
|
|
size_t dim = cm.getDimensions();
|
|
image = Image(dim * 2, dim);
|
|
CubemapUtils::cubemapToEquirectangular(js, image, cm);
|
|
}
|
|
|
|
if (g_type == OutputType::OCTAHEDRON) {
|
|
size_t dim = cm.getDimensions();
|
|
image = Image(dim, dim);
|
|
CubemapUtils::cubemapToOctahedron(js, image, cm);
|
|
}
|
|
|
|
saveImage(g_sh_filename, ImageEncoder::chooseFormat(g_sh_filename.getName()),
|
|
image, g_compression);
|
|
}
|
|
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") + ".hdr");
|
|
CubemapUtils::highlight(image);
|
|
saveImage(filePath, ImageEncoder::Format::HDR, image, "");
|
|
}
|
|
|
|
{ // save a file with the "other one" (irradiance or radiance)
|
|
std::unique_ptr<filament::math::float3[]> sh
|
|
= CubemapSH::computeSH(js, inputCubemap, g_sh_compute, !g_sh_irradiance);
|
|
CubemapSH::renderSH(js, cm, sh, g_sh_compute);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath =
|
|
outputDir + (basename + "_sh" + (!g_sh_irradiance ? "_i" : "_r") + ".hdr");
|
|
CubemapUtils::highlight(image);
|
|
saveImage(filePath, ImageEncoder::Format::HDR, image, "");
|
|
}
|
|
}
|
|
}
|
|
// Stash the computed coefficients in case we need to use them at a later stage (e.g. KTX gen)
|
|
g_sh_coefficients = std::move(sh);
|
|
}
|
|
|
|
void outputSh(std::ostream& out,
|
|
const std::unique_ptr<filament::math::float3[]>& 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 UTILS_UNUSED outputSpectrum(std::ostream& out,
|
|
const std::unique_ptr<filament::math::float3[]>& 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);
|
|
float L = dot(sh[i], float3{ 0.2126, 0.7152, 0.0722 });
|
|
out << std::fixed << std::setprecision(15) << std::setw(18) << sq(L) << std::endl;
|
|
}
|
|
}
|
|
|
|
void iblMipmapPrefilter(utils::JobSystem& js, 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::HDR;
|
|
std::string ext = ImageEncoder::chooseExtension(debug_format);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath = outputDir + (basename + "_is_m" + (std::to_string(level) + ext));
|
|
saveImage(filePath, debug_format, img, g_compression);
|
|
}
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
|
|
if (g_type == OutputType::EQUIRECT) {
|
|
size_t dim = dst.getDimensions();
|
|
Image image(dim * 2, dim);
|
|
CubemapUtils::cubemapToEquirectangular(js, image, dst);
|
|
std::string filename = outputDir + ("is_m" + std::to_string(level) + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
continue;
|
|
}
|
|
|
|
if (g_type == OutputType::OCTAHEDRON) {
|
|
size_t dim = dst.getDimensions();
|
|
Image image(dim, dim);
|
|
CubemapUtils::cubemapToOctahedron(js, image, dst);
|
|
std::string filename = outputDir + ("is_m" + std::to_string(level) + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
continue;
|
|
}
|
|
|
|
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);
|
|
saveImage(filename, g_format, dst.getImageForFace(face), g_compression);
|
|
}
|
|
}
|
|
}
|
|
|
|
static float lodToPerceptualRoughness(float lod) noexcept {
|
|
// Inverse perceptualRoughness-to-LOD mapping:
|
|
// The LOD-to-perceptualRoughness mapping is a quadratic fit for
|
|
// log2(perceptualRoughness)+iblMaxMipLevel when iblMaxMipLevel is 4.
|
|
// We found empirically that this mapping works very well for a 256 cubemap with 5 levels used,
|
|
// but also scales well for other iblMaxMipLevel values.
|
|
const float a = 2.0f;
|
|
const float b = -1.0f;
|
|
return (lod != 0)
|
|
? saturate((std::sqrt(a * a + 4.0f * b * lod) - a) / (2.0f * b))
|
|
: 0.0f;
|
|
}
|
|
|
|
void iblRoughnessPrefilter(
|
|
utils::JobSystem& js, const utils::Path& iname, const std::vector<Cubemap>& levels,
|
|
bool prefilter, const utils::Path& dir) {
|
|
utils::Path outputDir = dir.getAbsolutePath();
|
|
if (g_type != OutputType::KTX) {
|
|
outputDir += iname.getNameWithoutExtension();
|
|
}
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
// DEBUG: enable this to generate pre-filter 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 = utils::ctz(g_output_size ? g_output_size : IBL_DEFAULT_SIZE);
|
|
size_t minLod = utils::ctz(g_min_lod_size ? g_min_lod_size : IBL_DEFAULT_MIN_LOD_SIZE);
|
|
if (minLod >= baseExp) {
|
|
minLod = 0;
|
|
}
|
|
|
|
size_t numSamples = g_num_samples;
|
|
const size_t numLevels = (baseExp + 1) - minLod;
|
|
|
|
// It's convenient to create an empty KTX bundle on the stack in this scope, regardless of
|
|
// whether KTX is requested. It does not consume memory if empty.
|
|
Ktx1Bundle container((uint32_t) numLevels, 1, true);
|
|
container.info() = {
|
|
.endianness = Ktx1Bundle::ENDIAN_DEFAULT,
|
|
.glType = Ktx1Bundle::R11F_G11F_B10F,
|
|
.glTypeSize = 1,
|
|
.glFormat = Ktx1Bundle::RGB,
|
|
.glInternalFormat = Ktx1Bundle::R11F_G11F_B10F,
|
|
.glBaseInternalFormat = Ktx1Bundle::R11F_G11F_B10F,
|
|
.pixelWidth = 1U << baseExp,
|
|
.pixelHeight = 1U << baseExp,
|
|
.pixelDepth = 0,
|
|
};
|
|
|
|
for (ssize_t i = baseExp; i >= ssize_t((baseExp + 1) - numLevels) ; --i) {
|
|
const size_t dim = 1U << (DEBUG_FULL_RESOLUTION ? baseExp : i); // NOLINT
|
|
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) { // NOLINT
|
|
numSamples *= 2;
|
|
}
|
|
}
|
|
|
|
const float lod = saturate(level / (numLevels - 1.0f));
|
|
// map the lod to a perceptualRoughness
|
|
const float perceptualRoughness = lodToPerceptualRoughness(lod);
|
|
const float roughness = perceptualRoughness * perceptualRoughness;
|
|
if (!g_quiet) {
|
|
std::cout << "Level " << level << std::setprecision(3)
|
|
<< ", roughness = " << roughness
|
|
<< ", roughness (perceptual) = " << perceptualRoughness
|
|
<< std::endl;
|
|
}
|
|
Image image;
|
|
Cubemap dst = CubemapUtils::create(image, dim);
|
|
|
|
ProgressUpdater updater(1);
|
|
if (!g_quiet) {
|
|
updater.start();
|
|
}
|
|
CubemapIBL::roughnessFilter(js, dst, levels, roughness, numSamples,
|
|
float3{ 1, 1, 1 }, prefilter,
|
|
[](size_t index, float v, void* userdata) {
|
|
if (!g_quiet) {
|
|
((ProgressUpdater*) userdata)->update(index, v);
|
|
}
|
|
}, &updater);
|
|
if (!g_quiet) {
|
|
updater.stop();
|
|
}
|
|
|
|
dst.makeSeamless();
|
|
|
|
if (g_debug) {
|
|
ImageEncoder::Format debug_format = ImageEncoder::Format::HDR;
|
|
std::string ext = ImageEncoder::chooseExtension(debug_format);
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
utils::Path filePath = outputDir + (basename + "_roughness_m" + (std::to_string(level) + ext));
|
|
saveImage(filePath, debug_format, image, g_compression);
|
|
}
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
|
|
if (g_type == OutputType::KTX) {
|
|
exportKtxFaces(container, (uint32_t) level, dst);
|
|
continue;
|
|
}
|
|
|
|
if (g_type == OutputType::EQUIRECT) {
|
|
Image outImage(dim * 2, dim);
|
|
CubemapUtils::cubemapToEquirectangular(js, outImage, dst);
|
|
std::string filename = outputDir + ("m" + std::to_string(level) + ext);
|
|
saveImage(filename, g_format, outImage, g_compression);
|
|
continue;
|
|
}
|
|
|
|
if (g_type == OutputType::OCTAHEDRON) {
|
|
Image outImage(dim, dim);
|
|
CubemapUtils::cubemapToOctahedron(js, outImage, dst);
|
|
std::string filename = outputDir + ("m" + std::to_string(level) + ext);
|
|
saveImage(filename, g_format, outImage, g_compression);
|
|
continue;
|
|
}
|
|
|
|
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);
|
|
saveImage(filename, g_format, dst.getImageForFace(face), g_compression);
|
|
}
|
|
}
|
|
|
|
if (g_type == OutputType::KTX) {
|
|
if (g_sh_coefficients) {
|
|
std::ostringstream sstr;
|
|
for (ssize_t l = 0; l < g_sh_compute; l++) {
|
|
for (ssize_t m = -l; m <= l; m++) {
|
|
auto v = g_sh_coefficients[CubemapSH::getShIndex(m, (size_t) l)];
|
|
sstr << v.r << " " << v.g << " " << v.b << "\n";
|
|
}
|
|
}
|
|
container.setMetadata("sh", sstr.str().c_str());
|
|
}
|
|
std::vector<uint8_t> fileContents(container.getSerializedLength());
|
|
container.serialize(fileContents.data(), (uint32_t) fileContents.size());
|
|
std::string filename = dir.getNameWithoutExtension() + "_ibl.ktx";
|
|
auto fullpath = outputDir + filename;
|
|
std::ofstream outputStream(fullpath.c_str(), std::ios::out | std::ios::binary);
|
|
outputStream.write((const char*) fileContents.data(), fileContents.size());
|
|
outputStream.close();
|
|
}
|
|
}
|
|
|
|
void iblDiffuseIrradiance(utils::JobSystem& js, 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();
|
|
}
|
|
|
|
const size_t baseExp = utils::ctz(g_output_size ? g_output_size : IBL_DEFAULT_SIZE);
|
|
size_t numSamples = g_num_samples;
|
|
const size_t dim = 1U << baseExp;
|
|
Image image;
|
|
Cubemap dst = CubemapUtils::create(image, dim);
|
|
|
|
ProgressUpdater updater(1);
|
|
if (!g_quiet) {
|
|
updater.start();
|
|
}
|
|
CubemapIBL::diffuseIrradiance(js, dst, levels, numSamples,
|
|
[](size_t index, float v, void* userdata) {
|
|
if (!g_quiet) {
|
|
((ProgressUpdater*) userdata)->update(index, v);
|
|
}
|
|
}, &updater);
|
|
if (!g_quiet) {
|
|
updater.stop();
|
|
}
|
|
|
|
dst.makeSeamless();
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
|
|
if (g_type == OutputType::EQUIRECT) {
|
|
size_t dim = dst.getDimensions();
|
|
Image image(dim * 2, dim);
|
|
CubemapUtils::cubemapToEquirectangular(js, image, dst);
|
|
std::string filename = outputDir + ("irradiance" + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
}
|
|
|
|
if (g_type == OutputType::OCTAHEDRON) {
|
|
size_t dim = dst.getDimensions();
|
|
Image image(dim, dim);
|
|
CubemapUtils::cubemapToOctahedron(js, image, dst);
|
|
std::string filename = outputDir + ("irradiance" + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
}
|
|
|
|
if (g_type == OutputType::FACES) {
|
|
for (size_t j = 0; j < 6; j++) {
|
|
Cubemap::Face face = (Cubemap::Face)j;
|
|
std::string filename =
|
|
outputDir + ("i_" + std::string(CubemapUtils::getFaceName(face)) + ext);
|
|
saveImage(filename, g_format, dst.getImageForFace(face), g_compression);
|
|
}
|
|
}
|
|
|
|
if (g_debug) {
|
|
ImageEncoder::Format debug_format = ImageEncoder::Format::HDR;
|
|
std::string basename = iname.getNameWithoutExtension();
|
|
std::string fileExt = ImageEncoder::chooseExtension(debug_format);
|
|
utils::Path filePath = outputDir + (basename + "_diffuse_irradiance" + fileExt);
|
|
saveImage(filePath, debug_format, image, "");
|
|
|
|
// this generates SHs from the importance-sampled version above. This is just used
|
|
// to compare the resuts and see if the later is better.
|
|
Image outImage;
|
|
Cubemap cm = CubemapUtils::create(outImage, dim);
|
|
auto sh = CubemapSH::computeSH(js, dst, g_sh_compute, false);
|
|
CubemapSH::renderSH(js, cm, sh, g_sh_compute);
|
|
filePath = outputDir + (basename + "_diffuse_irradiance_sh" + fileExt);
|
|
saveImage(filePath, debug_format, outImage, "");
|
|
}
|
|
}
|
|
|
|
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(utils::JobSystem& js, const utils::Path& filename, size_t size, bool multiscatter,
|
|
bool cloth) {
|
|
Image image(size, size);
|
|
CubemapIBL::DFG(js, image, multiscatter, cloth);
|
|
|
|
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 half3 d = half3(*static_cast<float3*>(image.getPixelRef(x, size - 1 - y)));
|
|
const uint16_t r = *reinterpret_cast<const uint16_t*>(&d.r);
|
|
const uint16_t g = *reinterpret_cast<const uint16_t*>(&d.g);
|
|
const uint16_t b = *reinterpret_cast<const uint16_t*>(&d.b);
|
|
outputStream << "0x" << std::setfill('0') << std::setw(4) << std::hex << r << ", ";
|
|
outputStream << "0x" << std::setfill('0') << std::setw(4) << std::hex << g << ", ";
|
|
if (g_dfg_cloth) {
|
|
outputStream << "0x" << std::setfill('0') << std::setw(4) << std::hex << b << ", ";
|
|
}
|
|
}
|
|
}
|
|
if (!isInclude) {
|
|
outputStream << std::endl << "};" << std::endl;
|
|
}
|
|
|
|
outputStream << std::endl;
|
|
outputStream.flush();
|
|
outputStream.close();
|
|
} else {
|
|
ImageEncoder::Format format = ImageEncoder::chooseFormat(filename.getName(), true);
|
|
saveImage(filename, format, image, g_compression);
|
|
}
|
|
}
|
|
|
|
void extractCubemapFaces(utils::JobSystem& js, const utils::Path& iname, const Cubemap& cm,
|
|
const utils::Path& dir) {
|
|
utils::Path outputDir(dir.getAbsolutePath());
|
|
if (g_type != OutputType::KTX) {
|
|
outputDir += iname.getNameWithoutExtension();
|
|
}
|
|
if (!outputDir.exists()) {
|
|
outputDir.mkdirRecursive();
|
|
}
|
|
|
|
if (g_type == OutputType::KTX) {
|
|
const uint32_t dim = (const uint32_t) cm.getDimensions();
|
|
Ktx1Bundle container(1, 1, true);
|
|
container.info() = {
|
|
.endianness = Ktx1Bundle::ENDIAN_DEFAULT,
|
|
.glType = Ktx1Bundle::R11F_G11F_B10F,
|
|
.glTypeSize = 1,
|
|
.glFormat = Ktx1Bundle::RGB,
|
|
.glInternalFormat = Ktx1Bundle::R11F_G11F_B10F,
|
|
.glBaseInternalFormat = Ktx1Bundle::R11F_G11F_B10F,
|
|
.pixelWidth = dim,
|
|
.pixelHeight = dim,
|
|
.pixelDepth = 0,
|
|
};
|
|
exportKtxFaces(container, 0, cm);
|
|
std::string filename = dir.getNameWithoutExtension() + "_skybox.ktx";
|
|
auto fullpath = outputDir + filename;
|
|
std::vector<uint8_t> fileContents(container.getSerializedLength());
|
|
container.serialize(fileContents.data(), (uint32_t) fileContents.size());
|
|
std::ofstream outputStream(fullpath.c_str(), std::ios::out | std::ios::binary);
|
|
outputStream.write((const char*) fileContents.data(), fileContents.size());
|
|
outputStream.close();
|
|
return;
|
|
}
|
|
|
|
std::string ext = ImageEncoder::chooseExtension(g_format);
|
|
|
|
if (g_type == OutputType::EQUIRECT) {
|
|
size_t dim = cm.getDimensions();
|
|
Image image(dim * 2, dim);
|
|
CubemapUtils::cubemapToEquirectangular(js, image, cm);
|
|
std::string filename = outputDir + ("skybox" + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
return;
|
|
}
|
|
|
|
if (g_type == OutputType::OCTAHEDRON) {
|
|
size_t dim = cm.getDimensions();
|
|
Image image(dim, dim);
|
|
CubemapUtils::cubemapToOctahedron(js, image, cm);
|
|
std::string filename = outputDir + ("skybox" + ext);
|
|
saveImage(filename, g_format, image, g_compression);
|
|
return;
|
|
}
|
|
|
|
for (size_t i = 0; i < 6; i++) {
|
|
Cubemap::Face face = (Cubemap::Face) i;
|
|
std::string filename(outputDir + (CubemapUtils::getFaceName(face) + ext));
|
|
saveImage(filename, g_format, cm.getImageForFace(face), g_compression);
|
|
}
|
|
}
|
|
|
|
// Converts a cmgen Image into a libimage LinearImage
|
|
static LinearImage toLinearImage(const Image& image) {
|
|
LinearImage linearImage((uint32_t) image.getWidth(), (uint32_t) image.getHeight(), 3);
|
|
|
|
// Copy row by row since the image has padding.
|
|
assert(image.getBytesPerPixel() == 12);
|
|
const size_t w = image.getWidth(), h = image.getHeight();
|
|
for (size_t row = 0; row < h; ++row) {
|
|
float* dst = linearImage.getPixelRef(0, (uint32_t) row);
|
|
float const* src = static_cast<float const*>(image.getPixelRef(0, row));
|
|
memcpy(dst, src, w * 12);
|
|
}
|
|
return linearImage;
|
|
}
|
|
|
|
static void saveImage(const std::string& path, ImageEncoder::Format format, const Image& image,
|
|
const std::string& compression) {
|
|
std::ofstream outputStream(path, std::ios::binary | std::ios::trunc);
|
|
if (!ImageEncoder::encode(outputStream, format, toLinearImage(image), compression, path)) {
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
static void exportKtxFaces(Ktx1Bundle& container, uint32_t miplevel, const Cubemap& cm) {
|
|
auto& info = container.info();
|
|
const uint32_t dim = (const uint32_t) cm.getDimensions();
|
|
for (uint32_t j = 0; j < 6; j++) {
|
|
KtxBlobIndex blobIndex {(uint32_t) miplevel, 0, j};
|
|
Cubemap::Face face;
|
|
switch (j) {
|
|
case 0: face = Cubemap::Face::PX; break;
|
|
case 1: face = Cubemap::Face::NX; break;
|
|
case 2: face = Cubemap::Face::PY; break;
|
|
case 3: face = Cubemap::Face::NY; break;
|
|
case 4: face = Cubemap::Face::PZ; break;
|
|
case 5: face = Cubemap::Face::NZ; break;
|
|
default: face = Cubemap::Face::PX; break; // make linters happy
|
|
}
|
|
LinearImage image = toLinearImage(cm.getImageForFace(face));
|
|
auto uintData = fromLinearToRGB_10_11_11_REV(image);
|
|
container.setBlob(blobIndex, uintData.get(), dim * dim * 4);
|
|
}
|
|
}
|