core_shell_microgelsmagnetic_modelticket-1257-vesicle-productticket_1156ticket_1265_superballticket_822_more_unit_tests
Last change
on this file since 610ef23 was
ff31782,
checked in by Paul Kienzle <pkienzle@…>, 7 years ago
|
add -ngauss option to compare in order to set the number of integration points
|
-
Property mode set to
100755
|
File size:
1.1 KB
|
Rev | Line | |
---|
[ff31782] | 1 | #!/usr/bin/env python |
---|
| 2 | from __future__ import division, print_function |
---|
| 3 | |
---|
| 4 | import sys |
---|
| 5 | |
---|
| 6 | import numpy as np |
---|
| 7 | from numpy.polynomial.legendre import leggauss |
---|
| 8 | |
---|
| 9 | def gengauss(n, path): |
---|
| 10 | z, w = leggauss(n) |
---|
| 11 | |
---|
| 12 | # Make sure array size is a multiple of 4 |
---|
| 13 | if n%4: |
---|
| 14 | array_size = n + (4 - n%4) |
---|
| 15 | z, w = [np.hstack((v, [0.]*(4-n%4))) for v in (z, w)] |
---|
| 16 | else: |
---|
| 17 | array_size = n |
---|
| 18 | |
---|
| 19 | with open(path, "w") as fid: |
---|
| 20 | fid.write("""\ |
---|
| 21 | // Generated by sasmodels.gengauss.gengauss(%d) |
---|
| 22 | |
---|
| 23 | #ifdef GAUSS_N |
---|
| 24 | # undef GAUSS_N |
---|
| 25 | # undef GAUSS_Z |
---|
| 26 | # undef GAUSS_W |
---|
| 27 | #endif |
---|
| 28 | #define GAUSS_N %d |
---|
| 29 | #define GAUSS_Z Gauss%dZ |
---|
| 30 | #define GAUSS_W Gauss%dWt |
---|
| 31 | |
---|
| 32 | """%(n, n, n, n)) |
---|
| 33 | |
---|
| 34 | if array_size != n: |
---|
| 35 | fid.write("// Note: using array size %d so that it is a multiple of 4\n\n"%array_size) |
---|
| 36 | |
---|
| 37 | fid.write("constant double Gauss%dWt[%d]={\n"%(n, array_size)) |
---|
| 38 | fid.write(",\n".join("\t% .15e"%v for v in w)) |
---|
| 39 | fid.write("\n};\n") |
---|
| 40 | |
---|
| 41 | fid.write("constant double Gauss%dZ[%d]={\n"%(n, array_size)) |
---|
| 42 | fid.write(",\n".join("\t% .15e"%v for v in z)) |
---|
| 43 | fid.write("\n};") |
---|
Note: See
TracBrowser
for help on using the repository browser.