author | Arseniy Alekseyev <aalekseyev@janestreet.com> |
Wed, 04 Jan 2023 19:13:41 +0000 | |
changeset 49886 | 3aa8e569478a |
parent 48854 | 8b8054b8e5a7 |
child 50979 | 4c5f6e95df84 |
permissions | -rw-r--r-- |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
1 |
// dagops.rs |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
2 |
// |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
3 |
// Copyright 2019 Georges Racinet <georges.racinet@octobus.net> |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
4 |
// |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
5 |
// This software may be used and distributed according to the terms of the |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
6 |
// GNU General Public License version 2 or any later version. |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
7 |
|
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
8 |
//! Bindings for the `hg::dagops` module provided by the |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
9 |
//! `hg-core` package. |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
10 |
//! |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
11 |
//! From Python, this will be seen as `mercurial.rustext.dagop` |
43945
f98f0e3ddaa1
rust-index: add a function to convert PyObject index for hg-core
Pierre-Yves David <pierre-yves.david@octobus.net>
parents:
43269
diff
changeset
|
12 |
use crate::{conversion::rev_pyiter_collect, exceptions::GraphError}; |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
13 |
use cpython::{PyDict, PyModule, PyObject, PyResult, Python}; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
14 |
use hg::dagops; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
15 |
use hg::Revision; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
16 |
use std::collections::HashSet; |
48854
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
17 |
use vcsgraph::ancestors::node_rank; |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
18 |
use vcsgraph::graph::{Parents, Rank}; |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
19 |
|
43945
f98f0e3ddaa1
rust-index: add a function to convert PyObject index for hg-core
Pierre-Yves David <pierre-yves.david@octobus.net>
parents:
43269
diff
changeset
|
20 |
use crate::revlog::pyindex_to_graph; |
f98f0e3ddaa1
rust-index: add a function to convert PyObject index for hg-core
Pierre-Yves David <pierre-yves.david@octobus.net>
parents:
43269
diff
changeset
|
21 |
|
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
22 |
/// Using the the `index`, return heads out of any Python iterable of Revisions |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
23 |
/// |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
24 |
/// This is the Rust counterpart for `mercurial.dagop.headrevs` |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
25 |
pub fn headrevs( |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
26 |
py: Python, |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
27 |
index: PyObject, |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
28 |
revs: PyObject, |
43269
33fe96a5c522
rust-cpython: removed now useless py_set() conversion
Georges Racinet <georges.racinet@octobus.net>
parents:
42609
diff
changeset
|
29 |
) -> PyResult<HashSet<Revision>> { |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
30 |
let mut as_set: HashSet<Revision> = rev_pyiter_collect(py, &revs)?; |
43945
f98f0e3ddaa1
rust-index: add a function to convert PyObject index for hg-core
Pierre-Yves David <pierre-yves.david@octobus.net>
parents:
43269
diff
changeset
|
31 |
dagops::retain_heads(&pyindex_to_graph(py, index)?, &mut as_set) |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
32 |
.map_err(|e| GraphError::pynew(py, e))?; |
43269
33fe96a5c522
rust-cpython: removed now useless py_set() conversion
Georges Racinet <georges.racinet@octobus.net>
parents:
42609
diff
changeset
|
33 |
Ok(as_set) |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
34 |
} |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
35 |
|
48854
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
36 |
/// Computes the rank, i.e. the number of ancestors including itself, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
37 |
/// of a node represented by its parents. |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
38 |
pub fn rank( |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
39 |
py: Python, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
40 |
index: PyObject, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
41 |
p1r: Revision, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
42 |
p2r: Revision, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
43 |
) -> PyResult<Rank> { |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
44 |
node_rank(&pyindex_to_graph(py, index)?, &Parents([p1r, p2r])) |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
45 |
.map_err(|e| GraphError::pynew_from_vcsgraph(py, e)) |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
46 |
} |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
47 |
|
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
48 |
/// Create the module, with `__package__` given from parent |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
49 |
pub fn init_module(py: Python, package: &str) -> PyResult<PyModule> { |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
50 |
let dotted_name = &format!("{}.dagop", package); |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
51 |
let m = PyModule::new(py, dotted_name)?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
52 |
m.add(py, "__package__", package)?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
53 |
m.add(py, "__doc__", "DAG operations - Rust implementation")?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
54 |
m.add( |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
55 |
py, |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
56 |
"headrevs", |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
57 |
py_fn!(py, headrevs(index: PyObject, revs: PyObject)), |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
58 |
)?; |
48854
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
59 |
m.add( |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
60 |
py, |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
61 |
"rank", |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
62 |
py_fn!(py, rank(index: PyObject, p1r: Revision, p2r: Revision)), |
8b8054b8e5a7
rust: expose rank computation function to python
pacien <pacien.trangirard@pacien.net>
parents:
43945
diff
changeset
|
63 |
)?; |
41694
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
64 |
|
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
65 |
let sys = PyModule::import(py, "sys")?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
66 |
let sys_modules: PyDict = sys.get(py, "modules")?.extract(py)?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
67 |
sys_modules.set_item(py, dotted_name, &m)?; |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
68 |
// Example C code (see pyexpat.c and import.c) will "give away the |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
69 |
// reference", but we won't because it will be consumed once the |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
70 |
// Rust PyObject is dropped. |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
71 |
Ok(m) |
0c7b353ce100
rust-cpython: binding for headrevs()
Georges Racinet <georges.racinet@octobus.net>
parents:
diff
changeset
|
72 |
} |