steffo/flyingsnake
steffo/flyingsnake
1
Fork 0
mirror of https://github.com/Steffo99/flyingsnake.git synced 2024-12-22 14:44:17 +00:00
Code Activity

Merge branch 'fs-1' of https://github.com/polandar/flyingsnake into polandar-fs-1

Browse source
This commit is contained in:
Steffo 2019-10-02 10:56:06 +02:00
commit 35a9d8310d
2 changed files with 83 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.gitignore vendored
View file

@ -5,3 +5,5 @@ dist/
**/__pycache__/ **/__pycache__/
setup/ setup/
build/ build/
venv/
.venv/

119
flyingsnake/__main__.py
View file

@ -4,6 +4,17 @@ import json
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
from .default_colors import DEFAULT_COLORS from .default_colors import DEFAULT_COLORS
DEFAULT_REGION_VALUE = 1_000_000
DEFAULT_COORDS = -1
def get_region_size(*, world, min_x, min_y, region_width, region_height):
min_x = max(0, min_x or 0)
min_y = max(0, min_y or 0)
max_x = min(world.size.x, (min_x or world.size.x) + region_width)
max_y = min(world.size.y, (min_y or world.size.y) + region_height)
return min_x, min_y, max_x, max_y
@c.command() @c.command()
@c.argument("input_file", type=c.Path(exists=True)) @c.argument("input_file", type=c.Path(exists=True))
@ -22,6 +33,14 @@ from .default_colors import DEFAULT_COLORS
help="Draw the liquids present in the world.", default=None) help="Draw the liquids present in the world.", default=None)
@c.option("--paint/--no-paint", "draw_paint", @c.option("--paint/--no-paint", "draw_paint",
help="Draw painted blocks with the paint color overlayed on them.", default=True) help="Draw painted blocks with the paint color overlayed on them.", default=True)
@c.option("-x", "--x_coord", "min_x",
help="Min x coord for custom world rendering", default=None)
@c.option("-y", "--y_coord", "min_y",
help="Min y coord for custom world rendering", default=None)
@c.option("-w", "--region_width", "region_width",
help="Width of region for custom world rendering", default=None)
@c.option("-h", "--region_height", "region_height",
help="Height of region for custom world rendering", default=None
def flyingsnake(input_file: str, def flyingsnake(input_file: str,
output_file: str, output_file: str,
colors_file: str, colors_file: str,
@ -30,7 +49,11 @@ def flyingsnake(input_file: str,
draw_walls: bool, draw_walls: bool,
draw_liquids: bool, draw_liquids: bool,
draw_wires: bool, draw_wires: bool,
draw_paint: bool): draw_paint: bool,
min_x: int,
min_y: int,
region_width: int,
region_height: int):
# If at least a draw flag is set to True, default everything else to False # If at least a draw flag is set to True, default everything else to False
if draw_background is True \ if draw_background is True \
or draw_blocks is True \ or draw_blocks is True \
@ -98,98 +121,118 @@ def flyingsnake(input_file: str,
c.echo("Parsing world...") c.echo("Parsing world...")
world = li.World.create_from_file(input_file) world = li.World.create_from_file(input_file)
min_x, min_y, max_x, max_y = get_region_size(world=world, min_x=min_x, min_y=min_y,
region_width=region_width, region_height=region_height)
c.echo(f"Rendering world coordinates between ({min_x}, {min_y}) to ({max_x}, {max_y}")
width = max_x - min_x
height = max_y - min_y
if draw_background: if draw_background:
c.echo("Drawing the background...") c.echo("Drawing the background...")
background = Image.new("RGBA", (world.size.x, world.size.y)) background = Image.new("RGBA", (width, height))
draw = ImageDraw.Draw(background) draw = ImageDraw.Draw(background)
draw.rectangle(((0, 0), (world.size.x, world.underground_level)), tuple(colors["Globals"]["Sky"])) curr_y = 0
draw.rectangle(((0, world.underground_level + 1), (world.size.x, world.cavern_level)), if min_y <= world.underground_level:
tuple(colors["Globals"]["Earth"])) sky_y = min(world.underground_level - min_y, height)
draw.rectangle(((0, world.cavern_level + 1), (world.size.x, world.size.y - 192)), draw.rectangle(((0, curr_y), (width, sky_y)), tuple(colors["Globals"]["Sky"]))
tuple(colors["Globals"]["Rock"])) curr_y = sky_y + 1
draw.rectangle(((0, world.size.y - 191), (world.size.x, world.size.y)), tuple(colors["Globals"]["Hell"])) if max_y > world.underground_level and min_y <= world.cavern_level:
earth_y = min(world.cavern_level - min_y, height)
draw.rectangle(((0, curr_y), (width, earth_y)), tuple(colors["Globals"]["Earth"]))
curr_y = earth_y + 1
edge_of_rock = world.size_y - 192
if max_y > world.cavern_level and min_y <= edge_of_rock:
rock_y = min(edge_of_rock - min_y, height)
draw.rectangle(((0, curr_y), (width, edge_of_rock + 1)), tuple(colors["Globals"]["Rock"]))
curr_y = rock_y + 1
if max_y > edge_of_rock:
draw.rectangle(((0, curr_y), (width, height)), tuple(colors["Globals"]["Hell"]))
del draw del draw
to_merge.append(background) to_merge.append(background)
if draw_walls: if draw_walls:
c.echo("Drawing walls...") c.echo("Drawing walls...")
walls = Image.new("RGBA", (world.size.x, world.size.y)) walls = Image.new("RGBA", (width, height))
draw = ImageDraw.Draw(walls) draw = ImageDraw.Draw(walls)
for x in range(world.size.x): for x in range(min_x, max_x):
for y in range(world.size.y): for y in range(min_y, max_y):
tile = world.tiles[x, y] tile = world.tiles[x, y]
if tile.wall: if tile.wall:
if draw_paint and tile.wall.paint: if draw_paint and tile.wall.paint:
color = tuple(colors["Paints"][str(tile.wall.paint)]) color = tuple(colors["Paints"][str(tile.wall.paint)])
else: else:
color = tuple(colors["Walls"][str(tile.wall.type.value)]) color = tuple(colors["Walls"][str(tile.wall.type.value)])
draw.point((x, y), color) draw.point((x - min_x, y - min_y), color)
if not x % 100: if not x % 100:
c.echo(f"{x} / {world.size.x} rows done") c.echo(f"{x} / {width} rows done")
del draw del draw
to_merge.append(walls) to_merge.append(walls)
if draw_liquids: if draw_liquids:
c.echo("Drawing liquids...") c.echo("Drawing liquids...")
liquids = Image.new("RGBA", (world.size.x, world.size.y)) liquids = Image.new("RGBA", (width, height))
draw = ImageDraw.Draw(liquids) draw = ImageDraw.Draw(liquids)
for x in range(world.size.x): for x in range(min_x, max_x):
for y in range(world.size.y): for y in range(min_y, max_y):
tile = world.tiles[x, y] tile = world.tiles[x, y]
if tile.liquid: if tile.liquid:
if tile.liquid.type == li.tiles.LiquidType.WATER: if tile.liquid.type == li.tiles.LiquidType.WATER:
draw.point((x, y), tuple(colors["Globals"]["Water"])) color = tuple(colors["Globals"]["Water"])
elif tile.liquid.type == li.tiles.LiquidType.LAVA: elif tile.liquid.type == li.tiles.LiquidType.LAVA:
draw.point((x, y), tuple(colors["Globals"]["Lava"])) color = tuple(colors["Globals"]["Lava"])
elif tile.liquid.type == li.tiles.LiquidType.HONEY: elif tile.liquid.type == li.tiles.LiquidType.HONEY:
draw.point((x, y), tuple(colors["Globals"]["Honey"])) color = tuple(colors["Globals"]["Honey"])
else:
continue
draw.point((x - min_x, y - min_y), color)
if not x % 100: if not x % 100:
c.echo(f"{x} / {world.size.x} rows done") c.echo(f"{x} / {width} rows done")
del draw del draw
to_merge.append(liquids) to_merge.append(liquids)
if draw_blocks: if draw_blocks:
c.echo("Drawing blocks...") c.echo("Drawing blocks...")
blocks = Image.new("RGBA", (world.size.x, world.size.y)) blocks = Image.new("RGBA", (width, height))
draw = ImageDraw.Draw(blocks) draw = ImageDraw.Draw(blocks)
for x in range(world.size.x): for x in range(min_x, max_x):
for y in range(world.size.y): for y in range(min_y, max_y):
tile = world.tiles[x, y] tile = world.tiles[x, y]
if tile.block: if tile.block:
if draw_paint and tile.block.paint: if draw_paint and tile.block.paint:
color = tuple(colors["Paints"][str(tile.block.paint)]) color = tuple(colors["Paints"][str(tile.block.paint)])
else: else:
color = tuple(colors["Blocks"][str(tile.block.type.value)]) color = tuple(colors["Blocks"][str(tile.block.type.value)])
draw.point((x, y), color) draw.point((x - min_x, y - min_y), color)
if not x % 100: if not x % 100:
c.echo(f"{x} / {world.size.x} rows done") c.echo(f"{x} / {width} rows done")
del draw del draw
to_merge.append(blocks) to_merge.append(blocks)
if draw_wires: if draw_wires:
c.echo("Drawing wires...") c.echo("Drawing wires...")
wires = Image.new("RGBA", (world.size.x, world.size.y)) wires = Image.new("RGBA", (width, height))
draw = ImageDraw.Draw(wires) draw = ImageDraw.Draw(wires)
for x in range(world.size.x): for x in range(min_x, max_x):
for y in range(world.size.y): for y in range(min_y, max_y):
tile = world.tiles[x, y] tile = world.tiles[x, y]
if tile.wiring: if tile.wiring:
if tile.wiring.red: if tile.wiring.red:
draw.point((x, y), tuple(colors["Globals"]["Wire"])) color = tuple(colors["Globals"]["Wire"])
if tile.wiring.blue: elif tile.wiring.blue:
draw.point((x, y), tuple(colors["Globals"]["Wire1"])) color = tuple(colors["Globals"]["Wire1"])
if tile.wiring.green: elif tile.wiring.green:
draw.point((x, y), tuple(colors["Globals"]["Wire2"])) color = tuple(colors["Globals"]["Wire2"])
if tile.wiring.yellow: elif tile.wiring.yellow:
draw.point((x, y), tuple(colors["Globals"]["Wire3"])) color = tuple(colors["Globals"]["Wire3"])
else:
continue
draw.point((x - min_x, y - min_y), color)
if not x % 100: if not x % 100:
c.echo(f"{x} / {world.size.x} rows done") c.echo(f"{x} / {width} rows done")
del draw del draw
to_merge.append(wires) to_merge.append(wires)
c.echo("Merging layers...") c.echo("Merging layers...")
final = Image.new("RGBA", (world.size.x, world.size.y)) final = Image.new("RGBA", (width, height))
while to_merge: while to_merge:
final = Image.alpha_composite(final, to_merge.pop(0)) final = Image.alpha_composite(final, to_merge.pop(0))