Homec4science
Diffusion Gear (master) Browse

Gear/PyChem/shared/debug_YieldsFinal.py
f7e36798d4e7master

Last Change
Raw File

debug_YieldsFinal.py
View Options

#!/usr/bin/env python
import pickle
import sys
from optparse import OptionParser
import Ptools as pt
import string
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser = pt.add_postscript_options(parser)
parser = pt.add_ftype_options(parser)
parser = pt.add_reduc_options(parser)
parser = pt.add_center_options(parser)
parser = pt.add_select_options(parser)
parser = pt.add_cmd_options(parser)
parser = pt.add_display_options(parser)
parser = pt.add_info_options(parser)
parser = pt.add_limits_options(parser)
parser = pt.add_log_options(parser)
parser.add_option("--x",
action="store",
dest="x",
type="string",
default = 'r',
help="x value to plot",
metavar=" STRING")
parser.add_option("--y",
action="store",
dest="y",
type="string",
default = 'T',
help="y value to plot",
metavar=" STRING")
parser.add_option("--z",
action="store",
dest="z",
type="string",
default = None,
help="z value to plot",
metavar=" STRING")
parser.add_option("--legend",
action="store_true",
dest="legend",
default = False,
help="add a legend")
parser.add_option("--elts",
action="store",
type="string",
dest="elts",
default = None,
help="element name")
(options, args) = parser.parse_args()
pt.check_files_number(args)
files = args
return files,options
#######################################
# MakePlot
#######################################
def MakePlot(files,opt):
datas = []
if len(files)>1:
colors = pt.Colors(n=len(files))
for file in files:
f = open(file,'r')
elts = pickle.load(f)
Ms1 = pickle.load(f)
Ms2 = pickle.load(f)
Ress = pickle.load(f)
Hcos = pickle.load(f)
datai = pickle.load(f)
Mi1 = Ms1
f.close()
###############################################
# info
###############################################
#print MetalFile
print elts
###############################################
# plot
###############################################
if opt.elts !=None:
elts = string.split(opt.elts,',')
if len(files)==1:
colors = pt.Colors(n=len(elts))
for elt in elts:
d = pt.DataPoints(Mi1,datai[elt],label="%s"%(elt),color=colors.get(),tpe='line')
#pt.plot(Mi1,datai[elt])
#pt.scatter(Mi1,datai[elt])
pt.plot(d.x,d.y,color=d.color,ls=d.linestyle)
datas.append(d)
# set limits and draw axis
xmin,xmax,ymin,ymax = pt.SetLimitsFromDataPoints(opt.xmin,opt.xmax,opt.ymin,opt.ymax,datas,opt.log)
# plot axis
pt.SetAxis(xmin,xmax,ymin,ymax,log=opt.log)
pt.xlabel('Mass',fontsize=pt.labelfont)
pt.ylabel("Yield",fontsize=pt.labelfont)
if opt.legend:
pt.LegendFromDataPoints(datas,loc='lower right')
########################################################################
# MAIN
########################################################################
if __name__ == '__main__':
files,opt = parse_options()
pt.InitPlot(files,opt)
pt.pcolors
MakePlot(files,opt)
pt.EndPlot(files,opt)
c4science · Help