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)

    # ホットキー定義をいくつか追加します。
    hotkey_definitions = hou.PluginHotkeyDefinitions()

    example_context = "h.pane.gview.state.sop.example"
    hotkey_definitions.addContext(example_context, "Example Operation", "These keys apply to the Example operations")
    example_category = "h.pane.gview.state.sop.example"
    hotkey_definitions.addCommandCategory(example_category, "Example Operation", "These commands apply to the Example operations")

    h1 = example_category + 'h1'
    hotkey_definitions.addCommand(h1, 'example 1', 'example 1')
    hotkey_definitions.addDefaultBinding(example_context, h1, ['1'])

    h2 = example_category + 'h2'
    hotkey_definitions.addCommand(h2, 'example 2', 'example 2')
    hotkey_definition.addDefaultBinding(example_context, h2, ['2'])

    h3 = example_category + 'h3'
    hotkey_definitions.addCommand(h3, 'example 3', 'example 3')
    hotkey_definitions.addDefaultBinding(example_context, h3, ['3'])

    h4 = example_category + 'h3'
    hotkey_definitions.addCommand(h4, 'example 4', 'example 4')
    hotkey_definitions.addDefaultBinding(example_context, 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)

    # ホットキー定義をViewer Stateに取り付けます。
    template.bindHotkeyDefinitions(hotkey_definitions)

    return t