Add --scale option to render.

For best results, use fractions like 0.5, 0.25.
This commit is contained in:
voussoir 2018-06-30 00:05:18 -07:00
parent e7fb923438
commit e053713e4f
2 changed files with 36 additions and 28 deletions

View file

@ -24,23 +24,14 @@ down and right are positive.
Commands: Commands:
update: - **update**: Download chunks into the database.
Download chunks into the database.
> pixelcanvas.py update ~100.~100--100.100 `> pixelcanvas.py update ~100.~100--100.100`
render: - **render**: Export a region as PNG.
Export an image as PNG.
> pixelcanvas.py render 0.0--100.100 <flags> `> pixelcanvas.py render 0.0--100.100 <flags>`
flags:
--show:
Instead of saving the image, display it on the screen.
https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.show
--update:
Update the chunks before exporting them.
So, for example: So, for example:
@ -51,11 +42,12 @@ So, for example:
> pixelcanvas.py render 0.0--100.100 > pixelcanvas.py render 0.0--100.100
> pixelcanvas.py render ~100.~100--100.100 --update > pixelcanvas.py render ~100.~100--100.100 --update
> pixelcanvas.py render ~1200.300--~900.600 --show > pixelcanvas.py render ~1200.300--~900.600 --show
> pixelcanvas.py render ~2000.2000--2000.2000 --scale 0.25
# To do # To do
Here are some things we might like to improve: Here are some things we might like to improve:
- Some way to get a statistics overview or visual map of which chunks we have in the database, so we know what we're missing. - Some way to get a statistics overview or visual map of which chunks we have in the database, so we know what we're missing.
- Render the image is it appeared at some point in the past, taking advantage of the `updated_at` column. - Render the image as it appeared at some point in the past, taking advantage of the `updated_at` column.
- Probably never going to happen: A GUI application to browse the db just like the site. - Probably never going to happen: A GUI application to browse the db just like the site.

View file

@ -119,6 +119,11 @@ render:
> pixelcanvas.py render 0.0--100.100 <flags> > pixelcanvas.py render 0.0--100.100 <flags>
flags: flags:
--scale <float>:
Render the image at a different scale.
For best results, use powers of 2 like 0.5, 0.25, etc.
This will disable the autocropping.
--show: --show:
Instead of saving the image, display it on the screen. Instead of saving the image, display it on the screen.
https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.show https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.show
@ -343,10 +348,14 @@ def rgb_to_image(matrix):
i = PIL.Image.frombytes(mode='RGB', size=(CHUNK_SIZE_PIX, CHUNK_SIZE_PIX), data=matrix) i = PIL.Image.frombytes(mode='RGB', size=(CHUNK_SIZE_PIX, CHUNK_SIZE_PIX), data=matrix)
return i return i
def chunk_to_image(chunk_data): def chunk_to_image(chunk_data, scale=1):
return rgb_to_image(chunk_to_rgb(chunk_data)) image = rgb_to_image(chunk_to_rgb(chunk_data))
if scale is not None and scale != 1:
new_size = (int(image.size[0] * scale), int(image.size[1] * scale))
image = image.resize(new_size, resample=PIL.Image.ANTIALIAS)
return image
def chunks_to_image(chunks): def chunks_to_image(chunks, scale=1):
''' '''
Combine all of the given chunks into a single image. Combine all of the given chunks into a single image.
''' '''
@ -357,13 +366,13 @@ def chunks_to_image(chunks):
max_y = max(chunk[1] for chunk in chunks) max_y = max(chunk[1] for chunk in chunks)
span_x = max_x - min_x + 1 span_x = max_x - min_x + 1
span_y = max_y - min_y + 1 span_y = max_y - min_y + 1
img_width = span_x * CHUNK_SIZE_PIX img_width = int(span_x * CHUNK_SIZE_PIX * scale)
img_height = span_y * CHUNK_SIZE_PIX img_height = int(span_y * CHUNK_SIZE_PIX * scale)
img = PIL.Image.new(mode='RGB', size=(img_width, img_height), color=WHITE) img = PIL.Image.new(mode='RGB', size=(img_width, img_height), color=WHITE)
for (chunk_x, chunk_y, chunk_data) in chunks: for (chunk_x, chunk_y, chunk_data) in chunks:
paste_x = (chunk_x - min_x) * CHUNK_SIZE_PIX paste_x = int((chunk_x - min_x) * CHUNK_SIZE_PIX * scale)
paste_y = (chunk_y - min_y) * CHUNK_SIZE_PIX paste_y = int((chunk_y - min_y) * CHUNK_SIZE_PIX * scale)
chunk_image = chunk_to_image(chunk_data) chunk_image = chunk_to_image(chunk_data, scale)
img.paste(chunk_image, (paste_x, paste_y)) img.paste(chunk_image, (paste_x, paste_y))
return img return img
@ -392,7 +401,7 @@ def crop_image(image, pixel_xy1, pixel_xy2):
#################################################################################################### ####################################################################################################
def parse_coordinates(coordinates): def parse_coordinate_string(coordinates):
''' '''
Convert the given '~100.~100--100.100' to ((-100, -100), (100, 100)). Convert the given '~100.~100--100.100' to ((-100, -100), (100, 100)).
''' '''
@ -416,20 +425,26 @@ def parse_coordinates(coordinates):
def render_argparse(args): def render_argparse(args):
if args.do_update: if args.do_update:
update_argparse(args) update_argparse(args)
coordinates = parse_coordinates(args.coordinates) coordinates = parse_coordinate_string(args.coordinates)
chunk_range = pixel_range_to_chunk_range(*coordinates) chunk_range = pixel_range_to_chunk_range(*coordinates)
chunks = [get_chunk(*chunk_xy) for chunk_xy in chunk_range_iterator(*chunk_range)] chunks = [get_chunk(*chunk_xy) for chunk_xy in chunk_range_iterator(*chunk_range)]
image = chunks_to_image(chunks) scale = float(args.scale)
image = crop_image(image, *coordinates) image = chunks_to_image(chunks, scale=scale)
if scale == 1:
image = crop_image(image, *coordinates)
if args.do_show: if args.do_show:
image.show() image.show()
else: else:
filename = f'{coordinates[0][0]}.{coordinates[0][1]}--{coordinates[1][0]}.{coordinates[1][1]}.png' (p1, p2) = coordinates
(p1x, p1y) = p1
(p2x, p2y) = p2
scale_s = f'_{scale}' if scale != 1 else ''
filename = f'{p1x}.{p1y}--{p2x}.{p2y}{scale_s}.png'
image.save(filename) image.save(filename)
log.debug('Wrote %s', filename) log.debug('Wrote %s', filename)
def update_argparse(args): def update_argparse(args):
coordinates = parse_coordinates(args.coordinates) coordinates = parse_coordinate_string(args.coordinates)
bigchunk_range = pixel_range_to_bigchunk_range(*coordinates) bigchunk_range = pixel_range_to_bigchunk_range(*coordinates)
download_bigchunk_range(*bigchunk_range) download_bigchunk_range(*bigchunk_range)
@ -445,6 +460,7 @@ p_render = subparsers.add_parser('render')
p_render.add_argument('coordinates') p_render.add_argument('coordinates')
p_render.add_argument('--update', dest='do_update', action='store_true') p_render.add_argument('--update', dest='do_update', action='store_true')
p_render.add_argument('--show', dest='do_show', action='store_true') p_render.add_argument('--show', dest='do_show', action='store_true')
p_render.add_argument('--scale', dest='scale', default=1)
p_render.set_defaults(func=render_argparse) p_render.set_defaults(func=render_argparse)
def main(argv): def main(argv):