TogglePopulateDiagramFromTreeSelectionAction.java example

Explorer
teiid-designer-master
/*
 * JBoss, Home of Professional Open Source.
 *
 * See the LEGAL.txt file distributed with this work for information regarding copyright ownership and licensing.
 *
 * See the AUTHORS.txt file distributed with this work for a full listing of individual contributors.
 */
package org.teiid.designer.mapping.ui.actions;

import org.eclipse.jface.viewers.ISelection;
import org.eclipse.swt.SWT;
import org.eclipse.ui.IWorkbenchPart;
import org.teiid.designer.diagram.ui.editor.DiagramEditor;
import org.teiid.designer.mapping.ui.UiConstants;
import org.teiid.designer.mapping.ui.UiPlugin;
import org.teiid.designer.mapping.ui.diagram.MappingDiagramUtil;
import org.teiid.designer.mapping.ui.editor.MappingDiagramBehavior;
import org.teiid.designer.ui.viewsupport.ModelIdentifier;



/**
 * TogglePopulateDiagramFromTreeSelectionAction
 *
 * @since 8.0
 */
public class TogglePopulateDiagramFromTreeSelectionAction extends MappingAction {

    private static String POPULATE_DIAGRAM_FROM_TREE_SELECTION_TOOLTIP 
        = UiConstants.Util.getString( "TogglePopulateDiagramFromTreeSelectionAction.action.tooltip" );  //$NON-NLS-1$
    private static String POPULATE_DIAGRAM_FROM_TREE_SELECTION_TEXT 
        = UiConstants.Util.getString( "TogglePopulateDiagramFromTreeSelectionAction.action.text" );  //$NON-NLS-1$
    private boolean logicalModel = false;
    
    ///////////////////////////////////////////////////////////////////////////////////////////////
    // CONSTRUCTORS
    ///////////////////////////////////////////////////////////////////////////////////////////////

    public TogglePopulateDiagramFromTreeSelectionAction() {
        super( UiPlugin.getDefault(), SWT.TOGGLE );

        setImageDescriptor( UiPlugin.getDefault().getImageDescriptor( UiConstants.Images.POPULATE_DIAGRAM_FROM_TREE_SELECTION ) );                        
        setToolTipText( POPULATE_DIAGRAM_FROM_TREE_SELECTION_TOOLTIP );
        setText( POPULATE_DIAGRAM_FROM_TREE_SELECTION_TEXT );
    }
    

    ///////////////////////////////////////////////////////////////////////////////////////////////
    // METHODS
    ///////////////////////////////////////////////////////////////////////////////////////////////
    
    @Override
    public void setDiagramEditor( DiagramEditor editor ) {
        super.setDiagramEditor( editor );
                
        // update tooltip if logical model type
        boolean logical = ModelIdentifier.isLogicalModel(editor.getCurrentModelResource());

        if (this.logicalModel != logical) {
            this.logicalModel = logical;
            String key = (logical ? "TogglePopulateDiagramFromTreeSelectionAction.logicalModel.action.tooltip" //$NON-NLS-1$
                                  : POPULATE_DIAGRAM_FROM_TREE_SELECTION_TOOLTIP);
            setToolTipText(UiConstants.Util.getString(key));
            key = (logical ? "TogglePopulateDiagramFromTreeSelectionAction.logicalModel.action.text" //$NON-NLS-1$
                            : POPULATE_DIAGRAM_FROM_TREE_SELECTION_TEXT);
            setText(UiConstants.Util.getString(key));
        }

        // update the button state
        updateButtonState();
    }

    /* (non-Javadoc)
     * @see org.eclipse.ui.ISelectionListener#selectionChanged(IWorkbenchPart, ISelection)
     */
    @Override
    public void selectionChanged(IWorkbenchPart thePart, ISelection theSelection) {

        super.selectionChanged(thePart, theSelection);
        determineEnablement();
    }

    
    
    private MappingDiagramBehavior getBehavior() {
        return MappingDiagramUtil.getCurrentMappingDiagramBehavior();
    }
    
    /* (non-Javadoc)
     * @see org.eclipse.jface.action.IAction#run()
     */
    @Override
    protected void doRun() {
        
        // get current state
        boolean bPopulateDiagramFromTreeSelection = getBehavior().getPopulateDiagramFromTreeSelectionState();
               
        // reverse it
        bPopulateDiagramFromTreeSelection = !bPopulateDiagramFromTreeSelection;
        
        getBehavior().setPopulateDiagramFromTreeSelectionState( bPopulateDiagramFromTreeSelection );

        // update the button
        updateButtonState();
        
        // refresh the diagram...
        editor.doRefreshDiagram();        
    }

    private void determineEnablement() {

        /*
         * jhTODO A refinement would be to disable this action when the tree is fully expanded
         */
        // always enable
        boolean enable = true;

        setEnabled(enable);
    }
    

    protected void updateButtonState() {

        setChecked( getBehavior().getPopulateDiagramFromTreeSelectionState() );
    }
    
}