smbinterp/bin/grid_regular.py

43 lines
1.0 KiB
Python
Executable File

#!/usr/bin/python2.6
import sys
import pickle
from grid.DD import simple_rect_grid, simple_random_grid
from baker import run_baker
from baker.tools import smberror
qfile = '/tmp/grid_regular.txt'
if __name__ == '__main__':
try:
rx = int(sys.argv[1])
ry = int(sys.argv[2])
except (IndexError, ValueError) as e:
print "problem with argv: %s" % e
rx = 4
ry = 4 * rx
source_mesh = simple_rect_grid(rx, ry)
# print source_mesh
X = [0.1, 0.1]
try:
(R, S) = source_mesh.get_simplex_and_nearest_points(X, extra_points=4)
print "R for nearest-neighbor:\n", R
print "S for nearest-neighbor:\n", S
print run_baker(X, R, S)
except smberror as e:
print "caught error: %s" % e
(R, S) = source_mesh.get_points_conn(X)
print "R for connectivity:\n", R
print "S for connectivity:\n", S
print run_baker(X, R, S)
print "repeating the above just using the grid object:"
print source_mesh.run_baker(X)
open(qfile, 'w').write(source_mesh.for_qhull())