"""승인된 물 소재 자산을 복사해 정식 기준 원본과 애니메이션 원본을 만든다."""
import argparse
import hashlib
import json
import math
from pathlib import Path
import sys
import bpy
from mathutils import Vector
ROOT=Path(__file__).resolve().parents[3]
PROJECT=ROOT/'projects/water-hydrogen-bond-surface-tension'
APPROVED=PROJECT/'stages/08-scenes/r3/generated-source/preview.blend'
APPROVED_HASH='ba0850a2148824cc8416704c03f14e5714ca01a16620a4eb26981e238af78c73'
parser=argparse.ArgumentParser()
parser.add_argument('--stage',choices=['references','images'],required=True)
args=parser.parse_args(sys.argv[sys.argv.index('--')+1:])


def clone(obj,name,scene,parent=None):
    result=obj.copy()
    if obj.data:
        result.data=obj.data.copy()
    result.name=name
    result.animation_data_clear()
    result.parent=parent
    scene.collection.objects.link(result)
    return result


def studio(name,base,bounds,scale):
    scene=bpy.data.scenes.new(name)
    scene.world=base.world
    scene.render.engine='BLENDER_EEVEE'
    scene.render.fps=50
    scene.render.resolution_x,scene.render.resolution_y=1080,1920
    scene.eevee.taa_render_samples=64
    scene.view_settings.view_transform='Standard'
    for obj in base.objects:
        if obj.type in ('CAMERA','LIGHT') or obj.get('role')=='background':
            item=clone(obj,name+obj.type+obj.name,scene)
            if item.type=='CAMERA':
                scene.camera=item
                item.data.ortho_scale=scale
    scene.frame_start,scene.frame_end=bounds
    return scene


def molecule(scene,name,location,size=1,rotation=0):
    template=bpy.data.objects['PairA']
    root=clone(template,name,scene)
    root.location=location
    root.rotation_euler=(0,0,rotation)
    root.scale=(size,)*3
    for child in template.children:
        clone(child,name+child.name[len('PairA'):],scene,root)
    return root


def droplet(scene,name,location,scale):
    obj=clone(bpy.data.objects['Droplet'],name,scene)
    obj.location=location
    obj.scale=scale
    return obj


def anchor(scene,name,location,label):
    obj=bpy.data.objects.new(name,None)
    scene.collection.objects.link(obj)
    obj.location=location
    obj['anchor']=label
    return obj


def mark(obj,label):
    obj['anchor']=label


def collection(name,objects):
    group=bpy.data.collections.new(name)
    group.use_fake_user=True
    for obj in objects:
        group.objects.link(obj)
    return group


def animate_network(scene,prefix,start,turn,end):
    # 온전한 분자를 움직이고 H···O 원통을 실제 원자 위치로 갱신한다.
    roots=[scene.objects[prefix+str(i)] for i in range(10)]
    for i,root in enumerate(roots):
        position=root.location.copy()
        for frame,delta in [(start,0),(turn,.09*math.sin(i+1)),(end,.06*math.cos(i))]:
            root.location=position+Vector((delta,0,0))
            root.keyframe_insert('location',frame=frame)
    template=bpy.data.objects['LiquidDash0_1_0']
    pairs=[(0,1),(1,2),(3,4),(3,7),(4,5),(4,6),(4,7),(5,6),(5,8),(8,9)]
    for i,j in pairs:
        hydrogen,oxygen=scene.objects[prefix+str(i)+'_H1'],scene.objects[prefix+str(j)+'_O']
        for segment in range(6):
            obj=clone(template,prefix+'Dash'+str(i)+'_'+str(j)+'_'+str(segment),scene)
            obj['hydrogen'],obj['oxygen']=hydrogen.name,oxygen.name
            for frame in [start,1779,turn,2257,end]:
                scene.frame_set(frame)
                bpy.context.view_layer.update()
                a,b=hydrogen.matrix_world.translation,oxygen.matrix_world.translation
                direction=b-a
                obj.location=a+direction*(.18+segment*.11+.031)
                obj.rotation_euler=direction.to_track_quat('Z','Y').to_euler()
                active=not ((i,j)==(3,4) and frame>=turn) and not ((i,j)==(3,7) and frame<turn) and not ((i,j)==(4,5) and frame>=2257) and not ((i,j)==(4,6) and frame<2257)
                obj.scale=(1 if active else .001,1 if active else .001,direction.length*.062/.1)
                for path in ['location','rotation_euler','scale']:
                    obj.keyframe_insert(path,frame=frame)


