d14ad05078
The Cryptomatte layers in the compositor are saved compressed and saved in half precision in the File Output node. That's because the EXR writer makes decision about compression and half float based on channel names. And Cryptomatte are currently saved using RGBA channel names like other color images. To fix this, we follow the Cycles convention of using lowercase rgba for Cryptomatte images, allowing them to be saved without compression and always in full precision since they are otherwise useless. Fixes #124208, #87988. Pull Request: https://projects.blender.org/blender/blender/pulls/124508
108 lines
3.9 KiB
C++
108 lines
3.9 KiB
C++
/* SPDX-FileCopyrightText: 2021 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "COM_MetaData.h"
|
|
|
|
#include "BKE_image.h"
|
|
|
|
#include "RE_pipeline.h"
|
|
|
|
namespace blender::compositor {
|
|
|
|
void MetaData::add(const blender::StringRef key, const blender::StringRef value)
|
|
{
|
|
entries_.add(key, value);
|
|
}
|
|
|
|
bool MetaData::is_cryptomatte_layer() const
|
|
{
|
|
return entries_.contains(META_DATA_KEY_CRYPTOMATTE_HASH) ||
|
|
entries_.contains(META_DATA_KEY_CRYPTOMATTE_CONVERSION) ||
|
|
entries_.contains(META_DATA_KEY_CRYPTOMATTE_MANIFEST);
|
|
}
|
|
|
|
void MetaData::add_cryptomatte_entry(const blender::StringRef layer_name,
|
|
const blender::StringRefNull key,
|
|
const blender::StringRef value)
|
|
{
|
|
add(blender::bke::cryptomatte::BKE_cryptomatte_meta_data_key(layer_name, key), value);
|
|
}
|
|
|
|
void MetaData::replace_hash_neutral_cryptomatte_keys(const blender::StringRef layer_name)
|
|
{
|
|
std::string cryptomatte_hash = entries_.pop_default(META_DATA_KEY_CRYPTOMATTE_HASH, "");
|
|
std::string cryptomatte_conversion = entries_.pop_default(META_DATA_KEY_CRYPTOMATTE_CONVERSION,
|
|
"");
|
|
std::string cryptomatte_manifest = entries_.pop_default(META_DATA_KEY_CRYPTOMATTE_MANIFEST, "");
|
|
|
|
if (cryptomatte_hash.length() || cryptomatte_conversion.length() ||
|
|
cryptomatte_manifest.length())
|
|
{
|
|
add_cryptomatte_entry(layer_name, "name", layer_name);
|
|
}
|
|
if (cryptomatte_hash.length()) {
|
|
add_cryptomatte_entry(layer_name, "hash", cryptomatte_hash);
|
|
}
|
|
if (cryptomatte_conversion.length()) {
|
|
add_cryptomatte_entry(layer_name, "conversion", cryptomatte_conversion);
|
|
}
|
|
if (cryptomatte_manifest.length()) {
|
|
add_cryptomatte_entry(layer_name, "manifest", cryptomatte_manifest);
|
|
}
|
|
}
|
|
|
|
void MetaData::add_to_render_result(RenderResult *render_result) const
|
|
{
|
|
for (MapItem<std::string, std::string> entry : entries_.items()) {
|
|
BKE_render_result_stamp_data(render_result, entry.key.c_str(), entry.value.c_str());
|
|
}
|
|
}
|
|
|
|
void MetaData::for_each_entry(
|
|
FunctionRef<void(const std::string &, const std::string &)> callback) const
|
|
{
|
|
for (MapItem<std::string, std::string> entry : entries_.items()) {
|
|
callback(entry.key, entry.value);
|
|
}
|
|
}
|
|
|
|
void MetaDataExtractCallbackData::add_meta_data(blender::StringRef key,
|
|
blender::StringRefNull value)
|
|
{
|
|
if (!meta_data) {
|
|
meta_data = std::make_unique<MetaData>();
|
|
}
|
|
meta_data->add(key, value);
|
|
}
|
|
|
|
void MetaDataExtractCallbackData::set_cryptomatte_keys(blender::StringRef cryptomatte_layer_name)
|
|
{
|
|
manifest_key = blender::bke::cryptomatte::BKE_cryptomatte_meta_data_key(cryptomatte_layer_name,
|
|
"manifest");
|
|
hash_key = blender::bke::cryptomatte::BKE_cryptomatte_meta_data_key(cryptomatte_layer_name,
|
|
"hash");
|
|
conversion_key = blender::bke::cryptomatte::BKE_cryptomatte_meta_data_key(cryptomatte_layer_name,
|
|
"conversion");
|
|
}
|
|
|
|
void MetaDataExtractCallbackData::extract_cryptomatte_meta_data(void *_data,
|
|
const char *propname,
|
|
char *propvalue,
|
|
int /*len*/)
|
|
{
|
|
MetaDataExtractCallbackData *data = static_cast<MetaDataExtractCallbackData *>(_data);
|
|
blender::StringRefNull key(propname);
|
|
if (key == data->hash_key) {
|
|
data->add_meta_data(META_DATA_KEY_CRYPTOMATTE_HASH, propvalue);
|
|
}
|
|
else if (key == data->conversion_key) {
|
|
data->add_meta_data(META_DATA_KEY_CRYPTOMATTE_CONVERSION, propvalue);
|
|
}
|
|
else if (key == data->manifest_key) {
|
|
data->add_meta_data(META_DATA_KEY_CRYPTOMATTE_MANIFEST, propvalue);
|
|
}
|
|
}
|
|
|
|
} // namespace blender::compositor
|