menu.py

"""
Dillo Tools - menu
"""


import dilloTools
import logging
from pymel.core import *


LOG = logging.getLogger('dilloTools')


__MENU_NAME__ = 'dilloToolsMenu'

def createMenu(tools, p=None):
    """Create and populate the Dillo Tools menu.
    This menu can be created in alternate locations by
    passing a menu or window to the p parameter"""
    LOG.debug('Building Menu...')
    if p is None:
        if '$gMainWindow' in melGlobals.keys():
            p = melGlobals['gMainWindow']
        else:
            LOG.warning('cannot add Dillo Tools menu, gMainWindow not found')
            return
    
    setParent(p)
    longName = '{0}|{1}'.format(p, __MENU_NAME__)
    if menu(longName, ex=True):
        deleteUI(__MENU_NAME__)
    
    with menu(__MENU_NAME__, l='Dillo Tools', p=p, to=True):
        createMenuTools(tools)
        createGuiShortcuts()
        menuItem(d=True)
        menuItem(l='v{0}'.format(dilloTools.getVersion()), en=False)


def createMenuTools(tools):
    for cat in tools['[cats]']:
        LOG.debug('  Adding Submenu: {0}'.format(cat))
        with menuItem(l=cat, subMenu=True, to=True):
            for tool in tools[cat]['[tools]']:
                toolData = tools[cat][tool]
                stp = toolData['sourceType']
                itemKargs = {
                    'ann':toolData['annotation'],
                    'command':toolData['command'],
                    'image':toolData['image'],
                    'label':tool,
                }
                LOG.debug('    Adding Tool: {0}'.format(tool))
                dilloMenuItem(stp, itemKargs)


def dilloMenuItem(stp, kwargs):
    """Create a menu item using mel or python depending on stp."""
    if stp == 'mel':
        kwargs['command'] = 'mel.eval("{0}")'.format(kwargs['command'])
    menuItem(**kwargs)


def createGuiShortcuts():
    menuItem(d=True)
    menuItem(l='Window', c=Callback(dilloTools.createWindow))
    with menuItem(l='Shelf', sm=True):
        menuItem(l='Build', c=Callback(dilloTools.createShelf))
        menuItem(l='Delete', c=Callback(dilloTools.deleteShelf))