def InitialImageCut(img, cdim, newcdim):
"""Crops image, so that the crop output can be used in GenDataset.
Parameters
----------
img : PIL.Image.Image
Image
cdim : list-like
Coordinate limits (x_min, x_max, y_min, y_max) of image.
newcdim : list-like
Crop boundaries (x_min, x_max, y_min, y_max). There is
currently NO CHECK that newcdim is within cdim!
Returns
-------
img : PIL.Image.Image
Cropped image
"""
x, y = trf.coord2pix(np.array(newcdim[:2]), np.array(newcdim[2:]), cdim,
img.shape, origin="upper")
# y is backward since origin is upper!
#box = [x[0], y[1], x[1], y[0]]
#img = img.crop(box)
#img.load()
#print(x,y)
img = img[int(x[0]):int(x[1]),int(y[1]):int(y[0])]
#print(img.shape)
return img
def WarpCraterLoc(craters, geoproj, oproj, oextent, imgdim, llbd=None,
origin="upper"):
"""Wrapper for WarpImage that adds padding to warped image to make it the
same size as the original.
Parameters
----------
craters : pandas.DataFrame
Crater info
geoproj : cartopy.crs.Geodetic instance
Input lat/long coordinate system
oproj : cartopy.crs.Projection instance
Output coordinate system
oextent : list-like
Coordinate limits (x_min, x_max, y_min, y_max)
of output
imgdim : list, tuple or ndarray
Length and height of image, in pixels
llbd : list-like
Long/lat limits (long_min, long_max,
lat_min, lat_max) of image
origin : "lower" or "upper"
Based on imshow convention for displaying image y-axis.
"upper" means that [0,0] is upper-left corner of image;
"lower" means it is bottom-left.
Returns
-------
ctr_wrp : pandas.DataFrame
DataFrame that includes pixel x, y positions
"""
# Get subset of craters within llbd limits
if llbd is None:
ctr_wrp = craters
else:
ctr_xlim = ((craters["Long"] >= llbd[0]) &
(craters["Long"] <= llbd[1]))
ctr_ylim = ((craters["Lat"] >= llbd[2]) &
(craters["Lat"] <= llbd[3]))
ctr_wrp = craters.loc[ctr_xlim & ctr_ylim, :].copy()
# Get output projection coords.
# [:,:2] becaus we don't need elevation data
# If statement is in case ctr_wrp has nothing in it
if ctr_wrp.shape[0]:
ilong = ctr_wrp["Long"].as_matrix()
ilat = ctr_wrp["Lat"].as_matrix()
res = oproj.transform_points(x=ilong, y=ilat,
src_crs=geoproj)[:, :2]
# Get output
ctr_wrp["x"], ctr_wrp["y"] = trf.coord2pix(res[:, 0], res[:, 1],
oextent, imgdim,
origin=origin)
else:
ctr_wrp["x"] = []
ctr_wrp["y"] = []
return ctr_wrp
def test_pix2coord(self, origin):
x, y = trf.coord2pix(self.cx, self.cy, self.cdim,
self.imgdim, origin=origin)
cx, cy = trf.pix2coord(x, y, self.cdim, self.imgdim,
origin=origin)
cxy = np.r_[cx, cy]
cxy_gt = np.r_[self.cx, self.cy]
assert np.all(np.isclose(cxy, cxy_gt, rtol=1e-7, atol=1e-10))
def test_coord2pix(self, origin):
x_gt = (self.imgdim[0] *
(self.cx - self.cdim[0]) / (self.cdim[1] - self.cdim[0]))
y_gt = (self.imgdim[1] *
(self.cy - self.cdim[2]) / (self.cdim[3] - self.cdim[2]))
yi_gt = (self.imgdim[1] *
(self.cdim[3] - self.cy) / (self.cdim[3] - self.cdim[2]))
x, y = trf.coord2pix(self.cx, self.cy, self.cdim,
self.imgdim, origin=origin)
if origin == "upper":
y_gt_curr = yi_gt
else:
y_gt_curr = y_gt
xy = np.r_[x, y]
xy_gt = np.r_[x_gt, y_gt_curr]
assert np.all(np.isclose(xy, xy_gt, rtol=1e-7, atol=1e-10))
def AddPlateCarree_XY(craters, imgdim, cdim=[-180., 180., -90., 90.],
origin="upper"):
"""Adds x and y pixel locations to craters dataframe.
Parameters
----------
craters : pandas.DataFrame
Crater info
imgdim : list, tuple or ndarray
Length and height of image, in pixels
cdim : list-like, optional
Coordinate limits (x_min, x_max, y_min, y_max) of image. Default is
[-180., 180., -90., 90.].
origin : "upper" or "lower", optional
Based on imshow convention for displaying image y-axis.
"upper" means that [0,0] is upper-left corner of image;
"lower" means it is bottom-left.
"""
x, y = trf.coord2pix(craters["Long"].as_matrix(),
craters["Lat"].as_matrix(),
cdim, imgdim, origin=origin)
craters["x"] = x
craters["y"] = y
