source: sasview/sansmodels/src/setup.py @ 96d19c6

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 96d19c6 was 96d19c6, checked in by Jae Cho <jhjcho@…>, 14 years ago

added models and unit tests

  • Property mode set to 100644
File size: 11.0 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 setup, Extension
70
71
72# Build the module name
73srcdir  = "sans/models/c_extensions"
74igordir = "libigor"
75
76print "Installing SANS models"
77
78
79setup(
80    name="models",
81    version = "0.4",
82    description = "Python module for SANS scattering models",
83    author = "Mathieu Doucet",
84    author_email = "doucet@nist.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   
93    packages = ["sans","sans.models","sans.models.test",
94                "sans_extension","sans.models.pyre"],
95   
96    ext_modules = [ Extension("sans_extension.c_models",
97     sources = [
98        "sans/models/c_models/c_models.cpp",
99        #srcdir+"/CSphereModel.c",
100        #srcdir+"/sphere.c",
101        "sans/models/c_models/CSphereModel.cpp",
102        "sans/models/c_models/COnionModel.cpp",
103        "sans/models/c_models/onion.cpp",
104        srcdir+"/onion.c",
105        "sans/models/c_models/CReflModel.cpp",
106        "sans/models/c_models/refl.cpp",
107        srcdir+"/refl.c",
108        srcdir+"/SquareWell.c",
109        "sans/models/c_models/CSquareWellStructure.cpp", 
110        "sans/models/c_models/SquareWell.cpp", 
111        srcdir+"/StickyHS.c",         
112        "sans/models/c_models/CStickyHSStructure.cpp", 
113        "sans/models/c_models/StickyHS.cpp",     
114        srcdir+"/Hardsphere.c",         
115        "sans/models/c_models/CHardsphereStructure.cpp", 
116        "sans/models/c_models/Hardsphere.cpp",   
117        srcdir+"/DiamCyl.c",         
118        "sans/models/c_models/CDiamCylFunc.cpp", 
119        "sans/models/c_models/DiamCyl.cpp",   
120        srcdir+"/DiamEllip.c",         
121        "sans/models/c_models/CDiamEllipFunc.cpp", 
122        "sans/models/c_models/DiamEllip.cpp",     
123        srcdir+"/HayterMSA.c",         
124        "sans/models/c_models/CHayterMSAStructure.cpp", 
125        "sans/models/c_models/HayterMSA.cpp",             
126        "sans/models/c_models/sphere.cpp",
127        srcdir+"/fuzzysphere.c",
128        "sans/models/c_models/CFuzzySphereModel.cpp",
129        "sans/models/c_models/fuzzysphere.cpp",
130        #srcdir+"/CCylinderModel.c",
131        "sans/models/c_models/CCylinderModel.cpp",
132        "sans/models/c_models/cylinder.cpp",
133        "sans/models/c_models/parameters.cpp",
134        "sans/models/c_models/dispersion_visitor.cpp",
135        srcdir+"/cylinder.c",
136        #srcdir+"/CParallelepiped.c",
137        "sans/models/c_models/CParallelepipedModel.cpp",
138        "sans/models/c_models/parallelepiped.cpp",
139        srcdir+"/parallelepiped.c",
140        "sans/models/c_models/CCSParallelepipedModel.cpp",
141        "sans/models/c_models/csparallelepiped.cpp",
142        srcdir+"/csparallelepiped.c",
143        #srcdir+"/CCoreShellCylinderModel.c",
144        "sans/models/c_models/CCoreShellCylinderModel.cpp",
145        "sans/models/c_models/coreshellcylinder.cpp",
146        srcdir+"/core_shell_cylinder.c",
147        #srcdir+"/CHollowCylinderModel.c",
148        "sans/models/c_models/CHollowCylinderModel.cpp",
149        "sans/models/c_models/hollowcylinder.cpp",
150        srcdir+"/hollow_cylinder.c",
151        #srcdir+"/CCoreShellModel.c",
152        #srcdir+"/core_shell.c",
153        "sans/models/c_models/CCoreShellModel.cpp",
154        "sans/models/c_models/coreshellsphere.cpp",
155        #srcdir+"/CEllipsoidModel.c",
156        "sans/models/c_models/CEllipsoidModel.cpp",
157        "sans/models/c_models/ellipsoid.cpp",       
158        srcdir+"/ellipsoid.c",
159        "sans/models/c_models/CCoreFourShellModel.cpp",
160        "sans/models/c_models/corefourshell.cpp",
161        srcdir+"/corefourshell.c",
162        #srcdir+"/CEllipticalCylinderModel.c",
163        "sans/models/c_models/CEllipticalCylinderModel.cpp",
164        "sans/models/c_models/ellipticalcylinder.cpp",               
165        srcdir+"/elliptical_cylinder.c",
166        #srcdir+"/CTriaxialEllipsoidModel.c",
167        "sans/models/c_models/CTriaxialEllipsoidModel.cpp",
168        "sans/models/c_models/triaxialellipsoid.cpp",               
169        srcdir+"/triaxial_ellipsoid.c",
170        #srcdir+"/CFlexibleCylinderModel.c",
171        "sans/models/c_models/CFlexibleCylinderModel.cpp",
172        "sans/models/c_models/flexiblecylinder.cpp",               
173        srcdir+"/flexible_cylinder.c",
174        "sans/models/c_models/CFlexCylEllipXModel.cpp",
175        "sans/models/c_models/flexcyl_ellipX.cpp",         
176        srcdir+"/flexcyl_ellipX.c",
177        "sans/models/c_models/CBarBellModel.cpp",
178        "sans/models/c_models/barbell.cpp",         
179        srcdir+"/barbell.c",
180        "sans/models/c_models/CCappedCylinderModel.cpp",
181        "sans/models/c_models/capcyl.cpp",         
182        srcdir+"/capcyl.c",
183        #srcdir+"/CStakedDisksModel.c",
184        "sans/models/c_models/CSCCrystalModel.cpp",
185        "sans/models/c_models/sc.cpp",               
186        srcdir+"/sc.c",
187        "sans/models/c_models/CFCCrystalModel.cpp",
188        "sans/models/c_models/fcc.cpp",               
189        srcdir+"/fcc.c",
190        "sans/models/c_models/CBCCrystalModel.cpp",
191        "sans/models/c_models/bcc.cpp",               
192        srcdir+"/bcc.c",
193        "sans/models/c_models/CStackedDisksModel.cpp",
194        "sans/models/c_models/stackeddisks.cpp",               
195        srcdir+"/stacked_disks.c",
196        #srcdir+"/CLamellarModel.c",
197        "sans/models/c_models/CLamellarModel.cpp",
198        "sans/models/c_models/lamellar.cpp",               
199        srcdir+"/lamellar.c",
200        #srcdir+"/CLamellarFFHGModel.c",
201        "sans/models/c_models/CLamellarFFHGModel.cpp",
202        "sans/models/c_models/lamellarFF_HG.cpp",               
203        srcdir+"/lamellarFF_HG.c",
204        #srcdir+"/CLamellarPSModel.c",
205        "sans/models/c_models/CLamellarPSModel.cpp",
206        "sans/models/c_models/lamellarPS.cpp",               
207        srcdir+"/lamellarPS.c",
208        #srcdir+"/CLamellarPSHGModel.c",
209        "sans/models/c_models/CLamellarPSHGModel.cpp",
210        "sans/models/c_models/lamellarPS_HG.cpp",               
211        srcdir+"/lamellarPS_HG.c",
212        "sans/models/c_models/CLamellarPCrystalModel.cpp",
213        "sans/models/c_models/lamellarPC.cpp",               
214        srcdir+"/lamellarPC.c",
215        #srcdir+"/COblateModel.c",
216        "sans/models/c_models/CCoreShellEllipsoidModel.cpp",
217        "sans/models/c_models/spheroid.cpp",   
218        srcdir+"/spheroid.c",             
219        #srcdir+"/COblateModel.c",
220        #"sans/models/c_models/COblateModel.cpp",
221        #"sans/models/c_models/oblate.cpp",               
222        #srcdir+"/oblate.c",
223        #srcdir+"/CProlateModel.c",
224        #"sans/models/c_models/CProlateModel.cpp",
225        #"sans/models/c_models/prolate.cpp",               
226        #srcdir+"/prolate.c",
227        #srcdir+"/CMultishellModel.c",
228        "sans/models/c_models/CMultiShellModel.cpp",
229        "sans/models/c_models/multishell.cpp",               
230        srcdir+"/multishell.c",
231        "sans/models/c_models/CSphereSLDModel.cpp",
232        "sans/models/c_models/spheresld.cpp",               
233        srcdir+"/spheresld.c",
234        #srcdir+"/CVesicleModel.c",
235        "sans/models/c_models/CVesicleModel.cpp",
236        "sans/models/c_models/vesicle.cpp",               
237        srcdir+"/vesicle.c",
238        #srcdir+"/CBinaryHSModel.c",
239        "sans/models/c_models/CBinaryHSModel.cpp",
240        "sans/models/c_models/binaryHS.cpp",               
241        "sans/models/c_models/CPoly_GaussCoil.cpp",
242        "sans/models/c_models/polygausscoil.cpp",   
243        srcdir+"/rpa.c",             
244        "sans/models/c_models/CRPAModel.cpp",
245        "sans/models/c_models/rpa.cpp", 
246        srcdir+"/fractal.c",             
247        "sans/models/c_models/CFractalModel.cpp",
248        "sans/models/c_models/fractal.cpp", 
249        srcdir+"/sld_cal.c",         
250        "sans/models/c_models/CSLDCalFunc.cpp", 
251        "sans/models/c_models/sld_cal.cpp", 
252        #gammainc function need to imported from somewhere 
253        #srcdir+"/polyexclvol.c",             
254        #"sans/models/c_models/CPolymerExclVolModel.cpp",
255        #"sans/models/c_models/polyexclvol.cpp", 
256        srcdir+"/libmultifunc/librefl.c",
257        srcdir+"/polygausscoil.c",             
258        srcdir+"/binaryHS.c",
259        srcdir+"/disperser.c",
260        igordir+"/libCylinder.c",
261        igordir+"/libStructureFactor.c",
262        igordir+"/libSphere.c",
263        igordir+"/libTwoPhase.c",
264        srcdir+"/gaussian.c",
265        srcdir+"/CGaussian.c",
266        srcdir+"/logNormal.c",
267        srcdir+"/CLogNormal.c",
268        srcdir+"/schulz.c",
269        srcdir+"/CSchulz.c",
270        srcdir+"/lorentzian.c",
271        srcdir+"/CLorentzian.c"
272            ],
273         include_dirs=[igordir,srcdir,"sans/models/c_models",numpy_incl_path])])
274       
Note: See TracBrowser for help on using the repository browser.