Skip to content

Commit 1ae5be0

Browse files
committed
fix: issues with color balance node
1 parent 94387b7 commit 1ae5be0

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

compositor/operator.py

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -102,17 +102,26 @@ def _set_color_balance_settings(self, node: CompositorNodeColorBalance
102102
"""
103103
if node.correction_method == 'LIFT_GAMMA_GAIN':
104104
lst = [NTPNodeSetting("correction_method", ST.ENUM),
105-
NTPNodeSetting("gain", ST.COLOR),
106-
NTPNodeSetting("gamma", ST.COLOR),
107-
NTPNodeSetting("lift", ST.COLOR)]
105+
NTPNodeSetting("gain", ST.VEC3, max_version_=(3, 5, 0)),
106+
NTPNodeSetting("gain", ST.COLOR, min_version_=(3, 5, 0)),
107+
NTPNodeSetting("gamma", ST.VEC3, max_version_=(3, 5, 0)),
108+
NTPNodeSetting("gamma", ST.COLOR, min_version_=(3, 5, 0)),
109+
NTPNodeSetting("lift", ST.VEC3, max_version_=(3, 5, 0)),
110+
NTPNodeSetting("lift", ST.COLOR, min_version_=(3, 5, 0))]
108111
else:
109112
lst = [NTPNodeSetting("correction_method", ST.ENUM),
110-
NTPNodeSetting("offset", ST.COLOR),
111-
NTPNodeSetting("offset_basis", ST.FLOAT),
112-
NTPNodeSetting("power", ST.COLOR),
113-
NTPNodeSetting("slope", ST.COLOR)]
114-
115-
self._node_infos['CompositorNodeColorBalance'].attributes_ = lst
113+
NTPNodeSetting("offset", ST.VEC3, max_version_=(3, 5, 0)),
114+
NTPNodeSetting("offset", ST.COLOR, min_version_=(3, 5, 0)),
115+
NTPNodeSetting("offset_basis", ST.FLOAT),
116+
NTPNodeSetting("power", ST.VEC3, max_version_=(3, 5, 0)),
117+
NTPNodeSetting("power", ST.COLOR, min_version_=(3, 5, 0)),
118+
NTPNodeSetting("slope", ST.VEC3, max_version_=(3, 5, 0)),
119+
NTPNodeSetting("slope", ST.COLOR, min_version_=(3, 5, 0))]
120+
121+
color_balance_info = self._node_infos['CompositorNodeColorBalance']
122+
self._node_infos['CompositorNodeColorBalance'] = color_balance_info._replace(attributes_ = lst)
123+
for setting in self._node_infos['CompositorNodeColorBalance'].attributes_:
124+
print(setting.name_)
116125

117126
def _process_node(self, node: Node, ntp_nt: NTP_NodeTree):
118127
"""

0 commit comments

Comments
 (0)