def MakeTable(Obs_data,Obs_xml,filename):
pressure_amanzi = []
coordinates = []
mymodel = model_linear_head_head_1d.createFromXML(filename)
for obs in Obs_data.observations.itervalues():
coordinates.append([obs.coordinate[0], obs.coordinate[2]])
pressure_amanzi.append(str(obs.data).rstrip(']').lstrip('['))
pressure_analytic = list(mymodel.pressure(numpy.array(coordinates)))
x = prettytable.PrettyTable(["x [m]", "z [m]", "Analytic [Pa]","Amanzi [Pa]"])
x.padding_width = 2
x.hrules = 1
x.horizontal_header_char="="
for coords, p_analytic, p_amanzi in zip(coordinates,pressure_analytic,pressure_amanzi):
x.add_row([coords[0],coords[1],"%.4f" % float(p_analytic),"%.4f" % float(p_amanzi)])
if os.path.exists("table_values.txt"):
os.remove("table_values.txt")
table_file = open("table_values.txt", "w+")
table_file.write('.. tabularcolumns:: ' + '|R|C|C|C|' + '\n\n')
table_file.write(x.get_string(sortby="x [m]"))
table_file.write('\n')
table_file.close()
def plotTestModel(filename, cmap, axes1, Obs_xml, Obs_data):
# Instantiate the analytic solution
mymodel = model_linear_head_head_1d.createFromXML(filename)
table_values = []
# Create a set of points to plot the solution
x = numpy.linspace(mymodel.x_0,mymodel.x_1,11)
coords = numpy.zeros((11,2))
coords[:,0] = x
# Plot a line for each z-coordinate in the observations
for (z_val, color) in cmap.iteritems():
coords[:,1] = z_val
pressure = mymodel.pressure(coords)
axes1.plot(x,pressure,color,label='$z = %0.2f $'%z_val)
axes1.legend(loc="upper right" , fancybox = True , shadow = True)