3D RWLS Cube
N
c
shown loops
shown steps
source loops
"""
def load_payload(path):
with Path(path).open("rb") as f:
payload = pickle.load(f)
if isinstance(payload, dict) and "loopsoup" in payload:
return payload
return {
"N": None,
"c": None,
"seed": None,
"dimension": 3,
"loopsoup": payload,
"stats": {},
}
def loop_diameter(loop):
points = np.asarray(loop, dtype=float)
if len(points) == 0:
return 0.0
span = points.max(axis=0) - points.min(axis=0)
return float(np.linalg.norm(span))
def filtered_loops(loops, min_length, min_diameter, max_loops, max_points):
candidates = []
for loop in loops:
length = len(loop) - 1
if length < min_length:
continue
diameter = loop_diameter(loop)
if diameter < min_diameter:
continue
candidates.append((length, diameter, loop))
candidates.sort(key=lambda item: (item[1], item[0]), reverse=True)
selected = []
total_points = 0
for _, _, loop in candidates:
if max_loops is not None and len(selected) >= max_loops:
break
if max_points is not None and total_points + len(loop) > max_points:
break
selected.append(loop)
total_points += len(loop)
return selected
def normalize_loop(loop, N):
scale = max(1, N - 1)
return [[round(p[0] / scale, 6), round(p[1] / scale, 6), round(p[2] / scale, 6)] for p in loop]
def build_view_data(payload, min_length, min_diameter, max_loops, max_points):
loops = payload["loopsoup"]
N = payload.get("N")
if N is None:
N = max(max(max(point) for point in loop) for loop in loops if loop) + 1
selected = filtered_loops(loops, min_length, min_diameter, max_loops, max_points)
normalized = [normalize_loop(loop, N) for loop in selected]
lengths = [len(loop) - 1 for loop in selected]
return {
"N": N,
"c": payload.get("c"),
"seed": payload.get("seed"),
"source_loop_count": len(loops),
"loops": normalized,
"shown_steps": int(sum(lengths)),
"max_loop_length": int(max(lengths) if lengths else 0),
"min_length": min_length,
"min_diameter": min_diameter,
}
def parse_args():
parser = argparse.ArgumentParser(description="Export a 3D RWLS pickle to an interactive HTML viewer.")
parser.add_argument("pickle_path", type=Path, help="Input .pkl generated by RWLS_3D.py.")
parser.add_argument("--output", type=Path, default=None, help="Output .html path.")
parser.add_argument("--min-length", type=int, default=2, help="Only show loops with at least this many steps.")
parser.add_argument("--min-diameter", type=float, default=0.0, help="Only show loops with this lattice diameter or larger.")
parser.add_argument("--max-loops", type=int, default=3000, help="Maximum number of loops to embed.")
parser.add_argument("--max-points", type=int, default=160000, help="Maximum number of path points to embed.")
return parser.parse_args()
def main():
args = parse_args()
payload = load_payload(args.pickle_path)
view_data = build_view_data(
payload,
min_length=args.min_length,
min_diameter=args.min_diameter,
max_loops=args.max_loops,
max_points=args.max_points,
)
output = args.output
if output is None:
output = args.pickle_path.with_suffix(".html")
output.parent.mkdir(parents=True, exist_ok=True)
# Unescape the template before inserting JSON so data is never rewritten.
html = HTML_TEMPLATE.replace("{{", "{").replace("}}", "}")
html = html.replace("__DATA__", json.dumps(view_data, separators=(",", ":")))
output.write_text(html, encoding="utf-8")
print(f"Source loops: {view_data['source_loop_count']}")
print(f"Shown loops: {len(view_data['loops'])}")
print(f"Shown steps: {view_data['shown_steps']}")
print(f"Saved viewer to {output}")
if __name__ == "__main__":
main()