gobj: fix TexturePool ambiguity when loading tex with other settings

Previously, if you called load_texture a second time with different parameters (such as an alpha filename), you would still get the old texture.
This commit is contained in:
rdb 2018-07-31 14:56:04 +02:00
parent 30721ba33b
commit 0cd69c8748
3 changed files with 160 additions and 112 deletions

View File

@ -275,3 +275,23 @@ PT(Texture) TexturePool::
make_texture(const std::string &extension) { make_texture(const std::string &extension) {
return get_global_ptr()->ns_make_texture(extension); return get_global_ptr()->ns_make_texture(extension);
} }
/**
* Defines relative ordering between LookupKey instances.
*/
INLINE bool TexturePool::LookupKey::
operator < (const LookupKey &other) const {
if (_fullpath != other._fullpath) {
return _fullpath < other._fullpath;
}
if (_alpha_fullpath != other._alpha_fullpath) {
return _alpha_fullpath < other._alpha_fullpath;
}
if (_primary_file_num_channels != other._primary_file_num_channels) {
return _primary_file_num_channels < other._primary_file_num_channels;
}
if (_alpha_file_channel != other._alpha_file_channel) {
return _alpha_file_channel < other._alpha_file_channel;
}
return _texture_type < other._texture_type;
}

View File

@ -177,16 +177,23 @@ bool TexturePool::
ns_has_texture(const Filename &orig_filename) { ns_has_texture(const Filename &orig_filename) {
MutexHolder holder(_lock); MutexHolder holder(_lock);
Filename filename; LookupKey key;
resolve_filename(filename, orig_filename, false, LoaderOptions()); resolve_filename(key._fullpath, orig_filename, false, LoaderOptions());
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
return true; return true;
} }
// It might still have been loaded with non-standard settings.
for (ti = _textures.begin(); ti != _textures.end(); ++ti) {
if (ti->first._fullpath == key._fullpath) {
return true;
}
}
return false; return false;
} }
@ -196,13 +203,14 @@ ns_has_texture(const Filename &orig_filename) {
Texture *TexturePool:: Texture *TexturePool::
ns_load_texture(const Filename &orig_filename, int primary_file_num_channels, ns_load_texture(const Filename &orig_filename, int primary_file_num_channels,
bool read_mipmaps, const LoaderOptions &options) { bool read_mipmaps, const LoaderOptions &options) {
Filename filename; LookupKey key;
key._primary_file_num_channels = primary_file_num_channels;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
resolve_filename(filename, orig_filename, read_mipmaps, options); resolve_filename(key._fullpath, orig_filename, read_mipmaps, options);
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
Texture *tex = (*ti).second; Texture *tex = (*ti).second;
@ -222,54 +230,54 @@ ns_load_texture(const Filename &orig_filename, int primary_file_num_channels,
BamCache *cache = BamCache::get_global_ptr(); BamCache *cache = BamCache::get_global_ptr();
bool compressed_cache_record = false; bool compressed_cache_record = false;
try_load_cache(tex, cache, filename, record, compressed_cache_record, try_load_cache(tex, cache, key._fullpath, record, compressed_cache_record,
options); options);
if (tex == nullptr) { if (tex == nullptr) {
// The texture was neither in the pool, nor found in the on-disk cache; it // The texture was neither in the pool, nor found in the on-disk cache; it
// needs to be loaded from its source image(s). // needs to be loaded from its source image(s).
gobj_cat.info() gobj_cat.info()
<< "Loading texture " << filename << "\n"; << "Loading texture " << key._fullpath << "\n";
string ext = downcase(filename.get_extension()); string ext = downcase(key._fullpath.get_extension());
if (ext == "txo" || ext == "bam") { if (ext == "txo" || ext == "bam") {
// Assume this is a txo file, which might conceivably contain a movie // Assume this is a txo file, which might conceivably contain a movie
// file or some other subclass of Texture. In that case, use // file or some other subclass of Texture. In that case, use
// make_from_txo() to load it instead of read(). // make_from_txo() to load it instead of read().
VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr(); VirtualFileSystem *vfs = VirtualFileSystem::get_global_ptr();
filename.set_binary(); key._fullpath.set_binary();
PT(VirtualFile) file = vfs->get_file(filename); PT(VirtualFile) file = vfs->get_file(key._fullpath);
if (file == nullptr) { if (file == nullptr) {
// No such file. // No such file.
gobj_cat.error() gobj_cat.error()
<< "Could not find " << filename << "\n"; << "Could not find " << key._fullpath << "\n";
return nullptr; return nullptr;
} }
if (gobj_cat.is_debug()) { if (gobj_cat.is_debug()) {
gobj_cat.debug() gobj_cat.debug()
<< "Reading texture object " << filename << "\n"; << "Reading texture object " << key._fullpath << "\n";
} }
istream *in = file->open_read_file(true); istream *in = file->open_read_file(true);
tex = Texture::make_from_txo(*in, filename); tex = Texture::make_from_txo(*in, key._fullpath);
vfs->close_read_file(in); vfs->close_read_file(in);
if (tex == nullptr) { if (tex == nullptr) {
return nullptr; return nullptr;
} }
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->clear_alpha_fullpath(); tex->clear_alpha_fullpath();
tex->set_keep_ram_image(false); tex->set_keep_ram_image(false);
} else { } else {
// Read it the conventional way. // Read it the conventional way.
tex = ns_make_texture(ext); tex = ns_make_texture(ext);
if (!tex->read(filename, Filename(), primary_file_num_channels, 0, if (!tex->read(key._fullpath, Filename(), primary_file_num_channels, 0,
0, 0, false, read_mipmaps, record, options)) { 0, 0, false, read_mipmaps, record, options)) {
// This texture was not found or could not be read. // This texture was not found or could not be read.
report_texture_unreadable(filename); report_texture_unreadable(key._fullpath);
return nullptr; return nullptr;
} }
} }
@ -304,8 +312,8 @@ ns_load_texture(const Filename &orig_filename, int primary_file_num_channels,
// Set the original filename, before we searched along the path. // Set the original filename, before we searched along the path.
nassertr(tex != nullptr, nullptr); nassertr(tex != nullptr, nullptr);
tex->set_filename(orig_filename); tex->set_filename(orig_filename);
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->_texture_pool_key = filename; tex->_texture_pool_key = key._fullpath;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
@ -313,7 +321,7 @@ ns_load_texture(const Filename &orig_filename, int primary_file_num_channels,
// Now look again--someone may have just loaded this texture in another // Now look again--someone may have just loaded this texture in another
// thread. // thread.
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
Texture *tex = (*ti).second; Texture *tex = (*ti).second;
@ -321,7 +329,7 @@ ns_load_texture(const Filename &orig_filename, int primary_file_num_channels,
return tex; return tex;
} }
_textures[filename] = tex; _textures[std::move(key)] = tex;
} }
if (store_record && tex->is_cacheable()) { if (store_record && tex->is_cacheable()) {
@ -357,16 +365,16 @@ ns_load_texture(const Filename &orig_filename,
read_mipmaps, options); read_mipmaps, options);
} }
Filename filename; LookupKey key;
Filename alpha_filename; key._primary_file_num_channels = primary_file_num_channels;
key._alpha_file_channel = alpha_file_channel;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
resolve_filename(filename, orig_filename, read_mipmaps, options); resolve_filename(key._fullpath, orig_filename, read_mipmaps, options);
resolve_filename(alpha_filename, orig_alpha_filename, read_mipmaps, options); resolve_filename(key._alpha_fullpath, orig_alpha_filename, read_mipmaps, options);
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
Texture *tex = (*ti).second; Texture *tex = (*ti).second;
@ -380,26 +388,26 @@ ns_load_texture(const Filename &orig_filename,
bool store_record = false; bool store_record = false;
// Can one of our texture filters supply the texture? // Can one of our texture filters supply the texture?
tex = pre_load(orig_filename, alpha_filename, primary_file_num_channels, tex = pre_load(orig_filename, orig_alpha_filename, primary_file_num_channels,
alpha_file_channel, read_mipmaps, options); alpha_file_channel, read_mipmaps, options);
BamCache *cache = BamCache::get_global_ptr(); BamCache *cache = BamCache::get_global_ptr();
bool compressed_cache_record = false; bool compressed_cache_record = false;
try_load_cache(tex, cache, filename, record, compressed_cache_record, try_load_cache(tex, cache, key._fullpath, record, compressed_cache_record,
options); options);
if (tex == nullptr) { if (tex == nullptr) {
// The texture was neither in the pool, nor found in the on-disk cache; it // The texture was neither in the pool, nor found in the on-disk cache; it
// needs to be loaded from its source image(s). // needs to be loaded from its source image(s).
gobj_cat.info() gobj_cat.info()
<< "Loading texture " << filename << " and alpha component " << "Loading texture " << key._fullpath << " and alpha component "
<< alpha_filename << std::endl; << key._alpha_fullpath << std::endl;
tex = ns_make_texture(filename.get_extension()); tex = ns_make_texture(key._fullpath.get_extension());
if (!tex->read(filename, alpha_filename, primary_file_num_channels, if (!tex->read(key._fullpath, key._alpha_fullpath, primary_file_num_channels,
alpha_file_channel, 0, 0, false, read_mipmaps, nullptr, alpha_file_channel, 0, 0, false, read_mipmaps, nullptr,
options)) { options)) {
// This texture was not found or could not be read. // This texture was not found or could not be read.
report_texture_unreadable(filename); report_texture_unreadable(key._fullpath);
return nullptr; return nullptr;
} }
@ -433,17 +441,17 @@ ns_load_texture(const Filename &orig_filename,
// Set the original filenames, before we searched along the path. // Set the original filenames, before we searched along the path.
nassertr(tex != nullptr, nullptr); nassertr(tex != nullptr, nullptr);
tex->set_filename(orig_filename); tex->set_filename(orig_filename);
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->set_alpha_filename(orig_alpha_filename); tex->set_alpha_filename(orig_alpha_filename);
tex->set_alpha_fullpath(alpha_filename); tex->set_alpha_fullpath(key._alpha_fullpath);
tex->_texture_pool_key = filename; tex->_texture_pool_key = key._fullpath;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
// Now look again. // Now look again.
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
Texture *tex = (*ti).second; Texture *tex = (*ti).second;
@ -451,7 +459,7 @@ ns_load_texture(const Filename &orig_filename,
return tex; return tex;
} }
_textures[filename] = tex; _textures[std::move(key)] = tex;
} }
if (store_record && tex->is_cacheable()) { if (store_record && tex->is_cacheable()) {
@ -482,18 +490,19 @@ ns_load_3d_texture(const Filename &filename_pattern,
Filename orig_filename(filename_pattern); Filename orig_filename(filename_pattern);
orig_filename.set_pattern(true); orig_filename.set_pattern(true);
Filename filename; LookupKey key;
key._texture_type = Texture::TT_3d_texture;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
resolve_filename(filename, orig_filename, read_mipmaps, options); resolve_filename(key._fullpath, orig_filename, read_mipmaps, options);
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
if ((*ti).second->get_texture_type() == Texture::TT_3d_texture) { // This texture was previously loaded.
// This texture was previously loaded, as a 3d texture Texture *tex = (*ti).second;
return (*ti).second; nassertr(!tex->get_fullpath().empty(), tex);
} return tex;
} }
} }
@ -503,7 +512,7 @@ ns_load_3d_texture(const Filename &filename_pattern,
BamCache *cache = BamCache::get_global_ptr(); BamCache *cache = BamCache::get_global_ptr();
bool compressed_cache_record = false; bool compressed_cache_record = false;
try_load_cache(tex, cache, filename, record, compressed_cache_record, try_load_cache(tex, cache, key._fullpath, record, compressed_cache_record,
options); options);
if (tex == nullptr || if (tex == nullptr ||
@ -511,12 +520,12 @@ ns_load_3d_texture(const Filename &filename_pattern,
// The texture was neither in the pool, nor found in the on-disk cache; it // The texture was neither in the pool, nor found in the on-disk cache; it
// needs to be loaded from its source image(s). // needs to be loaded from its source image(s).
gobj_cat.info() gobj_cat.info()
<< "Loading 3-d texture " << filename << "\n"; << "Loading 3-d texture " << key._fullpath << "\n";
tex = ns_make_texture(filename.get_extension()); tex = ns_make_texture(key._fullpath.get_extension());
tex->setup_3d_texture(); tex->setup_3d_texture();
if (!tex->read(filename, 0, 0, true, read_mipmaps, options)) { if (!tex->read(key._fullpath, 0, 0, true, read_mipmaps, options)) {
// This texture was not found or could not be read. // This texture was not found or could not be read.
report_texture_unreadable(filename); report_texture_unreadable(key._fullpath);
return nullptr; return nullptr;
} }
store_record = (record != nullptr); store_record = (record != nullptr);
@ -545,23 +554,23 @@ ns_load_3d_texture(const Filename &filename_pattern,
// Set the original filename, before we searched along the path. // Set the original filename, before we searched along the path.
nassertr(tex != nullptr, nullptr); nassertr(tex != nullptr, nullptr);
tex->set_filename(filename_pattern); tex->set_filename(filename_pattern);
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->_texture_pool_key = filename; tex->_texture_pool_key = key._fullpath;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
// Now look again. // Now look again.
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
if ((*ti).second->get_texture_type() == Texture::TT_3d_texture) { // This texture was previously loaded.
// This texture was previously loaded, as a 3d texture Texture *tex = (*ti).second;
return (*ti).second; nassertr(!tex->get_fullpath().empty(), tex);
} return tex;
} }
_textures[filename] = tex; _textures[std::move(key)] = tex;
} }
if (store_record && tex->is_cacheable()) { if (store_record && tex->is_cacheable()) {
@ -583,21 +592,19 @@ ns_load_2d_texture_array(const Filename &filename_pattern,
Filename orig_filename(filename_pattern); Filename orig_filename(filename_pattern);
orig_filename.set_pattern(true); orig_filename.set_pattern(true);
Filename filename; LookupKey key;
Filename unique_filename; //differentiate 3d-textures from 2d-texture arrays key._texture_type = Texture::TT_2d_texture_array;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
resolve_filename(filename, orig_filename, read_mipmaps, options); resolve_filename(key._fullpath, orig_filename, read_mipmaps, options);
// Differentiate from preloaded 3d textures
unique_filename = filename + ".2DARRAY";
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(unique_filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
if ((*ti).second->get_texture_type() == Texture::TT_2d_texture_array) { // This texture was previously loaded.
// This texture was previously loaded, as a 2d texture array Texture *tex = (*ti).second;
return (*ti).second; nassertr(!tex->get_fullpath().empty(), tex);
} return tex;
} }
} }
@ -607,7 +614,7 @@ ns_load_2d_texture_array(const Filename &filename_pattern,
BamCache *cache = BamCache::get_global_ptr(); BamCache *cache = BamCache::get_global_ptr();
bool compressed_cache_record = false; bool compressed_cache_record = false;
try_load_cache(tex, cache, filename, record, compressed_cache_record, try_load_cache(tex, cache, key._fullpath, record, compressed_cache_record,
options); options);
if (tex == nullptr || if (tex == nullptr ||
@ -615,12 +622,12 @@ ns_load_2d_texture_array(const Filename &filename_pattern,
// The texture was neither in the pool, nor found in the on-disk cache; it // The texture was neither in the pool, nor found in the on-disk cache; it
// needs to be loaded from its source image(s). // needs to be loaded from its source image(s).
gobj_cat.info() gobj_cat.info()
<< "Loading 2-d texture array " << filename << "\n"; << "Loading 2-d texture array " << key._fullpath << "\n";
tex = ns_make_texture(filename.get_extension()); tex = ns_make_texture(key._fullpath.get_extension());
tex->setup_2d_texture_array(); tex->setup_2d_texture_array();
if (!tex->read(filename, 0, 0, true, read_mipmaps, options)) { if (!tex->read(key._fullpath, 0, 0, true, read_mipmaps, options)) {
// This texture was not found or could not be read. // This texture was not found or could not be read.
report_texture_unreadable(filename); report_texture_unreadable(key._fullpath);
return nullptr; return nullptr;
} }
store_record = (record != nullptr); store_record = (record != nullptr);
@ -649,23 +656,23 @@ ns_load_2d_texture_array(const Filename &filename_pattern,
// Set the original filename, before we searched along the path. // Set the original filename, before we searched along the path.
nassertr(tex != nullptr, nullptr); nassertr(tex != nullptr, nullptr);
tex->set_filename(filename_pattern); tex->set_filename(filename_pattern);
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->_texture_pool_key = unique_filename; tex->_texture_pool_key = key._fullpath;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
// Now look again. // Now look again.
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(unique_filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
if ((*ti).second->get_texture_type() == Texture::TT_2d_texture_array) { // This texture was previously loaded.
// This texture was previously loaded, as a 2d texture array Texture *tex = (*ti).second;
return (*ti).second; nassertr(!tex->get_fullpath().empty(), tex);
} return tex;
} }
_textures[unique_filename] = tex; _textures[std::move(key)] = tex;
} }
if (store_record && tex->is_cacheable()) { if (store_record && tex->is_cacheable()) {
@ -687,16 +694,19 @@ ns_load_cube_map(const Filename &filename_pattern, bool read_mipmaps,
Filename orig_filename(filename_pattern); Filename orig_filename(filename_pattern);
orig_filename.set_pattern(true); orig_filename.set_pattern(true);
Filename filename; LookupKey key;
key._texture_type = Texture::TT_cube_map;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
resolve_filename(filename, orig_filename, read_mipmaps, options); resolve_filename(key._fullpath, orig_filename, read_mipmaps, options);
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
return (*ti).second; Texture *tex = (*ti).second;
nassertr(!tex->get_fullpath().empty(), tex);
return tex;
} }
} }
@ -706,7 +716,7 @@ ns_load_cube_map(const Filename &filename_pattern, bool read_mipmaps,
BamCache *cache = BamCache::get_global_ptr(); BamCache *cache = BamCache::get_global_ptr();
bool compressed_cache_record = false; bool compressed_cache_record = false;
try_load_cache(tex, cache, filename, record, compressed_cache_record, try_load_cache(tex, cache, key._fullpath, record, compressed_cache_record,
options); options);
if (tex == nullptr || if (tex == nullptr ||
@ -714,12 +724,12 @@ ns_load_cube_map(const Filename &filename_pattern, bool read_mipmaps,
// The texture was neither in the pool, nor found in the on-disk cache; it // The texture was neither in the pool, nor found in the on-disk cache; it
// needs to be loaded from its source image(s). // needs to be loaded from its source image(s).
gobj_cat.info() gobj_cat.info()
<< "Loading cube map texture " << filename << "\n"; << "Loading cube map texture " << key._fullpath << "\n";
tex = ns_make_texture(filename.get_extension()); tex = ns_make_texture(key._fullpath.get_extension());
tex->setup_cube_map(); tex->setup_cube_map();
if (!tex->read(filename, 0, 0, true, read_mipmaps, options)) { if (!tex->read(key._fullpath, 0, 0, true, read_mipmaps, options)) {
// This texture was not found or could not be read. // This texture was not found or could not be read.
report_texture_unreadable(filename); report_texture_unreadable(key._fullpath);
return nullptr; return nullptr;
} }
store_record = (record != nullptr); store_record = (record != nullptr);
@ -748,21 +758,23 @@ ns_load_cube_map(const Filename &filename_pattern, bool read_mipmaps,
// Set the original filename, before we searched along the path. // Set the original filename, before we searched along the path.
nassertr(tex != nullptr, nullptr); nassertr(tex != nullptr, nullptr);
tex->set_filename(filename_pattern); tex->set_filename(filename_pattern);
tex->set_fullpath(filename); tex->set_fullpath(key._fullpath);
tex->_texture_pool_key = filename; tex->_texture_pool_key = key._fullpath;
{ {
MutexHolder holder(_lock); MutexHolder holder(_lock);
// Now look again. // Now look again.
Textures::const_iterator ti; Textures::const_iterator ti;
ti = _textures.find(filename); ti = _textures.find(key);
if (ti != _textures.end()) { if (ti != _textures.end()) {
// This texture was previously loaded. // This texture was previously loaded.
return (*ti).second; Texture *tex = (*ti).second;
nassertr(!tex->get_fullpath().empty(), tex);
return tex;
} }
_textures[filename] = tex; _textures[std::move(key)] = tex;
} }
if (store_record && tex->is_cacheable()) { if (store_record && tex->is_cacheable()) {
@ -819,15 +831,22 @@ ns_add_texture(Texture *tex) {
if (!tex->_texture_pool_key.empty()) { if (!tex->_texture_pool_key.empty()) {
ns_release_texture(tex); ns_release_texture(tex);
} }
string filename = tex->get_fullpath();
if (filename.empty()) { Texture::CDReader tex_cdata(tex->_cycler);
if (tex_cdata->_fullpath.empty()) {
gobj_cat.error() << "Attempt to call add_texture() on an unnamed texture.\n"; gobj_cat.error() << "Attempt to call add_texture() on an unnamed texture.\n";
return;
} }
LookupKey key;
key._fullpath = tex_cdata->_fullpath;
key._alpha_fullpath = tex_cdata->_alpha_fullpath;
key._alpha_file_channel = tex_cdata->_alpha_file_channel;
key._texture_type = tex_cdata->_texture_type;
// We blow away whatever texture was there previously, if any. // We blow away whatever texture was there previously, if any.
tex->_texture_pool_key = filename; tex->_texture_pool_key = key._fullpath;
_textures[filename] = tex; _textures[key] = tex;
nassertv(!tex->get_fullpath().empty());
} }
/** /**
@ -837,13 +856,13 @@ void TexturePool::
ns_release_texture(Texture *tex) { ns_release_texture(Texture *tex) {
MutexHolder holder(_lock); MutexHolder holder(_lock);
if (!tex->_texture_pool_key.empty()) { Textures::iterator ti;
Textures::iterator ti; for (ti = _textures.begin(); ti != _textures.end(); ++ti) {
ti = _textures.find(tex->_texture_pool_key); if (tex == (*ti).second) {
if (ti != _textures.end() && (*ti).second == tex) {
_textures.erase(ti); _textures.erase(ti);
tex->_texture_pool_key = string();
break;
} }
tex->_texture_pool_key = string();
} }
// Blow away the cache of resolved relative filenames. // Blow away the cache of resolved relative filenames.
@ -886,7 +905,7 @@ ns_garbage_collect() {
if (tex->get_ref_count() == 1) { if (tex->get_ref_count() == 1) {
if (gobj_cat.is_debug()) { if (gobj_cat.is_debug()) {
gobj_cat.debug() gobj_cat.debug()
<< "Releasing " << (*ti).first << "\n"; << "Releasing " << (*ti).first._fullpath << "\n";
} }
++num_released; ++num_released;
tex->_texture_pool_key = string(); tex->_texture_pool_key = string();
@ -927,14 +946,14 @@ ns_list_contents(ostream &out) const {
total_ram_size = 0; total_ram_size = 0;
for (ti = _textures.begin(); ti != _textures.end(); ++ti) { for (ti = _textures.begin(); ti != _textures.end(); ++ti) {
Texture *tex = (*ti).second; Texture *tex = (*ti).second;
out << (*ti).first << "\n"; out << (*ti).first._fullpath << "\n";
out << " (count = " << tex->get_ref_count() out << " (count = " << tex->get_ref_count()
<< ", ram = " << tex->get_ram_image_size() << ", ram = " << tex->get_ram_image_size()
<< ", size = " << tex->get_ram_page_size() << ", size = " << tex->get_ram_page_size()
<< ", w = " << tex->get_x_size() << ", w = " << tex->get_x_size()
<< ", h = " << tex->get_y_size() << ", h = " << tex->get_y_size()
<< ")\n"; << ")\n";
nassertv(tex->_texture_pool_key == (*ti).first); nassertv(tex->_texture_pool_key == (*ti).first._fullpath);
total_ram_size += tex->get_ram_image_size(); total_ram_size += tex->get_ram_image_size();
total_size += tex->get_ram_page_size(); total_size += tex->get_ram_page_size();
} }

View File

@ -149,8 +149,17 @@ private:
static TexturePool *_global_ptr; static TexturePool *_global_ptr;
Mutex _lock; Mutex _lock;
typedef pmap<Filename, PT(Texture)> Textures; struct LookupKey {
Textures _textures; // indexed by fullpath Filename _fullpath;
Filename _alpha_fullpath;
int _primary_file_num_channels = 0;
int _alpha_file_channel = 0;
Texture::TextureType _texture_type = Texture::TT_2d_texture;
INLINE bool operator < (const LookupKey &other) const;
};
typedef pmap<LookupKey, PT(Texture)> Textures;
Textures _textures;
typedef pmap<Filename, Filename> RelpathLookup; typedef pmap<Filename, Filename> RelpathLookup;
RelpathLookup _relpath_lookup; RelpathLookup _relpath_lookup;