source: sasview/sansmodels/src/setup.py @ 8f19b69

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 8f19b69 was 87615a48, checked in by Jae Cho <jhjcho@…>, 13 years ago

added c-extension for smearer2d_helper(but locked it for now)

  • Property mode set to 100644
File size: 11.9 KB
Line 
1"""
2 Installation script for SANS models
3
4  - To compile and install:
5      python setup.py install
6  - To create distribution:
7      python setup.py bdist_wininst
8  - To create odb files:
9      python setup.py odb
10
11"""
12import sys
13import os
14   
15from numpy.distutils.misc_util import get_numpy_include_dirs
16numpy_incl_path = os.path.join(get_numpy_include_dirs()[0], "numpy")
17
18def createODBcontent(class_name):
19    """
20        Return the content of the Pyre odb file for a given class
21        @param class_name: Name of the class to write an odb file for [string]
22        @return: content of the file [string]
23    """
24    content  = "\"\"\"\n"
25    content += "  Facility for SANS model\n\n"
26    content += "  WARNING: THIS FILE WAS AUTOGENERATED AT INSTALL TIME\n"
27    content += "           DO NOT MODIFY\n\n"
28    content += "  This code was written as part of the DANSE project\n"
29    content += "  http://danse.us/trac/sans/\n"
30    content += "  @copyright 2007:"
31    content += "  SANS/DANSE Group (University of Tennessee), for the DANSE project\n\n"
32    content += "\"\"\"\n"
33    content += "def model():\n"
34    content += "    from ScatteringIntensityFactory import ScatteringIntensityFactory\n"
35    content += "    from sans.models.%s import %s\n" % (class_name, class_name)
36    content += "    return ScatteringIntensityFactory(%s)('%s')\n"\
37                 % (class_name, class_name)
38
39    return content
40
41def createODBfiles():
42    """
43       Create odb files for all available models
44    """
45    from sans.models.ModelFactory import ModelFactory
46   
47    class_list = ModelFactory().getAllModels()
48    for name in class_list:
49        odb = open("sans/models/pyre/%s.odb" % name, 'w')
50        odb.write(createODBcontent(name))
51        odb.close()
52        print "sans/models/pyre/%s.odb created" % name
53       
54#
55# Proceed with installation
56#
57
58# First, create the odb files
59if len(sys.argv) > 1 and sys.argv[1].lower() == 'odb':
60    print "Creating odb files"
61    try:
62        createODBfiles()
63    except:   
64        print "ERROR: could not create odb files"
65        print sys.exc_value
66    sys.exit()
67
68# Then build and install the modules
69from distutils.core import Extension, setup
70#from setuptools import setup#, find_packages
71
72# Build the module name
73srcdir  = "sans/models/c_extensions"
74igordir = "libigor"
75
76print "Installing SANS models"
77
78
79dist = setup(
80    name="models",
81    version = "0.9.1",
82    description = "Python module for SANS scattering models",
83    author = "SANS/DANSE",
84    author_email = "sansdanse@gmail.gov",
85    url = "http://danse.us/trac/sans",
86   
87    # Place this module under the sans package
88    #ext_package = "sans",
89   
90    # Use the pure python modules
91    package_dir = {"sans_extension":"sans/models/c_extensions",
92                   "sans.models.media":"media"},
93    package_data={'sans.models.media': ['*.gif','*.jpg','*.png','*.html']},
94    packages = ["sans","sans.models","sans.models.test",
95                "sans_extension","sans.models.pyre",
96                "sans.models.media"],
97   
98    ext_modules = [ Extension("sans_extension.c_models",
99     sources = [
100        "sans/models/c_models/c_models.cpp",
101        #srcdir+"/CSphereModel.c",
102        #srcdir+"/sphere.c",
103        "sans/models/c_models/CSphereModel.cpp",
104        "sans/models/c_models/COnionModel.cpp",
105        "sans/models/c_models/onion.cpp",
106        srcdir+"/onion.c",
107        "sans/models/c_models/CReflModel.cpp",
108        "sans/models/c_models/refl.cpp",
109        srcdir+"/refl.c",
110        "sans/models/c_models/CReflAdvModel.cpp",
111        "sans/models/c_models/refl_adv.cpp",
112        srcdir+"/refl_adv.c",
113        #srcdir+"/SquareWell.c",
114        "sans/models/c_models/CSquareWellStructure.cpp", 
115        "sans/models/c_models/SquareWell.cpp", 
116        #srcdir+"/StickyHS.c",         
117        "sans/models/c_models/CStickyHSStructure.cpp", 
118        "sans/models/c_models/StickyHS.cpp",             
119        "sans/models/c_models/CHardsphereStructure.cpp", 
120        "sans/models/c_models/Hardsphere.cpp",   
121        #srcdir+"/DiamCyl.c",         
122        "sans/models/c_models/CDiamCylFunc.cpp", 
123        "sans/models/c_models/DiamCyl.cpp",   
124        #srcdir+"/DiamEllip.c",         
125        "sans/models/c_models/CDiamEllipFunc.cpp", 
126        "sans/models/c_models/DiamEllip.cpp",     
127        #srcdir+"/HayterMSA.c",         
128        "sans/models/c_models/CHayterMSAStructure.cpp", 
129        "sans/models/c_models/HayterMSA.cpp",             
130        "sans/models/c_models/sphere.cpp",
131        srcdir+"/fuzzysphere.c",
132        "sans/models/c_models/CFuzzySphereModel.cpp",
133        "sans/models/c_models/fuzzysphere.cpp",
134        #srcdir+"/CCylinderModel.c",
135        "sans/models/c_models/CCylinderModel.cpp",
136        "sans/models/c_models/cylinder.cpp",
137        "sans/models/c_models/parameters.cpp",
138        "sans/models/c_models/dispersion_visitor.cpp",
139        srcdir+"/cylinder.c",
140        #srcdir+"/CParallelepiped.c",
141        "sans/models/c_models/CParallelepipedModel.cpp",
142        "sans/models/c_models/parallelepiped.cpp",
143        srcdir+"/parallelepiped.c",
144        "sans/models/c_models/CCSParallelepipedModel.cpp",
145        "sans/models/c_models/csparallelepiped.cpp",
146        srcdir+"/csparallelepiped.c",
147        #srcdir+"/CCoreShellCylinderModel.c",
148        "sans/models/c_models/CCoreShellCylinderModel.cpp",
149        "sans/models/c_models/coreshellcylinder.cpp",
150        srcdir+"/core_shell_cylinder.c",
151        #srcdir+"/CHollowCylinderModel.c",
152        "sans/models/c_models/CHollowCylinderModel.cpp",
153        "sans/models/c_models/hollowcylinder.cpp",
154        srcdir+"/hollow_cylinder.c",
155        #srcdir+"/CCoreShellModel.c",
156        #srcdir+"/core_shell.c",
157        "sans/models/c_models/CCoreShellModel.cpp",
158        "sans/models/c_models/coreshellsphere.cpp",
159        #srcdir+"/CEllipsoidModel.c",
160        "sans/models/c_models/CEllipsoidModel.cpp",
161        "sans/models/c_models/ellipsoid.cpp",       
162        srcdir+"/ellipsoid.c",
163        "sans/models/c_models/CCoreFourShellModel.cpp",
164        "sans/models/c_models/corefourshell.cpp",
165        #srcdir+"/corefourshell.c",
166        #srcdir+"/CEllipticalCylinderModel.c",
167        "sans/models/c_models/CEllipticalCylinderModel.cpp",
168        "sans/models/c_models/ellipticalcylinder.cpp",               
169        srcdir+"/elliptical_cylinder.c",
170        #srcdir+"/CTriaxialEllipsoidModel.c",
171        "sans/models/c_models/CTriaxialEllipsoidModel.cpp",
172        "sans/models/c_models/triaxialellipsoid.cpp",               
173        srcdir+"/triaxial_ellipsoid.c",
174        #srcdir+"/CFlexibleCylinderModel.c",
175        "sans/models/c_models/CFlexibleCylinderModel.cpp",
176        "sans/models/c_models/flexiblecylinder.cpp",               
177        srcdir+"/flexible_cylinder.c",
178        "sans/models/c_models/CFlexCylEllipXModel.cpp",
179        "sans/models/c_models/flexcyl_ellipX.cpp",         
180        srcdir+"/flexcyl_ellipX.c",
181        "sans/models/c_models/CBarBellModel.cpp",
182        "sans/models/c_models/barbell.cpp",         
183        srcdir+"/barbell.c",
184        "sans/models/c_models/CCappedCylinderModel.cpp",
185        "sans/models/c_models/capcyl.cpp",         
186        srcdir+"/capcyl.c",
187        #srcdir+"/CStakedDisksModel.c",
188        "sans/models/c_models/CSCCrystalModel.cpp",
189        "sans/models/c_models/sc.cpp",               
190        srcdir+"/sc.c",
191        "sans/models/c_models/CFCCrystalModel.cpp",
192        "sans/models/c_models/fcc.cpp",               
193        srcdir+"/fcc.c",
194        "sans/models/c_models/CBCCrystalModel.cpp",
195        "sans/models/c_models/bcc.cpp",               
196        srcdir+"/bcc.c",
197        "sans/models/c_models/CStackedDisksModel.cpp",
198        "sans/models/c_models/stackeddisks.cpp",               
199        srcdir+"/stacked_disks.c",
200        #srcdir+"/CLamellarModel.c",
201        "sans/models/c_models/CLamellarModel.cpp",
202        "sans/models/c_models/lamellar.cpp",               
203        srcdir+"/lamellar.c",
204        #srcdir+"/CLamellarFFHGModel.c",
205        "sans/models/c_models/CLamellarFFHGModel.cpp",
206        "sans/models/c_models/lamellarFF_HG.cpp",               
207        #srcdir+"/lamellarFF_HG.c",
208        #srcdir+"/CLamellarPSModel.c",
209        "sans/models/c_models/CLamellarPSModel.cpp",
210        "sans/models/c_models/lamellarPS.cpp",               
211        srcdir+"/lamellarPS.c",
212        #srcdir+"/CLamellarPSHGModel.c",
213        "sans/models/c_models/CLamellarPSHGModel.cpp",
214        "sans/models/c_models/lamellarPS_HG.cpp",               
215        #srcdir+"/lamellarPS_HG.c",
216        "sans/models/c_models/CLamellarPCrystalModel.cpp",
217        "sans/models/c_models/lamellarPC.cpp",               
218        #srcdir+"/lamellarPC.c",
219        #srcdir+"/COblateModel.c",
220        "sans/models/c_models/CCoreShellEllipsoidModel.cpp",
221        "sans/models/c_models/spheroid.cpp",   
222        srcdir+"/spheroid.c",             
223        #srcdir+"/COblateModel.c",
224        #"sans/models/c_models/COblateModel.cpp",
225        #"sans/models/c_models/oblate.cpp",               
226        #srcdir+"/oblate.c",
227        #srcdir+"/CProlateModel.c",
228        #"sans/models/c_models/CProlateModel.cpp",
229        #"sans/models/c_models/prolate.cpp",               
230        #srcdir+"/prolate.c",
231        #srcdir+"/CMultishellModel.c",
232        "sans/models/c_models/CMultiShellModel.cpp",
233        "sans/models/c_models/multishell.cpp",               
234        #srcdir+"/multishell.c",
235        "sans/models/c_models/CSphereSLDModel.cpp",
236        "sans/models/c_models/spheresld.cpp",               
237        srcdir+"/spheresld.c",
238        #srcdir+"/CVesicleModel.c",
239        "sans/models/c_models/CVesicleModel.cpp",
240        "sans/models/c_models/vesicle.cpp",               
241        #srcdir+"/vesicle.c",
242        #srcdir+"/CBinaryHSModel.c",
243        "sans/models/c_models/CBinaryHSModel.cpp",
244        "sans/models/c_models/binaryHS.cpp",               
245        "sans/models/c_models/CPoly_GaussCoil.cpp",
246        "sans/models/c_models/polygausscoil.cpp",   
247        srcdir+"/rpa.c",             
248        "sans/models/c_models/CRPAModel.cpp",
249        "sans/models/c_models/rpa.cpp", 
250        #srcdir+"/fractal.c",             
251        "sans/models/c_models/CFractalModel.cpp",
252        "sans/models/c_models/fractal.cpp", 
253        srcdir+"/sld_cal.c",         
254        "sans/models/c_models/CSLDCalFunc.cpp", 
255        "sans/models/c_models/sld_cal.cpp", 
256        #gammainc function need to imported from somewhere 
257        #srcdir+"/polyexclvol.c",             
258        #"sans/models/c_models/CPolymerExclVolModel.cpp",
259        #"sans/models/c_models/polyexclvol.cpp", 
260        srcdir+"/libmultifunc/librefl.c",
261        #srcdir+"/polygausscoil.c",             
262        #srcdir+"/binaryHS.c",
263        srcdir+"/disperser.c",
264        igordir+"/libCylinder.c",
265        igordir+"/libStructureFactor.c",
266        igordir+"/libSphere.c",
267        igordir+"/libTwoPhase.c",
268        srcdir+"/gaussian.c",
269        srcdir+"/CGaussian.c",
270        srcdir+"/logNormal.c",
271        srcdir+"/CLogNormal.c",
272        srcdir+"/schulz.c",
273        srcdir+"/CSchulz.c",
274        srcdir+"/lorentzian.c",
275        srcdir+"/CLorentzian.c"
276            ],
277        include_dirs=[igordir,srcdir,"sans/models/c_models",numpy_incl_path]),       
278        # Smearer extension
279        Extension("sans_extension.smearer",
280                   sources = [
281        "sans/models/c_smearer/smearer_module.cpp",
282        "sans/models/c_smearer/smearer.cpp",
283        ],
284        include_dirs=["sans/models/c_smearer",numpy_incl_path]),
285        Extension("sans_extension.smearer2d_helper",
286                  sources = [
287        "sans/models/c_smearer/smearer2d_helper_module.cpp",
288        "sans/models/c_smearer/smearer2d_helper.cpp",
289        ],
290        include_dirs=["sans/models/c_smearer",numpy_incl_path]
291        )
292        ]
293    )
294       
Note: See TracBrowser for help on using the repository browser.