''' Converter classes that take StimCollection classes and return their
constituent elements as iterables. '''
from pliers.stimuli.video import VideoStim, VideoFrameCollectionStim
from pliers.stimuli.image import ImageStim
from pliers.stimuli.text import ComplexTextStim, TextStim
from .base import Converter
class StimCollectionIterator(Converter):
''' Base class for all StimCollectionIterators. '''
VERSION = '1.0'
def _convert(self, stim):
return stim.__iter__()
[docs]class VideoFrameIterator(StimCollectionIterator):
''' Iterates frames in a VideoStim as ImageStims. '''
_input_type = VideoStim
_output_type = ImageStim
[docs]class VideoFrameCollectionIterator(StimCollectionIterator):
''' Iterates frames in a DerivedVideoStim as ImageStims. '''
# TODO: use VideoFrameIterator for both VideoStim and DerivedVideoStim,
# but this may require reworking _input_type to handle disjunction rather
# than the current conjunction, or making the get_converter() code walk
# up the hierarchy and use superclass iterators.
_input_type = VideoFrameCollectionStim
_output_type = ImageStim
[docs]class ComplexTextIterator(StimCollectionIterator):
''' Iterates elements in a ComplexTextStim as TextStims. '''
_input_type = ComplexTextStim
_output_type = TextStim