core/org.eclipse.ptp.ui/src/org/eclipse/ptp/ui/managers/MachineManager.java - ptp/org.eclipse.ptp - Git at Google

 /*******************************************************************************
  * Copyright (c) 2005 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.ui.managers;

 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;

 import org.eclipse.ptp.core.elements.IPElement;
 import org.eclipse.ptp.core.elements.IPMachine;
 import org.eclipse.ptp.core.elements.IPNode;
 import org.eclipse.ptp.core.elements.IPProcess;
 import org.eclipse.ptp.core.elements.IPUniverse;
 import org.eclipse.ptp.core.elements.IResourceManager;
 import org.eclipse.ptp.core.elements.attributes.NodeAttributes;
 import org.eclipse.ptp.core.elements.attributes.ProcessAttributes;
 import org.eclipse.ptp.internal.ui.ParallelImages;
 import org.eclipse.ptp.ui.IMachineManager;
 import org.eclipse.ptp.ui.IRuntimeModelPresentation;
 import org.eclipse.ptp.ui.PTPUIPlugin;
 import org.eclipse.ptp.ui.messages.Messages;
 import org.eclipse.ptp.ui.model.Element;
 import org.eclipse.ptp.ui.model.ElementHandler;
 import org.eclipse.ptp.ui.model.IElement;
 import org.eclipse.ptp.ui.model.IElementHandler;
 import org.eclipse.ptp.ui.model.IElementSet;
 import org.eclipse.swt.graphics.Image;

 /**
  * @author clement chu
  *
  */
 public class MachineManager extends AbstractElementManager implements IMachineManager {
 	private Map<String, IPMachine> machineList = new HashMap<String, IPMachine>();
 	protected IPMachine cur_machine = null;
 	protected final String DEFAULT_TITLE = Messages.MachineManager_0;

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#addMachine(org.eclipse.ptp.core.elements.IPMachine)
 	 */
 	public void addMachine(IPMachine mac) {
 		if (mac != null) {
 			IElementHandler handler;
 			if (!machineList.containsKey(mac.getID())) {
 				handler = new ElementHandler();
 				machineList.put(mac.getID(), mac);
 				setElementHandler(mac.getID(), handler);
 			} else {
 				handler = getElementHandler(mac.getID());
 			}
 			List<IElement> elements = new ArrayList<IElement>();
 			IElementSet set = handler.getSetRoot();
 			for (IPNode node : mac.getNodes()) {
 				elements.add(createNodeElement(set, node.getID(), node.getName(), node));
 			}
 			set.addElements(elements.toArray(new IElement[0]));
 		}
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#addNode(org.eclipse.ptp.core.elements.IPNode)
 	 */
 	public void addNode(IPNode node) {
 		addMachine(node.getMachine());
 		IElementHandler elementHandler = getElementHandler(node.getMachine().getID());
 		IElementSet set = elementHandler.getSetRoot();
 		set.addElements(new IElement[] { createNodeElement(set, node.getID(), node.getName(), node) });
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.managers.AbstractElementManager#clear()
 	 */
 	@Override
 	public void clear() {
 		if (machineList != null) {
 			machineList.clear();
 		}
 		super.clear();
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#findMachineById(java.lang.String)
 	 */
 	public IPMachine findMachineById(String id) {
 		return (IPMachine) machineList.get(id);
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#findNode(java.lang.String)
 	 */
 	public IPNode findNode(String id) {
 		IPMachine machine = getCurrentMachine();
 		if (machine == null) {
 			System.out.println(Messages.MachineManager_1);
 			return null;
 		}
 		return machine.getNodeById(id);
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#getCurrentMachine()
 	 */
 	public IPMachine getCurrentMachine() {
 		return cur_machine;
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#getFullyQualifiedName(java.lang.String)
 	 */
 	public String getFullyQualifiedName(String id) {
 		if (id.equals(EMPTY_ID)) {
 			return DEFAULT_TITLE;
 		}
 		//TODO check that this is what should happen. Can we just use cur_machine?
 		IPMachine machine = getCurrentMachine();
 		if (machine != null) {
 			IResourceManager rm = machine.getResourceManager();
 			if (rm != null) {
 				return rm.getName() + ": " + machine.getName(); //$NON-NLS-1$
 			}
 		}
 		return ""; //$NON-NLS-1$
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#getMachines()
 	 */
 	public IPMachine[] getMachines() {
 		return machineList.values().toArray(new IPMachine[0]);
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#getName(java.lang.String)
 	 */
 	public String getName(String id) {
 		if (cur_machine == null)
 			return ""; //$NON-NLS-1$
 		return cur_machine.getName();
 	}

 	/**
 	 * Get node status.
 	 *
 	 * Currently the node status is determined as follows:
 	 * 	- if the node is up:
 	 * 		- if there are *any* running processes on the node: NODE_RUNNING
 	 * 		- if there are no running processes, but one or more exited processes on the node: NODE_EXITED
 	 *  - if the node is down: NODE_DOWN
 	 *  - if the node is error: NODE_ERROR
 	 *
 	 *  TODO: in the future, the machine view should be linked to the jobs view. The node state should only
 	 *  be shown as NODE_RUNNING if any processes belonging to the current job are running.
 	 *
 	 * @param node
 	 * @return
 	 */
 	private int getNodeState(IPNode node) {
 		if (node != null) {
 			NodeAttributes.State nodeState = node.getState();

 			if (nodeState == NodeAttributes.State.UP) {
 				IPProcess[] procs = node.getProcesses();
 				if (procs.length > 0) {
 					for (IPProcess proc : procs) {
 						if (proc.getState() != ProcessAttributes.State.COMPLETED) {
 							return NodeAttributes.State.values().length;
 						}
 					}
 					return NodeAttributes.State.values().length + 1;
 				}

 			}

 			return nodeState.ordinal();
 		}
 		return NodeAttributes.State.UNKNOWN.ordinal();
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.managers.AbstractElementManager#getImage(org.eclipse.ptp.ui.model.IElement)
 	 */
 	public Image getImage(IElement element) {
 		IPMachine machine = getCurrentMachine();
 		if (machine != null) {
 			IResourceManager rm = machine.getResourceManager();
 			final IRuntimeModelPresentation presentation = PTPUIPlugin.getDefault().getRuntimeModelPresentation(rm.getResourceManagerId());
 			if (presentation != null) {
 				final Image image = presentation.getImage(element);
 				if (image != null) {
 					return image;
 				}
 			}
 			IPNode node = machine.getNodeById(element.getID());
 			return ParallelImages.nodeImages[getNodeState(node)][element.isSelected() ? 1 : 0];
 		}
 		return null;
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#initial()
 	 */
 	public IPElement initial(IPUniverse universe) {
 		for (IResourceManager rm : universe.getResourceManagers()) {
 			for (IPMachine machine : rm.getMachines()) {
 				addMachine(machine);
 			}
 		}

 		setCurrentSetId(IElementHandler.SET_ROOT_ID);
 		return null;
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#removeMachine(org.eclipse.ptp.core.elements.IPMachine)
 	 */
 	public void removeMachine(IPMachine machine) {
 		machineList.remove(machine.getID());
 		IElementHandler handler = getElementHandler(machine.getID());
 		if (handler != null) {
 			IElementSet set = handler.getSetRoot();
 			for (IPNode node : machine.getNodes()) {
 				IElement element = set.getElementByID(node.getID());
 				if (element != null) {
 					set.removeElement(node.getID());
 				}
 			}
 		}
 		removeElementHandler(machine.getID());
 		if (cur_machine == machine) {
 			cur_machine = null;
 		}
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#removeNode(org.eclipse.ptp.core.elements.IPNode)
 	 */
 	public void removeNode(IPNode node) {
 		IElementHandler elementHandler = getElementHandler(node.getMachine().getID());
 		IElementSet set = elementHandler.getSetRoot();
 		IElement element = set.getElementByID(node.getID());
 		if (element != null) {
 			set.removeElement(node.getID());
 		}
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#setCurrentSetId(java.lang.String)
 	 */
 	public void setCurrentSetId(String set_id) {
 		cur_set_id = set_id;
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IMachineManager#setMachine(org.eclipse.ptp.core.elements.IPMachine)
 	 */
 	public void setMachine(IPMachine machine) {
 		if (machine != cur_machine) {
 			cur_machine = machine;
 			addMachine(machine);
 		}
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#shutdown()
 	 */
 	public void shutdown() {
 		clear();
 		modelPresentation = null;
 		super.shutdown();
 	}

 	/* (non-Javadoc)
 	 * @see org.eclipse.ptp.ui.IElementManager#size()
 	 */
 	public int size() {
 		return machineList.size();
 	}

 	/**
 	 * @param set
 	 * @param key
 	 * @param name
 	 * @return
 	 */
 	protected IElement createNodeElement(IElementSet set, String key, String name, IPNode node) {
 		return new Element(set, key, name, node) {
 			public int compareTo(IElement e) {
 				return getID().compareTo(e.getID());
 			}
 		};
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2005 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.ui.managers;

	import java.util.ArrayList;
	import java.util.HashMap;
	import java.util.List;
	import java.util.Map;

	import org.eclipse.ptp.core.elements.IPElement;
	import org.eclipse.ptp.core.elements.IPMachine;
	import org.eclipse.ptp.core.elements.IPNode;
	import org.eclipse.ptp.core.elements.IPProcess;
	import org.eclipse.ptp.core.elements.IPUniverse;
	import org.eclipse.ptp.core.elements.IResourceManager;
	import org.eclipse.ptp.core.elements.attributes.NodeAttributes;
	import org.eclipse.ptp.core.elements.attributes.ProcessAttributes;
	import org.eclipse.ptp.internal.ui.ParallelImages;
	import org.eclipse.ptp.ui.IMachineManager;
	import org.eclipse.ptp.ui.IRuntimeModelPresentation;
	import org.eclipse.ptp.ui.PTPUIPlugin;
	import org.eclipse.ptp.ui.messages.Messages;
	import org.eclipse.ptp.ui.model.Element;
	import org.eclipse.ptp.ui.model.ElementHandler;
	import org.eclipse.ptp.ui.model.IElement;
	import org.eclipse.ptp.ui.model.IElementHandler;
	import org.eclipse.ptp.ui.model.IElementSet;
	import org.eclipse.swt.graphics.Image;

	/**
	* @author clement chu
	*
	*/
	public class MachineManager extends AbstractElementManager implements IMachineManager {
	private Map<String, IPMachine> machineList = new HashMap<String, IPMachine>();
	protected IPMachine cur_machine = null;
	protected final String DEFAULT_TITLE = Messages.MachineManager_0;

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#addMachine(org.eclipse.ptp.core.elements.IPMachine)
	*/
	public void addMachine(IPMachine mac) {
	if (mac != null) {
	IElementHandler handler;
	if (!machineList.containsKey(mac.getID())) {
	handler = new ElementHandler();
	machineList.put(mac.getID(), mac);
	setElementHandler(mac.getID(), handler);
	} else {
	handler = getElementHandler(mac.getID());
	}
	List<IElement> elements = new ArrayList<IElement>();
	IElementSet set = handler.getSetRoot();
	for (IPNode node : mac.getNodes()) {
	elements.add(createNodeElement(set, node.getID(), node.getName(), node));
	}
	set.addElements(elements.toArray(new IElement[0]));
	}
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#addNode(org.eclipse.ptp.core.elements.IPNode)
	*/
	public void addNode(IPNode node) {
	addMachine(node.getMachine());
	IElementHandler elementHandler = getElementHandler(node.getMachine().getID());
	IElementSet set = elementHandler.getSetRoot();
	set.addElements(new IElement[] { createNodeElement(set, node.getID(), node.getName(), node) });
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.managers.AbstractElementManager#clear()
	*/
	@Override
	public void clear() {
	if (machineList != null) {
	machineList.clear();
	}
	super.clear();
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#findMachineById(java.lang.String)
	*/
	public IPMachine findMachineById(String id) {
	return (IPMachine) machineList.get(id);
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#findNode(java.lang.String)
	*/
	public IPNode findNode(String id) {
	IPMachine machine = getCurrentMachine();
	if (machine == null) {
	System.out.println(Messages.MachineManager_1);
	return null;
	}
	return machine.getNodeById(id);
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#getCurrentMachine()
	*/
	public IPMachine getCurrentMachine() {
	return cur_machine;
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#getFullyQualifiedName(java.lang.String)
	*/
	public String getFullyQualifiedName(String id) {
	if (id.equals(EMPTY_ID)) {
	return DEFAULT_TITLE;
	}
	//TODO check that this is what should happen. Can we just use cur_machine?
	IPMachine machine = getCurrentMachine();
	if (machine != null) {
	IResourceManager rm = machine.getResourceManager();
	if (rm != null) {
	return rm.getName() + ": " + machine.getName(); //$NON-NLS-1$
	}
	}
	return ""; //$NON-NLS-1$
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#getMachines()
	*/
	public IPMachine[] getMachines() {
	return machineList.values().toArray(new IPMachine[0]);
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#getName(java.lang.String)
	*/
	public String getName(String id) {
	if (cur_machine == null)
	return ""; //$NON-NLS-1$
	return cur_machine.getName();
	}

	/**
	* Get node status.
	*
	* Currently the node status is determined as follows:
	* - if the node is up:
	* - if there are any running processes on the node: NODE_RUNNING
	* - if there are no running processes, but one or more exited processes on the node: NODE_EXITED
	* - if the node is down: NODE_DOWN
	* - if the node is error: NODE_ERROR
	*
	* TODO: in the future, the machine view should be linked to the jobs view. The node state should only
	* be shown as NODE_RUNNING if any processes belonging to the current job are running.
	*
	* @param node
	* @return
	*/
	private int getNodeState(IPNode node) {
	if (node != null) {
	NodeAttributes.State nodeState = node.getState();

	if (nodeState == NodeAttributes.State.UP) {
	IPProcess[] procs = node.getProcesses();
	if (procs.length > 0) {
	for (IPProcess proc : procs) {
	if (proc.getState() != ProcessAttributes.State.COMPLETED) {
	return NodeAttributes.State.values().length;
	}
	}
	return NodeAttributes.State.values().length + 1;
	}

	}

	return nodeState.ordinal();
	}
	return NodeAttributes.State.UNKNOWN.ordinal();
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.managers.AbstractElementManager#getImage(org.eclipse.ptp.ui.model.IElement)
	*/
	public Image getImage(IElement element) {
	IPMachine machine = getCurrentMachine();
	if (machine != null) {
	IResourceManager rm = machine.getResourceManager();
	final IRuntimeModelPresentation presentation = PTPUIPlugin.getDefault().getRuntimeModelPresentation(rm.getResourceManagerId());
	if (presentation != null) {
	final Image image = presentation.getImage(element);
	if (image != null) {
	return image;
	}
	}
	IPNode node = machine.getNodeById(element.getID());
	return ParallelImages.nodeImages[getNodeState(node)][element.isSelected() ? 1 : 0];
	}
	return null;
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#initial()
	*/
	public IPElement initial(IPUniverse universe) {
	for (IResourceManager rm : universe.getResourceManagers()) {
	for (IPMachine machine : rm.getMachines()) {
	addMachine(machine);
	}
	}

	setCurrentSetId(IElementHandler.SET_ROOT_ID);
	return null;
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#removeMachine(org.eclipse.ptp.core.elements.IPMachine)
	*/
	public void removeMachine(IPMachine machine) {
	machineList.remove(machine.getID());
	IElementHandler handler = getElementHandler(machine.getID());
	if (handler != null) {
	IElementSet set = handler.getSetRoot();
	for (IPNode node : machine.getNodes()) {
	IElement element = set.getElementByID(node.getID());
	if (element != null) {
	set.removeElement(node.getID());
	}
	}
	}
	removeElementHandler(machine.getID());
	if (cur_machine == machine) {
	cur_machine = null;
	}
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#removeNode(org.eclipse.ptp.core.elements.IPNode)
	*/
	public void removeNode(IPNode node) {
	IElementHandler elementHandler = getElementHandler(node.getMachine().getID());
	IElementSet set = elementHandler.getSetRoot();
	IElement element = set.getElementByID(node.getID());
	if (element != null) {
	set.removeElement(node.getID());
	}
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#setCurrentSetId(java.lang.String)
	*/
	public void setCurrentSetId(String set_id) {
	cur_set_id = set_id;
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IMachineManager#setMachine(org.eclipse.ptp.core.elements.IPMachine)
	*/
	public void setMachine(IPMachine machine) {
	if (machine != cur_machine) {
	cur_machine = machine;
	addMachine(machine);
	}
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#shutdown()
	*/
	public void shutdown() {
	clear();
	modelPresentation = null;
	super.shutdown();
	}

	/* (non-Javadoc)
	* @see org.eclipse.ptp.ui.IElementManager#size()
	*/
	public int size() {
	return machineList.size();
	}

	/**
	* @param set
	* @param key
	* @param name
	* @return
	*/
	protected IElement createNodeElement(IElementSet set, String key, String name, IPNode node) {
	return new Element(set, key, name, node) {
	public int compareTo(IElement e) {
	return getID().compareTo(e.getID());
	}
	};
	}
	}