match mrcal

Author: mcm001

heatmap-gen.py

@@ -11,14 +11,16 @@
 plt.figure()
 plt.title("Projection Uncertainty (in pixels), looking out to infinity")
 
-sc = plt.scatter(x, y, c=z)
-plt.colorbar(sc, label="z")
-
-# force heat range to go from 0 to 10
-plt.clim(0, 10)
+# Create contour plot with 1px increments
+levels = np.arange(0, 11, 1)  # 0, 1, 2, ..., 10
+contour = plt.tricontour(x, y, z, levels=levels, colors='black', linewidths=0.5)
+contourf = plt.tricontourf(x, y, z, levels=levels, cmap='viridis')
+plt.clabel(contour, inline=True, fontsize=8, fmt='%1.0f px')
+plt.colorbar(contourf, label="Uncertainty (px)")
 
 plt.xlabel("x")
 plt.ylabel("y")
+plt.gca().invert_yaxis()
 plt.axis("equal")
 plt.tight_layout()