Skip to content

Commit 85fdcac

Browse files
committed
#22 修正。
1 parent 4906392 commit 85fdcac

File tree

1 file changed

+24
-7
lines changed

1 file changed

+24
-7
lines changed

node/draw_node/node_image_alpha_blend.py

Lines changed: 24 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# -*- coding: utf-8 -*-
33
import time
44
import re
5+
import copy
56
import cv2
67
import numpy as np
78
import dearpygui.dearpygui as dpg
@@ -333,23 +334,39 @@ def close(self, node_id):
333334

334335
def get_setting_dict(self, node_id):
335336
tag_node_name = str(node_id) + ':' + self.node_tag
336-
input_value03_tag = tag_node_name + ':' + self.TYPE_INT + ':Input03Value'
337+
input_value03_tag = tag_node_name + ':' + self.TYPE_FLOAT + ':Input03Value'
338+
input_value04_tag = tag_node_name + ':' + self.TYPE_FLOAT + ':Input04Value'
339+
input_value05_tag = tag_node_name + ':' + self.TYPE_INT + ':Input05Value'
337340

338-
kernel_size = dpg_get_value(input_value03_tag)
341+
alpha_val = dpg_get_value(input_value03_tag)
342+
beta_val = dpg_get_value(input_value04_tag)
343+
gamma_val = dpg_get_value(input_value05_tag)
339344

340345
pos = dpg.get_item_pos(tag_node_name)
341346

342347
setting_dict = {}
343348
setting_dict['ver'] = self._ver
344349
setting_dict['pos'] = pos
345-
setting_dict[input_value03_tag] = kernel_size
350+
setting_dict[input_value03_tag] = alpha_val
351+
setting_dict[input_value04_tag] = beta_val
352+
setting_dict[input_value05_tag] = gamma_val
346353

347354
return setting_dict
348355

349356
def set_setting_dict(self, node_id, setting_dict):
350357
tag_node_name = str(node_id) + ':' + self.node_tag
351-
input_value03_tag = tag_node_name + ':' + self.TYPE_INT + ':Input02Value'
352-
353-
kernel_size = int(setting_dict[input_value03_tag])
358+
input_value03_tag = tag_node_name + ':' + self.TYPE_FLOAT + ':Input03Value'
359+
input_value04_tag = tag_node_name + ':' + self.TYPE_FLOAT + ':Input04Value'
360+
input_value05_tag = tag_node_name + ':' + self.TYPE_INT + ':Input05Value'
354361

355-
dpg_set_value(input_value03_tag, kernel_size)
362+
if input_value03_tag in setting_dict:
363+
alpha_val = float(setting_dict[input_value03_tag])
364+
dpg_set_value(input_value03_tag, alpha_val)
365+
366+
if input_value04_tag in setting_dict:
367+
beta_val = float(setting_dict[input_value04_tag])
368+
dpg_set_value(input_value04_tag, beta_val)
369+
370+
if input_value05_tag in setting_dict:
371+
gamma_val = int(setting_dict[input_value05_tag])
372+
dpg_set_value(input_value05_tag, gamma_val)

0 commit comments

Comments
 (0)