org.eclipse.papyrus.dgts.wizard.editor.providers.DgtsDiagramDefinitionLabelProvider.java Source code

Introduction

Here is the source code for org.eclipse.papyrus.dgts.wizard.editor.providers.DgtsDiagramDefinitionLabelProvider.java

Source

/*******************************************************************************
 * Copyright  2013 Atos.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *   
 * Contributors:
 * Vincent Lartigaut (Atos) vincent.lartigaut@atos.net - Vincent Lartigaut - initial API and implementation
 * Guilhem Desq (Atos) guilhem.desq@atos.net -  Guilhem Desq - initial API and implementation
 ******************************************************************************/

package org.eclipse.papyrus.dgts.wizard.editor.providers;

import java.util.Collection;

import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.papyrus.commands.CreationCommandDescriptor;
import org.eclipse.papyrus.commands.CreationCommandRegistry;
import org.eclipse.papyrus.commands.ICreationCommandRegistry;
import org.eclipse.papyrus.infra.core.editor.BackboneException;
import org.eclipse.swt.graphics.Image;

import DiagramGlobalToolService.DiagramDefinition;

/**
 * Label provider for Diagram
 * 
 * @author gdesq
 * 
 */
public class DgtsDiagramDefinitionLabelProvider extends LabelProvider {

    public Image getImage(Object object) {
        if (object instanceof DiagramDefinition) {
            DiagramDefinition diagram = (DiagramDefinition) object;
            ICreationCommandRegistry creationCommandRegistry = CreationCommandRegistry
                    .getInstance(org.eclipse.papyrus.infra.core.Activator.PLUGIN_ID);

            Collection<CreationCommandDescriptor> descriptors = creationCommandRegistry.getCommandDescriptors();

            for (CreationCommandDescriptor descriptor : descriptors) {
                try {
                    if (descriptor.getCommand().getCreatedDiagramType().equals(diagram.getDiagramType())) {
                        ImageDescriptor image = (descriptor.getIcon());
                        // image is an optional attribute
                        if (image != null) {
                            return new Image(null, image.getImageData());
                        }

                    }
                } catch (BackboneException e) {
                    // Do nothing
                }

            }

        }
        return null;
    }

    public String getText(Object object) {
        if (object instanceof DiagramDefinition) {
            return ((DiagramDefinition) object).getDiagramType();
        }
        return null;
    }
}