91 lines
7.4 KiB
Python
91 lines
7.4 KiB
Python
"""10521 water-only Source-shaped fine RGB material; never a generator return."""
|
|
from pathlib import Path
|
|
import argparse,hashlib,io,json,os,stat,struct
|
|
import numpy as np
|
|
from PIL import Image
|
|
METHOD='10521-Source-water-only-quiet-genuine-RGB-fineband-preserve-nonwater-v1'
|
|
PARAMS=dict(source_sigma=1.0,donor_band_sigma=3.0,donor_band_clip=16.0,donor_band_gain=0.5)
|
|
DEPENDENCY='3284cf6f85ced107dfc364ea90e15aed341d478bea7e308024b5bc5218b03092'
|
|
MAX_INPUT=16*1024*1024
|
|
|
|
def read_contained(root,name,limit=MAX_INPUT):
|
|
if not isinstance(name,str)or name.startswith('/')or any(x in('','.','..')for x in name.split('/')):raise ValueError('contained canonical relative route required')
|
|
root=Path(root).absolute()
|
|
if'..'in root.parts:raise ValueError('canonical root required')
|
|
flags=os.O_RDONLY|os.O_CLOEXEC|os.O_NOFOLLOW|os.O_NONBLOCK;fd=os.open('/',flags|os.O_DIRECTORY)
|
|
try:
|
|
for part in root.parts[1:]+tuple(name.split('/')[:-1]):n=os.open(part,flags|os.O_DIRECTORY,dir_fd=fd);os.close(fd);fd=n
|
|
source=os.open(name.split('/')[-1],flags,dir_fd=fd)
|
|
try:
|
|
before=os.fstat(source)
|
|
if not stat.S_ISREG(before.st_mode)or not 0<=before.st_size<=limit:raise ValueError('bounded regular input required')
|
|
chunks=[];size=0
|
|
while True:
|
|
chunk=os.read(source,min(65536,limit+1-size))
|
|
if not chunk:break
|
|
chunks.append(chunk);size+=len(chunk)
|
|
if size>limit:raise ValueError('bounded EOF required')
|
|
after=os.fstat(source);key=lambda v:(v.st_dev,v.st_ino,v.st_size,v.st_mtime_ns,v.st_mode)
|
|
if key(before)!=key(after)or size!=before.st_size:raise ValueError('input changed during read')
|
|
return b''.join(chunks)
|
|
finally:os.close(source)
|
|
finally:os.close(fd)
|
|
|
|
ROOT=next((p for p in Path(__file__).resolve().parents if(p/'tools/hd_terrain_cleanup.py').is_file()),None)
|
|
if ROOT is None:raise RuntimeError('game root containing reviewed math required')
|
|
b=read_contained(ROOT,'tools/hd_terrain_cleanup.py',1024*1024)
|
|
if hashlib.sha256(b).hexdigest()!=DEPENDENCY:raise RuntimeError('reviewed math dependency changed')
|
|
ns={'__name__':'reviewed_water_math','__file__':str(ROOT/'tools/hd_terrain_cleanup.py')};exec(compile(b,ns['__file__'],'exec'),ns)
|
|
scale_nx,gaussian,resize_float,palette_rgb=(ns[n]for n in('scale_nx','gaussian','resize_float','palette_rgb'))
|
|
|
|
def decode_png(payload):
|
|
with Image.open(io.BytesIO(payload))as im:
|
|
if im.format!='PNG'or im.width*im.height>4*1024*1024 or getattr(im,'n_frames',1)!=1 or im.mode not in('RGB','RGBA'):raise ValueError('bounded single opaque PNG required')
|
|
if im.mode=='RGBA'and im.getchannel('A').getextrema()!=(255,255):raise ValueError('opaque PNG required')
|
|
return np.array(im.convert('RGB'))
|
|
|
|
def derive(indices,palette,old,donor):
|
|
if indices.shape!=(32,32)or indices.dtype!=np.uint8 or old.shape!=(724,724,3)or old.dtype!=np.uint8:raise ValueError('exact Source32 and724RGB required')
|
|
if palette.shape!=(256,3)or palette.dtype!=np.uint8 or donor.shape!=(887,887,3)or donor.dtype!=np.uint8:raise ValueError('exact palette and frozen water donor field required')
|
|
water=((indices>=149)&(indices<=159))|((indices>=240)&(indices<=244))
|
|
if not water[:16].all()or water[16:].any()or (indices>=245).any():raise ValueError('Source10521 straight upper-water geometry required')
|
|
guide=scale_nx(indices,4);mask=(((guide>=149)&(guide<=159))|((guide>=240)&(guide<=244))).astype(float)
|
|
owner=np.array(Image.fromarray(mask.astype(np.uint8)).resize((724,724),Image.Resampling.NEAREST),dtype=bool)
|
|
if not owner[:362].all()or owner[362:].any():raise ValueError('Source ownership changed dock anchor')
|
|
weights=resize_float(gaussian(mask,1.0),(724,724));numerator=resize_float(gaussian(palette[guide].astype(float)*mask[...,None],1.0),(724,724));base=numerator/np.maximum(weights[...,None],1e-12)
|
|
field=np.array(Image.fromarray(donor).resize((724,724),Image.Resampling.LANCZOS),dtype=float)
|
|
band=np.clip(field-gaussian(field,3.0),-16.0,16.0)*0.5
|
|
candidate=np.clip(np.rint(base+band),0,255).astype(np.uint8);result=old.copy();result[owner]=candidate[owner]
|
|
protected=np.array(Image.fromarray((guide>=240).astype(np.uint8)).resize((724,724),Image.Resampling.NEAREST),dtype=bool)
|
|
exact=np.array(Image.fromarray(palette[guide]).resize((724,724),Image.Resampling.NEAREST));result[protected]=exact[protected]
|
|
retained=set()
|
|
for s in range(1,5):
|
|
yy,xx=np.where(scale_nx(indices,s)>=240);retained.update((int(x)//s,int(y)//s)for x,y in zip(xx,yy))
|
|
changed=int(np.any(result!=np.clip(np.rint(base),0,255).astype(np.uint8),axis=2)[owner&~protected].sum())
|
|
if changed==0 or not np.array_equal(result[~owner],old[~owner]):raise ValueError('genuine fine detail and unchanged nonwater required')
|
|
return result,dict(water_pixels=int(owner.sum()),nonwater_changed_pixels=0,genuine_fine_band_changed_water_pixels=changed,old_to_new_water_changed_pixels=int(np.any(result!=old,axis=2)[owner].sum()),Original_protected_RGB_exact=bool(np.array_equal(result[protected],exact[protected])),original_detail_cells=[list(p)for p in sorted(retained)],Original_coarse_geometry_and_retained_cells_HD_credit=0,donor_coarse_shapes_colors_not_copied=True)
|
|
|
|
def run(recipe_path,output):
|
|
recipe_path=Path(recipe_path).absolute();root=recipe_path.parent;raw=read_contained(root,recipe_path.name,64*1024);recipe=json.loads(raw)
|
|
if set(recipe)!= {'method','asset','inputs','donor_box','parameters','dependency','scope'}or recipe['method']!=METHOD or recipe['asset']!='10521'or recipe['parameters']!=PARAMS or any(type(v)is not float for v in recipe['parameters'].values())or recipe['donor_box']!=[0,0,887,887]or recipe['dependency']!={'path':'tools/hd_terrain_cleanup.py','sha256':DEPENDENCY}:raise ValueError('exact reviewed10521 recipe required')
|
|
if set(recipe['inputs'])!={'source','palette','old_png','donor'}:raise ValueError('exact input roles required')
|
|
payloads={};pins={}
|
|
for name,binding in recipe['inputs'].items():
|
|
if set(binding)!={'path','sha256'}:raise ValueError('exact path/SHA input binding required')
|
|
payload=read_contained(root,binding['path']);digest=hashlib.sha256(payload).hexdigest()
|
|
if digest!=binding['sha256']:raise ValueError('input hash differs')
|
|
payloads[name]=payload;pins[name]=digest
|
|
source=payloads['source']
|
|
if len(source)!=1044 or struct.unpack_from('<5i',source)!=(1,1,0,32,32):raise ValueError('exact opaque Source32 PIC required')
|
|
donor=decode_png(payloads['donor'])
|
|
if donor.shape[0]<887 or donor.shape[1]<887:raise ValueError('donor crop outside image')
|
|
result,metrics=derive(np.frombuffer(source[20:],np.uint8).reshape(32,32),palette_rgb(payloads['palette']),decode_png(payloads['old_png']),donor[:887,:887])
|
|
out=Path(output).absolute()
|
|
if '..'in out.parts or out.exists()or not out.is_relative_to(root)or any(p.is_symlink()for p in[out,*out.parents]):raise ValueError('fresh contained output directory required')
|
|
tool=read_contained(Path(__file__).parent,Path(__file__).name,1024*1024)
|
|
out.mkdir();p=out/'10521-water-remediated.png';Image.fromarray(result).save(p)
|
|
record=dict(status='success',method=METHOD,recipe_sha256=hashlib.sha256(raw).hexdigest(),tool_sha256=hashlib.sha256(tool).hexdigest(),inputs=pins,output_sha256=hashlib.sha256(p.read_bytes()).hexdigest(),metrics=metrics,new_generation_calls=0,derived_not_generator_return=True)
|
|
(out/'execution-receipt.json').write_text(json.dumps(record,indent=2)+'\n');return record
|
|
if __name__=='__main__':
|
|
ap=argparse.ArgumentParser();ap.add_argument('--recipe',required=True);ap.add_argument('--output',required=True);a=ap.parse_args();print(json.dumps(run(a.recipe,a.output),indent=2))
|