Note: The default ITS GitLab runner is a shared resource and is subject to slowdowns during heavy usage.
You can run your own GitLab runner that is dedicated just to your group if you need to avoid processing delays.

Commit 5644e875 authored by Camilla Dodge Koslowsky Harris's avatar Camilla Dodge Koslowsky Harris
Browse files

inital commit of the prototype- need to remove pandas things

parent 028d8ce3
......@@ -195,3 +195,163 @@ def tecplot_interpolate(
if verbose:
print('Collecting parameters')
## collect the geometry parameters
geometry_params = {
## assign defaults for shell
if verbose:
print('Adding defaults')
if 'shell' in geometry_params['kind']:
geometry_params['center'] = geometry_params.get('center',(0.0,0.0,0.0))
geometry_params['npolar'] = geometry_params.get('npolar',181)
geometry_params['nazimuth'] = geometry_params.get('nazimuth',359)
## check that we support the geometry
geometry_param_names = {
if geometry_params['kind'] not in geometry_param_names:
raise ValueError(f'Geometry {geometry_params["kind"]} not supported!')
## check that we've gotten the right /required/ geometry arguments
for param in geometry_param_names[geometry_params['kind']]:
if param not in geometry_params:
raise TypeError(
f'Geometry {geometry_params["kind"]} '
f'requires argument {param}!')
## check that we support the file type to save as
file_types = (
if write_as not in file_types:
raise ValueError(f'File type {write_as} not supported!')
## describe the interpolation we're about to do on the data
if verbose:
print('Geometry to be interpolated:')
for key,value in geometry_params.items():
print(f'\t{key}: {value.__repr__()}')
## check whether we are using equations
## check that the equations file is there
use_equations = not (tecplot_equation_file_path is None)
if use_equations:
f = open(tecplot_equation_file_path,'r')
if verbose:
print('Applying equations from file:')
if verbose:
print('Not applying any equations')
## check patterns
if not (tecplot_variable_pattern is None) and verbose:
print(f'Applying pattern {tecplot_variable_pattern} to variables')
## check that the tecplot file is there
if not os.path.exists(tecplot_plt_file_path):
raise FileNotFoundError(
f'Tecplot file does not exist: {tecplot_plt_file_path}')
## load the tecplot data
if verbose:
print('Loading tecplot data')
batsrus =
## describe the loaded tecplot data
if verbose:
print('Loaded tecplot data with variables:')
## apply equations
if verbose:
print('Applying equations to data')
if verbose:
print('Variables after equations:')
## create geometry zone
if 'shell' in geometry_params['kind']:
geometry_points = shell_geometry(geometry_params)
elif 'line' in geometry_params['kind']:
geometry_points = line_geometry(geometry_params)
elif 'rectprism' in geometry_params['kind']:
geometry_points = rectprism_geometry(geometry_params)
elif 'trajectory' in geometry_params['kind']:
geometry_points = trajectory_geometry(geometry_params)
source_zone = list(batsrus.zones())
new_zone = batsrus.add_ordered_zone('geometry',geometry_points['npoints'])
for i,d in zip((0,1,2),('X','Y','Z')):
new_zone.values(i)[:] = geometry_points[d][:]
## interpolate variables on to the geometry
if verbose:
print('Interpolating variables:')
variables = list(batsrus.variables(re.compile(tecplot_variable_pattern)))
if verbose:
for var in variables:
## add variables for shell and trajectory cases
if 'shell' in geometry_params['kind']:
new_zone.values('polar')[:] = geometry_points['polar']
new_zone.values('azimuthal')[:] = geometry_points['azimuthal']
if 'trajectory' in geometry_params['kind']:
new_zone.values('time')[:] = geometry_points['time']
## add auxiliary data
if ('trajectory' in geometry_params['kind']
and 'pandas' in geometry_params['trajectory_format']):
{'trajectory_data': type(geometry_params['trajectory_data'])}
## construct default filename
if filename == None:
filename = tecplot_plt_file_path[:-4] + f'_{geometry_params["kind"]}'
## save zone
if verbose:
print(f'Writing {filename}')
if 'hdf5' in write_as:
elif 'pandas_ascii' in write_as:
elif 'pandas_hdf5' in write_as:
elif 'tecplot_ascii' in write_as:
elif 'tecplot_plt' in write_as:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment