from husd.backgroundrenderer import BackgroundRenderer


# レンダークラスは、husd.backgroundrenderer.BackgroundRendererをサブクラス化する必要があります。
# おそらく実装するレンダーのタイプに基づいてレンダークラスの名前を付けたいことでしょう。
# 例えば、CloudServiceRendererです。

class MyCustomRenderer(BackgroundRenderer):
    # デフォルトの __init__ 実装では、いくつか便利なインスタンス変数をセットアップします:
    # self._lopnet_path:
    #     (str) LOPネットワークのノードパス。例えば、"/stage"。
    # self._item_id:
    #     (str) このレンダーの固有なID文字列。この文字列を使用して、例えばレンダーファームマネージャのジョブ名などの名前または識別子を構築することができます。

    # 以下のヘルパー関数をコールすることができます:
    #
    # self.updateImagePath(image_path)
    #     指定したパスの画像を使用してスナップショットを更新します。
    #
    # self.updateMetadata(metadata)
    #     指定した辞書を使用してスナップショットのメタデータを更新します。
    #     辞書のほとんどのキーは任意ですが、いくつかのキーがHoudiniのUIで使用されています:
    #     "totalClockTime":
    #         (float) 現在の経過時間(秒)。
    #         Render Galleryビューアは、これをレンダー統計内に表示します。
    #     "percentDone":
    #         (float) 完了率(0-100)で、これもレンダー統計内に表示されます。
    #     "peakMemory":
    #         (int) 最大メモリ使用量。

    def pollFrequency(self):
        # システムが(この(浮動小数点)秒数の間隔で)このオブジェクトに対してメソッドをコールしてアップデートを取得する頻度。
        return 1.0

    def getConfigOptions(self, old_options):
        # このメソッドは、構成用インターフェースを起動し、ユーザからオプション値を取得してから、
        # そのオプションのキー/値を辞書として返すことを目的にしています。
        # この引数は、このメソッドが最後にコールされた時に返された内容を含んだ辞書になっているので、
        # ユーザーが以前に入力した値をオプションUIに入力することができます。
        # このメソッドは、ユーザがメニューからレンダープラグインを選択した場合にのみコールされます。
        # ユーザはBackgroundボタンをクリックするだけで、
        # システムは、このメソッドをコールして最初に構成用UIを表示するのではなく、以前の値で直接startBackgroundRender()をコールします。
        return {}

    def startBackgroundRender(self, usd_filepath, options):
        # このメソッドでは、バックグランドレンダラーを開始します。
        # usd_filepath: レンダリングするUSDファイルのファイルパス。
        # options引数は辞書です:
        # - options["plugin"]は、getConfigOptions()が返すユーザオプション辞書。
        # - options["viewport"]["rendersettings_path"]は、スナップショットで使用するRenderSettings Primのシーングラフパス。
        # - options["viewport"]["camera_path"]は、スナップショットで使用するCamera Primのシーングラフパス。
        # - options["viewport"]["res_x"]は、スナップショットRender Productのピクセル幅。
        # - options["viewport"]["res_y"]は、スナップショットRender Productのピクセル高さ。
        pass

    def isRenderFinished(self):
        # システムは、このメソッドを定期的にコールします。
        # このオブジェクトで開始されたレンダリングが終了したらTrueを返してください。
        #
        # このメソッド内でself.updateImagePath(image_path)やself.updateMetadata(metadata_dict)をコールすると、
        # レンダリングの進捗に関するフィードバックを送信することができます。
        pass

    def mouseClick(self, x, y):
        # ユーザがレンダープレビューウィンドウ内をクリックした時にコールされます。
        # xとyは正規化された座標が入るので、例えば、x=0.0は左端、x=0.5は中央、x=1.0は右端を意味します。
        # マウスクリックの通常の挙動では、ユーザがクリックした画像の部分にレンダリングが集中します。
        pass

    def stopBackgroundRender(self):
        # ユーザがライブスナップショットを右クリックしてStop Renderingを選択するとコールされます。
        # このメソッドは、レンダリングが停止されるまで戻りません。
        pass


# プラグインファイルには必ず登録関数を入れなければなりません。これは、Houdiniがプラグインファイルを読み込んだ時に検索される関数です。

def registerBackgroundRenderers(manager):
    # 1つ目の引数は、レンダータイプをわかりやすくしたラベルです。
    # 2つ目の引数は、バックグラウンドレンダークラスです。
    manager.registerBackgroundRenderer('Cloud Service Render', MyCustomRenderer)

import datetime
import json
import subprocess
import tempfile

from husd.backgroundrenderer import BackgroundRenderer


class FastRender(BackgroundRenderer):
    def pollFrequency(self):
        # システムは、2秒おきにこのレンダリングの更新をチェックします。
        return 2.0

    def startBackgroundRender(self, usd_filepath, options):
        self._popen = subprocess.Popen(
            ["fastrender",
             "--image", self._image_filepath]
        )
        # レンダリングが開始された時刻を記録します。
        self._start_datetime = datetime.datetime.utcnow() 

    def isRenderFinished(self):
        # レンダリングのメタデータを更新します。
        # レンダープロセスでレンダリング統計情報を段階的に出力する何かしらの方法がある場合
        # (例えば、stdoutへの出力を解析したり、定期的にJSONファイルを書き出すことができたり)、
        # その方法を使ってメタデータ辞書を収集することができます。
        metadata = {}

        # そのレンダープロセスのWall Clock Time(プロセスの開始から終了までの時間)を独自に非常に簡単に追跡することができます。
        time_delta = datetime.datetime.utcnow() - self._start_datetime
        metadata["totalClockTime"] = time_delta.total_seconds()

        # このサンプルでは、レンダープロセスが段階的にディスク上の画像ファイルを更新することを想定しているので、
        # 画像が更新されたことをシステムに伝え続ける必要があります。
        self.updateImagePath(self._imagepath)

        # Popen.poll()は、レンダープロセスが終了したら戻りコードを返し、
        # それ以外の場合はNoneを返します。
        rc = self._popen.poll()
        return rc is not None

    def stopBackgroundRender(self):
        self._popen.kill()


def registerBackgroundRenderers(manager):
    # 1つ目の引数は、レンダータイプをわかりやすくしたラベルです。
    # 2つ目の引数は、バックグラウンドレンダークラスです。
    manager.registerBackgroundRenderer('Fast Render', FastRender)

def isRenderFinished():
    # ... メタデータを更新します ...

    # Popen.poll()は、レンダープロセスが終了したら戻りコードを返し、
    # それ以外の場合はNoneを返します。
    rc = self._popen.poll()
    if rc is not None:
        # レンダリングが終了したら、スナップショット画像を更新するようにHoudiniに伝えます。
        self.updateImagePath(self._imagepath)
        return True
    else:
        return False