Spaces:
Build error
Build error
File size: 2,223 Bytes
28451f7 |
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 84 85 |
/*
* SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*
* 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.
*/
/** @file vec_json.h
* @author Thomas Müller, NVIDIA
* @brief Conversion between tcnn's vector / matrix / quaternion types
* and nlohmann::json.
*/
#pragma once
#include <tiny-cuda-nn/common.h>
#include <json/json.hpp>
namespace tcnn {
template <typename T, uint32_t N, uint32_t M>
void to_json(nlohmann::json& j, const tmat<T, N, M>& mat) {
for (int row = 0; row < M; ++row) {
nlohmann::json column = nlohmann::json::array();
for (int col = 0; col < N; ++col) {
column.push_back(mat[col][row]);
}
j.push_back(column);
}
}
template <typename T, uint32_t N, uint32_t M>
void from_json(const nlohmann::json& j, tmat<T, N, M>& mat) {
for (std::size_t row = 0; row < M; ++row) {
const auto& jrow = j.at(row);
for (std::size_t col = 0; col < N; ++col) {
const auto& value = jrow.at(col);
mat[col][row] = value.get<T>();
}
}
}
template <typename T, uint32_t N>
void to_json(nlohmann::json& j, const tvec<T, N>& v) {
for (uint32_t i = 0; i < N; ++i) {
j.push_back(v[i]);
}
}
template <typename T, uint32_t N>
void from_json(const nlohmann::json& j, tvec<T, N>& v) {
for (uint32_t i = 0; i < N; ++i) {
v[i] = j.at(i).get<T>();
}
}
template <typename T>
void to_json(nlohmann::json& j, const tquat<T>& q) {
j.push_back(q.x);
j.push_back(q.y);
j.push_back(q.z);
j.push_back(q.w);
}
template <typename T>
void from_json(const nlohmann::json& j, tquat<T>& q) {
q.x = j.at(0).get<T>();
q.y = j.at(1).get<T>();
q.z = j.at(2).get<T>();
q.w = j.at(3).get<T>();
}
}
|