core/plugins/org.eclipse.hawk.epsilon/src/org/eclipse/hawk/epsilon/emc/wrappers/GraphEdgeWrapper.java

/*******************************************************************************
 * Copyright (c) 2011-2016 The University of York.
 * 
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the Eclipse
 * Public License, v. 2.0 are satisfied: GNU General Public License, version 3.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-3.0
 *
 * Contributors:
 *     Antonio Garcia-Dominguez - initial API and implementation
 ******************************************************************************/
package org.eclipse.hawk.epsilon.emc.wrappers;

import java.util.Objects;
import java.util.Set;

import org.eclipse.epsilon.eol.exceptions.EolRuntimeException;
import org.eclipse.hawk.core.graph.IGraphEdge;
import org.eclipse.hawk.epsilon.emc.EOLQueryEngine;

/**
 * Wraps an edge produced during a Hawk query. Has a few useful synonyms
 * to support common names and the literal terms used within Hawk.
 */
public class GraphEdgeWrapper {

	private IGraphEdge edge;
	private EOLQueryEngine containerModel;

	public GraphEdgeWrapper(IGraphEdge r, EOLQueryEngine m) {
		this.edge = r;
		this.containerModel = m;
	}

	public IGraphEdge getEdge() {
		return edge;
	}

	public String getType() {
		return edge.getType();
	}

	public String getName() {
		return getType();
	}

	public Object getEndNode() {
		return containerModel.wrap(edge.getEndNode());
	}

	public Object getTarget() {
		return getEndNode();
	}

	public Object getStartNode() {
		return containerModel.wrap(edge.getStartNode());
	}

	public Object getSource() {
		return getStartNode();
	}

	public EOLQueryEngine getContainerModel() {
		return containerModel;
	}

	public Set<String> getPropertyKeys() {
		return edge.getPropertyKeys();
	}

	public Object getProperty(String name) {
		return edge.getProperty(name);
	}

	public Object getFeature(String name) throws EolRuntimeException {
		return containerModel.getPropertyGetter().invoke(this, name);
	}

	@Override
	public int hashCode() {
		return Objects.hash(containerModel, edge);
	}

	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		GraphEdgeWrapper other = (GraphEdgeWrapper) obj;
		return Objects.equals(containerModel, other.containerModel) && Objects.equals(edge, other.edge);
	}

}