Initial commit: RHINE LAB · ANALYSIS OS：三维界面与动效实验（本仓库不含个人归档索引数据）

reference/measure-boot.py

@@ -0,0 +1,70 @@
+"""Measure the local 25 fps reference; generated images stay in the ignored folder."""
+import sys, json
+from pathlib import Path
+sys.path.insert(0, str(Path(__file__).resolve().parents[1] / '.tools'))
+import cv2
+import numpy as np
+from PIL import Image, ImageDraw
+
+root = Path(__file__).resolve().parent
+out = root / 'boot-frames'
+angles = np.linspace(0, 2*np.pi, 1440, endpoint=False)
+radii = np.arange(85, 2201, .5)
+mx = (960 + radii[:, None]*np.cos(angles)).astype(np.float32)
+my = (540 + radii[:, None]*np.sin(angles)).astype(np.float32)
+
+def segments(mask):
+    # Close sub-degree compression holes, then unwrap at a gap.
+    mask = np.array(mask, dtype=np.uint8)
+    mask = cv2.morphologyEx(np.tile(mask, 3)[None,:], cv2.MORPH_CLOSE, np.ones((1,7),np.uint8))[0,1440:2880]
+    gaps = np.where(mask == 0)[0]
+    if len(gaps) == 0: return [(0,360)]
+    origin = gaps[0]
+    rolled = np.roll(mask, -origin)
+    edges = np.diff(np.r_[0,rolled,0].astype(int))
+    return sorted([(((a+origin)%1440)/4,(b-a)/4) for a,b in zip(np.where(edges==1)[0],np.where(edges==-1)[0]) if b-a>12], key=lambda x:-x[1])
+
+rows=[]
+for f in range(487,569):
+    im=cv2.imread(str(out/f'{f:04}.jpg'))
+    polar=cv2.remap(im,mx,my,cv2.INTER_LINEAR,borderMode=cv2.BORDER_CONSTANT,borderValue=(128,128,128))
+    black=polar.max(axis=2)<100
+    white=polar.min(axis=2)>248
+    row={'frame':f}
+    # Outer arcs occupy much more of a concentric radius than text/background.
+    for name,mask,lo,hi in [('outer',black,240,2200),('white',white,220,1500),('inner',black,85,140)]:
+        valid=(radii>=lo)&(radii<=hi)
+        counts=mask.sum(axis=1)*valid
+        idx=int(np.argmax(counts))
+        row[name]=[float(radii[idx]),segments(mask[max(0,idx-1):idx+2].any(axis=0))[:2]]
+    orange=(im[:,:,2]>190)&(im[:,:,1]>65)&(im[:,:,1]<190)&(im[:,:,0]<100)
+    n,labels,stats,centers=cv2.connectedComponentsWithStats(orange.astype(np.uint8))
+    dots=[(float(x),float(y),int(s[4])) for s,(x,y) in zip(stats[1:],centers[1:]) if s[4]>8 and 500<x<1420 and 200<y<880]
+    row['dots']=dots
+    rows.append(row)
+(out/'measurements.json').write_text(json.dumps(rows,indent=2))
+brand=[]
+for f in range(277,316):
+    im=cv2.imread(str(out/f'{f:04}.jpg'),0)
+    bands=[]
+    for top,bottom in [(120,158),(163,181),(187,215)]:
+        roi=im[top:bottom,40:700]
+        yy,xx=np.where(roi<185)
+        bands.append([int(xx.min()+40),int(np.percentile(roi[yy,xx],10))] if len(xx) else None)
+    brand.append([f,*bands])
+(out/'brand-measurements.json').write_text(json.dumps(brand,indent=2))
+company=[]
+for f in range(588,622):
+    im=cv2.imread(str(out/f'{f:04}.jpg'),0)
+    xs=np.where(im[472,748:1170]<170)[0]
+    company.append([f,int(xs.max()+748) if len(xs) else None])
+(out/'company-measurements.json').write_text(json.dumps(company,indent=2))
+for name,frames,crop,size,cols in [('ring-detail',range(518,546),(655,235,1265,845),(366,366),7),('company-detail',range(588,609),(748,468,1168,530),(630,93),3),('brand-detail',range(277,307),(40,110,650,225),(610,115),3)]:
+    frames=list(frames);w,h=size
+    sheet=Image.new('RGB',(cols*w,((len(frames)+cols-1)//cols)*(h+25)),'#242424');draw=ImageDraw.Draw(sheet)
+    for i,f in enumerate(frames):
+        x=i%cols*w;y=i//cols*(h+25)
+        sheet.paste(Image.open(out/f'{f:04}.jpg').crop(crop).resize(size),(x,y+25))
+        draw.text((x+8,y+5),f'{f} / {f/25:.2f}s',fill='white')
+    sheet.save(out/f'{name}.jpg',quality=96)
+print(json.dumps(rows[::5],indent=2))