from __future__ import print_function
import hou


class ExampleState(object):
    def __init__(self, state_name, scene_viewer):
        self.state_name = state_name
        self.scene_viewer = scene_viewer

    # メニューコールバック
    def onMenuAction(self, kwargs):
        action = kwargs["menu_item"]
        if action == 'toggle1':
            print("Set toggle 1 to", kwargs["toggle1"])

        elif action == 'toggle2':
            print("Set toggle 2 to", kwargs["toggle2"])

        elif action == 'action':
            print("Chose action 1")

        elif action == 'radio_strip':
            print("Set the radio strip to", kwargs["radio_strip"])

        elif action == 'submenu_toggle1':
            print("Set the submenu toggle to", kwargs["submenu_toggle1"])


def createViewerStateTemplate(state_type, state_label):     
    template = hou.ViewerStateTemplate(
        state_type, state_label, hou.sopNodeTypeCategory()
    )
    template.bindFactory(ExampleState)

    # ホットキーをいくつか作成します。
    example_context = "h.pane.gview.state.sop.example"
    hou.hotkeys.addContext(example_context, "Example Operation", "These keys apply to the Example operations")

    h1 = example_context + 'h1'
    hou.hotkeys.addCommand(h1, 'example 1', 'example 1')
    hou.hotkeys.addAssignment(h1, '1')            

    h2 = example_context + 'h2'
    hou.hotkeys.addCommand(h2, 'example 2', 'example 2')
    hou.hotkeys.addAssignment(h2, '2')

    h3 = example_context + 'h3'
    hou.hotkeys.addCommand(h3, 'example 3', 'example 3')
    hou.hotkeys.addAssignment(h3, '3')

    h4 = example_context + 'h3'
    hou.hotkeys.addCommand(h4, 'example 4', 'example 4')
    hou.hotkeys.addAssignment(h4, '4')

    # Viewer Stateのポップアップメニューを定義します。
    m = hou.ViewerStateMenu('menu', 'Example')

    m.addSeparator()
    m.addToggleItem( 'toggle1', 'Toggle1', True, hotkey=h1 )
    m.addToggleItem( 'toggle2', 'Toggle2', False )
    m.addSeparator()
    m.addActionItem( 'action', 'Action1', hotkey=h2 )
    m.addSeparator()
    m.addRadioStrip( 'radio_strip', 'Radio', "radio_item1" )
    m.addRadioStripItem( 'radio_strip', 'radio_item1', 'Radio1', hotkey=h3 )
    m.addRadioStripItem( 'radio_strip', 'radio_item2', 'Radio2', hotkey=h4 )
    m.addSeparator()
    subm = hou.ViewerStateMenu('submenu', 'More Example...')
    subm.addToggleItem( 'submenu_toggle1', 'Sub toggle1', True )
    m.addMenu( subm )
    m.addSeparator()

    # ポップアップルメニューをViewer Stateに取り付けます。
    template.bindMenu(m)

    return t