from pathlib import Path
from collections import Counter
import json,zipfile
from ezdxf import recover
from build_sample_grounds import ROOT,OUTPUT,save
from pierzainer_ground.dxf_terrain import convert_dwg_to_dxf,read_source_document
from pyproj import CRS

manifest=json.loads((ROOT/'data/input/samples_20261102/recovered/source_manifest.json').read_text(encoding='utf-8'))
results=[]
for record in manifest:
    source=Path(record['local']);cache=ROOT/'data/cache/recovered_sources';cache.mkdir(exist_ok=True)
    dxf=convert_dwg_to_dxf(source,cache);doc,_=read_source_document(dxf,cache)
    layers={};blocks=[]
    for b in doc.blocks:
        counts=Counter(e.dxftype() for e in b)
        if len(b)>100:blocks.append(dict(name=b.name,types=dict(counts)))
    for e in doc.modelspace():
        stats=layers.setdefault(e.dxf.layer,dict(count=0,types=Counter(),heights=[]))
        stats['count']+=1;stats['types'][e.dxftype()]+=1
        if e.dxftype()=='POINT':stats['heights'].append(float(e.dxf.location.z))
        elif e.dxftype()=='LWPOLYLINE':stats['heights'].append(float(e.dxf.elevation))
        elif e.dxftype()=='POLYLINE':stats['heights'].append(float(e.dxf.elevation.z))
    for name,stats in layers.items():
        h=stats.pop('heights');stats['z_range']=[min(h),max(h)] if h else None
    result={**record,'dxf':str(dxf),'header_bounds':[list(doc.header.get('$EXTMIN',(0,0,0))),list(doc.header.get('$EXTMAX',(0,0,0)))],
            'layers':layers,'blocks':blocks}
    if record.get('archive_entry'):
        with zipfile.ZipFile(record['source']) as archive:
            prjs=[i for i in archive.infolist() if i.filename.lower().endswith('.prj')]
            result['projection_files']=[]
            for p in prjs:
                text=archive.read(p).decode('utf-8-sig',errors='replace')
                try:epsg=CRS.from_wkt(text).to_epsg()
                except Exception:epsg=None
                result['projection_files'].append(dict(entry=p.filename,wkt=text,epsg=epsg))
    results.append(result);save(OUTPUT/'recovered_source_inspection.json',results)
    print(json.dumps({'samples':record['samples'],'file':source.name,'bounds':result['header_bounds'],'blocks':blocks[:4],
                      'top_layers':sorted(layers.items(),key=lambda pair:pair[1]['count'],reverse=True)[:7],
                      'prj':result.get('projection_files')},ensure_ascii=False),flush=True)