ex-3: implement variable bins in the plot

ex-3/main.c

@@ -100,6 +100,11 @@ int main(int argc, char **argv) {
   // print histogram to stdout
   if (opts.show_hist || opts.only_hist) {
     fputs("\n# histogram\n", stderr);
+    /* Print the bins on the first line:
+     * this is needed by plot.py to make a
+     * 2D histogram plot.
+     */
+    fprintf(stdout, "%ld %ld\n", opts.bins_theta, opts.bins_phi);
     gsl_histogram2d_fprintf(stdout, hist, "%g", "%g");
   }
 

ex-3/plot.py

@@ -5,8 +5,8 @@ from mpl_toolkits.mplot3d import Axes3D
 from matplotlib import cm
 import sys
 
+bins = tuple(map(int, sys.stdin.readline().split()))
 xedges, yedges, counts = loadtxt(sys.stdin, unpack=True, usecols=[0,2,4])
-bins = (30, 60)
 counts = counts.reshape(bins)
 
 plt.rcParams['font.size'] = 15