org.jbpt.pm.structure.ProcessModel2NetSystem Maven / Gradle / Ivy

Go to download
package org.jbpt.pm.structure;

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

import org.jbpt.petri.NetSystem;
import org.jbpt.petri.Node;
import org.jbpt.petri.Place;
import org.jbpt.petri.Transition;
import org.jbpt.pm.AndGateway;
import org.jbpt.pm.ControlFlow;
import org.jbpt.pm.FlowNode;
import org.jbpt.pm.OrGateway;
import org.jbpt.pm.ProcessModel;
import org.jbpt.pm.XorGateway;
import org.jbpt.throwable.TransformationException;

public class ProcessModel2NetSystem {
	
	public static NetSystem transform(ProcessModel pm) throws TransformationException {
		if (pm.getGateways(OrGateway.class).size() > 0) throw new TransformationException();
		
		NetSystem sys = new NetSystem();
		
		sys.setId(pm.getId());
		sys.setName(pm.getName());
		sys.setDescription(pm.getDescription());
		sys.setTag(pm.getTag());
		
		Map map = new HashMap();
		
		// the process is transformed edge by edge
		for (ControlFlow flow : pm.getControlFlow()) {
			FlowNode src = flow.getSource();
			FlowNode tgt = flow.getTarget();
			
			if (isXORGateway(src)) {
				if (isXORGateway(tgt)) {
					Place psrc = (Place) getNode(src, sys, map);
					Place ptgt = (Place) getNode(tgt, sys, map);
					Transition inter = new Transition();
					inter.setId(psrc.getId() + "_" + ptgt.getId());
					sys.addFlow(psrc, inter);
					sys.addFlow(inter, ptgt);
				}
				else {
					Place psrc = (Place) getNode(src, sys, map);
					Transition ptgt = (Transition) getNode(tgt, sys, map);
	
					Place pintp = new Place();
					pintp.setId(psrc.getId() + "_p_" + ptgt.getId());
					Transition pintt = new Transition(); 
					pintt.setId(psrc.getId() + "_t_" + ptgt.getId());
					sys.addFlow(psrc, pintt);
					sys.addFlow(pintt, pintp);
					sys.addFlow(pintp, ptgt);
				}  
			}
			else {
				if (isXORGateway(tgt)) {
					Transition psrc = (Transition) getNode(src, sys, map);					
					Place ptgt = (Place) getNode(tgt, sys, map);
					sys.addFlow(psrc, ptgt);
				}
				else {						
					Transition psrc = (Transition) getNode(src, sys, map);
					Transition ptgt = (Transition) getNode(tgt, sys, map);
					Place p = new Place();
					p.setId(psrc.getId() + "_" + ptgt.getId());
					sys.addFlow(psrc, p);
					sys.addFlow(p, ptgt);
				} 
			} 
		}
		
		List sources = new ArrayList();
		List sinks = new ArrayList();
		
		for (FlowNode node : pm.getVertices()) {
			if (pm.getIncomingEdges(node).size() == 0) {
				// nodes without an incoming edge
				if (isXORGateway(node)) {
					// XOR place needs an additional transition in front
					Transition t = new Transition();
					sys.addFlow(t, (Place)getNode(node, sys, map));
					sources.add(t);
				} else
					// AND or normal task transition
					sources.add(getNode(node, sys, map));
			}
			if (pm.getOutgoingEdges(node).size() == 0) {
				// nodes without outgoing edge
				if (isXORGateway(node)) {
					// XOR place needs an additional following transition
					Transition t = new Transition();
					sys.addFlow((Place)getNode(node, sys, map), t);
					sinks.add(t);
				} else 
					// AND or normal task transition
					sinks.add(getNode(node, sys, map));
			}
		}
		// create according entry and exit places for the transitions without incoming / outgoing edges
		for (org.jbpt.petri.Node node:sources) {
			Place p = new Place();
			sys.addFlow(p, (Transition)node);
		}
		for (org.jbpt.petri.Node node:sinks) {
			Place p = new Place();
			sys.addFlow((Transition)node, p);
		}
		
		sys.loadNaturalMarking();
		return sys;
	}
	
	private static boolean isANDGateway(FlowNode node) {
		return (node instanceof AndGateway);
	}
	
	private static boolean isXORGateway(FlowNode node) {
		return (node instanceof XorGateway);
	}
	
	private static Node getNode(FlowNode node, NetSystem sys, Map map) {
		Node res = map.get(node);
		if (res==null) {
			if (isXORGateway(node)) res = new Place();
			else res = new Transition();
			
			res.setId(node.getId());
			
			if (!node.getName().equals("")) {
				res.setName(node.getName());
				res.setLabel(node.getName());
			}
			
			res.setDescription(node.getDescription());
			res.setTag(node.getTag());
			
			map.put(node, res);
		}
		return res;
	}

}