def topng(inp, date=None):
"""Convert *inp* into a PNG file. Input file can be a step5mask
file (produces greyscale PNG), or if *date* is supplied it can be a
subbox file."""
# :todo: move into proper module.
from landmask import centrein
resolution = 0.25
width = 360 / resolution
height = 180 / resolution
assert int(width) == width
assert int(height) == height
width = int(width)
height = int(height)
cells = eqarea.grid8k()
if not date:
# Mask file in text format.
values = gio.maskboxes(inp, cells)
colour = greyscale
isgrey = True
else:
values = extractdate(inp, cells, date)
colour = colourscale
isgrey = False
if isgrey:
black = 0
else:
black = (0, 0, 0)
# an array of rows:
a = [[black] * width for _ in range(height)]
for v, box in values:
v = colour(v)
for x, y in centrein(box, resolution):
a[y][x] = v
# For colour images each row of *a* is of the form:
# [(R,G,B), (R,G,B), ...] we want to flatten it to:
# [R,G,B,R,G,B,...]
if not isgrey:
a = [list(itertools.chain(*row)) for row in a]
try:
outpath = inp.name + '.png'
except:
outpath = 'out.png'
w = png.Writer(width=width,
height=height,
greyscale=isgrey,
alpha=False,
bitdepth=8)
w.write(open(outpath, 'wb'), a)
def topng(inp, date=None):
"""Convert *inp* into a PNG file. Input file can be a step5mask
file (produces greyscale PNG), or if *date* is supplied it can be a
subbox file."""
# :todo: move into proper module.
from landmask import centrein
resolution = 0.25
width = 360 / resolution
height = 180 / resolution
assert int(width) == width
assert int(height) == height
width = int(width)
height = int(height)
cells = eqarea.grid8k()
if not date:
# Mask file in text format.
values = gio.maskboxes(inp, cells)
colour = greyscale
isgrey = True
else:
values = extractdate(inp, cells, date)
colour = colourscale
isgrey = False
if isgrey:
black = 0
else:
black = (0, 0, 0)
# an array of rows:
a = [[black] * width for _ in range(height)]
for v, box in values:
v = colour(v)
for x, y in centrein(box, resolution):
a[y][x] = v
# For colour images each row of *a* is of the form:
# [(R,G,B), (R,G,B), ...] we want to flatten it to:
# [R,G,B,R,G,B,...]
if not isgrey:
a = [list(itertools.chain(*row)) for row in a]
try:
outpath = inp.name + ".png"
except:
outpath = "out.png"
w = png.Writer(width=width, height=height, greyscale=isgrey, alpha=False, bitdepth=8)
w.write(open(outpath, "wb"), a)
def maskmap(argv):
"""[command] [--class css-class] mask Draws an SVG map of the
cells in mask."""
opts,arg = getopt.getopt(argv[1:], '', ['class='])
option = dict((o[2:],v) for o,v in opts)
from code import eqarea
maskfile = arg[0]
out = sys.stdout
def transform(p):
u"""Transform p=(lat,lon) into (x,y) used for map system. Which
in this case is Plate Carr\xe9e with x from 0 to 720, and y from
0 to 360."""
lat,lon = p
y = (90+lat)*2
x = (180+lon)*2
return x,y
out.write("""<svg
xmlns="http://www.w3.org/2000/svg"
xmlns:xlink="http://www.w3.org/1999/xlink"
version="1.1">\n""")
out.write("""<defs>
<style type="text/css">
path.m0 { stroke: none; fill: rgb(0,0,0); fill-opacity: 0.0; }
path.m1 { stroke: none; fill: rgb(255,0,0); fill-opacity: 0.4; }
</style>\n</defs>
""")
class_ = option.get('class', 'mask')
out.write("<g class='%s' transform='translate(0,360) scale(1,-1)'>\n" %
class_)
cells = eqarea.grid8k()
for v,box in gio.maskboxes(open(maskfile), cells):
s,n,w,e = box
corners = [(n,w), (n,e), (s,e), (s,w)]
corners = map(transform, corners)
out.write("<path class='m%.0f'" % v +
" d='M %.2f %.2f L" % corners[0] +
" %.2f"*6 % tuple(itertools.chain(*corners[1:])) +
" z' />\n")
out.write("</g>\n")
out.write("</svg>\n")
def cells(inp, date=None, trend=True):
"""
Yield a series of (value, rect) pairs.
"""
subboxes = eqarea.grid8k()
if not date and not trend:
# Mask file in text format.
values = gio.maskboxes(inp, subboxes)
return values
assert not (date and trend)
if date:
values = extract_date(inp, subboxes, date)
elif trend:
values = extract_trend(inp, subboxes)
return values