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 468bf7a6 authored by Camilla Dodge Koslowsky Harris's avatar Camilla Dodge Koslowsky Harris
Browse files

very simple h5 save format implemented

parent 31746013
......@@ -24,6 +24,7 @@ __email__ = 'cdha@umich.edu'
import os
import re
import h5py
import numpy as np
import tecplot
......@@ -217,6 +218,15 @@ def _trajectory_geometry(geometry_params: dict) -> dict:
def _save_hdf5(filename, geometry_params, new_zone, variables) -> None:
"""Save the aux data and a subset of the variables in hdf5 format.
"""
column_names = [var.name for var in variables]
tp_data = []
for var in variables:
tp_data.append(new_zone.values(var)[:])
tp_data_np = np.array(tp_data).transpose()
with h5py.File(filename, 'w-') as h5_file:
h5_file['data'] = tp_data_np
h5_file['data'].attrs.update(geometry_params)
h5_file['data'].attrs['names'] = column_names
def _save_csv(filename, geometry_params, new_zone, variables) -> None:
......
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