Пример #1
0
def run_tardis(configuration_dict, atom_data=None):
    """
    This function is one of the core functions to run TARDIS from a given
    config object.

    It will return a model object containing

    Parameters
    ----------

    configuration_dict: ~dict

    atom_data: ~tardis.atomic.AtomData
        Atomic data to use for this TARDIS simulation. If set to None, the
        atomic data will be loaded according to keywords set in the configuration
        [default=None]
    """

    tardis_config = config_reader.Configuration.from_config_dict(
        configuration_dict, atom_data=atom_data)
    radial1d_mdl = model.Radial1DModel(tardis_config)

    simulation.run_radial1d(radial1d_mdl)

    return radial1d_mdl
Пример #2
0
    def setup(self):
        self.atom_data_filename = pytest.config.getvalue("atomic-dataset")
        assert os.path.exists(self.atom_data_filename)
        self.config_yaml = yaml.load(open("tardis/io/tests/data/tardis_configv1_verysimple.yml"))
        self.config_yaml["atom_data"] = self.atom_data_filename

        self.config = TARDISConfiguration.from_config_dict(self.config_yaml)
        self.model = model.Radial1DModel(self.config)
        simulation.run_radial1d(self.model)
Пример #3
0
    def setup(self):
        self.atom_data_filename = os.path.expanduser(os.path.expandvars(
            pytest.config.getvalue('atomic-dataset')))
        assert os.path.exists(self.atom_data_filename), ("{0} atomic datafiles "
                                                         "does not seem to "
                                                         "exist".format(
            self.atom_data_filename))
        self.config_yaml = yaml.load(open('tardis/io/tests/data/tardis_configv1_verysimple.yml'))
        self.config_yaml['atom_data'] = self.atom_data_filename

        self.config = Configuration.from_config_dict(self.config_yaml)
        self.model = model.Radial1DModel(self.config)
        simulation.run_radial1d(self.model)
Пример #4
0
    def setup(self):
        self.atom_data_filename = os.path.expanduser(os.path.expandvars(
            pytest.config.getvalue('atomic-dataset')))
        assert os.path.exists(self.atom_data_filename), ("{0} atomic datafiles "
                                                         "does not seem to "
                                                         "exist".format(
            self.atom_data_filename))
        self.config_yaml = yaml.load(open('tardis/io/tests/data/tardis_configv1_verysimple.yml'))
        self.config_yaml['atom_data'] = self.atom_data_filename

        self.config = Configuration.from_config_dict(self.config_yaml)
        self.model = model.Radial1DModel(self.config)
        simulation.run_radial1d(self.model)
Пример #5
0
    def evaluate(self, *args, **kwargs):
        config = self._get_config_from_args(args)
        radial1d_mdl = model.Radial1DModel(config)

        simulation.run_radial1d(radial1d_mdl)
        runner = radial1d_mdl.runner

        param_names = ['t_inner']
        param_values = [radial1d_mdl.t_inner]
        self.mdl = radial1d_mdl
        return (runner.emitted_packet_nu,
                runner.emitted_packet_luminosity,
                runner.virt_packet_nus * u.Hz,
                (runner.virt_packet_energies /
                 runner.time_of_simulation),
                param_names, param_values)
Пример #6
0
def run_tardis(config, atom_data=None):
    """
    This function is one of the core functions to run TARDIS from a given
    config object.

    It will return a model object containing

    Parameters
    ----------

    config: ~str or ~dict
        filename of configuration yaml file or dictionary

    atom_data: ~str or ~tardis.atomic.AtomData
        if atom_data is a string it is interpreted as a path to a file storing
        the atomic data. Atomic data to use for this TARDIS simulation. If set to None, the
        atomic data will be loaded according to keywords set in the configuration
        [default=None]
    """
    import yaml

    from tardis.io import config_reader
    from tardis import model, simulation, atomic
    import os

    try:
        config_dict = yaml.load(open(config))
        config_dirname = os.path.dirname(config)
    except TypeError:
        config_dict = config
        config_dirname = ''

    if atom_data is not None:
        try:
            atom_data = atomic.AtomData.from_hdf5(atom_data)
        except TypeError:
            atom_data = atom_data

    tardis_config = config_reader.Configuration.from_config_dict(
        config_dict, atom_data=atom_data, config_dirname=config_dirname)
    radial1d_mdl = model.Radial1DModel(tardis_config)

    simulation.run_radial1d(radial1d_mdl)

    return radial1d_mdl
Пример #7
0
def run_tardis(config, atom_data=None):
    """
    This function is one of the core functions to run TARDIS from a given
    config object.

    It will return a model object containing

    Parameters
    ----------

    config: ~str or ~dict
        filename of configuration yaml file or dictionary

    atom_data: ~str or ~tardis.atomic.AtomData
        if atom_data is a string it is interpreted as a path to a file storing
        the atomic data. Atomic data to use for this TARDIS simulation. If set to None, the
        atomic data will be loaded according to keywords set in the configuration
        [default=None]
    """
    import yaml

    from tardis.io import config_reader
    from tardis import model, simulation, atomic

    try:
        config_dict = yaml.load(open(config))
    except TypeError:
        config_dict = config

    if atom_data is not None:
        try:
            atom_data = atomic.AtomData.from_hdf5(atom_data)
        except TypeError:
            atom_data = atom_data

    tardis_config = config_reader.Configuration.from_config_dict(
        config_dict, atom_data=atom_data)
    radial1d_mdl = model.Radial1DModel(tardis_config)

    simulation.run_radial1d(radial1d_mdl)

    return radial1d_mdl
Пример #8
0
 def run_tardis(self, config):
     mdl = Radial1DModel(config)
     run_radial1d(mdl)
     return mdl