MatchSearchTool.java example

Explorer
Henshin-Editor-master
/*******************************************************************************
 * Copyright (c) 2010-2015 Henshin developers. 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:
 *     TU Berlin, University of Luxembourg, SES S.A.
 *******************************************************************************/
/**
 * 
 */
package de.tub.tfs.henshin.editor.tools;

import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import org.eclipse.draw2d.ColorConstants;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.eclipse.emf.henshin.interpreter.Engine;
import org.eclipse.emf.henshin.interpreter.Match;
import org.eclipse.emf.henshin.interpreter.impl.EngineImpl;
import org.eclipse.emf.henshin.interpreter.util.HenshinEGraph;
import org.eclipse.emf.henshin.model.Graph;
import org.eclipse.emf.henshin.model.Module;
import org.eclipse.emf.henshin.model.Node;
import org.eclipse.emf.henshin.model.Rule;
import org.eclipse.gef.tools.AbstractTool;
import org.eclipse.swt.widgets.Shell;

import de.tub.tfs.henshin.editor.editparts.graph.graphical.GraphEditPart;
import de.tub.tfs.henshin.editor.editparts.graph.graphical.NodeEditPart;
import de.tub.tfs.henshin.editor.interfaces.Messages;
import de.tub.tfs.henshin.editor.ui.graph.GraphView;
import de.tub.tfs.henshin.editor.util.DialogUtil;
import de.tub.tfs.henshin.editor.util.HenshinSelectionUtil;
import de.tub.tfs.henshin.editor.util.HenshinUtil;
import de.tub.tfs.henshin.editor.util.ModelUtil;

/**
 * @author huuloi
 *
 */
public class MatchSearchTool extends AbstractTool {
	
	private Graph graph;
	
	public MatchSearchTool() {
	}
	
	@Override
	public void activate() {

		Module transformationSystem = ModelUtil.getModelRoot(graph, Module.class);
		List<Rule> rules = HenshinUtil.getRules( transformationSystem);
		GraphView graphView = HenshinSelectionUtil.getInstance().getActiveGraphView(graph);
		Shell shell = graphView.getSite().getShell();
		// open rule selection dialog
		Rule rule = DialogUtil.runRuleChoiceDialog(shell, rules);
		
		// do search
		HenshinEGraph henshinGraph = new HenshinEGraph(graph);
		Engine engine = new EngineImpl();
		//RuleApplication ruleApplication = new RuleApplicationImpl(engine,henshinGraph, rule,null);
		//ruleApplication.apply();
		
		Iterable<Match> matches = engine.findMatches(rule, henshinGraph, null);
		Set<Node> nodeMapping = new HashSet<Node>();
		//for (Match match : matche) {
			nodeMapping.addAll(rule.getLhs().getNodes());
		//}
		
		List<NodeEditPart> nodeEditParts = HenshinSelectionUtil.getInstance().getNodeEditParts(graph);
		for (NodeEditPart nodeEditPart : nodeEditParts) {
			Node node = nodeEditPart.getCastedModel();
			Iterator<Node> iterator = nodeMapping.iterator();
			while (iterator.hasNext()) {
				Node mappedNode = iterator.next();
				if (EcoreUtil.equals(node, mappedNode)) {
					nodeEditPart.getFigure().setBackgroundColor(ColorConstants.lightBlue);
				}
				else if (!EcoreUtil.equals(node, mappedNode) && nodeEditPart.getFigure().getBackgroundColor() != nodeEditPart.getDefaultColor()) {
					nodeEditPart.getFigure().setBackgroundColor(nodeEditPart.getDefaultColor());
				}
			}
		}
		
		// refresh
		((GraphEditPart)graphView.getCurrentGraphPage().getCurrentViewer().getEditPartRegistry().get(graph)).refresh();
	}

	@Override
	protected String getCommandName() {
		return Messages.MATCH_SEARCH;
	}

	public Graph getGraph() {
		return graph;
	}

	public void setGraph(Graph graph) {
		this.graph = graph;
	}
	
}