tardis.io.util module

class tardis.io.util.HDFWriterMixin(*args, **kwargs)

Bases: object

static convert_to_snake_case(s)

property full_hdf_properties

get_properties()

to_hdf(file_path_or_buf, path='', name=None, overwrite=False)

Parameters

file_path_or_bufstr or pandas.io.pytables.HDFStore

Path or buffer to the HDF file

pathstr

Path inside the HDF file to store the elements

namestr

Group inside the HDF file to which the elements need to be saved

overwritebool

If the HDF file path already exists, whether to overwrite it or not

static to_hdf_util(path_or_buf, path, elements, overwrite, complevel=9, complib='blosc')

A function to uniformly store TARDIS data to an HDF file.

Scalars will be stored in a Series under path/scalars 1D arrays will be stored under path/property_name as distinct Series 2D arrays will be stored under path/property_name as distinct DataFrames

Units will be stored as their CGS value

Parameters

path_or_bufstr or pandas.io.pytables.HDFStore

Path or buffer to the HDF file

pathstr

Path inside the HDF file to store the elements

elementsdict

A dict of property names and their values to be stored.

overwritebool

If the HDF file path already exists, whether to overwrite it or not

Notes

overwrite option doesn’t have any effect when path_or_buf is an HDFStore because the user decides on the mode in which they have opened the HDFStore (‘r’, ‘w’ or ‘a’).

class tardis.io.util.MockRegexPattern(target_type)

Bases: object

A mock class to be used in place of a compiled regular expression when a type check is needed instead of a regex match.

Notes

This is usually a lot slower than regex matching.

match(text)

Parameters

text :

A string to be passed to target_type for conversion.

Returns

bool

Returns True if text can be converted to target_type, otherwise returns False

class tardis.io.util.PlasmaWriterMixin(*args, **kwargs)

Bases: tardis.io.util.HDFWriterMixin

get_properties()

to_hdf(file_path_or_buf, path='', name=None, collection=None, overwrite=False)

Parameters

file_path_or_bufstr or pandas.io.pytables.HDFStore

Path or buffer to the HDF file

pathstr

Path inside the HDF file to store the elements

namestr

Group inside the HDF file to which the elements need to be saved

collection :

None or a PlasmaPropertyCollection of which members are the property types which will be stored. If None then all types of properties will be stored. This acts like a filter, for example if a value of property_collections.basic_inputs is given, only those input parameters will be stored to the HDF file.

overwritebool

If the HDF file path already exists, whether to overwrite it or not

class tardis.io.util.YAMLLoader(stream)

Bases: yaml.loader.Loader

A custom YAML loader containing all the constructors required to properly parse the tardis configuration.

construct_quantity(node)

A constructor for converting quantity-like YAML nodes to astropy.units.Quantity objects.

Parameters

node :

The YAML node to be constructed

Returns

astropy.units.Quantity

mapping_constructor(node)

tardis.io.util.assert_equality(item1, item2)

tardis.io.util.check_equality(item1, item2)

tardis.io.util.download_from_url(url, dst)

kindly used from https://gist.github.com/wy193777/0e2a4932e81afc6aa4c8f7a2984f34e2 @param: url to download file @param: dst place to put the file

tardis.io.util.get_internal_data_path(fname)

Get internal data path of TARDIS

Returns

data_pathstr

internal data path of TARDIS

tardis.io.util.quantity_from_str(text)

Convert a string to astropy.units.Quantity

Parameters

text :

The string to convert to astropy.units.Quantity

Returns

astropy.units.Quantity

tardis.io.util.traverse_configs(base, other, func, *args)

Recursively traverse a base dict or list along with another one calling func for leafs of both objects.

Parameters

base :

The object on which the traversing is done

other :

The object which is traversed along with base

func :

A function called for each leaf of base and the correspnding leaf of other Signature: func(item1, item2, *args)

args :

Arguments passed into func

tardis.io.util.yaml_load_config_file(filename)

tardis.io.util.yaml_load_file(filename, loader=<class 'yaml.loader.Loader'>)