def curtain_for_line(data, X, Y, coords, day=0, numk=0): """ extracts curtain data from a grid given pairs of lon, lat input: H (a pf.Header with a FD data object from H.read_grid) if the read_grid method has not been called, we'll call it below. flighttrack (a numpy array with datetime, lon, lat, elv values) nspec: is optional for reference to the FD dict. index: is optional in case numpointspec > 1 get_track: if True extracts the points along the flight track rather than the curtain. output: curtain (a 2-d array with shape (len(flighttrack), len(H.outheight) TODO:: add interpolation, extract a track, not curtain (e.g. z points) """ grid = data.data_cube grid = grid[:, :, :, day, numk] curtain = np.zeros((len(coords), grid.shape[2])) for i, (x, y) in enumerate(coords): I = closest(x, X) J = closest(y, Y) curtain[i] = grid[I, J, :] curtain = np.nan_to_num(curtain) # curtain = np.ma.fix_invalid(np.array(curtain,dtype=float)) return curtain.T
def curtain_for_line(data, X, Y, coords, day=0, numk=0): """ extracts curtain data from a grid given pairs of lon, lat input: H (a pf.Header with a FD data object from H.read_grid) if the read_grid method has not been called, we'll call it below. flighttrack (a numpy array with datetime, lon, lat, elv values) nspec: is optional for reference to the FD dict. index: is optional in case numpointspec > 1 get_track: if True extracts the points along the flight track rather than the curtain. output: curtain (a 2-d array with shape (len(flighttrack), len(H.outheight) TODO:: add interpolation, extract a track, not curtain (e.g. z points) """ grid = data.data_cube grid = grid[:, :, :, day, numk] curtain = np.zeros((len(coords), grid.shape[2])) for i, (x, y) in enumerate(coords): I = closest(x, X) J = closest(y, Y) curtain[i] = grid[I, J, :] curtain = np.nan_to_num(curtain) # curtain = np.ma.fix_invalid(np.array(curtain,dtype=float)) return curtain.T
def closest_date(self, dateval, fmt=None): """ given a datestring or datetime, tries to find the closest date. if passed a list, assumes it is a list of datetimes """ if isinstance(dateval, str): if not fmt: if len(dateval) == 8: fmt = '%Y%m%d' if len(dateval) == 14: fmt = '%Y%m%d%H%M%S' else: raise IOError("no format provided for datestring") print("Assuming date format: {0}".format(fmt)) dateval = dt.datetime.strptime(dateval, fmt) return closest(dateval, self['available_dates_dt'])