Examples with SearchState mb.statix.generator.SearchState used on opensource projects

Example 1 with SearchState

use of mb.statix.generator.SearchState in project nabl by metaborg.

the class StatixGenerate method logTrace.

@SuppressWarnings("unused")
private static void logTrace(ILogger log, Level lvl, SearchElement node, int maxDepth, Function1<SearchState, String> pp) {
    if (node instanceof SearchNodes) {
        SearchNodes<?> nodes = (SearchNodes<?>) node;
        log.log(lvl, " * {}", nodes.desc());
        logTrace(log, lvl, nodes.parent(), maxDepth, pp);
    } else {
        SearchNode<?> traceNode = (SearchNode<?>) node;
        int depth = 0;
        do {
            log.log(lvl, " * [{}] {}", traceNode.id(), traceNode.desc());
            if ((depth++ == 0 || (TRACE && depth <= maxDepth)) && traceNode.output() instanceof SearchState) {
                SearchState state = (SearchState) ((SearchNode<?>) traceNode).output();
                state.print(ln -> log.log(lvl, "   {}", ln), (t, u) -> u.toString(t));
            }
        } while ((traceNode = traceNode.parent()) != null);
        log.log(lvl, " # depth {}", depth);
    }
}

Example 2 with SearchState

use of mb.statix.generator.SearchState in project nabl by metaborg.

the class StatixGenerate method run.

public void run(String file) throws MetaborgException, InterruptedException {
    final FileObject resource = STX.S.resolve(file);
    TermFormatter tf = ITerm::toString;
    try {
        final ILanguageImpl lang = STX.cli.loadLanguage(STX.project.location());
        final IContext context = STX.S.contextService.get(resource, STX.project, lang);
        tf = StatixGenerator.pretty(STX.S, context, "pp-generated");
    } catch (MetaborgException e) {
    // ignore
    }
    final TermFormatter _tf = tf;
    final Function1<SearchState, String> pretty = (s) -> _tf.format(project(VAR, s));
    final DescriptiveStatistics hitStats = new DescriptiveStatistics();
    final DescriptiveStatistics missStats = new DescriptiveStatistics();
    final StreamProgressPrinter progress = new StreamProgressPrinter(System.err, 80, out -> {
        long hits = hitStats.getN();
        long all = hits + missStats.getN();
        out.println(" " + hits + "/" + all + " " + summary(hitStats));
    });
    final SearchLogger<SearchState, SearchState> searchLog = new SearchLogger<SearchState, SearchState>() {

        @Override
        public void init(long seed, SearchStrategy<SearchState, SearchState> strategy, Iterable<IConstraint> constraints) {
            log.info("seed {}", seed);
            log.info("strategy {}", strategy);
            log.info("constraints {}", constraints);
        }

        @Override
        public void success(SearchNode<SearchState> n) {
            progress.step('+');
            addSize(n.output(), hitStats);
            logSuccess(log, Level.Debug, n, pretty);
        }

        @Override
        public void failure(SearchNodes<?> nodes) {
            progress.step('.');
            SearchNode<?> parentNode = nodes.parent();
            if (parentNode != null && parentNode.output() instanceof SearchState) {
                addSize((SearchState) parentNode.output(), missStats);
            }
            logFailure(log, Level.Debug, nodes, pretty);
        }

        private void addSize(SearchState s, DescriptiveStatistics stats) {
            s.state().unifier().size(project(VAR, s)).ifFinite(size -> {
                stats.addValue(size.doubleValue());
            });
        }
    };
    final StatixGenerator statixGen = new StatixGenerator(STX.S, STX.context, resource);
    // Paret.addFragments(statixGen.spec());
    final Spec spec = statixGen.spec();
    final RandomTermGenerator rtg = new RandomTermGenerator(spec, statixGen.constraint(), new Paret(spec).search(), searchLog);
    final Stream<SearchState> resultStream = rtg.apply().nodes().map(sn -> {
        searchLog.success(sn);
        return sn.output();
    });
    log.info("Generating random terms.");
    final List<SearchState> results = Lists.newArrayList(resultStream.limit(COUNT).iterator());
    progress.done();
    results.forEach(s -> {
        System.out.println(pretty.apply(s));
    });
    log.info("Generated {} random terms.", results.size());
    logStatsInfo("hits", hitStats);
    logStatsInfo("misses", missStats);
}

