def __init__(self, filename):
""" Initialise from a file
"""
points = list()
for line in open(defaultExtension(filename, ".probe"), "r"):
line = line.strip()
if len(line) > 0:
parts = line.split(" ")
if len(parts) <> 3:
raise Exception("Unrecognised probe file format")
parts = list([ float(val) for val in parts ])
points.append(parts)
# Do some verification
if len(points) < 3:
raise Exception("File appears to be truncated")
self.xstart, self.xend, self.xsteps = points[0]
self.ystart, self.yend, self.ysteps = points[1]
self.zmin, self.zmax, self.feed = points[2]
points = points[3:]
if len(points) <> (self.xsteps * self.ysteps):
raise Exception("File appears to be truncated")
# Sort points into an X/Y array
self.points = points
self.pdict = dict()
total = 0.0
for x, y, z in points:
if not self.pdict.has_key(x):
self.pdict[x] = dict()
if not self.pdict[x].has_key(y):
self.pdict[x][y] = z
else:
raise Exception("Duplicate point found - %0.4f, %0.4f, %0.4f" % (x, y, z))
self.xvals = sorted(self.pdict.keys())
self.yvals = sorted(self.pdict[self.xvals[0]].keys())
# Calculate average and median
self.zlevels = sorted([ p[2] for p in points ])
self.zcount = len(self.zlevels)
self.average = sum(self.zlevels) / len(points)
index = self.zcount // 2
if len(self.zlevels) % 2:
self.median = self.zlevels[index]
else:
self.median = (self.zlevels[index - 1] + self.zlevels[index]) / 2.0
plt.imshow(my_data, cmap = 'copper')
plt.colorbar()
fig.set_size_inches((16, 8))
plt.savefig(filename, dpi = 100)
#--- Main program
if __name__ == "__main__":
# Set up program options
parser = OptionParser()
parser.add_option("-i", "--image", action="store_true", dest="image", default=False)
options, args = parser.parse_args()
# Check positional arguments
if len(args) <> 1:
print USAGE.strip() % argv[0]
exit(1)
# Load the probe file
# try:
probe = ProbeFile(args[0])
# except Exception , ex:
# LOG.FATAL("Could not load file '%s' - %s" % (args[0], ex))
# Display basic information
print "Probe from X %0.4f -> %0.4f, Y %0.4f -> %0.4f" % (probe.xstart, probe.xend, probe.ystart, probe.yend)
print "Z levels range from %0.4f to %0.4f" % (min(probe.zlevels), max(probe.zlevels))
print " Diff: %0.4f, Avg: %0.4f, Med: %0.4f" % (max(probe.zlevels) - min(probe.zlevels), probe.average, probe.median)
# Generate a plot if requested
if options.image:
probe.generateImage(defaultExtension(args[0], ".png", True))