54 lines
2.9 KiB
Python
54 lines
2.9 KiB
Python
"""Sample the continuous infinity contour against the reference's black ink."""
|
|
import sys,json
|
|
from pathlib import Path
|
|
root=Path(__file__).resolve().parent
|
|
sys.path.insert(0,str(root.parent/'.tools'))
|
|
import cv2
|
|
import numpy as np
|
|
from PIL import Image,ImageDraw
|
|
out=root/'boot-frames'
|
|
# Continuous version of the shared Bezier, with the printed logo's small gap filled.
|
|
curves=[[(295,73),(295,41),(273,15),(240,15)],[(240,15),(221,15),(207,23),(192,38)],[(192,38),(186,43),(181,47),(176,52)],[(176,52),(127,96),(103,128),(70,128)],[(70,128),(38,128),(15,101),(15,70)],[(15,70),(15,39),(37,15),(70,15)],[(70,15),(103,15),(127,48),(156,75)],[(156,75),(182,99),(208,128),(240,128)],[(240,128),(273,128),(295,105),(295,73)]]
|
|
p=[]
|
|
for c in curves:
|
|
t=np.linspace(0,1,300)[:,None];a,b,d,e=np.array(c)
|
|
p.extend((1-t)**3*a+3*(1-t)**2*t*b+3*(1-t)*t*t*d+t**3*e)
|
|
p=np.array(p);dist=np.r_[0,np.cumsum(np.linalg.norm(np.diff(p,axis=0),axis=1))]
|
|
params=np.linspace(0,1,2000,endpoint=False)
|
|
points=np.c_[np.interp(params*dist[-1],dist,p[:,0]),np.interp(params*dist[-1],dist,p[:,1])]
|
|
results=[]
|
|
for f in range(229,487):
|
|
im=cv2.imread(str(out/f'{f:04}.jpg'),0)
|
|
# Fit the final-size contour to its source bounding box; translation is
|
|
# measured from stable leftmost ink once the left loop has been drawn.
|
|
crop=im[464:608,490:1130]
|
|
yy,xx=np.where(crop<90)
|
|
left=float(xx.min()+490) if len(xx) and f>=238 else 815
|
|
x=(points[:,0]-2)*(.955)+left
|
|
y=(points[:,1]-2)*(.955)+465
|
|
gray=cv2.remap(im,x.astype(np.float32)[None,:],y.astype(np.float32)[None,:],cv2.INTER_LINEAR)[0]
|
|
mask=(gray<130).astype(np.uint8)
|
|
# Remove tiny raster interruptions; do not fill a real moving cut.
|
|
mask=cv2.morphologyEx(np.tile(mask,3)[None,:],cv2.MORPH_CLOSE,np.ones((1,8),np.uint8))[0,2000:4000]
|
|
changes=np.diff(np.r_[0,mask,0].astype(int))
|
|
runs=[(int(a),int(b)) for a,b in zip(np.where(changes==1)[0],np.where(changes==-1)[0]) if b-a>4]
|
|
results.append({'frame':f,'left':left,'runs':runs,'coverage':round(float(mask.mean()),4)})
|
|
if 247<=f<=268:
|
|
mark=im[475:593,int(left):int(left)+294]
|
|
n,labels,stats,centers=cv2.connectedComponentsWithStats((mark<110).astype(np.uint8))
|
|
symbols=[]
|
|
for s,(cx,cy) in zip(stats[1:],centers[1:]):
|
|
if s[4]<4 or s[4]>2500 or not 30<cy<95:continue
|
|
symbols.append([round(float(cx/.955+2),2),round(float((cy+10)/.955+2),2),int(s[2]),int(s[3]),int(s[4])])
|
|
results[-1]['symbols']=symbols
|
|
if f in [238,260,350,460]:print(f,'bbox',(int(xx.min()+490),int(yy.min()+464),int(xx.max()+490),int(yy.max()+464)))
|
|
(out/'logo-measurements.json').write_text(json.dumps(results,indent=2))
|
|
for a,b in [(229,265),(300,487)]:
|
|
rows=[r for r in results if a<=r['frame']<b]
|
|
img=Image.new('RGB',(1100,len(rows)*12),'#eeeeeb');draw=ImageDraw.Draw(img)
|
|
for i,row in enumerate(rows):
|
|
draw.text((0,i*12),str(row['frame']),fill='black')
|
|
for start,end in row['runs']:draw.rectangle((70+start/2,i*12,70+end/2,i*12+10),fill='black')
|
|
img.save(out/f'logo-ink-{a}.jpg')
|
|
print('Contour length',dist[-1])
|