From c323511b05918335e0349c8f9be12d3f38a945d9 Mon Sep 17 00:00:00 2001 From: Stephen Mardson McQuay Date: Wed, 26 Jan 2011 23:01:18 -0700 Subject: [PATCH] added functionality for supporting the GL front-end --- interp/grid/__init__.py | 31 ++++++++++++++++++++++++++++++- 1 file changed, 30 insertions(+), 1 deletion(-) diff --git a/interp/grid/__init__.py b/interp/grid/__init__.py index a935cba..e88cc03 100644 --- a/interp/grid/__init__.py +++ b/interp/grid/__init__.py @@ -2,6 +2,8 @@ import sys import re from collections import defaultdict +from xml.dom.minidom import Document + import numpy as np from scipy.spatial import KDTree @@ -23,7 +25,7 @@ class grid(object): q = array (1D) of physical values """ - if verts: + if verts != None: self.verts = np.array(verts) self.tree = KDTree(self.verts) if q: @@ -142,6 +144,33 @@ class grid(object): r += "%s\n" % v return r + def normalize_q(self, new_max = 0.1): + largest_number = np.max(np.abs(self.q)) + self.q *= new_max/largest_number + + + + def get_xml(self): + doc = Document() + ps = doc.createElement("points") + doc.appendChild(ps) + for i in zip(self.verts, self.q): + p = doc.createElement("point") + + p.setAttribute("x", str(i[0][0])) + p.setAttribute('y', str(i[0][1])) + p.setAttribute('z', str(i[0][2])) + p.setAttribute('q', str(i[1] )) + ps.appendChild(p) + + return doc + + def toxml(self): + return self.get_xml().toxml() + def toprettyxml(self): + return self.get_xml().toprettyxml() + + class delaunay_grid(grid):