comparison experiments/scripts/cnbh-syllables/results_plotting/spider_plot.py @ 100:ae195c41c7bd

- Python results plotting (finally).
- Proper results reporting script.
- Test on ALL talkers. The results script then generates a summary based on all the various subsets.
- Fixed chown users (hopefully sudos to be deleted entirely soon)
- More...

+#!/usr/bin/env python
+"""
+spider_plot.py
+
+Created by Thomas Walters on 2010-09-12.
+Copyright 2010 Google. All rights reserved.
+"""
+
+import numpy as np
+import pylab as p
+import mpl_toolkits.mplot3d.axes3d as p3
+import matplotlib as mpl
+from matplotlib import cm
+import matplotlib.ticker as ticker 
+
+total_value_count=185
+
+central_vtl=15
+central_vtl_scaling=112.32
+
+# Read in a file with lines in the form
+# Pitch Scale Percentage
+xs=[]
+ys=[]
+zs=[]
+f = open('plottable_results.txt', 'r')
+for line in f:
+  if line[0] != "#":
+    values = line.strip().split(' ')
+    xs.append(central_vtl*central_vtl_scaling/float(values[1]))
+    ys.append(float(values[0]))
+    zs.append(float(values[2]))
+
+
+# Define a tiny sphere, centered on the origin, which
+# we'll shift to the desired position. 
+u=np.r_[0:2*np.pi:50j]
+v=np.r_[0:np.pi:50j]
+sx=0.01*np.outer(np.cos(u),np.sin(v))
+sy=0.01*np.outer(np.sin(u),np.sin(v))
+sz=2.5*np.outer(np.ones(np.size(u)),np.cos(v))
+
+fig=p.figure()
+ax = p3.Axes3D(fig, azim=-80, elev=60)
+
+colormap = cm.get_cmap('jet', 100)
+
+# Note: here I fake out the lack of proper logarihmic scales on 3D axes in
+# matplotlib by just plotting log values on a linear scale and renaming
+# the labels.
+# (This doesn't work: ax.w_yaxis.set_scale('log') ax.w_xaxis.set_scale('log'))
+
+# Plot the values seven at a time as dark lines.
+# These are the individual spokes of the spoke pattern.
+n=7
+for i in xrange(0,8):
+  ax.plot(np.log(xs[i*n:(i+1)*n]), np.log(ys[i*n:(i+1)*n]), zs[i*n:(i+1)*n], color=[0,0,0])
+
+for x,y,z in zip(xs,ys,zs):
+  ax.plot(np.log([x, x]), np.log([y, y]), [z, 0], color=[0.8,0.8,0.8])
+  ax.plot_surface(sx+np.log(x),sy+np.log(y),sz+z, color=colormap(int(z)), linewidth=0)
+
+ax.set_ylabel('GPR/Hz')
+ax.set_xlabel('VTL/cm')
+ax.set_zlabel('Percent correct')
+ax.set_ylim3d(np.log([131,225]))
+ax.set_xlim3d(np.log([9.9, 22.1]))
+ax.set_zlim3d([-1, 101])
+ax.w_zaxis.set_major_locator(ticker.FixedLocator([0, 20, 40, 60, 80, 100]))
+
+ax.w_xaxis.set_major_locator(ticker.FixedLocator(np.log([10,15,22])))
+ax.w_xaxis.set_ticklabels(['10', '15', '22'])
+ax.w_yaxis.set_major_locator(ticker.FixedLocator(np.log([132, 172, 224])))
+ax.w_yaxis.set_ticklabels(['132', '172', '224'])
+
+#for a in ax.w_xaxis.get_ticklines()+ax.w_xaxis.get_ticklabels(): 
+#    a.set_visible(False) 
+
+#for a in ax.w_yaxis.get_ticklines()+ax.w_yaxis.get_ticklabels(): 
+#    a.set_visible(False) 
+
+
+#p.show()
+p.savefig('results.png')
+