def run(cosmo, data, command_line):
"""
First rudimentary implementation
The :mod:`mcmc` module is used as previously, except the call to
:func:`mcmc.chain`, or :func:`nested_sampling.run` is now within
this function, instead of from within :mod:`MontePython`.
In the long term, this function should contain any potential hybrid scheme,
and any chain communication (defining different roles, etc)
"""
if command_line.method == 'MH':
import mcmc
mcmc.chain(cosmo, data, command_line)
data.out.close()
elif command_line.method == 'NS':
import nested_sampling as ns
ns.run(cosmo, data, command_line)
elif command_line.method == 'CH':
import cosmo_hammer as hammer
hammer.run(cosmo, data, command_line)
else:
raise io_mp.ConfigurationError(
"Sampling method %s not understood" % command_line.method)
def run(cosmo, data, command_line):
"""
Depending on the choice of sampler, dispatch the appropriate information
The :mod:`mcmc` module is used as previously, except the call to
:func:`mcmc.chain`, or :func:`nested_sampling.run` is now within
this function, instead of from within :mod:`MontePython`.
In the long term, this function should contain any potential hybrid scheme.
"""
if command_line.method == 'MH':
import mcmc
mcmc.chain(cosmo, data, command_line)
data.out.close()
elif command_line.method == 'NS':
import nested_sampling as ns
ns.run(cosmo, data, command_line)
elif command_line.method == 'CH':
import cosmo_hammer as hammer
hammer.run(cosmo, data, command_line)
elif command_line.method == 'IS':
import importance_sampling as ims
ims.run(cosmo, data, command_line)
elif command_line.method == 'Der':
import add_derived as der
der.run(cosmo, data, command_line)
else:
raise io_mp.ConfigurationError(
"Sampling method %s not understood" % command_line.method)
def run(cosmo, data, command_line):
"""
Depending on the choice of sampler, dispatch the appropriate information
The :mod:`mcmc` module is used as previously, except the call to
:func:`mcmc.chain`, or :func:`nested_sampling.run` is now within
this function, instead of from within :mod:`MontePython`.
In the long term, this function should contain any potential hybrid scheme.
"""
if command_line.method == 'MH':
import mcmc
mcmc.chain(cosmo, data, command_line)
data.out.close()
elif command_line.method == 'NS':
import nested_sampling as ns
ns.run(cosmo, data, command_line)
elif command_line.method == 'CH':
import cosmo_hammer as hammer
hammer.run(cosmo, data, command_line)
elif command_line.method == 'IS':
import importance_sampling as ims
ims.run(cosmo, data, command_line)
elif command_line.method == 'Der':
import add_derived as der
der.run(cosmo, data, command_line)
else:
raise io_mp.ConfigurationError("Sampling method %s not understood" %
command_line.method)
def main():
"""
Main call of the function
This function recovers the input from the command line arguments, from
:mod:`parser_mp`, the parameter files.
It then extracts the path of the used Monte Python code, assuming a
standard setting (the data folder is in the same directory as the code
folder).
It finally proceeds to initialize a :class:`data` instance, a cosmological
code instance, and runs the Markov chain.
.. note::
A possible parallelization would take place here.
"""
# Parsing line argument
command_line = parser_mp.parse()
# Default configuration
path = {}
# On execution, sys.path contains all the standard locations for the
# libraries, plus, on the first position (index 0), the directory from
# where the code is executed. By default, then, the data folder is located
# in the same root directory. Any setting in the configuration file will
# overwrite this one.
path['MontePython'] = sys.path[0] + '/'
path['data'] = path['MontePython'][:-5] + 'data/'
# Configuration file, defaulting to default.conf in your root directory.
# This can be changed with the command line option -conf. All changes will
# be stored into the log.param of your folder, and hence will be reused for
# an ulterior run in the same directory
conf_file = path['MontePython'][:-5] + command_line.config_file
if os.path.isfile(conf_file):
for line in open(conf_file):
exec(line)
for key, value in path.iteritems():
if not value.endswith('/'):
path[key] = value + '/'
else:
io_mp.message(
"You must provide a .conf file (default.conf by default in your \
montepython directory that specifies the correct locations for your \
data folder, Class (, Clik), etc...",
"error")
sys.stdout.write('Running MontePython version 1.2\n')
# If the info flag was used, read a potential chain (or set of chains) to
# be analysed with default procedure. If the argument is a .info file, then
# it will extract information from it (plots to compute, chains to analyse,
# etc...)
if command_line.files is not None:
from analyze import analyze # analysis module, only invoked when analyzing
analyze(command_line)
exit()
# If the restart flag was used, load the cosmology directly from the
# log.param file, and append to the existing chain.
if command_line.restart is not None:
if command_line.restart[0] == '/':
folder = ''
else:
folder = './'
for elem in command_line.restart.split("/")[:-1]:
folder += ''.join(elem+'/')
command_line.param = folder+'log.param'
command_line.folder = folder
sys.stdout.write('Reading {0} file'.format(command_line.restart))
Data = data.data(command_line, path)
# Else, fill in data, starting from parameter file. If output folder
# already exists, the input parameter file was automatically replaced by
# the existing log.param. This prevents you to run different things in a
# same folder.
else:
Data = data.data(command_line, path)
# Overwrite arguments from parameter file with the command line
if command_line.N is None:
try:
command_line.N = Data.N
except AttributeError:
io_mp.message(
"You did not provide a number of steps, neither via \
command line, nor in %s" % command_line.param,
"error")
# Creating the file that will contain the chain
io_mp.create_output_files(command_line, Data)
# If there is a conflict between the log.param value and the .conf file,
# exiting.
if Data.path != path:
io_mp.message(
"Your log.param file is in contradiction with your .conf file, \
please check your path in these two places.",
"error")
# Loading up the cosmological backbone. For the moment, only Class has been
# wrapped.
# Importing the python-wrapped Class from the correct folder, defined in
# the .conf file, or overwritten at this point by the log.param.
# If the cosmological code is Class, do the following to import all
# relevant quantities
if Data.cosmological_module_name == 'Class':
try:
for elem in os.listdir(Data.path['cosmo']+"python/build"):
if elem.find("lib.") != -1:
classy_path = path['cosmo']+"python/build/"+elem
except OSError:
io_mp.message(
"You probably did not compile the python wrapper of Class. \
Please go to /path/to/class/python/ and do\n\
..]$ python setup.py build",
"error")
# Inserting the previously found path into the list of folders to
# search for python modules.
sys.path.insert(1, classy_path)
try:
from classy import Class
except ImportError:
io_mp.message(
"You must have compiled the classy.pyx file. Please go to \
/path/to/class/python and run the command\n\
python setup.py build",
"error")
cosmo = Class()
else:
io_mp.message(
"Unrecognised cosmological module. \
Be sure to define the correct behaviour in MontePython.py \
and data.py, to support a new one",
"error")
# MCMC chain
mcmc.chain(cosmo, Data, command_line)
# Closing up the file
Data.out.close()
