fixed # 修复作物绘制位置异常的BUG

scripts/export_render_assets.py

@@ -4,20 +4,65 @@ import argparse
 import hashlib
 from io import BytesIO
 import json
+import math
 from pathlib import Path
 from zipfile import ZIP_STORED, ZipFile, ZipInfo
 
 from PIL import Image
 
 
+def fit_grid(source, target):
+    if len(source) != 30 or len(target) != 30:
+        raise ValueError('Expected all 30 reference land centers')
+    design = [[x, y, 1] for x, y in source]
+    normal = [[sum(row[i]*row[j] for row in design)
+               for j in range(3)] for i in range(3)]
+
+    def solve(values):
+        matrix = [row[:] + [value] for row, value in zip(normal, values)]
+        for i in range(3):
+            pivot = max(range(i, 3), key=lambda k: abs(matrix[k][i]))
+            matrix[i], matrix[pivot] = matrix[pivot], matrix[i]
+            divisor = matrix[i][i]
+            if abs(divisor) < 1e-8:
+                raise ValueError('Reference grid is degenerate')
+            matrix[i] = [v/divisor for v in matrix[i]]
+            for j in range(3):
+                if i != j:
+                    factor = matrix[j][i]
+                    matrix[j] = [v-factor*w for v, w in zip(
+                        matrix[j], matrix[i],
+                    )]
+        return [row[-1] for row in matrix]
+
+    affine = [solve([sum(row[i]*point[k]
+                        for row, point in zip(design, target))
+                     for i in range(3)]) for k in range(2)]
+    errors = [math.hypot(*[
+        sum(a*b for a, b in zip(row, point))-value
+        for row, value in zip(affine, dest)
+    ]) for point, dest in zip(design, target)]
+    if max(errors) > 10:
+        raise ValueError('Local grid no longer matches reference geometry')
+    return [row[:2] for row in affine], {
+        'rms': math.sqrt(sum(e*e for e in errors)/len(errors)),
+        'maximum': max(errors), 'points': len(errors),
+    }
+
+
 def apply_positions(root, metadata):
     document = json.loads((
         root / "src/farm_render/data/crop_positions.json"
     ).read_bytes())
-    if document.get("schema") != 2:
+    if document.get("schema") != 4:
         raise ValueError("Unsupported crop position schema")
-    if document.get("mode") != "full_canvas_bottom_center":
+    if document.get("mode") != "visible_contact_center":
         raise ValueError("Unsupported crop placement mode")
+    w, h = metadata['soil_size']
+    target = [[p['x']+w/2, p['y']+h/2] for p in metadata['positions']]
+    transform, errors = fit_grid(document['source_land_centers'], target)
+    document['scene_transform'] = transform
+    document['grid_fit'] = errors
     metadata["placement"] = document
 
 

scripts/preview_crop_mapping.py

@@ -0,0 +1,79 @@
+"""离线校验统一地块映射：成熟胡萝卜与随机五种作物各阶段。"""
+
+import argparse
+from dataclasses import replace
+from pathlib import Path
+import random
+import sys
+
+from PIL import Image, ImageDraw, ImageFont
+
+ROOT = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT / 'src'))
+
+from ci_font import choose_font
+from farm_core import Catalogue, Identity
+from farm_core.domain import Account, Plot, Snapshot
+from farm_render import FarmRenderer, RenderOptions
+from farm_render.canvas import Canvas
+
+
+def main():
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument('--output', type=Path, required=True)
+    args = parser.parse_args()
+    args.output.mkdir(parents=True, exist_ok=True)
+    renderer = FarmRenderer(choose_font())
+    who = Identity('preview', 'local', 'qq', 'crop-mapping')
+    catalogue = Catalogue.bundled()
+    carrot = catalogue.get('胡萝卜')
+    data = Snapshot(Account('成熟胡萝卜 · 统一公式校验', 11, 0, 6), (), tuple(
+        Plot(i, carrot, 100, pest_count=3, pest_spawned=3, maturity_pests=3)
+        for i in range(1, 4)
+    ))
+    result = renderer.render(who, data, 200, RenderOptions.high(
+        format='JPEG', theme='autumn',
+    ))
+    (args.output / 'carrot-mature-1536.jpg').write_bytes(result.content)
+    clean = replace(data, plots=tuple(
+        replace(plot, pest_count=0) for plot in data.plots
+    ))
+    result = renderer.render(who, clean, 200, RenderOptions.high(
+        format='JPEG', theme='autumn',
+    ))
+    (args.output / 'carrot-mature-clear-1536.jpg').write_bytes(result.content)
+    candidates = [p for p in catalogue if p.buyable
+                  and p.name in renderer.assets.metadata['plants']
+                  and f'plant/{p.name}/4.png' in
+                  renderer.assets.metadata['images']]
+    chosen = random.Random(20261021).sample(candidates, 5)
+    font = ImageFont.truetype(str(renderer.font_path), 24)
+    sheet = Image.new('RGB', (1536, 6*270), '#f4efdf')
+    draw = ImageDraw.Draw(sheet)
+    for row, plant in enumerate([carrot]+chosen):
+        phases = list(dict.fromkeys(map(int,
+            renderer.assets.metadata['plants'][plant.name]['phase'].split(','),
+        )))
+        draw.text((20, row*270+10), f'{plant.name}  #{plant.plant_id}',
+                  font=font, fill='#513819')
+        for stage, now in enumerate([0]+phases):
+            snapshot = Snapshot(data.account, (), (
+                Plot(1, plant, plant.growth_seconds),
+            ))
+            scene = renderer.scene(who, snapshot, now)
+            canvas = Canvas.new((1024, 624), '#f4efdf')
+            pos = renderer.assets.metadata['positions'][0]
+            soil = renderer.assets.image('soil/普通土地.png', (78, 52))
+            canvas.paste(soil, (round(pos['x']/3), round(pos['y']/3)))
+            renderer._plants(canvas, scene, RenderOptions())
+            tile = canvas.image.crop((390, 100, 600, 310))
+            x, y = stage*250+10, row*270+45
+            sheet.paste(tile.convert('RGB'), (x, y))
+            draw.text((x+10, y+190), f'阶段 {stage}', font=font,
+                      fill='#513819')
+    sheet.save(args.output / 'six-crops-stages.jpg', quality=90)
+    print('Crop mapping previews generated')
+
+
+if __name__ == '__main__':
+    main()