Source code for mr_utils.gadgetron.configs.distributed

'''Example generation of distributed gadget configs.'''

from mr_utils.gadgetron import GadgetronConfig

def distributed_default():
    # pylint: disable=C0301
    '''Generates distributed_default.xml.

    Generates [1]_.

    References
    ==========
    .. [1] https://github.com/gadgetron/gadgetron/blob/master/gadgets/distributed/config/distributed_default.xml
    '''
    # pylint: enable=C0301

    config = GadgetronConfig()
    config.add_reader('1008', 'GadgetIsmrmrdAcquisitionMessageReader')
    config.add_reader('1022', 'MRIImageReader')
    config.add_writer('1022', 'MRIImageWriter')
    config.add_writer('1008', 'GadgetIsmrmrdAcquisitionMessageWriter')
    config.add_gadget(
        'Distribute',
        'IsmrmrdAcquisitionDistributeGadget',
        dll='gadgetron_distributed',
        props=[
            ('parallel_dimension', 'repetition'),
            ('use_this_node_for_compute', 'true')
        ])
    config.add_gadget('RemoveROOversampling')
    config.add_gadget('AccTrig', 'AcquisitionAccumulateTriggerGadget', props=[
        ('trigger_dimension', 'repetition'),
        ('sorting_dimension', 'slice')
    ])
    config.add_gadget('Buff', 'BucketToBufferGadget', props=[
        ('N_dimension', ''),
        ('S_dimension', ''),
        ('split_slices', 'true')
    ])
    config.add_gadget('SimpleRecon')
    config.add_gadget('ImageArraySplit')
    config.add_gadget('Collect', dll='gadgetron_distributed')
    config.add_gadget('Extract')
    config.add_gadget('Sort', 'ImageSortGadget', props=[
        ('sorting_dimension', 'repetition')
    ])
    config.add_gadget('ImageFinish')
    return config

def distributed_image_default():
    # pylint: disable=C0301
    '''Generates distributed_image_default.xml.

    Generates [2]_.

    References
    ==========
    .. [2] https://github.com/gadgetron/gadgetron/blob/master/gadgets/distributed/config/distributed_image_default.xml
    '''
    # pylint: enable=C0301

    config = GadgetronConfig()
    config.add_reader('1008', 'GadgetIsmrmrdAcquisitionMessageReader')
    config.add_reader('1022', 'MRIImageReader')
    config.add_writer('1022', 'MRIImageWriter')
    config.add_writer('1008', 'GadgetIsmrmrdAcquisitionMessageWriter')
    config.add_gadget('RemoveROOversampling')
    config.add_gadget('AccTrig', 'AcquisitionAccumulateTriggerGadget', props=[
        ('trigger_dimension', 'repetition'),
        ('sorting_dimension', 'slice')
    ])
    config.add_gadget('Buff', 'BucketToBufferGadget', props=[
        ('N_dimension', ''),
        ('S_dimension', ''),
        ('split_slices', 'true')
    ])
    config.add_gadget('SimpleRecon')
    config.add_gadget('ImageArraySplit')

    config.add_gadget(
        'Distribute',
        'IsmrmrdImageDistributeGadget',
        dll='gadgetron_distributed',
        props=[
            ('parallel_dimension', 'repetition'),
            ('use_this_node_for_compute', 'true'),
            ('single_package_mode', 'true')
        ])
    config.add_gadget('Extract')
    config.add_gadget('Collect', dll='gadgetron_distributed')
    config.add_gadget('ImageSort', props=[
        ('sorting_dimension', 'repetition')
    ])
    config.add_gadget('ImageFinish')
    return config

if __name__ == '__main__':
    pass