1
Fork 0
mirror of https://github.com/Steffo99/flyingsnake.git synced 2024-12-23 06:54:19 +00:00

Update __main__.py

This commit is contained in:
Garvita Jain 2019-09-28 23:35:12 +05:30 committed by GitHub
parent 9d0b231f08
commit 51f31013dc
Signed by: github
GPG key ID: 4AEE18F83AFDEB23

View file

@ -116,17 +116,17 @@ def flyingsnake(input_file: str,
c.echo("Drawing walls...") c.echo("Drawing walls...")
walls = Image.new("RGBA", (world.size.x, world.size.y)) walls = Image.new("RGBA", (world.size.x, world.size.y))
draw = ImageDraw.Draw(walls) draw = ImageDraw.Draw(walls)
with click.progressbar(world.tiles, length = world.size) as bar:
for x in range(world.size.x): for x in range(world.size.x):
for y in range(world.size.y): for y in range(world.size.y):
tile = world.tiles[x, y] tile = bar[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, y), color)
if not x % 100:
c.echo(f"{x} / {world.size.x} rows done")
del draw del draw
to_merge.append(walls) to_merge.append(walls)
@ -134,9 +134,10 @@ def flyingsnake(input_file: str,
c.echo("Drawing liquids...") c.echo("Drawing liquids...")
liquids = Image.new("RGBA", (world.size.x, world.size.y)) liquids = Image.new("RGBA", (world.size.x, world.size.y))
draw = ImageDraw.Draw(liquids) draw = ImageDraw.Draw(liquids)
with click.progressbar(world.tiles, length = world.size) as bar:
for x in range(world.size.x): for x in range(world.size.x):
for y in range(world.size.y): for y in range(world.size.y):
tile = world.tiles[x, y] tile = bar[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"])) draw.point((x, y), tuple(colors["Globals"]["Water"]))
@ -144,8 +145,7 @@ def flyingsnake(input_file: str,
draw.point((x, y), tuple(colors["Globals"]["Lava"])) draw.point((x, y), 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"])) draw.point((x, y), tuple(colors["Globals"]["Honey"]))
if not x % 100:
c.echo(f"{x} / {world.size.x} rows done")
del draw del draw
to_merge.append(liquids) to_merge.append(liquids)
@ -153,17 +153,17 @@ def flyingsnake(input_file: str,
c.echo("Drawing blocks...") c.echo("Drawing blocks...")
blocks = Image.new("RGBA", (world.size.x, world.size.y)) blocks = Image.new("RGBA", (world.size.x, world.size.y))
draw = ImageDraw.Draw(blocks) draw = ImageDraw.Draw(blocks)
with click.progressbar(world.tiles, length = world.size) as bar:
for x in range(world.size.x): for x in range(world.size.x):
for y in range(world.size.y): for y in range(world.size.y):
tile = world.tiles[x, y] tile = bar[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, y), color)
if not x % 100:
c.echo(f"{x} / {world.size.x} rows done")
del draw del draw
to_merge.append(blocks) to_merge.append(blocks)
@ -171,9 +171,10 @@ def flyingsnake(input_file: str,
c.echo("Drawing wires...") c.echo("Drawing wires...")
wires = Image.new("RGBA", (world.size.x, world.size.y)) wires = Image.new("RGBA", (world.size.x, world.size.y))
draw = ImageDraw.Draw(wires) draw = ImageDraw.Draw(wires)
with c.progressbar(world.tiles, length = world.size) as bar:
for x in range(world.size.x): for x in range(world.size.x):
for y in range(world.size.y): for y in range(world.size.y):
tile = world.tiles[x, y] tile = bar[x, y]
if tile.wiring: if tile.wiring:
if tile.wiring.red: if tile.wiring.red:
draw.point((x, y), tuple(colors["Globals"]["Wire"])) draw.point((x, y), tuple(colors["Globals"]["Wire"]))
@ -183,8 +184,7 @@ def flyingsnake(input_file: str,
draw.point((x, y), tuple(colors["Globals"]["Wire2"])) draw.point((x, y), tuple(colors["Globals"]["Wire2"]))
if tile.wiring.yellow: if tile.wiring.yellow:
draw.point((x, y), tuple(colors["Globals"]["Wire3"])) draw.point((x, y), tuple(colors["Globals"]["Wire3"]))
if not x % 100:
c.echo(f"{x} / {world.size.x} rows done")
del draw del draw
to_merge.append(wires) to_merge.append(wires)