2014-02-10 01:10:30 +00:00
/**************************************************************************/
/* resource_loader.cpp */
/**************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* https://godotengine.org */
/**************************************************************************/
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
/* */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the */
/* "Software"), to deal in the Software without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of the Software, and to */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions: */
/* */
/* The above copyright notice and this permission notice shall be */
/* included in all copies or substantial portions of the Software. */
/* */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/**************************************************************************/
2018-01-04 23:50:27 +00:00
2014-02-10 01:10:30 +00:00
# include "resource_loader.h"
2018-09-11 16:13:45 +00:00
2020-11-07 22:33:38 +00:00
# include "core/config/project_settings.h"
2024-09-05 07:48:13 +00:00
# include "core/core_bind.h"
2021-06-11 12:51:48 +00:00
# include "core/io/file_access.h"
2019-02-12 12:30:56 +00:00
# include "core/io/resource_importer.h"
2023-09-06 19:02:52 +00:00
# include "core/object/script_language.h"
2023-02-20 18:00:26 +00:00
# include "core/os/condition_variable.h"
2018-09-11 16:13:45 +00:00
# include "core/os/os.h"
2024-02-04 15:59:54 +00:00
# include "core/os/safe_binary_mutex.h"
2020-11-07 22:33:38 +00:00
# include "core/string/print_string.h"
2024-08-15 07:00:47 +00:00
# include "core/string/translation_server.h"
2020-11-07 22:33:38 +00:00
# include "core/variant/variant_parser.h"
2024-06-28 09:16:59 +00:00
# include "servers/rendering_server.h"
2018-09-11 16:13:45 +00:00
2020-02-28 11:27:04 +00:00
# ifdef DEBUG_LOAD_THREADED
# define print_lt(m_text) print_line(m_text)
# else
# define print_lt(m_text)
# endif
2018-06-11 00:59:53 +00:00
Ref < ResourceFormatLoader > ResourceLoader : : loader [ ResourceLoader : : MAX_LOADERS ] ;
2014-02-10 01:10:30 +00:00
int ResourceLoader : : loader_count = 0 ;
2017-01-26 00:55:59 +00:00
bool ResourceFormatLoader : : recognize_path ( const String & p_path , const String & p_for_type ) const {
2022-10-11 09:40:04 +00:00
bool ret = false ;
if ( GDVIRTUAL_CALL ( _recognize_path , p_path , p_for_type , ret ) ) {
return ret ;
}
2017-01-26 00:55:59 +00:00
String extension = p_path . get_extension ( ) ;
2016-03-08 23:00:52 +00:00
2014-02-10 01:10:30 +00:00
List < String > extensions ;
2021-12-09 09:42:46 +00:00
if ( p_for_type . is_empty ( ) ) {
2017-01-26 00:55:59 +00:00
get_recognized_extensions ( & extensions ) ;
} else {
get_recognized_extensions_for_type ( p_for_type , & extensions ) ;
}
2021-07-24 13:46:25 +00:00
for ( const String & E : extensions ) {
2021-07-16 03:45:57 +00:00
if ( E . nocasecmp_to ( extension ) = = 0 ) {
2014-02-10 01:10:30 +00:00
return true ;
2020-05-14 14:41:43 +00:00
}
2014-02-10 01:10:30 +00:00
}
2016-03-08 23:00:52 +00:00
2014-02-10 01:10:30 +00:00
return false ;
}
2018-06-11 00:59:53 +00:00
bool ResourceFormatLoader : : handles_type ( const String & p_type ) const {
2022-09-29 09:53:28 +00:00
bool success = false ;
2022-10-18 16:47:44 +00:00
GDVIRTUAL_CALL ( _handles_type , p_type , success ) ;
return success ;
2018-06-11 00:59:53 +00:00
}
2022-07-14 12:18:18 +00:00
void ResourceFormatLoader : : get_classes_used ( const String & p_path , HashSet < StringName > * r_classes ) {
Vector < String > ret ;
if ( GDVIRTUAL_CALL ( _get_classes_used , p_path , ret ) ) {
for ( int i = 0 ; i < ret . size ( ) ; i + + ) {
r_classes - > insert ( ret [ i ] ) ;
}
return ;
}
String res = get_resource_type ( p_path ) ;
if ( ! res . is_empty ( ) ) {
r_classes - > insert ( res ) ;
}
}
2018-06-11 00:59:53 +00:00
String ResourceFormatLoader : : get_resource_type ( const String & p_path ) const {
2021-08-22 01:52:44 +00:00
String ret ;
2022-10-18 16:47:44 +00:00
GDVIRTUAL_CALL ( _get_resource_type , p_path , ret ) ;
return ret ;
2018-06-11 00:59:53 +00:00
}
2023-01-19 18:12:25 +00:00
String ResourceFormatLoader : : get_resource_script_class ( const String & p_path ) const {
String ret ;
GDVIRTUAL_CALL ( _get_resource_script_class , p_path , ret ) ;
return ret ;
}
2021-07-23 19:01:18 +00:00
ResourceUID : : ID ResourceFormatLoader : : get_resource_uid ( const String & p_path ) const {
2022-09-29 09:53:28 +00:00
int64_t uid = ResourceUID : : INVALID_ID ;
2022-10-18 16:47:44 +00:00
GDVIRTUAL_CALL ( _get_resource_uid , p_path , uid ) ;
return uid ;
2021-07-23 19:01:18 +00:00
}
2014-02-10 01:10:30 +00:00
void ResourceFormatLoader : : get_recognized_extensions_for_type ( const String & p_type , List < String > * p_extensions ) const {
2021-12-09 09:42:46 +00:00
if ( p_type . is_empty ( ) | | handles_type ( p_type ) ) {
2014-02-10 01:10:30 +00:00
get_recognized_extensions ( p_extensions ) ;
2020-05-14 14:41:43 +00:00
}
2014-02-10 01:10:30 +00:00
}
void ResourceLoader : : get_recognized_extensions_for_type ( const String & p_type , List < String > * p_extensions ) {
for ( int i = 0 ; i < loader_count ; i + + ) {
loader [ i ] - > get_recognized_extensions_for_type ( p_type , p_extensions ) ;
}
}
2018-08-10 18:57:43 +00:00
bool ResourceFormatLoader : : exists ( const String & p_path ) const {
2022-09-29 09:53:28 +00:00
bool success = false ;
2021-08-22 01:52:44 +00:00
if ( GDVIRTUAL_CALL ( _exists , p_path , success ) ) {
return success ;
}
2022-10-18 16:47:44 +00:00
return FileAccess : : exists ( p_path ) ; // By default just check file.
2018-08-10 18:57:43 +00:00
}
2018-08-12 10:44:38 +00:00
2018-06-11 00:59:53 +00:00
void ResourceFormatLoader : : get_recognized_extensions ( List < String > * p_extensions ) const {
2021-08-22 01:52:44 +00:00
PackedStringArray exts ;
if ( GDVIRTUAL_CALL ( _get_recognized_extensions , exts ) ) {
const String * r = exts . ptr ( ) ;
for ( int i = 0 ; i < exts . size ( ) ; + + i ) {
p_extensions - > push_back ( r [ i ] ) ;
2018-06-11 00:59:53 +00:00
}
}
}
2022-05-02 23:43:50 +00:00
Ref < Resource > ResourceFormatLoader : : 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 ) {
2021-08-22 01:52:44 +00:00
Variant res ;
if ( GDVIRTUAL_CALL ( _load , p_path , p_original_path , p_use_sub_threads , p_cache_mode , res ) ) {
2021-05-05 08:56:26 +00:00
if ( res . get_type ( ) = = Variant : : INT ) { // Error code, abort.
2020-05-14 14:41:43 +00:00
if ( r_error ) {
2018-06-11 00:59:53 +00:00
* r_error = ( Error ) res . operator int64_t ( ) ;
2020-05-14 14:41:43 +00:00
}
2022-05-02 23:43:50 +00:00
return Ref < Resource > ( ) ;
2021-05-05 08:56:26 +00:00
} else { // Success, pass on result.
2020-05-14 14:41:43 +00:00
if ( r_error ) {
2018-06-11 00:59:53 +00:00
* r_error = OK ;
2020-05-14 14:41:43 +00:00
}
2018-06-11 00:59:53 +00:00
return res ;
}
2014-02-10 01:10:30 +00:00
}
2020-02-28 11:27:04 +00:00
2022-05-02 23:43:50 +00:00
ERR_FAIL_V_MSG ( Ref < Resource > ( ) , " Failed to load resource ' " + p_path + " '. ResourceFormatLoader::load was not implemented for this resource type. " ) ;
2014-02-10 01:10:30 +00:00
}
2015-08-23 23:15:56 +00:00
void ResourceFormatLoader : : get_dependencies ( const String & p_path , List < String > * p_dependencies , bool p_add_types ) {
2021-08-22 01:52:44 +00:00
PackedStringArray deps ;
if ( GDVIRTUAL_CALL ( _get_dependencies , p_path , p_add_types , deps ) ) {
const String * r = deps . ptr ( ) ;
for ( int i = 0 ; i < deps . size ( ) ; + + i ) {
p_dependencies - > push_back ( r [ i ] ) ;
2018-06-11 00:59:53 +00:00
}
}
}
2022-05-13 13:04:37 +00:00
Error ResourceFormatLoader : : rename_dependencies ( const String & p_path , const HashMap < String , String > & p_map ) {
2021-08-22 01:52:44 +00:00
Dictionary deps_dict ;
2021-08-09 20:13:42 +00:00
for ( KeyValue < String , String > E : p_map ) {
deps_dict [ E . key ] = E . value ;
2021-08-22 01:52:44 +00:00
}
2018-06-11 00:59:53 +00:00
2023-01-15 19:33:20 +00:00
Error err = OK ;
2022-10-18 16:47:44 +00:00
GDVIRTUAL_CALL ( _rename_dependencies , p_path , deps_dict , err ) ;
2023-01-15 19:33:20 +00:00
return err ;
2018-06-11 00:59:53 +00:00
}
void ResourceFormatLoader : : _bind_methods ( ) {
2021-02-19 12:35:31 +00:00
BIND_ENUM_CONSTANT ( CACHE_MODE_IGNORE ) ;
BIND_ENUM_CONSTANT ( CACHE_MODE_REUSE ) ;
BIND_ENUM_CONSTANT ( CACHE_MODE_REPLACE ) ;
2024-02-22 11:53:19 +00:00
BIND_ENUM_CONSTANT ( CACHE_MODE_IGNORE_DEEP ) ;
BIND_ENUM_CONSTANT ( CACHE_MODE_REPLACE_DEEP ) ;
2021-08-22 01:52:44 +00:00
GDVIRTUAL_BIND ( _get_recognized_extensions ) ;
2022-10-11 09:40:04 +00:00
GDVIRTUAL_BIND ( _recognize_path , " path " , " type " ) ;
2021-08-22 01:52:44 +00:00
GDVIRTUAL_BIND ( _handles_type , " type " ) ;
GDVIRTUAL_BIND ( _get_resource_type , " path " ) ;
2023-01-19 18:12:25 +00:00
GDVIRTUAL_BIND ( _get_resource_script_class , " path " ) ;
2021-08-22 01:52:44 +00:00
GDVIRTUAL_BIND ( _get_resource_uid , " path " ) ;
GDVIRTUAL_BIND ( _get_dependencies , " path " , " add_types " ) ;
GDVIRTUAL_BIND ( _rename_dependencies , " path " , " renames " ) ;
GDVIRTUAL_BIND ( _exists , " path " ) ;
2022-07-14 12:18:18 +00:00
GDVIRTUAL_BIND ( _get_classes_used , " path " ) ;
2021-08-22 01:52:44 +00:00
GDVIRTUAL_BIND ( _load , " path " , " original_path " , " use_sub_threads " , " cache_mode " ) ;
2014-02-10 01:10:30 +00:00
}
///////////////////////////////////
2024-07-10 10:01:22 +00:00
// These are used before and after a wait for a WorkerThreadPool task
// because that can lead to another load started in the same thread,
// something we must treat as a different stack for the purposes
// of tracking nesting.
# define PREPARE_FOR_WTP_WAIT \
int load_nesting_backup = ResourceLoader : : load_nesting ; \
Vector < String > load_paths_stack_backup = ResourceLoader : : load_paths_stack ; \
ResourceLoader : : load_nesting = 0 ; \
ResourceLoader : : load_paths_stack . clear ( ) ;
# define RESTORE_AFTER_WTP_WAIT \
DEV_ASSERT ( ResourceLoader : : load_nesting = = 0 ) ; \
DEV_ASSERT ( ResourceLoader : : load_paths_stack . is_empty ( ) ) ; \
ResourceLoader : : load_nesting = load_nesting_backup ; \
ResourceLoader : : load_paths_stack = load_paths_stack_backup ; \
load_paths_stack_backup . clear ( ) ;
2023-03-05 00:09:18 +00:00
// This should be robust enough to be called redundantly without issues.
void ResourceLoader : : LoadToken : : clear ( ) {
2023-05-10 09:23:07 +00:00
WorkerThreadPool : : TaskID task_to_await = 0 ;
2023-03-05 00:09:18 +00:00
2024-08-28 09:27:01 +00:00
{
MutexLock thread_load_lock ( thread_load_mutex ) ;
// User-facing tokens shouldn't be deleted until completely claimed.
DEV_ASSERT ( user_rc = = 0 & & user_path . is_empty ( ) ) ;
2024-09-05 06:35:05 +00:00
if ( ! local_path . is_empty ( ) ) {
if ( task_if_unregistered ) {
memdelete ( task_if_unregistered ) ;
task_if_unregistered = nullptr ;
} else {
DEV_ASSERT ( thread_load_tasks . has ( local_path ) ) ;
ThreadLoadTask & load_task = thread_load_tasks [ local_path ] ;
if ( load_task . task_id & & ! load_task . awaited ) {
task_to_await = load_task . task_id ;
}
// Removing a task which is still in progress would be catastrophic.
// Tokens must be alive until the task thread function is done.
DEV_ASSERT ( load_task . status = = THREAD_LOAD_FAILED | | load_task . status = = THREAD_LOAD_LOADED ) ;
thread_load_tasks . erase ( local_path ) ;
2024-08-28 09:27:01 +00:00
}
2024-09-05 06:35:05 +00:00
local_path . clear ( ) ; // Mark as already cleared.
2023-05-10 09:23:07 +00:00
}
2023-03-05 00:09:18 +00:00
}
2023-05-10 09:23:07 +00:00
// If task is unused, await it here, locally, now the token data is consistent.
if ( task_to_await ) {
2024-07-10 10:01:22 +00:00
PREPARE_FOR_WTP_WAIT
2023-05-10 09:23:07 +00:00
WorkerThreadPool : : get_singleton ( ) - > wait_for_task_completion ( task_to_await ) ;
2024-07-10 10:01:22 +00:00
RESTORE_AFTER_WTP_WAIT
2023-03-05 00:09:18 +00:00
}
}
ResourceLoader : : LoadToken : : ~ LoadToken ( ) {
clear ( ) ;
}
2022-05-02 23:43:50 +00:00
Ref < Resource > ResourceLoader : : _load ( const String & p_path , const String & p_original_path , const String & p_type_hint , ResourceFormatLoader : : CacheMode p_cache_mode , Error * r_error , bool p_use_sub_threads , float * r_progress ) {
2024-04-01 05:56:47 +00:00
const String & original_path = p_original_path . is_empty ( ) ? p_path : p_original_path ;
2023-03-05 00:09:18 +00:00
load_nesting + + ;
2024-06-19 05:27:33 +00:00
if ( load_paths_stack . size ( ) ) {
2024-08-28 09:27:01 +00:00
MutexLock thread_load_lock ( thread_load_mutex ) ;
2024-06-19 05:27:33 +00:00
const String & parent_task_path = load_paths_stack . get ( load_paths_stack . size ( ) - 1 ) ;
2024-04-01 05:56:47 +00:00
HashMap < String , ThreadLoadTask > : : Iterator E = thread_load_tasks . find ( parent_task_path ) ;
// Avoid double-tracking, for progress reporting, resources that boil down to a remapped path containing the real payload (e.g., imported resources).
bool is_remapped_load = original_path = = parent_task_path ;
if ( E & & ! is_remapped_load ) {
2024-01-29 17:27:57 +00:00
E - > value . sub_tasks . insert ( p_original_path ) ;
2023-03-05 00:09:18 +00:00
}
}
2024-06-19 05:27:33 +00:00
load_paths_stack . push_back ( original_path ) ;
2017-08-31 21:57:03 +00:00
// Try all loaders and pick the first match for the type hint
2023-03-05 00:09:18 +00:00
bool found = false ;
Ref < Resource > res ;
2017-08-31 21:57:03 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
if ( ! loader [ i ] - > recognize_path ( p_path , p_type_hint ) ) {
continue ;
}
found = true ;
2024-04-01 05:56:47 +00:00
res = loader [ i ] - > load ( p_path , original_path , r_error , p_use_sub_threads , r_progress , p_cache_mode ) ;
2023-03-05 00:09:18 +00:00
if ( ! res . is_null ( ) ) {
break ;
2017-08-31 21:57:03 +00:00
}
2023-03-05 00:09:18 +00:00
}
2017-08-31 21:57:03 +00:00
2024-06-19 05:27:33 +00:00
load_paths_stack . resize ( load_paths_stack . size ( ) - 1 ) ;
2024-06-25 12:12:40 +00:00
res_ref_overrides . erase ( load_nesting ) ;
2023-03-05 00:09:18 +00:00
load_nesting - - ;
if ( ! res . is_null ( ) ) {
2017-08-31 21:57:03 +00:00
return res ;
}
2022-05-02 23:43:50 +00:00
ERR_FAIL_COND_V_MSG ( found , Ref < Resource > ( ) ,
2020-08-22 20:19:08 +00:00
vformat ( " Failed loading resource: %s. Make sure resources have been imported by opening the project in the editor at least once. " , p_path ) ) ;
2019-08-15 02:57:49 +00:00
2020-01-08 14:11:16 +00:00
# ifdef TOOLS_ENABLED
2022-03-23 09:08:58 +00:00
Ref < FileAccess > file_check = FileAccess : : create ( FileAccess : : ACCESS_RESOURCES ) ;
2023-07-21 02:45:34 +00:00
ERR_FAIL_COND_V_MSG ( ! file_check - > file_exists ( p_path ) , Ref < Resource > ( ) , vformat ( " Resource file not found: %s (expected type: %s) " , p_path , p_type_hint ) ) ;
2020-01-08 14:11:16 +00:00
# endif
2023-07-21 02:45:34 +00:00
ERR_FAIL_V_MSG ( Ref < Resource > ( ) , vformat ( " No loader found for resource: %s (expected type: %s) " , p_path , p_type_hint ) ) ;
2017-08-31 21:57:03 +00:00
}
2024-07-10 10:53:14 +00:00
// This implementation must allow re-entrancy for a task that started awaiting in a deeper stack frame.
2024-09-05 17:06:38 +00:00
// The load task token must be manually re-referenced before this is called, which includes threaded runs.
2024-07-10 10:53:14 +00:00
void ResourceLoader : : _run_load_task ( void * p_userdata ) {
2020-02-28 11:27:04 +00:00
ThreadLoadTask & load_task = * ( ThreadLoadTask * ) p_userdata ;
2023-05-10 09:23:07 +00:00
2024-08-28 09:27:01 +00:00
{
MutexLock thread_load_lock ( thread_load_mutex ) ;
if ( cleaning_tasks ) {
load_task . status = THREAD_LOAD_FAILED ;
return ;
}
2023-05-10 09:23:07 +00:00
}
2024-09-05 07:48:13 +00:00
ThreadLoadTask * curr_load_task_backup = curr_load_task ;
curr_load_task = & load_task ;
2023-03-05 00:09:18 +00:00
// Thread-safe either if it's the current thread or a brand new one.
2024-06-13 08:26:33 +00:00
CallQueue * own_mq_override = nullptr ;
2023-05-10 09:23:07 +00:00
if ( load_nesting = = 0 ) {
2024-06-19 05:27:33 +00:00
DEV_ASSERT ( load_paths_stack . is_empty ( ) ) ;
2023-05-10 08:00:33 +00:00
if ( ! Thread : : is_main_thread ( ) ) {
2024-06-13 08:26:33 +00:00
// Let the caller thread use its own, for added flexibility. Provide one otherwise.
if ( MessageQueue : : get_singleton ( ) = = MessageQueue : : get_main_singleton ( ) ) {
own_mq_override = memnew ( CallQueue ) ;
MessageQueue : : set_thread_singleton_override ( own_mq_override ) ;
}
2023-05-23 21:28:47 +00:00
set_current_thread_safe_for_nodes ( true ) ;
2023-05-10 08:00:33 +00:00
}
2020-02-28 11:27:04 +00:00
}
2023-03-05 00:09:18 +00:00
// --
2020-02-28 11:27:04 +00:00
2024-07-10 11:51:02 +00:00
bool xl_remapped = false ;
const String & remapped_path = _path_remap ( load_task . local_path , & xl_remapped ) ;
2024-07-29 14:31:57 +00:00
print_verbose ( " Loading resource: " + remapped_path ) ;
2024-07-15 09:43:27 +00:00
Error load_err = OK ;
2024-07-10 11:51:02 +00:00
Ref < Resource > res = _load ( remapped_path , remapped_path ! = load_task . local_path ? load_task . local_path : String ( ) , load_task . type_hint , load_task . cache_mode , & load_err , load_task . use_sub_threads , & load_task . progress ) ;
2024-07-09 16:41:24 +00:00
if ( MessageQueue : : get_singleton ( ) ! = MessageQueue : : get_main_singleton ( ) ) {
2024-06-13 08:26:33 +00:00
MessageQueue : : get_singleton ( ) - > flush ( ) ;
2023-05-19 08:46:49 +00:00
}
2023-03-05 00:09:18 +00:00
2024-07-29 14:31:57 +00:00
if ( res . is_null ( ) ) {
print_verbose ( " Failed loading resource: " + remapped_path ) ;
}
2023-03-05 00:09:18 +00:00
thread_load_mutex . lock ( ) ;
load_task . resource = res ;
2019-02-16 20:38:09 +00:00
2024-07-15 09:43:27 +00:00
load_task . progress = 1.0 ; // It was fully loaded at this point, so force progress to 1.0.
load_task . error = load_err ;
2020-02-28 11:27:04 +00:00
if ( load_task . error ! = OK ) {
load_task . status = THREAD_LOAD_FAILED ;
2019-01-27 22:24:55 +00:00
} else {
2020-02-28 11:27:04 +00:00
load_task . status = THREAD_LOAD_LOADED ;
2019-01-27 22:24:55 +00:00
}
2024-07-29 17:20:55 +00:00
if ( load_task . cond_var & & load_task . need_wait ) {
2023-02-20 18:00:26 +00:00
load_task . cond_var - > notify_all ( ) ;
2020-02-28 11:27:04 +00:00
}
2024-07-29 17:20:55 +00:00
load_task . need_wait = false ;
2019-02-16 20:38:09 +00:00
2024-02-22 11:53:19 +00:00
bool ignoring = load_task . cache_mode = = ResourceFormatLoader : : CACHE_MODE_IGNORE | | load_task . cache_mode = = ResourceFormatLoader : : CACHE_MODE_IGNORE_DEEP ;
bool replacing = load_task . cache_mode = = ResourceFormatLoader : : CACHE_MODE_REPLACE | | load_task . cache_mode = = ResourceFormatLoader : : CACHE_MODE_REPLACE_DEEP ;
2024-07-29 11:40:29 +00:00
bool unlock_pending = true ;
2020-02-28 11:27:04 +00:00
if ( load_task . resource . is_valid ( ) ) {
2024-07-29 11:40:29 +00:00
// From now on, no critical section needed as no one will write to the task anymore.
// Moreover, the mutex being unlocked is a requirement if some of the calls below
// that set the resource up invoke code that in turn requests resource loading.
thread_load_mutex . unlock ( ) ;
unlock_pending = false ;
2024-02-22 11:53:19 +00:00
if ( ! ignoring ) {
2024-07-10 10:53:14 +00:00
ResourceCache : : lock . lock ( ) ; // Check and operations must happen atomically.
bool pending_unlock = true ;
Ref < Resource > old_res = ResourceCache : : get_ref ( load_task . local_path ) ;
if ( old_res . is_valid ( ) ) {
if ( old_res ! = load_task . resource ) {
// Resource can already exists at this point for two reasons:
// a) The load uses replace mode.
// b) There were more than one load in flight for the same path because of deadlock prevention.
// Either case, we want to keep the resource that was already there.
ResourceCache : : lock . unlock ( ) ;
pending_unlock = false ;
if ( replacing ) {
old_res - > copy_from ( load_task . resource ) ;
}
2024-01-09 14:48:48 +00:00
load_task . resource = old_res ;
}
2024-07-10 10:53:14 +00:00
} else {
load_task . resource - > set_path ( load_task . local_path ) ;
}
if ( pending_unlock ) {
ResourceCache : : lock . unlock ( ) ;
2024-01-09 14:48:48 +00:00
}
2024-02-22 11:53:19 +00:00
} else {
2023-08-05 00:07:16 +00:00
load_task . resource - > set_path_cache ( load_task . local_path ) ;
2023-03-05 00:09:18 +00:00
}
2019-02-16 20:38:09 +00:00
2024-07-10 11:51:02 +00:00
if ( xl_remapped ) {
2020-02-28 11:27:04 +00:00
load_task . resource - > set_as_translation_remapped ( true ) ;
2020-05-14 14:41:43 +00:00
}
2019-02-16 20:38:09 +00:00
2020-02-28 11:27:04 +00:00
# ifdef TOOLS_ENABLED
load_task . resource - > set_edited ( false ) ;
if ( timestamp_on_load ) {
2024-07-10 11:51:02 +00:00
uint64_t mt = FileAccess : : get_modified_time ( remapped_path ) ;
2020-02-28 11:27:04 +00:00
//printf("mt %s: %lli\n",remapped_path.utf8().get_data(),mt);
load_task . resource - > set_last_modified_time ( mt ) ;
}
# endif
2019-01-27 22:24:55 +00:00
2020-02-28 11:27:04 +00:00
if ( _loaded_callback ) {
_loaded_callback ( load_task . resource , load_task . local_path ) ;
}
2024-02-22 11:53:19 +00:00
} else if ( ! ignoring ) {
2024-01-09 14:48:48 +00:00
Ref < Resource > existing = ResourceCache : : get_ref ( load_task . local_path ) ;
if ( existing . is_valid ( ) ) {
load_task . resource = existing ;
load_task . status = THREAD_LOAD_LOADED ;
load_task . progress = 1.0 ;
2024-07-29 11:40:29 +00:00
thread_load_mutex . unlock ( ) ;
unlock_pending = false ;
2024-01-09 14:48:48 +00:00
if ( _loaded_callback ) {
_loaded_callback ( load_task . resource , load_task . local_path ) ;
}
}
2020-02-28 11:27:04 +00:00
}
2015-08-23 23:15:56 +00:00
2024-09-05 17:06:38 +00:00
// It's safe now to let the task go in case no one else was grabbing the token.
load_task . load_token - > unreference ( ) ;
2024-07-29 11:40:29 +00:00
if ( unlock_pending ) {
thread_load_mutex . unlock ( ) ;
}
2023-05-10 08:00:33 +00:00
2023-07-03 11:54:55 +00:00
if ( load_nesting = = 0 ) {
2024-06-13 08:26:33 +00:00
if ( own_mq_override ) {
2024-06-13 08:14:14 +00:00
MessageQueue : : set_thread_singleton_override ( nullptr ) ;
2024-06-13 08:26:33 +00:00
memdelete ( own_mq_override ) ;
2023-07-03 11:54:55 +00:00
}
2024-06-19 05:27:33 +00:00
DEV_ASSERT ( load_paths_stack . is_empty ( ) ) ;
2023-05-10 08:00:33 +00:00
}
2024-09-05 07:48:13 +00:00
curr_load_task = curr_load_task_backup ;
2020-02-28 11:27:04 +00:00
}
2020-05-14 12:29:06 +00:00
2021-07-23 19:01:18 +00:00
static String _validate_local_path ( const String & p_path ) {
ResourceUID : : ID uid = ResourceUID : : get_singleton ( ) - > text_to_id ( p_path ) ;
if ( uid ! = ResourceUID : : INVALID_ID ) {
return ResourceUID : : get_singleton ( ) - > get_id_path ( uid ) ;
2021-08-29 23:43:47 +00:00
} else if ( p_path . is_relative_path ( ) ) {
2024-03-30 11:49:52 +00:00
return ( " res:// " + p_path ) . simplify_path ( ) ;
2020-05-14 14:41:43 +00:00
} else {
2021-07-23 19:01:18 +00:00
return ProjectSettings : : get_singleton ( ) - > localize_path ( p_path ) ;
2020-05-14 14:41:43 +00:00
}
2021-07-23 19:01:18 +00:00
}
2014-02-10 01:10:30 +00:00
2023-03-05 00:09:18 +00:00
Error ResourceLoader : : load_threaded_request ( const String & p_path , const String & p_type_hint , bool p_use_sub_threads , ResourceFormatLoader : : CacheMode p_cache_mode ) {
2024-07-15 08:30:02 +00:00
Ref < ResourceLoader : : LoadToken > token = _load_start ( p_path , p_type_hint , p_use_sub_threads ? LOAD_THREAD_DISTRIBUTE : LOAD_THREAD_SPAWN_SINGLE , p_cache_mode , true ) ;
return token . is_valid ( ) ? OK : FAILED ;
}
2023-03-05 00:09:18 +00:00
2024-07-15 08:30:02 +00:00
ResourceLoader : : LoadToken * ResourceLoader : : _load_threaded_request_reuse_user_token ( const String & p_path ) {
HashMap < String , LoadToken * > : : Iterator E = user_load_tokens . find ( p_path ) ;
if ( E ) {
print_verbose ( " load_threaded_request(): Another threaded load for resource path ' " + p_path + " ' has been initiated. Not an error. " ) ;
LoadToken * token = E - > value ;
token - > user_rc + + ;
return token ;
2023-03-05 00:09:18 +00:00
} else {
2024-07-15 08:30:02 +00:00
return nullptr ;
2023-03-05 00:09:18 +00:00
}
}
2019-01-27 22:24:55 +00:00
2024-07-15 08:30:02 +00:00
void ResourceLoader : : _load_threaded_request_setup_user_token ( LoadToken * p_token , const String & p_path ) {
p_token - > user_path = p_path ;
p_token - > reference ( ) ; // Extra RC until all user requests have been gotten.
p_token - > user_rc = 1 ;
user_load_tokens [ p_path ] = p_token ;
print_lt ( " REQUEST: user load tokens: " + itos ( user_load_tokens . size ( ) ) ) ;
}
2023-03-05 00:09:18 +00:00
Ref < Resource > ResourceLoader : : load ( const String & p_path , const String & p_type_hint , ResourceFormatLoader : : CacheMode p_cache_mode , Error * r_error ) {
if ( r_error ) {
* r_error = OK ;
2020-02-28 11:27:04 +00:00
}
2014-02-10 01:10:30 +00:00
2024-07-15 08:30:02 +00:00
LoadThreadMode thread_mode = LOAD_THREAD_FROM_CURRENT ;
if ( WorkerThreadPool : : get_singleton ( ) - > get_caller_task_id ( ) ! = WorkerThreadPool : : INVALID_TASK_ID ) {
// If user is initiating a single-threaded load from a WorkerThreadPool task,
// we instead spawn a new task so there's a precondition that a load in a pool task
// is always initiated by the engine. That makes certain aspects simpler, such as
// cyclic load detection and awaiting.
thread_mode = LOAD_THREAD_SPAWN_SINGLE ;
}
Ref < LoadToken > load_token = _load_start ( p_path , p_type_hint , thread_mode , p_cache_mode ) ;
2023-03-05 00:09:18 +00:00
if ( ! load_token . is_valid ( ) ) {
if ( r_error ) {
* r_error = FAILED ;
2020-02-28 11:27:04 +00:00
}
2023-03-05 00:09:18 +00:00
return Ref < Resource > ( ) ;
2020-02-28 11:27:04 +00:00
}
2018-09-02 16:59:33 +00:00
2023-03-05 00:09:18 +00:00
Ref < Resource > res = _load_complete ( * load_token . ptr ( ) , r_error ) ;
return res ;
}
2020-02-28 11:27:04 +00:00
2024-07-15 08:30:02 +00:00
Ref < ResourceLoader : : LoadToken > ResourceLoader : : _load_start ( const String & p_path , const String & p_type_hint , LoadThreadMode p_thread_mode , ResourceFormatLoader : : CacheMode p_cache_mode , bool p_for_user ) {
2023-03-05 00:09:18 +00:00
String local_path = _validate_local_path ( p_path ) ;
2020-02-28 11:27:04 +00:00
2024-06-24 09:25:57 +00:00
bool ignoring_cache = p_cache_mode = = ResourceFormatLoader : : CACHE_MODE_IGNORE | | p_cache_mode = = ResourceFormatLoader : : CACHE_MODE_IGNORE_DEEP ;
2023-03-05 00:09:18 +00:00
Ref < LoadToken > load_token ;
2024-08-13 10:41:11 +00:00
bool must_not_register = false ;
2023-03-05 00:09:18 +00:00
ThreadLoadTask * load_task_ptr = nullptr ;
{
MutexLock thread_load_lock ( thread_load_mutex ) ;
2020-02-28 11:27:04 +00:00
2024-07-15 08:30:02 +00:00
if ( p_for_user ) {
LoadToken * existing_token = _load_threaded_request_reuse_user_token ( p_path ) ;
if ( existing_token ) {
return Ref < LoadToken > ( existing_token ) ;
}
}
2024-06-24 09:25:57 +00:00
if ( ! ignoring_cache & & thread_load_tasks . has ( local_path ) ) {
2023-03-05 00:09:18 +00:00
load_token = Ref < LoadToken > ( thread_load_tasks [ local_path ] . load_token ) ;
2024-07-09 16:41:24 +00:00
if ( load_token . is_valid ( ) ) {
2024-08-28 11:53:39 +00:00
if ( p_for_user ) {
// Load task exists, with no user tokens at the moment.
// Let's "attach" to it.
_load_threaded_request_setup_user_token ( load_token . ptr ( ) , p_path ) ;
}
2024-07-09 16:41:24 +00:00
return load_token ;
} else {
2023-03-05 00:09:18 +00:00
// The token is dying (reached 0 on another thread).
// Ensure it's killed now so the path can be safely reused right away.
thread_load_tasks [ local_path ] . load_token - > clear ( ) ;
2020-02-28 11:27:04 +00:00
}
2023-03-05 00:09:18 +00:00
}
2022-06-22 11:46:46 +00:00
2023-03-05 00:09:18 +00:00
load_token . instantiate ( ) ;
load_token - > local_path = local_path ;
2024-07-15 08:30:02 +00:00
if ( p_for_user ) {
_load_threaded_request_setup_user_token ( load_token . ptr ( ) , p_path ) ;
}
2022-06-22 11:46:46 +00:00
2023-03-05 00:09:18 +00:00
//create load task
{
ThreadLoadTask load_task ;
load_task . load_token = load_token . ptr ( ) ;
load_task . local_path = local_path ;
load_task . type_hint = p_type_hint ;
load_task . cache_mode = p_cache_mode ;
load_task . use_sub_threads = p_thread_mode = = LOAD_THREAD_DISTRIBUTE ;
2024-01-09 14:48:48 +00:00
if ( p_cache_mode = = ResourceFormatLoader : : CACHE_MODE_REUSE ) {
2023-03-05 00:09:18 +00:00
Ref < Resource > existing = ResourceCache : : get_ref ( local_path ) ;
if ( existing . is_valid ( ) ) {
//referencing is fine
load_task . resource = existing ;
load_task . status = THREAD_LOAD_LOADED ;
load_task . progress = 1.0 ;
2024-07-15 08:30:02 +00:00
DEV_ASSERT ( ! thread_load_tasks . has ( local_path ) ) ;
2023-03-05 00:09:18 +00:00
thread_load_tasks [ local_path ] = load_task ;
return load_token ;
}
2020-02-28 11:27:04 +00:00
}
2024-09-05 06:35:05 +00:00
// If we want to ignore cache, but there's another task loading it, we can't add this one to the map.
2024-08-13 10:41:11 +00:00
must_not_register = ignoring_cache & & thread_load_tasks . has ( local_path ) ;
if ( must_not_register ) {
2024-09-05 06:35:05 +00:00
load_token - > task_if_unregistered = memnew ( ThreadLoadTask ( load_task ) ) ;
load_task_ptr = load_token - > task_if_unregistered ;
2023-03-05 00:09:18 +00:00
} else {
2024-06-24 09:25:57 +00:00
DEV_ASSERT ( ! thread_load_tasks . has ( local_path ) ) ;
HashMap < String , ResourceLoader : : ThreadLoadTask > : : Iterator E = thread_load_tasks . insert ( local_path , load_task ) ;
load_task_ptr = & E - > value ;
2023-03-05 00:09:18 +00:00
}
2018-09-02 16:59:33 +00:00
}
2020-02-28 11:27:04 +00:00
2024-09-05 17:06:38 +00:00
// It's important to keep the token alive because until the load completes,
// which includes before the thread start, it may happen that no one is grabbing
// the token anymore so it's released.
load_task_ptr - > load_token - > reference ( ) ;
2024-09-05 06:35:05 +00:00
if ( p_thread_mode = = LOAD_THREAD_FROM_CURRENT ) {
2024-07-10 10:53:14 +00:00
// The current thread may happen to be a thread from the pool.
WorkerThreadPool : : TaskID tid = WorkerThreadPool : : get_singleton ( ) - > get_caller_task_id ( ) ;
if ( tid ! = WorkerThreadPool : : INVALID_TASK_ID ) {
load_task_ptr - > task_id = tid ;
} else {
load_task_ptr - > thread_id = Thread : : get_caller_id ( ) ;
}
2023-03-05 00:09:18 +00:00
} else {
2024-07-10 10:53:14 +00:00
load_task_ptr - > task_id = WorkerThreadPool : : get_singleton ( ) - > add_native_task ( & ResourceLoader : : _run_load_task , load_task_ptr ) ;
2023-03-05 00:09:18 +00:00
}
2024-07-15 08:30:02 +00:00
} // MutexLock(thread_load_mutex).
2017-12-14 18:33:54 +00:00
2024-09-05 06:35:05 +00:00
if ( p_thread_mode = = LOAD_THREAD_FROM_CURRENT ) {
2024-07-10 10:53:14 +00:00
_run_load_task ( load_task_ptr ) ;
2023-03-05 00:09:18 +00:00
}
2020-02-28 11:27:04 +00:00
2023-03-05 00:09:18 +00:00
return load_token ;
2020-02-28 11:27:04 +00:00
}
float ResourceLoader : : _dependency_get_progress ( const String & p_path ) {
if ( thread_load_tasks . has ( p_path ) ) {
ThreadLoadTask & load_task = thread_load_tasks [ p_path ] ;
2024-01-05 18:17:51 +00:00
float current_progress = 0.0 ;
2020-02-28 11:27:04 +00:00
int dep_count = load_task . sub_tasks . size ( ) ;
if ( dep_count > 0 ) {
2022-05-18 23:43:40 +00:00
for ( const String & E : load_task . sub_tasks ) {
2024-01-05 18:17:51 +00:00
current_progress + = _dependency_get_progress ( E ) ;
2020-02-28 11:27:04 +00:00
}
2024-01-05 18:17:51 +00:00
current_progress / = float ( dep_count ) ;
current_progress * = 0.5 ;
current_progress + = load_task . progress * 0.5 ;
2020-02-28 11:27:04 +00:00
} else {
2024-01-05 18:17:51 +00:00
current_progress = load_task . progress ;
2019-01-27 22:24:55 +00:00
}
2024-01-05 18:17:51 +00:00
load_task . max_reported_progress = MAX ( load_task . max_reported_progress , current_progress ) ;
return load_task . max_reported_progress ;
2020-02-28 11:27:04 +00:00
} else {
return 1.0 ; //assume finished loading it so it no longer exists
2017-08-31 21:57:03 +00:00
}
2020-02-28 11:27:04 +00:00
}
2017-06-28 20:00:18 +00:00
2020-02-28 11:27:04 +00:00
ResourceLoader : : ThreadLoadStatus ResourceLoader : : load_threaded_get_status ( const String & p_path , float * r_progress ) {
2024-07-09 16:41:24 +00:00
bool ensure_progress = false ;
ThreadLoadStatus status = THREAD_LOAD_IN_PROGRESS ;
{
MutexLock thread_load_lock ( thread_load_mutex ) ;
2014-02-10 01:10:30 +00:00
2024-07-09 16:41:24 +00:00
if ( ! user_load_tokens . has ( p_path ) ) {
print_verbose ( " load_threaded_get_status(): No threaded load for resource path ' " + p_path + " ' has been initiated or its result has already been collected. " ) ;
return THREAD_LOAD_INVALID_RESOURCE ;
}
2023-03-05 00:09:18 +00:00
2024-07-09 16:41:24 +00:00
String local_path = _validate_local_path ( p_path ) ;
2024-08-13 10:52:08 +00:00
ERR_FAIL_COND_V_MSG ( ! thread_load_tasks . has ( local_path ) , THREAD_LOAD_INVALID_RESOURCE , " Bug in ResourceLoader logic, please report. " ) ;
2023-03-05 00:09:18 +00:00
2024-07-09 16:41:24 +00:00
ThreadLoadTask & load_task = thread_load_tasks [ local_path ] ;
status = load_task . status ;
if ( r_progress ) {
* r_progress = _dependency_get_progress ( local_path ) ;
}
2019-01-27 22:24:55 +00:00
2024-07-09 16:41:24 +00:00
// Support userland polling in a loop on the main thread.
if ( Thread : : is_main_thread ( ) & & status = = THREAD_LOAD_IN_PROGRESS ) {
uint64_t frame = Engine : : get_singleton ( ) - > get_process_frames ( ) ;
if ( frame = = load_task . last_progress_check_main_thread_frame ) {
ensure_progress = true ;
} else {
load_task . last_progress_check_main_thread_frame = frame ;
}
2024-06-28 09:16:59 +00:00
}
}
2024-07-09 16:41:24 +00:00
if ( ensure_progress ) {
_ensure_load_progress ( ) ;
}
2020-02-28 11:27:04 +00:00
return status ;
2014-02-10 01:10:30 +00:00
}
2020-05-14 12:29:06 +00:00
2022-05-02 23:43:50 +00:00
Ref < Resource > ResourceLoader : : load_threaded_get ( const String & p_path , Error * r_error ) {
2023-03-05 00:09:18 +00:00
if ( r_error ) {
* r_error = OK ;
2018-08-10 18:57:43 +00:00
}
2023-03-05 00:09:18 +00:00
Ref < Resource > res ;
{
MutexLock thread_load_lock ( thread_load_mutex ) ;
2018-08-10 18:57:43 +00:00
2023-03-05 00:09:18 +00:00
if ( ! user_load_tokens . has ( p_path ) ) {
print_verbose ( " load_threaded_get(): No threaded load for resource path ' " + p_path + " ' has been initiated or its result has already been collected. " ) ;
2023-02-26 16:17:41 +00:00
if ( r_error ) {
2023-03-05 00:09:18 +00:00
* r_error = ERR_INVALID_PARAMETER ;
2023-02-26 16:17:41 +00:00
}
return Ref < Resource > ( ) ;
2018-08-10 18:57:43 +00:00
}
2023-03-05 00:09:18 +00:00
LoadToken * load_token = user_load_tokens [ p_path ] ;
2024-07-15 08:30:02 +00:00
DEV_ASSERT ( load_token - > user_rc > = 1 ) ;
2024-06-28 09:16:59 +00:00
// Support userland requesting on the main thread before the load is reported to be complete.
if ( Thread : : is_main_thread ( ) & & ! load_token - > local_path . is_empty ( ) ) {
const ThreadLoadTask & load_task = thread_load_tasks [ load_token - > local_path ] ;
while ( load_task . status = = THREAD_LOAD_IN_PROGRESS ) {
2024-06-29 11:01:01 +00:00
thread_load_lock . temp_unlock ( ) ;
2024-07-09 16:41:24 +00:00
bool exit = ! _ensure_load_progress ( ) ;
2024-06-28 09:16:59 +00:00
OS : : get_singleton ( ) - > delay_usec ( 1000 ) ;
2024-06-29 11:01:01 +00:00
thread_load_lock . temp_relock ( ) ;
2024-07-09 16:41:24 +00:00
if ( exit ) {
break ;
}
2024-06-28 09:16:59 +00:00
}
}
2023-03-05 00:09:18 +00:00
res = _load_complete_inner ( * load_token , r_error , thread_load_lock ) ;
2024-07-15 08:30:02 +00:00
load_token - > user_rc - - ;
if ( load_token - > user_rc = = 0 ) {
load_token - > user_path . clear ( ) ;
user_load_tokens . erase ( p_path ) ;
if ( load_token - > unreference ( ) ) {
memdelete ( load_token ) ;
load_token = nullptr ;
}
2023-03-05 00:09:18 +00:00
}
2018-08-10 18:57:43 +00:00
}
2023-03-05 00:09:18 +00:00
print_lt ( " GET: user load tokens: " + itos ( user_load_tokens . size ( ) ) ) ;
2020-02-28 11:27:04 +00:00
2023-03-05 00:09:18 +00:00
return res ;
}
2020-02-28 11:27:04 +00:00
2023-03-05 00:09:18 +00:00
Ref < Resource > ResourceLoader : : _load_complete ( LoadToken & p_load_token , Error * r_error ) {
MutexLock thread_load_lock ( thread_load_mutex ) ;
return _load_complete_inner ( p_load_token , r_error , thread_load_lock ) ;
2018-08-10 18:57:43 +00:00
}
2023-03-05 00:09:18 +00:00
Ref < Resource > ResourceLoader : : _load_complete_inner ( LoadToken & p_load_token , Error * r_error , MutexLock < SafeBinaryMutex < BINARY_MUTEX_TAG > > & p_thread_load_lock ) {
2020-05-14 14:41:43 +00:00
if ( r_error ) {
2023-03-05 00:09:18 +00:00
* r_error = OK ;
2020-05-14 14:41:43 +00:00
}
2014-02-10 01:10:30 +00:00
2024-09-05 06:35:05 +00:00
ThreadLoadTask * load_task_ptr = nullptr ;
if ( p_load_token . task_if_unregistered ) {
load_task_ptr = p_load_token . task_if_unregistered ;
} else {
2023-03-05 00:09:18 +00:00
if ( ! thread_load_tasks . has ( p_load_token . local_path ) ) {
if ( r_error ) {
* r_error = ERR_BUG ;
}
2024-08-13 10:52:08 +00:00
ERR_FAIL_V_MSG ( Ref < Resource > ( ) , " Bug in ResourceLoader logic, please report. " ) ;
2023-03-05 00:09:18 +00:00
}
2014-02-10 01:10:30 +00:00
2023-03-05 00:09:18 +00:00
ThreadLoadTask & load_task = thread_load_tasks [ p_load_token . local_path ] ;
2016-05-29 23:22:00 +00:00
2023-03-05 00:09:18 +00:00
if ( load_task . status = = THREAD_LOAD_IN_PROGRESS ) {
2023-05-10 09:23:07 +00:00
DEV_ASSERT ( ( load_task . task_id = = 0 ) ! = ( load_task . thread_id = = 0 ) ) ;
2024-07-10 10:53:14 +00:00
if ( ( load_task . task_id ! = 0 & & load_task . task_id = = WorkerThreadPool : : get_singleton ( ) - > get_caller_task_id ( ) ) | |
2023-05-10 09:23:07 +00:00
( load_task . thread_id ! = 0 & & load_task . thread_id = = Thread : : get_caller_id ( ) ) ) {
2023-03-05 00:09:18 +00:00
// Load is in progress, but it's precisely this thread the one in charge.
// That means this is a cyclic load.
2020-02-28 11:27:04 +00:00
if ( r_error ) {
2023-03-05 00:09:18 +00:00
* r_error = ERR_BUSY ;
2020-02-28 11:27:04 +00:00
}
2022-05-02 23:43:50 +00:00
return Ref < Resource > ( ) ;
2020-02-28 11:27:04 +00:00
}
2019-01-27 22:24:55 +00:00
2024-06-12 11:45:31 +00:00
bool loader_is_wtp = load_task . task_id ! = 0 ;
if ( loader_is_wtp ) {
2023-05-16 22:00:45 +00:00
// Loading thread is in the worker pool.
2024-06-29 11:01:01 +00:00
p_thread_load_lock . temp_unlock ( ) ;
2024-07-15 08:30:02 +00:00
2024-07-10 10:01:22 +00:00
PREPARE_FOR_WTP_WAIT
2024-07-15 08:30:02 +00:00
Error wait_err = WorkerThreadPool : : get_singleton ( ) - > wait_for_task_completion ( load_task . task_id ) ;
2024-07-10 10:01:22 +00:00
RESTORE_AFTER_WTP_WAIT
2024-06-12 11:45:31 +00:00
2024-07-15 08:30:02 +00:00
DEV_ASSERT ( ! wait_err | | wait_err = = ERR_BUSY ) ;
if ( wait_err = = ERR_BUSY ) {
// The WorkerThreadPool has reported that the current task wants to await on an older one.
// That't not allowed for safety, to avoid deadlocks. Fortunately, though, in the context of
// resource loading that means that the task to wait for can be restarted here to break the
// cycle, with as much recursion into this process as needed.
// When the stack is eventually unrolled, the original load will have been notified to go on.
2024-09-05 17:06:38 +00:00
load_task . load_token - > reference ( ) ;
2024-07-15 08:30:02 +00:00
_run_load_task ( & load_task ) ;
2023-05-16 22:00:45 +00:00
}
2024-07-15 08:30:02 +00:00
2024-06-29 11:01:01 +00:00
p_thread_load_lock . temp_relock ( ) ;
2024-07-15 08:30:02 +00:00
load_task . awaited = true ;
DEV_ASSERT ( load_task . status = = THREAD_LOAD_FAILED | | load_task . status = = THREAD_LOAD_LOADED ) ;
} else if ( load_task . need_wait ) {
// Loading thread is main or user thread.
if ( ! load_task . cond_var ) {
load_task . cond_var = memnew ( ConditionVariable ) ;
2023-05-10 09:23:07 +00:00
}
2024-07-15 08:30:02 +00:00
load_task . awaiters_count + + ;
do {
load_task . cond_var - > wait ( p_thread_load_lock ) ;
DEV_ASSERT ( thread_load_tasks . has ( p_load_token . local_path ) & & p_load_token . get_reference_count ( ) ) ;
} while ( load_task . need_wait ) ;
load_task . awaiters_count - - ;
if ( load_task . awaiters_count = = 0 ) {
memdelete ( load_task . cond_var ) ;
load_task . cond_var = nullptr ;
}
DEV_ASSERT ( load_task . status = = THREAD_LOAD_FAILED | | load_task . status = = THREAD_LOAD_LOADED ) ;
2023-05-10 09:23:07 +00:00
}
2020-02-28 11:27:04 +00:00
}
2023-03-05 00:09:18 +00:00
if ( cleaning_tasks ) {
load_task . resource = Ref < Resource > ( ) ;
load_task . error = FAILED ;
2019-01-27 22:24:55 +00:00
}
2024-09-05 06:35:05 +00:00
load_task_ptr = & load_task ;
}
2024-09-06 15:43:05 +00:00
p_thread_load_lock . temp_unlock ( ) ;
2024-09-05 06:35:05 +00:00
Ref < Resource > resource = load_task_ptr - > resource ;
if ( r_error ) {
* r_error = load_task_ptr - > error ;
2014-02-10 01:10:30 +00:00
}
2024-09-05 07:48:13 +00:00
if ( resource . is_valid ( ) ) {
if ( curr_load_task ) {
// A task awaiting another => Let the awaiter accumulate the resource changed connections.
DEV_ASSERT ( curr_load_task ! = load_task_ptr ) ;
for ( const ThreadLoadTask : : ResourceChangedConnection & rcc : load_task_ptr - > resource_changed_connections ) {
curr_load_task - > resource_changed_connections . push_back ( rcc ) ;
}
} else {
// A leaf task being awaited => Propagate the resource changed connections.
if ( Thread : : is_main_thread ( ) ) {
// On the main thread it's safe to migrate the connections to the standard signal mechanism.
for ( const ThreadLoadTask : : ResourceChangedConnection & rcc : load_task_ptr - > resource_changed_connections ) {
if ( rcc . callable . is_valid ( ) ) {
rcc . source - > connect_changed ( rcc . callable , rcc . flags ) ;
}
}
} else {
// On non-main threads, we have to queue and call it done when processed.
if ( ! load_task_ptr - > resource_changed_connections . is_empty ( ) ) {
for ( const ThreadLoadTask : : ResourceChangedConnection & rcc : load_task_ptr - > resource_changed_connections ) {
if ( rcc . callable . is_valid ( ) ) {
MessageQueue : : get_main_singleton ( ) - > push_callable ( callable_mp ( rcc . source , & Resource : : connect_changed ) . bind ( rcc . callable , rcc . flags ) ) ;
}
}
core_bind : : Semaphore done ;
2024-09-12 09:37:07 +00:00
MessageQueue : : get_main_singleton ( ) - > push_callable ( callable_mp ( & done , & core_bind : : Semaphore : : post ) . bind ( 1 ) ) ;
2024-09-05 07:48:13 +00:00
done . wait ( ) ;
}
}
}
}
2024-09-06 15:43:05 +00:00
p_thread_load_lock . temp_relock ( ) ;
2024-09-05 06:35:05 +00:00
return resource ;
2020-02-28 11:27:04 +00:00
}
2014-02-10 01:10:30 +00:00
2024-06-28 09:16:59 +00:00
bool ResourceLoader : : _ensure_load_progress ( ) {
// Some servers may need a new engine iteration to allow the load to progress.
// Since the only known one is the rendering server (in single thread mode), let's keep it simple and just sync it.
// This may be refactored in the future to support other servers and have less coupling.
if ( OS : : get_singleton ( ) - > get_render_thread_mode ( ) = = OS : : RENDER_SEPARATE_THREAD ) {
return false ; // Not needed.
}
RenderingServer : : get_singleton ( ) - > sync ( ) ;
return true ;
}
2024-09-05 07:48:13 +00:00
void ResourceLoader : : resource_changed_connect ( Resource * p_source , const Callable & p_callable , uint32_t p_flags ) {
print_lt ( vformat ( " %d \t %ud:%s \t " FUNCTION_STR " \t %d " , Thread : : get_caller_id ( ) , p_source - > get_instance_id ( ) , p_source - > get_class ( ) , p_callable . get_object_id ( ) ) ) ;
MutexLock lock ( thread_load_mutex ) ;
for ( const ThreadLoadTask : : ResourceChangedConnection & rcc : curr_load_task - > resource_changed_connections ) {
if ( unlikely ( rcc . source = = p_source & & rcc . callable = = p_callable ) ) {
return ;
}
}
ThreadLoadTask : : ResourceChangedConnection rcc ;
rcc . source = p_source ;
rcc . callable = p_callable ;
rcc . flags = p_flags ;
curr_load_task - > resource_changed_connections . push_back ( rcc ) ;
}
void ResourceLoader : : resource_changed_disconnect ( Resource * p_source , const Callable & p_callable ) {
print_lt ( vformat ( " %d \t %ud:%s \t " FUNCTION_STR " t%d " , Thread : : get_caller_id ( ) , p_source - > get_instance_id ( ) , p_source - > get_class ( ) , p_callable . get_object_id ( ) ) ) ;
MutexLock lock ( thread_load_mutex ) ;
for ( uint32_t i = 0 ; i < curr_load_task - > resource_changed_connections . size ( ) ; + + i ) {
const ThreadLoadTask : : ResourceChangedConnection & rcc = curr_load_task - > resource_changed_connections [ i ] ;
if ( unlikely ( rcc . source = = p_source & & rcc . callable = = p_callable ) ) {
curr_load_task - > resource_changed_connections . remove_at_unordered ( i ) ;
return ;
}
}
}
void ResourceLoader : : resource_changed_emit ( Resource * p_source ) {
print_lt ( vformat ( " %d \t %ud:%s \t " FUNCTION_STR , Thread : : get_caller_id ( ) , p_source - > get_instance_id ( ) , p_source - > get_class ( ) ) ) ;
MutexLock lock ( thread_load_mutex ) ;
for ( const ThreadLoadTask : : ResourceChangedConnection & rcc : curr_load_task - > resource_changed_connections ) {
if ( unlikely ( rcc . source = = p_source ) ) {
rcc . callable . call ( ) ;
}
}
}
2024-06-25 12:12:40 +00:00
Ref < Resource > ResourceLoader : : ensure_resource_ref_override_for_outer_load ( const String & p_path , const String & p_res_type ) {
ERR_FAIL_COND_V ( load_nesting = = 0 , Ref < Resource > ( ) ) ; // It makes no sense to use this from nesting level 0.
const String & local_path = _validate_local_path ( p_path ) ;
HashMap < String , Ref < Resource > > & overrides = res_ref_overrides [ load_nesting - 1 ] ;
HashMap < String , Ref < Resource > > : : Iterator E = overrides . find ( local_path ) ;
if ( E ) {
return E - > value ;
} else {
Object * obj = ClassDB : : instantiate ( p_res_type ) ;
ERR_FAIL_NULL_V ( obj , Ref < Resource > ( ) ) ;
Ref < Resource > res ( obj ) ;
if ( ! res . is_valid ( ) ) {
memdelete ( obj ) ;
ERR_FAIL_V ( Ref < Resource > ( ) ) ;
}
overrides [ local_path ] = res ;
return res ;
}
}
Ref < Resource > ResourceLoader : : get_resource_ref_override ( const String & p_path ) {
DEV_ASSERT ( p_path = = _validate_local_path ( p_path ) ) ;
HashMap < int , HashMap < String , Ref < Resource > > > : : Iterator E = res_ref_overrides . find ( load_nesting ) ;
if ( ! E ) {
return nullptr ;
}
HashMap < String , Ref < Resource > > : : Iterator F = E - > value . find ( p_path ) ;
if ( ! F ) {
return nullptr ;
}
return F - > value ;
}
2020-02-28 11:27:04 +00:00
bool ResourceLoader : : exists ( const String & p_path , const String & p_type_hint ) {
2021-07-23 19:01:18 +00:00
String local_path = _validate_local_path ( p_path ) ;
2020-02-28 11:27:04 +00:00
if ( ResourceCache : : has ( local_path ) ) {
return true ; // If cached, it probably exists
2019-01-27 22:24:55 +00:00
}
2020-02-28 11:27:04 +00:00
bool xl_remapped = false ;
String path = _path_remap ( local_path , & xl_remapped ) ;
2019-08-15 02:57:49 +00:00
2020-02-28 11:27:04 +00:00
// Try all loaders and pick the first match for the type hint
for ( int i = 0 ; i < loader_count ; i + + ) {
if ( ! loader [ i ] - > recognize_path ( path , p_type_hint ) ) {
continue ;
}
2020-05-14 14:41:43 +00:00
if ( loader [ i ] - > exists ( path ) ) {
2020-02-28 11:27:04 +00:00
return true ;
2020-05-14 14:41:43 +00:00
}
2020-02-28 11:27:04 +00:00
}
return false ;
2014-02-10 01:10:30 +00:00
}
2018-06-11 00:59:53 +00:00
void ResourceLoader : : add_resource_format_loader ( Ref < ResourceFormatLoader > p_format_loader , bool p_at_front ) {
ERR_FAIL_COND ( p_format_loader . is_null ( ) ) ;
2014-02-10 01:10:30 +00:00
ERR_FAIL_COND ( loader_count > = MAX_LOADERS ) ;
2018-06-11 00:59:53 +00:00
2016-07-20 00:40:05 +00:00
if ( p_at_front ) {
for ( int i = loader_count ; i > 0 ; i - - ) {
loader [ i ] = loader [ i - 1 ] ;
}
loader [ 0 ] = p_format_loader ;
loader_count + + ;
} else {
loader [ loader_count + + ] = p_format_loader ;
}
2014-02-10 01:10:30 +00:00
}
2018-06-11 00:59:53 +00:00
void ResourceLoader : : remove_resource_format_loader ( Ref < ResourceFormatLoader > p_format_loader ) {
ERR_FAIL_COND ( p_format_loader . is_null ( ) ) ;
// Find loader
int i = 0 ;
for ( ; i < loader_count ; + + i ) {
2020-05-14 14:41:43 +00:00
if ( loader [ i ] = = p_format_loader ) {
2018-06-11 00:59:53 +00:00
break ;
2020-05-14 14:41:43 +00:00
}
2018-06-11 00:59:53 +00:00
}
ERR_FAIL_COND ( i > = loader_count ) ; // Not found
// Shift next loaders up
for ( ; i < loader_count - 1 ; + + i ) {
loader [ i ] = loader [ i + 1 ] ;
}
loader [ loader_count - 1 ] . unref ( ) ;
- - loader_count ;
}
2017-09-20 23:59:19 +00:00
int ResourceLoader : : get_import_order ( const String & p_path ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2017-09-20 23:59:19 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2017-09-20 23:59:19 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2017-09-20 23:59:19 +00:00
return loader [ i ] - > get_import_order ( p_path ) ;
}
return 0 ;
}
2019-04-19 18:54:33 +00:00
String ResourceLoader : : get_import_group_file ( const String & p_path ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2019-04-19 18:54:33 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2019-04-19 18:54:33 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2019-04-19 18:54:33 +00:00
return loader [ i ] - > get_import_group_file ( p_path ) ;
}
return String ( ) ; //not found
}
2017-08-29 22:50:58 +00:00
bool ResourceLoader : : is_import_valid ( const String & p_path ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2017-08-29 22:50:58 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2017-08-29 22:50:58 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2017-08-29 22:50:58 +00:00
return loader [ i ] - > is_import_valid ( p_path ) ;
}
return false ; //not found
}
2019-03-04 14:06:15 +00:00
bool ResourceLoader : : is_imported ( const String & p_path ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2019-03-04 14:06:15 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2019-03-04 14:06:15 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2019-03-04 14:06:15 +00:00
return loader [ i ] - > is_imported ( p_path ) ;
}
return false ; //not found
}
2015-08-23 23:15:56 +00:00
void ResourceLoader : : get_dependencies ( const String & p_path , List < String > * p_dependencies , bool p_add_types ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2015-08-23 23:15:56 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2015-08-23 23:15:56 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2015-08-23 23:15:56 +00:00
2017-01-26 00:55:59 +00:00
loader [ i ] - > get_dependencies ( local_path , p_dependencies , p_add_types ) ;
2015-08-23 23:15:56 +00:00
}
}
2022-05-13 13:04:37 +00:00
Error ResourceLoader : : rename_dependencies ( const String & p_path , const HashMap < String , String > & p_map ) {
2021-07-23 19:01:18 +00:00
String local_path = _path_remap ( _validate_local_path ( p_path ) ) ;
2015-05-04 13:17:24 +00:00
2014-02-10 01:10:30 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
2020-05-14 14:41:43 +00:00
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
2014-02-10 01:10:30 +00:00
continue ;
2020-05-14 14:41:43 +00:00
}
2014-02-10 01:10:30 +00:00
2017-01-26 00:55:59 +00:00
return loader [ i ] - > rename_dependencies ( local_path , p_map ) ;
2014-02-10 01:10:30 +00:00
}
2015-08-23 23:15:56 +00:00
return OK ; // ??
2014-02-10 01:10:30 +00:00
}
2022-07-14 12:18:18 +00:00
void ResourceLoader : : get_classes_used ( const String & p_path , HashSet < StringName > * r_classes ) {
String local_path = _validate_local_path ( p_path ) ;
for ( int i = 0 ; i < loader_count ; i + + ) {
if ( ! loader [ i ] - > recognize_path ( local_path ) ) {
continue ;
}
return loader [ i ] - > get_classes_used ( p_path , r_classes ) ;
}
}
2014-02-10 01:10:30 +00:00
String ResourceLoader : : get_resource_type ( const String & p_path ) {
2021-07-23 19:01:18 +00:00
String local_path = _validate_local_path ( p_path ) ;
2015-05-04 13:17:24 +00:00
2014-02-10 01:10:30 +00:00
for ( int i = 0 ; i < loader_count ; i + + ) {
String result = loader [ i ] - > get_resource_type ( local_path ) ;
2021-12-09 09:42:46 +00:00
if ( ! result . is_empty ( ) ) {
2014-02-10 01:10:30 +00:00
return result ;
2019-10-03 20:39:08 +00:00
}
2014-02-10 01:10:30 +00:00
}
return " " ;
}
2017-06-28 20:00:18 +00:00
2023-01-19 18:12:25 +00:00
String ResourceLoader : : get_resource_script_class ( const String & p_path ) {
String local_path = _validate_local_path ( p_path ) ;
for ( int i = 0 ; i < loader_count ; i + + ) {
String result = loader [ i ] - > get_resource_script_class ( local_path ) ;
if ( ! result . is_empty ( ) ) {
return result ;
}
}
return " " ;
}
2021-07-23 19:01:18 +00:00
ResourceUID : : ID ResourceLoader : : get_resource_uid ( const String & p_path ) {
String local_path = _validate_local_path ( p_path ) ;
for ( int i = 0 ; i < loader_count ; i + + ) {
ResourceUID : : ID id = loader [ i ] - > get_resource_uid ( local_path ) ;
if ( id ! = ResourceUID : : INVALID_ID ) {
return id ;
}
}
return ResourceUID : : INVALID_ID ;
}
2017-06-28 20:00:18 +00:00
String ResourceLoader : : _path_remap ( const String & p_path , bool * r_translation_remapped ) {
2017-12-14 18:33:54 +00:00
String new_path = p_path ;
2017-06-28 20:00:18 +00:00
2019-12-04 15:50:43 +00:00
if ( translation_remaps . has ( p_path ) ) {
// translation_remaps has the following format:
2020-02-17 21:06:54 +00:00
// { "res://path.png": PackedStringArray( "res://path-ru.png:ru", "res://path-de.png:de" ) }
2019-12-04 15:50:43 +00:00
// To find the path of the remapped resource, we extract the locale name after
// the last ':' to match the project locale.
2017-12-14 18:33:54 +00:00
2022-07-29 14:30:29 +00:00
// An extra remap may still be necessary afterwards due to the text -> binary converter on export.
2017-06-28 20:00:18 +00:00
String locale = TranslationServer : : get_singleton ( ) - > get_locale ( ) ;
2019-12-04 15:50:43 +00:00
ERR_FAIL_COND_V_MSG ( locale . length ( ) < 2 , p_path , " Could not remap path ' " + p_path + " ' for translation as configured locale ' " + locale + " ' is invalid. " ) ;
2017-06-28 20:00:18 +00:00
2019-12-04 15:50:43 +00:00
Vector < String > & res_remaps = * translation_remaps . getptr ( new_path ) ;
2021-09-23 11:08:50 +00:00
int best_score = 0 ;
2019-12-04 15:50:43 +00:00
for ( int i = 0 ; i < res_remaps . size ( ) ; i + + ) {
2020-07-03 13:26:22 +00:00
int split = res_remaps [ i ] . rfind ( " : " ) ;
2019-12-04 15:50:43 +00:00
if ( split = = - 1 ) {
2017-06-28 20:00:18 +00:00
continue ;
2019-12-04 15:50:43 +00:00
}
2020-02-13 15:42:49 +00:00
String l = res_remaps [ i ] . substr ( split + 1 ) . strip_edges ( ) ;
2021-09-23 11:08:50 +00:00
int score = TranslationServer : : get_singleton ( ) - > compare_locales ( locale , l ) ;
2022-01-24 16:58:16 +00:00
if ( score > 0 & & score > = best_score ) {
2019-12-04 15:50:43 +00:00
new_path = res_remaps [ i ] . left ( split ) ;
2021-09-23 11:08:50 +00:00
best_score = score ;
if ( score = = 10 ) {
break ; // Exact match, skip the rest.
}
2019-12-04 15:50:43 +00:00
}
}
if ( r_translation_remapped ) {
* r_translation_remapped = true ;
2017-06-28 20:00:18 +00:00
}
2022-01-04 17:52:32 +00:00
// Fallback to p_path if new_path does not exist.
2023-11-12 12:16:37 +00:00
if ( ! FileAccess : : exists ( new_path + " .import " ) & & ! FileAccess : : exists ( new_path ) ) {
2022-01-04 17:52:32 +00:00
WARN_PRINT ( vformat ( " Translation remap '%s' does not exist. Falling back to '%s'. " , new_path , p_path ) ) ;
new_path = p_path ;
}
2017-06-28 20:00:18 +00:00
}
2017-12-14 18:33:54 +00:00
if ( path_remaps . has ( new_path ) ) {
new_path = path_remaps [ new_path ] ;
2022-07-29 14:30:29 +00:00
} else {
2019-12-04 15:50:43 +00:00
// Try file remap.
2024-06-12 00:11:10 +00:00
// Usually, there's no remap file and FileAccess::exists() is faster than FileAccess::open().
if ( FileAccess : : exists ( new_path + " .remap " ) ) {
Error err ;
Ref < FileAccess > f = FileAccess : : open ( new_path + " .remap " , FileAccess : : READ , & err ) ;
if ( f . is_valid ( ) ) {
VariantParser : : StreamFile stream ;
stream . f = f ;
String assign ;
Variant value ;
VariantParser : : Tag next_tag ;
int lines = 0 ;
String error_text ;
while ( true ) {
assign = Variant ( ) ;
next_tag . fields . clear ( ) ;
next_tag . name = String ( ) ;
err = VariantParser : : parse_tag_assign_eof ( & stream , lines , error_text , next_tag , assign , value , nullptr , true ) ;
if ( err = = ERR_FILE_EOF ) {
break ;
} else if ( err ! = OK ) {
ERR_PRINT ( " Parse error: " + p_path + " .remap: " + itos ( lines ) + " error: " + error_text + " . " ) ;
break ;
}
2017-12-18 14:21:13 +00:00
2024-06-12 00:11:10 +00:00
if ( assign = = " path " ) {
new_path = value ;
break ;
} else if ( next_tag . name ! = " remap " ) {
break ;
}
2017-12-18 14:21:13 +00:00
}
}
}
}
2017-12-14 18:33:54 +00:00
return new_path ;
2017-06-28 20:00:18 +00:00
}
String ResourceLoader : : import_remap ( const String & p_path ) {
if ( ResourceFormatImporter : : get_singleton ( ) - > recognize_path ( p_path ) ) {
return ResourceFormatImporter : : get_singleton ( ) - > get_internal_resource_path ( p_path ) ;
}
return p_path ;
}
String ResourceLoader : : path_remap ( const String & p_path ) {
return _path_remap ( p_path ) ;
}
void ResourceLoader : : reload_translation_remaps ( ) {
List < Resource * > to_reload ;
2024-08-28 09:27:01 +00:00
{
MutexLock lock ( ResourceCache : : lock ) ;
SelfList < Resource > * E = remapped_list . first ( ) ;
2017-06-28 20:00:18 +00:00
2024-08-28 09:27:01 +00:00
while ( E ) {
to_reload . push_back ( E - > self ( ) ) ;
E = E - > next ( ) ;
}
}
2017-06-28 20:00:18 +00:00
//now just make sure to not delete any of these resources while changing locale..
while ( to_reload . front ( ) ) {
to_reload . front ( ) - > get ( ) - > reload_from_file ( ) ;
to_reload . pop_front ( ) ;
}
}
void ResourceLoader : : load_translation_remaps ( ) {
2021-02-17 16:44:49 +00:00
if ( ! ProjectSettings : : get_singleton ( ) - > has_setting ( " internationalization/locale/translation_remaps " ) ) {
2017-07-26 13:03:13 +00:00
return ;
2020-05-14 14:41:43 +00:00
}
2017-07-26 13:03:13 +00:00
2022-10-18 14:43:37 +00:00
Dictionary remaps = GLOBAL_GET ( " internationalization/locale/translation_remaps " ) ;
2017-06-28 20:00:18 +00:00
List < Variant > keys ;
remaps . get_key_list ( & keys ) ;
2021-07-24 13:46:25 +00:00
for ( const Variant & E : keys ) {
2021-07-16 03:45:57 +00:00
Array langs = remaps [ E ] ;
2017-06-28 20:00:18 +00:00
Vector < String > lang_remaps ;
lang_remaps . resize ( langs . size ( ) ) ;
2023-12-24 12:44:21 +00:00
String * lang_remaps_ptrw = lang_remaps . ptrw ( ) ;
for ( const Variant & lang : langs ) {
* lang_remaps_ptrw + + = lang ;
2017-06-28 20:00:18 +00:00
}
2021-07-16 03:45:57 +00:00
translation_remaps [ String ( E ) ] = lang_remaps ;
2017-06-28 20:00:18 +00:00
}
}
void ResourceLoader : : clear_translation_remaps ( ) {
translation_remaps . clear ( ) ;
2020-08-30 19:44:39 +00:00
while ( remapped_list . first ( ) ! = nullptr ) {
remapped_list . remove ( remapped_list . first ( ) ) ;
}
2017-06-28 20:00:18 +00:00
}
2022-12-06 18:51:06 +00:00
void ResourceLoader : : clear_thread_load_tasks ( ) {
2023-03-05 00:09:18 +00:00
// Bring the thing down as quickly as possible without causing deadlocks or leaks.
2024-06-29 11:01:01 +00:00
MutexLock thread_load_lock ( thread_load_mutex ) ;
2023-03-05 00:09:18 +00:00
cleaning_tasks = true ;
while ( true ) {
bool none_running = true ;
if ( thread_load_tasks . size ( ) ) {
for ( KeyValue < String , ResourceLoader : : ThreadLoadTask > & E : thread_load_tasks ) {
if ( E . value . status = = THREAD_LOAD_IN_PROGRESS ) {
2024-07-29 17:20:55 +00:00
if ( E . value . cond_var & & E . value . need_wait ) {
2023-03-05 00:09:18 +00:00
E . value . cond_var - > notify_all ( ) ;
}
2024-07-29 17:20:55 +00:00
E . value . need_wait = false ;
2023-03-05 00:09:18 +00:00
none_running = false ;
2022-12-06 18:51:06 +00:00
}
}
}
2023-03-05 00:09:18 +00:00
if ( none_running ) {
break ;
}
2024-06-29 11:01:01 +00:00
thread_load_lock . temp_unlock ( ) ;
2023-03-05 00:09:18 +00:00
OS : : get_singleton ( ) - > delay_usec ( 1000 ) ;
2024-06-29 11:01:01 +00:00
thread_load_lock . temp_relock ( ) ;
2022-12-06 18:51:06 +00:00
}
2023-03-05 00:09:18 +00:00
2023-10-22 14:59:46 +00:00
while ( user_load_tokens . begin ( ) ) {
2024-07-15 08:30:02 +00:00
LoadToken * user_token = user_load_tokens . begin ( ) - > value ;
user_load_tokens . remove ( user_load_tokens . begin ( ) ) ;
DEV_ASSERT ( user_token - > user_rc > 0 & & ! user_token - > user_path . is_empty ( ) ) ;
user_token - > user_path . clear ( ) ;
user_token - > user_rc = 0 ;
user_token - > unreference ( ) ;
2023-03-05 00:09:18 +00:00
}
2022-12-06 18:51:06 +00:00
thread_load_tasks . clear ( ) ;
2023-03-05 00:09:18 +00:00
cleaning_tasks = false ;
2022-12-06 18:51:06 +00:00
}
2017-12-14 18:33:54 +00:00
void ResourceLoader : : load_path_remaps ( ) {
2020-05-14 14:41:43 +00:00
if ( ! ProjectSettings : : get_singleton ( ) - > has_setting ( " path_remap/remapped_paths " ) ) {
2017-12-14 18:33:54 +00:00
return ;
2020-05-14 14:41:43 +00:00
}
2017-12-14 18:33:54 +00:00
2022-10-18 14:43:37 +00:00
Vector < String > remaps = GLOBAL_GET ( " path_remap/remapped_paths " ) ;
2017-12-14 18:33:54 +00:00
int rc = remaps . size ( ) ;
ERR_FAIL_COND ( rc & 1 ) ; //must be even
2020-02-17 21:06:54 +00:00
const String * r = remaps . ptr ( ) ;
2017-12-14 18:33:54 +00:00
for ( int i = 0 ; i < rc ; i + = 2 ) {
path_remaps [ r [ i ] ] = r [ i + 1 ] ;
}
}
void ResourceLoader : : clear_path_remaps ( ) {
path_remaps . clear ( ) ;
}
2018-10-29 19:36:31 +00:00
void ResourceLoader : : set_load_callback ( ResourceLoadedCallback p_callback ) {
_loaded_callback = p_callback ;
}
2020-04-01 23:20:12 +00:00
ResourceLoadedCallback ResourceLoader : : _loaded_callback = nullptr ;
2018-10-29 19:36:31 +00:00
2024-01-09 01:36:19 +00:00
Ref < ResourceFormatLoader > ResourceLoader : : _find_custom_resource_format_loader ( const String & path ) {
2018-06-11 00:59:53 +00:00
for ( int i = 0 ; i < loader_count ; + + i ) {
if ( loader [ i ] - > get_script_instance ( ) & & loader [ i ] - > get_script_instance ( ) - > get_script ( ) - > get_path ( ) = = path ) {
return loader [ i ] ;
}
}
return Ref < ResourceFormatLoader > ( ) ;
}
2024-01-09 01:36:19 +00:00
bool ResourceLoader : : add_custom_resource_format_loader ( const String & script_path ) {
2020-05-14 14:41:43 +00:00
if ( _find_custom_resource_format_loader ( script_path ) . is_valid ( ) ) {
2018-06-11 00:59:53 +00:00
return false ;
2020-05-14 14:41:43 +00:00
}
2018-06-11 00:59:53 +00:00
Ref < Resource > res = ResourceLoader : : load ( script_path ) ;
ERR_FAIL_COND_V ( res . is_null ( ) , false ) ;
ERR_FAIL_COND_V ( ! res - > is_class ( " Script " ) , false ) ;
Ref < Script > s = res ;
StringName ibt = s - > get_instance_base_type ( ) ;
bool valid_type = ClassDB : : is_parent_class ( ibt , " ResourceFormatLoader " ) ;
2023-11-23 15:19:24 +00:00
ERR_FAIL_COND_V_MSG ( ! valid_type , false , vformat ( " Failed to add a custom resource loader, script '%s' does not inherit 'ResourceFormatLoader'. " , script_path ) ) ;
2018-06-11 00:59:53 +00:00
2021-06-17 22:03:09 +00:00
Object * obj = ClassDB : : instantiate ( ibt ) ;
2023-11-23 15:19:24 +00:00
ERR_FAIL_NULL_V_MSG ( obj , false , vformat ( " Failed to add a custom resource loader, cannot instantiate '%s'. " , ibt ) ) ;
2018-06-11 00:59:53 +00:00
2020-11-15 14:14:06 +00:00
Ref < ResourceFormatLoader > crl = Object : : cast_to < ResourceFormatLoader > ( obj ) ;
2020-02-13 19:03:10 +00:00
crl - > set_script ( s ) ;
2018-06-11 00:59:53 +00:00
ResourceLoader : : add_resource_format_loader ( crl ) ;
return true ;
}
2022-04-28 20:49:10 +00:00
void ResourceLoader : : set_create_missing_resources_if_class_unavailable ( bool p_enable ) {
create_missing_resources_if_class_unavailable = p_enable ;
}
2018-06-11 00:59:53 +00:00
void ResourceLoader : : add_custom_loaders ( ) {
// Custom loaders registration exploits global class names
String custom_loader_base_class = ResourceFormatLoader : : get_class_static ( ) ;
List < StringName > global_classes ;
ScriptServer : : get_global_class_list ( & global_classes ) ;
2021-07-24 13:46:25 +00:00
for ( const StringName & class_name : global_classes ) {
2019-03-09 03:47:27 +00:00
StringName base_class = ScriptServer : : get_global_class_native_base ( class_name ) ;
2018-06-11 00:59:53 +00:00
if ( base_class = = custom_loader_base_class ) {
String path = ScriptServer : : get_global_class_path ( class_name ) ;
add_custom_resource_format_loader ( path ) ;
}
}
}
void ResourceLoader : : remove_custom_loaders ( ) {
2020-03-17 06:33:00 +00:00
Vector < Ref < ResourceFormatLoader > > custom_loaders ;
2018-06-11 00:59:53 +00:00
for ( int i = 0 ; i < loader_count ; + + i ) {
if ( loader [ i ] - > get_script_instance ( ) ) {
custom_loaders . push_back ( loader [ i ] ) ;
}
}
for ( int i = 0 ; i < custom_loaders . size ( ) ; + + i ) {
remove_resource_format_loader ( custom_loaders [ i ] ) ;
}
}
2023-03-05 00:09:18 +00:00
bool ResourceLoader : : is_cleaning_tasks ( ) {
MutexLock lock ( thread_load_mutex ) ;
return cleaning_tasks ;
}
2023-05-10 09:23:07 +00:00
void ResourceLoader : : initialize ( ) { }
2019-01-27 22:24:55 +00:00
2023-03-05 00:09:18 +00:00
void ResourceLoader : : finalize ( ) { }
2019-01-27 22:24:55 +00:00
2020-04-01 23:20:12 +00:00
ResourceLoadErrorNotify ResourceLoader : : err_notify = nullptr ;
DependencyErrorNotify ResourceLoader : : dep_err_notify = nullptr ;
2014-02-10 01:10:30 +00:00
2022-04-28 20:49:10 +00:00
bool ResourceLoader : : create_missing_resources_if_class_unavailable = false ;
2014-02-10 01:10:30 +00:00
bool ResourceLoader : : abort_on_missing_resource = true ;
2016-07-20 00:40:05 +00:00
bool ResourceLoader : : timestamp_on_load = false ;
2017-06-28 20:00:18 +00:00
2023-03-05 00:09:18 +00:00
thread_local int ResourceLoader : : load_nesting = 0 ;
2024-06-19 05:27:33 +00:00
thread_local Vector < String > ResourceLoader : : load_paths_stack ;
2024-06-25 12:12:40 +00:00
thread_local HashMap < int , HashMap < String , Ref < Resource > > > ResourceLoader : : res_ref_overrides ;
2024-09-05 07:48:13 +00:00
thread_local ResourceLoader : : ThreadLoadTask * ResourceLoader : : curr_load_task = nullptr ;
2023-03-05 00:09:18 +00:00
2024-07-18 12:54:58 +00:00
SafeBinaryMutex < ResourceLoader : : BINARY_MUTEX_TAG > & _get_res_loader_mutex ( ) {
return ResourceLoader : : thread_load_mutex ;
}
2023-02-20 18:00:26 +00:00
template < >
2024-07-18 12:54:58 +00:00
thread_local SafeBinaryMutex < ResourceLoader : : BINARY_MUTEX_TAG > : : TLSData SafeBinaryMutex < ResourceLoader : : BINARY_MUTEX_TAG > : : tls_data ( _get_res_loader_mutex ( ) ) ;
2023-03-05 00:09:18 +00:00
SafeBinaryMutex < ResourceLoader : : BINARY_MUTEX_TAG > ResourceLoader : : thread_load_mutex ;
2020-02-28 11:27:04 +00:00
HashMap < String , ResourceLoader : : ThreadLoadTask > ResourceLoader : : thread_load_tasks ;
2023-03-05 00:09:18 +00:00
bool ResourceLoader : : cleaning_tasks = false ;
HashMap < String , ResourceLoader : : LoadToken * > ResourceLoader : : user_load_tokens ;
2020-02-28 11:27:04 +00:00
2017-06-28 20:00:18 +00:00
SelfList < Resource > : : List ResourceLoader : : remapped_list ;
2020-03-17 06:33:00 +00:00
HashMap < String , Vector < String > > ResourceLoader : : translation_remaps ;
2017-12-14 18:33:54 +00:00
HashMap < String , String > ResourceLoader : : path_remaps ;
2018-10-05 02:00:02 +00:00
2020-04-01 23:20:12 +00:00
ResourceLoaderImport ResourceLoader : : import = nullptr ;