NukeMultiThreadRender/menu.py at master · rameshbabuvfx/NukeMultiThreadRender · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
from multiThreadRender import MultiThreadRender
from multiThreadRender import UpdateRenderWidget
from nukescripts import panels


def add_render_knob():
    """
    Adds user knob in write node.

    :return: None.
    """
    node = nuke.thisNode()
    root_node = nuke.root()
    try:
        node['RenderThread'].value()
        render_knob = "exist"
    except:
        render_knob = "not exist"

    if render_knob == "not exist":
        render_tab = nuke.Tab_Knob("RenderThread", "RenderThread")
        range_first_knob = nuke.Int_Knob("custom_first", "Frame Range")
        range_last_knob = nuke.Int_Knob("custom_last", "")
        range_last_knob.clearFlag(nuke.STARTLINE)
        divider_line = nuke.Text_Knob("divider", "")
        thread_knob = nuke.PyScript_Knob("SubmitRender")
        thread_knob.setFlag(nuke.STARTLINE)

        range_first_knob.setValue(int(root_node['first_frame'].value()))
        range_last_knob.setValue(int(root_node['last_frame'].value()))
        thread_knob.setValue("""
try:
    nuke.scriptSave()
    UpdateRenderWidget(render_panel, nuke.thisNode())
except Exception as error:
    print(error)
        """)
        node.addKnob(render_tab)
        node.addKnob(range_first_knob)
        node.addKnob(range_last_knob)
        node.addKnob(divider_line)
        node.addKnob(thread_knob)


class TestPanel(nukescripts.PythonPanel):
    def __init__(self):
        super(TestPanel, self).__init__(title="MultiRenderPanel", id="com.example.WidgetPanel")
        self.customKnob = nuke.PyCustom_Knob("Render Thread Panel", "", "WidgetKnob()")
        self.addKnob(self.customKnob)


class WidgetKnob:
    def makeUI(self):
        self.widget = MultiThreadRender()
        return self.widget


def load_python_panel(python_panel):
    python_panel.addToPane()


pane_menu = nuke.menu("Pane")
render_panel = TestPanel()
pane_menu.addCommand("Multi Thread Render", "load_python_panel(render_panel)")


nuke.addOnCreate(add_render_knob, nodeClass="Write")