2 * Copyright (c) 2023 Samsung Electronics Co., Ltd.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
19 #include <dali-scene3d/public-api/loader/environment-map-loader.h>
22 #include <dali-scene3d/public-api/loader/ktx-loader.h>
25 #include <dali/devel-api/adaptor-framework/image-loading.h>
33 const std::string_view KTX_EXTENSION = ".ktx";
35 static constexpr uint32_t NUMBER_OF_ENVIRONMENT_MAP_TYPE = 5;
36 static constexpr uint32_t NUMBER_OF_CUBE_MAP_TYPE = 4;
39 * @brief cube map face index
40 * Cube map layer order is as fallows:
41 * POSITIVE_X, NEGATIVE_X, POSITIVE_Y, NEGATIVE_Y, POSITIVE_Z, NEGATIVE_Z. @see CubeMapLayer
42 * The indices are for 4 kind of environment cube map. Cross_horizontal, Array_horizontal, Cross_vertical, and Array_vertical.
44 static constexpr uint32_t CUBEMAP_INDEX_X[NUMBER_OF_CUBE_MAP_TYPE][6] = {{2, 0, 1, 1, 1, 3}, {0, 1, 2, 3, 4, 5}, {1, 1, 1, 1, 0, 2}, {0, 0, 0, 0, 0, 0}};
45 static constexpr uint32_t CUBEMAP_INDEX_Y[NUMBER_OF_CUBE_MAP_TYPE][6] = {{1, 1, 0, 2, 1, 1}, {0, 0, 0, 0, 0, 0}, {1, 3, 0, 2, 1, 1}, {0, 1, 2, 3, 4, 5}};
47 static constexpr Vector2 NUMBER_OF_CUBE_FACE[NUMBER_OF_CUBE_MAP_TYPE] =
55 static constexpr float expectedAspectRatios[NUMBER_OF_ENVIRONMENT_MAP_TYPE] =
66 CROSS_HORIZONTAL = 0, // Cross horizontal style cube map
67 ARRAY_HORIZONTAL, // array horizontal style cube map
68 CROSS_VERTICAL, // Cross vertical style cube map
69 ARRAY_VERTICAL, // array vertical style cube map
73 uint8_t* GetCroppedBuffer(uint8_t* sourceBuffer, uint32_t bytesPerPixel, uint32_t width, uint32_t height, uint32_t xOffset, uint32_t yOffset, uint32_t xFaceSize, uint32_t yFaceSize)
75 uint32_t byteSize = bytesPerPixel * xFaceSize * yFaceSize;
76 uint8_t* destBuffer = reinterpret_cast<uint8_t*>(malloc(byteSize + 4u));
78 int32_t srcStride = width * bytesPerPixel;
79 int32_t destStride = xFaceSize * bytesPerPixel;
80 int32_t srcOffset = xOffset * bytesPerPixel + yOffset * srcStride;
81 int32_t destOffset = 0;
82 for(uint16_t row = yOffset; row < yOffset + yFaceSize; ++row)
84 memcpy(destBuffer + destOffset, sourceBuffer + srcOffset, destStride);
85 srcOffset += srcStride;
86 destOffset += destStride;
92 PixelData GetCubeFace(Devel::PixelBuffer pixelBuffer, uint32_t faceIndex, CubeType cubeType, float faceWidth, float faceHeight)
97 uint8_t* imageBuffer = pixelBuffer.GetBuffer();
98 uint32_t bytesPerPixel = Pixel::GetBytesPerPixel(pixelBuffer.GetPixelFormat());
99 uint32_t imageWidth = pixelBuffer.GetWidth();
100 uint32_t imageHeight = pixelBuffer.GetHeight();
102 uint32_t xOffset = CUBEMAP_INDEX_X[cubeType][faceIndex] * static_cast<uint32_t>(faceWidth);
103 uint32_t yOffset = CUBEMAP_INDEX_Y[cubeType][faceIndex] * static_cast<uint32_t>(faceHeight);
105 uint32_t finalFaceWidth = (xOffset + static_cast<uint32_t>(faceWidth) < imageWidth) ? static_cast<uint32_t>(faceWidth) : imageWidth - xOffset;
106 uint32_t finalFaceHeight = (yOffset + static_cast<uint32_t>(faceHeight) < imageHeight) ? static_cast<uint32_t>(faceHeight) : imageHeight - yOffset;
107 uint8_t* tempImageBuffer = GetCroppedBuffer(imageBuffer, bytesPerPixel, imageWidth, imageHeight, xOffset, yOffset, finalFaceWidth, finalFaceHeight);
108 pixelData = PixelData::New(tempImageBuffer, finalFaceWidth * finalFaceHeight * bytesPerPixel, finalFaceWidth, finalFaceHeight, pixelBuffer.GetPixelFormat(), PixelData::FREE);
114 * @brief Loads environment map data texture from an image url.
116 * @param[in] environmentMapUrl The environment map file url.
117 * @param[out] environmentMapData The data structure with all pixel data objects.
118 * @return bool True if the loading is succeded.
120 bool LoadEnvironmentMapData(const std::string& environmentMapUrl, Scene3D::Loader::EnvironmentMapData& environmentMapData)
122 // Diffuse Environment Map
123 if(environmentMapUrl.empty())
128 Devel::PixelBuffer pixelBuffer = LoadImageFromFile(environmentMapUrl);
131 CubeType cubeType = NONE;
132 uint32_t imageWidth = pixelBuffer.GetWidth();
133 uint32_t imageHeight = pixelBuffer.GetHeight();
135 * If the environment map type is not EQUIRECTANGULAR,
136 * The type should be defined internally.
137 * DALi checkes aspect ratio of input image and find the closest type.
138 * If the environment map type is CUBEMAP, DALi determines which of the following styles is closest:
139 * cross horizontal, cross vertical, array horizontal, and array vertical.
140 * When the environment map type is AUTO, it finds the closest type, including the Equirectangular type.
142 if(environmentMapData.GetEnvironmentMapType() != Scene3D::EnvironmentMapType::EQUIRECTANGULAR)
144 float aspectRatio = (float)imageWidth / (float)imageHeight;
146 float minDistance = FLT_MAX;
147 uint32_t typeCount = (environmentMapData.GetEnvironmentMapType() == Scene3D::EnvironmentMapType::CUBEMAP) ? NUMBER_OF_CUBE_MAP_TYPE : NUMBER_OF_ENVIRONMENT_MAP_TYPE;
148 for(uint32_t i = 0; i < typeCount; ++i)
150 float distance = fabs(aspectRatio - expectedAspectRatios[i]);
151 if(distance < minDistance)
153 minDistance = distance;
154 cubeType = static_cast<CubeType>(i);
161 float faceWidth = imageWidth, faceHeight = imageHeight;
162 faceWidth /= NUMBER_OF_CUBE_FACE[static_cast<uint32_t>(cubeType)].x;
163 faceHeight /= NUMBER_OF_CUBE_FACE[static_cast<uint32_t>(cubeType)].y;
164 environmentMapData.mPixelData.resize(6);
165 for(uint32_t i = 0; i < 6; ++i)
167 environmentMapData.mPixelData[i].resize(1);
169 for(uint32_t i = 0; i < 6; ++i)
171 environmentMapData.mPixelData[i][0] = GetCubeFace(pixelBuffer, i, cubeType, faceWidth, faceHeight);
173 environmentMapData.SetEnvironmentMapType(Scene3D::EnvironmentMapType::CUBEMAP);
177 environmentMapData.mPixelData.resize(1);
178 environmentMapData.mPixelData[0].resize(1);
179 environmentMapData.mPixelData[0][0] = Devel::PixelBuffer::Convert(pixelBuffer);
180 environmentMapData.SetEnvironmentMapType(Scene3D::EnvironmentMapType::EQUIRECTANGULAR);
193 bool LoadEnvironmentMap(const std::string& environmentMapUrl, EnvironmentMapData& environmentMapData)
195 std::filesystem::path modelPath(environmentMapUrl);
196 std::string extension = modelPath.extension();
197 std::transform(extension.begin(), extension.end(), extension.begin(), ::tolower);
199 return (extension == KTX_EXTENSION) ? Dali::Scene3D::Loader::LoadKtxData(environmentMapUrl, environmentMapData) : LoadEnvironmentMapData(environmentMapUrl, environmentMapData);
202 } // namespace Loader
203 } // namespace Scene3D