#!/usr/bin/python

import sys
import pickle

from grid        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())