214 lines
7.2 KiB
Python
214 lines
7.2 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Optional, Union
|
|
import asyncio
|
|
from httpx import AsyncClient
|
|
from asyncio.queues import Queue
|
|
from rich.progress import TaskID, Progress
|
|
from zipfile import ZipFile
|
|
from PIL import Image
|
|
|
|
INPUT_PATH = Path("input/")
|
|
TEXTURES_CACHE = INPUT_PATH / "textures_cache"
|
|
GLASS_CACHE_SEAMLESS = INPUT_PATH / "glass_seamless"
|
|
OUTPUT_PATH = Path("output/")
|
|
GLASS_OUTPUT_PATH = (
|
|
OUTPUT_PATH / "assets/minecraft/optifine/ctm/greenhouse/glass"
|
|
)
|
|
|
|
TEXTURES_PATH: str = "assets/minecraft/textures"
|
|
SEAM_TEXTURE_NAME: str = "block.dark_oak_planks"
|
|
GLASS_COLORS: list[str] = [
|
|
"black",
|
|
"blue",
|
|
"brown",
|
|
"cyan",
|
|
"gray",
|
|
"green",
|
|
"light_blue",
|
|
"light_gray",
|
|
"lime",
|
|
"magenta",
|
|
"orange",
|
|
"pink",
|
|
"purple",
|
|
"red",
|
|
"white",
|
|
"yellow",
|
|
]
|
|
|
|
meta_url: str = (
|
|
"https://piston-meta.mojang.com/v1/packages/715ccf3330885e75b205124f09f8712542cbe7e0/1.20.1.json"
|
|
)
|
|
meta_filename = INPUT_PATH / f"meta-{meta_url.split('/')[-1]}"
|
|
|
|
|
|
class AsyncDownloader:
|
|
def __init__(
|
|
self, files: Optional[list[tuple[str, Path]]] = None, maxfiles: int = 8
|
|
):
|
|
self._queue: Queue[tuple[str, Path]] = Queue(maxfiles)
|
|
self._client = AsyncClient()
|
|
self._progress: Optional[Progress] = None
|
|
self._tasks: list[TaskID] = []
|
|
|
|
for url, path in files or []:
|
|
self._queue.put_nowait((url, path))
|
|
|
|
def add_file(self, url: str, path: Union[Path, str]):
|
|
self._queue.put_nowait((url, Path(path)))
|
|
|
|
async def __aenter__(self):
|
|
await self._client.__aenter__()
|
|
self._progress = Progress()
|
|
self._progress.__enter__()
|
|
for i in range(self._queue.maxsize):
|
|
self._tasks.append(
|
|
self._progress.add_task(f"Downloader {i + 1}", start=False)
|
|
)
|
|
return self
|
|
|
|
async def _worker(self, task: TaskID):
|
|
if not self._progress:
|
|
return
|
|
while not self._queue.empty():
|
|
url, path = await self._queue.get()
|
|
with path.open("wb") as fout:
|
|
async with self._client.stream("GET", url) as req:
|
|
size = int(req.headers.get("Content-Length", 0))
|
|
self._progress.start_task(task)
|
|
self._progress.update(
|
|
task,
|
|
total=size,
|
|
completed=0,
|
|
description=f"{path.name}",
|
|
)
|
|
async for chunk in req.aiter_bytes(8192):
|
|
self._progress.advance(task, fout.write(chunk))
|
|
self._progress.stop_task(task)
|
|
|
|
async def run(self):
|
|
await asyncio.gather(*[self._worker(task) for task in self._tasks])
|
|
|
|
async def __aexit__(self, a, b, c):
|
|
if self._progress is None:
|
|
raise ValueError("how did that happen?")
|
|
self._progress.__exit__(a, b, c)
|
|
await self._client.__aexit__()
|
|
|
|
|
|
async def download_if_missing(url: str, path: Union[Path, str]):
|
|
path = Path(path)
|
|
if path.exists():
|
|
return
|
|
|
|
with Progress() as progress:
|
|
task = progress.add_task(f"Downloading {path}")
|
|
with Path(path).open("wb") as fout:
|
|
async with AsyncClient() as client:
|
|
async with client.stream("GET", url) as req:
|
|
progress.update(
|
|
task,
|
|
total=int(req.headers.get("Content-Length", 0)),
|
|
completed=0,
|
|
)
|
|
async for chunk in req.aiter_bytes(8192):
|
|
progress.advance(task, fout.write(chunk))
|
|
|
|
|
|
async def main():
|
|
INPUT_PATH.mkdir(exist_ok=True)
|
|
TEXTURES_CACHE.mkdir(exist_ok=True)
|
|
GLASS_CACHE_SEAMLESS.mkdir(exist_ok=True)
|
|
OUTPUT_PATH.mkdir(exist_ok=True)
|
|
|
|
GLASS_OUTPUT_PATH.mkdir(exist_ok=True, parents=True)
|
|
|
|
print("[ + ] Getting client info")
|
|
await download_if_missing(meta_url, meta_filename)
|
|
|
|
with open(meta_filename, "r") as fp:
|
|
meta: dict = json.load(fp)
|
|
|
|
print("[ + ] Getting client JAR")
|
|
client_filename = INPUT_PATH / f"client-{meta['id']}.jar"
|
|
await download_if_missing(
|
|
meta["downloads"]["client"]["url"], client_filename
|
|
)
|
|
|
|
# "assets/minecraft/textures/block/orange_stained_glass.png"
|
|
with ZipFile(client_filename) as zipf:
|
|
|
|
def extract_texture(name: str) -> Path:
|
|
path = f"{TEXTURES_PATH}/{name.replace('.', '/')}.png"
|
|
out_path = TEXTURES_CACHE / f"{name}.png"
|
|
if out_path.exists():
|
|
print("[---] Extracted", name, "already")
|
|
return out_path
|
|
|
|
print("[...] Extracting", name)
|
|
with zipf.open(path, "r") as fp_in:
|
|
with out_path.open("wb") as fp_out:
|
|
while chunk := fp_in.read():
|
|
fp_out.write(chunk)
|
|
return out_path
|
|
|
|
extract_texture(SEAM_TEXTURE_NAME)
|
|
for color in GLASS_COLORS:
|
|
extract_texture(f"block.{color}_stained_glass")
|
|
|
|
print("[ + ] Removing seam on glass textures")
|
|
for color in GLASS_COLORS:
|
|
in_path = TEXTURES_CACHE / f"block.{color}_stained_glass.png"
|
|
out_file = GLASS_CACHE_SEAMLESS / f"{color}.png"
|
|
if out_file.exists():
|
|
print("[---] SKIP", out_file.name)
|
|
continue
|
|
|
|
print("[...]", out_file.name)
|
|
with Image.open(in_path).convert("RGBA") as im:
|
|
im.paste(im.crop((1, 1, 2, 15)), (0, 0))
|
|
im.paste(im.crop((1, 1, 2, 15)), (0, 2))
|
|
im.paste(im.crop((1, 1, 2, 15)), (15, 0))
|
|
im.paste(im.crop((1, 1, 2, 15)), (15, 2))
|
|
im.paste(im.crop((1, 1, 15, 2)), (1, 0))
|
|
im.paste(im.crop((1, 1, 15, 2)), (1, 15))
|
|
im.save(out_file)
|
|
|
|
print("[ + ] Loading connected textures masks")
|
|
ctm_list: list[Image.Image] = []
|
|
with Image.open("ctm.png") as im:
|
|
for i in range(47):
|
|
ox, oy = (i % 8) * 16, (i // 8) * 16
|
|
ctm_list.append(im.crop((ox, oy, ox + 16, oy + 16)).convert("1"))
|
|
|
|
border_texture = Image.open(TEXTURES_CACHE / f"{SEAM_TEXTURE_NAME}.png")
|
|
|
|
print("[ + ] Creating connected textures")
|
|
for color in GLASS_COLORS:
|
|
out_path = GLASS_OUTPUT_PATH / color
|
|
out_path.mkdir(exist_ok=True)
|
|
with (out_path / "glass.properties").open("w") as fp:
|
|
texture_name = f"{color}_greenhouse_glass"
|
|
fp.write("method=ctm\n")
|
|
fp.write(f"matchBlocks=seasonextras:{texture_name}\n")
|
|
fp.write(f"tiles=0-46\n")
|
|
fp.write(f"connect=block\n")
|
|
fp.write(f"resourceCondition=textures/block/{texture_name}.png\n")
|
|
with Image.open(GLASS_CACHE_SEAMLESS / f"{color}.png") as glass:
|
|
for i in range(47):
|
|
ctm = Image.composite(border_texture, glass, ctm_list[i])
|
|
ctm.save(out_path / f"{i}.png")
|
|
with (OUTPUT_PATH / "pack.mcmeta").open("w") as fp:
|
|
json.dump({
|
|
"pack": {
|
|
"pack_format": 15,
|
|
"description": "CTM support for Fabric Seasons Extras. Also, nicer textures in general"
|
|
}
|
|
}, fp, indent=2, ensure_ascii=False)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|