| """Batch measurement cache builder; malformed meshes do not abort the batch.""" |
| import argparse |
| import json |
| import sys |
| from pathlib import Path |
| sys.path.insert(0, str(Path(__file__).resolve().parents[1])) |
| from labeler.dataset import discover, cached_measure |
|
|
|
|
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--data', required=True) |
| parser.add_argument('--out', default='cache') |
| parser.add_argument('--method', choices=['pca', 'minimum', 'axis_aligned'], default='pca') |
| parser.add_argument('--trim', type=float, default=0.0) |
| args = parser.parse_args() |
| objects = discover(args.data) |
| if not objects: |
| parser.error('No mesh files found.') |
| failed = [] |
| for item in objects: |
| try: |
| cached_measure(item, args.out, args.method, args.trim) |
| print(f"OK {item['object_id']}") |
| except Exception as exc: |
| failed.append({'object_id': item['object_id'], 'error': str(exc)}) |
| print(f"FAILED {item['object_id']}: {exc}", file=sys.stderr) |
| Path(args.out).mkdir(parents=True, exist_ok=True) |
| (Path(args.out) / 'failures.json').write_text(json.dumps(failed, indent=2)) |
| return bool(failed) |
|
|
|
|
| if __name__ == '__main__': |
| sys.exit(main()) |
|
|
