|
2 | 2 | # -*- coding: utf-8 -*-
|
3 | 3 | import time
|
4 | 4 | import re
|
| 5 | +import copy |
5 | 6 | import cv2
|
6 | 7 | import numpy as np
|
7 | 8 | import dearpygui.dearpygui as dpg
|
@@ -333,23 +334,39 @@ def close(self, node_id):
|
333 | 334 |
|
334 | 335 | def get_setting_dict(self, node_id):
|
335 | 336 | 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' |
337 | 340 |
|
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) |
339 | 344 |
|
340 | 345 | pos = dpg.get_item_pos(tag_node_name)
|
341 | 346 |
|
342 | 347 | setting_dict = {}
|
343 | 348 | setting_dict['ver'] = self._ver
|
344 | 349 | 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 |
346 | 353 |
|
347 | 354 | return setting_dict
|
348 | 355 |
|
349 | 356 | def set_setting_dict(self, node_id, setting_dict):
|
350 | 357 | 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' |
354 | 361 |
|
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