steffo/flyingsnake
steffo/flyingsnake
1
Fork 0
mirror of https://github.com/Steffo99/flyingsnake.git synced 2024-12-22 06:34:18 +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__/
setup/
build/
venv/
.venv/

119
flyingsnake/__main__.py
View file

@ -4,6 +4,17 @@ import json
from PIL import Image, ImageDraw
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.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)
@c.option("--paint/--no-paint", "draw_paint",
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,
output_file: str,
colors_file: str,
@ -30,7 +49,11 @@ def flyingsnake(input_file: str,
draw_walls: bool,
draw_liquids: 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 draw_background is True \
or draw_blocks is True \
@ -98,98 +121,118 @@ def flyingsnake(input_file: str,
c.echo("Parsing world...")
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:
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.rectangle(((0, 0), (world.size.x, world.underground_level)), tuple(colors["Globals"]["Sky"]))
draw.rectangle(((0, world.underground_level + 1), (world.size.x, world.cavern_level)),
tuple(colors["Globals"]["Earth"]))
draw.rectangle(((0, world.cavern_level + 1), (world.size.x, world.size.y - 192)),
tuple(colors["Globals"]["Rock"]))
draw.rectangle(((0, world.size.y - 191), (world.size.x, world.size.y)), tuple(colors["Globals"]["Hell"]))
curr_y = 0
if min_y <= world.underground_level:
sky_y = min(world.underground_level - min_y, height)
draw.rectangle(((0, curr_y), (width, sky_y)), tuple(colors["Globals"]["Sky"]))
curr_y = sky_y + 1
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
to_merge.append(background)
if draw_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)
for x in range(world.size.x):
for y in range(world.size.y):
for x in range(min_x, max_x):
for y in range(min_y, max_y):
tile = world.tiles[x, y]
if tile.wall:
if draw_paint and tile.wall.paint:
color = tuple(colors["Paints"][str(tile.wall.paint)])
else:
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:
c.echo(f"{x} / {world.size.x} rows done")
c.echo(f"{x} / {width} rows done")
del draw
to_merge.append(walls)
if draw_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)
for x in range(world.size.x):
for y in range(world.size.y):
for x in range(min_x, max_x):
for y in range(min_y, max_y):
tile = world.tiles[x, y]
if tile.liquid:
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:
draw.point((x, y), tuple(colors["Globals"]["Lava"]))
color = tuple(colors["Globals"]["Lava"])
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:
c.echo(f"{x} / {world.size.x} rows done")
c.echo(f"{x} / {width} rows done")
del draw
to_merge.append(liquids)
if draw_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)
for x in range(world.size.x):
for y in range(world.size.y):
for x in range(min_x, max_x):
for y in range(min_y, max_y):
tile = world.tiles[x, y]
if tile.block:
if draw_paint and tile.block.paint:
color = tuple(colors["Paints"][str(tile.block.paint)])
else:
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:
c.echo(f"{x} / {world.size.x} rows done")
c.echo(f"{x} / {width} rows done")
del draw
to_merge.append(blocks)
if draw_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)
for x in range(world.size.x):
for y in range(world.size.y):
for x in range(min_x, max_x):
for y in range(min_y, max_y):
tile = world.tiles[x, y]
if tile.wiring:
if tile.wiring.red:
draw.point((x, y), tuple(colors["Globals"]["Wire"]))
if tile.wiring.blue:
draw.point((x, y), tuple(colors["Globals"]["Wire1"]))
if tile.wiring.green:
draw.point((x, y), tuple(colors["Globals"]["Wire2"]))
if tile.wiring.yellow:
draw.point((x, y), tuple(colors["Globals"]["Wire3"]))
color = tuple(colors["Globals"]["Wire"])
elif tile.wiring.blue:
color = tuple(colors["Globals"]["Wire1"])
elif tile.wiring.green:
color = tuple(colors["Globals"]["Wire2"])
elif tile.wiring.yellow:
color = tuple(colors["Globals"]["Wire3"])
else:
continue
draw.point((x - min_x, y - min_y), color)
if not x % 100:
c.echo(f"{x} / {world.size.x} rows done")
c.echo(f"{x} / {width} rows done")
del draw
to_merge.append(wires)
c.echo("Merging layers...")
final = Image.new("RGBA", (world.size.x, world.size.y))
final = Image.new("RGBA", (width, height))
while to_merge:
final = Image.alpha_composite(final, to_merge.pop(0))