comparison mercurial/setdiscovery.py @ 39171:abce899c985f

setdiscovery: pass head revisions into sample functions

This eliminates the last remaining consumer of heads() and
related functionality in dagutil.

Differential Revision: https://phab.mercurial-scm.org/D4319

39170:754f389b87f2

        for p in dag.parents(curr):
            if not revs or p in revs:
                dist.setdefault(p, d + 1)
                visit.append(p)

def _takequicksample(repo, dag, revs, size):
    """takes a quick sample of size <size>

    It is meant for initial sampling and focuses on querying heads and close
    ancestors of heads.

    :dag: a dag object
    :revs: set of revs to discover
    :size: the maximum size of the sample"""
    sample = set(repo.revs('heads(%ld)', revs))

    if len(sample) >= size:
        return _limitsample(sample, size)

    _updatesample(dag, None, dag.heads(), sample, quicksamplesize=size)
    return sample

def _takefullsample(repo, dag, revs, size):
    sample = set(repo.revs('heads(%ld)', revs))

    # update from heads
    _updatesample(dag, revs, dag.headsetofconnecteds(revs), sample)
    # update from roots

            samplefunc = _takequicksample
            targetsize = initialsamplesize
        if len(undecided) < targetsize:
            sample = list(undecided)
        else:
            sample = samplefunc(local, dag, undecided, targetsize)

        roundtrips += 1
        progress.update(roundtrips)
        ui.debug("query %i; still undecided: %i, sample size is: %i\n"
                 % (roundtrips, len(undecided), len(sample)))

39171:abce899c985f

        for p in dag.parents(curr):
            if not revs or p in revs:
                dist.setdefault(p, d + 1)
                visit.append(p)

def _takequicksample(repo, dag, headrevs, revs, size):
    """takes a quick sample of size <size>

    It is meant for initial sampling and focuses on querying heads and close
    ancestors of heads.

    :dag: a dag object
    :headrevs: set of head revisions in local DAG to consider
    :revs: set of revs to discover
    :size: the maximum size of the sample"""
    sample = set(repo.revs('heads(%ld)', revs))

    if len(sample) >= size:
        return _limitsample(sample, size)

    _updatesample(dag, None, headrevs, sample, quicksamplesize=size)
    return sample

def _takefullsample(repo, dag, headrevs, revs, size):
    sample = set(repo.revs('heads(%ld)', revs))

    # update from heads
    _updatesample(dag, revs, dag.headsetofconnecteds(revs), sample)
    # update from roots

            samplefunc = _takequicksample
            targetsize = initialsamplesize
        if len(undecided) < targetsize:
            sample = list(undecided)
        else:
            sample = samplefunc(local, dag, ownheads, undecided, targetsize)

        roundtrips += 1
        progress.update(roundtrips)
        ui.debug("query %i; still undecided: %i, sample size is: %i\n"
                 % (roundtrips, len(undecided), len(sample)))