From bdaeb3c92f3ba5c2744ce69fa3aadd3135f33699 Mon Sep 17 00:00:00 2001
From: Kayla Man <moonyuet2008@hotmail.com>
Date: Wed, 13 Sep 2023 19:08:09 +0800
Subject: [PATCH] add loadError for loaders which uses external plugins

---
 openpype/hosts/max/api/lib.py                    | 16 ++++++++++++++++
 .../hosts/max/plugins/load/load_model_usd.py     |  8 ++++++--
 .../hosts/max/plugins/load/load_pointcloud.py    |  4 ++--
 .../max/plugins/load/load_redshift_proxy.py      |  8 ++++++--
 4 files changed, 30 insertions(+), 6 deletions(-)

diff --git a/openpype/hosts/max/api/lib.py b/openpype/hosts/max/api/lib.py
index 8287341456..4a150067e1 100644
--- a/openpype/hosts/max/api/lib.py
+++ b/openpype/hosts/max/api/lib.py
@@ -407,3 +407,19 @@ def object_transform_set(container_children):
         name = f"{node.name}.scale"
         transform_set[name] = node.scale
     return transform_set
+
+
+def get_plugins() -> list:
+    """Get all loaded plugins in 3dsMax
+
+    Returns:
+        plugin_info_list: a list of loaded plugins
+    """
+    manager = rt.PluginManager
+    count = manager.pluginDllCount
+    plugin_info_list = []
+    for p in range(1, count + 1):
+        plugin_info = manager.pluginDllName(p)
+        plugin_info_list.append(plugin_info)
+
+    return plugin_info_list
diff --git a/openpype/hosts/max/plugins/load/load_model_usd.py b/openpype/hosts/max/plugins/load/load_model_usd.py
index 09566c2e78..67d0e75259 100644
--- a/openpype/hosts/max/plugins/load/load_model_usd.py
+++ b/openpype/hosts/max/plugins/load/load_model_usd.py
@@ -1,12 +1,13 @@
 import os
 
 from pymxs import runtime as rt
-
+from openpype.pipeline.load import LoadError
 from openpype.hosts.max.api import lib
 from openpype.hosts.max.api.lib import (
     unique_namespace,
     get_namespace,
-    object_transform_set
+    object_transform_set,
+    get_plugins
 )
 from openpype.hosts.max.api.lib import maintained_selection
 from openpype.hosts.max.api.pipeline import (
@@ -29,6 +30,9 @@ class ModelUSDLoader(load.LoaderPlugin):
 
     def load(self, context, name=None, namespace=None, data=None):
         # asset_filepath
+        plugin_info = get_plugins()
+        if "usdimport.dli" not in plugin_info:
+            raise LoadError("No USDImporter loaded/installed in Max..")
         filepath = os.path.normpath(self.filepath_from_context(context))
         import_options = rt.USDImporter.CreateOptions()
         base_filename = os.path.basename(filepath)
diff --git a/openpype/hosts/max/plugins/load/load_pointcloud.py b/openpype/hosts/max/plugins/load/load_pointcloud.py
index 2309d3aebf..e0317a2e22 100644
--- a/openpype/hosts/max/plugins/load/load_pointcloud.py
+++ b/openpype/hosts/max/plugins/load/load_pointcloud.py
@@ -2,7 +2,8 @@ import os
 
 from openpype.hosts.max.api import lib, maintained_selection
 from openpype.hosts.max.api.lib import (
-    unique_namespace
+    unique_namespace,
+
 )
 from openpype.hosts.max.api.pipeline import (
     containerise,
@@ -25,7 +26,6 @@ class PointCloudLoader(load.LoaderPlugin):
     def load(self, context, name=None, namespace=None, data=None):
         """load point cloud by tyCache"""
         from pymxs import runtime as rt
-
         filepath = os.path.normpath(self.filepath_from_context(context))
         obj = rt.tyCache()
         obj.filename = filepath
diff --git a/openpype/hosts/max/plugins/load/load_redshift_proxy.py b/openpype/hosts/max/plugins/load/load_redshift_proxy.py
index 36bc7ac9e2..daf6d3e169 100644
--- a/openpype/hosts/max/plugins/load/load_redshift_proxy.py
+++ b/openpype/hosts/max/plugins/load/load_redshift_proxy.py
@@ -5,6 +5,7 @@ from openpype.pipeline import (
     load,
     get_representation_path
 )
+from openpype.pipeline.load import LoadError
 from openpype.hosts.max.api.pipeline import (
     containerise,
     update_custom_attribute_data,
@@ -12,7 +13,8 @@ from openpype.hosts.max.api.pipeline import (
 )
 from openpype.hosts.max.api import lib
 from openpype.hosts.max.api.lib import (
-    unique_namespace
+    unique_namespace,
+    get_plugins
 )
 
 
@@ -28,7 +30,9 @@ class RedshiftProxyLoader(load.LoaderPlugin):
 
     def load(self, context, name=None, namespace=None, data=None):
         from pymxs import runtime as rt
-
+        plugin_info = get_plugins()
+        if "redshift4max.dlr" not in plugin_info:
+            raise LoadError("Redshift not loaded/installed in Max..")
         filepath = self.filepath_from_context(context)
         rs_proxy = rt.RedshiftProxy()
         rs_proxy.file = filepath