Source code for bbprep._internal.selectors.all_selector

import stk

from .selector import Selector



[docs]
class AllSelector(Selector):
    """Select all atom ids."""


[docs]
    def select_atoms(self, molecule: stk.BuildingBlock) -> tuple[int, ...]:
        return tuple(i.get_id() for i in molecule.get_atoms())





[docs]
class AllNonHSelector(Selector):
    """Select all atom ids."""


[docs]
    def select_atoms(self, molecule: stk.BuildingBlock) -> tuple[int, ...]:
        return tuple(
            i.get_id()
            for i in molecule.get_atoms()
            if i.get_atomic_number() != 1
        )