core/org.eclipse.ptp.ui/src/org/eclipse/ptp/internal/ui/adapters/ResourceManagerWorkbenchAdapter.java

/*******************************************************************************
 * Copyright (c) 2007 The Regents of the University of California. 
 * This material was produced under U.S. Government contract W-7405-ENG-36 
 * for Los Alamos National Laboratory, which is operated by the University 
 * of California for the U.S. Department of Energy. The U.S. Government has 
 * rights to use, reproduce, and distribute this software. NEITHER THE 
 * GOVERNMENT NOR THE UNIVERSITY MAKES ANY WARRANTY, EXPRESS OR IMPLIED, OR 
 * ASSUMES ANY LIABILITY FOR THE USE OF THIS SOFTWARE. If software is modified 
 * to produce derivative works, such modified software should be clearly marked, 
 * so as not to confuse it with the version available from LANL.
 * 
 * Additionally, 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
 * 
 * LA-CC 04-115
 *******************************************************************************/
package org.eclipse.ptp.internal.ui.adapters;

import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.ptp.core.elementcontrols.IResourceManagerControl;
import org.eclipse.ptp.core.elements.IPMachine;
import org.eclipse.ptp.core.elements.IPQueue;
import org.eclipse.ptp.core.elements.IResourceManager;
import org.eclipse.ptp.core.elements.attributes.ResourceManagerAttributes;
import org.eclipse.ptp.internal.ui.ParallelImages;
import org.eclipse.ptp.ui.IRuntimeModelPresentation;
import org.eclipse.ptp.ui.PTPUIPlugin;
import org.eclipse.ptp.ui.messages.Messages;
import org.eclipse.ptp.utils.ui.ImageImageDescriptor;
import org.eclipse.swt.graphics.Image;
import org.eclipse.ui.model.IWorkbenchAdapter;
import org.eclipse.ui.model.IWorkbenchAdapter2;
import org.eclipse.ui.model.WorkbenchAdapter;

public class ResourceManagerWorkbenchAdapter extends WorkbenchAdapter {

	private class ChildContainer extends WorkbenchAdapter implements IAdaptable {

		private final Object parent;

		private final String name;

		private final Object[] children;

		public ChildContainer(Object parent, String name, Object[] children) {
			this.parent = parent;
			this.name = name;
			this.children = children;
		}

		public Object getAdapter(@SuppressWarnings("unchecked") Class adapter) {
			if (adapter == IWorkbenchAdapter.class || adapter == IWorkbenchAdapter2.class) {
				return this;
			}
			return null;
		}

		/* (non-Javadoc)
		 * @see org.eclipse.ui.model.WorkbenchAdapter#getChildren(java.lang.Object)
		 */
		@Override
		public Object[] getChildren(Object object) {
			return children.clone();
		}

		/* (non-Javadoc)
		 * @see org.eclipse.ui.model.WorkbenchAdapter#getLabel(java.lang.Object)
		 */
		@Override
		public String getLabel(Object object) {
			return name;
		}

		/* (non-Javadoc)
		 * @see org.eclipse.ui.model.WorkbenchAdapter#getParent(java.lang.Object)
		 */
		@Override
		public Object getParent(Object object) {
			return parent;
		}
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ui.model.WorkbenchAdapter#getChildren(java.lang.Object)
	 */
	@Override
	public Object[] getChildren(Object parentElement) {
		IResourceManager rm = getResourceManager(parentElement);
		if (rm != null) {
			if (rm.getState() != ResourceManagerAttributes.State.STARTED) {
				return new Object[0];
			}
			IPMachine[] machines = rm.getMachines();
			IPQueue[] queues = rm.getQueues();
			return new Object[] {
					makeChildContainer(parentElement, Messages.ResourceManagerWorkbenchAdapter_0, machines),
					makeChildContainer(parentElement, Messages.ResourceManagerWorkbenchAdapter_1, queues) };
		}
		return null;
	}

	/**
	 * @param parentElement
	 * @return
	 */
	private IResourceManager getResourceManager(Object parentElement) {
		IResourceManager rm = null;
		if (parentElement instanceof IAdaptable) {
			rm = (IResourceManager) ((IAdaptable) parentElement).getAdapter(IResourceManager.class);
		}
		return rm;
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ui.model.WorkbenchAdapter#getImageDescriptor(java.lang.Object)
	 */
	@Override
	public ImageDescriptor getImageDescriptor(Object object) {
		final IRuntimeModelPresentation presentation = PTPUIPlugin.getDefault().getRuntimeModelPresentation(((IResourceManager)object).getResourceManagerId());
		if (presentation != null) {
			final Image image = presentation.getImage(object);
			if (image != null) {
				return new ImageImageDescriptor(image);
			}
		}
		final ResourceManagerAttributes.State state = ((IResourceManager) object).getState();
		return new ImageImageDescriptor(ParallelImages.rmImages[state.ordinal()]);
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ui.model.WorkbenchAdapter#getLabel(java.lang.Object)
	 */
	@Override
	public String getLabel(Object object) {
		final IRuntimeModelPresentation presentation = PTPUIPlugin.getDefault().getRuntimeModelPresentation(((IResourceManager)object).getResourceManagerId());
		if (presentation != null) {
			final String label = presentation.getText(object);
			if (label != null) {
				return label;
			}
		}
		final IResourceManagerControl resourceManager = (IResourceManagerControl) object;
		final String type = resourceManager.getConfiguration().getType();
		if (type == null) {
			return resourceManager.getName();
		}
		return resourceManager.getName() + " (" + type + ")"; //$NON-NLS-1$ //$NON-NLS-2$
	}
	
	/* (non-Javadoc)
	 * @see org.eclipse.ui.model.WorkbenchAdapter#getParent(java.lang.Object)
	 */
	@Override
	public Object getParent(Object object) {
		IResourceManagerControl rm = (IResourceManagerControl) getResourceManager(object);
		if (rm != null) {
			return rm.getParent();
		}
		return null;
	}

	private ChildContainer makeChildContainer(Object parent, String name,
			final Object[] children) {
		final ChildContainer container = new ChildContainer(parent, name,
				children);
		return container;
	}

}