"""Read immutable mesh and source textures; write reference-coordinate evidence only."""
import argparse
import json
from pathlib import Path
import sys

import bpy
import numpy as np

sys.path.insert(0, str(Path(__file__).resolve().parent))
from reference_registration_raster import rasterize

parser = argparse.ArgumentParser()
parser.add_argument('--folder', required=True)
parser.add_argument('--front-axis', choices=['-Y', '+Y', '-X', '+X'], default='-Y')
args = parser.parse_args(sys.argv[sys.argv.index('--') + 1:])
folder = Path(args.folder)
bpy.ops.wm.read_factory_settings(use_empty=True)
bpy.ops.import_scene.gltf(filepath=str(folder / 'source.glb'))
objects = [obj for obj in bpy.context.scene.objects if obj.type == 'MESH']
parts = []
for item in json.loads((folder / 'inputs.json').read_text()):
    matches = [obj for obj in objects
               if obj.name == item['part_id'] or obj.name.startswith(item['part_id'] + ' ')]
    if len(matches) != 1:
        raise ValueError('Cannot identify unique diagnostic component ' + item['part_id'])
    obj = matches[0]
    obj.data.calc_loop_triangles()
    if obj.data.uv_layers.active is None:
        raise ValueError('Registration diagnostic requires source UVs')
    parts.append({
        'vertices': np.array([obj.matrix_world @ vertex.co for vertex in obj.data.vertices]),
        'faces': np.array([triangle.vertices[:] for triangle in obj.data.loop_triangles]),
        'uvs': np.array([[obj.data.uv_layers.active.data[i].uv[:] for i in triangle.loops]
                         for triangle in obj.data.loop_triangles]),
        'texture': np.load(folder / item['file']),
        'alpha_mode': item.get('alpha_mode', 'OPAQUE'),
        'alpha_cutoff': item.get('alpha_cutoff', .5),
        'base_color_factor': item.get('base_color_factor', [1, 1, 1, 1]),
    })
reference = np.load(folder / 'reference.npz')
front = {'-Y': [0, -1, 0], '+Y': [0, 1, 0], '-X': [-1, 0, 0], '+X': [1, 0, 0]}[args.front_axis]
result = rasterize(parts, reference['bounds'], reference['mask'].shape, front)
np.savez_compressed(folder / 'registration-raster.npz', **result)