gp_sims: start gp_sims_panel

pyana/examples/gp_sims.py

@@ -1,7 +1,7 @@
 import os, argparse, logging
 from .utils import getWorkDirs, checkSymLink, getEnergy4Key, particleLabel4Key
 from collections import OrderedDict
-from ..ccsgp.ccsgp import make_plot
+from ..ccsgp.ccsgp import make_plot, make_panel
 from ..ccsgp.config import default_colors
 import numpy as np
 
@@ -57,6 +57,39 @@ def gp_sims(version):
           }
       )
 
+def gp_sims_panel(version):
+  """panel plot of cocktail simulations at all energies, includ. total
+
+  :param version: plot version / input subdir name
+  :type version: str
+  """
+  inDir, outDir = getWorkDirs()
+  inDir = os.path.join(inDir, version, 'cocktail_contribs')
+  energies = [19, 27, 39, 62]
+  mesons = ['pion', 'eta', 'etap', 'rho', 'omega', 'phi', 'jpsi']
+  data = OrderedDict((energy, [
+      np.loadtxt(open(os.path.join(inDir, m+str(energy)+'.dat'), 'rb'))
+      for m in mesons
+  ]) for energy in energies)
+  for v in data.values():
+      for d in v:
+          d[:,2:] = 0
+  make_panel(
+      dpt_dict = OrderedDict((
+          ' '.join([getEnergy4Key(str(energy)), 'GeV']),
+          [ data[energy], [
+              'with lines lc %s lw 4 lt 1' % default_colors[-i-2]
+              for i in xrange(len(mesons))
+          ], mesons]
+      ) for energy in energies),
+      name = os.path.join(outDir, 'sims_panel'),
+      ylog = True, xr = [0.,3.2], yr = [1e-6,9],
+      ylabel = '1/N@_{mb}^{evt} dN@_{ee}^{acc.}/dM_{ee} [ (GeV/c^2)^{-1} ]',
+      xlabel = 'invariant dielectron mass, M_{ee} (GeV/c^{2})',
+      #layout = '2x2', size = '8in,8in',
+      size = '5in,12in',
+  )
+
 if __name__ == '__main__':
   checkSymLink()
   parser = argparse.ArgumentParser()
@@ -67,4 +100,5 @@ def gp_sims(version):
   logging.basicConfig(
     format='%(message)s', level=getattr(logging, loglevel)
   )
-  gp_sims(args.version)
+  #gp_sims(args.version)
+  gp_sims_panel(args.version)