00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015 #include "modelPool.h"
00016 #include "loader.h"
00017 #include "config_pgraph.h"
00018 #include "lightMutexHolder.h"
00019
00020
00021 ModelPool *ModelPool::_global_ptr = (ModelPool *)NULL;
00022
00023
00024
00025
00026
00027
00028
00029
00030 void ModelPool::
00031 write(ostream &out) {
00032 get_ptr()->ns_list_contents(out);
00033 }
00034
00035
00036
00037
00038
00039
00040 bool ModelPool::
00041 ns_has_model(const Filename &filename) {
00042 LightMutexHolder holder(_lock);
00043 Models::const_iterator ti;
00044 ti = _models.find(filename);
00045 if (ti != _models.end() && (*ti).second != (ModelRoot *)NULL) {
00046
00047 return true;
00048 }
00049
00050 return false;
00051 }
00052
00053
00054
00055
00056
00057
00058 ModelRoot *ModelPool::
00059 ns_load_model(const Filename &filename, const LoaderOptions &options) {
00060 {
00061 LightMutexHolder holder(_lock);
00062 Models::const_iterator ti;
00063 ti = _models.find(filename);
00064 if (ti != _models.end()) {
00065
00066 return (*ti).second;
00067 }
00068 }
00069
00070 LoaderOptions new_options(options);
00071 new_options.set_flags((new_options.get_flags() | LoaderOptions::LF_no_ram_cache) &
00072 ~(LoaderOptions::LF_search | LoaderOptions::LF_report_errors));
00073
00074 Loader *model_loader = Loader::get_global_ptr();
00075 PT(PandaNode) panda_node = model_loader->load_sync(filename, new_options);
00076 PT(ModelRoot) node;
00077
00078 if (panda_node.is_null()) {
00079
00080
00081 } else {
00082 if (panda_node->is_of_type(ModelRoot::get_class_type())) {
00083 node = DCAST(ModelRoot, panda_node);
00084
00085 } else {
00086
00087 node = new ModelRoot(filename);
00088 node->add_child(panda_node);
00089 }
00090 node->set_fullpath(filename);
00091 }
00092
00093 {
00094 LightMutexHolder holder(_lock);
00095
00096
00097
00098 Models::const_iterator ti;
00099 ti = _models.find(filename);
00100 if (ti != _models.end()) {
00101
00102 return (*ti).second;
00103 }
00104
00105 _models[filename] = node;
00106 }
00107
00108 return node;
00109 }
00110
00111
00112
00113
00114
00115
00116 void ModelPool::
00117 ns_add_model(const Filename &filename, ModelRoot *model) {
00118 LightMutexHolder holder(_lock);
00119
00120 _models[filename] = model;
00121 }
00122
00123
00124
00125
00126
00127
00128 void ModelPool::
00129 ns_release_model(const Filename &filename) {
00130 LightMutexHolder holder(_lock);
00131 Models::iterator ti;
00132 ti = _models.find(filename);
00133 if (ti != _models.end()) {
00134 _models.erase(ti);
00135 }
00136 }
00137
00138
00139
00140
00141
00142
00143 void ModelPool::
00144 ns_add_model(ModelRoot *model) {
00145 LightMutexHolder holder(_lock);
00146
00147 _models[model->get_fullpath()] = model;
00148 }
00149
00150
00151
00152
00153
00154
00155 void ModelPool::
00156 ns_release_model(ModelRoot *model) {
00157 LightMutexHolder holder(_lock);
00158 Models::iterator ti;
00159 ti = _models.find(model->get_fullpath());
00160 if (ti != _models.end()) {
00161 _models.erase(ti);
00162 }
00163 }
00164
00165
00166
00167
00168
00169
00170 void ModelPool::
00171 ns_release_all_models() {
00172 LightMutexHolder holder(_lock);
00173 _models.clear();
00174 }
00175
00176
00177
00178
00179
00180
00181 int ModelPool::
00182 ns_garbage_collect() {
00183 LightMutexHolder holder(_lock);
00184
00185 int num_released = 0;
00186 Models new_set;
00187
00188 Models::iterator ti;
00189 for (ti = _models.begin(); ti != _models.end(); ++ti) {
00190 ModelRoot *node = (*ti).second;
00191 if (node == (ModelRoot *)NULL ||
00192 node->get_model_ref_count() == 1) {
00193 if (loader_cat.is_debug()) {
00194 loader_cat.debug()
00195 << "Releasing " << (*ti).first << "\n";
00196 }
00197 ++num_released;
00198 } else {
00199 new_set.insert(new_set.end(), *ti);
00200 }
00201 }
00202
00203 _models.swap(new_set);
00204 return num_released;
00205 }
00206
00207
00208
00209
00210
00211
00212 void ModelPool::
00213 ns_list_contents(ostream &out) const {
00214 LightMutexHolder holder(_lock);
00215
00216 out << "model pool contents:\n";
00217
00218 Models::const_iterator ti;
00219 int num_models = 0;
00220 for (ti = _models.begin(); ti != _models.end(); ++ti) {
00221 if ((*ti).second != NULL) {
00222 ++num_models;
00223 out << (*ti).first << "\n"
00224 << " (count = " << (*ti).second->get_model_ref_count()
00225 << ")\n";
00226 }
00227 }
00228
00229 out << "total number of models: " << num_models << " (plus "
00230 << _models.size() - num_models << " entries for nonexistent files)\n";
00231 }
00232
00233
00234
00235
00236
00237
00238
00239 ModelPool *ModelPool::
00240 get_ptr() {
00241 if (_global_ptr == (ModelPool *)NULL) {
00242 _global_ptr = new ModelPool;
00243 }
00244 return _global_ptr;
00245 }