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.IModelManager;
import org.eclipse.ptp.core.PTPCorePlugin;
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.rmsystem.IResourceManagerFactory;
import org.eclipse.ptp.ui.UIMessage;
import org.eclipse.ptp.ui.utils.ImageImageDescriptor;
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,
							UIMessage.getResourceString("ResourceManagerView.Machines"), machines), //$NON-NLS-1$
					makeChildContainer(parentElement,
							UIMessage.getResourceString("ResourceManagerView.Queues"), queues) }; //$NON-NLS-1$
		}
		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) {
		ResourceManagerAttributes.State status = ((IResourceManager) object).getState();
		return new ImageImageDescriptor(ParallelImages.rmImages[status.ordinal()]);
	}

	/* (non-Javadoc)
	 * @see org.eclipse.ui.model.WorkbenchAdapter#getLabel(java.lang.Object)
	 */
	@Override
	public String getLabel(Object object) {
		final IResourceManagerControl resourceManager = (IResourceManagerControl) object;
		final IModelManager modelManager = PTPCorePlugin.getDefault().getModelManager();
		final String resourceManagerId = resourceManager.getConfiguration().getResourceManagerId();
		if (resourceManagerId == null)
			return resourceManager.getName();
		IResourceManagerFactory factory = modelManager.getResourceManagerFactory(
				resourceManagerId);
		return resourceManager.getName() + " (" + factory.getName() + ")"; //$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;
	}

}