comparison mercurial/revlog.py @ 51101:be56d322ee89

revlog: move the `rawtext` method on the inner object

This is one of the current core goal of the inner object: having a "simple" API
for some of the core revlog action, like getting a raw text, with the inner
object able to do that part internally, on its own.

51100:8ec2de9c6770

                    msg %= comp_mode
                    raise error.RevlogError(msg)

        return l


class revlog:
    """
    the underlying revision storage object

            rev = nodeorrev
        else:
            rev = self.rev(nodeorrev)
        return self._sidedata(rev)

    def _revisiondata(self, nodeorrev, raw=False):
        # deal with <nodeorrev> argument type
        if isinstance(nodeorrev, int):
            rev = nodeorrev
            node = self.node(rev)

            self.checkhash(text, node, rev=rev)
        if not validated:
            self._inner._revisioncache = (node, rev, rawtext)

        return text

    def _rawtext(self, node, rev):
        """return the possibly unvalidated rawtext for a revision

        returns (rev, rawtext, validated)
        """

        # revision in the cache (could be useful to apply delta)
        cachedrev = None
        # An intermediate text to apply deltas to
        basetext = None

        # Check if we have the entry in cache
        # The cache entry looks like (node, rev, rawtext)
        if self._inner._revisioncache:
            if self._inner._revisioncache[0] == node:
                return (rev, self._inner._revisioncache[2], True)
            cachedrev = self._inner._revisioncache[1]

        if rev is None:
            rev = self.rev(node)

        chain, stopped = self._inner._deltachain(rev, stoprev=cachedrev)
        if stopped:
            basetext = self._inner._revisioncache[2]

        # drop cache to save memory, the caller is expected to
        # update self._inner._revisioncache after validating the text
        self._inner._revisioncache = None

        targetsize = None
        rawsize = self.index[rev][2]
        if 0 <= rawsize:
            targetsize = 4 * rawsize

        bins = self._inner._chunks(chain, targetsize=targetsize)
        if basetext is None:
            basetext = bytes(bins[0])
            bins = bins[1:]

        rawtext = mdiff.patches(basetext, bins)
        del basetext  # let us have a chance to free memory early
        return (rev, rawtext, False)

    def _sidedata(self, rev):
        """Return the sidedata for a given revision number."""
        index_entry = self.index[rev]
        sidedata_offset = index_entry[8]

51101:be56d322ee89

                    msg %= comp_mode
                    raise error.RevlogError(msg)

        return l

    def raw_text(self, node, rev):
        """return the possibly unvalidated rawtext for a revision

        returns (rev, rawtext, validated)
        """

        # revision in the cache (could be useful to apply delta)
        cachedrev = None
        # An intermediate text to apply deltas to
        basetext = None

        # Check if we have the entry in cache
        # The cache entry looks like (node, rev, rawtext)
        if self._revisioncache:
            cachedrev = self._revisioncache[1]

        chain, stopped = self._deltachain(rev, stoprev=cachedrev)
        if stopped:
            basetext = self._revisioncache[2]

        # drop cache to save memory, the caller is expected to
        # update self._inner._revisioncache after validating the text
        self._revisioncache = None

        targetsize = None
        rawsize = self.index[rev][2]
        if 0 <= rawsize:
            targetsize = 4 * rawsize

        bins = self._chunks(chain, targetsize=targetsize)
        if basetext is None:
            basetext = bytes(bins[0])
            bins = bins[1:]

        rawtext = mdiff.patches(basetext, bins)
        del basetext  # let us have a chance to free memory early
        return (rev, rawtext, False)


class revlog:
    """
    the underlying revision storage object

            rev = nodeorrev
        else:
            rev = self.rev(nodeorrev)
        return self._sidedata(rev)

    def _rawtext(self, node, rev):
        """return the possibly unvalidated rawtext for a revision

        returns (rev, rawtext, validated)
        """
        # Check if we have the entry in cache
        # The cache entry looks like (node, rev, rawtext)
        if self._inner._revisioncache:
            if self._inner._revisioncache[0] == node:
                return (rev, self._inner._revisioncache[2], True)

        if rev is None:
            rev = self.rev(node)

        return self._inner.raw_text(node, rev)

    def _revisiondata(self, nodeorrev, raw=False):
        # deal with <nodeorrev> argument type
        if isinstance(nodeorrev, int):
            rev = nodeorrev
            node = self.node(rev)

            self.checkhash(text, node, rev=rev)
        if not validated:
            self._inner._revisioncache = (node, rev, rawtext)

        return text

    def _sidedata(self, rev):
        """Return the sidedata for a given revision number."""
        index_entry = self.index[rev]
        sidedata_offset = index_entry[8]