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 | """ |
---|
12 | import sys |
---|
13 | import os |
---|
14 | |
---|
15 | from numpy.distutils.misc_util import get_numpy_include_dirs |
---|
16 | numpy_incl_path = os.path.join(get_numpy_include_dirs()[0], "numpy") |
---|
17 | |
---|
18 | def 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 | |
---|
41 | def 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 |
---|
59 | if 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 |
---|
69 | from distutils.core import setup, Extension |
---|
70 | |
---|
71 | |
---|
72 | # Build the module name |
---|
73 | srcdir = "sans/models/c_extensions" |
---|
74 | igordir = "libigor" |
---|
75 | |
---|
76 | print "Installing SANS models" |
---|
77 | |
---|
78 | |
---|
79 | setup( |
---|
80 | name="models", |
---|
81 | version = "0.1", |
---|
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 | srcdir+"/SquareWell.c", |
---|
103 | "sans/models/c_models/CSquareWellStructure.cpp", |
---|
104 | "sans/models/c_models/SquareWell.cpp", |
---|
105 | srcdir+"/StickyHS.c", |
---|
106 | "sans/models/c_models/CStickyHSStructure.cpp", |
---|
107 | "sans/models/c_models/StickyHS.cpp", |
---|
108 | srcdir+"/Hardsphere.c", |
---|
109 | "sans/models/c_models/CHardsphereStructure.cpp", |
---|
110 | "sans/models/c_models/Hardsphere.cpp", |
---|
111 | srcdir+"/DiamCyl.c", |
---|
112 | "sans/models/c_models/CDiamCylFunc.cpp", |
---|
113 | "sans/models/c_models/DiamCyl.cpp", |
---|
114 | srcdir+"/DiamEllip.c", |
---|
115 | "sans/models/c_models/CDiamEllipFunc.cpp", |
---|
116 | "sans/models/c_models/DiamEllip.cpp", |
---|
117 | srcdir+"/HayterMSA.c", |
---|
118 | "sans/models/c_models/CHayterMSAStructure.cpp", |
---|
119 | "sans/models/c_models/HayterMSA.cpp", |
---|
120 | "sans/models/c_models/sphere.cpp", |
---|
121 | srcdir+"/fuzzysphere.c", |
---|
122 | "sans/models/c_models/CFuzzySphereModel.cpp", |
---|
123 | "sans/models/c_models/fuzzysphere.cpp", |
---|
124 | #srcdir+"/CCylinderModel.c", |
---|
125 | "sans/models/c_models/CCylinderModel.cpp", |
---|
126 | "sans/models/c_models/cylinder.cpp", |
---|
127 | "sans/models/c_models/parameters.cpp", |
---|
128 | "sans/models/c_models/dispersion_visitor.cpp", |
---|
129 | srcdir+"/cylinder.c", |
---|
130 | #srcdir+"/CParallelepiped.c", |
---|
131 | "sans/models/c_models/CParallelepipedModel.cpp", |
---|
132 | "sans/models/c_models/parallelepiped.cpp", |
---|
133 | srcdir+"/parallelepiped.c", |
---|
134 | #srcdir+"/CCoreShellCylinderModel.c", |
---|
135 | "sans/models/c_models/CCoreShellCylinderModel.cpp", |
---|
136 | "sans/models/c_models/coreshellcylinder.cpp", |
---|
137 | srcdir+"/core_shell_cylinder.c", |
---|
138 | #srcdir+"/CHollowCylinderModel.c", |
---|
139 | "sans/models/c_models/CHollowCylinderModel.cpp", |
---|
140 | "sans/models/c_models/hollowcylinder.cpp", |
---|
141 | srcdir+"/hollow_cylinder.c", |
---|
142 | #srcdir+"/CCoreShellModel.c", |
---|
143 | #srcdir+"/core_shell.c", |
---|
144 | "sans/models/c_models/CCoreShellModel.cpp", |
---|
145 | "sans/models/c_models/coreshellsphere.cpp", |
---|
146 | #srcdir+"/CEllipsoidModel.c", |
---|
147 | "sans/models/c_models/CEllipsoidModel.cpp", |
---|
148 | "sans/models/c_models/ellipsoid.cpp", |
---|
149 | srcdir+"/ellipsoid.c", |
---|
150 | "sans/models/c_models/CCoreFourShellModel.cpp", |
---|
151 | "sans/models/c_models/corefourshell.cpp", |
---|
152 | srcdir+"/corefourshell.c", |
---|
153 | #srcdir+"/CEllipticalCylinderModel.c", |
---|
154 | "sans/models/c_models/CEllipticalCylinderModel.cpp", |
---|
155 | "sans/models/c_models/ellipticalcylinder.cpp", |
---|
156 | srcdir+"/elliptical_cylinder.c", |
---|
157 | #srcdir+"/CTriaxialEllipsoidModel.c", |
---|
158 | "sans/models/c_models/CTriaxialEllipsoidModel.cpp", |
---|
159 | "sans/models/c_models/triaxialellipsoid.cpp", |
---|
160 | srcdir+"/triaxial_ellipsoid.c", |
---|
161 | #srcdir+"/CFlexibleCylinderModel.c", |
---|
162 | "sans/models/c_models/CFlexibleCylinderModel.cpp", |
---|
163 | "sans/models/c_models/flexiblecylinder.cpp", |
---|
164 | srcdir+"/flexible_cylinder.c", |
---|
165 | "sans/models/c_models/CFlexCylEllipXModel.cpp", |
---|
166 | "sans/models/c_models/flexcyl_ellipX.cpp", |
---|
167 | srcdir+"/flexcyl_ellipX.c", |
---|
168 | #srcdir+"/CStakedDisksModel.c", |
---|
169 | "sans/models/c_models/CStackedDisksModel.cpp", |
---|
170 | "sans/models/c_models/stackeddisks.cpp", |
---|
171 | srcdir+"/stacked_disks.c", |
---|
172 | #srcdir+"/CLamellarModel.c", |
---|
173 | "sans/models/c_models/CLamellarModel.cpp", |
---|
174 | "sans/models/c_models/lamellar.cpp", |
---|
175 | srcdir+"/lamellar.c", |
---|
176 | #srcdir+"/CLamellarFFHGModel.c", |
---|
177 | "sans/models/c_models/CLamellarFFHGModel.cpp", |
---|
178 | "sans/models/c_models/lamellarFF_HG.cpp", |
---|
179 | srcdir+"/lamellarFF_HG.c", |
---|
180 | #srcdir+"/CLamellarPSModel.c", |
---|
181 | "sans/models/c_models/CLamellarPSModel.cpp", |
---|
182 | "sans/models/c_models/lamellarPS.cpp", |
---|
183 | srcdir+"/lamellarPS.c", |
---|
184 | #srcdir+"/CLamellarPSHGModel.c", |
---|
185 | "sans/models/c_models/CLamellarPSHGModel.cpp", |
---|
186 | "sans/models/c_models/lamellarPS_HG.cpp", |
---|
187 | srcdir+"/lamellarPS_HG.c", |
---|
188 | #srcdir+"/COblateModel.c", |
---|
189 | "sans/models/c_models/CCoreShellEllipsoidModel.cpp", |
---|
190 | "sans/models/c_models/spheroid.cpp", |
---|
191 | srcdir+"/spheroid.c", |
---|
192 | #srcdir+"/COblateModel.c", |
---|
193 | #"sans/models/c_models/COblateModel.cpp", |
---|
194 | #"sans/models/c_models/oblate.cpp", |
---|
195 | #srcdir+"/oblate.c", |
---|
196 | #srcdir+"/CProlateModel.c", |
---|
197 | #"sans/models/c_models/CProlateModel.cpp", |
---|
198 | #"sans/models/c_models/prolate.cpp", |
---|
199 | #srcdir+"/prolate.c", |
---|
200 | #srcdir+"/CMultishellModel.c", |
---|
201 | "sans/models/c_models/CMultiShellModel.cpp", |
---|
202 | "sans/models/c_models/multishell.cpp", |
---|
203 | srcdir+"/multishell.c", |
---|
204 | #srcdir+"/CVesicleModel.c", |
---|
205 | "sans/models/c_models/CVesicleModel.cpp", |
---|
206 | "sans/models/c_models/vesicle.cpp", |
---|
207 | srcdir+"/vesicle.c", |
---|
208 | #srcdir+"/CBinaryHSModel.c", |
---|
209 | "sans/models/c_models/CBinaryHSModel.cpp", |
---|
210 | "sans/models/c_models/binaryHS.cpp", |
---|
211 | "sans/models/c_models/CPoly_GaussCoil.cpp", |
---|
212 | "sans/models/c_models/polygausscoil.cpp", |
---|
213 | srcdir+"/fractal.c", |
---|
214 | "sans/models/c_models/CFractalModel.cpp", |
---|
215 | "sans/models/c_models/fractal.cpp", |
---|
216 | |
---|
217 | #gammainc function need to imported from somewhere |
---|
218 | #srcdir+"/polyexclvol.c", |
---|
219 | #"sans/models/c_models/CPolymerExclVolModel.cpp", |
---|
220 | #"sans/models/c_models/polyexclvol.cpp", |
---|
221 | |
---|
222 | srcdir+"/polygausscoil.c", |
---|
223 | srcdir+"/binaryHS.c", |
---|
224 | srcdir+"/disperser.c", |
---|
225 | igordir+"/libCylinder.c", |
---|
226 | igordir+"/libStructureFactor.c", |
---|
227 | igordir+"/libSphere.c", |
---|
228 | igordir+"/libTwoPhase.c", |
---|
229 | srcdir+"/gaussian.c", |
---|
230 | srcdir+"/CGaussian.c", |
---|
231 | srcdir+"/logNormal.c", |
---|
232 | srcdir+"/CLogNormal.c", |
---|
233 | srcdir+"/schulz.c", |
---|
234 | srcdir+"/CSchulz.c", |
---|
235 | srcdir+"/lorentzian.c", |
---|
236 | srcdir+"/CLorentzian.c" |
---|
237 | ], |
---|
238 | include_dirs=[igordir,srcdir,"sans/models/c_models",numpy_incl_path])]) |
---|
239 | |
---|