Source code for molsysmt.form.file_mmtf.iterators

from molsysmt._private.exceptions import NotImplementedIteratorError
from ..mmtf_MMTFDecoder.iterators import StructuresIterator as StructuresIterator_mmtf_MMTFDecoder
from ..mmtf_MMTFDecoder.iterators import TopologyIterator as TopologyIterator_mmtf_MMTFDecoder
from .to_mmtf_MMTFDecoder import to_mmtf_MMTFDecoder
from molsysmt._private.digestion import digest



class StructuresIterator(StructuresIterator_mmtf_MMTFDecoder):



    @digest(form='file:mmtf')
    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_mmtf_MMTFDecoder(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_mmtf_MMTFDecoder):



    @digest(form='file:mmtf')
    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_mmtf_MMTFDecoder(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)