c7a33a62a2
Instead of relying on them being included indirectly. Pull Request: https://projects.blender.org/blender/blender/pulls/134406
112 lines
3.6 KiB
C++
112 lines
3.6 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "image.hh"
|
|
#include "usd_private.hh"
|
|
|
|
#include <pxr/imaging/hio/imageRegistry.h>
|
|
|
|
#include "BLI_fileops.h"
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_path_utils.hh"
|
|
#include "BLI_string.h"
|
|
|
|
#include "BKE_image.hh"
|
|
#include "BKE_image_format.hh"
|
|
#include "BKE_image_save.hh"
|
|
#include "BKE_main.hh"
|
|
#include "BKE_packedFile.hh"
|
|
|
|
#include "hydra_scene_delegate.hh"
|
|
|
|
namespace blender::io::hydra {
|
|
|
|
static std::string cache_image_file(
|
|
Main *bmain, Scene *scene, Image *image, ImageUser *iuser, bool check_exist)
|
|
{
|
|
std::string file_path;
|
|
ImageSaveOptions opts;
|
|
if (BKE_image_save_options_init(&opts, bmain, scene, image, iuser, false, false)) {
|
|
const char *file_ext[BKE_IMAGE_PATH_EXT_MAX];
|
|
file_ext[0] = BLI_path_extension_or_end(image->id.name);
|
|
if (!pxr::HioImageRegistry::GetInstance().IsSupportedImageFile(image->id.name)) {
|
|
BKE_image_path_ext_from_imformat(&scene->r.im_format, file_ext);
|
|
BKE_image_format_free(&opts.im_format);
|
|
BKE_image_format_copy(&opts.im_format, &scene->r.im_format);
|
|
}
|
|
|
|
char file_name[FILE_MAX];
|
|
SNPRINTF(file_name, "img_%p%s", image, file_ext[0]);
|
|
|
|
file_path = blender::io::usd::get_image_cache_file(file_name);
|
|
if (check_exist && BLI_exists(file_path.c_str())) {
|
|
BKE_image_save_options_free(&opts);
|
|
return file_path;
|
|
}
|
|
|
|
opts.save_copy = true;
|
|
STRNCPY(opts.filepath, file_path.c_str());
|
|
if (BKE_image_save(nullptr, bmain, image, iuser, &opts)) {
|
|
CLOG_INFO(LOG_HYDRA_SCENE, 1, "%s -> %s", image->id.name, file_path.c_str());
|
|
}
|
|
else {
|
|
CLOG_ERROR(LOG_HYDRA_SCENE, "Can't save %s", file_path.c_str());
|
|
file_path = "";
|
|
}
|
|
}
|
|
BKE_image_save_options_free(&opts);
|
|
return file_path;
|
|
}
|
|
|
|
std::string cache_or_get_image_file(Main *bmain, Scene *scene, Image *image, ImageUser *iuser)
|
|
{
|
|
char str[FILE_MAX];
|
|
std::string file_path;
|
|
bool do_check_extension = false;
|
|
if (image->source == IMA_SRC_GENERATED) {
|
|
file_path = cache_image_file(bmain, scene, image, iuser, false);
|
|
}
|
|
else if (BKE_image_has_packedfile(image)) {
|
|
do_check_extension = true;
|
|
std::string dir_path = blender::io::usd::image_cache_file_path();
|
|
char *cached_path;
|
|
char subfolder[FILE_MAXDIR];
|
|
SNPRINTF(subfolder, "unpack_%p", image);
|
|
LISTBASE_FOREACH (ImagePackedFile *, ipf, &image->packedfiles) {
|
|
char path[FILE_MAX];
|
|
BLI_path_join(
|
|
path, sizeof(path), dir_path.c_str(), subfolder, BLI_path_basename(ipf->filepath));
|
|
cached_path = BKE_packedfile_unpack_to_file(nullptr,
|
|
BKE_main_blendfile_path(bmain),
|
|
dir_path.c_str(),
|
|
path,
|
|
ipf->packedfile,
|
|
PF_WRITE_LOCAL);
|
|
|
|
/* Take first successfully unpacked image. */
|
|
if (cached_path != nullptr) {
|
|
if (file_path.empty()) {
|
|
file_path = cached_path;
|
|
}
|
|
MEM_freeN(cached_path);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
do_check_extension = true;
|
|
BKE_image_user_file_path_ex(bmain, iuser, image, str, false, true);
|
|
file_path = str;
|
|
}
|
|
|
|
if (do_check_extension && !pxr::HioImageRegistry::GetInstance().IsSupportedImageFile(file_path))
|
|
{
|
|
file_path = cache_image_file(bmain, scene, image, iuser, true);
|
|
}
|
|
|
|
CLOG_INFO(LOG_HYDRA_SCENE, 1, "%s -> %s", image->id.name, file_path.c_str());
|
|
return file_path;
|
|
}
|
|
|
|
} // namespace blender::io::hydra
|