def savefig(fname, loc=1, decorate=True, **kwargs):
import io
import os
import sys
from PIL import Image
import matplotlib.pyplot as plt
try:
from pydecorate import DecoratorAGG
pydecorate = True
except ImportError:
pydecorate = False
plt.savefig(fname, **kwargs)
if pydecorate and decorate:
img = Image.open(fname)
dc = DecoratorAGG(img)
if loc == 1:
dc.align_bottom()
elif loc == 2:
dc.align_bottom()
dc.align_right()
elif loc == 3:
dc.align_right()
# sys.argv[0])[-5] + 'data/MONET_logo.png'
# print(os.path.basename(__file__))
logo = os.path.abspath(__file__)[:-17] + 'data/MONET-logo.png'
# print(logo)
dc.add_logo(logo)
if fname.split('.')[-1] == 'png':
img.save(fname, "PNG")
elif fname.split('.')[-1] == 'jpg':
img.save(fname, "JPEG")
def show_swath_pycoast(self, start, period=None):
"""
A helper method that displays the orbital swath
starting at datetime start, for a period number of minutes.
If, start is iterable, then the method assumes it is an iterable
of datetimes, plotting a number of swaths at those times.
"""
# test if start is iterable, EAFP style:
try:
for e in start:
pass
except TypeError:
start = [start]
start.sort()
from PIL import Image
from pycoast import ContourWriterAGG
from pydecorate import DecoratorAGG
img = Image.new('RGB', (650, 650))
proj4_string = ""
for x in self.working_projection:
proj4_string += "+%s=%s "%(x,self.working_projection[x])
area_extent = (-6700000.0, -6700000.0, 6700000.0, 6700000.0)
area_def = (proj4_string, area_extent)
cw = ContourWriterAGG()
cw.add_grid(img, area_def, (10.0,10.0),(2.0,2.0), fill='blue',
outline='gray', outline_opacity=130, minor_outline=None, write_text=False)
# Plot granules
for t in start:
# fetch the coordinates
xys_segs = self.swath_working_projection(t, period)
for xys in xys_segs:
lls = self.proj(xys[0],xys[1],inverse=True)
cw.add_polygon(img, area_def, zip(lls[0], lls[1]), outline="blue", fill="blue", fill_opacity=70, width=1)
cw.add_coastlines(img, area_def, resolution='l')
aoi_coords = zip(*self.aoi)
## TODO: Handle single point case properly
if len(aoi_coords) == 1:
x, y = aoi_coords[0]
d = 0.5
line_coords = [(x-d,y),(x+d,y)]
cw.add_line(img, area_def, line_coords,
outline="red", fill="red", fill_opacity=100, width=2)
elif len(aoi_coords) == 2:
cw.add_line(img, area_def, aoi_coords, outline="red", fill="red", fill_opacity=100, width=10)
else:
cw.add_polygon(img, area_def, aoi_coords, outline="red", fill="red", fill_opacity=100, width=2)
# Decorate
dc = DecoratorAGG(img)
text = "Granules from time: %s + %.2f min."%(start[0].strftime('%Y.%m.%d %H:%M:%S'),
(start[-1]-start[0]).total_seconds()/60.0)
dc.align_bottom()
dc.add_text(text,height=0)
img.show()
def decorate_pilimg(self, pilimg):
"""Apply decorations to an image
Parameters
----------
pilimg : PIL.Image
"""
dc = DecoratorAGG(pilimg)
dc.align_bottom()
self.apply_colorbar(dc)
self.apply_label(dc)
def plot(data, text, minmax=None, transpose=False,
filename=None, position=(-16.743860721,64.915348712,712.4),
tilt=None, target_position=None, target_name="", vfov=18.5, tick_distance=50.0):
if transpose:
data = data.transpose()
if minmax is None:
rainbow.set_range(data.min(),data.max())
else:
rainbow.set_range(minmax[0],minmax[1])
img = Image(data, mode="L")
img.colorize(rainbow)
img = img.pil_image()
# Decoration
dc = DecoratorAGG(img)
dc.align_bottom()
#dc.add_logo("/home/sat/dev/pydecorate/logos/VI_Logo_Transp.png")
#dc.add_logo("Nicarnica-Aviation-22-oct-300x102.png")
try:
dc.add_logo("/home/master/bin/futurevolc_logo.png", height=47)
dc.add_logo("/home/master/bin/nicarnica_logo.png")
dc.add_logo("/home/master/bin/vi_logo.png")
except IOError:
dc.add_logo("bin/futurevolc_logo.png", height=47)
dc.add_logo("bin/nicarnica_logo.png")
dc.add_logo("bin/vi_logo.png")
dc.add_text(text, font=font)
tick_space = 5.0
#dc.add_scale(rainbow, extend=True, unit='°C', tick_marks=tick_space, minor_tick_marks=tick_space)
# target distance and km/px
ny = data.shape[0]
nx = data.shape[1]
distance = distance_longlat(target_position,position)
# plot grid
lines, texts = prepare_graticule(nx, ny, distance=distance, vfov=vfov,
tilt=tilt, tick_distance=tick_distance,
height=position[2],target_height=target_position[2],
target_name=target_name)
draw_lines(img, lines)
draw_texts(img, texts)
if filename == None:
img.show()
else:
img.save(filename, "JPEG", quality=90)
def savefig(fname, loc=1, decorate=True, **kwargs):
"""save figure and add the MONET logo .
Parameters
----------
fname : str
output file name.
loc : int
the location for the monet logo.
decorate : bool
Description of parameter `decorate`.
**kwargs : dict
kwargs for the matplotlib.pyplot.savefig function.
Returns
-------
type
Description of returned object.
"""
import io
import os
import sys
from PIL import Image
import matplotlib.pyplot as plt
try:
from pydecorate import DecoratorAGG
pydecorate = True
except ImportError:
pydecorate = False
plt.savefig(fname, **kwargs)
if pydecorate and decorate:
img = Image.open(fname)
dc = DecoratorAGG(img)
if loc == 1:
dc.align_bottom()
elif loc == 2:
dc.align_bottom()
dc.align_right()
elif loc == 3:
dc.align_right()
# sys.argv[0])[-5] + 'data/MONET_logo.png'
# print(os.path.basename(__file__))
logo = os.path.abspath(__file__)[:-17] + 'data/MONET-logo.png'
# print(logo)
dc.add_logo(logo)
if fname.split('.')[-1] == 'png':
img.save(fname, "PNG")
elif fname.split('.')[-1] == 'jpg':
img.save(fname, "JPEG")
def test_style_retention():
# import aggdraw
from PIL import Image
from trollimage.colormap import rdbu
from pydecorate import DecoratorAGG
# font = aggdraw.Font("navy", DEJAVU_FONT, size=20)
# font_scale = aggdraw.Font("black", DEJAVU_FONT, size=12)
rdbu.colors = rdbu.colors[::-1]
rdbu.set_range(-90, 10)
img = Image.open(os.path.join(REPOS_ROOT, "BMNG_clouds_201109181715_areaT2.png"))
dc = DecoratorAGG(img)
# dc.write_vertically()
# dc.add_logo("logos/pytroll_light_big.png")
# dc.add_logo("logos/NASA_Logo.gif",margins=[10,10],bg='yellow')
# dc.add_logo("logos/pytroll_light_big.png")
# font = aggdraw.Font("blue", DEJAVU_FONT, size=16)
# dc.add_text("Some text",font=font)
# dc.align_right()
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit="K")
# dc.align_bottom()
# dc.add_scale(rdbu, extend=True, tick_marks=2.0, line_opacity=100, width=60)
# dc.align_right()
# dc.write_vertically()
dc.align_bottom()
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit="K")
# dc.align_left()
# dc.add_scale(rdbu, extend=True, font=font_scale, tick_marks=2.0, minor_tick_marks=1.0,
# line_opacity=100, width=60, unit='K')
# img.show()
img.save("style_retention.png")
font = aggdraw.Font("blue",
"/usr/share/fonts/truetype/ttf-dejavu/DejaVuSerif.ttf",
size=16)
#dc.add_text("Some text",font=font)
#dc.align_right()
print rdbu.values
print rdbu.colors
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit='K')
#dc.align_bottom()
#dc.add_scale(rdbu, extend=True, tick_marks=2.0, line_opacity=100, width=60)
#dc.align_right()
#dc.write_vertically()
dc.align_bottom()
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit='K')
#dc.align_left()
#dc.add_scale(rdbu, extend=True, font=font_scale, tick_marks=2.0, minor_tick_marks=1.0, line_opacity=100, width=60, unit='K')
img.show()
#img.save("style_retention.png")
exit()
#dc.align_right()
#dc.align_bottom()
#dc.add_logo("logos/pytroll_light_big.png")
#dc.add_logo("logos/NASA_Logo.gif")
#dc.add_text("This is manually\nplaced text\nover here.",cursor=[400,480])
dc.new_line()
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 or args.add_colorbar):
LOG.error(
"Please specify one of the '--add-X' options to modify the image")
return -1
# we may be dealing with large images that look like decompression bombs
# let's turn off the check for the image size in PIL/Pillow
Image.MAX_IMAGE_PIXELS = None
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('RGBA' if gtiff.RasterCount in (
2, 4) else 'RGB')
area_def = (proj4_str, area_extent)
cw = ContourWriterAGG(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,
width=args.coastlines_width,
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,
width=args.rivers_width,
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,
width=args.borders_width)
if args.add_grid:
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)
font_path = find_font(args.grid_font, args.grid_text_size)
font = Font(outline, font_path, size=args.grid_text_size)
cw.add_grid(img,
area_def,
args.grid_D,
args.grid_d,
font,
fill=fill,
outline=outline,
minor_outline=minor_outline,
write_text=args.grid_text,
width=args.grid_width,
lon_placement=args.grid_lon_placement,
lat_placement=args.grid_lat_placement)
if args.add_colorbar:
from pydecorate import DecoratorAGG
font_color = args.colorbar_text_color
font_color = font_color[0] if len(font_color) == 1 else tuple(
int(x) for x in font_color)
font_path = find_font(args.colorbar_font, args.colorbar_text_size)
# this actually needs an aggdraw font
font = Font(font_color, font_path, size=args.colorbar_text_size)
band_count = gtiff.RasterCount
if band_count not in [1, 2]:
raise ValueError("Can't add colorbar to RGB/RGBA image")
# figure out what colormap we are dealing with
band = gtiff.GetRasterBand(1)
cmap = get_colormap(band, band_count)
# figure out our limits
vmin = args.colorbar_min
vmax = args.colorbar_max
metadata = gtiff.GetMetadata_Dict()
vmin = vmin or metadata.get('min_in')
vmax = vmax or metadata.get('max_in')
if isinstance(vmin, str):
vmin = float(vmin)
if isinstance(vmax, str):
vmax = float(vmax)
if vmin is None or vmax is None:
data = gtiff.GetRasterBand(1).ReadAsArray()
vmin = vmin or np.iinfo(data.dtype).min
vmax = vmax or np.iinfo(data.dtype).max
cmap.set_range(vmin, vmax)
dc = DecoratorAGG(img)
if args.colorbar_align == 'top':
dc.align_top()
elif args.colorbar_align == 'bottom':
dc.align_bottom()
elif args.colorbar_align == 'left':
dc.align_left()
elif args.colorbar_align == 'right':
dc.align_right()
if args.colorbar_vertical:
dc.write_vertically()
else:
dc.write_horizontally()
if args.colorbar_width is None or args.colorbar_height is None:
LOG.warning("'--colorbar-width' or '--colorbar-height' were "
"not specified. Forcing '--colorbar-extend'.")
args.colorbar_extend = True
kwargs = {}
if args.colorbar_width:
kwargs['width'] = args.colorbar_width
if args.colorbar_height:
kwargs['height'] = args.colorbar_height
dc.add_scale(cmap,
extend=args.colorbar_extend,
font=font,
line=font_color,
tick_marks=args.colorbar_tick_marks,
title=args.colorbar_title,
unit=args.colorbar_units,
**kwargs)
img.save(output_filename)
def show_swath_pycoast(self, start, period=None):
"""
A helper method that displays the orbital swath
starting at datetime start, for a period number of minutes.
If, start is iterable, then the method assumes it is an iterable
of datetimes, plotting a number of swaths at those times.
"""
# test if start is iterable, EAFP style:
try:
for e in start:
pass
except TypeError:
start = [start]
start.sort()
from PIL import Image
from pycoast import ContourWriterAGG
from pydecorate import DecoratorAGG
img = Image.new('RGB', (650, 650))
proj4_string = ""
for x in self.working_projection:
proj4_string += "+%s=%s " % (x, self.working_projection[x])
area_extent = (-6700000.0, -6700000.0, 6700000.0, 6700000.0)
area_def = (proj4_string, area_extent)
cw = ContourWriterAGG()
cw.add_grid(img,
area_def, (10.0, 10.0), (2.0, 2.0),
fill='blue',
outline='gray',
outline_opacity=130,
minor_outline=None,
write_text=False)
# Plot granules
for t in start:
# fetch the coordinates
xys_segs = self.swath_working_projection(t, period)
for xys in xys_segs:
lls = self.proj(xys[0], xys[1], inverse=True)
cw.add_polygon(img,
area_def,
zip(lls[0], lls[1]),
outline="blue",
fill="blue",
fill_opacity=70,
width=1)
cw.add_coastlines(img, area_def, resolution='l')
aoi_coords = zip(*self.aoi)
## TODO: Handle single point case properly
if len(aoi_coords) == 1:
x, y = aoi_coords[0]
d = 0.5
line_coords = [(x - d, y), (x + d, y)]
cw.add_line(img,
area_def,
line_coords,
outline="red",
fill="red",
fill_opacity=100,
width=2)
elif len(aoi_coords) == 2:
cw.add_line(img,
area_def,
aoi_coords,
outline="red",
fill="red",
fill_opacity=100,
width=10)
else:
cw.add_polygon(img,
area_def,
aoi_coords,
outline="red",
fill="red",
fill_opacity=100,
width=2)
# Decorate
dc = DecoratorAGG(img)
text = "Granules from time: %s + %.2f min." % (
start[0].strftime('%Y.%m.%d %H:%M:%S'),
(start[-1] - start[0]).total_seconds() / 60.0)
dc.align_bottom()
dc.add_text(text, height=0)
img.show()
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 or args.add_colorbar):
LOG.error("Please specify one of the '--add-X' options to modify the image")
return -1
# we may be dealing with large images that look like decompression bombs
# let's turn off the check for the image size in PIL/Pillow
Image.MAX_IMAGE_PIXELS = None
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 = ContourWriterAGG(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:
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)
font_path = find_font(args.grid_font, args.grid_text_size)
font = Font(outline, font_path, size=args.grid_text_size)
cw.add_grid(img, area_def, args.grid_D, args.grid_d, font,
fill=fill, outline=outline, minor_outline=minor_outline,
write_text=args.grid_text,
lon_placement=args.grid_lon_placement,
lat_placement=args.grid_lat_placement)
if args.add_colorbar:
from pydecorate import DecoratorAGG
font_color = args.colorbar_text_color
font_color = font_color[0] if len(font_color) == 1 else tuple(int(x) for x in font_color)
font_path = find_font(args.colorbar_font, args.colorbar_text_size)
# this actually needs an aggdraw font
font = Font(font_color, font_path, size=args.colorbar_text_size)
band_count = gtiff.RasterCount
if band_count not in [1, 2]:
raise ValueError("Can't add colorbar to RGB/RGBA image")
# figure out what colormap we are dealing with
band = gtiff.GetRasterBand(1)
cmap = get_colormap(band, band_count)
# figure out our limits
vmin = args.colorbar_min
vmax = args.colorbar_max
metadata = gtiff.GetMetadata_Dict()
vmin = vmin or metadata.get('min_in')
vmax = vmax or metadata.get('max_in')
if isinstance(vmin, str):
vmin = float(vmin)
if isinstance(vmax, str):
vmax = float(vmax)
if vmin is None or vmax is None:
data = gtiff.GetRasterBand(1).ReadAsArray()
vmin = vmin or np.iinfo(data.dtype).min
vmax = vmax or np.iinfo(data.dtype).max
cmap.set_range(vmin, vmax)
dc = DecoratorAGG(img)
if args.colorbar_align == 'top':
dc.align_top()
elif args.colorbar_align == 'bottom':
dc.align_bottom()
elif args.colorbar_align == 'left':
dc.align_left()
elif args.colorbar_align == 'right':
dc.align_right()
if args.colorbar_vertical:
dc.write_vertically()
else:
dc.write_horizontally()
if args.colorbar_width is None or args.colorbar_height is None:
LOG.warning("'--colorbar-width' or '--colorbar-height' were "
"not specified. Forcing '--colorbar-extend'.")
args.colorbar_extend = True
kwargs = {}
if args.colorbar_width:
kwargs['width'] = args.colorbar_width
if args.colorbar_height:
kwargs['height'] = args.colorbar_height
dc.add_scale(cmap, extend=args.colorbar_extend,
font=font,
line=font_color,
tick_marks=args.colorbar_tick_marks,
title=args.colorbar_title,
unit=args.colorbar_units,
**kwargs)
img.save(output_filename)
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 or args.add_colorbar):
LOG.error("Please specify one of the '--add-X' options to modify the image")
return -1
if args.cache_dir and not os.path.isdir(args.cache_dir):
LOG.info(f"Creating cache directory: {args.cache_dir}")
os.makedirs(args.cache_dir, exist_ok=True)
# we may be dealing with large images that look like decompression bombs
# let's turn off the check for the image size in PIL/Pillow
Image.MAX_IMAGE_PIXELS = None
# gather all options into a single dictionary that we can pass to pycoast
pycoast_options = _args_to_pycoast_dict(args)
for input_tiff, output_filename in zip(args.input_tiff, args.output_filename):
LOG.info("Creating {} from {}".format(output_filename, input_tiff))
img = Image.open(input_tiff)
img_bands = img.getbands()
num_bands = len(img_bands)
# P = palette which we assume to be an RGBA colormap
img = img.convert("RGBA" if num_bands in (2, 4) or "P" in img_bands else "RGB")
if pycoast_options:
area_id = os.path.splitext(input_tiff[0])[0]
area_def = get_area_def_from_raster(input_tiff, area_id=area_id)
cw = ContourWriterAGG(args.shapes_dir)
cw.add_overlay_from_dict(pycoast_options, area_def, background=img)
if args.add_colorbar:
from pydecorate import DecoratorAGG
font_color = args.colorbar_text_color
font_color = font_color[0] if len(font_color) == 1 else tuple(int(x) for x in font_color)
font_path = find_font(args.colorbar_font, args.colorbar_text_size)
# this actually needs an aggdraw font
font = Font(font_color, font_path, size=args.colorbar_text_size)
if num_bands not in (1, 2):
raise ValueError("Can't add colorbar to RGB/RGBA image")
# figure out what colormap we are dealing with
rio_ds = rasterio.open(input_tiff)
input_dtype = np.dtype(rio_ds.meta["dtype"])
rio_ct = _get_rio_colormap(rio_ds, 1)
cmap = get_colormap(input_dtype, rio_ct, num_bands)
# figure out our limits
vmin = args.colorbar_min
vmax = args.colorbar_max
metadata = rio_ds.tags()
vmin = vmin or metadata.get("min_in")
vmax = vmax or metadata.get("max_in")
if isinstance(vmin, str):
vmin = float(vmin)
if isinstance(vmax, str):
vmax = float(vmax)
if vmin is None or vmax is None:
vmin = vmin or np.iinfo(input_dtype).min
vmax = vmax or np.iinfo(input_dtype).max
cmap.set_range(vmin, vmax)
dc = DecoratorAGG(img)
if args.colorbar_align == "top":
dc.align_top()
elif args.colorbar_align == "bottom":
dc.align_bottom()
elif args.colorbar_align == "left":
dc.align_left()
elif args.colorbar_align == "right":
dc.align_right()
if args.colorbar_vertical:
dc.write_vertically()
else:
dc.write_horizontally()
if args.colorbar_width is None or args.colorbar_height is None:
LOG.warning(
"'--colorbar-width' or '--colorbar-height' were " "not specified. Forcing '--colorbar-extend'."
)
args.colorbar_extend = True
kwargs = {}
if args.colorbar_width:
kwargs["width"] = args.colorbar_width
if args.colorbar_height:
kwargs["height"] = args.colorbar_height
dc.add_scale(
cmap,
extend=args.colorbar_extend,
font=font,
line=font_color,
tick_marks=args.colorbar_tick_marks,
title=args.colorbar_title,
unit=args.colorbar_units,
**kwargs,
)
img.save(output_filename)
#dc.add_logo("logos/pytroll_light_big.png")
font=aggdraw.Font("blue","/usr/share/fonts/truetype/ttf-dejavu/DejaVuSerif.ttf",size=16)
#dc.add_text("Some text",font=font)
#dc.align_right()
print rdbu.values
print rdbu.colors
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit='K')
#dc.align_bottom()
#dc.add_scale(rdbu, extend=True, tick_marks=2.0, line_opacity=100, width=60)
#dc.align_right()
#dc.write_vertically()
dc.align_bottom()
dc.add_scale(rdbu, extend=True, tick_marks=5.0, line_opacity=100, unit='K')
#dc.align_left()
#dc.add_scale(rdbu, extend=True, font=font_scale, tick_marks=2.0, minor_tick_marks=1.0, line_opacity=100, width=60, unit='K')
img.show()
#img.save("style_retention.png")
exit()
