File size: 2,361 Bytes
7873319 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | /*
* Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
*
* NVIDIA CORPORATION and its licensors retain all intellectual property
* and proprietary rights in and to this software, related documentation
* and any modifications thereto. Any use, reproduction, disclosure or
* distribution of this software and related documentation without an express
* license agreement from NVIDIA CORPORATION is strictly prohibited.
*/
/** @file common_device.cu
* @author Thomas Müller, NVIDIA
*/
#include <neural-graphics-primitives/common_device.cuh>
#include <neural-graphics-primitives/tinyexr_wrapper.h>
#include <unsupported/Eigen/MatrixFunctions>
#include <stb_image/stb_image.h>
using namespace Eigen;
using namespace tcnn;
NGP_NAMESPACE_BEGIN
Matrix<float, 3, 4> log_space_lerp(const Matrix<float, 3, 4>& begin, const Matrix<float, 3, 4>& end, float t) {
Matrix4f A = Matrix4f::Identity();
A.block<3,4>(0,0) = begin;
Matrix4f B = Matrix4f::Identity();
B.block<3,4>(0,0) = end;
Matrix4f log_space_a_to_b = (B * A.inverse()).log();
return ((log_space_a_to_b * t).exp() * A).block<3,4>(0,0);
}
GPUMemory<float> load_exr(const std::string& filename, int& width, int& height) {
float* out; // width * height * RGBA
load_exr(&out, &width, &height, filename.c_str());
ScopeGuard mem_guard{[&]() { free(out); }};
GPUMemory<float> result(width * height * 4);
result.copy_from_host(out);
return result;
}
GPUMemory<float> load_stbi(const std::string& filename, int& width, int& height) {
bool is_hdr = stbi_is_hdr(filename.c_str());
void* data; // width * height * RGBA
int comp;
if (is_hdr) {
data = stbi_loadf(filename.c_str(), &width, &height, &comp, 4);
} else {
data = stbi_load(filename.c_str(), &width, &height, &comp, 4);
}
if (!data) {
throw std::runtime_error{std::string{stbi_failure_reason()}};
}
ScopeGuard mem_guard{[&]() { stbi_image_free(data); }};
if (width == 0 || height == 0) {
throw std::runtime_error{"Image has zero pixels."};
}
GPUMemory<float> result(width * height * 4);
if (is_hdr) {
result.copy_from_host((float*)data);
} else {
GPUMemory<uint8_t> bytes(width * height * 4);
bytes.copy_from_host((uint8_t*)data);
linear_kernel(from_rgba32<float>, 0, nullptr, width * height, bytes.data(), result.data(), false, false, 0);
}
return result;
}
NGP_NAMESPACE_END
|