Skip to content
Snippets Groups Projects
Select Git revision
  • 6e103987a8c7487e6f66ad72282c0437cb7f6fe7
  • master default protected
  • use_flight_altitude
  • distribute
4 results

romio.py

Blame
    • romio.py 1.46 KiB
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59 
    

    

    

    import xarray as xr
import numpy as np
from util.geos_nav import GEOSNavigation
from netCDF4 import Dataset
import pickle


def romio_to_fgf(filename, goes_e_w='EAST'):
    ds = xr.open_dataset(filename, engine='cfgrib')

    lons = ds['longitude']
    lats = ds['latitude']

    lons = lons.values
    lats = lats.values

    lat_array, lon_array = np.meshgrid(lats, lons, indexing='ij')

    nav = GEOSNavigation()

    if goes_e_w == 'EAST':
        nav = GEOSNavigation()
    elif goes_e_w == 'WEST':
        nav = GEOSNavigation(sub_lon=-137.0)

    cc, ll = nav.earth_to_lc_s(lon_array.flatten(), lat_array.flatten())

    cc = cc.reshape(lon_array.shape)
    ll = ll.reshape(lon_array.shape)

    f = open('/home/rink/elems.pkl', 'wb')
    pickle.dump(cc, f)
    f.close()

    f = open('/home/rink/lines.pkl', 'wb')
    pickle.dump(ll, f)
    f.close()

    ds.close()

    return cc, ll


def create_file(filename, fgf_elem, fgf_line):
    dim_x_len = fgf_elem.shape[1]
    dim_y_len = fgf_elem.shape[0]

    rootgrp = Dataset(filename, 'w', format='NETCDF4')
    dim_x = rootgrp.createDimension('romio_x', size=dim_x_len)
    dim_y = rootgrp.createDimension('romio_y', size=dim_y_len)

    romio_fgf_elems = rootgrp.createVariable('romio_fgf_elems', 'f4', ['romio_y', 'romio_x'])
    romio_fgf_lines = rootgrp.createVariable('romio_fgf_lines', 'f4', ['romio_y', 'romio_x'])

    romio_fgf_elems[:, :] = fgf_elem[:, :]
    romio_fgf_lines[:, :] = fgf_line[:, :]

    rootgrp.close()