Skip to content

Commit

Permalink
Bake blend material on import
Browse files Browse the repository at this point in the history
  • Loading branch information
luboslenco committed Jan 6, 2025
1 parent 04fc2d7 commit 05be043
Show file tree
Hide file tree
Showing 2 changed files with 80 additions and 2 deletions.
81 changes: 79 additions & 2 deletions base/sources/import_blend_material.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,83 @@

let _import_blend_material_path: string;

function import_blend_material_run(path: string) {
let b: buffer_t = data_get_blob(path);

data_delete_blob(path);
_import_blend_material_path = path;

ui_box_show_custom(function (ui: ui_t) {
if (ui_tab(ui_handle(__ID__), tr("Import Material"))) {
import_blend_mesh_ui();

ui_row2();
if (ui_button(tr("Cancel"))) {
ui_box_hide();
}
if (ui_button(tr("Import")) || ui.is_return_down) {

ui_box_hide();

if (config_raw.blender == null || config_raw.blender == "") {
console_error(tr("Blender executable path not set"));
return;
}

_import_blend_material();
}
}
});
}

function _import_blend_material() {

let current: image_t = _g2_current;
let g2_in_use: bool = _g2_in_use;
if (g2_in_use) g2_end();

console_toast(tr("Baking material"));

if (g2_in_use) g2_begin(current);

app_notify_on_init(function () {

let save: string;
if (path_is_protected()) {
save = iron_save_path();
}
else {
save = path_data();
}
save += "blender";
file_create_directory(save);

let py: string = "\
import bpy;\n\
bpy.context.scene.render.engine = 'CYCLES'\n\
bpy.context.scene.cycles.samples = 4\n\
bpy.ops.mesh.primitive_plane_add()\n\
plane = bpy.context.selected_objects[0]\n\
img = bpy.data.images.new('bake', 2048, 2048)\n\
img.file_format = 'JPEG'\n\
for mat in bpy.data.materials:\n\
if mat.name == 'Dots Stroke':\n\
continue\n\
plane.data.materials.append(mat)\n\
node = mat.node_tree.nodes.new(type='ShaderNodeTexImage')\n\
node.image = img\n\
mat.node_tree.nodes.active = node\n\
bpy.ops.object.bake(type='DIFFUSE')\n\
img.filepath_raw = '" + save + "/blender_base.jpg'\n\
img.save()\n\
bpy.ops.object.bake(type='ROUGHNESS')\n\
img.filepath_raw = '" + save + "/blender_rough.jpg'\n\
img.save()\n\
bpy.ops.object.bake(type='NORMAL')\n\
img.filepath_raw = '" + save + "/blender_nor.jpg'\n\
img.save()\n\
break\n\
";

iron_sys_command("\"" + config_raw.blender + "\" \"" + _import_blend_material_path + "\" -b --python-expr \"" + py + "\"");
import_folder_run(save);
});
}
1 change: 1 addition & 0 deletions base/sources/import_blend_mesh.ts
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ function import_blend_mesh_ui() {
if (ui_button("...")) {
ui_files_show("", false, false, function (path: string) {
config_raw.blender = path;
config_save();
});
}
}
Expand Down

0 comments on commit 05be043

Please sign in to comment.