import stateutils
import soptoolutils


# アセットのイベントコールバックとは違い、このツールスクリプトは、["type"]からアセットタイプの参照を取得していません(その代わりに、kwargs["toolname"]からツール名を取得しています)。
# そのため、そのツール名からノード名を解釈する方法を用意するか、または、単に以下のように明示的にノード名を指定する必要があります。
nodetypename = "mynamespace::myasset::2.0"
# アセットの"ベース"名を使って、新しいオブジェクトの名前を付けます。
basename = "myasset"

# アクティブなペインの参照を取得します。このツールをネットワークエディタのTabメニューから選択した場合、そのペインはNetworkEditorになります。
pane = stateutils.activePane(kwargs)
if isinstance(pane, hou.SceneViewer):
    network = pane.pwd()
    if network.childTypeCategory() != hou.sopNodeTypeCategory():
        # まだオブジェクトの中にいないので、新しいSOPの作成先となる場所をユーザに尋ねます。
        objects = pane.selectObjects(
            prompt='Select objects',
            quick_select=False,
            use_existing_selection=True,
            allow_multisel=False,
            allowed_types=['geo']
        )
        if objects:
            # ユーザが2つ以上のオブジェクトを選択した場合は、1番目に選択したオブジェクトのみが取得されます。
            network = objects[0]
            # オブジェクトにジャンプします。
            pane.setPwd(network)
        else:
            # ユーザがオブジェクトを選択せずにEnterを押した場合は、
            # 新しいオブジェクトを作成します。
            cname = basename + "_object1"
            network = hou.node("/obj").createNode("geo", node_name=cname)

        # ネットワークの中に入ります。
        pane.setPwd(network)

    # 選択したネットワークの中に新しいノードを作成します(空っぽのリストは、コンポーネントが選択されていないことを意味します)。
    newnode = stateutils.createFilterSop(kwargs, nodetypename, network, [])
    # ノードタイプに関連付けられているデフォルトのステート(アセットが正しくセットアップされていれば、あなたのカスタムPythonステート)を起動します。
    pane.enterCurrentNodeState()
    # 別の方法として、以下のようにあなたのカスタムステートを明示的に設定することができます。
    # pane.setCurrentState("my_state_name")

else:
    # ビューア以外で操作した場合は、ローレベルの関数で代用します。
    soptoolutils.genericTool(kwargs, nodetypename)

import stateutils

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

        self._node = None
        self._pressed = False
        self._index = 0    

    def _point_count(self):
        multiparm = self._node.parm("points")
        # これは、マルチパラメータのインスタンスの数を取得する方法です。
        return multiparm.evalAsInt()

    def _insert_point(self):
        index = self._point_count()
        multiparm = self._node.parm("points")
        multiparm.insertMultiParmInstance(index)
        return index

    def _set(self, index, position):
        self._node.parm("usept%d" % index).set(1)
        self._node.parmTuple("pt%d" % index).set(position)

    def _start(self):
        if not self._pressed:
            self.scene_viewer.beginStateUndo("Add point")
            self._index = self._insert_point()
        self._pressed = True

    def _finish(self):
        if self._pressed:
            self.scene_viewer.endStateUndo()
        self._pressed = False

    def onMouseEvent(self, kwargs):
        node = self._node = kwargs["node"]
        ui_event = kwargs["ui_event"]
        device = ui_event.device()
        origin, direction = ui_event.ray()

        # ジオメトリまたは地面との交差を求めます。
        intersected = -1
        inputs = node.inputs()
        # このノードに入力があれば、ジオメトリの交差のみを求めます。
        if inputs and inputs[0]:
            geometry = inputs[0].geometry()
            intersected, position, _, _ = stateutils.sopGeometryIntersection(geometry, origin, direction)
        if intersected < 0:
            position = stateutils.cplaneIntersection(self.scene_viewer, origin, direction)

        # LMBクリックされたら、ポイントを作成/移動させます。
        if device.isLeftButton():
            self._start()
            self._set(self._index, position)
        else:
            self._finish()

    def onInterrupt(self, kwargs):
        self._finish()