editor_import_export.cpp 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721
  1. /*************************************************************************/
  2. /* editor_import_export.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* http://www.godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2014 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 "version.h"
  30. #include "editor_import_export.h"
  31. #include "script_language.h"
  32. #include "globals.h"
  33. #include "os/file_access.h"
  34. #include "os/dir_access.h"
  35. #include "tools/editor/editor_file_system.h"
  36. #include "io/resource_loader.h"
  37. #include "editor_node.h"
  38. #include "editor_settings.h"
  39. #include "io/config_file.h"
  40. #include "io/resource_saver.h"
  41. #include "io/md5.h"
  42. #include "io_plugins/editor_texture_import_plugin.h"
  43. String EditorImportPlugin::validate_source_path(const String& p_path) {
  44. String gp = Globals::get_singleton()->globalize_path(p_path);
  45. String rp = Globals::get_singleton()->get_resource_path();
  46. if (!rp.ends_with("/"))
  47. rp+="/";
  48. return rp.path_to_file(gp);
  49. }
  50. String EditorImportPlugin::expand_source_path(const String& p_path) {
  51. if (p_path.is_rel_path()) {
  52. return Globals::get_singleton()->get_resource_path().plus_file(p_path).simplify_path();
  53. } else {
  54. return p_path;
  55. }
  56. }
  57. void EditorImportPlugin::_bind_methods() {
  58. ObjectTypeDB::add_virtual_method(get_type_static(),MethodInfo(Variant::STRING,"get_name"));
  59. ObjectTypeDB::add_virtual_method(get_type_static(),MethodInfo(Variant::STRING,"get_visible_name"));
  60. ObjectTypeDB::add_virtual_method(get_type_static(),MethodInfo("import_dialog",PropertyInfo(Variant::STRING,"from")));
  61. ObjectTypeDB::add_virtual_method(get_type_static(),MethodInfo(Variant::INT,"import",PropertyInfo(Variant::STRING,"path"),PropertyInfo(Variant::OBJECT,"from",PROPERTY_HINT_RESOURCE_TYPE,"ResourceImportMetaData")));
  62. ObjectTypeDB::add_virtual_method(get_type_static(),MethodInfo(Variant::RAW_ARRAY,"custom_export",PropertyInfo(Variant::STRING,"path")));
  63. }
  64. String EditorImportPlugin::get_name() const {
  65. if (get_script_instance() && get_script_instance()->has_method("get_name")) {
  66. return get_script_instance()->call("get_name");
  67. }
  68. ERR_FAIL_V("");
  69. }
  70. String EditorImportPlugin::get_visible_name() const {
  71. if (get_script_instance() && get_script_instance()->has_method("get_visible_name")) {
  72. return get_script_instance()->call("get_visible_name");
  73. }
  74. ERR_FAIL_V("");
  75. }
  76. void EditorImportPlugin::import_dialog(const String& p_from) {
  77. if (get_script_instance() && get_script_instance()->has_method("import_dialog")) {
  78. get_script_instance()->call("import_dialog",p_from);
  79. return;
  80. }
  81. ERR_FAIL();
  82. }
  83. Error EditorImportPlugin::import(const String& p_path, const Ref<ResourceImportMetadata>& p_from) {
  84. if (get_script_instance() && get_script_instance()->has_method("import")) {
  85. return Error(get_script_instance()->call("import",p_path,p_from).operator int());
  86. }
  87. ERR_FAIL_V(ERR_UNAVAILABLE);
  88. }
  89. Vector<uint8_t> EditorImportPlugin::custom_export(const String& p_path, const Ref<EditorExportPlatform> &p_platform) {
  90. if (get_script_instance() && get_script_instance()->has_method("custom_export")) {
  91. get_script_instance()->call("custom_export",p_path);
  92. }
  93. return Vector<uint8_t>();
  94. }
  95. EditorImportPlugin::EditorImportPlugin() {
  96. }
  97. /////////////////////////////////////////////////////////////////////////////////////////////////////
  98. void EditorExportPlugin::_bind_methods() {
  99. BIND_VMETHOD( MethodInfo("custom_export:Dictionary",PropertyInfo(Variant::STRING,"name",PROPERTY_HINT_RESOURCE_TYPE,"EditorExportPlatformPC")) );
  100. }
  101. Vector<uint8_t> EditorExportPlugin::custom_export(String& p_path,const Ref<EditorExportPlatform> &p_platform) {
  102. if (get_script_instance()) {
  103. Variant d = get_script_instance()->call("custom_export",p_path,p_platform);
  104. if (d.get_type()==Variant::NIL)
  105. return Vector<uint8_t>();
  106. ERR_FAIL_COND_V(d.get_type()!=Variant::DICTIONARY,Vector<uint8_t>());
  107. Dictionary dict=d;
  108. ERR_FAIL_COND_V(!dict.has("name"),Vector<uint8_t>());
  109. ERR_FAIL_COND_V(!dict.has("data"),Vector<uint8_t>());
  110. p_path=dict["name"];
  111. return dict["data"];
  112. }
  113. return Vector<uint8_t>();
  114. }
  115. EditorExportPlugin::EditorExportPlugin() {
  116. }
  117. /////////////////////////////////////////////////////////////////////////////////////////////////////
  118. static void _add_to_list(EditorFileSystemDirectory *p_efsd,Set<StringName>& r_list) {
  119. for(int i=0;i<p_efsd->get_subdir_count();i++) {
  120. _add_to_list(p_efsd->get_subdir(i),r_list);
  121. }
  122. for(int i=0;i<p_efsd->get_file_count();i++) {
  123. r_list.insert(p_efsd->get_file_path(i));
  124. }
  125. }
  126. struct __EESortDepCmp {
  127. _FORCE_INLINE_ bool operator()(const StringName& p_l,const StringName& p_r) const {
  128. return p_l.operator String() < p_r.operator String();
  129. }
  130. };
  131. static void _add_files_with_filter(DirAccess *da,const List<String>& p_filters,Set<StringName>& r_list) {
  132. List<String> files;
  133. List<String> dirs;
  134. da->list_dir_begin();
  135. String f = da->get_next();
  136. while(f!="") {
  137. print_line("HOHO: "+f);
  138. if (da->current_is_dir())
  139. dirs.push_back(f);
  140. else
  141. files.push_back(f);
  142. f=da->get_next();
  143. }
  144. String r = da->get_current_dir().replace("\\","/");
  145. if (!r.ends_with("/"))
  146. r+="/";
  147. print_line("AT: "+r);
  148. for(List<String>::Element *E=files.front();E;E=E->next()) {
  149. String fullpath=r+E->get();
  150. for(const List<String>::Element *F=p_filters.front();F;F=F->next()) {
  151. if (fullpath.matchn(F->get())) {
  152. r_list.insert(fullpath);
  153. print_line("Added: "+fullpath);
  154. }
  155. }
  156. }
  157. da->list_dir_end();
  158. for(List<String>::Element *E=dirs.front();E;E=E->next()) {
  159. if (E->get().begins_with("."))
  160. continue;
  161. da->change_dir(E->get());
  162. _add_files_with_filter(da,p_filters,r_list);
  163. da->change_dir("..");
  164. }
  165. }
  166. static void _add_filter_to_list(Set<StringName>& r_list,const String& p_filter) {
  167. if (p_filter=="")
  168. return;
  169. Vector<String> split = p_filter.split(",");
  170. List<String> filters;
  171. for(int i=0;i<split.size();i++) {
  172. String f = split[i].strip_edges();
  173. if (f.empty())
  174. continue;
  175. filters.push_back(f);
  176. }
  177. DirAccess *da = DirAccess::open("res://");
  178. _add_files_with_filter(da,filters,r_list);
  179. memdelete(da);
  180. }
  181. Vector<uint8_t> EditorExportPlatform::get_exported_file(String& p_fname) const {
  182. Ref<EditorExportPlatform> ep=EditorImportExport::get_singleton()->get_export_platform(get_name());
  183. for(int i=0;i<EditorImportExport::get_singleton()->get_export_plugin_count();i++) {
  184. Vector<uint8_t> data = EditorImportExport::get_singleton()->get_export_plugin(i)->custom_export(p_fname,ep);
  185. if (data.size())
  186. return data;
  187. }
  188. FileAccess *f = FileAccess::open(p_fname,FileAccess::READ);
  189. ERR_FAIL_COND_V(!f,Vector<uint8_t>());
  190. Vector<uint8_t> ret;
  191. ret.resize(f->get_len());
  192. int rbs = f->get_buffer(ret.ptr(),ret.size());
  193. memdelete(f);
  194. return ret;
  195. }
  196. Vector<StringName> EditorExportPlatform::get_dependencies(bool p_bundles) const {
  197. Set<StringName> exported;
  198. if (FileAccess::exists("res://engine.cfg"))
  199. exported.insert("res://engine.cfg");
  200. if (EditorImportExport::get_singleton()->get_export_filter()!=EditorImportExport::EXPORT_SELECTED) {
  201. String filter;
  202. if (EditorImportExport::get_singleton()->get_export_filter()==EditorImportExport::EXPORT_ALL) {
  203. _add_filter_to_list(exported,"*");
  204. } else {
  205. _add_to_list(EditorFileSystem::get_singleton()->get_filesystem(),exported);
  206. _add_filter_to_list(exported,EditorImportExport::get_singleton()->get_export_custom_filter());
  207. }
  208. } else {
  209. Map<String,Map<String,String> > remapped_paths;
  210. Set<String> scene_extensions;
  211. Set<String> resource_extensions;
  212. {
  213. List<String> l;
  214. // SceneLoader::get_recognized_extensions(&l);
  215. // for(List<String>::Element *E=l.front();E;E=E->next()) {
  216. //
  217. // scene_extensions.insert(E->get());
  218. // }
  219. ResourceLoader::get_recognized_extensions_for_type("",&l);
  220. for(List<String>::Element *E=l.front();E;E=E->next()) {
  221. resource_extensions.insert(E->get());
  222. }
  223. }
  224. List<StringName> toexport;
  225. EditorImportExport::get_singleton()->get_export_file_list(&toexport);
  226. print_line("TO EXPORT: "+itos(toexport.size()));
  227. for (List<StringName>::Element *E=toexport.front();E;E=E->next()) {
  228. print_line("DEP: "+String(E->get()));
  229. exported.insert(E->get());
  230. if (p_bundles && EditorImportExport::get_singleton()->get_export_file_action(E->get())==EditorImportExport::ACTION_BUNDLE) {
  231. print_line("NO BECAUSE OF BUNDLE!");
  232. continue; //no dependencies needed to be copied
  233. }
  234. List<String> testsubs;
  235. testsubs.push_back(E->get());
  236. while(testsubs.size()) {
  237. //recursive subdep search!
  238. List<String> deplist;
  239. ResourceLoader::get_dependencies(testsubs.front()->get(),&deplist);
  240. testsubs.pop_front();
  241. List<String> subdeps;
  242. for (List<String>::Element *F=deplist.front();F;F=F->next()) {
  243. StringName dep = F->get();
  244. if (exported.has(dep) || EditorImportExport::get_singleton()->get_export_file_action(dep)!=EditorImportExport::ACTION_NONE)
  245. continue; //dependency added or to be added
  246. print_line(" SUBDEP: "+String(dep));
  247. exported.insert(dep);
  248. testsubs.push_back(dep);
  249. }
  250. }
  251. }
  252. _add_filter_to_list(exported,EditorImportExport::get_singleton()->get_export_custom_filter());
  253. }
  254. Vector<StringName> ret;
  255. ret.resize(exported.size());
  256. int idx=0;
  257. for(Set<StringName>::Element *E=exported.front();E;E=E->next()) {
  258. ret[idx++]=E->get();
  259. }
  260. SortArray<StringName,__EESortDepCmp> sort; //some platforms work better if this is sorted
  261. sort.sort(ret.ptr(),ret.size());
  262. return ret;
  263. }
  264. ///////////////////////////////////////
  265. bool EditorExportPlatformPC::_set(const StringName& p_name, const Variant& p_value) {
  266. String n = p_name;
  267. if (n=="custom_binary/release") {
  268. custom_release_binary=p_value;
  269. } else if (n=="custom_binary/debug") {
  270. custom_debug_binary=p_value;
  271. } else if (n=="resources/pack_mode") {
  272. export_mode=ExportMode(int(p_value));
  273. } else if (n=="binary/64_bits") {
  274. use64=p_value;
  275. } else
  276. return false;
  277. return true;
  278. }
  279. bool EditorExportPlatformPC::_get(const StringName& p_name,Variant &r_ret) const {
  280. String n = p_name;
  281. if (n=="custom_binary/release") {
  282. r_ret=custom_release_binary;
  283. } else if (n=="custom_binary/debug") {
  284. r_ret=custom_debug_binary;
  285. } else if (n=="resources/pack_mode") {
  286. r_ret=export_mode;
  287. } else if (n=="binary/64_bits") {
  288. r_ret=use64;
  289. } else
  290. return false;
  291. return true;
  292. }
  293. void EditorExportPlatformPC::_get_property_list( List<PropertyInfo> *p_list) const {
  294. p_list->push_back( PropertyInfo( Variant::STRING, "custom_binary/debug", PROPERTY_HINT_FILE,binary_extension));
  295. p_list->push_back( PropertyInfo( Variant::STRING, "custom_binary/release", PROPERTY_HINT_FILE,binary_extension));
  296. p_list->push_back( PropertyInfo( Variant::INT, "resources/pack_mode", PROPERTY_HINT_ENUM,"Single Exec.,Exec+Pack (.pck),Copy,Bundles (Optical)"));
  297. p_list->push_back( PropertyInfo( Variant::BOOL, "binary/64_bits"));
  298. }
  299. static void _exp_add_dep(Map<StringName,List<StringName> > &deps,const StringName& p_path) {
  300. if (deps.has(p_path))
  301. return; //already done
  302. deps.insert(p_path,List<StringName>());
  303. List<StringName> &deplist=deps[p_path];
  304. Set<StringName> depset;
  305. List<String> dl;
  306. ResourceLoader::get_dependencies(p_path,&dl);
  307. //added in order so child dependencies are always added bfore parent dependencies
  308. for (List<String>::Element *E=dl.front();E;E=E->next()) {
  309. if (!deps.has(E->get()))
  310. _exp_add_dep(deps,E->get());
  311. for(List<StringName>::Element *F=deps[E->get()].front();F;F=F->next()) {
  312. if (!depset.has(F->get())) {
  313. depset.insert(F->get());
  314. deplist.push_back(F->get());
  315. }
  316. }
  317. if (!depset.has(E->get())) {
  318. depset.insert(E->get());
  319. deplist.push_back(E->get());
  320. }
  321. }
  322. }
  323. Error EditorExportPlatform::export_project_files(EditorExportSaveFunction p_func, void* p_udata,bool p_make_bundles) {
  324. /* ALL FILES AND DEPENDENCIES */
  325. Vector<StringName> files=get_dependencies(p_make_bundles);
  326. Map<StringName,List<StringName> > deps;
  327. if (false) {
  328. for(int i=0;i<files.size();i++) {
  329. _exp_add_dep(deps,files[i]);
  330. }
  331. }
  332. /* GROUP ATLAS */
  333. List<StringName> groups;
  334. EditorImportExport::get_singleton()->image_export_get_groups(&groups);
  335. Map<StringName,StringName> remap_files;
  336. Set<StringName> saved;
  337. int counter=0;
  338. for(List<StringName>::Element *E=groups.front();E;E=E->next()) {
  339. if (!EditorImportExport::get_singleton()->image_export_group_get_make_atlas(E->get()))
  340. continue; //uninterested, only process for atlas!
  341. List<StringName> atlas_images;
  342. EditorImportExport::get_singleton()->image_export_get_images_in_group(E->get(),&atlas_images);
  343. atlas_images.sort_custom<StringName::AlphCompare>();
  344. for (List<StringName>::Element *F=atlas_images.front();F;) {
  345. List<StringName>::Element *N=F->next();
  346. if (!FileAccess::exists(F->get())) {
  347. atlas_images.erase(F);
  348. }
  349. F=N;
  350. }
  351. if (atlas_images.size()<=1)
  352. continue;
  353. int group_format=0;
  354. float group_lossy_quality=EditorImportExport::get_singleton()->image_export_group_get_lossy_quality(E->get());
  355. int group_shrink=EditorImportExport::get_singleton()->image_export_group_get_shrink(E->get());
  356. group_shrink*=EditorImportExport::get_singleton()->get_export_image_shrink();
  357. switch(EditorImportExport::get_singleton()->image_export_group_get_image_action(E->get())) {
  358. case EditorImportExport::IMAGE_ACTION_NONE: {
  359. switch(EditorImportExport::get_singleton()->get_export_image_action()) {
  360. case EditorImportExport::IMAGE_ACTION_NONE: {
  361. group_format=EditorTextureImportPlugin::IMAGE_FORMAT_COMPRESS_DISK_LOSSLESS; //?
  362. } break; //use default
  363. case EditorImportExport::IMAGE_ACTION_COMPRESS_DISK: {
  364. group_format=EditorTextureImportPlugin::IMAGE_FORMAT_COMPRESS_DISK_LOSSY;
  365. } break; //use default
  366. case EditorImportExport::IMAGE_ACTION_COMPRESS_RAM: {
  367. group_format=EditorTextureImportPlugin::IMAGE_FORMAT_COMPRESS_RAM;
  368. } break; //use default
  369. }
  370. group_lossy_quality=EditorImportExport::get_singleton()->get_export_image_quality();
  371. } break; //use default
  372. case EditorImportExport::IMAGE_ACTION_COMPRESS_DISK: {
  373. group_format=EditorTextureImportPlugin::IMAGE_FORMAT_COMPRESS_DISK_LOSSY;
  374. } break; //use default
  375. case EditorImportExport::IMAGE_ACTION_COMPRESS_RAM: {
  376. group_format=EditorTextureImportPlugin::IMAGE_FORMAT_COMPRESS_RAM;
  377. } break; //use default
  378. }
  379. String image_list_md5;
  380. {
  381. MD5_CTX ctx;
  382. MD5Init(&ctx);
  383. for (List<StringName>::Element *F=atlas_images.front();F;F=F->next()) {
  384. String p = F->get();
  385. MD5Update(&ctx,(unsigned char*)p.utf8().get_data(),p.utf8().length());
  386. }
  387. MD5Final(&ctx);
  388. image_list_md5=String::md5(ctx.digest);
  389. }
  390. //ok see if cached
  391. String md5;
  392. bool atlas_valid=true;
  393. String atlas_name;
  394. {
  395. MD5_CTX ctx;
  396. MD5Init(&ctx);
  397. String path = Globals::get_singleton()->get_resource_path()+"::"+String(E->get())+"::"+get_name();
  398. MD5Update(&ctx,(unsigned char*)path.utf8().get_data(),path.utf8().length());
  399. MD5Final(&ctx);
  400. md5 = String::md5(ctx.digest);
  401. }
  402. FileAccess *f=NULL;
  403. if (!FileAccess::exists(EditorSettings::get_singleton()->get_settings_path()+"/tmp/atlas-"+md5)) {
  404. print_line("NO MD5 INVALID");
  405. atlas_valid=false;
  406. }
  407. if (atlas_valid)
  408. f=FileAccess::open(EditorSettings::get_singleton()->get_settings_path()+"/tmp/atlas-"+md5,FileAccess::READ);
  409. if (atlas_valid) {
  410. //compare options
  411. Dictionary options;
  412. options.parse_json(f->get_line());
  413. if (!options.has("lossy_quality") || float(options["lossy_quality"])!=group_lossy_quality)
  414. atlas_valid=false;
  415. else if (!options.has("shrink") || int(options["shrink"])!=group_shrink)
  416. atlas_valid=false;
  417. else if (!options.has("image_format") || int(options["image_format"])!=group_format)
  418. atlas_valid=false;
  419. if (!atlas_valid)
  420. print_line("JSON INVALID");
  421. }
  422. if (atlas_valid) {
  423. //check md5 of list of image /names/
  424. if (f->get_line().strip_edges()!=image_list_md5) {
  425. atlas_valid=false;
  426. print_line("IMAGE MD5 INVALID!");
  427. }
  428. }
  429. Vector<Rect2> rects;
  430. bool resave_deps=false;
  431. if (atlas_valid) {
  432. //check if images were not modified
  433. for (List<StringName>::Element *F=atlas_images.front();F;F=F->next()) {
  434. Vector<String> slices = f->get_line().strip_edges().split("::");
  435. if (slices.size()!=10) {
  436. atlas_valid=false;
  437. print_line("CANT SLICE IN 10");
  438. break;
  439. }
  440. uint64_t mod_time = slices[0].to_int64();
  441. uint64_t file_mod_time = FileAccess::get_modified_time(F->get());
  442. if (mod_time!=file_mod_time) {
  443. String image_md5 = slices[1];
  444. String file_md5 = FileAccess::get_md5(F->get());
  445. if (image_md5!=file_md5) {
  446. atlas_valid=false;
  447. print_line("IMAGE INVALID "+slices[0]);
  448. break;
  449. } else {
  450. resave_deps=true;
  451. }
  452. }
  453. if (atlas_valid) {
  454. //push back region and margin
  455. rects.push_back(Rect2(slices[2].to_float(),slices[3].to_float(),slices[4].to_float(),slices[5].to_float()));
  456. rects.push_back(Rect2(slices[6].to_float(),slices[7].to_float(),slices[8].to_float(),slices[9].to_float()));
  457. }
  458. }
  459. }
  460. if (f) {
  461. memdelete(f);
  462. f=NULL;
  463. }
  464. print_line("ATLAS VALID? "+itos(atlas_valid)+" RESAVE DEPS? "+itos(resave_deps));
  465. if (!atlas_valid) {
  466. rects.clear();
  467. //oh well, atlas is not valid. need to make new one....
  468. String dst_file = EditorSettings::get_singleton()->get_settings_path()+"/tmp/atlas-"+md5+".tex";
  469. Ref<ResourceImportMetadata> imd = memnew( ResourceImportMetadata );
  470. //imd->set_editor();
  471. for (List<StringName>::Element *F=atlas_images.front();F;F=F->next()) {
  472. imd->add_source(EditorImportPlugin::validate_source_path(F->get()),FileAccess::get_md5(F->get()));
  473. }
  474. imd->set_option("format",group_format);
  475. int flags=0;
  476. if (Globals::get_singleton()->get("image_loader/filter"))
  477. flags|=EditorTextureImportPlugin::IMAGE_FLAG_FILTER;
  478. if (!Globals::get_singleton()->get("image_loader/gen_mipmaps"))
  479. flags|=EditorTextureImportPlugin::IMAGE_FLAG_NO_MIPMAPS;
  480. if (!Globals::get_singleton()->get("image_loader/repeat"))
  481. flags|=EditorTextureImportPlugin::IMAGE_FLAG_REPEAT;
  482. flags|=EditorTextureImportPlugin::IMAGE_FLAG_FIX_BORDER_ALPHA;
  483. imd->set_option("flags",flags);
  484. imd->set_option("quality",group_lossy_quality);
  485. imd->set_option("atlas",true);
  486. imd->set_option("crop",true);
  487. imd->set_option("shrink",group_shrink);
  488. Ref<EditorTextureImportPlugin> plugin = EditorImportExport::get_singleton()->get_import_plugin_by_name("texture_atlas");
  489. Error err = plugin->import2(dst_file,imd,get_image_compression(),true);
  490. if (err) {
  491. EditorNode::add_io_error("Error saving atlas! "+dst_file.get_file());
  492. return ERR_CANT_CREATE;
  493. }
  494. ERR_FAIL_COND_V(imd->get_option("rects")==Variant(),ERR_BUG);
  495. Array r_rects=imd->get_option("rects");
  496. rects.resize(r_rects.size());
  497. for(int i=0;i<r_rects.size();i++) {
  498. //get back region and margins
  499. rects[i]=r_rects[i];
  500. }
  501. resave_deps=true;
  502. }
  503. //atlas is valid (or it was just saved i guess), create the atex files and save them
  504. if (resave_deps) {
  505. f=FileAccess::open(EditorSettings::get_singleton()->get_settings_path()+"/tmp/atlas-"+md5,FileAccess::WRITE);
  506. Dictionary options;
  507. options["lossy_quality"]=group_lossy_quality;
  508. options["shrink"]=EditorImportExport::get_singleton()->image_export_group_get_shrink(E->get());
  509. options["image_format"]=group_format;
  510. f->store_line(options.to_json());
  511. f->store_line(image_list_md5);
  512. }
  513. //go through all ATEX files
  514. {
  515. Ref<ImageTexture> atlas = memnew( ImageTexture ); //fake atlas!
  516. String atlas_path="res://atlas-"+md5+".tex";
  517. atlas->set_path(atlas_path);
  518. int idx=0;
  519. for (List<StringName>::Element *F=atlas_images.front();F;F=F->next()) {
  520. String p = F->get();
  521. Ref<AtlasTexture> atex = memnew(AtlasTexture);
  522. atex->set_atlas(atlas);
  523. Rect2 region=rects[idx++];
  524. Rect2 margin=rects[idx++];
  525. atex->set_region(region);
  526. atex->set_margin(margin);
  527. String path = EditorSettings::get_singleton()->get_settings_path()+"/tmp/tmpatlas.atex";
  528. Error err = ResourceSaver::save(path,atex);
  529. if (err!=OK) {
  530. EditorNode::add_io_error("Could not save atlas subtexture: "+path);
  531. return ERR_CANT_CREATE;
  532. }
  533. Vector<uint8_t> data = FileAccess::get_file_as_array(path);
  534. String dst_path = F->get().operator String().basename()+".atex";
  535. err = p_func(p_udata,dst_path,data,counter++,files.size());
  536. saved.insert(dst_path);
  537. if (err)
  538. return err;
  539. if (f) {
  540. //recreating deps..
  541. String depline;
  542. // depline=String(F->get())+"::"+itos(FileAccess::get_modified_time(F->get()))+"::"+FileAccess::get_md5(F->get()); name unneccesary by top md5
  543. depline=itos(FileAccess::get_modified_time(F->get()))+"::"+FileAccess::get_md5(F->get());
  544. depline+="::"+itos(region.pos.x)+"::"+itos(region.pos.y)+"::"+itos(region.size.x)+"::"+itos(region.size.y);
  545. depline+="::"+itos(margin.pos.x)+"::"+itos(margin.pos.y)+"::"+itos(margin.size.x)+"::"+itos(margin.size.y);
  546. f->store_line(depline);
  547. }
  548. remap_files[F->get()]=dst_path;
  549. }
  550. Vector<uint8_t> atlas_data = FileAccess::get_file_as_array(EditorSettings::get_singleton()->get_settings_path()+"/tmp/atlas-"+md5+".tex");
  551. Error err = p_func(p_udata,atlas_path,atlas_data,counter,files.size());
  552. saved.insert(atlas_path);
  553. if (err)
  554. return err;
  555. }
  556. if (f) {
  557. memdelete(f);
  558. }
  559. }
  560. StringName engine_cfg="res://engine.cfg";
  561. for(int i=0;i<files.size();i++) {
  562. if (remap_files.has(files[i]) || files[i]==engine_cfg) //gonna be remapped (happened before!)
  563. continue; //from atlas?
  564. String src=files[i];
  565. Vector<uint8_t> buf = get_exported_file(src);
  566. ERR_CONTINUE( saved.has(src) );
  567. Error err = p_func(p_udata,src,buf,counter++,files.size());
  568. if (err)
  569. return err;
  570. saved.insert(src);
  571. if (src!=String(files[i]))
  572. remap_files[files[i]]=src;
  573. }
  574. {
  575. //make binary engine.cfg config
  576. Map<String,Variant> custom;
  577. if (remap_files.size()) {
  578. Vector<String> remapsprop;
  579. for(Map<StringName,StringName>::Element *E=remap_files.front();E;E=E->next()) {
  580. print_line("REMAP: "+String(E->key())+" -> "+E->get());
  581. remapsprop.push_back(E->key());
  582. remapsprop.push_back(E->get());
  583. }
  584. custom["remap/all"]=remapsprop;
  585. }
  586. //add presaved dependencies
  587. for(Map<StringName,List<StringName> >::Element *E=deps.front();E;E=E->next()) {
  588. if (E->get().size()==0)
  589. continue; //no deps
  590. String key;
  591. Vector<StringName> deps;
  592. //if bundle continue (when bundles supported obviously)
  593. if (remap_files.has(E->key())) {
  594. key=remap_files[E->key()];
  595. } else {
  596. key=E->key();
  597. }
  598. deps.resize(E->get().size());
  599. int i=0;
  600. for(List<StringName>::Element *F=E->get().front();F;F=F->next()) {
  601. deps[i++]=F->get();
  602. print_line(" -"+String(F->get()));
  603. }
  604. NodePath prop(deps,true,String()); //seems best to use this for performance
  605. custom["deps/"+key.md5_text()]=prop;
  606. }
  607. String remap_file="engine.cfb";
  608. String engine_cfb =EditorSettings::get_singleton()->get_settings_path()+"/tmp/tmp"+remap_file;
  609. Globals::get_singleton()->save_custom(engine_cfb,custom);
  610. Vector<uint8_t> data = FileAccess::get_file_as_array(engine_cfb);
  611. Error err = p_func(p_udata,"res://"+remap_file,data,counter,files.size());
  612. if (err)
  613. return err;
  614. }
  615. return OK;
  616. }
  617. Error EditorExportPlatform::save_pack_file(void *p_userdata,const String& p_path, const Vector<uint8_t>& p_data,int p_file,int p_total) {
  618. PackData *pd = (PackData*)p_userdata;
  619. CharString cs=p_path.utf8();
  620. pd->f->store_32(cs.length());
  621. pd->f->store_buffer((uint8_t*)cs.get_data(),cs.length());
  622. TempData td;
  623. td.pos=pd->f->get_pos();;
  624. td.ofs=pd->ftmp->get_pos();
  625. td.size=p_data.size();
  626. pd->file_ofs.push_back(td);
  627. pd->f->store_64(0); //ofs
  628. pd->f->store_64(0); //size
  629. {
  630. MD5_CTX ctx;
  631. MD5Init(&ctx);
  632. MD5Update(&ctx,(unsigned char*)p_data.ptr(),p_data.size());
  633. MD5Final(&ctx);
  634. pd->f->store_buffer(ctx.digest,16);
  635. }
  636. pd->ep->step("Storing File: "+p_path,2+p_file*100/p_total);
  637. pd->count++;
  638. pd->ftmp->store_buffer(p_data.ptr(),p_data.size());
  639. return OK;
  640. }
  641. Error EditorExportPlatform::save_pack(FileAccess *dst,bool p_make_bundles) {
  642. EditorProgress ep("savepack","Packing",102);
  643. String tmppath = EditorSettings::get_singleton()->get_settings_path()+"/tmp/packtmp";
  644. FileAccess *tmp = FileAccess::open(tmppath,FileAccess::WRITE);
  645. uint64_t ofs_begin = dst->get_pos();
  646. dst->store_32(0x43504447); //GDPK
  647. dst->store_32(0); //pack version
  648. dst->store_32(VERSION_MAJOR);
  649. dst->store_32(VERSION_MINOR);
  650. dst->store_32(VERSION_REVISION);
  651. for(int i=0;i<16;i++) {
  652. //reserved
  653. dst->store_32(0);
  654. }
  655. size_t fcountpos = dst->get_pos();
  656. dst->store_32(0);
  657. PackData pd;
  658. pd.ep=&ep;
  659. pd.f=dst;
  660. pd.ftmp=tmp;
  661. pd.count=0;
  662. Error err = export_project_files(save_pack_file,&pd,p_make_bundles);
  663. memdelete(tmp);
  664. if (err)
  665. return err;
  666. size_t ofsplus = dst->get_pos();
  667. //append file
  668. tmp = FileAccess::open(tmppath,FileAccess::READ);
  669. ERR_FAIL_COND_V(!tmp,ERR_CANT_OPEN;)
  670. const int bufsize=16384;
  671. uint8_t buf[bufsize];
  672. while(true) {
  673. int got = tmp->get_buffer(buf,bufsize);
  674. if (got<=0)
  675. break;
  676. dst->store_buffer(buf,got);
  677. }
  678. memdelete(tmp);
  679. dst->store_64(dst->get_pos()-ofs_begin);
  680. dst->store_32(0x43504447); //GDPK
  681. //fix offsets
  682. dst->seek(fcountpos);
  683. dst->store_32(pd.count);
  684. for(int i=0;i<pd.file_ofs.size();i++) {
  685. dst->seek(pd.file_ofs[i].pos);
  686. dst->store_64(pd.file_ofs[i].ofs+ofsplus);
  687. dst->store_64(pd.file_ofs[i].size);
  688. }
  689. return OK;
  690. }
  691. Error EditorExportPlatformPC::export_project(const String& p_path, bool p_debug, bool p_dumb) {
  692. EditorProgress ep("export","Exporting for "+get_name(),102);
  693. const int BUFSIZE = 32768;
  694. ep.step("Setting Up..",0);
  695. String exe_path = EditorSettings::get_singleton()->get_settings_path()+"/templates/";
  696. if (use64) {
  697. if (p_debug)
  698. exe_path+=custom_debug_binary!=""?custom_debug_binary:debug_binary64;
  699. else
  700. exe_path+=custom_release_binary!=""?custom_release_binary:release_binary64;
  701. } else {
  702. if (p_debug)
  703. exe_path+=custom_debug_binary!=""?custom_debug_binary:debug_binary32;
  704. else
  705. exe_path+=custom_release_binary!=""?custom_release_binary:release_binary32;
  706. }
  707. FileAccess *src_exe=FileAccess::open(exe_path,FileAccess::READ);
  708. if (!src_exe) {
  709. EditorNode::add_io_error("Couldn't read source executable at:\n "+exe_path);
  710. return ERR_FILE_CANT_READ;
  711. }
  712. FileAccess *dst=FileAccess::open(p_path,FileAccess::WRITE);
  713. if (!dst) {
  714. EditorNode::add_io_error("Can't copy executable file to:\n "+p_path);
  715. return ERR_FILE_CANT_WRITE;
  716. }
  717. uint8_t buff[32768];
  718. while(true) {
  719. int c = src_exe->get_buffer(buff,BUFSIZE);
  720. if (c>0) {
  721. dst->store_buffer(buff,c);
  722. } else {
  723. break;
  724. }
  725. }
  726. if (export_mode!=EXPORT_EXE) {
  727. String dstfile=p_path.replace_first("res://","").replace("\\","/");
  728. if (dstfile.find("/")!=-1)
  729. dstfile=dstfile.get_base_dir()+"/data.pck";
  730. else
  731. dstfile="data.pck";
  732. memdelete(dst);
  733. dst=FileAccess::open(dstfile,FileAccess::WRITE);
  734. if (!dst) {
  735. EditorNode::add_io_error("Can't write data pack to:\n "+p_path);
  736. return ERR_FILE_CANT_WRITE;
  737. }
  738. }
  739. memdelete(src_exe);
  740. Error err = save_pack(dst,export_mode==EXPORT_BUNDLES);
  741. memdelete(dst);
  742. return err;
  743. }
  744. void EditorExportPlatformPC::set_binary_extension(const String& p_extension) {
  745. binary_extension=p_extension;
  746. }
  747. bool EditorExportPlatformPC::can_export(String *r_error) const {
  748. String err;
  749. bool valid=true;
  750. String exe_path = EditorSettings::get_singleton()->get_settings_path()+"/templates/";
  751. if (!FileAccess::exists(exe_path+debug_binary32) || !FileAccess::exists(exe_path+release_binary32)) {
  752. valid=false;
  753. err="No 32 bits export templates found.\nDownload and install export templates.\n";
  754. }
  755. if (!FileAccess::exists(exe_path+debug_binary64) || !FileAccess::exists(exe_path+release_binary64)) {
  756. valid=false;
  757. err="No 64 bits export templates found.\nDownload and install export templates.\n";
  758. }
  759. if (custom_debug_binary!="" && !FileAccess::exists(custom_debug_binary)) {
  760. valid=false;
  761. err+="Custom debug binary not found.\n";
  762. }
  763. if (custom_release_binary!="" && !FileAccess::exists(custom_release_binary)) {
  764. valid=false;
  765. err+="Custom release binary not found.\n";
  766. }
  767. if (r_error)
  768. *r_error=err;
  769. return valid;
  770. }
  771. EditorExportPlatformPC::EditorExportPlatformPC() {
  772. export_mode=EXPORT_PACK;
  773. use64=true;
  774. }
  775. ///////////////////////////////////////////////////////////////////////////////////////////////////////
  776. EditorImportExport* EditorImportExport::singleton=NULL;
  777. void EditorImportExport::add_import_plugin(const Ref<EditorImportPlugin>& p_plugin) {
  778. by_idx[ p_plugin->get_name() ]=plugins.size();
  779. plugins.push_back(p_plugin);
  780. }
  781. int EditorImportExport::get_import_plugin_count() const{
  782. return plugins.size();
  783. }
  784. Ref<EditorImportPlugin> EditorImportExport::get_import_plugin(int p_idx) const{
  785. ERR_FAIL_INDEX_V(p_idx,plugins.size(),Ref<EditorImportPlugin>());
  786. return plugins[p_idx];
  787. }
  788. Ref<EditorImportPlugin> EditorImportExport::get_import_plugin_by_name(const String& p_string) const{
  789. ERR_FAIL_COND_V( !by_idx.has(p_string), Ref<EditorImportPlugin>());
  790. return plugins[ by_idx[p_string] ];
  791. }
  792. void EditorImportExport::add_export_plugin(const Ref<EditorExportPlugin>& p_plugin) {
  793. export_plugins.push_back(p_plugin);
  794. }
  795. int EditorImportExport::get_export_plugin_count() const{
  796. return export_plugins.size();
  797. }
  798. Ref<EditorExportPlugin> EditorImportExport::get_export_plugin(int p_idx) const{
  799. ERR_FAIL_INDEX_V(p_idx,export_plugins.size(),Ref<EditorExportPlugin>());
  800. return export_plugins[p_idx];
  801. }
  802. void EditorImportExport::set_export_file_action(const StringName& p_file, FileAction p_action) {
  803. if (p_action==ACTION_NONE) {
  804. files.erase(p_file);
  805. } else {
  806. files[p_file]=p_action;
  807. }
  808. }
  809. EditorImportExport::FileAction EditorImportExport::get_export_file_action(const StringName& p_file) const{
  810. if (files.has(p_file))
  811. return files[p_file];
  812. return ACTION_NONE;
  813. }
  814. void EditorImportExport::get_export_file_list(List<StringName> *p_files){
  815. for (Map<StringName,FileAction>::Element *E=files.front();E;E=E->next()) {
  816. p_files->push_back(E->key());
  817. }
  818. }
  819. void EditorImportExport::add_export_platform(const Ref<EditorExportPlatform>& p_export) {
  820. exporters[p_export->get_name()]=p_export;
  821. }
  822. void EditorImportExport::get_export_platforms(List<StringName> *r_platforms) {
  823. for (Map<StringName,Ref<EditorExportPlatform> >::Element *E=exporters.front();E;E=E->next()) {
  824. r_platforms->push_back(E->key());
  825. }
  826. }
  827. Ref<EditorExportPlatform> EditorImportExport::get_export_platform(const StringName& p_platform) {
  828. if (exporters.has(p_platform)) {
  829. return exporters[p_platform];
  830. } else {
  831. return Ref<EditorExportPlatform>();
  832. }
  833. }
  834. bool EditorImportExport::poll_export_platforms() {
  835. bool changed=false;
  836. for (Map<StringName,Ref<EditorExportPlatform> >::Element *E=exporters.front();E;E=E->next()) {
  837. if (E->get()->poll_devices())
  838. changed=true;
  839. }
  840. return changed;
  841. }
  842. void EditorImportExport::set_export_filter(ExportFilter p_enable) {
  843. export_filter=p_enable;
  844. }
  845. EditorImportExport::ExportFilter EditorImportExport::get_export_filter() const{
  846. return export_filter;
  847. }
  848. void EditorImportExport::set_export_custom_filter(const String& p_custom_filter){
  849. export_custom_filter=p_custom_filter;
  850. }
  851. String EditorImportExport::get_export_custom_filter() const{
  852. return export_custom_filter;
  853. }
  854. void EditorImportExport::set_export_image_action(ImageAction p_action) {
  855. image_action=p_action;
  856. }
  857. EditorImportExport::ImageAction EditorImportExport::get_export_image_action() const{
  858. return image_action;
  859. }
  860. void EditorImportExport::set_export_image_shrink(int p_shrink) {
  861. image_shrink=p_shrink;
  862. }
  863. int EditorImportExport::get_export_image_shrink() const{
  864. return image_shrink;
  865. }
  866. void EditorImportExport::set_export_image_quality(float p_quality){
  867. image_action_compress_quality=p_quality;
  868. }
  869. float EditorImportExport::get_export_image_quality() const{
  870. return image_action_compress_quality;
  871. }
  872. void EditorImportExport::image_export_group_create(const StringName& p_name) {
  873. ERR_FAIL_COND(image_groups.has(p_name));
  874. ImageGroup ig;
  875. ig.action=IMAGE_ACTION_NONE; //default
  876. ig.make_atlas=false;
  877. ig.shrink=1;
  878. ig.lossy_quality=0.7;
  879. image_groups[p_name]=ig;
  880. }
  881. bool EditorImportExport::image_export_has_group(const StringName& p_name) const {
  882. return image_groups.has(p_name);
  883. }
  884. void EditorImportExport::image_export_get_groups(List<StringName> *r_name) const {
  885. for (Map<StringName,ImageGroup>::Element *E=image_groups.front();E;E=E->next()) {
  886. r_name->push_back(E->key());
  887. }
  888. }
  889. void EditorImportExport::image_export_group_remove(const StringName& p_name){
  890. ERR_FAIL_COND(!image_groups.has(p_name));
  891. image_groups.erase(p_name);
  892. }
  893. void EditorImportExport::image_export_group_set_image_action(const StringName& p_export_group,ImageAction p_action){
  894. ERR_FAIL_COND(!image_groups.has(p_export_group));
  895. image_groups[p_export_group].action=p_action;
  896. }
  897. EditorImportExport::ImageAction EditorImportExport::image_export_group_get_image_action(const StringName& p_export_group) const{
  898. ERR_FAIL_COND_V(!image_groups.has(p_export_group),IMAGE_ACTION_NONE);
  899. return image_groups[p_export_group].action;
  900. }
  901. void EditorImportExport::image_export_group_set_make_atlas(const StringName& p_export_group,bool p_make){
  902. ERR_FAIL_COND(!image_groups.has(p_export_group));
  903. image_groups[p_export_group].make_atlas=p_make;
  904. }
  905. bool EditorImportExport::image_export_group_get_make_atlas(const StringName& p_export_group) const{
  906. ERR_FAIL_COND_V(!image_groups.has(p_export_group),false);
  907. return image_groups[p_export_group].make_atlas;
  908. }
  909. void EditorImportExport::image_export_group_set_shrink(const StringName& p_export_group,int p_amount){
  910. ERR_FAIL_COND(!image_groups.has(p_export_group));
  911. image_groups[p_export_group].shrink=p_amount;
  912. }
  913. int EditorImportExport::image_export_group_get_shrink(const StringName& p_export_group) const{
  914. ERR_FAIL_COND_V(!image_groups.has(p_export_group),1);
  915. return image_groups[p_export_group].shrink;
  916. }
  917. void EditorImportExport::image_export_group_set_lossy_quality(const StringName& p_export_group,float p_amount){
  918. ERR_FAIL_COND(!image_groups.has(p_export_group));
  919. image_groups[p_export_group].lossy_quality=p_amount;
  920. }
  921. float EditorImportExport::image_export_group_get_lossy_quality(const StringName& p_export_group) const{
  922. ERR_FAIL_COND_V(!image_groups.has(p_export_group),1);
  923. return image_groups[p_export_group].lossy_quality;
  924. }
  925. StringName EditorImportExport::image_get_export_group(const StringName& p_image) const {
  926. if (image_group_files.has(p_image))
  927. return image_group_files[p_image];
  928. else
  929. return StringName();
  930. }
  931. void EditorImportExport::image_add_to_export_group(const StringName& p_image,const StringName& p_export_group) {
  932. bool emptygroup = String(p_export_group)==String();
  933. ERR_FAIL_COND(!emptygroup && !image_groups.has(p_export_group));
  934. if (emptygroup)
  935. image_group_files.erase(p_image);
  936. else
  937. image_group_files[p_image]=p_export_group;
  938. }
  939. void EditorImportExport::image_export_get_images_in_group(const StringName& p_group,List<StringName> *r_images) const {
  940. for (Map<StringName,StringName>::Element *E=image_group_files.front();E;E=E->next()) {
  941. if (p_group==E->get())
  942. r_images->push_back(E->key());
  943. }
  944. }
  945. void EditorImportExport::load_config() {
  946. Ref<ConfigFile> cf = memnew( ConfigFile );
  947. Error err = cf->load("res://export.cfg");
  948. if (err!=OK)
  949. return; //no export config to be loaded!
  950. export_custom_filter=cf->get_value("export_filter","filter");
  951. String t=cf->get_value("export_filter","type");
  952. if (t=="selected")
  953. export_filter=EXPORT_SELECTED;
  954. else if (t=="resources")
  955. export_filter=EXPORT_RESOURCES;
  956. else if (t=="all")
  957. export_filter=EXPORT_ALL;
  958. if (cf->has_section("convert_images")) {
  959. String ci = "convert_images";
  960. String action = cf->get_value(ci,"action");
  961. if (action=="none")
  962. image_action=IMAGE_ACTION_NONE;
  963. else if (action=="compress_ram")
  964. image_action=IMAGE_ACTION_COMPRESS_RAM;
  965. else if (action=="compress_disk")
  966. image_action=IMAGE_ACTION_COMPRESS_DISK;
  967. image_action_compress_quality = cf->get_value(ci,"compress_quality");
  968. if (cf->has_section_key(ci,"shrink"))
  969. image_shrink = cf->get_value(ci,"shrink");
  970. else
  971. image_shrink=1;
  972. String formats=cf->get_value(ci,"formats");
  973. Vector<String> f = formats.split(",");
  974. image_formats.clear();
  975. for(int i=0;i<f.size();i++) {
  976. image_formats.insert(f[i].strip_edges());
  977. }
  978. }
  979. if (cf->has_section("export_filter_files")) {
  980. String eff = "export_filter_files";
  981. List<String> k;
  982. cf->get_section_keys(eff,&k);
  983. for(List<String>::Element *E=k.front();E;E=E->next()) {
  984. String val = cf->get_value(eff,E->get());
  985. if (val=="copy") {
  986. files[E->get()]=ACTION_COPY;
  987. } else if (val=="bundle") {
  988. files[E->get()]=ACTION_BUNDLE;
  989. }
  990. }
  991. }
  992. List<String> sect;
  993. cf->get_sections(&sect);
  994. for(List<String>::Element *E=sect.front();E;E=E->next()) {
  995. String s = E->get();
  996. if (!s.begins_with("platform:"))
  997. continue;
  998. String p = s.substr(s.find(":")+1,s.length());
  999. if (!exporters.has(p))
  1000. continue;
  1001. Ref<EditorExportPlatform> ep = exporters[p];
  1002. if (!ep.is_valid()) {
  1003. continue;
  1004. }
  1005. List<String> keys;
  1006. cf->get_section_keys(s,&keys);
  1007. for(List<String>::Element *F=keys.front();F;F=F->next()) {
  1008. ep->set(F->get(),cf->get_value(s,F->get()));
  1009. }
  1010. }
  1011. //save image groups
  1012. if (cf->has_section("image_groups")) {
  1013. sect.clear();
  1014. cf->get_section_keys("image_groups",&sect);
  1015. for(List<String>::Element *E=sect.front();E;E=E->next()) {
  1016. Dictionary d = cf->get_value("image_groups",E->get());
  1017. ImageGroup g;
  1018. g.action=IMAGE_ACTION_NONE;
  1019. g.make_atlas=false;
  1020. g.lossy_quality=0.7;
  1021. g.shrink=1;
  1022. if (d.has("action")) {
  1023. String action=d["action"];
  1024. if (action=="compress_ram")
  1025. g.action=IMAGE_ACTION_COMPRESS_RAM;
  1026. else if (action=="compress_disk")
  1027. g.action=IMAGE_ACTION_COMPRESS_DISK;
  1028. }
  1029. if (d.has("atlas"))
  1030. g.make_atlas=d["atlas"];
  1031. if (d.has("lossy_quality"))
  1032. g.lossy_quality=d["lossy_quality"];
  1033. if (d.has("shrink")) {
  1034. g.shrink=d["shrink"];
  1035. g.shrink=CLAMP(g.shrink,1,8);
  1036. }
  1037. image_groups[E->get()]=g;
  1038. }
  1039. if (cf->has_section_key("image_group_files","files")) {
  1040. Vector<String> sa=cf->get_value("image_group_files","files");
  1041. if (sa.size()%2==0) {
  1042. for(int i=0;i<sa.size();i+=2) {
  1043. image_group_files[sa[i]]=sa[i+1];
  1044. }
  1045. }
  1046. }
  1047. }
  1048. if (cf->has_section("script")) {
  1049. if (cf->has_section_key("script","action")) {
  1050. String action = cf->get_value("script","action");
  1051. if (action=="compile")
  1052. script_action=SCRIPT_ACTION_COMPILE;
  1053. else if (action=="encrypt")
  1054. script_action=SCRIPT_ACTION_ENCRYPT;
  1055. else
  1056. script_action=SCRIPT_ACTION_NONE;
  1057. }
  1058. if (cf->has_section_key("script","encrypt_key")) {
  1059. script_key = cf->get_value("script","encrypt_key");
  1060. }
  1061. }
  1062. }
  1063. void EditorImportExport::save_config() {
  1064. Ref<ConfigFile> cf = memnew( ConfigFile );
  1065. switch(export_filter) {
  1066. case EXPORT_SELECTED: cf->set_value("export_filter","type","selected"); break;
  1067. case EXPORT_RESOURCES: cf->set_value("export_filter","type","resources"); break;
  1068. case EXPORT_ALL: cf->set_value("export_filter","type","all"); break;
  1069. }
  1070. cf->set_value("export_filter","filter",export_custom_filter);
  1071. String file_action_section = "export_filter_files";
  1072. for (Map<StringName,FileAction>::Element *E=files.front();E;E=E->next()) {
  1073. String f=E->key();
  1074. String a;
  1075. switch (E->get()) {
  1076. case ACTION_NONE: {}
  1077. case ACTION_COPY: a="copy"; break;
  1078. case ACTION_BUNDLE: a="bundle"; break;
  1079. }
  1080. cf->set_value(file_action_section,f,a);
  1081. }
  1082. for (Map<StringName,Ref<EditorExportPlatform> >::Element *E=exporters.front();E;E=E->next()) {
  1083. String pname = "platform:"+String(E->key());
  1084. Ref<EditorExportPlatform> ep = E->get();
  1085. List<PropertyInfo> pl;
  1086. ep->get_property_list(&pl);
  1087. for (List<PropertyInfo>::Element *F=pl.front();F;F=F->next()) {
  1088. cf->set_value(pname,F->get().name,ep->get(F->get().name));
  1089. }
  1090. }
  1091. switch(image_action) {
  1092. case IMAGE_ACTION_NONE: cf->set_value("convert_images","action","none"); break;
  1093. case IMAGE_ACTION_COMPRESS_RAM: cf->set_value("convert_images","action","compress_ram"); break;
  1094. case IMAGE_ACTION_COMPRESS_DISK: cf->set_value("convert_images","action","compress_disk"); break;
  1095. }
  1096. cf->set_value("convert_images","shrink",image_shrink);
  1097. cf->set_value("convert_images","compress_quality",image_action_compress_quality);
  1098. String formats;
  1099. for(Set<String>::Element *E=image_formats.front();E;E=E->next()) {
  1100. if (E!=image_formats.front())
  1101. formats+=",";
  1102. formats+=E->get();
  1103. }
  1104. cf->set_value("convert_images","formats",formats);
  1105. //save image groups
  1106. for(Map<StringName,ImageGroup>::Element *E=image_groups.front();E;E=E->next()) {
  1107. Dictionary d;
  1108. switch(E->get().action) {
  1109. case IMAGE_ACTION_NONE: d["action"]="default"; break;
  1110. case IMAGE_ACTION_COMPRESS_RAM: d["action"]="compress_ram"; break;
  1111. case IMAGE_ACTION_COMPRESS_DISK: d["action"]="compress_disk"; break;
  1112. }
  1113. d["atlas"]=E->get().make_atlas;
  1114. d["shrink"]=E->get().shrink;
  1115. d["lossy_quality"]=E->get().lossy_quality;
  1116. cf->set_value("image_groups",E->key(),d);
  1117. }
  1118. if (image_groups.size() && image_group_files.size()){
  1119. Vector<String> igfsave;
  1120. igfsave.resize(image_group_files.size()*2);
  1121. int idx=0;
  1122. for (Map<StringName,StringName>::Element *E=image_group_files.front();E;E=E->next()) {
  1123. igfsave[idx++]=E->key();
  1124. igfsave[idx++]=E->get();
  1125. }
  1126. cf->set_value("image_group_files","files",igfsave);
  1127. }
  1128. switch(script_action) {
  1129. case SCRIPT_ACTION_NONE: cf->set_value("script","action","none"); break;
  1130. case SCRIPT_ACTION_COMPILE: cf->set_value("script","action","compile"); break;
  1131. case SCRIPT_ACTION_ENCRYPT: cf->set_value("script","action","encrypt"); break;
  1132. }
  1133. cf->set_value("script","encrypt_key",script_key);
  1134. cf->save("res://export.cfg");
  1135. }
  1136. void EditorImportExport::script_set_action(ScriptAction p_action) {
  1137. script_action=p_action;
  1138. }
  1139. EditorImportExport::ScriptAction EditorImportExport::script_get_action() const{
  1140. return script_action;
  1141. }
  1142. void EditorImportExport::script_set_encryption_key(const String& p_key){
  1143. script_key=p_key;
  1144. }
  1145. String EditorImportExport::script_get_encryption_key() const{
  1146. return script_key;
  1147. }
  1148. void EditorImportExport::_bind_methods() {
  1149. ObjectTypeDB::bind_method(_MD("image_export_group_create"),&EditorImportExport::image_export_group_create);
  1150. ObjectTypeDB::bind_method(_MD("image_export_group_remove"),&EditorImportExport::image_export_group_remove);
  1151. ObjectTypeDB::bind_method(_MD("image_export_group_set_image_action"),&EditorImportExport::image_export_group_set_image_action);
  1152. ObjectTypeDB::bind_method(_MD("image_export_group_set_make_atlas"),&EditorImportExport::image_export_group_set_make_atlas);
  1153. ObjectTypeDB::bind_method(_MD("image_export_group_set_shrink"),&EditorImportExport::image_export_group_set_shrink);
  1154. ObjectTypeDB::bind_method(_MD("image_export_group_get_image_action"),&EditorImportExport::image_export_group_get_image_action);
  1155. ObjectTypeDB::bind_method(_MD("image_export_group_get_make_atlas"),&EditorImportExport::image_export_group_get_make_atlas);
  1156. ObjectTypeDB::bind_method(_MD("image_export_group_get_shrink"),&EditorImportExport::image_export_group_get_shrink);
  1157. ObjectTypeDB::bind_method(_MD("image_add_to_export_group"),&EditorImportExport::image_add_to_export_group);
  1158. ObjectTypeDB::bind_method(_MD("script_set_action"),&EditorImportExport::script_set_action);
  1159. ObjectTypeDB::bind_method(_MD("script_set_encryption_key"),&EditorImportExport::script_set_encryption_key);
  1160. ObjectTypeDB::bind_method(_MD("script_get_action"),&EditorImportExport::script_get_action);
  1161. ObjectTypeDB::bind_method(_MD("script_get_encryption_key"),&EditorImportExport::script_get_encryption_key);
  1162. }
  1163. EditorImportExport::EditorImportExport() {
  1164. export_filter=EXPORT_RESOURCES;
  1165. singleton=this;
  1166. image_action=IMAGE_ACTION_NONE;
  1167. image_action_compress_quality=0.7;
  1168. image_formats.insert("png");
  1169. image_shrink=1;
  1170. script_action=SCRIPT_ACTION_COMPILE;
  1171. }