sclass MetaTransformer { sinterface StructureHandler { default O transform(O o, IF1 recurse) { null; } void visit(O o, IVF1 recurse); } L structureHandlers; //bool keepUnknown; // assume true for now Set seen; // initialize to identityHashSet() if you want to check for cycles *() {} *(StructureHandler... handlers) { structureHandlers = asList(handlers); } void addStructureHandler aka add(StructureHandler sh) { structureHandlers.add(sh); } // if f returns null, go through structure // if f returns an object, do not recurse into it // TODO: honor seen O transform(IF1 f, O o) { ping(); O x = f.get(o); ifdef MetaTransformer_debug printVars_str("MetaTransformer.transform", +f, +o, +x); endifdef if (x != null) ret x; IF1 recurse = liftFunction(f); fOr (StructureHandler h : structureHandlers) { ping(); ifdef MetaTransformer_debug print("Calling structure handler " + h + " on " + o); endifdef try object h.transform(o, recurse); } //ret keepUnknown ? o : null; ret o; } // transform without result void visit(IVF1 f, O o) { ping(); if (o == null) ret; if (seen != null && !seen.add(o)) ret; f.get(o); IVF1 recurse = x -> visit(f, x); fOr (StructureHandler h : structureHandlers) { ping(); ifdef MetaTransformer_debug print("Calling structure handler " + h + " on " + o); endifdef h.visit(o, recurse); } } // lift transformer function to handle structures IF1 liftFunction(IF1 f) { ret o -> transform(f, o); } // check if any element satisfies a predicate. // Note: might even be faster without the cancel point logic bool any(IPred pred, O o) { new Flag flag; withCancelPoint(cp -> visit(x -> { if (pred.get(x)) { flag.raise(); cancelTo(cp); } }, o) ); ret flag.isUp(); } void addVisitor(IVF1 visitor) { if (visitor == null) ret; addStructureHandler(new StructureHandler { public void visit(O o, IVF1 recurse) { visitor.get(o); } }); } void avoidCycles { seen = identityHashSet(); } }