def test_grid_nh(self):
grid_img = Image.open(
os.path.join(os.path.dirname(__file__), 'grid_nh.png'))
grid_data = np.array(grid_img)
img = Image.new('RGB', (425, 425))
proj4_string = '+proj=laea +lat_0=90 +lon_0=0 +a=6371228.0 +units=m'
area_extent = (-5326849.0625, -5326849.0625, 5326849.0625,
5326849.0625)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l', level=4)
font = ImageFont.truetype(
os.path.join(os.path.dirname(__file__), 'test_data',
'DejaVuSerif.ttf'), 10)
cw.add_grid(img,
area_def, (10.0, 10.0), (2.0, 2.0),
font=font,
fill='blue',
outline='blue',
minor_outline='blue',
lon_placement='tblr',
lat_placement='')
res = np.array(img)
self.failUnless(fft_metric(grid_data, res),
'Writing of nh grid failed')
def test_grid(self):
grid_img = Image.open(
os.path.join(os.path.dirname(__file__), 'grid_europe.png'))
grid_data = np.array(grid_img)
img = Image.new('RGB', (640, 480))
proj4_string = \
'+proj=stere +lon_0=8.00 +lat_0=50.00 +lat_ts=50.00 +ellps=WGS84'
area_extent = (-3363403.31, -2291879.85, 2630596.69, 2203620.1)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l', level=4)
font = ImageFont.truetype(
os.path.join(os.path.dirname(__file__), 'test_data',
'DejaVuSerif.ttf'), 16)
cw.add_grid(img,
area_def, (10.0, 10.0), (2.0, 2.0),
font=font,
fill='blue',
write_text=False,
outline='blue',
minor_outline='blue')
res = np.array(img)
self.assertTrue(fft_metric(grid_data, res), 'Writing of grid failed')
def test_dateline_boundary_cross(self):
dl_img = Image.open(
os.path.join(os.path.dirname(__file__),
'dateline_boundary_cross.png'))
dl_data = np.array(dl_img)
img = Image.new('RGB', (640, 480))
proj4_string = '+proj=stere +lon_0=140.00 +lat_0=60.00 +lat_ts=50.00 +ellps=WGS84'
area_extent = (-3363403.31, -2291879.85, 2630596.69, 2203620.1)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l', level=4)
font = ImageFont.truetype(
os.path.join(os.path.dirname(__file__), 'test_data',
'DejaVuSerif.ttf'), 16)
cw.add_grid(img,
area_def, (10.0, 10.0), (2.0, 2.0),
font=font,
fill='blue',
outline='blue',
minor_outline='blue',
lon_placement='b',
lat_placement='lr')
res = np.array(img)
self.failUnless(fft_metric(dl_data, res),
'Writing of dateline boundary crossing data failed')
def test_grid_geos(self):
geos_img = Image.open(os.path.join(os.path.dirname(__file__), 'grid_geos.png'))
geos_data = np.array(geos_img)
img = Image.new('RGB', (425, 425))
proj4_string = '+proj=geos +lon_0=0.0 +a=6378169.00 +b=6356583.80 +h=35785831.0'
area_extent = (-5570248.4773392612, -5567248.074173444, 5567248.074173444, 5570248.4773392612)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l')
cw.add_grid(img, area_def, (10.0, 10.0), (2.0, 2.0), fill='blue', outline='blue', minor_outline='blue',
write_text=False)
res = np.array(img)
self.failUnless(fft_metric(geos_data, res), 'Writing of geos contours failed')
def test_grid(self):
grid_img = Image.open(os.path.join(os.path.dirname(__file__),
'grid_europe.png'))
grid_data = np.array(grid_img)
img = Image.new('RGB', (640, 480))
proj4_string = '+proj=stere +lon_0=8.00 +lat_0=50.00 +lat_ts=50.00 +ellps=WGS84'
area_extent = (-3363403.31, -2291879.85, 2630596.69, 2203620.1)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l', level=4)
font = ImageFont.truetype(os.path.join(os.path.dirname(__file__), 'test_data', 'DejaVuSerif.ttf'), 16)
cw.add_grid(img, area_def, (10.0, 10.0), (2.0, 2.0), font=font, fill='blue',
outline='blue', minor_outline='blue')
res = np.array(img)
self.failUnless(fft_metric(grid_data, res), 'Writing of grid failed')
def test_grid_nh(self):
grid_img = Image.open(os.path.join(os.path.dirname(__file__),
'grid_nh.png'))
grid_data = np.array(grid_img)
img = Image.new('RGB', (425, 425))
proj4_string = '+proj=laea +lat_0=90 +lon_0=0 +a=6371228.0 +units=m'
area_extent = (-5326849.0625, -5326849.0625, 5326849.0625, 5326849.0625)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l', level=4)
font = ImageFont.truetype(os.path.join(os.path.dirname(__file__), 'test_data', 'DejaVuSerif.ttf'), 10)
cw.add_grid(img, area_def, (10.0, 10.0), (2.0, 2.0), font=font, fill='blue',
outline='blue', minor_outline='blue',
lon_placement='tblr', lat_placement='')
res = np.array(img)
self.failUnless(fft_metric(grid_data, res), 'Writing of nh grid failed')
def test_grid_geos(self):
geos_img = Image.open(
os.path.join(os.path.dirname(__file__), 'grid_geos.png'))
geos_data = np.array(geos_img)
img = Image.new('RGB', (425, 425))
proj4_string = '+proj=geos +lon_0=0.0 +a=6378169.00 +b=6356583.80 +h=35785831.0'
area_extent = (-5570248.4773392612, -5567248.074173444,
5567248.074173444, 5570248.4773392612)
area_def = (proj4_string, area_extent)
cw = ContourWriter(gshhs_root_dir)
cw.add_coastlines(img, area_def, resolution='l')
cw.add_grid(img,
area_def, (10.0, 10.0), (2.0, 2.0),
fill='blue',
outline='blue',
minor_outline='blue',
write_text=False)
res = np.array(img)
self.failUnless(fft_metric(geos_data, res),
'Writing of geos contours failed')
def main():
parser = get_parser()
args = parser.parse_args()
levels = [logging.ERROR, logging.WARN, logging.INFO, logging.DEBUG]
logging.basicConfig(level=levels[min(3, args.verbosity)])
if args.output_filename is None:
args.output_filename = [x[:-3] + "png" for x in args.input_tiff]
else:
assert len(args.output_filename) == len(args.input_tiff), "Output filenames must be equal to number of input tiffs"
if not (args.add_borders or args.add_coastlines or args.add_grid or args.add_rivers):
LOG.error("Please specify one of the '--add-X' options to modify the image")
return -1
for input_tiff, output_filename in zip(args.input_tiff, args.output_filename):
LOG.info("Creating {} from {}".format(output_filename, input_tiff))
gtiff = gdal.Open(input_tiff)
proj4_str = osr.SpatialReference(gtiff.GetProjection()).ExportToProj4()
ul_x, res_x, _, ul_y, _, res_y = gtiff.GetGeoTransform()
half_pixel_x = res_x / 2.
half_pixel_y = res_y / 2.
area_extent = (
ul_x - half_pixel_x, # lower-left X
ul_y + res_y * gtiff.RasterYSize - half_pixel_y, # lower-left Y
ul_x + res_x * gtiff.RasterXSize + half_pixel_x, # upper-right X
ul_y + half_pixel_y, # upper-right Y
)
img = Image.open(input_tiff).convert('RGB')
area_def = (proj4_str, area_extent)
cw = ContourWriter(args.shapes_dir)
if args.add_coastlines:
outline = args.coastlines_outline[0] if len(args.coastlines_outline) == 1 else tuple(int(x) for x in args.coastlines_outline)
if args.coastlines_fill:
fill = args.coastlines_fill[0] if len(args.coastlines_fill) == 1 else tuple(int(x) for x in args.coastlines_fill)
else:
fill = None
cw.add_coastlines(img, area_def, resolution=args.coastlines_resolution, level=args.coastlines_level,
outline=outline, fill=fill)
if args.add_rivers:
outline = args.rivers_outline[0] if len(args.rivers_outline) == 1 else tuple(int(x) for x in args.rivers_outline)
cw.add_rivers(img, area_def,
resolution=args.rivers_resolution, level=args.rivers_level,
outline=outline)
if args.add_borders:
outline = args.borders_outline[0] if len(args.borders_outline) == 1 else tuple(int(x) for x in args.borders_outline)
cw.add_borders(img, area_def, resolution=args.borders_resolution, level=args.borders_level, outline=outline)
if args.add_grid:
try:
font = ImageFont.truetype(args.grid_font, args.grid_text_size)
except IOError:
font_path = get_resource_filename('polar2grid.fonts', args.grid_font)
if not os.path.exists(font_path):
raise ValueError("Font path does not exist: {}".format(font_path))
font = ImageFont.truetype(font_path, args.grid_text_size)
outline = args.grid_outline[0] if len(args.grid_outline) == 1 else tuple(int(x) for x in args.grid_outline)
minor_outline = args.grid_minor_outline[0] if len(args.grid_minor_outline) == 1 else tuple(int(x) for x in args.grid_minor_outline)
fill = args.grid_fill[0] if len(args.grid_fill) == 1 else tuple(int(x) for x in args.grid_fill)
cw.add_grid(img, area_def, args.grid_D, args.grid_d, font,
fill=fill, outline=outline, minor_outline=minor_outline,
lon_placement=args.grid_lon_placement,
lat_placement=args.grid_lat_placement)
img.save(output_filename)
def embellish(basDir, GSHHS_ROOT, imgStr, ii, dateSnap, timeSnap):
"""
What does this definition do?
Embellishes the image with custom graphics
:param basDir: Base directory path
:param GSHHS_ROOT: GSHHS installation folder
:param imgStr: Complete path of the output image data as string
:param ii: Channel name as string
:param dateSnap: Date (YYYYMMDD) name as string
:param timeSnap: Time (HHMM) name as string
:return: Image with all the decorations..
References:
[1]. https://stackoverflow.com/questions/18522295/python-pil-change-greyscale-tif-to-rgb
"""
import os, sys
import aggdraw, PIL
from PIL import Image, ImageFont
from pydecorate import DecoratorAGG as dag
from pycoast import ContourWriter
from satpy.resample import get_area_def
img = Image.open(imgStr, mode = "r")
# Convert the image into RGB if it's not
if ("L" or "A" in img.getbands()):
rgbimg = Image.new("RGBA", img.size)
rgbimg.paste(img)
img = rgbimg
else:
print("\n It was already an RGB image, so no need to convert!")
# end if-confition
# Add logos
dc = dag(img)
# dc.add_logo(basDir + 'logo/NCMRWF.png', height = 75.0)
dc.add_logo(basDir + 'logo/NCMRWF.png', height = 75.0, bg='yellow')
# Add basic text information
capStr = ii
textStr = "MSG-1: SEVIRI [" + capStr + "]\n" + dateSnap + '\n' + timeSnap + ' UTC'
# fontClr = aggdraw.Font((255, 255, 255), "/usr/share/fonts/truetype/DejaVuSerif.ttf", size = 18)
fontClr = aggdraw.Font('blue', "/usr/share/fonts/truetype/DejaVuSerif.ttf", size = 18)
dc.add_text(textStr, font = fontClr)
# Adding partner's logos
dc.align_right()
dc.add_logo(basDir + 'logo/imdlogo.jpg', height=75.0)
dc.align_left()
dc.align_bottom()
dc.add_logo(basDir + 'logo/eumetsatLogo.jpg', height=50.0)
dc.align_right()
dc.add_logo(basDir + 'logo/pytroll-logo.png', height = 50.0)
# Add coastline, grid to the image
# Set up projection info for coastlines
proj4_str = '+proj=merc +lon_0=75.0 + lat_0=17.5 + lat_ts=17.5 +ellps=WGS84'
area_extent = (-3339584.72, -1111475.10, 3339584.72, 5591295.92)
area_def = (proj4_str, area_extent)
# Add the shape file directory or the GSHHS root path
cw = ContourWriter(GSHHS_ROOT)
cw.add_coastlines(img, area_def, resolution = 'h', level = 1)
cw.add_shapefile_shapes(img, area_def, GSHHS_ROOT + 'India/Admin2.shp', outline = 'white')
# Add gridlines
fontClr2 = ImageFont.truetype("/usr/share/fonts/truetype/DejaVuSerif.ttf", 14)
cw.add_grid(img, area_def, (10.0, 10.0), (2.0, 2.0), fontClr2, fill = "white", outline = 'lightblue',
minor_outline = 'lightblue')
# Save the image
img.save(imgStr)
# return the image object
return img
def main():
parser = get_parser()
args = parser.parse_args()
levels = [logging.ERROR, logging.WARN, logging.INFO, logging.DEBUG]
logging.basicConfig(level=levels[min(3, args.verbosity)])
if args.output_filename is None:
args.output_filename = [x[:-3] + "png" for x in args.input_tiff]
else:
assert len(args.output_filename) == len(
args.input_tiff
), "Output filenames must be equal to number of input tiffs"
if not (args.add_borders or args.add_coastlines or args.add_grid
or args.add_rivers):
LOG.error(
"Please specify one of the '--add-X' options to modify the image")
return -1
for input_tiff, output_filename in zip(args.input_tiff,
args.output_filename):
LOG.info("Creating {} from {}".format(output_filename, input_tiff))
gtiff = gdal.Open(input_tiff)
proj4_str = osr.SpatialReference(gtiff.GetProjection()).ExportToProj4()
ul_x, res_x, _, ul_y, _, res_y = gtiff.GetGeoTransform()
half_pixel_x = res_x / 2.
half_pixel_y = res_y / 2.
area_extent = (
ul_x - half_pixel_x, # lower-left X
ul_y + res_y * gtiff.RasterYSize - half_pixel_y, # lower-left Y
ul_x + res_x * gtiff.RasterXSize + half_pixel_x, # upper-right X
ul_y + half_pixel_y, # upper-right Y
)
p = Proj(proj4_str)
if p.is_latlong():
# convert lat/lons to radians
area_extent = p(area_extent[0], area_extent[1]) + p(
area_extent[2], area_extent[3])
img = Image.open(input_tiff).convert('RGB')
area_def = (proj4_str, area_extent)
cw = ContourWriter(args.shapes_dir)
if args.add_coastlines:
outline = args.coastlines_outline[0] if len(
args.coastlines_outline) == 1 else tuple(
int(x) for x in args.coastlines_outline)
if args.coastlines_fill:
fill = args.coastlines_fill[0] if len(
args.coastlines_fill) == 1 else tuple(
int(x) for x in args.coastlines_fill)
else:
fill = None
cw.add_coastlines(img,
area_def,
resolution=args.coastlines_resolution,
level=args.coastlines_level,
outline=outline,
fill=fill)
if args.add_rivers:
outline = args.rivers_outline[0] if len(
args.rivers_outline) == 1 else tuple(
int(x) for x in args.rivers_outline)
cw.add_rivers(img,
area_def,
resolution=args.rivers_resolution,
level=args.rivers_level,
outline=outline)
if args.add_borders:
outline = args.borders_outline[0] if len(
args.borders_outline) == 1 else tuple(
int(x) for x in args.borders_outline)
cw.add_borders(img,
area_def,
resolution=args.borders_resolution,
level=args.borders_level,
outline=outline)
if args.add_grid:
try:
font = ImageFont.truetype(args.grid_font, args.grid_text_size)
except IOError:
font_path = get_resource_filename('polar2grid.fonts',
args.grid_font)
if not os.path.exists(font_path):
raise ValueError(
"Font path does not exist: {}".format(font_path))
font = ImageFont.truetype(font_path, args.grid_text_size)
outline = args.grid_outline[0] if len(
args.grid_outline) == 1 else tuple(
int(x) for x in args.grid_outline)
minor_outline = args.grid_minor_outline[0] if len(
args.grid_minor_outline) == 1 else tuple(
int(x) for x in args.grid_minor_outline)
fill = args.grid_fill[0] if len(args.grid_fill) == 1 else tuple(
int(x) for x in args.grid_fill)
cw.add_grid(img,
area_def,
args.grid_D,
args.grid_d,
font,
fill=fill,
outline=outline,
minor_outline=minor_outline,
lon_placement=args.grid_lon_placement,
lat_placement=args.grid_lat_placement)
img.save(output_filename)