Source code for molsysmt.form.file_bcif_gz.iterators

from molsysmt._private.exceptions import NotImplementedIteratorError
from ..molsysmt_MolSys.iterators import StructuresIterator as StructuresIterator_molsysmt_MolSys
from ..molsysmt_MolSys.iterators import TopologyIterator as TopologyIterator_molsysmt_MolSys
from .to_molsysmt_MolSys import to_molsysmt_MolSys
from molsysmt._private.digestion import digest



class StructuresIterator(StructuresIterator_molsysmt_MolSys):



    @digest(form='file:bcif.gz')
    def __init__(self, molecular_system, atom_indices='all', start=0, step=1, stop=None, chunk=1, structure_indices=None,
            output_type='values', skip_digestion=False, **kwargs):

        molecular_system = to_molsysmt_MolSys(molecular_system, skip_digestion=True)

        super().__init__(molecular_system, atom_indices=atom_indices, start=start, step=step, stop=stop,
                         chunk=chunk, structure_indices=structure_indices, output_type=output_type,
                         skip_digestion=True, **kwargs)





class TopologyIterator(TopologyIterator_molsysmt_MolSys):



    @digest(form='file:bcif.gz')
    def __init__(self, molecular_system, element='atom', indices='all', start=0, step=1, stop=None, chunk=1,
            output_type='values', skip_digestion=False, **kwargs):

        molecular_system = to_molsysmt_MolSys(molecular_system)

        super().__init__(molecular_system, element=element, indices=indices, start=start, step=step, stop=stop,
                chunk=chunk, output_type=output_type, skip_digestion=True, **kwargs)