Homec4science
Diffusion pNbody (master) Browse

pNbody/scripts/gmkgmov
8e5aa76087famaster

Last Change
Raw File

gmkgmov
View Options

#!/usr/bin/env python
# -*- coding: iso-8859-1 -*-
import matplotlib
matplotlib.use('Agg')
from optparse import OptionParser
import os,sys
from pNbody import Mkgmov
from pNbody import mpi
def parse_options():
usage = "usage: %prog [options] file"
parser = OptionParser(usage=usage)
parser.add_option("-p",
action="store",
dest="parameterfile",
type="string",
default = None,
help="parameterfile file",
metavar=" FILE")
parser.add_option("--format",
action="store",
dest="format",
type="string",
default = None,
help="output file format",
metavar=" FILE")
parser.add_option("--imdir",
action="store",
dest="imdir",
type="string",
default = None,
help="outputdirectory for fits files",
metavar=" DIRECTORY")
parser.add_option("--pio",
action="store_true",
dest="pio",
default = False,
help="parallele io",
metavar=" BOOL")
parser.add_option("--compress",
action="store_true",
dest="compress",
default = True,
help="compress output files",
metavar=" BOOL")
(options, args) = parser.parse_args()
files = args
return files,options
#######################################################################
#
# M A I N
#
#######################################################################
files = None
movie = None
opt = None
if mpi.mpi_IsMaster():
##############################
# parse options and check dirs
##############################
files,opt = parse_options()
movie = Mkgmov.Movie(parameterfile=opt.parameterfile,format=opt.format,imdir=opt.imdir,timesteps=None,pio=opt.pio,compress=opt.compress)
# broadcast files and parameters
files = mpi.mpi_bcast(files,root=0)
opt = mpi.mpi_bcast(opt,root=0)
movie = mpi.mpi_bcast(movie,root=0)
###################################################################################
###################################################################################
##
## main loop over all files
##
###################################################################################
###################################################################################
for ifile,file in enumerate(files):
movie.dumpimage(file=file)
c4science · Help