DeepGPR 0.0.1__tar.gz → 0.0.4__tar.gz

{deepgpr-0.0.1/DeepGPR.egg-info → deepgpr-0.0.4}/PKG-INFO

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: DeepGPR
-Version: 0.0.1
+Version: 0.0.4
 Summary: PyTorch and CUDA for GPR FWI
 Author-email: Lei Liu <liulei990222@gmail.com>
 Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX :: Linux
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 
@@ -59,7 +60,6 @@ peak_time = 1 / freq
 source_amplitudes = torch.zeros((1,nt,1),device=device)
 source_amplitudes[0,:,0]=DeepGPR.ricker(freq, nt, dt, peak_time).to(device)
 
-DeepGPR.plot_survey_geometry(er,source_location, receiver_location,dx)
 
 #forward modeling
 r = DeepGPR.compute(
@@ -79,8 +79,7 @@ ax[1].imshow(er.grad.detach())
 ax[1].set_title("Gradient")
 plt.show()
 ```
-![result](./Fig/example1.png)
 
-![result](./Fig/example2.png)
+![result](./Fig/example.png)
 
 There are more examples in the ./examples.

{deepgpr-0.0.1 → deepgpr-0.0.4}/README.md

@@ -49,7 +49,6 @@ peak_time = 1 / freq
 source_amplitudes = torch.zeros((1,nt,1),device=device)
 source_amplitudes[0,:,0]=DeepGPR.ricker(freq, nt, dt, peak_time).to(device)
 
-DeepGPR.plot_survey_geometry(er,source_location, receiver_location,dx)
 
 #forward modeling
 r = DeepGPR.compute(
@@ -69,8 +68,7 @@ ax[1].imshow(er.grad.detach())
 ax[1].set_title("Gradient")
 plt.show()
 ```
-![result](./Fig/example1.png)
 
-![result](./Fig/example2.png)
+![result](./Fig/example.png)
 
 There are more examples in the ./examples.

{deepgpr-0.0.1 → deepgpr-0.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "DeepGPR" 
-version = "0.0.1"              
+version = "0.0.4"              
 authors = [
   { name="Lei Liu", email="liulei990222@gmail.com" },
 ]
@@ -13,8 +13,13 @@ readme = "README.md"
 requires-python = ">=3.7"
 classifiers = [
     "Programming Language :: Python :: 3",
-    "Operating System :: OS Independent"
+    "Operating System :: Microsoft :: Windows",
+    "Operating System :: POSIX :: Linux"
 ]
 
-[tool.setuptools]
-packages = ["DeepGPR"]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+"DeepGPR" = ["lib/*.cu", "lib/*.so", "lib/*.dll"]

{deepgpr-0.0.1 → deepgpr-0.0.4/src/DeepGPR.egg-info}/PKG-INFO

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: DeepGPR
-Version: 0.0.1
+Version: 0.0.4
 Summary: PyTorch and CUDA for GPR FWI
 Author-email: Lei Liu <liulei990222@gmail.com>
 Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: OS Independent
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX :: Linux
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 
@@ -59,7 +60,6 @@ peak_time = 1 / freq
 source_amplitudes = torch.zeros((1,nt,1),device=device)
 source_amplitudes[0,:,0]=DeepGPR.ricker(freq, nt, dt, peak_time).to(device)
 
-DeepGPR.plot_survey_geometry(er,source_location, receiver_location,dx)
 
 #forward modeling
 r = DeepGPR.compute(
@@ -79,8 +79,7 @@ ax[1].imshow(er.grad.detach())
 ax[1].set_title("Gradient")
 plt.show()
 ```
-![result](./Fig/example1.png)
 
-![result](./Fig/example2.png)
+![result](./Fig/example.png)
 
 There are more examples in the ./examples.

deepgpr-0.0.4/src/DeepGPR.egg-info/SOURCES.txt

@@ -0,0 +1,6 @@
+README.md
+pyproject.toml
+src/DeepGPR.egg-info/PKG-INFO
+src/DeepGPR.egg-info/SOURCES.txt
+src/DeepGPR.egg-info/dependency_links.txt
+src/DeepGPR.egg-info/top_level.txt

deepgpr-0.0.4/src/DeepGPR.egg-info/top_level.txt

@@ -0,0 +1 @@
+lib

deepgpr-0.0.1/DeepGPR/__init__.py

@@ -1,147 +0,0 @@
-import ctypes
-import os
-import subprocess
-import platform
-from pathlib import Path
-
-
-system_name = platform.system()
-lib_dir = Path(__file__).parent / 'lib'
-cu_file = lib_dir / 'deepgpr.cu'
-
-if system_name == "Windows":
-    lib_extension = ".dll"
-    lib_filename = f'deepgpr{lib_extension}'
-    lib_path_obj = lib_dir / lib_filename
-    
-    # Windows 下直接调用 nvcc 的命令参数
-    nvcc_cmd = [
-        'nvcc', '-shared', 
-        '-o', str(lib_path_obj), 
-        str(cu_file)
-    ]
-else:
-    lib_extension = ".so"
-    lib_filename = f'deepgpr{lib_extension}'
-    lib_path_obj = lib_dir / lib_filename
-    
-    # Linux 下直接调用 nvcc 的命令参数 (-fPIC 和 ABI 设置)
-    nvcc_cmd = [
-        'nvcc', '-shared', '-Xcompiler', '-fPIC', 
-        '-D_GLIBCXX_USE_CXX11_ABI=0', 
-        '-o', str(lib_path_obj), 
-        str(cu_file)
-    ]
-
-# 2. 检查动态库是否存在，不存在则直接使用 Python 调用 nvcc 进行编译
-if not lib_path_obj.is_file():
-    print(f'Compiling CUDA extension for {system_name} directly via nvcc...')
-    try:
-        # 直接执行 nvcc 命令，完全摆脱 make 依赖
-        subprocess.run(nvcc_cmd, check=True)
-    except FileNotFoundError:
-        raise RuntimeError(
-            "Compilation failed: 'nvcc' command not found. "
-            "Please ensure NVIDIA CUDA Toolkit is installed and 'nvcc' is added to your system PATH."
-        )
-    except subprocess.CalledProcessError as e:
-        raise RuntimeError(f"Compilation failed with error code {e.returncode}.")
-
-# 3. 加载编译好的动态链接库 (变量名改为 c_lib，避免与 lib 文件夹冲突)
-lib_path = str(lib_path_obj)
-c_lib = ctypes.cdll.LoadLibrary(lib_path)
-
-# 4. 定义 C 函数的参数和返回值类型
-c_lib.forward.argtypes = [
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    
-    ctypes.c_float,ctypes.c_int,ctypes.c_int,ctypes.c_int,ctypes.c_float,
-    ctypes.POINTER(ctypes.c_int), ctypes.POINTER(ctypes.c_float),
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-    ctypes.POINTER(ctypes.c_int), ctypes.POINTER(ctypes.c_float),
-    ctypes.c_int 
-]
-c_lib.forward.restype = None  # 修复了绑定错误
-
-c_lib.backward.argtypes = [
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),#16
-    
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),#40
-
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), 
-    ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float), #58
-    
-    ctypes.c_float,ctypes.c_int,ctypes.c_int,ctypes.c_int,ctypes.c_float,
-    ctypes.c_int, ctypes.c_int, ctypes.c_int, ctypes.c_int, 
-    ctypes.POINTER(ctypes.c_int), ctypes.POINTER(ctypes.c_float),
-    ctypes.c_int ,ctypes.POINTER(ctypes.c_float), ctypes.POINTER(ctypes.c_float),
-    ctypes.c_int, ctypes.c_int 
-]
-c_lib.backward.restype = None  # 修复了绑定错误
-
-__all__ = ['c_lib']
-
-# ==============================================================================
-# 第二步：在动态库环境就绪后再导入其余模块
-# ==============================================================================
-from .common import *
-from .compute2 import *
-from .multiscale import *
-from .visual import *