diff options
Diffstat (limited to 'plotter')
| -rwxr-xr-x | plotter/horizontal.py | 8 | ||||
| -rwxr-xr-x | plotter/vertical_from_grib.py | 8 |
2 files changed, 14 insertions, 2 deletions
diff --git a/plotter/horizontal.py b/plotter/horizontal.py index a81b372..e83b683 100755 --- a/plotter/horizontal.py +++ b/plotter/horizontal.py @@ -37,8 +37,12 @@ def run(source, plots, output='.'): misc.create_output_dir(output) data = xr.load_dataset(source, engine='cfgrib') + index = [] + for plot in plots: - _plot(data, output, **plot) + index.append(_plot(data, output, **plot)) + + return index def _plot(data, output, name, layers, area = None): index = [] @@ -88,6 +92,8 @@ def _plot(data, output, name, layers, area = None): with open(os.path.join(output, f'{name}.index.json'), 'w') as f: f.write(json.dumps(index, indent=4)) + return { 'name': name, 'indexfile': f'{name}.index.json' } + def _layer(data, layertype, **kwargs): layertypes={ 'raster': { diff --git a/plotter/vertical_from_grib.py b/plotter/vertical_from_grib.py index 8aa851a..929782e 100755 --- a/plotter/vertical_from_grib.py +++ b/plotter/vertical_from_grib.py @@ -29,8 +29,12 @@ def run(source, plots, output='.'): misc.create_output_dir(output) data = xr.load_dataset(source, engine='cfgrib') + index = [] + for plot in plots: - _plot(data, output, **plot) + index.append(_plot(data, output, **plot)) + + return index def _plot(data, output, lat, lon, name, analysis=None): for_temp = data.sel(latitude=lat, longitude = lon, method='nearest') @@ -84,5 +88,7 @@ def _plot(data, output, lat, lon, name, analysis=None): with open(os.path.join(output, f'skewt_{name}.index.json'), 'w') as f: f.write(json.dumps(index, indent=4)) + return {'name': name, 'indexfile': f'skewt_{name}.index.json'} + if __name__ == '__main__': run(**config) |