Source code for zBuilder.builders.deltaMush

from zBuilder.builder import Builder
import zBuilder.zMaya as mz

import maya.cmds as mc

import logging

logger = logging.getLogger(__name__)


class DeltaMush(Builder):
    """To capture a deltaMush
    """

    def __init__(self):
        Builder.__init__(self)

    @Builder.time_this
    def retrieve_from_scene(self, *args, **kwargs):
        # parse args------------------------------------------------------------
        selection = mz.parse_maya_node_for_selection(args)

        # kwargs----------------------------------------------------------------
        get_mesh = kwargs.get('get_mesh', True)
        get_maps = kwargs.get('get_map_names', True)

        hist = mc.listHistory(selection)
        delta_mushes = mc.ls(hist, type='deltaMush')[::-1]

        if not delta_mushes:
            raise StandardError('No delta mushes found, aborting!')

        for delta_mush in delta_mushes:
            parameter = self.node_factory(delta_mush)
            self.bundle.add_parameter(parameter)
        self.stats()

    @Builder.time_this
    def build(self, *args, **kwargs):
        logger.info('Applying deltaMush....')
        attr_filter = kwargs.get('attr_filter', None)
        interp_maps = kwargs.get('interp_maps', 'auto')
        name_filter = kwargs.get('name_filter', list())

        parameters = self.get_scene_items(name_filter=name_filter,
                                         type_filter='deltaMush')
        for parameter in parameters:
            parameter.build(attr_filter=attr_filter, interp_maps=interp_maps)