if args.stage=='references':
    assert hashlib.sha256(APPROVED.read_bytes()).hexdigest()==APPROVED_HASH, '승인 원본 해시 불일치'
    bpy.ops.wm.open_mainfile(filepath=str(APPROVED))
    pair=bpy.data.scenes['BondPair']
    drop=bpy.data.scenes['DropletStretch']
    interface=bpy.data.scenes['Interface']
    opening=studio('OpeningDroplet',drop,(1,397),10.5)
    obj=droplet(opening,'OpeningWater',(0,0,0),(1,1,1))
    mark(obj,'opening_drop')
    anchor(opening,'OpeningMoleculeInset',(1.65,0,1.7),'molecule_inset')
    single=studio('SingleMolecule',pair,(397,905),9)
    molecule(single,'SingleWater',(0,0,-.4),1.8)
    for suffix,label in [('O','single_O_delta_minus'),('H1','single_H_delta_plus'),('H2','single_H2_delta_plus')]:
        mark(single.objects['SingleWater_'+suffix],label)
    network=studio('LiquidNetwork',interface,(1637,2469),13)
    for i in range(10):
        old=interface.objects['Liquid'+str(i)]
        molecule(network,'Network'+str(i),(old.location.x,old.location.y,old.location.z+1.3),.58,old.rotation_euler.z)
    mark(network.objects['Network4'],'network_interior')
    comparison=studio('EqualVolumeComparison',drop,(4077,4370),12.5)
    size=.72
    droplet(comparison,'ComparisonLong',(-1.65,0,0),(size*1.65,size/math.sqrt(1.65),size/math.sqrt(1.65)))
    droplet(comparison,'ComparisonRound',(1.65,0,0),(size,)*3)
    mark(comparison.objects['ComparisonLong'],'equal_volume_left')
    mark(comparison.objects['ComparisonRound'],'equal_volume_right')
    conditions=studio('ShapeConditions',drop,(4370,4889),12.5)
    droplet(conditions,'SmallFreeDrop',(-1.65,0,.2),(.58,)*3)
    # 큰 접촉 방울은 같은 부피 비교가 아닌 별도 조건 모형이다.
    contact=droplet(conditions,'LargeContactDrop',(1.55,0,-.275),(1.02,1.02,.75/1.3))
    mark(conditions.objects['SmallFreeDrop'],'small_free_condition')
    mark(contact,'large_contact_condition')
    bpy.context.window.scene=conditions
    bpy.ops.mesh.primitive_cube_add(size=1,location=(1.55,0,-1.15))
    floor=bpy.context.object
    floor.name='ContactFloor'
    floor.scale=(3.1,2.5,.25)
    floor_material=bpy.data.materials.new('ContactFloorCream')
    floor_material.diffuse_color=(.78,.73,.62,1)
    floor_material.use_nodes=True
    floor_material.node_tree.nodes['Principled BSDF'].inputs['Base Color'].default_value=(.78,.73,.62,1)
    floor_material.node_tree.nodes['Principled BSDF'].inputs['Roughness'].default_value=.8
    floor.data.materials.append(floor_material)
    floor['role']='condition_floor'
    mark(floor,'contact_floor')
    answer=studio('AnswerDroplet',drop,(4889,5529),12.5)
    droplet(answer,'AnswerWater',(0,0,-.7),(1,)*3)
    mark(answer.objects['AnswerWater'],'answer_macro_drop')
    for i,position in enumerate([(-1.1,0,1.8),(0,.1,2.05),(1.1,0,1.8)]):
        molecule(answer,'AnswerMolecule'+str(i),position,.55,i*.2)
    mark(answer.objects['AnswerMolecule0_H1'],'answer_inset_H')
    mark(answer.objects['AnswerMolecule1_O'],'answer_inset_O')
    mark(answer.objects['AnswerMolecule2'],'answer_inset_extent')
    mark(pair.objects['PairA_H1'],'pair_H_delta_plus')
    mark(pair.objects['PairB_O'],'pair_O_delta_minus')
    mark(interface.objects['Liquid4'],'water_interior')
    mark(interface.objects['Liquid1'],'water_surface')
    anchor(interface,'InterfaceAirAnchor',(0,0,1.45),'air_region')
    anchor(interface,'InterfaceWaterAnchor',(0,0,-2.95),'cutaway_water_region')
    mark(drop.objects['Droplet'],'same_volume_drop')
    collection('StudioAssets',[o for s in bpy.data.scenes for o in s.objects if o.get('role')=='background'])
    collection('StyleAssets',[o for s in bpy.data.scenes for o in s.objects if o.get('role')=='background'])
    collection('DropletStudioAssets',[o for s in bpy.data.scenes if s.name in ['OpeningDroplet','DropletStretch','AnswerDroplet'] for o in s.objects if o.get('role')=='background'])
    collection('MoleculeStudioAssets',[o for s in bpy.data.scenes if s.name in ['SingleMolecule','BondPair','LiquidNetwork'] for o in s.objects if o.get('role')=='background'])
    collection('ComparisonStudioAssets',[o for s in bpy.data.scenes if s.name in ['EqualVolumeComparison','ShapeConditions'] for o in s.objects if o.get('role')=='background'])
    collection('MoleculeAssets',[o for o in bpy.data.objects if o.get('role') in ('molecule','atom','covalent','hydrogen_bond')])
    collection('DropletAssets',[o for o in bpy.data.objects if o.get('role')=='same_volume_water'])
    collection('InterfaceAssets',[interface.objects['WaterCutawayBody']])
    # 실제 기준 원본에도 액체 연결 자산을 구성하여 후속 원본이 새 경로만 참조하지 않게 한다.
    bpy.context.window.scene=network
    animate_network(network,'Network',1637,2019,2469)
    bpy.context.window.scene=answer
    # 답 회수에는 거시 방울과 공간적으로 분리된 분자 개념도가 공존한다.
    for i,j in [(0,1),(1,2)]:
        template=bpy.data.objects['LiquidDash0_1_0']
        h,o=answer.objects['AnswerMolecule'+str(i)+'_H1'],answer.objects['AnswerMolecule'+str(j)+'_O']
        bpy.context.view_layer.update()
        a,b=h.matrix_world.translation,o.matrix_world.translation
        for n in range(6):
            obj=clone(template,'AnswerDash'+str(i)+'_'+str(j)+'_'+str(n),answer)
            obj['hydrogen'],obj['oxygen']=h.name,o.name
            direction=b-a
            obj.location=a+direction*(.18+n*.11+.031)
            obj.rotation_euler=direction.to_track_quat('Z','Y').to_euler()
            obj.scale=(1,1,direction.length*.062/.1)
    for scene in bpy.data.scenes:
        scene['approved_source_sha256']=APPROVED_HASH
    destination=PROJECT/'stages/10-references/r2/generated-source/references.blend'
