mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-25 05:14:40 +01:00
fixing nuke format updating in node graph
also making code compliance with flake8
This commit is contained in:
Jakub Jezek
parent
8356bf98e5
commit
ef71fc1716
1 changed files with 42 additions and 32 deletions
|
|
@ -2250,16 +2250,15 @@ Reopening Nuke should synchronize these paths and resolve any discrepancies.
|
|||
log.warning(msg)
|
||||
nuke.message(msg)
|
||||
return
|
||||
data = self._asset_entity["data"]
|
||||
|
||||
log.debug("__ asset data: `{}`".format(data))
|
||||
asset_data = self._asset_entity["data"]
|
||||
|
||||
missing_cols = []
|
||||
check_cols = ["fps", "frameStart", "frameEnd",
|
||||
"handleStart", "handleEnd"]
|
||||
|
||||
for col in check_cols:
|
||||
if col not in data:
|
||||
if col not in asset_data:
|
||||
missing_cols.append(col)
|
||||
|
||||
if len(missing_cols) > 0:
|
||||
|
|
@ -2271,12 +2270,12 @@ Reopening Nuke should synchronize these paths and resolve any discrepancies.
|
|||
return
|
||||
|
||||
# get handles values
|
||||
handle_start = data["handleStart"]
|
||||
handle_end = data["handleEnd"]
|
||||
handle_start = asset_data["handleStart"]
|
||||
handle_end = asset_data["handleEnd"]
|
||||
|
||||
fps = float(data["fps"])
|
||||
frame_start_handle = int(data["frameStart"]) - handle_start
|
||||
frame_end_handle = int(data["frameEnd"]) + handle_end
|
||||
fps = float(asset_data["fps"])
|
||||
frame_start_handle = int(asset_data["frameStart"]) - handle_start
|
||||
frame_end_handle = int(asset_data["frameEnd"]) + handle_end
|
||||
|
||||
self._root_node["lock_range"].setValue(False)
|
||||
self._root_node["fps"].setValue(fps)
|
||||
|
|
@ -2284,21 +2283,18 @@ Reopening Nuke should synchronize these paths and resolve any discrepancies.
|
|||
self._root_node["last_frame"].setValue(frame_end_handle)
|
||||
self._root_node["lock_range"].setValue(True)
|
||||
|
||||
# setting active viewers
|
||||
try:
|
||||
nuke.frame(int(data["frameStart"]))
|
||||
except Exception as e:
|
||||
log.warning("no viewer in scene: `{}`".format(e))
|
||||
# update node graph so knobs are updated
|
||||
update_node_graph()
|
||||
|
||||
range = '{0}-{1}'.format(
|
||||
int(data["frameStart"]),
|
||||
int(data["frameEnd"])
|
||||
frame_range = '{0}-{1}'.format(
|
||||
int(asset_data["frameStart"]),
|
||||
int(asset_data["frameEnd"])
|
||||
)
|
||||
|
||||
for node in nuke.allNodes(filter="Viewer"):
|
||||
node['frame_range'].setValue(range)
|
||||
node['frame_range'].setValue(frame_range)
|
||||
node['frame_range_lock'].setValue(True)
|
||||
node['frame_range'].setValue(range)
|
||||
node['frame_range'].setValue(frame_range)
|
||||
node['frame_range_lock'].setValue(True)
|
||||
|
||||
if not ASSIST:
|
||||
|
|
@ -2320,12 +2316,9 @@ Reopening Nuke should synchronize these paths and resolve any discrepancies.
|
|||
"""Set resolution to project resolution."""
|
||||
log.info("Resetting resolution")
|
||||
project_name = legacy_io.active_project()
|
||||
project = get_project(project_name)
|
||||
asset_name = legacy_io.Session["AVALON_ASSET"]
|
||||
asset = get_asset_by_name(project_name, asset_name)
|
||||
asset_data = asset.get('data', {})
|
||||
asset_data = self._asset_entity["data"]
|
||||
|
||||
data = {
|
||||
format_data = {
|
||||
"width": int(asset_data.get(
|
||||
'resolutionWidth',
|
||||
asset_data.get('resolution_width'))),
|
||||
|
|
@ -2335,37 +2328,40 @@ Reopening Nuke should synchronize these paths and resolve any discrepancies.
|
|||
"pixel_aspect": asset_data.get(
|
||||
'pixelAspect',
|
||||
asset_data.get('pixel_aspect', 1)),
|
||||
"name": project["name"]
|
||||
"name": project_name
|
||||
}
|
||||
|
||||
if any(x for x in data.values() if x is None):
|
||||
if any(x_ for x_ in format_data.values() if x_ is None):
|
||||
msg = ("Missing set shot attributes in DB."
|
||||
"\nContact your supervisor!."
|
||||
"\n\nWidth: `{width}`"
|
||||
"\nHeight: `{height}`"
|
||||
"\nPixel Asspect: `{pixel_aspect}`").format(**data)
|
||||
"\nPixel Aspect: `{pixel_aspect}`").format(**format_data)
|
||||
log.error(msg)
|
||||
nuke.message(msg)
|
||||
|
||||
existing_format = None
|
||||
for format in nuke.formats():
|
||||
if data["name"] == format.name():
|
||||
if format_data["name"] == format.name():
|
||||
existing_format = format
|
||||
break
|
||||
|
||||
if existing_format:
|
||||
# Enforce existing format to be correct.
|
||||
existing_format.setWidth(data["width"])
|
||||
existing_format.setHeight(data["height"])
|
||||
existing_format.setPixelAspect(data["pixel_aspect"])
|
||||
existing_format.setWidth(format_data["width"])
|
||||
existing_format.setHeight(format_data["height"])
|
||||
existing_format.setPixelAspect(format_data["pixel_aspect"])
|
||||
else:
|
||||
format_string = self.make_format_string(**data)
|
||||
format_string = self.make_format_string(**format_data)
|
||||
log.info("Creating new format: {}".format(format_string))
|
||||
nuke.addFormat(format_string)
|
||||
|
||||
nuke.root()["format"].setValue(data["name"])
|
||||
nuke.root()["format"].setValue(format_data["name"])
|
||||
log.info("Format is set.")
|
||||
|
||||
# update node graph so knobs are updated
|
||||
update_node_graph()
|
||||
|
||||
def make_format_string(self, **kwargs):
|
||||
if kwargs.get("r"):
|
||||
return (
|
||||
|
|
@ -2484,6 +2480,20 @@ def get_dependent_nodes(nodes):
|
|||
return connections_in, connections_out
|
||||
|
||||
|
||||
def update_node_graph():
|
||||
# Resetting frame will update knob values
|
||||
try:
|
||||
root_node_lock = nuke.root()["lock_range"].value()
|
||||
nuke.root()["lock_range"].setValue(not root_node_lock)
|
||||
nuke.root()["lock_range"].setValue(root_node_lock)
|
||||
|
||||
current_frame = nuke.frame()
|
||||
nuke.frame(1)
|
||||
nuke.frame(int(current_frame))
|
||||
except Exception as error:
|
||||
log.warning(error)
|
||||
|
||||
|
||||
def find_free_space_to_paste_nodes(
|
||||
nodes,
|
||||
group=nuke.root(),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue