# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@octobus.net>
# Date 2023-09-18 23:24:10
# Node ID 810446d2d5efe149e8af541ccf19bdabcf42475c
# Parent  793a058f64bd283cf6cf6f4d45a704f7037d76b6

debug-delta-chaing: add a parameter to select revision to look at

This allows for much faster runtime when we are interrested in some revisions only.

diff --git a/mercurial/debugcommands.py b/mercurial/debugcommands.py
--- a/mercurial/debugcommands.py
+++ b/mercurial/debugcommands.py
@@ -746,7 +746,16 @@ def debugdate(ui, date, range=None, **op
 
 @command(
     b'debugdeltachain',
-    cmdutil.debugrevlogopts + cmdutil.formatteropts,
+    [
+        (
+            b'r',
+            b'rev',
+            [],
+            _('restrict processing to these revlog revisions'),
+        ),
+    ]
+    + cmdutil.debugrevlogopts
+    + cmdutil.formatteropts,
     _(b'-c|-m|FILE'),
     optionalrepo=True,
 )
@@ -797,12 +806,16 @@ def debugdeltachain(ui, repo, file_=None
 
     The sparse read can be enabled with experimental.sparse-read = True
     """
+    revs = None
+    revs_opt = opts.pop('rev', [])
+    if revs_opt:
+        revs = [int(r) for r in revs_opt]
     revlog = cmdutil.openrevlog(
         repo, b'debugdeltachain', file_, pycompat.byteskwargs(opts)
     )
     fm = ui.formatter(b'debugdeltachain', pycompat.byteskwargs(opts))
 
-    lines = revlog_debug.debug_delta_chain(revlog)
+    lines = revlog_debug.debug_delta_chain(revlog, revs=revs)
     # first entry is the header
     header = next(lines)
     fm.plain(header)
diff --git a/mercurial/revlogutils/debug.py b/mercurial/revlogutils/debug.py
--- a/mercurial/revlogutils/debug.py
+++ b/mercurial/revlogutils/debug.py
@@ -801,7 +801,7 @@ class DeltaChainAuditor:
         return p1, p2, compsize, uncompsize, deltatype, chain, chain_size
 
 
-def debug_delta_chain(revlog):
+def debug_delta_chain(revlog, revs=None):
     auditor = DeltaChainAuditor(revlog)
     r = revlog
     start = r.start
@@ -818,8 +818,14 @@ def debug_delta_chain(revlog):
     header += b'\n'
     yield header
 
+    if revs is None:
+        all_revs = iter(r)
+    else:
+        revlog_size = len(r)
+        all_revs = sorted(rev for rev in revs if rev < revlog_size)
+
     chainbases = {}
-    for rev in r:
+    for rev in all_revs:
         p1, p2, comp, uncomp, deltatype, chain, chainsize = auditor.revinfo(rev)
         chainbase = chain[0]
         chainid = chainbases.setdefault(chainbase, len(chainbases) + 1)
diff --git a/tests/test-completion.t b/tests/test-completion.t
--- a/tests/test-completion.t
+++ b/tests/test-completion.t
@@ -296,7 +296,7 @@ Show all commands + options
   debugdag: tags, branches, dots, spaces
   debugdata: changelog, manifest, dir
   debugdate: extended
-  debugdeltachain: changelog, manifest, dir, template
+  debugdeltachain: rev, changelog, manifest, dir, template
   debugdirstateignorepatternshash: 
   debugdirstate: nodates, dates, datesort, docket, all
   debugdiscovery: old, nonheads, rev, seed, local-as-revs, remote-as-revs, ssh, remotecmd, insecure, template