Homec4science
Diffusion blackdynamite (features/zeo) Browse

R3127/python/BlackDynamite/graphhelper.py
c29dd8695b6bfeatures/zeo

Last Change
Raw File

graphhelper.py
View Options

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -*- py-which-shell: "python"; -*-
import base
import run
import job
import runselector
import jobselector
import matplotlib.pyplot as plt
import re
import bdparser
import sys
import run
import numpy as np
class GraphHelper(object):
"""
"""
def getMeasures(self,run_list):
myresults = []
add_req = []
if (self.frequency):
add_req += ["step %% {0} = 0".format(self.frequency)]
if (self.start):
add_req += ["step > {0}".format(self.start) ]
if (self.end):
add_req += ["step < {0}".format(self.end) ]
for r,j in run_list:
print ("retreive data from run " + r["run_name"])
res = r.getScalarQuantities(self.quantities,add_req)
for key , value in res:
if (value == None):
del res[key]
myresults.append([r,j,res])
return myresults
def selectGraphs(self):
run_list = self.runSelector.selectRuns(self.run_constraints,self.job_constraints,
self.sort_by)
results = self.getMeasures(run_list)
return results
def makeGraphs(self,fig=None,**kwargs):
results = self.selectGraphs()
if fig is None:
fig = plt.figure(figsize=self.figsize)
for r,j,data in results:
self.makeCurve(data,fig=fig,myrun=r,myjob=j,**kwargs)
return fig
def replaceRunAndJobsParameters(self,name,myrun,myjob):
res = name
# print (res)
codes = [["%r."+key,myrun[key]] for key in myrun.entries.keys()]
codes += [["%j."+key,myjob[key]] for key in myjob.entries.keys()]
for code,val in codes:
res = res.replace(code,str(val))
return res
def generateLabels(self,results,myrun,myjob):
labels = []
names = [r[0] for r in results]
for i in range(0,len(results)):
name = results[i][0]
if (not self.legend or i >= len(self.legend) or not self.legend[i]):
labels.append(self.replaceRunAndJobsParameters(name,myrun,myjob))
continue
# print (self.legend[i])
head_legend = self.legend[i].replace("{","{{")
head_legend = head_legend.replace("}","}}")
head_legend = re.sub(r"(%)([0-9]+)",r'{\2}',head_legend).format(*names)
# print (head_legend)
head_legend = self.replaceRunAndJobsParameters(head_legend,myrun,myjob)
# print (head_legend)
# if (not head_legend.find("%") == -1):
# print("unknown variable name. Possible variables are:")
# print "\n".join([c[0] for c in codes])
# sys.exit(-1)
# print (head_legend)
labels.append(head_legend)
return labels
def makeComposedQuantity(self,results,myrun,myjob):
vecs = [r[1] for r in results]
names = [r[0] for r in results]
# print (vecs[0].shape)
new_results = []
for comp in self.using:
exprs = comp.split(":")
tmp_res = []
for i in [0,1]:
e = re.sub(r"(%)([0-9]+)\.(x)",r"vecs[\2][:,0]",exprs[i])
e = re.sub(r"(%)([0-9]+)\.(y)",r"vecs[\2][:,1]",e)
e = self.replaceRunAndJobsParameters(e,myrun,myjob)
try:
tmp_res.append(eval(e))
except:
print (names)
print ("invalid expression: '" + exprs[i] + "'")
print ("invalid expression: '" + e + "'")
sys.exit(-1)
name = re.sub(r"(%)([0-9]+)\.([x|y])",r'(" + str(names[\2]) + ")',exprs[1])
res = np.zeros((tmp_res[0].shape[0],2))
res[:,0] = tmp_res[0]
res[:,1] = tmp_res[1]
# print (res.shape)
expr = re.sub(r"(%)([0-9]+)",r"vecs[\2]",comp)
# res[0] = eval(expr)
# print (name)
name = "\"" + name + "\""
# print (name)
name = eval(name)
# print (name)
new_results.append([name,res])
return new_results
def makeCurve(self,results,myrun=None,myjob=None,fig=None,**kwargs):
if fig is None:
fig = plt.figure()
axe=fig.add_subplot(1,1,1)
if self.xrange:
axe.set_xlim(self.xrange)
if self.yrange:
axe.set_ylim(self.yrange)
if (self.xlabel):
axe.set_xlabel(self.xlabel)
if (self.ylabel):
axe.set_ylabel(self.ylabel)
if (self.title):
t = self.replaceRunAndJobsParameters(self.title,myrun,myjob)
axe.set_title(t)
axe.grid(True,linewidth=0.1)
if (self.using):
results = self.makeComposedQuantity(results,myrun,myjob)
labels = self.generateLabels(results,myrun,myjob)
# print (labels)
for i in range(0,len(results)):
name = results[i][0]
vec = results[i][1]
label = labels[i]
# print (self.quantities)
# print (name)
style=dict()
if (self.marker is not None):
style["marker"] = self.marker
if self.blackwhite:
width_index = self.cycle_index/len(self.linestyle_cycle)
style_index = self.cycle_index%len(self.linestyle_cycle)
self.cycle_index += 1
style["linewidth"] = self.linewidth_cycle[width_index]
style["linestyle"] = self.linestyle_cycle[style_index]
style["color"] = 'k'
axe.plot(vec[:,0]/self.xscale, vec[:,1]/self.yscale,
label=label, **style)
axe.legend(loc='best')
if (self.fileout):
fig.savefig(self.fileout)
return fig
def setJobConstraint(self,**params):
self.job_constraints = []
if ("job_constraints" in params):
self.job_constraints = params["job_constraints"]
def setRunConstraint(self,**params):
self.run_constraints = []
if ("run_constraints" in params):
self.run_constraints = params["run_constraints"]
def setQuantity(self,**params):
if ("quantity" in params):
self.quantities = params["quantity"]
else:
print ("quantity should be provided using option --quantity")
self.quantities = "__BLACKDYNAMITE_ERROR__"
def __init__ (self,base,**params):
self.setJobConstraint(**params)
self.setRunConstraint(**params)
self.setQuantity(**params)
self.base = base
self.runSelector = runselector.RunSelector(self.base)
self.fig = None
self.xrange = None
self.yrange = None
self.xlabel = None
self.ylabel = None
self.xscale = None
self.yscale = None
self.fileout = None
self.title = None
self.using = None
self.frequency = None
self.start = None
self.end = None
self.figsize = None
self.blackwhite = None
self.legend = None
self.sort_by = None
self.marker = None
### set the members if keys are present in params
members = set(self.__dict__.keys())
p = set(params.keys())
for key in members & p:
setattr(self,key,params[key])
if("list_quantities" in params):
myrun = run.Run(base)
print ("list of possible quantities:\n")
print("\n".join(myrun.listQuantities()))
sys.exit(0)
if("list_parameters" in params):
myjob = job.Job(base)
myjob.prepare()
print ("****************************************************************")
print ("Job parameters:")
print ("****************************************************************")
params = [str(j[0]) + ": " + str(j[1]) for j in myjob.types.iteritems() ]
print("\n".join(params))
myrun = run.Run(base)
myrun.prepare()
print ("****************************************************************")
print ("Run parameters:")
print ("****************************************************************")
params = [str(j[0]) + ": " + str(j[1]) for j in myrun.types.iteritems() ]
print("\n".join(params))
sys.exit(0)
self.linewidth_cycle = [1, 2, 4]
self.linestyle_cycle = ['-', '--', '-.']
self.cycle_index = 0
################################################################
class GraphParser(bdparser.BDParser):
"""
"""
def declareBDParameters(self,additional_params=None,default_additional_params=None):
tmp_params = self.graph_params
if (additional_params):
tmp_params.update(additional_params)
tmp_default_params = self.graph_default_params
if (default_additional_params):
tmp_default_params.update(default_additional_params)
bdparser.BDParser.declareBDParameters(self,additional_params=tmp_params,default_additional_params=tmp_default_params)
def __init__ (self):
bdparser.BDParser.__init__(self)
self.graph_params = {}
self.graph_default_params = {}
self.graph_params["quantity"] = [str]
self.help["quantity"] = "Specify the quantity to be outputed"
self.graph_params["xrange"] = [float]
self.help["xrange"] = "Specify range of values in the X direction"
self.graph_params["yrange"] = [float]
self.help["yrange"] = "Specify range of values in the Y direction"
self.graph_params["sort_by"] = [str]
self.help["sort_by"] = "Specify a study parameter to be used in sorting the curves"
self.graph_params["xlabel"] = str
self.help["xlabel"] = "Specify the label for the X axis"
self.graph_params["ylabel"] = str
self.help["ylabel"] = "Specify the label for the Y axis"
self.graph_params["xscale"] = float
self.graph_default_params["xscale"] = 1.
self.help["xscale"] = "Specify a scale factor for the X axis"
self.graph_params["yscale"] = float
self.graph_default_params["yscale"] = 1.
self.help["yscale"] = "Specify a scale factor for the Y axis"
self.graph_params["title"] = str
self.help["title"] = "Specify title for the graph"
self.graph_params["legend"] = [str]
self.help["legend"] = "Specify a legend for the curves. The syntax can use %%j.param or %%r.param to use get job and run values"
self.graph_default_params["legend"] = None
self.graph_params["using"] = [str]
self.help["using"] = "Allow to combine several quantities. The syntax uses python syntax where %%quantity1.column1:%%quantity2.column2 is the python numpy vector provided by quantity number (provided using the --quantities option) and column number (x or y). The sytax is comparable to the GNUPlot one in using the ':' to separate X from Y axis"
self.graph_params["list_quantities"] = None
self.help["list_quantities"] = "Request to list the possible quantities to be plotted"
self.graph_params["list_parameters"] = None
self.help["list_parameters"] = "Request to list the possible job/run parameters"
self.graph_params["frequency"] = None
self.help["frequency"] = "Set a frequency at which the quantity values should be retreived (helpful when the amount of data is very large)"
self.graph_params["start"] = None
self.help["start"] = "Set the start X value for the graph"
self.graph_params["end"] = None
self.help["end"] = "Set the end X value for the graph"
self.graph_params["figsize"] = None
self.graph_default_params["figsize"] = None
self.graph_params["blackwhite"] = None
self.help["blackwhite"] = "Request a black and white graph generation"
self.graph_default_params["blackwhite"] = False
self.help["blackwhite"] = "Request to plot a black and white graph"
self.graph_params["marker"] = str
self.help["marker"] = "Request a specific marker (matplotlib option)"
################################################################
c4science · Help