smbinterp/bin/test3d-connectivity.py

#!/usr/bin/env python

import sys
from interp.grid.DDD import random_grid
from interp.baker import get_phis, run_baker
from interp.tools import exact_func, improved_answer, log
from glob import glob
from os import remove

import numpy as np

import pylab

def draw_gb(bad, good):
  pylab.xlabel('total runs')
  pylab.ylabel('count')
  pylab.grid(True)
  pylab.plot(bad)
  pylab.plot(good)
  pylab.legend(('bad', 'good'))

  pylab.show()

FILE_PREFIX='/tmp/qhull-'


if __name__ == '__main__':
  for g in glob('%s*' % FILE_PREFIX):
    remove(g)

  try:
    total_points   = int(sys.argv[1])
    random_points  = int(sys.argv[2])
    total_tries    = int(sys.argv[3])
  except:
    print "usage: app.py [total points in random grid] [number of random points to attempt] [total attempts]"
    sys.exit(1)


  print "total points", total_points

  d = {True: 0, False: 0}
  good = []
  bad = []
  for cur_try in xrange(total_tries):
    g = random_grid(total_points)
    open('%s%0.3d.txt' % (FILE_PREFIX, cur_try), 'w').write(g.for_qhull())

    for i in xrange(random_points):
      X = [np.random.rand(), np.random.rand(), np.random.rand()]
      exact = exact_func(X)

      try:
        try:
          answer =  g.run_baker(X)
          d[improved_answer(answer, exact)] += 1
        except ValueError as e:
          log.error(e)
      except Exception as e:
        print e
        print d

    print d
    bad.append(d[False])
    good.append(d[True])

  draw_gb(bad = bad, good = good)