def simulation(parameter_filename, simulation_type, find_outputs=False):
"""
Loads a simulation time series object of the specified
simulation type.
"""
if simulation_type not in simulation_time_series_registry:
raise YTSimulationNotIdentified(simulation_type)
if os.path.exists(parameter_filename):
valid_file = True
elif os.path.exists(os.path.join(ytcfg.get("yt", "test_data_dir"),
parameter_filename)):
parameter_filename = os.path.join(ytcfg.get("yt", "test_data_dir"),
parameter_filename)
valid_file = True
else:
valid_file = False
if not valid_file:
raise YTOutputNotIdentified((parameter_filename, simulation_type),
dict(find_outputs=find_outputs))
return simulation_time_series_registry[simulation_type](parameter_filename,
find_outputs=find_outputs)
def load_simulation(fn, simulation_type, find_outputs=False):
"""
Load a simulation time series object of the specified simulation type.
Parameters
----------
fn : str, os.Pathlike, or byte (types supported by os.path.expandusers)
Name of the data file or directory.
simulation_type : str
E.g. 'Enzo'
find_outputs : bool
Defaults to False
Raises
------
FileNotFoundError
If fn is not found.
yt.utilities.exceptions.YTSimulationNotIdentified
If simulation_type is unknown.
"""
fn = str(lookup_on_disk_data(fn))
try:
cls = simulation_time_series_registry[simulation_type]
except KeyError as e:
raise YTSimulationNotIdentified(simulation_type) from e
return cls(fn, find_outputs=find_outputs)
def load_simulation(fn, simulation_type, find_outputs=False):
"""
Load a simulation time series object of the specified simulation type.
Parameters
----------
fn : str, os.Pathlike, or byte (types supported by os.path.expandusers)
Name of the data file or directory.
simulation_type : str
E.g. 'Enzo'
find_outputs : bool
Defaults to False
Raises
------
FileNotFoundError
If fn is not found.
yt.utilities.exceptions.YTSimulationNotIdentified
If simulation_type is unknown.
"""
if not os.path.exists(fn):
alt_fn = os.path.join(ytcfg.get("yt", "test_data_dir"), fn)
if os.path.exists(alt_fn):
fn = alt_fn
else:
raise FileNotFoundError(f"No such file or directory: '{fn}'")
try:
cls = simulation_time_series_registry[simulation_type]
except KeyError as e:
raise YTSimulationNotIdentified(simulation_type) from e
return cls(fn, find_outputs=find_outputs)