Files
armorpaint/paint/sources/ui/tab_materials.c
T
2026-08-21 22:40:25 +02:00

450 lines
16 KiB
C

#include "../global.h"
i32 _tab_materials_draw_slots;
i32 tab_materials_drag_pos = -1;
void tab_materials_button_nodes() {
if (ui_button(tr("Nodes"), UI_ALIGN_CENTER, "")) {
ui_base_show_material_nodes();
}
else if (g_ui->is_hovered) {
ui_tooltip(string_tmp("%s (%s)", tr("Show Node Editor"), (char *)any_map_get(g_keymap, "toggle_node_editor")));
}
}
void tab_materials_draw_slots_update_fill_layers(void *_) {
layers_update_fill_layers();
}
void tab_materials_update_material() {
ui_header_handle->redraws = 2;
ui_nodes_hwnd->redraws = 2;
ui_nodes_group_stack = any_array_create_from_raw((void *[]){}, 0);
make_material_parse_paint_material(true);
util_render_make_material_preview();
bool decal = context_is_decal();
if (decal) {
util_render_make_decal_preview();
}
base_update_workflow_nodes();
}
void tab_materials_draw_slots_duplicate(void *_) {
i32 i = _tab_materials_draw_slots;
g_context->material = slot_material_create(g_project->_->materials->buffer[0]->data, NULL);
any_array_push(g_project->_->materials, g_context->material);
ui_node_canvas_t *cloned = util_clone_canvas(g_project->_->materials->buffer[i]->canvas);
g_context->material->canvas = cloned;
tab_materials_update_material();
history_duplicate_material();
}
void tab_materials_update_material_pointers(ui_node_t_array_t *nodes, i32 i) {
for (i32 i = 0; i < nodes->length; ++i) {
ui_node_t *n = nodes->buffer[i];
if (string_equals(n->type, "MATERIAL")) {
if (n->buttons->buffer[0]->default_value->buffer[0] == i) {
n->buttons->buffer[0]->default_value->buffer[0] = 9999; // Material deleted
}
else if (n->buttons->buffer[0]->default_value->buffer[0] > i) {
n->buttons->buffer[0]->default_value->buffer[0]--; // Offset by deleted material
}
}
}
}
void tab_materials_delete_material(slot_material_t *m) {
i32 i = array_index_of(g_project->_->materials, m);
for (i32 i = 0; i < g_project->_->layers->length; ++i) {
slot_layer_t *l = g_project->_->layers->buffer[i];
if (l->fill_material == m) {
l->fill_material = NULL;
}
}
history_delete_material();
context_select_material(i == g_project->_->materials->length - 1 ? i - 1 : i + 1);
array_splice(g_project->_->materials, i, 1);
tab_meshes_on_material_deleted(i);
ui_base_hwnds->buffer[1]->redraws = 2;
for (i32 i = 0; i < g_project->_->materials->length; ++i) {
slot_material_t *m = g_project->_->materials->buffer[i];
tab_materials_update_material_pointers(m->canvas->nodes, i);
}
for (i32 i = 0; i < m->canvas->nodes->length; ++i) {
ui_node_t *n = m->canvas->nodes->buffer[i];
ui_viewnodes_on_node_remove(n);
}
}
void tab_materials_draw_slots_menu() {
i32 i = _tab_materials_draw_slots;
slot_material_t *m = g_project->_->materials->buffer[i];
if (ui_menu_button(tr("To Fill Layer"), "", ICON_SPHERE)) {
context_select_material(i);
layers_create_fill_layer(UV_TYPE_UVMAP, mat4_nan(), -1);
}
if (ui_menu_button(tr("Export"), "", ICON_EXPORT)) {
context_select_material(i);
box_export_show_material();
}
if (ui_menu_button(tr("Bake"), "", ICON_BAKE)) {
context_select_material(i);
box_export_show_bake_material();
}
if (ui_menu_button(tr("Duplicate"), "ctrl+d", ICON_DUPLICATE)) {
sys_notify_on_next_frame(&tab_materials_draw_slots_duplicate, NULL);
}
if (g_project->_->materials->length > 1 && ui_menu_button(tr("Delete"), "delete", ICON_DELETE)) {
tab_materials_delete_material(m);
}
if (g_config->experimental && ui_menu_button(tr("Inspect as Script"), "", ICON_SEARCH)) {
node_shader_dump_to_script = true;
context_select_material(i);
}
ui_handle_t *base_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *opac_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *nor_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *occ_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *rough_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *met_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *height_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *emis_handle = ui_nest(ui_handle(__ID__), m->id);
ui_handle_t *subs_handle = ui_nest(ui_handle(__ID__), m->id);
base_handle->b = m->paint_base;
opac_handle->b = m->paint_opac;
nor_handle->b = m->paint_nor;
occ_handle->b = m->paint_occ;
rough_handle->b = m->paint_rough;
met_handle->b = m->paint_met;
height_handle->b = m->paint_height;
emis_handle->b = m->paint_emis;
subs_handle->b = m->paint_subs;
ui_menu_separator();
ui_menu_align();
ui_menu_label(tr("Opacity Mode"), NULL);
ui_menu_align();
ui_handle_t *opac_mode_handle = ui_handle(__ID__);
opac_mode_handle->i = m->paint_opac_mode;
string_array_t *opac_mode_items = any_array_create_from_raw(
(void *[]){
tr("Alpha"),
tr("Translucency"),
},
2);
m->paint_opac_mode = ui_inline_radio(opac_mode_handle, opac_mode_items, UI_ALIGN_LEFT);
ui_menu_separator();
ui_menu_align();
ui_menu_label(tr("Channels"), NULL);
ui_menu_align();
ui_row2();
m->paint_base = ui_check(base_handle, tr("Base Color"), "");
m->paint_opac = ui_check(opac_handle, tr("Opacity"), "");
if (g_config->workflow == WORKFLOW_PBR) {
ui_row2();
m->paint_nor = ui_check(nor_handle, tr("Normal"), "");
m->paint_height = ui_check(height_handle, tr("Height"), "");
ui_row2();
m->paint_rough = ui_check(rough_handle, tr("Roughness"), "");
m->paint_met = ui_check(met_handle, tr("Metallic"), "");
ui_row2();
m->paint_emis = ui_check(emis_handle, tr("Emission"), "");
m->paint_subs = ui_check(subs_handle, tr("Subsurface"), "");
m->paint_occ = ui_check(occ_handle, tr("Occlusion"), "");
}
if (base_handle->changed || opac_handle->changed || nor_handle->changed || occ_handle->changed || rough_handle->changed || met_handle->changed ||
height_handle->changed || emis_handle->changed || subs_handle->changed || opac_mode_handle->changed) {
make_material_parse_paint_material(true);
if (opac_mode_handle->changed) {
sys_notify_on_next_frame(util_render_make_material_preview, NULL);
}
ui_menu_keep_open = true;
}
}
void tab_materials_draw_slots(bool mini) {
i32 slotw = math_floor(51 * UI_SCALE() + g_config->window_scale * 2);
i32 num = math_floor(g_ui->_window_w / (float)slotw);
if (num == 0) {
return;
}
bool drag_pos_set = false;
f32 uix = 0.0;
f32 uiy = 0.0;
i32 imgw_val = math_floor(50 * UI_SCALE());
for (i32 row = 0; row < math_floor(math_ceil(g_project->_->materials->length / (float)num)); ++row) {
i32 mult = g_config->show_asset_names ? 2 : 1;
f32_array_t *ar = f32_array_create_from_raw_tmp(NULL, num * mult);
for (i32 i = 0; i < num * mult; ++i) {
ar->buffer[i] = 1 / (float)num;
}
ui_row(ar);
g_ui->_x += 2;
f32 off = g_config->show_asset_names ? UI_ELEMENT_OFFSET() * 10.0 : 6;
if (row > 0) {
g_ui->_y += off;
}
for (i32 j = 0; j < num; ++j) {
i32 imgw = math_floor(50 * UI_SCALE());
i32 i = j + row * num;
if (i >= g_project->_->materials->length) {
ui_end_element_of_size(imgw);
if (g_config->show_asset_names) {
ui_end_element_of_size(0);
}
continue;
}
gpu_texture_t *img = UI_SCALE() > 1 ? g_project->_->materials->buffer[i]->image : g_project->_->materials->buffer[i]->image_icon;
gpu_texture_t *img_full = g_project->_->materials->buffer[i]->image;
// Highligh selected
if (g_context->material == g_project->_->materials->buffer[i]) {
if (mini) {
f32 w = g_ui->_w / (float)UI_SCALE();
ui_rect(0, -2, w - 2, w - 4, g_theme->HIGHLIGHT_COL, 3);
}
else {
i32 off = row % 2 == 1 ? 1 : 0;
i32 w = 50 + math_floor(g_config->window_scale * 2);
ui_fill(-1, -2, w + 3, 2, g_theme->HIGHLIGHT_COL);
ui_fill(-1, w - off, w + 3, 2 + off, g_theme->HIGHLIGHT_COL);
ui_fill(-1, -2, 2, w + 3, g_theme->HIGHLIGHT_COL);
ui_fill(w + 1, -2, 2, w + 4, g_theme->HIGHLIGHT_COL);
}
}
// Draw material icon
uix = g_ui->_x;
uiy = g_ui->_y;
i32 tile = UI_SCALE() > 1 ? 100 : 50;
f32 imgh = mini ? ui_sidebar_default_w_mini * 0.85 * UI_SCALE() : 50 * UI_SCALE();
if (base_drag_material != NULL && tab_materials_drag_pos == i) {
ui_fill(-1, -2, 2, imgw_val + 4, g_theme->HIGHLIGHT_COL);
}
ui_state_t state = g_project->_->materials->buffer[i]->preview_ready
? ui_image(img, 0xffffffff, imgh)
: ui_sub_image(resource_get("icons.k"), 0xffffffff, -1.0, tile, tile, tile, tile);
if (state == UI_STATE_HOVERED && base_drag_material != NULL) {
tab_materials_drag_pos = (mouse_x > uix + g_ui->_window_x + imgw_val / 2.0) ? i + 1 : i;
drag_pos_set = true;
}
// Draw material numbers when selecting a material via keyboard shortcut
bool is_typing = g_ui->is_typing;
if (!is_typing) {
if (i < 9 && operator_shortcut(any_map_get(g_keymap, "select_material"), SHORTCUT_TYPE_DOWN)) {
char *number = i32_to_string(i + 1);
i32 width = draw_string_width(g_font, g_ui->font_size, number) + 10;
i32 height = draw_font_height(g_font, g_ui->font_size);
draw_set_color(g_theme->TEXT_COL);
draw_filled_rect(uix, uiy, width, height);
draw_set_color(g_theme->BUTTON_COL);
draw_string(number, uix + 5, uiy);
}
}
// Select material
if (state == UI_STATE_STARTED && g_ui->input_y > g_ui->_window_y) {
if (g_context->material != g_project->_->materials->buffer[i]) {
context_select_material(i);
if (g_context->tool == TOOL_TYPE_MATERIAL) {
sys_notify_on_next_frame(&tab_materials_draw_slots_update_fill_layers, NULL);
}
}
base_drag_off_x = -(mouse_x - uix - g_ui->_window_x - 3);
base_drag_off_y = -(mouse_y - uiy - g_ui->_window_y + 1);
base_drag_material = g_context->material;
// Double click to show nodes
if (sys_time() - g_context->select_time < 0.2) {
ui_base_show_material_nodes();
base_drag_material = NULL;
base_is_dragging = false;
}
g_context->select_time = sys_time();
}
// Context menu
if (g_ui->is_hovered && g_ui->input_released_r) {
context_select_material(i);
_tab_materials_draw_slots = i;
ui_menu_draw(&tab_materials_draw_slots_menu, -1, -1);
}
if (g_ui->is_hovered) {
ui_tooltip_image(img_full, 0);
if (i < 9) {
i32 i1 = i + 1;
ui_tooltip(
string_tmp("%s - (%s %d)", g_project->_->materials->buffer[i]->canvas->name, (char *)any_map_get(g_keymap, "select_material"), i1));
}
else {
ui_tooltip(g_project->_->materials->buffer[i]->canvas->name);
}
}
if (g_config->show_asset_names) {
g_ui->_x = uix;
g_ui->_y += slotw * 0.9;
ui_text(g_project->_->materials->buffer[i]->canvas->name, UI_ALIGN_CENTER, 0x00000000);
if (g_ui->is_hovered) {
if (i < 9) {
i32 i1 = i + 1;
ui_tooltip(
string_tmp("%s - (%s %d)", g_project->_->materials->buffer[i]->canvas->name, (char *)any_map_get(g_keymap, "select_material"), i1));
}
else {
ui_tooltip(g_project->_->materials->buffer[i]->canvas->name);
}
}
g_ui->_y -= slotw * 0.9;
if (i == g_project->_->materials->length - 1) {
g_ui->_y += j == num - 1 ? imgw : imgw + UI_ELEMENT_H() + UI_ELEMENT_OFFSET();
}
}
}
g_ui->_y += mini ? 0 : 6;
}
if (base_drag_material != NULL && tab_materials_drag_pos == g_project->_->materials->length) {
g_ui->_x = uix;
g_ui->_y = uiy;
ui_fill(imgw_val + 1, -2, 2, imgw_val + 4, g_theme->HIGHLIGHT_COL);
}
if (!drag_pos_set) {
tab_materials_drag_pos = -1;
}
bool in_focus = g_ui->input_x > g_ui->_window_x && g_ui->input_x < g_ui->_window_x + g_ui->_window_w && g_ui->input_y > g_ui->_window_y &&
g_ui->input_y < g_ui->_window_y + g_ui->_window_h;
if (in_focus && g_ui->is_delete_down && g_project->_->materials->length > 1) {
g_ui->is_delete_down = false;
tab_materials_delete_material(g_context->material);
}
if (in_focus && g_ui->is_ctrl_down && g_ui->is_key_pressed && g_ui->key_code == KEY_CODE_D) {
_tab_materials_draw_slots = array_index_of(g_project->_->materials, g_context->material);
sys_notify_on_next_frame(&tab_materials_draw_slots_duplicate, NULL);
}
if (in_focus) {
i32 i = array_index_of(g_project->_->materials, g_context->material);
if (g_ui->is_key_pressed && g_ui->key_code == KEY_CODE_UP) {
if (i > 0) {
context_select_material(i - 1);
}
}
if (g_ui->is_key_pressed && g_ui->key_code == KEY_CODE_DOWN) {
if (i < g_project->_->materials->length - 1) {
context_select_material(i + 1);
}
}
}
}
void tab_materials_button_new_on_next_frame(void *_) {
g_context->material = slot_material_create(g_project->_->materials->buffer[0]->data, NULL);
any_array_push(g_project->_->materials, g_context->material);
tab_materials_update_material();
history_new_material();
}
void tab_materials_button_new(char *text) {
if (ui_icon_button(text, ICON_PLUS, UI_ALIGN_CENTER)) {
sys_notify_on_next_frame(&tab_materials_button_new_on_next_frame, NULL);
}
}
void tab_materials_draw_mini(ui_handle_t *htab) {
ui_set_hovered_tab_name(tr("Materials"));
ui_begin_sticky();
ui_separator(5, true);
tab_materials_button_nodes();
tab_materials_button_new("");
ui_end_sticky();
ui_separator(3, false);
tab_materials_draw_slots(true);
}
void tab_materials_draw_full(ui_handle_t *htab) {
if (ui_tab(htab, tr("Materials"), false, -1, false)) {
ui_begin_sticky();
f32_array_t *row = f32_array_create_from_raw_tmp(
(f32[]){
-70,
-70,
-70,
},
3);
ui_row(row);
tab_materials_button_new(tr("New"));
if (ui_icon_button(tr("Import"), ICON_IMPORT, UI_ALIGN_CENTER)) {
project_import_material();
}
tab_materials_button_nodes();
ui_end_sticky();
ui_separator(3, false);
tab_materials_draw_slots(false);
}
}
void tab_materials_draw(ui_handle_t *htab) {
bool mini = g_ui->_window_w <= ui_sidebar_w_mini;
mini ? tab_materials_draw_mini(htab) : tab_materials_draw_full(htab);
}
void tab_materials_accept_material_drop(slot_material_t *material) {
if (tab_materials_drag_pos == -1) {
return;
}
i32 mat_pos = array_index_of(g_project->_->materials, material);
if (mat_pos != -1 && math_abs(mat_pos - tab_materials_drag_pos) > 0) {
array_remove(g_project->_->materials, material);
i32 new_pos = tab_materials_drag_pos - mat_pos > 0 ? tab_materials_drag_pos - 1 : tab_materials_drag_pos;
array_insert(g_project->_->materials, new_pos, material);
tab_meshes_on_material_reordered(mat_pos, new_pos);
}
}
void tab_materials_accept_swatch_drop(swatch_color_t *swatch) {
g_context->material = slot_material_create(g_project->_->materials->buffer[0]->data, NULL);
for (i32 i = 0; i < g_context->material->canvas->nodes->length; ++i) {
ui_node_t *node = g_context->material->canvas->nodes->buffer[i];
if (string_equals(node->type, "RGB")) {
node->outputs->buffer[0]->default_value = f32_array_create_xyzw(color_get_rb(swatch->base) / 255.0, color_get_gb(swatch->base) / 255.0,
color_get_bb(swatch->base) / 255.0, color_get_ab(swatch->base) / 255.0);
}
else if (string_equals(node->type, "OUTPUT_MATERIAL_PBR")) {
node->inputs->buffer[1]->default_value->buffer[0] = swatch->opacity;
node->inputs->buffer[2]->default_value->buffer[0] = swatch->occlusion;
node->inputs->buffer[3]->default_value->buffer[0] = swatch->roughness;
node->inputs->buffer[4]->default_value->buffer[0] = swatch->metallic;
node->inputs->buffer[7]->default_value->buffer[0] = swatch->height;
}
}
any_array_push(g_project->_->materials, g_context->material);
tab_materials_update_material();
history_new_material();
}