Source code for getdist.parampriors

import os
import numpy as np

[docs]class ParamBounds: """ Class for holding list of parameter bounds (e.g. for plotting, or hard priors). A limit is None if not specified, denoted by 'N' if read from a string or file :ivar names: list of parameter names :ivar lower: dict of lower limits, indexed by parameter name :ivar upper: dict of upper limits, indexed by parameter name """ def __init__(self, fileName=None): """ :param fileName: optional file name to read from """ self.names = [] self.lower = {} self.upper = {} if fileName is not None: self.loadFromFile(fileName) def loadFromFile(self, fileName): self.filenameLoadedFrom = os.path.split(fileName)[1] extension = os.path.splitext(fileName)[-1] if extension in ('.ranges', '.bounds'): with open(fileName, encoding='utf-8-sig') as f: for line in f: strings = [text.strip() for text in line.split()] if len(strings) == 3: self.setRange(strings[0], strings[1:]) elif extension in ('.yaml', '.yml'): from getdist.cobaya_interface import get_range, get_info_params, is_parameter_with_range info_params = get_info_params(fileName) for p, info in info_params.items(): if is_parameter_with_range(info): self.setRange(p, get_range(info)) else: raise ValueError('ParamBounds must be loaded from .bounds, .ranges or .yaml/.yml file, ' 'not %s' % fileName) def __str__(self): s = '' for name in self.names: valMin = self.getLower(name) if valMin is not None: lim1 = "%15.7E" % valMin else: lim1 = " N" valMax = self.getUpper(name) if valMax is not None: lim2 = "%15.7E" % valMax else: lim2 = " N" s += "%22s%17s%17s\n" % (name, lim1, lim2) return s
[docs] def saveToFile(self, fileName): """ Save to a plain text file :param fileName: file name to save to """ with open(fileName, 'w', encoding='utf-8') as f: f.write(str(self))
def _check_name(self, name): if not isinstance(name, str): raise ValueError('"name" must be a parameter name string not %s: %s' % (type(name), name)) def setFixed(self, name, value): self.setRange(name, (value, value)) def setRange(self, name, strings): self._check_name(name) if strings[0] != 'N' and strings[0] is not None and strings[0] != -np.inf: self.lower[name] = float(strings[0]) if strings[1] != 'N' and strings[1] is not None and strings[1] != np.inf: self.upper[name] = float(strings[1]) if name not in self.names: self.names.append(name)
[docs] def getUpper(self, name): """ :param name: parameter name :return: upper limit, or None if not specified """ self._check_name(name) return self.upper.get(name, None)
[docs] def getLower(self, name): """ :param name: parameter name :return: lower limit, or None if not specified """ self._check_name(name) return self.lower.get(name, None)
[docs] def fixedValue(self, name): """ :param name: parameter name :return: if range has zero width return fixed value else return None """ lower = self.lower.get(name, None) if lower is not None: higher = self.upper.get(name, None) if higher is not None: if higher == lower: return lower return None
[docs] def fixedValueDict(self): """ :return: dictionary of fixed parameter values """ res = {} for name in self.names: value = self.fixedValue(name) if value is not None: res[name] = value return res