Changeset c2c51a2 in sasmodels for sasmodels/generate.py

Timestamp:

May 6, 2016 1:13:55 AM (8 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, costrafo411, magnetic_model, release_v0.94, release_v0.95, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

306e354

Parents:

948b8c1

Message:

allow sasmodels to look in sasmodels-data for builtin model c source

File:

• sasmodels/generate.py (modified) (5 diffs)

sasmodels/generate.py

@@ -210 +210 @@
 import sys
 from os.path import abspath, dirname, join as joinpath, exists, basename, \
-    splitext
+    splitext, isdir
 import re
 import string
@@ -224 +224 @@
 Parameter = namedtuple('Parameter', PARAMETER_FIELDS)
 
-C_KERNEL_TEMPLATE_PATH = joinpath(dirname(__file__), 'kernel_template.c')
+SIBLING_DIR = 'sasmodels-data'
+PACKAGE_PATH = abspath(dirname(__file__))
+SIBLING_PATH = abspath(joinpath(PACKAGE_PATH, '..', 'sasmodels-data'))
+DATA_PATH = SIBLING_PATH if isdir(SIBLING_PATH) else PACKAGE_PATH
+MODEL_PATH = joinpath(DATA_PATH, 'models')
+C_KERNEL_TEMPLATE_FILE = joinpath(DATA_PATH, 'kernel_template.c')
+
 
 F16 = np.dtype('float16')
@@ -336 +342 @@
     Return a list of the sources file paths for the module.
     """
-    search_path = [dirname(model_info['filename']),
-                   abspath(joinpath(dirname(__file__), 'models'))]
+    search_path = [dirname(model_info['filename']), MODEL_PATH]
     return [_search(search_path, f) for f in model_info['source']]
 
@@ -457 +462 @@
     global C_KERNEL_TEMPLATE
     if C_KERNEL_TEMPLATE is None:
-        with open(C_KERNEL_TEMPLATE_PATH) as fid:
+        with open(C_KERNEL_TEMPLATE_FILE) as fid:
             C_KERNEL_TEMPLATE = fid.read()
 
@@ -466 +471 @@
               for p in ('#line 1 "%s"'%f.replace('\\', '\\\\'), open(f).read())
               ]
-    source.append('#line 133 "%s"'%C_KERNEL_TEMPLATE_PATH.replace('\\', '\\\\'))
+    source.append('#line 133 "%s"'%C_KERNEL_TEMPLATE_FILE.replace('\\', '\\\\'))
 
     # Prepare defines