Revize 4f50a73f

Introduced GraphFactory

Class prepares instance of Graph from data loaded by BaseDOTLoader

sources/imiger-dot-converter/src/main/java/cz/zcu/kiv/imiger/plugin/dot/BaseGraphFactory.java

@@ +1 @@
+package cz.zcu.kiv.imiger.plugin.dot;
+
+import cz.zcu.kiv.imiger.plugin.dot.dto.EdgeDTO;
+import cz.zcu.kiv.imiger.plugin.dot.dto.VertexDTO;
+import cz.zcu.kiv.imiger.plugin.dot.loader.BaseDOTLoader;
+import cz.zcu.kiv.imiger.vo.*;
+
+import java.util.*;
+
+/**
+ * Date: 31.03.2019
+ *
+ * @author Martin Matas
+ */
+public abstract class BaseGraphFactory<V extends VertexDTO, E extends EdgeDTO> {
+
+	protected List<VertexArchetype> vertexArchetypes;
+	protected List<EdgeArchetype> edgeArchetypes;
+	protected List<AttributeType> attributeTypes;
+	protected List<VertexDTO> vertices;
+	protected List<EdgeDTO> edges;
+
+
+	protected BaseGraphFactory(BaseDOTLoader<V, E> dotLoader) {
+		loadData(dotLoader);
+	}
+
+	protected abstract void loadData(BaseDOTLoader<V, E> dotLoader);
+
+	public abstract Graph createGraph();
+
+}

sources/imiger-dot-converter/src/main/java/cz/zcu/kiv/imiger/plugin/dot/GraphFactory.java

@@ +1 @@
+package cz.zcu.kiv.imiger.plugin.dot;
+
+import cz.zcu.kiv.imiger.plugin.dot.dto.EdgeDTO;
+import cz.zcu.kiv.imiger.plugin.dot.dto.VertexDTO;
+import cz.zcu.kiv.imiger.plugin.dot.loader.BaseDOTLoader;
+import cz.zcu.kiv.imiger.vo.*;
+
+import java.util.*;
+
+/**
+ * Date: 31.03.2019
+ *
+ * @author Martin Matas
+ */
+public class GraphFactory extends BaseGraphFactory<VertexDTO, EdgeDTO> {
+
+	private static final String EDGE_ARCHETYPE_NAME = "Edge";
+	private static final String VERTEX_ARCHETYPE_NAME = "Vertex";
+	private static final String ARCHETYPE_TEXT = "";
+	private static final int DEFAULT_ARCHETYPE_INDEX = 0;
+
+	public GraphFactory(BaseDOTLoader<VertexDTO,EdgeDTO> dotLoader) {
+		super(dotLoader);
+		vertices = new ArrayList<>();
+		edges = new ArrayList<>();
+		attributeTypes = new ArrayList<>();
+		prepareEdgeArchetypes();
+		prepareVertexArchetypes();
+	}
+
+	@Override
+	protected void loadData(BaseDOTLoader<VertexDTO,EdgeDTO> dotLoader) {
+		List<VertexDTO> loadedVertices = dotLoader.getVertices();
+		List<EdgeDTO> loadedEdges = dotLoader.getEdges();
+		Set<AttributeType> loadedAttributeTypes = dotLoader.getAttributeTypes();
+
+		if (loadedVertices != null) {
+			vertices = new ArrayList<>(loadedVertices);
+		}
+
+		if (loadedVertices != null) {
+			edges = new ArrayList<>(loadedEdges);
+		}
+
+		if (loadedVertices != null) {
+			attributeTypes = new ArrayList<>(loadedAttributeTypes);
+		}
+	}
+
+	@Override
+	public Graph createGraph() {
+		Graph graph = new Graph();
+
+		graph.setEdgeArchetypes(edgeArchetypes);
+		graph.setVertexArchetypes(vertexArchetypes);
+		graph.setAttributeTypes(attributeTypes);
+		addEdgesToGraph(graph, edges);
+		addVerticesToGraph(graph, vertices);
+
+		return graph;
+	}
+
+	private void addVerticesToGraph(Graph graph, List<VertexDTO> vertices) {
+		for (VertexDTO v : vertices) {
+
+			List<String[]> attributes = new ArrayList<>();
+			attributes.add((String[]) v.getAttributes().values().toArray());
+			Vertex vertex = new Vertex(v.getId(), v.getName(), DEFAULT_ARCHETYPE_INDEX, "", attributes);
+			graph.getVertices().add(vertex);
+		}
+	}
+
+	private void addEdgesToGraph(Graph graph, List<EdgeDTO> edges) {
+		for (EdgeDTO e : edges) {
+
+			List<String[]> attributes = new ArrayList<>();
+			attributes.add((String[]) e.getAttributes().values().toArray());
+			SubedgeInfo subedgeInfo = new SubedgeInfo(e.getId(), DEFAULT_ARCHETYPE_INDEX, attributes);
+
+			List<SubedgeInfo> subedgeInfos = new ArrayList<>();
+			subedgeInfos.add(subedgeInfo);
+
+			Edge eddge = new Edge(e.getId(), e.getIdFrom(), e.getIdTo(), e.getName(), subedgeInfos);
+			graph.getEdges().add(eddge);
+		}
+	}
+
+	private void prepareEdgeArchetypes() {
+		edgeArchetypes= new ArrayList<>();
+		edgeArchetypes.add(new EdgeArchetype(EDGE_ARCHETYPE_NAME, ARCHETYPE_TEXT));
+	}
+
+	private void prepareVertexArchetypes() {
+		vertexArchetypes = new ArrayList<>();
+		vertexArchetypes.add(new VertexArchetype(VERTEX_ARCHETYPE_NAME, ARCHETYPE_TEXT));
+	}
+
+}