Homec4science
Diffusion Gtools (master) Browse

gtools/dev/gplotall.py
8cfdf441e0b6master

Last Change
Raw File

gplotall.py
View Options

#!/usr/bin/env python
'''
Extract and plot energy and mass values contained in the
output Gadget file called by default "energy.txt".
Yves Revaz
ven jun 9 10:43:59 CEST 2006
'''
from numpy import *
from pNbody import *
import string
import sys
import os
import copy as docopy
from pNbody.libutil import histogram
from pNbody import libgrid
from optparse import OptionParser
from Gtools import *
from Gtools import io
import Ptools as pt
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser = pt.add_postscript_options(parser)
parser = pt.add_limits_options(parser)
parser = pt.add_log_options(parser)
parser.add_option("--x",
action="store",
type="string",
dest="x",
default = 'R',
help="x")
parser.add_option("--y",
action="store",
type="string",
dest="y",
default = 'vct',
help="y")
parser.add_option("--mode",
action="store",
dest="mode",
type="string",
default = 'vc',
help="mode",
metavar=" NAME")
(options, args) = parser.parse_args()
if len(args) == 0:
print "you must specify a filename"
sys.exit(0)
files = args
return files,options
#######################################
# MakePlot
#######################################
def MakePlot(dirs,opt):
# some inits
palette = pt.GetPalette()
colors = pt.SetColorsForFiles(files,palette)
labels = []
#######################################
# LOOP
#######################################
yt = None
# read files
for file in files:
######################################
# open file
######################################
stats = pt.io.read_hdf5(file)
###################################
# now, plot
###################################
x = stats[opt.x]
y = stats[opt.y]
xmin,xmax,ymin,ymax = pt.SetLimits(opt.xmin,opt.xmax,opt.ymin,opt.ymax,x,y,opt.log)
pt.plot(x,y)
if opt.mode=='sum':
if yt ==None:
yt = y
else:
yt = yt + y
elif opt.mode=='sum2':
if yt ==None:
yt = y**2
else:
yt = yt + y**2
if opt.mode=='sum':
pt.plot(x,yt,'k')
xmin,xmax,ymin,ymax = pt.SetLimits(opt.xmin,opt.xmax,opt.ymin,opt.ymax,x,yt,opt.log)
elif opt.mode=='sum2':
yt = where(yt>=0,sqrt(yt),0)
pt.plot(x,yt,'k')
xmin,xmax,ymin,ymax = pt.SetLimits(opt.xmin,opt.xmax,opt.ymin,opt.ymax,x,yt,opt.log)
pt.SetAxis(xmin,xmax,ymin,ymax,log=opt.log)
pt.show()
if __name__ == '__main__':
files,opt = parse_options()
pt.InitPlot(files,opt)
#pt.figure(figsize=(8*2,6*2))
#pt.figure(dpi=10)
pt.pcolors
#fig = pt.gcf()
#fig.subplots_adjust(left=0.1)
#fig.subplots_adjust(right=1)
#fig.subplots_adjust(bottom=0.12)
#fig.subplots_adjust(top=0.95)
#fig.subplots_adjust(wspace=0.25)
#fig.subplots_adjust(hspace=0.02)
MakePlot(files,opt)
pt.EndPlot(files,opt)
c4science · Help