LumAPI 1.1.2__tar.gz → 1.1.4__tar.gz

lumapi-1.1.4/LumAPI/__init__.py

@@ -0,0 +1,21 @@
+# LumAPI/__init__.py
+
+# 显式导入所有需要对外暴露的类和函数
+from .lumapi import (
+    lumapi, lumerical, LumFuncBase,
+    FDTD, MODE, DEVICE, INTERCONNECT,
+    savemat, loadmat, 
+    create_cmap, set_colorbar_range,
+    Estimate_focal, Kirchhoff, 
+    RayleighSommerfeld_Scalar, RayleighSommerfeld_Vector, AngularSpectrum_Vector
+)
+
+# 定义 import * 时对外暴露的接口白名单（这是静态检查器识别 import * 的关键）
+__all__ = [
+    'lumapi', 'lumerical', 'LumFuncBase',
+    'FDTD', 'MODE', 'DEVICE', 'INTERCONNECT',
+    'savemat', 'loadmat', 
+    'create_cmap', 'set_colorbar_range',
+    'Estimate_focal', 'Kirchhoff', 
+    'RayleighSommerfeld_Scalar', 'RayleighSommerfeld_Vector', 'AngularSpectrum_Vector'
+]

{lumapi-1.1.2 → lumapi-1.1.4}/LumAPI/lumapi.py

@@ -1002,7 +1002,12 @@ def detect_version(lumerical_root):
 
 def get_lumapi_path(lumerical_root, version):
     """从Lumerical根路径和版本获取lumapi.py路径"""
-    return os.path.join(lumerical_root, version, "api", "python", "lumapi.py")
+    base = os.path.join(lumerical_root, version)
+    p1 = os.path.join(base, "Lumerical", "api", "python", "lumapi.py")
+    if os.path.exists(p1): return p1
+    p2 = os.path.join(base, "api", "python", "lumapi.py")
+    if os.path.exists(p2): return p2
+    return None
 
 def validate_path(lumerical_root: str, version: str = None) -> object:
     """验证Lumerical路径有效性并返回lumapi对象
@@ -1031,7 +1036,7 @@ def validate_path(lumerical_root: str, version: str = None) -> object:
         # 获取lumapi.py的完整路径
         lumapi_path = get_lumapi_path(lumerical_root, version)
         
-        if not os.path.exists(lumapi_path):
+        if not lumapi_path:
             print(f"错误：在指定路径未找到 lumapi.py 文件（查找路径：{lumapi_path}）")
             return None
             
@@ -1081,6 +1086,10 @@ class lumerical:
         except Exception:
             self.lumapi = None
 
+    def __bool__(self):
+        """判断配置是否成功，允许直接使用 if lumapi: 来判断"""
+        return self.lumapi is not None
+
     def _check_config_and_prompt(self):
         """核心逻辑：检查配置状态，若无效则引导用户使用 LumAPI 命令"""
         if self.lumapi is None: