mercurial-scm/hg-stable: rust/hg-cpython/src/ancestors.rs annotate

annotate rust/hg-cpython/src/ancestors.rs @ 42172:c3484ddbdb96 stable 5.0.1

manifest: add some documentation to _lazymanifest python code It was not particularly easy figuring out the design of this class and keeping track of how the pieces work. So might as well write some of it down for the next person.

author	Matt Harbison <matt_harbison@yahoo.com>
date	Thu, 23 May 2019 22:50:11 -0400
parents	060c030c9993
children	326fdce22fb2

rev	line source
40978 5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	1 // ancestors.rs
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	2 //
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	3 // Copyright 2018 Georges Racinet <gracinet@anybox.fr>
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	4 //
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	5 // This software may be used and distributed according to the terms of the
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	6 // GNU General Public License version 2 or any later version.
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	7
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	8 //! Bindings for the `hg::ancestors` module provided by the
40978 5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	9 //! `hg-core` crate. From Python, this will be seen as `rustext.ancestor`
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	10 //! and can be used as replacement for the the pure `ancestor` Python module.
dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	11 //!
dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	12 //! # Classes visible from Python:
dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	13 //! - [`LazyAncestors`] is the Rust implementation of
41243 5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	14 //! `mercurial.ancestor.lazyancestors`. The only difference is that it is
5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	15 //! instantiated with a C `parsers.index` instance instead of a parents
5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	16 //! function.
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	17 //!
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	18 //! - [`MissingAncestors`] is the Rust implementation of
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	19 //! `mercurial.ancestor.incrementalmissingancestors`.
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	20 //!
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	21 //! API differences:
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	22 //! + it is instantiated with a C `parsers.index`
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	23 //! instance instead of a parents function.
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	24 //! + `MissingAncestors.bases` is a method returning a tuple instead of
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	25 //! a set-valued attribute. We could return a Python set easily if our
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	26 //! [PySet PR](https://github.com/dgrunwald/rust-cpython/pull/165)
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	27 //! is accepted.
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	28 //!
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	29 //! - [`AncestorsIterator`] is the Rust counterpart of the
41243 5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	30 //! `ancestor._lazyancestorsiter` Python generator. From Python, instances of
5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	31 //! this should be mainly obtained by calling `iter()` on a [`LazyAncestors`]
5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	32 //! instance.
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	33 //!
dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	34 //! [`LazyAncestors`]: struct.LazyAncestors.html
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	35 //! [`MissingAncestors`]: struct.MissingAncestors.html
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	36 //! [`AncestorsIterator`]: struct.AncestorsIterator.html
41704 060c030c9993 rust-cpython: moved py_set() utility to conversion module Georges Racinet <georges.racinet@octobus.net> parents: 41246 diff changeset	37 use crate::conversion::{py_set, rev_pyiter_collect};
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	38 use cindex::Index;
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	39 use cpython::{
41243 5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	40 ObjectProtocol, PyClone, PyDict, PyList, PyModule, PyObject, PyResult,
41704 060c030c9993 rust-cpython: moved py_set() utility to conversion module Georges Racinet <georges.racinet@octobus.net> parents: 41246 diff changeset	41 Python, PythonObject, ToPyObject,
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	42 };
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	43 use exceptions::GraphError;
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	44 use hg::Revision;
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	45 use hg::{
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	46 AncestorsIterator as CoreIterator, LazyAncestors as CoreLazy,
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	47 MissingAncestors as CoreMissing,
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	48 };
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	49 use std::cell::RefCell;
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	50 use std::collections::HashSet;
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	51
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	52 py_class!(pub class AncestorsIterator \|py\| {
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	53 data inner: RefCell<Box<CoreIterator<Index>>>;
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	54
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	55 def __next__(&self) -> PyResult<Option<Revision>> {
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	56 match self.inner(py).borrow_mut().next() {
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	57 Some(Err(e)) => Err(GraphError::pynew(py, e)),
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	58 None => Ok(None),
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	59 Some(Ok(r)) => Ok(Some(r)),
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	60 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	61 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	62
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	63 def __contains__(&self, rev: Revision) -> PyResult<bool> {
41186 1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	64 self.inner(py).borrow_mut().contains(rev)
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	65 .map_err(\|e\| GraphError::pynew(py, e))
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	66 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	67
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	68 def __iter__(&self) -> PyResult<Self> {
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	69 Ok(self.clone_ref(py))
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	70 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	71
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	72 def __new__(_cls, index: PyObject, initrevs: PyObject, stoprev: Revision,
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	73 inclusive: bool) -> PyResult<AncestorsIterator> {
41187 3bf6979a1785 rust-cpython: generalised conversion function Georges Racinet <georges.racinet@octobus.net> parents: 41186 diff changeset	74 let initvec: Vec<Revision> = rev_pyiter_collect(py, &initrevs)?;
41186 1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	75 let ait = CoreIterator::new(
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	76 Index::new(py, index)?,
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	77 initvec,
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	78 stoprev,
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	79 inclusive,
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	80 )
1b4b94bac8a0 rust-cpython: style consistency leftovers Georges Racinet <georges.racinet@octobus.net> parents: 41185 diff changeset	81 .map_err(\|e\| GraphError::pynew(py, e))?;
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	82 AncestorsIterator::from_inner(py, ait)
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	83 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	84
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	85 });
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	86
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	87 impl AncestorsIterator {
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	88 pub fn from_inner(py: Python, ait: CoreIterator<Index>) -> PyResult<Self> {
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	89 Self::create_instance(py, RefCell::new(Box::new(ait)))
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	90 }
d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	91 }
40978 5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	92
41184 dcf818267bc1 rust-cpython: rustdoc improvements Georges Racinet <georges.racinet@octobus.net> parents: 41117 diff changeset	93 py_class!(pub class LazyAncestors \|py\| {
41117 b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	94 data inner: RefCell<Box<CoreLazy<Index>>>;
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	95
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	96 def __contains__(&self, rev: Revision) -> PyResult<bool> {
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	97 self.inner(py)
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	98 .borrow_mut()
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	99 .contains(rev)
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	100 .map_err(\|e\| GraphError::pynew(py, e))
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	101 }
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	102
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	103 def __iter__(&self) -> PyResult<AncestorsIterator> {
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	104 AncestorsIterator::from_inner(py, self.inner(py).borrow().iter())
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	105 }
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	106
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	107 def __bool__(&self) -> PyResult<bool> {
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	108 Ok(!self.inner(py).borrow().is_empty())
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	109 }
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	110
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	111 def __new__(_cls, index: PyObject, initrevs: PyObject, stoprev: Revision,
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	112 inclusive: bool) -> PyResult<Self> {
41187 3bf6979a1785 rust-cpython: generalised conversion function Georges Racinet <georges.racinet@octobus.net> parents: 41186 diff changeset	113 let initvec: Vec<Revision> = rev_pyiter_collect(py, &initrevs)?;
41117 b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	114
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	115 let lazy =
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	116 CoreLazy::new(Index::new(py, index)?, initvec, stoprev, inclusive)
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	117 .map_err(\|e\| GraphError::pynew(py, e))?;
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	118
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	119 Self::create_instance(py, RefCell::new(Box::new(lazy)))
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	120 }
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	121
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	122 });
b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	123
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	124 py_class!(pub class MissingAncestors \|py\| {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	125 data inner: RefCell<Box<CoreMissing<Index>>>;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	126
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	127 def __new__(_cls, index: PyObject, bases: PyObject) -> PyResult<MissingAncestors> {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	128 let bases_vec: Vec<Revision> = rev_pyiter_collect(py, &bases)?;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	129 let inner = CoreMissing::new(Index::new(py, index)?, bases_vec);
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	130 MissingAncestors::create_instance(py, RefCell::new(Box::new(inner)))
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	131 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	132
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	133 def hasbases(&self) -> PyResult<bool> {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	134 Ok(self.inner(py).borrow().has_bases())
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	135 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	136
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	137 def addbases(&self, bases: PyObject) -> PyResult<PyObject> {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	138 let mut inner = self.inner(py).borrow_mut();
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	139 let bases_vec: Vec<Revision> = rev_pyiter_collect(py, &bases)?;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	140 inner.add_bases(bases_vec);
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	141 // cpython doc has examples with PyResult<()> but this gives me
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	142 // the trait `cpython::ToPyObject` is not implemented for `()`
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	143 // so let's return an explicit None
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	144 Ok(py.None())
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	145 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	146
41243 5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	147 def bases(&self) -> PyResult<PyObject> {
5257e6299d4c rust-cpython: set conversion for MissingAncestors.bases() Georges Racinet <georges.racinet@octobus.net> parents: 41240 diff changeset	148 py_set(py, self.inner(py).borrow().get_bases())
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	149 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	150
41246 619ee4039bd4 rust: MissingAncestors.basesheads() Georges Racinet <georges.racinet@octobus.net> parents: 41243 diff changeset	151 def basesheads(&self) -> PyResult<PyObject> {
619ee4039bd4 rust: MissingAncestors.basesheads() Georges Racinet <georges.racinet@octobus.net> parents: 41243 diff changeset	152 let inner = self.inner(py).borrow();
619ee4039bd4 rust: MissingAncestors.basesheads() Georges Racinet <georges.racinet@octobus.net> parents: 41243 diff changeset	153 py_set(py, &inner.bases_heads().map_err(\|e\| GraphError::pynew(py, e))?)
619ee4039bd4 rust: MissingAncestors.basesheads() Georges Racinet <georges.racinet@octobus.net> parents: 41243 diff changeset	154 }
619ee4039bd4 rust: MissingAncestors.basesheads() Georges Racinet <georges.racinet@octobus.net> parents: 41243 diff changeset	155
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	156 def removeancestorsfrom(&self, revs: PyObject) -> PyResult<PyObject> {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	157 let mut inner = self.inner(py).borrow_mut();
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	158 // this is very lame: we convert to a Rust set, update it in place
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	159 // and then convert back to Python, only to have Python remove the
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	160 // excess (thankfully, Python is happy with a list or even an iterator)
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	161 // Leads to improve this:
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	162 // - have the CoreMissing instead do something emit revisions to
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	163 // discard
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	164 // - define a trait for sets of revisions in the core and implement
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	165 // it for a Python set rewrapped with the GIL marker
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	166 let mut revs_pyset: HashSet<Revision> = rev_pyiter_collect(py, &revs)?;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	167 inner.remove_ancestors_from(&mut revs_pyset)
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	168 .map_err(\|e\| GraphError::pynew(py, e))?;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	169
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	170 // convert as Python list
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	171 let mut remaining_pyint_vec: Vec<PyObject> = Vec::with_capacity(
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	172 revs_pyset.len());
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	173 for rev in revs_pyset {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	174 remaining_pyint_vec.push(rev.to_py_object(py).into_object());
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	175 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	176 let remaining_pylist = PyList::new(py, remaining_pyint_vec.as_slice());
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	177 revs.call_method(py, "intersection_update", (remaining_pylist, ), None)
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	178 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	179
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	180 def missingancestors(&self, revs: PyObject) -> PyResult<PyList> {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	181 let mut inner = self.inner(py).borrow_mut();
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	182 let revs_vec: Vec<Revision> = rev_pyiter_collect(py, &revs)?;
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	183 let missing_vec = match inner.missing_ancestors(revs_vec) {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	184 Ok(missing) => missing,
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	185 Err(e) => {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	186 return Err(GraphError::pynew(py, e));
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	187 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	188 };
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	189 // convert as Python list
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	190 let mut missing_pyint_vec: Vec<PyObject> = Vec::with_capacity(
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	191 missing_vec.len());
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	192 for rev in missing_vec {
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	193 missing_pyint_vec.push(rev.to_py_object(py).into_object());
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	194 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	195 Ok(PyList::new(py, missing_pyint_vec.as_slice()))
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	196 }
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	197 });
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	198
006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	199 /// Create the module, with __package__ given from parent
40978 5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	200 pub fn init_module(py: Python, package: &str) -> PyResult<PyModule> {
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	201 let dotted_name = &format!("{}.ancestor", package);
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	202 let m = PyModule::new(py, dotted_name)?;
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	203 m.add(py, "__package__", package)?;
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	204 m.add(
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	205 py,
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	206 "__doc__",
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	207 "Generic DAG ancestor algorithms - Rust implementation",
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	208 )?;
41056 d9f439fcdb4c rust-cpython: binding for AncestorsIterator Georges Racinet <gracinet@anybox.fr> parents: 40978 diff changeset	209 m.add_class::<AncestorsIterator>(py)?;
41117 b31a41f24864 rust-cpython: binding for LazyAncestors Georges Racinet <gracinet@anybox.fr> parents: 41056 diff changeset	210 m.add_class::<LazyAncestors>(py)?;
41188 006c9ce486fa rust-cpython: bindings for MissingAncestors Georges Racinet <georges.racinet@octobus.net> parents: 41187 diff changeset	211 m.add_class::<MissingAncestors>(py)?;
40978 5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	212
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	213 let sys = PyModule::import(py, "sys")?;
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	214 let sys_modules: PyDict = sys.get(py, "modules")?.extract(py)?;
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	215 sys_modules.set_item(py, dotted_name, &m)?;
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	216 // Example C code (see pyexpat.c and import.c) will "give away the
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	217 // reference", but we won't because it will be consumed once the
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	218 // Rust PyObject is dropped.
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	219 Ok(m)
5532823e8c18 rust-cpython: start cpython crate bindings Georges Racinet <gracinet@anybox.fr> parents: diff changeset	220 }

Mercurial > public > mercurial-scm > hg-stable

annotate rust/hg-cpython/src/ancestors.rs @ 42172:c3484ddbdb96 stable 5.0.1