From bdfa4a7f6eaa2c9c50909a65f5a33dd5ec1736d5 Mon Sep 17 00:00:00 2001 From: Alex Gerveshi <15710060+agerveshi@users.noreply.github.com> Date: Thu, 14 Sep 2023 00:52:28 +0000 Subject: [PATCH] ocio config job bundle test Signed-off-by: Alex Gerveshi <15710060+agerveshi@users.noreply.github.com> --- .../expected_job_bundle/asset_references.yaml | 11 + .../expected_job_bundle/parameter_values.yaml | 19 ++ .../ocio/expected_job_bundle/template.yaml | 227 ++++++++++++++++++ .../ocio/scene/config.ocio | 25 ++ job_bundle_output_tests/ocio/scene/ocio.nk | 75 ++++++ 5 files changed, 357 insertions(+) create mode 100644 job_bundle_output_tests/ocio/expected_job_bundle/asset_references.yaml create mode 100644 job_bundle_output_tests/ocio/expected_job_bundle/parameter_values.yaml create mode 100644 job_bundle_output_tests/ocio/expected_job_bundle/template.yaml create mode 100644 job_bundle_output_tests/ocio/scene/config.ocio create mode 100644 job_bundle_output_tests/ocio/scene/ocio.nk diff --git a/job_bundle_output_tests/ocio/expected_job_bundle/asset_references.yaml b/job_bundle_output_tests/ocio/expected_job_bundle/asset_references.yaml new file mode 100644 index 0000000..ca11f4d --- /dev/null +++ b/job_bundle_output_tests/ocio/expected_job_bundle/asset_references.yaml @@ -0,0 +1,11 @@ +assetReferences: + inputs: + directories: + - /normalized/job/bundle/dir/luts + - /tmp/luts + filenames: + - /normalized/job/bundle/dir/config.ocio + - /normalized/job/bundle/dir/ocio.nk + outputs: + directories: + - output diff --git a/job_bundle_output_tests/ocio/expected_job_bundle/parameter_values.yaml b/job_bundle_output_tests/ocio/expected_job_bundle/parameter_values.yaml new file mode 100644 index 0000000..7fae662 --- /dev/null +++ b/job_bundle_output_tests/ocio/expected_job_bundle/parameter_values.yaml @@ -0,0 +1,19 @@ +parameterValues: +- name: deadline:priority + value: 50 +- name: deadline:targetTaskRunStatus + value: READY +- name: deadline:maxFailedTasksCount + value: 100 +- name: deadline:maxRetriesPerTask + value: 5 +- name: Frames + value: 1-100 +- name: NukeScriptFile + value: /normalized/job/bundle/dir/ocio.nk +- name: ProxyMode + value: 'false' +- name: NukeVersion + value: 14.0v5 +- name: RezPackages + value: nuke-13 deadline_cloud_for_nuke diff --git a/job_bundle_output_tests/ocio/expected_job_bundle/template.yaml b/job_bundle_output_tests/ocio/expected_job_bundle/template.yaml new file mode 100644 index 0000000..91c6ddc --- /dev/null +++ b/job_bundle_output_tests/ocio/expected_job_bundle/template.yaml @@ -0,0 +1,227 @@ +specificationVersion: jobtemplate-2023-09 +name: ocio.nk +parameterDefinitions: +- name: NukeScriptFile + type: PATH + objectType: FILE + dataFlow: IN + userInterface: + control: CHOOSE_INPUT_FILE + label: Nuke Script File + fileFilters: + - label: Nuke Script Files + patterns: + - '*.nk' + - label: All Files + patterns: + - '*' + description: The Nuke script file to render. +- name: Frames + type: STRING + description: The frames to render. E.g. 1-3,8,11-15 + minLength: 1 +- name: WriteNode + type: STRING + userInterface: + control: DROPDOWN_LIST + label: Write Node + description: Which write node to render ('All Write Nodes' for all of them) + default: All Write Nodes + allowedValues: + - All Write Nodes + - Write1 +- name: View + type: STRING + userInterface: + control: DROPDOWN_LIST + description: Which view to render ('All Views' for all of them) + default: All Views + allowedValues: + - All Views + - main +- name: ProxyMode + type: STRING + userInterface: + control: CHECK_BOX + label: Proxy Mode + description: Render in Proxy Mode. + default: 'false' + allowedValues: + - 'true' + - 'false' +- name: ContinueOnError + type: STRING + userInterface: + control: CHECK_BOX + label: Continue On Error + description: Continue processing when errors occur. + default: 'false' + allowedValues: + - 'true' + - 'false' +- name: NukeVersion + type: STRING + userInterface: + control: LINE_EDIT + label: Nuke Version + description: The version of Nuke. +- name: RezPackages + type: STRING + userInterface: + control: LINE_EDIT + label: Rez Packages + description: A space-separated list of Rez packages to install + default: nuke-13 deadline_cloud_for_nuke +jobEnvironments: +- name: Rez + description: Initializes and destroys the Rez environment for the job. + script: + actions: + onEnter: + command: '{{ Env.File.Enter }}' + onExit: + command: '{{ Env.File.Exit }}' + embeddedFiles: + - name: Enter + filename: rez-enter.sh + type: TEXT + runnable: true + data: | + #!/bin/env bash + + set -euo pipefail + + if [ ! -z "{{Param.RezPackages}}" ]; then + echo "Rez Package List:" + echo " {{Param.RezPackages}}" + + # Create the environment + /usr/local/bin/deadline-rez init \ + -d "{{Session.WorkingDirectory}}" \ + {{Param.RezPackages}} + + # Capture the environment's vars + {{Env.File.InitialVars}} + . /usr/local/bin/deadline-rez activate \ + -d "{{Session.WorkingDirectory}}" + {{Env.File.CaptureVars}} + else + echo "No Rez Packages, skipping environment creation." + fi + - name: Exit + filename: rez-exit.sh + type: TEXT + runnable: true + data: | + #!/bin/env bash + + set -euo pipefail + + if [ ! -z "{{Param.RezPackages}}" ]; then + echo "Rez Package List:" + echo " {{Param.RezPackages}}" + + /usr/local/bin/deadline-rez destroy \ + -d "{{ Session.WorkingDirectory }}" + else + echo "No Rez Packages, skipping environment teardown." + fi + - name: InitialVars + filename: initial-vars.sh + type: TEXT + runnable: true + data: | + #!/usr/bin/env python3 + import os, json + envfile = "{{Session.WorkingDirectory}}/.envInitial" + with open(envfile, "w", encoding="utf8") as f: + json.dump(dict(os.environ), f) + - name: CaptureVars + filename: capture-vars.sh + type: TEXT + runnable: true + data: | + #!/usr/bin/env python3 + import os, json, sys + envfile = "{{Session.WorkingDirectory}}/.envInitial" + if os.path.isfile(envfile): + with open(envfile, "r", encoding="utf8") as f: + before = json.load(f) + else: + print("No initial environment found, must run Env.File.CaptureVars script first") + sys.exit(1) + after = dict(os.environ) + + put = {k: v for k, v in after.items() if v != before.get(k)} + delete = {k for k in before if k not in after} + + for k, v in put.items(): + print(f"updating {k}={v}") + print(f"openjd_env: {k}={v}") + for k in delete: + print(f"openjd_unset_env: {k}") +steps: +- name: Render + parameterSpace: + taskParameterDefinitions: + - name: Frame + type: INT + range: '{{Param.Frames}}' + stepEnvironments: + - name: Nuke + description: Runs Nuke in the background with a script file loaded. + script: + embeddedFiles: + - name: initData + filename: init-data.yaml + type: TEXT + data: | + continue_on_error: {{Param.ContinueOnError}} + proxy: {{Param.ProxyMode}} + script_file: '{{Param.NukeScriptFile}}' + version: '{{Param.NukeVersion}}' + write_nodes: + - '{{Param.WriteNode}}' + views: + - '{{Param.View}}' + actions: + onEnter: + command: NukeAdaptor + args: + - daemon + - start + - --path-mapping-rules + - file://{{Session.PathMappingRulesFile}} + - --connection-file + - '{{Session.WorkingDirectory}}/connection.json' + - --init-data + - file://{{ Env.File.initData }} + cancelation: + mode: NOTIFY_THEN_TERMINATE + onExit: + command: NukeAdaptor + args: + - daemon + - stop + - --connection-file + - '{{ Session.WorkingDirectory }}/connection.json' + cancelation: + mode: NOTIFY_THEN_TERMINATE + script: + embeddedFiles: + - name: runData + filename: run-data.yaml + type: TEXT + data: 'frame: {{Task.Param.Frame}}' + actions: + onRun: + command: NukeAdaptor + args: + - daemon + - run + - --connection-file + - '{{Session.WorkingDirectory}}/connection.json' + - --run-data + - file://{{ Task.File.runData }} + cancelation: + mode: NOTIFY_THEN_TERMINATE diff --git a/job_bundle_output_tests/ocio/scene/config.ocio b/job_bundle_output_tests/ocio/scene/config.ocio new file mode 100644 index 0000000..54cf9b6 --- /dev/null +++ b/job_bundle_output_tests/ocio/scene/config.ocio @@ -0,0 +1,25 @@ +ocio_profile_version: 2.1 + +environment: + {} +search_path: + - luts + - /tmp/luts +strictparsing: true +luma: [0.2126, 0.7152, 0.0722] +name: deadline-cloud-for-nuke-test-config + +roles: + {} + +file_rules: + - ! {name: Default, colorspace: default} + +displays: + {} + +active_displays: [] +active_views: [] + +colorspaces: + [] \ No newline at end of file diff --git a/job_bundle_output_tests/ocio/scene/ocio.nk b/job_bundle_output_tests/ocio/scene/ocio.nk new file mode 100644 index 0000000..8d55835 --- /dev/null +++ b/job_bundle_output_tests/ocio/scene/ocio.nk @@ -0,0 +1,75 @@ +#! /opt/nuke/Nuke14.0v5/libnuke-14.0.5.so -nx +#write_info Write1 file:"output/output_%04d.png" format:"2048 1556 1" chans:":rgba.red:rgba.green:rgba.blue:" framerange:"1 100" fps:"0" colorspace:"matte_paint" datatype:"8 bit" transfer:"unknown" views:"main" colorManagement:"OCIO" +version 14.0 v5 +define_window_layout_xml { + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +} +Root { + inputs 0 + name /home/rocky/src/deadline-cloud-for-nuke/job_bundle_output_tests/ocio/scene/ocio.nk + format "2048 1556 0 0 2048 1556 1 2K_Super_35(full-ap)" + proxy_type scale + proxy_format "1024 778 0 0 1024 778 1 1K_Super_35(full-ap)" + colorManagement OCIO + OCIO_config custom + customOCIOConfigPath /home/rocky/src/deadline-cloud-for-nuke/job_bundle_output_tests/ocio/scene/config.ocio + defaultViewerLUT "OCIO LUTs" + workingSpaceLUT scene_linear + monitorLut 0 + monitorOutLUT "sRGB (default)" + int8Lut matte_paint + int16Lut texture_paint + logLut compositing_log + floatLut scene_linear +} +ColorBars { + inputs 0 + name ColorBars1 + xpos -53 + ypos -134 +} +set Nca0bef0 [stack 0] +Write { + file output/output_####.png + colorspace matte_paint + file_type png + checkHashOnRead false + name Write1 + xpos 11 + ypos -17 +} +push $Nca0bef0 +Viewer { + frame_range 1-100 + name Viewer1 + xpos -139 + ypos 52 +}