comparison mercurial/commands.py @ 29913:9cb950276d27

debugrevspec: add option to print parsed tree at given stages

"-p <stage>" is useful for investigating parsing stages. With -p option, a
transformed tree is printed no matter if it is changed or not, which allows
us to know valid stage names by "-p all".

29912:1c1c2bce2f97

                     + fmt % pcfmt(nump2, numdeltas))
            ui.write(('deltas against other : ') + fmt % pcfmt(numother,
                                                             numdeltas))

@command('debugrevspec',
    [('', 'optimize', None, _('print parsed tree after optimizing'))],
    ('REVSPEC'))
def debugrevspec(ui, repo, expr, **opts):
    """parse and apply a revision specification

    Use --verbose to print the parsed tree before and after aliases
    expansion.
    """
    stages = [
        ('parsed', lambda tree: tree),
        ('expanded', lambda tree: revset.expandaliases(ui, tree)),
        ('concatenated', revset.foldconcat),
        ('analyzed', revset.analyze),
        ('optimized', revset.optimize),
    ]

    showalways = set(['parsed'])
    showchanged = set(['expanded', 'concatenated'])
    if opts['optimize']:
        showalways.add('optimized')

    printedtree = None
    tree = revset.parse(expr, lookup=repo.__contains__)
    for n, f in stages:
        tree = f(tree)
        if n in showalways or (n in showchanged and tree != printedtree):
            if n != 'parsed':
                ui.note(("* %s:\n") % n)
            ui.note(revset.prettyformat(tree), "\n")
            printedtree = tree

    func = revset.match(ui, expr, repo)
    revs = func(repo)
    if ui.verbose:

29913:9cb950276d27

                     + fmt % pcfmt(nump2, numdeltas))
            ui.write(('deltas against other : ') + fmt % pcfmt(numother,
                                                             numdeltas))

@command('debugrevspec',
    [('', 'optimize', None, _('print parsed tree after optimizing')),
     ('p', 'show-stage', [],
      _('print parsed tree at the given stage'), _('NAME')),
     ],
    ('REVSPEC'))
def debugrevspec(ui, repo, expr, **opts):
    """parse and apply a revision specification

    Use -p/--show-stage option to print the parsed tree at the given stages.
    Use -p all to print tree at every stage.
    """
    stages = [
        ('parsed', lambda tree: tree),
        ('expanded', lambda tree: revset.expandaliases(ui, tree)),
        ('concatenated', revset.foldconcat),
        ('analyzed', revset.analyze),
        ('optimized', revset.optimize),
    ]
    stagenames = set(n for n, f in stages)

    showalways = set()
    showchanged = set()
    if ui.verbose and not opts['show_stage']:
        # show parsed tree by --verbose (deprecated)
        showalways.add('parsed')
        showchanged.update(['expanded', 'concatenated'])
        if opts['optimize']:
            showalways.add('optimized')
    if opts['show_stage'] and opts['optimize']:
        raise error.Abort(_('cannot use --optimize with --show-stage'))
    if opts['show_stage'] == ['all']:
        showalways.update(stagenames)
    else:
        for n in opts['show_stage']:
            if n not in stagenames:
                raise error.Abort(_('invalid stage name: %s') % n)
        showalways.update(opts['show_stage'])

    printedtree = None
    tree = revset.parse(expr, lookup=repo.__contains__)
    for n, f in stages:
        tree = f(tree)
        if n in showalways or (n in showchanged and tree != printedtree):
            if opts['show_stage'] or n != 'parsed':
                ui.write(("* %s:\n") % n)
            ui.write(revset.prettyformat(tree), "\n")
            printedtree = tree

    func = revset.match(ui, expr, repo)
    revs = func(repo)
    if ui.verbose: