B d @sHdZddlZddlmZddlmZddlmZd ddZdd d Z dS)z This module provides functions for drawing samples from a standalone .ini file in a Python script, rather than in the command line. N)InterpolatingConfigParser) transforms) distributionspriorcCshtj||ddd\}}t|dr2t|d}nd}tj|||d}t||}tj|f|d|iS)aLoads a prior distribution from the given config file. Parameters ---------- cp : pycbc.workflow.WorkflowConfigParser The config file to read. sections : list of str, optional The sections to retrieve the prior from. If ``None`` (the default), will look in sections starting with 'prior'. Returns ------- distributions.JointDistribution The prior distribution. variable_params static_params) prior_sectionZ vargs_sectionZ sargs_sectionwaveform_transformsN)rZ static_args constraints) rZread_params_from_configanyget_subsectionsrread_transforms_from_configZread_constraints_from_configZread_distributions_from_configZJointDistribution)cprrrr r distsrf/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/distributions/utils.pyprior_from_config#s     rMcCsrtj|t}t|d}|||t|d}|jt |d}t | drnt |d}t ||}|S)a Generate sampling points from a standalone .ini file. Parameters ---------- path : str The path to the .ini file. num : int The number of samples. seed: int The random seed for sampling. Returns -------- samples : pycbc.io.record.FieldArray The parameter values and names of sample(s). Examples -------- Draw a sample from the distribution defined in the .ini file: >>> import numpy as np >>> from pycbc.distributions.utils import draw_samples_from_config >>> # A path to the .ini file. >>> CONFIG_PATH = "./pycbc_bbh_prior.ini" >>> random_seed = np.random.randint(low=0, high=2**32-1) >>> sample = draw_samples_from_config( >>> path=CONFIG_PATH, num=1, seed=random_seed) >>> # Print all parameters. >>> print(sample.fieldnames) >>> print(sample) >>> # Print a certain parameter, for example 'mass1'. >>> print(sample[0]['mass1']) r)r)sizer )nprandomseedropen read_filecloserZrvsintr r rr Zapply_transforms)pathnumrZ config_parserfileZ prior_distsZsamplesr rrrdraw_samples_from_configMs%     r!)r)rr) __doc__numpyrZpycbc.types.configrZpycbcrrrr!rrrrs     *