comparison mercurial/discovery.py @ 18103:83cb1a1a705d

clfilter: use filtering in `visiblebranchmap`

Here is the first real use of changelog filtering. The change is very small to
allow testing the new filter with a setup close to the original one.

We replace custom post processing on branchmap function by call to the
standard code pass on a filtering repo.

In later coming will have wider usage of filtering that will make the dedicated
function useless.

18102:3c7b67b76190

    return vheads


def visiblebranchmap(repo):
    """return a branchmap for the visible set"""
    # XXX Recomputing this data on the fly is very slow.  We should build a
    # XXX cached version while computing the standard branchmap version.
    sroots = repo._phasecache.phaseroots[phases.secret]
    if sroots or repo.obsstore:
        vbranchmap = {}
        for branch, nodes in  repo.branchmap().iteritems():
            # search for secret heads.
            for n in nodes:
                if repo[n].phase() >= phases.secret:
                    nodes = None
                    break
            # if secret heads were found we must compute them again
            if nodes is None:
                s = repo.set('heads(branch(%s) - secret() - extinct())',
                             branch)
                nodes = [c.node() for c in s]
            vbranchmap[branch] = nodes
    else:
        vbranchmap = repo.branchmap()
    return vbranchmap

18103:83cb1a1a705d

    return vheads


def visiblebranchmap(repo):
    """return a branchmap for the visible set"""
    return repo.filtered('unserved').branchmap()