molframe-query 0.6.1

Typed and textual structural selection plans for molframe
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! Narrowing a scan with the statistics every chunk already carries.
//!
//! Each chunk records which elements it holds. A predicate that asks for one
//! element can therefore skip every chunk whose summary excludes it, without
//! reading a single atom of it. Only statistics that are exact are used:
//! `resid` is not pruned, because a chunk's residue bounds are hierarchy row
//! indices, not author sequence identifiers.

use molframe_core::Element;
use molframe_core::selection::AtomSelection;
use molframe_core::structure::Structure;

/// The atoms of every chunk that may hold any of `elements`.
pub(crate) fn chunks_with_any_element(
    structure: &Structure,
    elements: &[Element],
) -> AtomSelection {
    AtomSelection::from_runs(
        structure
            .data()
            .chunks
            .iter()
            .filter(|chunk| {
                elements
                    .iter()
                    .any(|element| !chunk.stats().excludes_element(*element))
            })
            .map(molframe_core::AtomChunk::atoms),
    )
}

#[cfg(test)]
#[path = "prune_tests.rs"]
mod tests;