else:
    artifact=json.loads((PROJECT/'stages/10-references/r2/artifact.json').read_text(encoding='utf-8'))
    result=artifact['payload']['result']
    source=ROOT/result['scene_file']['path']
    assert result['scene_file']['verified'] and hashlib.sha256(source.read_bytes()).hexdigest()==result['scene_file']['sha256']
    assert len(result['items'])==7 and all((ROOT/item['media']['path']).is_file() and item['media']['verified'] for item in result['items'])
    bpy.ops.wm.open_mainfile(filepath=str(source))
    for scene in bpy.data.scenes:
        scene['reference_source_sha256']=result['scene_file']['sha256']
    drop=bpy.data.objects['Droplet']
    expression='1+0.85*min(max((frame-3173)/200,0),1)*(1-min(max((frame-3657)/196,0),1))'
    drop.animation_data.drivers[0].driver.expression=expression
    for axis in [1,2]:
        drop.animation_data.drivers[axis].driver.expression='('+expression+')**(-0.5)'
    # 액체 점선의 위치·방향을 프레임마다 실제 H/O 평가 위치에 맞춰 굽는다.
    network=bpy.data.scenes['LiquidNetwork']
    bpy.context.window.scene=network
    segments=[obj for obj in network.objects if obj.get('role')=='hydrogen_bond']
    for obj in segments:
        obj.animation_data_clear()
    for frame in range(1637,2470):
        network.frame_set(frame)
        bpy.context.view_layer.update()
        for obj in segments:
            h,o=network.objects[obj['hydrogen']],network.objects[obj['oxygen']]
            a,b=h.matrix_world.translation,o.matrix_world.translation
            direction=b-a
            segment=int(obj.name.rsplit('_',1)[1])
            obj.location=a+direction*(.18+segment*.11+.031)
            obj.rotation_euler=direction.to_track_quat('Z','Y').to_euler()
            pair=(h.parent.name,o.parent.name)
            first=min(max((frame-1779)/240,0),1)
            second=min(max((frame-2019)/238,0),1)
            weight={('Network3','Network4'):1-first,('Network3','Network7'):first,('Network4','Network5'):1-second,('Network4','Network6'):second}.get(pair,1)
            obj.scale=(max(weight,.001),max(weight,.001),direction.length*.062/.1)
            for path in ['location','rotation_euler','scale']:
                obj.keyframe_insert(path,frame=frame)
    # 답의 수소결합은 영구 끈이 아니며 연결이 약해졌다가 다시 나타난다.
    answer=bpy.data.scenes['AnswerDroplet']
    for obj in answer.objects:
        if obj.get('role')=='hydrogen_bond':
            original=obj.scale.z
            for frame,active in [(4889,1),(5100,1),(5160,.001),(5229,1),(5529,1)]:
                obj.scale=(active,active,original)
                obj.keyframe_insert('scale',frame=frame)
    destination=PROJECT/'stages/11-images/r2/generated-source/animation.blend'
for scene in bpy.data.scenes:
    bpy.context.window.scene=scene
    scene.frame_set(scene.frame_start)
destination.parent.mkdir(parents=True,exist_ok=True)
assert not destination.exists(), '제출 원본 덮어쓰기 금지'
bpy.ops.wm.save_as_mainfile(filepath=str(destination))
print('정식 원본 저장',destination)