import hou
import viewerstate.utils as su

# パラメータダイアログにのみ表示される事前定義されたテキストサイズの値を使ってメニューパラメータを定義します。
menu_item_info = [
    ('text_size_05', '0.5', 'BUTTONS_calc'), 
    ('text_size_1', '1.0', 'BUTTONS_calc'), 
    ('text_size_2', '2.0', 'BUTTONS_calc'), 
    ('text_size_5', '5.0', 'BUTTONS_calc')]

class State(object):
    def __init__(self, state_name, scene_viewer):
        self.state_name = state_name
        self.scene_viewer = scene_viewer        
        self.text_scale = hou.Vector3()
        self.menu_item_dict = { menu_items[0]:float(menu_items[1]) for menu_items in menu_item_info}

        # ビューポート内にテキスト文字列を描画するためのDrawable。
        self.text_drawable = hou.TextDrawable(self.scene_viewer, 'text_drawable_name',
            params = {'margins': (10.0,10.0)} )
        self.text_drawable.show(True)

    def onEnter(self, kwargs):
        """
        ステートに入すると、いくつかのパラメータを無効にします。
        """
        node = kwargs['node']
        state_parms = kwargs['state_parms']

        # テキストスケールを設定します。
        size = state_parms['text_size']['value']
        self.text_scale = hou.Vector3(size,size,size)

        # テキストサイズのプリセットメニューを非表示にして、トグルボタンのチェックを外します。
        state_parms['text_size_menu']['visible'] = False
        state_parms['enable_text_size_menu']['value'] = False

    def onParmChangeEvent(self, kwargs):
        """
        パラメータの変更に反応させます。
        """
        parm_name = kwargs['parm_name']
        parm_value = kwargs['parm_value']
        state_parms = kwargs['state_parms']

        if parm_name == 'text_size':
            # テキストスケールを設定します。
            self.text_scale = hou.Vector3(parm_value,parm_value,parm_value)            
        elif parm_name == 'text_size_menu':
            # 選択したプリセットでテキストサイズを設定します。
            size = self.menu_item_dict[parm_value]
            self.text_scale = hou.Vector3(size,size,size)
            state_parms['text_size']['value'] = size
        elif parm_name == 'text_size_reset':
            # テキストサイズをリセットします。
            size = 1.0
            self.text_scale = hou.Vector3(size,size,size)
            state_parms['text_size']['value'] = size
        elif parm_name == 'enable_text_size_menu':
            # テキストサイズメニューを有効または無効にします。
            state_parms['text_size_menu']['visible'] = parm_value

        # テキストを強制的に再描画します。
        self.scene_viewer.curViewport().draw()

    def onDraw( self, kwargs ):
        # ビューポート内の左下(デフォルトの原点)にテキストを描画します。
        handle = kwargs['draw_handle']
        state_parms = kwargs['state_parms']

        params = {
            'text': state_parms['text']['value'],
            'scale' : self.text_scale }

        self.text_drawable.draw( handle, params )


def createViewerStateTemplate():
    """登録するViewerステートテンプレートを作成して返すための必須エントリポイント。"""

    state_typename = kwargs["type"].definition().sections()["DefaultState"].contents()
    state_label = "State parms demo"
    state_cat = hou.objNodeTypeCategory()

    template = hou.ViewerStateTemplate(state_typename, state_label, state_cat)
    template.bindFactory(State)
    template.bindIcon('$HFS/houdini/pic/minimizedicon.pic')

    # 描画するテキストを格納するための文字列パラメータを定義します。
    template.bindParameter( hou.parmTemplateType.String, name="text", label="Text",
        default_value="Lorem ipsum dolor sit amet." )

    # 最小/最大範囲を持ったテキストサイズ浮動小数点パラメータを定義します。
    template.bindParameter( hou.parmTemplateType.Float, name="text_size", 
        label="Text Size", min_limit=0.5, max_limit=5, 
        default_value=0.75 )

    # ファクトリー値でテキストサイズをリセットするためのボタンパラメータを定義します。
    template.bindParameter( hou.parmTemplateType.Button, name="text_size_reset", 
        label="Reset Text Size" )

    # テキストサイズのプリセットメニューを表示/非表示するためのトグルパラメータを定義します。
    template.bindParameter( hou.parmTemplateType.Toggle, name="enable_text_size_menu", 
        label="Enable Text Size Menu", default_value=True, align=True )

    # カーソルサイズのプリセット値でメニューパラメータを定義します。
    template.bindParameter( hou.parmTemplateType.Menu, name="text_size_menu", 
        label="Text Size Presets", default_value='text_size_1', 
        menu_items=menu_item_info, toolbox=False )

    return template