texture_loader_ktx.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613
  1. /**************************************************************************/
  2. /* texture_loader_ktx.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /**************************************************************************/
  30. #include "texture_loader_ktx.h"
  31. #include "core/io/file_access.h"
  32. #include "core/io/file_access_memory.h"
  33. #include "scene/resources/image_texture.h"
  34. #include "servers/rendering/rendering_server.h"
  35. #include <ktx.h>
  36. #include <vk_format.h>
  37. KTX_error_code ktx_read(ktxStream *stream, void *dst, const ktx_size_t count) {
  38. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  39. (*f)->get_buffer(reinterpret_cast<uint8_t *>(dst), count);
  40. return KTX_SUCCESS;
  41. }
  42. KTX_error_code ktx_skip(ktxStream *stream, const ktx_size_t count) {
  43. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  44. for (ktx_size_t i = 0; i < count; ++i) {
  45. (*f)->get_8();
  46. }
  47. return KTX_SUCCESS;
  48. }
  49. KTX_error_code ktx_write(ktxStream *stream, const void *src, const ktx_size_t size, const ktx_size_t count) {
  50. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  51. (*f)->store_buffer(reinterpret_cast<const uint8_t *>(src), size * count);
  52. return KTX_SUCCESS;
  53. }
  54. KTX_error_code ktx_getpos(ktxStream *stream, ktx_off_t *const offset) {
  55. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  56. *offset = (*f)->get_position();
  57. return KTX_SUCCESS;
  58. }
  59. KTX_error_code ktx_setpos(ktxStream *stream, const ktx_off_t offset) {
  60. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  61. (*f)->seek(offset);
  62. return KTX_SUCCESS;
  63. }
  64. KTX_error_code ktx_getsize(ktxStream *stream, ktx_size_t *const size) {
  65. Ref<FileAccess> *f = reinterpret_cast<Ref<FileAccess> *>(stream->data.custom_ptr.address);
  66. *size = (*f)->get_length();
  67. return KTX_SUCCESS;
  68. }
  69. void ktx_destruct(ktxStream *stream) {
  70. (void)stream;
  71. }
  72. static Ref<Image> load_from_file_access(Ref<FileAccess> f, Error *r_error) {
  73. ktxStream ktx_stream;
  74. ktx_stream.read = ktx_read;
  75. ktx_stream.skip = ktx_skip;
  76. ktx_stream.write = ktx_write;
  77. ktx_stream.getpos = ktx_getpos;
  78. ktx_stream.setpos = ktx_setpos;
  79. ktx_stream.getsize = ktx_getsize;
  80. ktx_stream.destruct = ktx_destruct;
  81. ktx_stream.type = eStreamTypeCustom;
  82. ktx_stream.data.custom_ptr.address = &f;
  83. ktx_stream.data.custom_ptr.allocatorAddress = nullptr;
  84. ktx_stream.data.custom_ptr.size = 0;
  85. ktx_stream.readpos = 0;
  86. ktx_stream.closeOnDestruct = false;
  87. ktxTexture *ktx_texture;
  88. KTX_error_code result = ktxTexture_CreateFromStream(&ktx_stream,
  89. KTX_TEXTURE_CREATE_LOAD_IMAGE_DATA_BIT,
  90. &ktx_texture);
  91. if (result != KTX_SUCCESS) {
  92. ERR_FAIL_V_MSG(Ref<Resource>(), "Invalid or unsupported KTX texture file.");
  93. }
  94. if (ktx_texture->numDimensions != 2) {
  95. ktxTexture_Destroy(ktx_texture);
  96. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported non-2D KTX texture file.");
  97. }
  98. if (ktx_texture->isCubemap || ktx_texture->numFaces != 1) {
  99. ktxTexture_Destroy(ktx_texture);
  100. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported cube map KTX texture file.");
  101. }
  102. if (ktx_texture->isArray) {
  103. ktxTexture_Destroy(ktx_texture);
  104. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported array KTX texture file.");
  105. }
  106. uint32_t width = ktx_texture->baseWidth;
  107. uint32_t height = ktx_texture->baseHeight;
  108. uint32_t mipmaps = ktx_texture->numLevels;
  109. Image::Format format;
  110. bool srgb = false;
  111. switch (ktx_texture->classId) {
  112. case ktxTexture1_c:
  113. switch (((ktxTexture1 *)ktx_texture)->glInternalformat) {
  114. case GL_LUMINANCE:
  115. format = Image::FORMAT_L8;
  116. break;
  117. case GL_LUMINANCE_ALPHA:
  118. format = Image::FORMAT_LA8;
  119. break;
  120. case GL_SRGB8:
  121. format = Image::FORMAT_RGB8;
  122. srgb = true;
  123. break;
  124. case GL_SRGB8_ALPHA8:
  125. format = Image::FORMAT_RGBA8;
  126. srgb = true;
  127. break;
  128. case GL_R8:
  129. case GL_R8UI:
  130. format = Image::FORMAT_R8;
  131. break;
  132. case GL_RG8:
  133. format = Image::FORMAT_RG8;
  134. break;
  135. case GL_RGB8:
  136. format = Image::FORMAT_RGB8;
  137. break;
  138. case GL_RGBA8:
  139. format = Image::FORMAT_RGBA8;
  140. break;
  141. case GL_RGBA4:
  142. format = Image::FORMAT_RGBA4444;
  143. break;
  144. case GL_RGB565:
  145. format = Image::FORMAT_RGB565;
  146. break;
  147. case GL_R32F:
  148. format = Image::FORMAT_RF;
  149. break;
  150. case GL_RG32F:
  151. format = Image::FORMAT_RGF;
  152. break;
  153. case GL_RGB32F:
  154. format = Image::FORMAT_RGBF;
  155. break;
  156. case GL_RGBA32F:
  157. format = Image::FORMAT_RGBAF;
  158. break;
  159. case GL_R16F:
  160. format = Image::FORMAT_RH;
  161. break;
  162. case GL_RG16F:
  163. format = Image::FORMAT_RGH;
  164. break;
  165. case GL_RGB16F:
  166. format = Image::FORMAT_RGBH;
  167. break;
  168. case GL_RGBA16F:
  169. format = Image::FORMAT_RGBAH;
  170. break;
  171. case GL_RGB9_E5:
  172. format = Image::FORMAT_RGBE9995;
  173. break;
  174. case GL_R16:
  175. format = Image::FORMAT_R16;
  176. break;
  177. case GL_RG16:
  178. format = Image::FORMAT_RG16;
  179. break;
  180. case GL_RGB16:
  181. format = Image::FORMAT_RGB16;
  182. break;
  183. case GL_RGBA16:
  184. format = Image::FORMAT_RGBA16;
  185. break;
  186. case GL_R16UI:
  187. format = Image::FORMAT_R16I;
  188. break;
  189. case GL_RG16UI:
  190. format = Image::FORMAT_RG16I;
  191. break;
  192. case GL_RGB16UI:
  193. format = Image::FORMAT_RGB16I;
  194. break;
  195. case GL_RGBA16UI:
  196. format = Image::FORMAT_RGBA16I;
  197. break;
  198. case GL_COMPRESSED_RGB_S3TC_DXT1_EXT:
  199. case GL_COMPRESSED_RGBA_S3TC_DXT1_EXT:
  200. format = Image::FORMAT_DXT1;
  201. break;
  202. case GL_COMPRESSED_RGBA_S3TC_DXT3_EXT:
  203. format = Image::FORMAT_DXT3;
  204. break;
  205. case GL_COMPRESSED_RGBA_S3TC_DXT5_EXT:
  206. format = Image::FORMAT_DXT5;
  207. break;
  208. case GL_COMPRESSED_RED_RGTC1:
  209. format = Image::FORMAT_RGTC_R;
  210. break;
  211. case GL_COMPRESSED_RG_RGTC2:
  212. format = Image::FORMAT_RGTC_RG;
  213. break;
  214. case GL_COMPRESSED_RGB_BPTC_UNSIGNED_FLOAT:
  215. format = Image::FORMAT_BPTC_RGBFU;
  216. break;
  217. case GL_COMPRESSED_RGBA_BPTC_UNORM:
  218. format = Image::FORMAT_BPTC_RGBA;
  219. break;
  220. case GL_ETC1_RGB8_OES:
  221. format = Image::FORMAT_ETC;
  222. break;
  223. case GL_COMPRESSED_R11_EAC:
  224. format = Image::FORMAT_ETC2_R11;
  225. break;
  226. // Decompression is not supported for this format.
  227. /*case GL_COMPRESSED_SIGNED_R11_EAC:
  228. format = Image::FORMAT_ETC2_R11S;
  229. break;*/
  230. case GL_COMPRESSED_RG11_EAC:
  231. format = Image::FORMAT_ETC2_RG11;
  232. break;
  233. // Decompression is not supported for this format.
  234. /*case GL_COMPRESSED_SIGNED_RG11_EAC:
  235. format = Image::FORMAT_ETC2_RG11S;
  236. break;*/
  237. case GL_COMPRESSED_RGB8_ETC2:
  238. format = Image::FORMAT_ETC2_RGB8;
  239. break;
  240. case GL_COMPRESSED_RGBA8_ETC2_EAC:
  241. format = Image::FORMAT_ETC2_RGBA8;
  242. break;
  243. // Decompression is not supported for this format.
  244. /*case GL_COMPRESSED_RGB8_PUNCHTHROUGH_ALPHA1_ETC2:
  245. format = Image::FORMAT_ETC2_RGB8A1;
  246. break;*/
  247. case GL_COMPRESSED_SRGB8_ALPHA8_ASTC_4x4_KHR:
  248. format = Image::FORMAT_ASTC_4x4;
  249. break;
  250. case GL_COMPRESSED_RGBA_ASTC_4x4_KHR:
  251. format = Image::FORMAT_ASTC_4x4_HDR;
  252. break;
  253. case GL_COMPRESSED_SRGB8_ALPHA8_ASTC_8x8_KHR:
  254. format = Image::FORMAT_ASTC_8x8;
  255. break;
  256. case GL_COMPRESSED_RGBA_ASTC_8x8_KHR:
  257. format = Image::FORMAT_ASTC_8x8_HDR;
  258. break;
  259. default:
  260. ktxTexture_Destroy(ktx_texture);
  261. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported format " + itos(((ktxTexture1 *)ktx_texture)->glInternalformat) + " of KTX1 texture file.");
  262. }
  263. break;
  264. case ktxTexture2_c: {
  265. ktxTexture2 *ktx_texture2 = reinterpret_cast<ktxTexture2 *>(ktx_texture);
  266. if (ktx_texture2->vkFormat == VK_FORMAT_UNDEFINED) {
  267. if (!ktxTexture2_NeedsTranscoding(ktx_texture2)) {
  268. ktxTexture_Destroy(ktx_texture);
  269. ERR_FAIL_V_MSG(Ref<Resource>(), "Invalid VK_FORMAT_UNDEFINED of KTX2 texture file.");
  270. }
  271. ktx_transcode_fmt_e ktxfmt;
  272. switch (ktxTexture2_GetNumComponents(ktx_texture2)) {
  273. case 1: {
  274. if (ktxTexture2_GetOETF_e(ktx_texture2) == KHR_DF_TRANSFER_SRGB) { // TODO srgb native support
  275. ktxfmt = KTX_TTF_RGBA32;
  276. } else if (RS::get_singleton()->has_os_feature("rgtc")) {
  277. ktxfmt = KTX_TTF_BC4_R;
  278. } else {
  279. ktxfmt = KTX_TTF_RGBA32;
  280. }
  281. break;
  282. }
  283. case 2: {
  284. if (ktxTexture2_GetOETF_e(ktx_texture2) == KHR_DF_TRANSFER_SRGB) { // TODO srgb native support
  285. ktxfmt = KTX_TTF_RGBA32;
  286. } else if (RS::get_singleton()->has_os_feature("rgtc")) {
  287. ktxfmt = KTX_TTF_BC5_RG;
  288. } else {
  289. ktxfmt = KTX_TTF_RGBA32;
  290. }
  291. break;
  292. }
  293. case 3: {
  294. if (ktxTexture2_GetOETF_e(ktx_texture2) == KHR_DF_TRANSFER_SRGB) { // TODO: srgb native support
  295. ktxfmt = KTX_TTF_RGBA32;
  296. } else if (RS::get_singleton()->has_os_feature("bptc")) {
  297. ktxfmt = KTX_TTF_BC7_RGBA;
  298. } else if (RS::get_singleton()->has_os_feature("s3tc")) {
  299. ktxfmt = KTX_TTF_BC1_RGB;
  300. } else if (RS::get_singleton()->has_os_feature("etc2")) {
  301. ktxfmt = KTX_TTF_ETC1_RGB;
  302. } else {
  303. ktxfmt = KTX_TTF_RGBA32;
  304. }
  305. break;
  306. }
  307. case 4: {
  308. if (ktxTexture2_GetOETF_e(ktx_texture2) == KHR_DF_TRANSFER_SRGB) { // TODO srgb native support
  309. ktxfmt = KTX_TTF_RGBA32;
  310. } else if (RS::get_singleton()->has_os_feature("astc")) {
  311. ktxfmt = KTX_TTF_ASTC_4x4_RGBA;
  312. } else if (RS::get_singleton()->has_os_feature("bptc")) {
  313. ktxfmt = KTX_TTF_BC7_RGBA;
  314. } else if (RS::get_singleton()->has_os_feature("s3tc")) {
  315. ktxfmt = KTX_TTF_BC3_RGBA;
  316. } else if (RS::get_singleton()->has_os_feature("etc2")) {
  317. ktxfmt = KTX_TTF_ETC2_RGBA;
  318. } else {
  319. ktxfmt = KTX_TTF_RGBA32;
  320. }
  321. break;
  322. }
  323. default: {
  324. ktxTexture_Destroy(ktx_texture);
  325. ERR_FAIL_V_MSG(Ref<Resource>(), "Invalid components of KTX2 texture file.");
  326. }
  327. }
  328. result = ktxTexture2_TranscodeBasis(ktx_texture2, ktxfmt, 0);
  329. if (result != KTX_SUCCESS) {
  330. ktxTexture_Destroy(ktx_texture);
  331. ERR_FAIL_V_MSG(Ref<Resource>(), "Failed to transcode KTX2 texture file.");
  332. }
  333. }
  334. switch (ktx_texture2->vkFormat) {
  335. case VK_FORMAT_R8_UNORM:
  336. format = Image::FORMAT_L8;
  337. break;
  338. case VK_FORMAT_R8G8_UNORM:
  339. format = Image::FORMAT_LA8;
  340. break;
  341. case VK_FORMAT_R8G8B8_SRGB:
  342. format = Image::FORMAT_RGB8;
  343. srgb = true;
  344. break;
  345. case VK_FORMAT_R8G8B8A8_SRGB:
  346. format = Image::FORMAT_RGBA8;
  347. srgb = true;
  348. break;
  349. case VK_FORMAT_R8_UINT:
  350. format = Image::FORMAT_R8;
  351. break;
  352. case VK_FORMAT_R8G8_UINT:
  353. format = Image::FORMAT_RG8;
  354. break;
  355. case VK_FORMAT_R8G8B8_UINT:
  356. format = Image::FORMAT_RGB8;
  357. break;
  358. case VK_FORMAT_R8G8B8A8_UINT:
  359. format = Image::FORMAT_RGBA8;
  360. break;
  361. case VK_FORMAT_R4G4B4A4_UNORM_PACK16:
  362. format = Image::FORMAT_RGBA4444;
  363. break;
  364. case VK_FORMAT_R5G6B5_UNORM_PACK16:
  365. format = Image::FORMAT_RGB565;
  366. break;
  367. case VK_FORMAT_R32_SFLOAT:
  368. format = Image::FORMAT_RF;
  369. break;
  370. case VK_FORMAT_R32G32_SFLOAT:
  371. format = Image::FORMAT_RGF;
  372. break;
  373. case VK_FORMAT_R32G32B32_SFLOAT:
  374. format = Image::FORMAT_RGBF;
  375. break;
  376. case VK_FORMAT_R32G32B32A32_SFLOAT:
  377. format = Image::FORMAT_RGBAF;
  378. break;
  379. case VK_FORMAT_R16_SFLOAT:
  380. format = Image::FORMAT_RH;
  381. break;
  382. case VK_FORMAT_R16G16_SFLOAT:
  383. format = Image::FORMAT_RGH;
  384. break;
  385. case VK_FORMAT_R16G16B16_SFLOAT:
  386. format = Image::FORMAT_RGBH;
  387. break;
  388. case VK_FORMAT_R16G16B16A16_SFLOAT:
  389. format = Image::FORMAT_RGBAH;
  390. break;
  391. case VK_FORMAT_E5B9G9R9_UFLOAT_PACK32:
  392. format = Image::FORMAT_RGBE9995;
  393. break;
  394. case VK_FORMAT_R16_UNORM:
  395. format = Image::FORMAT_R16;
  396. break;
  397. case VK_FORMAT_R16G16_UNORM:
  398. format = Image::FORMAT_RG16;
  399. break;
  400. case VK_FORMAT_R16G16B16_UNORM:
  401. format = Image::FORMAT_RGB16;
  402. break;
  403. case VK_FORMAT_R16G16B16A16_UNORM:
  404. format = Image::FORMAT_RGBA16;
  405. break;
  406. case VK_FORMAT_R16_UINT:
  407. format = Image::FORMAT_R16I;
  408. break;
  409. case VK_FORMAT_R16G16_UINT:
  410. format = Image::FORMAT_RG16I;
  411. break;
  412. case VK_FORMAT_R16G16B16_UINT:
  413. format = Image::FORMAT_RGB16I;
  414. break;
  415. case VK_FORMAT_R16G16B16A16_UINT:
  416. format = Image::FORMAT_RGBA16I;
  417. break;
  418. case VK_FORMAT_BC1_RGB_UNORM_BLOCK:
  419. case VK_FORMAT_BC1_RGBA_UNORM_BLOCK:
  420. format = Image::FORMAT_DXT1;
  421. break;
  422. case VK_FORMAT_BC2_UNORM_BLOCK:
  423. format = Image::FORMAT_DXT3;
  424. break;
  425. case VK_FORMAT_BC3_UNORM_BLOCK:
  426. format = Image::FORMAT_DXT5;
  427. break;
  428. case VK_FORMAT_BC4_UNORM_BLOCK:
  429. format = Image::FORMAT_RGTC_R;
  430. break;
  431. case VK_FORMAT_BC5_UNORM_BLOCK:
  432. format = Image::FORMAT_RGTC_RG;
  433. break;
  434. case VK_FORMAT_BC6H_UFLOAT_BLOCK:
  435. format = Image::FORMAT_BPTC_RGBFU;
  436. break;
  437. case VK_FORMAT_BC6H_SFLOAT_BLOCK:
  438. format = Image::FORMAT_BPTC_RGBF;
  439. break;
  440. case VK_FORMAT_BC7_UNORM_BLOCK:
  441. format = Image::FORMAT_BPTC_RGBA;
  442. break;
  443. case VK_FORMAT_EAC_R11_UNORM_BLOCK:
  444. format = Image::FORMAT_ETC2_R11;
  445. break;
  446. // Decompression is not supported for this format.
  447. /*case VK_FORMAT_EAC_R11_SNORM_BLOCK:
  448. format = Image::FORMAT_ETC2_R11S;
  449. break;*/
  450. case VK_FORMAT_EAC_R11G11_UNORM_BLOCK:
  451. format = Image::FORMAT_ETC2_RG11;
  452. break;
  453. // Decompression is not supported for this format.
  454. /*case VK_FORMAT_EAC_R11G11_SNORM_BLOCK:
  455. format = Image::FORMAT_ETC2_RG11S;
  456. break;*/
  457. case VK_FORMAT_ETC2_R8G8B8_UNORM_BLOCK:
  458. format = Image::FORMAT_ETC2_RGB8;
  459. break;
  460. case VK_FORMAT_ETC2_R8G8B8A8_UNORM_BLOCK:
  461. format = Image::FORMAT_ETC2_RGBA8;
  462. break;
  463. // Decompression is not supported for this format.
  464. /*case VK_FORMAT_ETC2_R8G8B8A1_UNORM_BLOCK:
  465. format = Image::FORMAT_ETC2_RGB8A1;
  466. break;*/
  467. case VK_FORMAT_ASTC_4x4_SRGB_BLOCK:
  468. format = Image::FORMAT_ASTC_4x4;
  469. break;
  470. case VK_FORMAT_ASTC_4x4_UNORM_BLOCK:
  471. format = Image::FORMAT_ASTC_4x4_HDR;
  472. break;
  473. case VK_FORMAT_ASTC_8x8_SRGB_BLOCK:
  474. format = Image::FORMAT_ASTC_8x8;
  475. break;
  476. case VK_FORMAT_ASTC_8x8_UNORM_BLOCK:
  477. format = Image::FORMAT_ASTC_8x8_HDR;
  478. break;
  479. default:
  480. ktxTexture_Destroy(ktx_texture);
  481. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported format " + itos(((ktxTexture2 *)ktx_texture)->vkFormat) + " of KTX2 texture file.");
  482. break;
  483. }
  484. break;
  485. }
  486. default:
  487. ktxTexture_Destroy(ktx_texture);
  488. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported version KTX texture file.");
  489. break;
  490. }
  491. Vector<uint8_t> src_data;
  492. // KTX use 4-bytes padding, don't use mipmaps if padding is effective
  493. // TODO: unpad dynamically
  494. int pixel_size = Image::get_format_pixel_size(format);
  495. int pixel_rshift = Image::get_format_pixel_rshift(format);
  496. int block = Image::get_format_block_size(format);
  497. int minw, minh;
  498. Image::get_format_min_pixel_size(format, minw, minh);
  499. int w = width;
  500. int h = height;
  501. for (uint32_t i = 0; i < mipmaps; ++i) {
  502. ktx_size_t mip_size = ktxTexture_GetImageSize(ktx_texture, i);
  503. size_t bw = w % block != 0 ? w + (block - w % block) : w;
  504. size_t bh = h % block != 0 ? h + (block - h % block) : h;
  505. size_t s = bw * bh;
  506. s *= pixel_size;
  507. s >>= pixel_rshift;
  508. if (mip_size != static_cast<ktx_size_t>(s)) {
  509. if (!i) {
  510. ktxTexture_Destroy(ktx_texture);
  511. ERR_FAIL_V_MSG(Ref<Resource>(), "Unsupported padded KTX texture file.");
  512. }
  513. mipmaps = 1;
  514. break;
  515. }
  516. w = MAX(minw, w >> 1);
  517. h = MAX(minh, h >> 1);
  518. }
  519. for (uint32_t i = 0; i < mipmaps; ++i) {
  520. ktx_size_t mip_size = ktxTexture_GetImageSize(ktx_texture, i);
  521. ktx_size_t offset;
  522. if (ktxTexture_GetImageOffset(ktx_texture, i, 0, 0, &offset) != KTX_SUCCESS) {
  523. ktxTexture_Destroy(ktx_texture);
  524. ERR_FAIL_V_MSG(Ref<Resource>(), "Invalid KTX texture file.");
  525. }
  526. int prev_size = src_data.size();
  527. src_data.resize(prev_size + mip_size);
  528. memcpy(src_data.ptrw() + prev_size, ktxTexture_GetData(ktx_texture) + offset, mip_size);
  529. }
  530. Ref<Image> img = memnew(Image(width, height, mipmaps - 1, format, src_data));
  531. if (srgb) {
  532. img->srgb_to_linear();
  533. }
  534. if (r_error) {
  535. *r_error = OK;
  536. }
  537. ktxTexture_Destroy(ktx_texture);
  538. return img;
  539. }
  540. Ref<Resource> ResourceFormatKTX::load(const String &p_path, const String &p_original_path, Error *r_error, bool p_use_sub_threads, float *r_progress, CacheMode p_cache_mode) {
  541. if (r_error) {
  542. *r_error = ERR_CANT_OPEN;
  543. }
  544. Error err;
  545. Ref<FileAccess> f = FileAccess::open(p_path, FileAccess::READ, &err);
  546. if (f.is_null()) {
  547. return Ref<Resource>();
  548. }
  549. Ref<FileAccess> fref(f);
  550. if (r_error) {
  551. *r_error = ERR_FILE_CORRUPT;
  552. }
  553. ERR_FAIL_COND_V_MSG(err != OK, Ref<Resource>(), "Unable to open KTX texture file '" + p_path + "'.");
  554. Ref<Image> img = load_from_file_access(f, r_error);
  555. Ref<ImageTexture> texture = ImageTexture::create_from_image(img);
  556. return texture;
  557. }
  558. static Ref<Image> _ktx_mem_loader_func(const uint8_t *p_ktx, int p_size) {
  559. Ref<FileAccessMemory> f;
  560. f.instantiate();
  561. f->open_custom(p_ktx, p_size);
  562. Error err;
  563. Ref<Image> img = load_from_file_access(f, &err);
  564. ERR_FAIL_COND_V(err, Ref<Image>());
  565. return img;
  566. }
  567. void ResourceFormatKTX::get_recognized_extensions(List<String> *p_extensions) const {
  568. p_extensions->push_back("ktx");
  569. p_extensions->push_back("ktx2");
  570. }
  571. bool ResourceFormatKTX::handles_type(const String &p_type) const {
  572. return ClassDB::is_parent_class(p_type, "Texture2D");
  573. }
  574. String ResourceFormatKTX::get_resource_type(const String &p_path) const {
  575. if (p_path.has_extension("ktx") || p_path.has_extension("ktx2")) {
  576. return "ImageTexture";
  577. }
  578. return "";
  579. }
  580. ResourceFormatKTX::ResourceFormatKTX() {
  581. Image::_ktx_mem_loader_func = _ktx_mem_loader_func;
  582. }