Example 3 with SearchState

use of mb.statix.generator.SearchState in project nabl by metaborg.

the class Expand method updateSearchState.

private Optional<SearchState> updateSearchState(SearchContext ctx, IConstraint predicate, ApplyResult result, SearchState input) {
    final IConstraint applyConstraint = result.body();
    final IState.Immutable applyState = input.state();
    final IUniDisunifier.Immutable applyUnifier = applyState.unifier();
    // update constraints
    final Set.Transient<IConstraint> constraints = input.constraints().asTransient();
    constraints.__insert(applyConstraint);
    constraints.__remove(predicate);
    // update completeness
    final ICompleteness.Transient completeness = input.completeness().melt();
    completeness.add(applyConstraint, ctx.spec(), applyUnifier);
    java.util.Set<CriticalEdge> removedEdges = completeness.remove(predicate, ctx.spec(), applyUnifier);
    // update delays
    final Map.Transient<IConstraint, Delay> delays = Map.Transient.of();
    input.delays().forEach((c, d) -> {
        if (!Sets.intersection(d.criticalEdges(), removedEdges).isEmpty()) {
            constraints.__insert(c);
        } else {
            delays.__put(c, d);
        }
    });
    // return new state
    final SearchState newState = input.replace(applyState, constraints.freeze(), delays.freeze(), completeness.freeze());
    return Optional.of(newState);
}

Example 4 with SearchState

use of mb.statix.generator.SearchState in project nabl by metaborg.

the class Fix method doApply.

@Override
protected SearchNodes<SearchState> doApply(SearchContext ctx, SearchNode<SearchState> node) {
    final AtomicInteger failureCount = new AtomicInteger();
    final Deque<Iterator<SearchNode<SearchState>>> stack = new LinkedList<>();
    final Action1<SearchNodes<SearchState>> push = ns -> {
        Iterator<SearchNode<SearchState>> it = ns.nodes().iterator();
        if (it.hasNext()) {
            stack.push(it);
            failureCount.set(0);
        } else {
            ctx.failure(ns);
            failureCount.incrementAndGet();
        }
    };
    final SearchNodes<SearchState> initNodes = infer.apply(ctx, node);
    push.apply(initNodes);
    final Stream<SearchNode<SearchState>> fixNodes = StreamUtil.generate(() -> {
        while (!stack.isEmpty()) {
            final Iterator<SearchNode<SearchState>> nodes = stack.peek();
            if (!nodes.hasNext()) {
                stack.pop();
                continue;
            }
            final SearchNode<SearchState> next = nodes.next();
            if (Streams.stream(next.output().constraintsAndDelays()).allMatch(c -> (c instanceof CUser && done.test((CUser) c)))) {
                return Optional.of(next);
            }
            final SearchNodes<SearchState> nextNodes = searchAndInfer.apply(ctx, next);
            push.apply(nextNodes);
            if (maxConsecutiveFailures >= 0 && failureCount.get() >= maxConsecutiveFailures) {
                // we're done here
                stack.clear();
            }
        }
        return Optional.empty();
    });
    return SearchNodes.of(node, () -> "fix(???)", fixNodes);
}

Example 5 with SearchState

use of mb.statix.generator.SearchState in project nabl by metaborg.

the class MapConstraints method doApply.

@Override
protected SearchNodes<SearchState> doApply(SearchContext ctx, SearchNode<SearchState> node) {
    final SearchState input = node.output();
    final IState.Immutable state = input.state();
    final Set.Immutable<IConstraint> constraints = input.constraints().stream().map(f::apply).collect(CapsuleCollectors.toSet());
    final ICompleteness.Transient completeness = Completeness.Transient.of();
    completeness.addAll(constraints, ctx.spec(), state.unifier());
    completeness.addAll(input.delays().keySet(), ctx.spec(), state.unifier());
    final SearchState output = input.replace(state, constraints, input.delays(), completeness.freeze());
    return SearchNodes.of(node, this::toString, new SearchNode<>(ctx.nextNodeId(), output, node, this.toString()));
}