editor_file_system.cpp 31 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394
  1. /*************************************************************************/
  2. /* editor_file_system.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* http://www.godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2016 Juan Linietsky, Ariel Manzur. */
  9. /* */
  10. /* Permission is hereby granted, free of charge, to any person obtaining */
  11. /* a copy of this software and associated documentation files (the */
  12. /* "Software"), to deal in the Software without restriction, including */
  13. /* without limitation the rights to use, copy, modify, merge, publish, */
  14. /* distribute, sublicense, and/or sell copies of the Software, and to */
  15. /* permit persons to whom the Software is furnished to do so, subject to */
  16. /* the following conditions: */
  17. /* */
  18. /* The above copyright notice and this permission notice shall be */
  19. /* included in all copies or substantial portions of the Software. */
  20. /* */
  21. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  22. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  23. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  24. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  25. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  26. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  27. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  28. /*************************************************************************/
  29. #include "editor_file_system.h"
  30. #include "globals.h"
  31. #include "io/resource_loader.h"
  32. #include "os/os.h"
  33. #include "os/file_access.h"
  34. #include "editor_node.h"
  35. #include "io/resource_saver.h"
  36. #include "editor_settings.h"
  37. #include "editor_resource_preview.h"
  38. EditorFileSystem *EditorFileSystem::singleton=NULL;
  39. void EditorFileSystemDirectory::sort_files() {
  40. files.sort_custom<FileInfoSort>();
  41. }
  42. int EditorFileSystemDirectory::find_file_index(const String& p_file) const {
  43. for(int i=0;i<files.size();i++) {
  44. if (files[i]->file==p_file)
  45. return i;
  46. }
  47. return -1;
  48. }
  49. int EditorFileSystemDirectory::find_dir_index(const String& p_dir) const{
  50. for(int i=0;i<subdirs.size();i++) {
  51. if (subdirs[i]->name==p_dir)
  52. return i;
  53. }
  54. return -1;
  55. }
  56. int EditorFileSystemDirectory::get_subdir_count() const {
  57. return subdirs.size();
  58. }
  59. EditorFileSystemDirectory *EditorFileSystemDirectory::get_subdir(int p_idx){
  60. ERR_FAIL_INDEX_V(p_idx,subdirs.size(),NULL);
  61. return subdirs[p_idx];
  62. }
  63. int EditorFileSystemDirectory::get_file_count() const{
  64. return files.size();
  65. }
  66. String EditorFileSystemDirectory::get_file(int p_idx) const{
  67. ERR_FAIL_INDEX_V(p_idx,files.size(),"");
  68. return files[p_idx]->file;
  69. }
  70. String EditorFileSystemDirectory::get_path() const {
  71. String p;
  72. const EditorFileSystemDirectory *d=this;
  73. while(d->parent) {
  74. p=d->name+"/"+p;
  75. d=d->parent;
  76. }
  77. return "res://"+p;
  78. }
  79. String EditorFileSystemDirectory::get_file_path(int p_idx) const {
  80. String file = get_file(p_idx);
  81. const EditorFileSystemDirectory *d=this;
  82. while(d->parent) {
  83. file=d->name+"/"+file;
  84. d=d->parent;
  85. }
  86. return "res://"+file;
  87. }
  88. bool EditorFileSystemDirectory::get_file_meta(int p_idx) const {
  89. ERR_FAIL_INDEX_V(p_idx,files.size(),"");
  90. return files[p_idx]->meta.enabled;
  91. }
  92. Vector<String> EditorFileSystemDirectory::get_file_deps(int p_idx) const {
  93. ERR_FAIL_INDEX_V(p_idx,files.size(),Vector<String>());
  94. return files[p_idx]->meta.deps;
  95. }
  96. Vector<String> EditorFileSystemDirectory::get_missing_sources(int p_idx) const {
  97. ERR_FAIL_INDEX_V(p_idx,files.size(),Vector<String>());
  98. Vector<String> missing;
  99. for(int i=0;i<files[p_idx]->meta.sources.size();i++) {
  100. if (files[p_idx]->meta.sources[i].missing)
  101. missing.push_back(files[p_idx]->meta.sources[i].path);
  102. }
  103. return missing;
  104. }
  105. bool EditorFileSystemDirectory::is_missing_sources(int p_idx) const {
  106. ERR_FAIL_INDEX_V(p_idx,files.size(),false);
  107. for(int i=0;i<files[p_idx]->meta.sources.size();i++) {
  108. if (files[p_idx]->meta.sources[i].missing)
  109. return true;
  110. }
  111. return false;
  112. }
  113. bool EditorFileSystemDirectory::have_sources_changed(int p_idx) const {
  114. ERR_FAIL_INDEX_V(p_idx,files.size(),false);
  115. return files[p_idx]->meta.sources_changed;
  116. }
  117. int EditorFileSystemDirectory::get_source_count(int p_idx) const {
  118. ERR_FAIL_INDEX_V(p_idx,files.size(),0);
  119. if (!files[p_idx]->meta.enabled)
  120. return 0;
  121. return files[p_idx]->meta.sources.size();
  122. }
  123. String EditorFileSystemDirectory::get_source_file(int p_idx,int p_source) const {
  124. ERR_FAIL_INDEX_V(p_idx,files.size(),String());
  125. ERR_FAIL_INDEX_V(p_source,files[p_idx]->meta.sources.size(),String());
  126. if (!files[p_idx]->meta.enabled)
  127. return String();
  128. return files[p_idx]->meta.sources[p_source].path;
  129. }
  130. bool EditorFileSystemDirectory::is_source_file_missing(int p_idx,int p_source) const {
  131. ERR_FAIL_INDEX_V(p_idx,files.size(),false);
  132. ERR_FAIL_INDEX_V(p_source,files[p_idx]->meta.sources.size(),false);
  133. if (!files[p_idx]->meta.enabled)
  134. return false;
  135. return files[p_idx]->meta.sources[p_source].missing;
  136. }
  137. StringName EditorFileSystemDirectory::get_file_type(int p_idx) const {
  138. ERR_FAIL_INDEX_V(p_idx,files.size(),"");
  139. return files[p_idx]->type;
  140. }
  141. String EditorFileSystemDirectory::get_name() {
  142. return name;
  143. }
  144. EditorFileSystemDirectory *EditorFileSystemDirectory::get_parent() {
  145. return parent;
  146. }
  147. void EditorFileSystemDirectory::_bind_methods() {
  148. ObjectTypeDB::bind_method(_MD("get_subdir_count"),&EditorFileSystemDirectory::get_subdir_count);
  149. ObjectTypeDB::bind_method(_MD("get_subdir","idx"),&EditorFileSystemDirectory::get_subdir);
  150. ObjectTypeDB::bind_method(_MD("get_file_count"),&EditorFileSystemDirectory::get_file_count);
  151. ObjectTypeDB::bind_method(_MD("get_file","idx"),&EditorFileSystemDirectory::get_file);
  152. ObjectTypeDB::bind_method(_MD("get_file_path","idx"),&EditorFileSystemDirectory::get_file_path);
  153. ObjectTypeDB::bind_method(_MD("get_file_type","idx"),&EditorFileSystemDirectory::get_file_type);
  154. ObjectTypeDB::bind_method(_MD("is_missing_sources","idx"),&EditorFileSystemDirectory::is_missing_sources);
  155. ObjectTypeDB::bind_method(_MD("get_name"),&EditorFileSystemDirectory::get_name);
  156. ObjectTypeDB::bind_method(_MD("get_path"),&EditorFileSystemDirectory::get_path);
  157. ObjectTypeDB::bind_method(_MD("get_parent:EditorFileSystemDirectory"),&EditorFileSystemDirectory::get_parent);
  158. ObjectTypeDB::bind_method(_MD("find_file_index","name"),&EditorFileSystemDirectory::find_file_index);
  159. ObjectTypeDB::bind_method(_MD("find_dir_index","name"),&EditorFileSystemDirectory::find_dir_index);
  160. }
  161. EditorFileSystemDirectory::EditorFileSystemDirectory() {
  162. modified_time=0;
  163. parent=NULL;
  164. }
  165. EditorFileSystemDirectory::~EditorFileSystemDirectory() {
  166. for(int i=0;i<files.size();i++) {
  167. memdelete(files[i]);
  168. }
  169. for(int i=0;i<subdirs.size();i++) {
  170. memdelete(subdirs[i]);
  171. }
  172. }
  173. EditorFileSystemDirectory::ImportMeta EditorFileSystem::_get_meta(const String& p_path) {
  174. Ref<ResourceImportMetadata> imd = ResourceLoader::load_import_metadata(p_path);
  175. EditorFileSystemDirectory::ImportMeta m;
  176. if (imd.is_null()) {
  177. m.enabled=false;
  178. m.sources_changed=false;
  179. } else {
  180. m.enabled=true;
  181. m.sources_changed=false;
  182. for(int i=0;i<imd->get_source_count();i++) {
  183. EditorFileSystemDirectory::ImportMeta::Source s;
  184. s.path=imd->get_source_path(i);
  185. s.md5=imd->get_source_md5(i);
  186. s.modified_time=0;
  187. s.missing=false;
  188. m.sources.push_back(s);
  189. }
  190. m.import_editor=imd->get_editor();
  191. }
  192. List<String> deps;
  193. ResourceLoader::get_dependencies(p_path,&deps);
  194. for(List<String>::Element *E=deps.front();E;E=E->next()) {
  195. m.deps.push_back(E->get());
  196. }
  197. return m;
  198. }
  199. void EditorFileSystem::_scan_filesystem() {
  200. ERR_FAIL_COND(!scanning || new_filesystem);
  201. //read .fscache
  202. String cpath;
  203. sources_changed.clear();
  204. file_cache.clear();
  205. String project=Globals::get_singleton()->get_resource_path();
  206. String fscache = EditorSettings::get_singleton()->get_project_settings_path().plus_file("filesystem_cache");
  207. FileAccess *f =FileAccess::open(fscache,FileAccess::READ);
  208. if (f) {
  209. //read the disk cache
  210. while(!f->eof_reached()) {
  211. String l = f->get_line().strip_edges();
  212. if (l==String())
  213. continue;
  214. if (l.begins_with("::")) {
  215. Vector<String> split = l.split("::");
  216. ERR_CONTINUE( split.size() != 3);
  217. String name = split[1];
  218. cpath=name;
  219. } else {
  220. Vector<String> split = l.split("::");
  221. ERR_CONTINUE( split.size() != 5);
  222. String name = split[0];
  223. String file;
  224. file=name;
  225. name=cpath.plus_file(name);
  226. FileCache fc;
  227. fc.type=split[1];
  228. fc.modification_time=split[2].to_int64();
  229. String meta = split[3].strip_edges();
  230. fc.meta.enabled=false;
  231. if (meta.find("<>")!=-1){
  232. Vector<String> spl = meta.split("<>");
  233. int sc = spl.size()-1;
  234. if (sc%3==0){
  235. fc.meta.enabled=true;
  236. fc.meta.import_editor=spl[0];
  237. fc.meta.sources.resize(sc/3);
  238. for(int i=0;i<fc.meta.sources.size();i++) {
  239. fc.meta.sources[i].path=spl[1+i*3+0];
  240. fc.meta.sources[i].md5=spl[1+i*3+1];
  241. fc.meta.sources[i].modified_time=spl[1+i*3+2].to_int64();
  242. }
  243. }
  244. }
  245. String deps = split[4].strip_edges();
  246. if (deps.length()) {
  247. Vector<String> dp = deps.split("<>");
  248. for(int i=0;i<dp.size();i++) {
  249. String path=dp[i];
  250. fc.meta.deps.push_back(path);
  251. }
  252. }
  253. file_cache[name]=fc;
  254. }
  255. }
  256. f->close();
  257. memdelete(f);
  258. }
  259. EditorProgressBG scan_progress("efs","ScanFS",1000);
  260. ScanProgress sp;
  261. sp.low=0;
  262. sp.hi=1;
  263. sp.progress=&scan_progress;
  264. new_filesystem = memnew( EditorFileSystemDirectory );
  265. new_filesystem->parent=NULL;
  266. DirAccess *d = DirAccess::create(DirAccess::ACCESS_RESOURCES);
  267. d->change_dir("res://");
  268. _scan_new_dir(new_filesystem,d,sp);
  269. file_cache.clear(); //clear caches, no longer needed
  270. memdelete(d);
  271. //save back the findings
  272. // String fscache = EditorSettings::get_singleton()->get_project_settings_path().plus_file("file_cache");
  273. f=FileAccess::open(fscache,FileAccess::WRITE);
  274. _save_filesystem_cache(new_filesystem,f);
  275. f->close();
  276. memdelete(f);
  277. scanning=false;
  278. }
  279. void EditorFileSystem::_thread_func(void *_userdata) {
  280. EditorFileSystem *sd = (EditorFileSystem*)_userdata;
  281. sd->_scan_filesystem();
  282. }
  283. bool EditorFileSystem::_update_scan_actions() {
  284. sources_changed.clear();
  285. bool fs_changed=false;
  286. for (List<ItemAction>::Element *E=scan_actions.front();E;E=E->next()) {
  287. ItemAction&ia = E->get();
  288. switch(ia.action) {
  289. case ItemAction::ACTION_NONE: {
  290. } break;
  291. case ItemAction::ACTION_DIR_ADD: {
  292. //print_line("*ACTION ADD DIR: "+ia.new_dir->get_name());
  293. int idx=0;
  294. for(int i=0;i<ia.dir->subdirs.size();i++) {
  295. if (ia.new_dir->name<ia.dir->subdirs[i]->name)
  296. break;
  297. idx++;
  298. }
  299. if (idx==ia.dir->subdirs.size()) {
  300. ia.dir->subdirs.push_back(ia.new_dir);
  301. } else {
  302. ia.dir->subdirs.insert(idx,ia.new_dir);
  303. }
  304. fs_changed=true;
  305. } break;
  306. case ItemAction::ACTION_DIR_REMOVE: {
  307. ERR_CONTINUE(!ia.dir->parent);
  308. //print_line("*ACTION REMOVE DIR: "+ia.dir->get_name());
  309. ia.dir->parent->subdirs.erase(ia.dir);
  310. memdelete( ia.dir );
  311. fs_changed=true;
  312. } break;
  313. case ItemAction::ACTION_FILE_ADD: {
  314. int idx=0;
  315. for(int i=0;i<ia.dir->files.size();i++) {
  316. if (ia.new_file->file<ia.dir->files[i]->file)
  317. break;
  318. idx++;
  319. }
  320. if (idx==ia.dir->files.size()) {
  321. ia.dir->files.push_back(ia.new_file);
  322. } else {
  323. ia.dir->files.insert(idx,ia.new_file);
  324. }
  325. fs_changed=true;
  326. //print_line("*ACTION ADD FILE: "+ia.new_file->file);
  327. } break;
  328. case ItemAction::ACTION_FILE_REMOVE: {
  329. int idx = ia.dir->find_file_index(ia.file);
  330. ERR_CONTINUE(idx==-1);
  331. memdelete( ia.dir->files[idx] );
  332. ia.dir->files.remove(idx);
  333. fs_changed=true;
  334. //print_line("*ACTION REMOVE FILE: "+ia.file);
  335. } break;
  336. case ItemAction::ACTION_FILE_SOURCES_CHANGED: {
  337. int idx = ia.dir->find_file_index(ia.file);
  338. ERR_CONTINUE(idx==-1);
  339. String full_path = ia.dir->get_file_path(idx);
  340. sources_changed.push_back(full_path);
  341. } break;
  342. }
  343. }
  344. scan_actions.clear();
  345. return fs_changed;
  346. }
  347. void EditorFileSystem::scan() {
  348. if (bool(Globals::get_singleton()->get("debug/disable_scan")))
  349. return;
  350. if (scanning || scanning_sources|| thread)
  351. return;
  352. abort_scan=false;
  353. if (!use_threads) {
  354. scanning=true;
  355. scan_total=0;
  356. _scan_filesystem();
  357. if (filesystem)
  358. memdelete(filesystem);
  359. // file_type_cache.clear();
  360. filesystem=new_filesystem;
  361. new_filesystem=NULL;
  362. _update_scan_actions();
  363. scanning=false;
  364. emit_signal("filesystem_changed");
  365. emit_signal("sources_changed",sources_changed.size()>0);
  366. } else {
  367. ERR_FAIL_COND(thread);
  368. set_process(true);
  369. Thread::Settings s;
  370. scanning=true;
  371. scan_total=0;
  372. s.priority=Thread::PRIORITY_LOW;
  373. thread = Thread::create(_thread_func,this,s);
  374. //tree->hide();
  375. //progress->show();
  376. }
  377. }
  378. bool EditorFileSystem::_check_meta_sources(EditorFileSystemDirectory::ImportMeta & p_meta) {
  379. if (p_meta.enabled) {
  380. for(int j=0;j<p_meta.sources.size();j++) {
  381. String src = EditorImportPlugin::expand_source_path(p_meta.sources[j].path);
  382. if (!FileAccess::exists(src)) {
  383. p_meta.sources[j].missing=true;
  384. continue;
  385. }
  386. p_meta.sources[j].missing=false;
  387. uint64_t mt = FileAccess::get_modified_time(src);
  388. if (mt!=p_meta.sources[j].modified_time) {
  389. //scan
  390. String md5 = FileAccess::get_md5(src);
  391. //print_line("checking: "+src);
  392. //print_line("md5: "+md5);
  393. //print_line("vs: "+p_meta.sources[j].md5);
  394. if (md5!=p_meta.sources[j].md5) {
  395. //really changed
  396. return true;
  397. }
  398. p_meta.sources[j].modified_time=mt;
  399. }
  400. }
  401. }
  402. return false;
  403. }
  404. void EditorFileSystem::ScanProgress::update(int p_current,int p_total) const {
  405. float ratio = low + ((hi-low)/p_total)*p_current;
  406. progress->step(ratio*1000);
  407. EditorFileSystem::singleton->scan_total=ratio;
  408. }
  409. EditorFileSystem::ScanProgress EditorFileSystem::ScanProgress::get_sub(int p_current,int p_total) const{
  410. ScanProgress sp=*this;
  411. float slice = (sp.hi-sp.low)/p_total;
  412. sp.low+=slice*p_current;
  413. sp.hi=slice;
  414. return sp;
  415. }
  416. void EditorFileSystem::_scan_new_dir(EditorFileSystemDirectory *p_dir,DirAccess *da,const ScanProgress& p_progress) {
  417. List<String> dirs;
  418. List<String> files;
  419. String cd = da->get_current_dir();
  420. p_dir->modified_time = FileAccess::get_modified_time(cd);
  421. da->list_dir_begin();
  422. while (true) {
  423. bool isdir;
  424. String f = da->get_next(&isdir);
  425. if (f=="")
  426. break;
  427. if (isdir) {
  428. if (f.begins_with(".")) //ignore hidden and . / ..
  429. continue;
  430. if (FileAccess::exists(cd.plus_file(f).plus_file("engine.cfg"))) // skip if another project inside this
  431. continue;
  432. dirs.push_back(f);
  433. } else {
  434. files.push_back(f);
  435. }
  436. }
  437. da->list_dir_end();
  438. dirs.sort();
  439. files.sort();
  440. int total = dirs.size()+files.size();
  441. int idx=0;
  442. for (List<String>::Element *E=dirs.front();E;E=E->next(),idx++) {
  443. if (da->change_dir(E->get())==OK) {
  444. String d = da->get_current_dir();
  445. if (d==cd || !d.begins_with(cd)) {
  446. da->change_dir(cd); //avoid recursion
  447. } else {
  448. EditorFileSystemDirectory *efd = memnew( EditorFileSystemDirectory );
  449. efd->parent=p_dir;
  450. efd->name=E->get();
  451. _scan_new_dir(efd,da,p_progress.get_sub(idx,total));
  452. int idx=0;
  453. for(int i=0;i<p_dir->subdirs.size();i++) {
  454. if (efd->name<p_dir->subdirs[i]->name)
  455. break;
  456. idx++;
  457. }
  458. if (idx==p_dir->subdirs.size()) {
  459. p_dir->subdirs.push_back(efd);
  460. } else {
  461. p_dir->subdirs.insert(idx,efd);
  462. }
  463. da->change_dir("..");
  464. }
  465. } else {
  466. ERR_PRINTS("Cannot go into subdir: "+E->get());
  467. }
  468. p_progress.update(idx,total);
  469. }
  470. for (List<String>::Element*E=files.front();E;E=E->next(),idx++) {
  471. String ext = E->get().extension().to_lower();
  472. if (!valid_extensions.has(ext))
  473. continue; //invalid
  474. EditorFileSystemDirectory::FileInfo *fi = memnew( EditorFileSystemDirectory::FileInfo );
  475. fi->file=E->get();
  476. String path = cd.plus_file(fi->file);
  477. FileCache *fc = file_cache.getptr(path);
  478. uint64_t mt = FileAccess::get_modified_time(path);
  479. if (fc && fc->modification_time == mt) {
  480. fi->meta=fc->meta;
  481. fi->type=fc->type;
  482. fi->modified_time=fc->modification_time;
  483. } else {
  484. fi->meta=_get_meta(path);
  485. fi->type=ResourceLoader::get_resource_type(path);
  486. fi->modified_time=mt;
  487. }
  488. if (fi->meta.enabled) {
  489. if (_check_meta_sources(fi->meta)) {
  490. ItemAction ia;
  491. ia.action=ItemAction::ACTION_FILE_SOURCES_CHANGED;
  492. ia.dir=p_dir;
  493. ia.file=E->get();
  494. scan_actions.push_back(ia);
  495. fi->meta.sources_changed=true;
  496. } else {
  497. fi->meta.sources_changed=false;
  498. }
  499. } else {
  500. fi->meta.sources_changed=true;
  501. }
  502. p_dir->files.push_back(fi);
  503. p_progress.update(idx,total);
  504. }
  505. }
  506. void EditorFileSystem::_scan_fs_changes(EditorFileSystemDirectory *p_dir,const ScanProgress& p_progress) {
  507. uint64_t current_mtime = FileAccess::get_modified_time(p_dir->get_path());
  508. bool updated_dir=false;
  509. //print_line("dir: "+p_dir->get_path()+" MODTIME: "+itos(p_dir->modified_time)+" CTIME: "+itos(current_mtime));
  510. if (current_mtime!=p_dir->modified_time) {
  511. updated_dir=true;
  512. p_dir->modified_time=current_mtime;
  513. //ooooops, dir changed, see what's going on
  514. //first mark everything as veryfied
  515. for(int i=0;i<p_dir->files.size();i++) {
  516. p_dir->files[i]->verified=false;
  517. }
  518. for(int i=0;i<p_dir->subdirs.size();i++) {
  519. p_dir->get_subdir(i)->verified=false;
  520. }
  521. //then scan files and directories and check what's different
  522. DirAccess *da = DirAccess::create(DirAccess::ACCESS_RESOURCES);
  523. String cd = p_dir->get_path();
  524. da->change_dir(cd);
  525. da->list_dir_begin();
  526. while (true) {
  527. bool isdir;
  528. String f = da->get_next(&isdir);
  529. if (f=="")
  530. break;
  531. if (isdir) {
  532. if (f.begins_with(".")) //ignore hidden and . / ..
  533. continue;
  534. int idx = p_dir->find_dir_index(f);
  535. if (idx==-1) {
  536. if (FileAccess::exists(cd.plus_file(f).plus_file("engine.cfg"))) // skip if another project inside this
  537. continue;
  538. EditorFileSystemDirectory *efd = memnew( EditorFileSystemDirectory );
  539. efd->parent=p_dir;
  540. efd->name=f;
  541. DirAccess *d = DirAccess::create(DirAccess::ACCESS_RESOURCES);
  542. d->change_dir(cd.plus_file(f));
  543. _scan_new_dir(efd,d,p_progress.get_sub(1,1));
  544. memdelete(d);
  545. ItemAction ia;
  546. ia.action=ItemAction::ACTION_DIR_ADD;
  547. ia.dir=p_dir;
  548. ia.file=f;
  549. ia.new_dir=efd;
  550. scan_actions.push_back(ia);
  551. } else {
  552. p_dir->subdirs[idx]->verified=true;
  553. }
  554. } else {
  555. String ext = f.extension().to_lower();
  556. if (!valid_extensions.has(ext))
  557. continue; //invalid
  558. int idx = p_dir->find_file_index(f);
  559. if (idx==-1) {
  560. //never seen this file, add actition to add it
  561. EditorFileSystemDirectory::FileInfo *fi = memnew( EditorFileSystemDirectory::FileInfo );
  562. fi->file=f;
  563. String path = cd.plus_file(fi->file);
  564. fi->modified_time=FileAccess::get_modified_time(path);
  565. fi->meta=_get_meta(path);
  566. fi->type=ResourceLoader::get_resource_type(path);
  567. {
  568. ItemAction ia;
  569. ia.action=ItemAction::ACTION_FILE_ADD;
  570. ia.dir=p_dir;
  571. ia.file=f;
  572. ia.new_file=fi;
  573. scan_actions.push_back(ia);
  574. }
  575. //take the chance and scan sources
  576. if (_check_meta_sources(fi->meta)) {
  577. ItemAction ia;
  578. ia.action=ItemAction::ACTION_FILE_SOURCES_CHANGED;
  579. ia.dir=p_dir;
  580. ia.file=f;
  581. scan_actions.push_back(ia);
  582. fi->meta.sources_changed=true;
  583. } else {
  584. fi->meta.sources_changed=false;
  585. }
  586. } else {
  587. p_dir->files[idx]->verified=true;
  588. }
  589. }
  590. }
  591. da->list_dir_end();
  592. memdelete(da);
  593. }
  594. for(int i=0;i<p_dir->files.size();i++) {
  595. if (updated_dir && !p_dir->files[i]->verified) {
  596. //this file was removed, add action to remove it
  597. ItemAction ia;
  598. ia.action=ItemAction::ACTION_FILE_REMOVE;
  599. ia.dir=p_dir;
  600. ia.file=p_dir->files[i]->file;
  601. scan_actions.push_back(ia);
  602. continue;
  603. }
  604. if (_check_meta_sources(p_dir->files[i]->meta)) {
  605. ItemAction ia;
  606. ia.action=ItemAction::ACTION_FILE_SOURCES_CHANGED;
  607. ia.dir=p_dir;
  608. ia.file=p_dir->files[i]->file;
  609. scan_actions.push_back(ia);
  610. p_dir->files[i]->meta.sources_changed=true;
  611. } else {
  612. p_dir->files[i]->meta.sources_changed=false;
  613. }
  614. EditorResourcePreview::get_singleton()->check_for_invalidation(p_dir->get_file_path(i));
  615. }
  616. for(int i=0;i<p_dir->subdirs.size();i++) {
  617. if (updated_dir && !p_dir->subdirs[i]->verified) {
  618. //this directory was removed, add action to remove it
  619. ItemAction ia;
  620. ia.action=ItemAction::ACTION_DIR_REMOVE;
  621. ia.dir=p_dir->subdirs[i];
  622. scan_actions.push_back(ia);
  623. continue;
  624. }
  625. _scan_fs_changes(p_dir->get_subdir(i),p_progress);
  626. }
  627. }
  628. void EditorFileSystem::_thread_func_sources(void *_userdata) {
  629. EditorFileSystem *efs = (EditorFileSystem*)_userdata;
  630. if (efs->filesystem) {
  631. EditorProgressBG pr("sources",TTR("ScanSources"),1000);
  632. ScanProgress sp;
  633. sp.progress=&pr;
  634. sp.hi=1;
  635. sp.low=0;
  636. efs->_scan_fs_changes(efs->filesystem,sp);
  637. }
  638. efs->scanning_sources_done=true;
  639. }
  640. void EditorFileSystem::get_changed_sources(List<String> *r_changed) {
  641. *r_changed=sources_changed;
  642. }
  643. void EditorFileSystem::scan_sources() {
  644. if (scanning || scanning_sources|| thread)
  645. return;
  646. sources_changed.clear();
  647. scanning_sources=true;
  648. scanning_sources_done=false;
  649. abort_scan=false;
  650. if (!use_threads) {
  651. if (filesystem) {
  652. EditorProgressBG pr("sources",TTR("ScanSources"),1000);
  653. ScanProgress sp;
  654. sp.progress=&pr;
  655. sp.hi=1;
  656. sp.low=0;
  657. scan_total=0;
  658. _scan_fs_changes(filesystem,sp);
  659. if (_update_scan_actions())
  660. emit_signal("filesystem_changed");
  661. }
  662. scanning_sources=false;
  663. scanning_sources_done=true;
  664. emit_signal("sources_changed",sources_changed.size()>0);
  665. } else {
  666. ERR_FAIL_COND(thread_sources);
  667. set_process(true);
  668. scan_total=0;
  669. Thread::Settings s;
  670. s.priority=Thread::PRIORITY_LOW;
  671. thread_sources = Thread::create(_thread_func_sources,this,s);
  672. //tree->hide();
  673. //print_line("SCAN BEGIN!");
  674. //progress->show();
  675. }
  676. }
  677. void EditorFileSystem::_notification(int p_what) {
  678. switch(p_what) {
  679. case NOTIFICATION_ENTER_TREE: {
  680. scan();
  681. } break;
  682. case NOTIFICATION_EXIT_TREE: {
  683. if (use_threads && thread) {
  684. //abort thread if in progress
  685. abort_scan=true;
  686. while(scanning) {
  687. OS::get_singleton()->delay_usec(1000);
  688. }
  689. Thread::wait_to_finish(thread);
  690. memdelete(thread);
  691. thread=NULL;
  692. WARN_PRINTS("Scan thread aborted...");
  693. set_process(false);
  694. }
  695. if (filesystem)
  696. memdelete(filesystem);
  697. if (new_filesystem)
  698. memdelete(new_filesystem);
  699. filesystem=NULL;
  700. new_filesystem=NULL;
  701. } break;
  702. case NOTIFICATION_PROCESS: {
  703. if (use_threads) {
  704. if (scanning_sources) {
  705. if (scanning_sources_done) {
  706. scanning_sources=false;
  707. set_process(false);
  708. Thread::wait_to_finish(thread_sources);
  709. memdelete(thread_sources);
  710. thread_sources=NULL;
  711. if (_update_scan_actions())
  712. emit_signal("filesystem_changed");
  713. //print_line("sources changed: "+itos(sources_changed.size()));
  714. emit_signal("sources_changed",sources_changed.size()>0);
  715. }
  716. } else if (!scanning) {
  717. set_process(false);
  718. if (filesystem)
  719. memdelete(filesystem);
  720. filesystem=new_filesystem;
  721. new_filesystem=NULL;
  722. Thread::wait_to_finish(thread);
  723. memdelete(thread);
  724. thread=NULL;
  725. _update_scan_actions();
  726. emit_signal("filesystem_changed");
  727. emit_signal("sources_changed",sources_changed.size()>0);
  728. //print_line("initial sources changed: "+itos(sources_changed.size()));
  729. } else {
  730. //progress->set_text("Scanning...\n"+itos(total*100)+"%");
  731. }
  732. }
  733. } break;
  734. }
  735. }
  736. bool EditorFileSystem::is_scanning() const {
  737. return scanning;
  738. }
  739. float EditorFileSystem::get_scanning_progress() const {
  740. return scan_total;
  741. }
  742. EditorFileSystemDirectory *EditorFileSystem::get_filesystem() {
  743. return filesystem;
  744. }
  745. void EditorFileSystem::_save_filesystem_cache(EditorFileSystemDirectory*p_dir,FileAccess *p_file) {
  746. if (!p_dir)
  747. return; //none
  748. p_file->store_line("::"+p_dir->get_path()+"::"+String::num(p_dir->modified_time));
  749. for(int i=0;i<p_dir->files.size();i++) {
  750. String s=p_dir->files[i]->file+"::"+p_dir->files[i]->type+"::"+String::num(p_dir->files[i]->modified_time)+"::";
  751. if (p_dir->files[i]->meta.enabled) {
  752. s+=p_dir->files[i]->meta.import_editor;
  753. for(int j=0;j<p_dir->files[i]->meta.sources.size();j++){
  754. s+="<>"+p_dir->files[i]->meta.sources[j].path;
  755. s+="<>"+p_dir->files[i]->meta.sources[j].md5;
  756. s+="<>"+String::num(p_dir->files[i]->meta.sources[j].modified_time);
  757. }
  758. }
  759. s+="::";
  760. for(int j=0;j<p_dir->files[i]->meta.deps.size();j++) {
  761. if (j>0)
  762. s+="<>";
  763. s+=p_dir->files[i]->meta.deps[j];
  764. }
  765. p_file->store_line(s);
  766. }
  767. for(int i=0;i<p_dir->subdirs.size();i++) {
  768. _save_filesystem_cache(p_dir->subdirs[i],p_file);
  769. }
  770. }
  771. bool EditorFileSystem::_find_file(const String& p_file,EditorFileSystemDirectory ** r_d, int &r_file_pos) const {
  772. //todo make faster
  773. if (!filesystem || scanning)
  774. return false;
  775. String f = Globals::get_singleton()->localize_path(p_file);
  776. if (!f.begins_with("res://"))
  777. return false;
  778. f=f.substr(6,f.length());
  779. f=f.replace("\\","/");
  780. Vector<String> path = f.split("/");
  781. if (path.size()==0)
  782. return false;
  783. String file=path[path.size()-1];
  784. path.resize(path.size()-1);
  785. EditorFileSystemDirectory *fs=filesystem;
  786. for(int i=0;i<path.size();i++) {
  787. int idx=-1;
  788. for(int j=0;j<fs->get_subdir_count();j++) {
  789. if (fs->get_subdir(j)->get_name()==path[i]) {
  790. idx=j;
  791. break;
  792. }
  793. }
  794. if (idx==-1) {
  795. //does not exist, create i guess?
  796. EditorFileSystemDirectory *efsd = memnew( EditorFileSystemDirectory );
  797. efsd->name=path[i];
  798. int idx2=0;
  799. for(int j=0;j<fs->get_subdir_count();j++) {
  800. if (efsd->name<fs->get_subdir(j)->get_name())
  801. break;
  802. idx2++;
  803. }
  804. if (idx2==fs->get_subdir_count())
  805. fs->subdirs.push_back(efsd);
  806. else
  807. fs->subdirs.insert(idx2,efsd);
  808. fs=efsd;
  809. } else {
  810. fs=fs->get_subdir(idx);
  811. }
  812. }
  813. int cpos=-1;
  814. for(int i=0;i<fs->files.size();i++) {
  815. if (fs->files[i]->file==file) {
  816. cpos=i;
  817. break;
  818. }
  819. }
  820. r_file_pos=cpos;
  821. *r_d=fs;
  822. if (cpos!=-1) {
  823. return true;
  824. } else {
  825. return false;
  826. }
  827. }
  828. String EditorFileSystem::get_file_type(const String& p_file) const {
  829. EditorFileSystemDirectory *fs=NULL;
  830. int cpos=-1;
  831. if (!_find_file(p_file,&fs,cpos)) {
  832. return "";
  833. }
  834. return fs->files[cpos]->type;
  835. }
  836. EditorFileSystemDirectory* EditorFileSystem::find_file(const String& p_file,int* r_index) const {
  837. if (!filesystem || scanning)
  838. return NULL;
  839. EditorFileSystemDirectory *fs=NULL;
  840. int cpos=-1;
  841. if (!_find_file(p_file,&fs,cpos)) {
  842. return NULL;
  843. }
  844. if (r_index)
  845. *r_index=cpos;
  846. return fs;
  847. }
  848. EditorFileSystemDirectory *EditorFileSystem::get_path(const String& p_path) {
  849. if (!filesystem || scanning)
  850. return NULL;
  851. String f = Globals::get_singleton()->localize_path(p_path);
  852. if (!f.begins_with("res://"))
  853. return NULL;
  854. f=f.substr(6,f.length());
  855. f=f.replace("\\","/");
  856. if (f==String())
  857. return filesystem;
  858. if (f.ends_with("/"))
  859. f=f.substr(0,f.length()-1);
  860. Vector<String> path = f.split("/");
  861. if (path.size()==0)
  862. return NULL;
  863. EditorFileSystemDirectory *fs=filesystem;
  864. for(int i=0;i<path.size();i++) {
  865. int idx=-1;
  866. for(int j=0;j<fs->get_subdir_count();j++) {
  867. if (fs->get_subdir(j)->get_name()==path[i]) {
  868. idx=j;
  869. break;
  870. }
  871. }
  872. if (idx==-1) {
  873. return NULL;
  874. } else {
  875. fs=fs->get_subdir(idx);
  876. }
  877. }
  878. return fs;
  879. }
  880. void EditorFileSystem::_resource_saved(const String& p_path){
  881. //print_line("resource saved: "+p_path);
  882. EditorFileSystem::get_singleton()->update_file(p_path);
  883. }
  884. String EditorFileSystem::_find_first_from_source(EditorFileSystemDirectory* p_dir,const String &p_src) const {
  885. for(int i=0;i<p_dir->files.size();i++) {
  886. for(int j=0;j<p_dir->files[i]->meta.sources.size();j++) {
  887. if (p_dir->files[i]->meta.sources[j].path==p_src)
  888. return p_dir->get_file_path(i);
  889. }
  890. }
  891. for(int i=0;i<p_dir->subdirs.size();i++) {
  892. String ret = _find_first_from_source(p_dir->subdirs[i],p_src);
  893. if (ret.length()>0)
  894. return ret;
  895. }
  896. return String();
  897. }
  898. String EditorFileSystem::find_resource_from_source(const String& p_path) const {
  899. if (filesystem)
  900. return _find_first_from_source(filesystem,p_path);
  901. return String();
  902. }
  903. void EditorFileSystem::update_file(const String& p_file) {
  904. EditorFileSystemDirectory *fs=NULL;
  905. int cpos=-1;
  906. if (!_find_file(p_file,&fs,cpos)) {
  907. if (!fs)
  908. return;
  909. }
  910. if (!FileAccess::exists(p_file)) {
  911. //was removed
  912. memdelete( fs->files[cpos] );
  913. fs->files.remove(cpos);
  914. call_deferred("emit_signal","filesystem_changed"); //update later
  915. return;
  916. }
  917. String type = ResourceLoader::get_resource_type(p_file);
  918. if (cpos==-1) {
  919. int idx=0;
  920. for(int i=0;i<fs->files.size();i++) {
  921. if (p_file<fs->files[i]->file)
  922. break;
  923. idx++;
  924. }
  925. EditorFileSystemDirectory::FileInfo *fi = memnew( EditorFileSystemDirectory::FileInfo );
  926. fi->file=p_file.get_file();
  927. if (idx==fs->files.size()) {
  928. fs->files.push_back(fi);
  929. } else {
  930. fs->files.insert(idx,fi);
  931. }
  932. cpos=idx;
  933. }
  934. //print_line("UPDATING: "+p_file);
  935. fs->files[cpos]->type=type;
  936. fs->files[cpos]->modified_time=FileAccess::get_modified_time(p_file);
  937. fs->files[cpos]->meta=_get_meta(p_file);
  938. EditorResourcePreview::get_singleton()->call_deferred("check_for_invalidation",p_file);
  939. call_deferred("emit_signal","filesystem_changed"); //update later
  940. }
  941. void EditorFileSystem::_bind_methods() {
  942. ObjectTypeDB::bind_method(_MD("get_filesystem:EditorFileSystemDirectory"),&EditorFileSystem::get_filesystem);
  943. ObjectTypeDB::bind_method(_MD("is_scanning"),&EditorFileSystem::is_scanning);
  944. ObjectTypeDB::bind_method(_MD("get_scanning_progress"),&EditorFileSystem::get_scanning_progress);
  945. ObjectTypeDB::bind_method(_MD("scan"),&EditorFileSystem::scan);
  946. ObjectTypeDB::bind_method(_MD("scan_sources"),&EditorFileSystem::scan_sources);
  947. ObjectTypeDB::bind_method(_MD("update_file","path"),&EditorFileSystem::update_file);
  948. ObjectTypeDB::bind_method(_MD("get_path:EditorFileSystemDirectory","path"),&EditorFileSystem::get_path);
  949. ObjectTypeDB::bind_method(_MD("get_file_type","path"),&EditorFileSystem::get_file_type);
  950. ADD_SIGNAL( MethodInfo("filesystem_changed") );
  951. ADD_SIGNAL( MethodInfo("sources_changed",PropertyInfo(Variant::BOOL,"exist")) );
  952. }
  953. EditorFileSystem::EditorFileSystem() {
  954. singleton=this;
  955. filesystem=memnew( EditorFileSystemDirectory ); //like, empty
  956. thread = NULL;
  957. scanning=false;
  958. use_threads=true;
  959. thread_sources=NULL;
  960. new_filesystem=NULL;
  961. scanning_sources=false;
  962. ResourceSaver::set_save_callback(_resource_saved);
  963. List<String> extensionsl;
  964. ResourceLoader::get_recognized_extensions_for_type("",&extensionsl);
  965. for(List<String>::Element *E = extensionsl.front();E;E=E->next()) {
  966. valid_extensions.insert(E->get());
  967. }
  968. scan_total=0;
  969. }
  970. EditorFileSystem::~EditorFileSystem() {
  971. }