package converter;
import bpcf.*;
import bpcf.impl.EndImpl;
import bpcf.impl.StartImpl;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.UUID;
/**
* Created by miguel on 06-05-2014.
*/
public class BPMNManager implements Manager {
private BpcfFactory factory;
private BPCtrlFlowModel bpCtrlFlowModel;
private HashMap<String, Object> elements;
private long numberOfElements = 1;
int n = 0;
public BPMNManager() {
factory = BpcfFactory.eINSTANCE;
bpCtrlFlowModel = factory.createBPCtrlFlowModel();
elements = new HashMap<String, Object>();
}
public void addActivity(String id, String name) {
System.out.println(".. Adding Activity " + name);
Activity activity = factory.createActivity();
activity.setId(numberOfElements);
numberOfElements++;
activity.setName(name);
bpCtrlFlowModel.getBpElements().add(activity);
elements.put(id, activity);
}
public void addFlow(BPElement one, BPElement two) {
one.getTarget().add(two);
two.getSource().add(one);
}
public void addFlow(String idFrom, String idTo) {
if (n < 4) {
n++;
}
if (elements.containsKey(idFrom) && elements.containsKey(idTo)) {
System.out.println(".. Adding flow from " + ((BPElement) elements.get(idFrom)).getName() + " to " + ((BPElement) elements.get(idTo)).getName());
addFlow((BPElement) elements.get(idFrom), (BPElement) elements.get(idTo));
}
}
public void addMerge(String id) {
addMerge(id, id);
}
public void addMerge(String id, String name) {
System.out.println(".. Adding Merge Element " + name);
Merge merge = factory.createMerge();
merge.setId(numberOfElements);
numberOfElements++;
merge.setName(name);
bpCtrlFlowModel.getBpElements().add(merge);
elements.put(id, merge);
}
public void addFork(String id) {
addFork(id, id);
}
public void addFork(String id, String name) {
System.out.println(".. Adding Fork Element " + name);
Fork fork = factory.createFork();
fork.setId(numberOfElements);
numberOfElements++;
fork.setName(name);
bpCtrlFlowModel.getBpElements().add(fork);
elements.put(id, fork);
}
public void addJoin(String id) {
addJoin(id, id);
}
public void addJoin(String id, String name) {
System.out.println(".. Adding Join Element " + name);
Join join = factory.createJoin();
join.setId(numberOfElements);
numberOfElements++;
join.setName(name);
bpCtrlFlowModel.getBpElements().add(join);
elements.put(id, join);
}
public void addDecision(String id) {
addDecision(id, id);
}
public void addDecision(String id, String name) {
System.out.println(".. Adding Decision Element " + name);
Decision decision = factory.createDecision();
decision.setId(numberOfElements);
numberOfElements++;
decision.setName(name);
bpCtrlFlowModel.getBpElements().add(decision);
elements.put(id, decision);
}
public void addStart(String id) {
addStart(id, "Start");
}
public void addStart(String id, String name) {
System.out.println(".. Adding Start Element " + name);
Start start = factory.createStart();
start.setName(name);
start.setId(numberOfElements);
numberOfElements++;
elements.put(id, start);
bpCtrlFlowModel.getBpElements().add(start);
bpCtrlFlowModel.getStartEvents().add(start);
// start.getTarget().add(elements.get(flowToId));
// elements.get(flowToId).getSource().add(start);
}
public void addArtificialStart(String flowToId) {
if (elements.containsKey(flowToId) && elements.get(flowToId).getClass() != StartImpl.class) {
System.out.println(".. Adding Infered Start Element flowing to " + ((BPElement) elements.get(flowToId)).getName());
String uuid = UUID.randomUUID().toString();
addStart(uuid, "START");
addFlow((BPElement) elements.get(uuid), (BPElement) elements.get(flowToId));
}
}
public void addFinish(String id) {
addFinish(id, "End");
}
public void addFinish(String id, String name) {
System.out.println(".. Adding Finish Element " + name);
// System.out.println("ADDING FINISH: " + id + " --> " + name);
End end = factory.createEnd();
end.setId(numberOfElements);
end.setName(name);
numberOfElements++;
elements.put(id, end);
bpCtrlFlowModel.getBpElements().add(end);
// elements.get(flowFrom).getTarget().add(end);
// end.getSource().add(elements.get(flowFrom));
}
public void addArtificialFinish(String flowFromId) {
if (elements.containsKey(flowFromId) && elements.get(flowFromId).getClass() != EndImpl.class) {
System.out.println(".. Adding Infered End Element flowing from " + ((BPElement) elements.get(flowFromId)).getName());
String uuid = UUID.randomUUID().toString();
addFinish(uuid, "END");
addFlow((BPElement) elements.get(flowFromId), (BPElement) elements.get(uuid));
}
}
public void saveModel(String path) {
System.out.println("Saving model in " + path + "...");
/*for (String a : elements.keySet())
System.out.println(a);*/
ResourceSet resSet = new ResourceSetImpl();
resSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put(
Resource.Factory.Registry.DEFAULT_EXTENSION, new org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl());
String newFileString = new File(path).getAbsolutePath();
// System.out.println(newFileString);
URI fileURI = URI.createFileURI(newFileString);
Resource resource = resSet.createResource(fileURI);
resource.getContents().add(bpCtrlFlowModel);
try {
resource.save(Collections.EMPTY_MAP);
} catch (IOException e) {
e.printStackTrace();
}
}
public HashMap<String, Object> getElements() {
return elements;
}
public static void main(String[] args) {
BpcfFactory factory = BpcfFactory.eINSTANCE;
BPCtrlFlowModel bpCtrlFlowModel = factory.createBPCtrlFlowModel();
// bpCtrlFlowModel.getBpElements().add(start);
BPElement start = factory.createStart();
Start start1 = factory.createStart();
// System.out.println(StartImpl.class);
// bpCtrlFlowModel.getStartEvents().add(start);
}
private static void saveEMFModel(EObject model, String filepath) {
ResourceSet resSet = new ResourceSetImpl();
resSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put(
Resource.Factory.Registry.DEFAULT_EXTENSION, new org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl());
String newFileString = new File(filepath).getAbsolutePath();
URI fileURI = URI.createFileURI(newFileString);
Resource resource = resSet.createResource(fileURI);
resource.getContents().add(model);
try {
resource.save(Collections.EMPTY_MAP);
} catch (IOException e) {
e.printStackTrace();
}
}
}