Slight re-organization of C/C++ tools. Significant modifications to support observational data. Python and pipeline scripts added

python_tools/void_python_tools/plotting/__init__.py

@@ -0,0 +1,2 @@
+from plotTools import *
+from plotDefs import *

python_tools/void_python_tools/plotting/plotDefs.py

@@ -0,0 +1,13 @@
+colorList = ['r', 'b', 'g', 'y', 'c', 'm', 'y', 
+             'brown', 'grey',
+                          'darkred', 'orange', 'pink', 'darkblue',
+                          'lightblue', 'chocolate',
+                          'indigo', 'lightseagreen', 'maroon', 'olive',
+                          'royalblue', 'palevioletred', 'seagreen', 'tomato',
+                          'aquamarine', 'darkslateblue',
+                          'khaki', 'lawngreen', 'mediumorchid',
+                           'orangered', 'thistle'
+                          'yellowgreen']
+
+linewidth = 4
+fontsize = 12

python_tools/void_python_tools/plotting/plotTools.py

@@ -0,0 +1,69 @@
+__all__=['plotNumberCounts']
+
+from void_python_tools.backend.classes import *
+from plotDefs import *
+import numpy as np
+import os
+import pylab as plt
+
+# -----------------------------------------------------------------------------
+def plotNumberCounts(workDir=None, sampleList=None, figDir=None, plotNameBase="numbercount", 
+                     showPlot=False, dataPortion=None, setName=None):
+
+  plt.clf()
+  plt.xlabel("Comoving Distance (Mpc/h)")
+  plt.ylabel("Number of Voids")
+
+  plotTitle = setName
+
+  plotName = plotNameBase
+
+  xMin = 1.e00
+  xMax = 0
+
+  for (iSample,sample) in enumerate(sampleList):
+
+    sampleName = sample.fullName
+    lineTitle = sampleName
+
+    filename = workDir+"/sample_"+sampleName+"/centers_"+dataPortion+"_"+\
+               sampleName+".out"
+    if not os.access(filename, os.F_OK):
+      print "File not found: ", filename
+      continue
+
+    data = np.loadtxt(filename, comments="#")
+    if data.ndim == 1:
+      print " Too few!"
+      continue
+
+    zMin = sample.zRange[0]
+    zMax = sample.zRange[1]
+
+    range = (zMin, zMax)
+    nbins = np.ceil((zMax-zMin)/0.1)
+
+    thisMax = np.max(data[:,5])
+    thisMin = np.min(data[:,5])
+    if thisMax > xMax: xMax = thisMax
+    if thisMin < xMin: xMin = thisMin
+
+    plt.hist(data[:,5], bins=nbins,
+             label=lineTitle, color=colorList[iSample],
+             histtype = "step", range=range,
+             linewidth=linewidth)
+
+  #plt.legend(title = "Samples", loc = "upper right")
+  plt.title(plotTitle)
+
+  plt.xlim(xMin, xMax)
+  #plt.xlim(xMin, xMax*1.4) # make room for legend
+
+  plt.savefig(figDir+"/fig_"+plotName+".pdf", bbox_inches="tight")
+  plt.savefig(figDir+"/fig_"+plotName+".eps", bbox_inches="tight")
+  plt.savefig(figDir+"/fig_"+plotName+".png", bbox_inches="tight")
+
+  if showPlot:
+    os.system("display %s" % figDir+"/fig_"+plotName+".png")
+
+