feat: pool set_mask + route
Write a per-slot grayscale mask sidecar (img_XXXX.mask.png) and record it on the slot. Add the multipart /grid_pool/set_mask route. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -112,6 +112,19 @@ def set_label(base_dir, pool_id, index, label):
|
||||
return m
|
||||
|
||||
|
||||
def set_mask(base_dir, pool_id, index, mask_bytes):
|
||||
m = read_manifest(base_dir, pool_id)
|
||||
if not (0 <= index < len(m["slots"])):
|
||||
return m
|
||||
img_name = m["slots"][index]["image"]
|
||||
mask_name = img_name.replace(".png", ".mask.png")
|
||||
with open(pool_dir(base_dir, pool_id) / mask_name, "wb") as f:
|
||||
f.write(mask_bytes)
|
||||
m["slots"][index]["mask"] = mask_name
|
||||
write_manifest(base_dir, pool_id, m)
|
||||
return m
|
||||
|
||||
|
||||
def rebuild_manifest(base_dir, pool_id):
|
||||
d = pool_dir(base_dir, pool_id)
|
||||
m = empty_manifest()
|
||||
|
||||
@@ -45,6 +45,21 @@ async def _label(request):
|
||||
return web.json_response(handlers.handle_label(_base(), body["pool_id"], int(body["index"]), body["label"]))
|
||||
|
||||
|
||||
@routes.post("/grid_pool/set_mask")
|
||||
async def _set_mask(request):
|
||||
reader = await request.multipart()
|
||||
pool_id, index, data = "default", 0, None
|
||||
async for part in reader:
|
||||
if part.name == "pool_id":
|
||||
pool_id = (await part.text())
|
||||
elif part.name == "index":
|
||||
index = int(await part.text())
|
||||
elif part.name == "mask":
|
||||
data = await part.read(decode=False)
|
||||
m = handlers.handle_set_mask(_base(), pool_id, index, data)
|
||||
return web.json_response(m)
|
||||
|
||||
|
||||
@routes.get("/grid_pool/list")
|
||||
async def _list(request):
|
||||
pool_id = request.query.get("pool_id", "default")
|
||||
|
||||
Reference in New Issue
Block a user