}
CsvReader reader = null;
try {
//Prepare attribute columns for the column names, creating the not already existing columns:
AttributeTable edgesTable = Lookup.getDefault().lookup(AttributeController.class).getModel().getEdgeTable();
String idColumn = null;
String sourceColumn = null;
String targetColumn = null;
String typeColumn = null;
ArrayList<AttributeColumn> columnsList = new ArrayList<AttributeColumn>();
HashMap<AttributeColumn, String> columnHeaders = new HashMap<AttributeColumn, String>();//Necessary because of column name case insensitivity, to map columns to its corresponding csv header.
for (int i = 0; i < columnNames.length; i++) {
//Separate first id column found from the list to use as id. If more are found later, the will not be in the list and be ignored.
if (columnNames[i].equalsIgnoreCase("id")) {
if (idColumn == null) {
idColumn = columnNames[i];
}
} else if (columnNames[i].equalsIgnoreCase("source") && sourceColumn == null) {//Separate first source column found from the list to use as source node id
sourceColumn = columnNames[i];
} else if (columnNames[i].equalsIgnoreCase("target") && targetColumn == null) {//Separate first target column found from the list to use as target node id
targetColumn = columnNames[i];
} else if (columnNames[i].equalsIgnoreCase("type") && typeColumn == null) {//Separate first type column found from the list to use as edge type (directed/undirected)
typeColumn = columnNames[i];
} else if (edgesTable.hasColumn(columnNames[i])) {
AttributeColumn column = edgesTable.getColumn(columnNames[i]);
columnsList.add(column);
columnHeaders.put(column, columnNames[i]);
} else {
AttributeColumn column = addAttributeColumn(edgesTable, columnNames[i], columnTypes[i]);
if (column != null) {
columnsList.add(column);
columnHeaders.put(column, columnNames[i]);
}
}
}
//Create edges:
GraphElementsController gec = Lookup.getDefault().lookup(GraphElementsController.class);
Graph graph = Lookup.getDefault().lookup(GraphController.class).getModel().getGraph();
String id = null;
Edge edge;
String sourceId, targetId;
Node source, target;
String type;
boolean directed;
Attributes edgeAttributes;
reader = new CsvReader(new FileInputStream(file), separator, charset);
reader.setTrimWhitespace(false);
reader.readHeaders();
while (reader.readRecord()) {
sourceId = reader.get(sourceColumn);
targetId = reader.get(targetColumn);
if (sourceId == null || sourceId.isEmpty() || targetId == null || targetId.isEmpty()) {
continue;//No correct source and target ids were provided, ignore row
}
graph.readLock();
source = graph.getNode(sourceId);
graph.readUnlock();
if (source == null) {
if (createNewNodes) {//Create new nodes when they don't exist already and option is enabled
if (source == null) {
source = gec.createNode(null, sourceId);
}
} else {
continue;//Ignore this edge row, since no new nodes should be created.
}
}
graph.readLock();
target = graph.getNode(targetId);
graph.readUnlock();
if (target == null) {
if (createNewNodes) {//Create new nodes when they don't exist already and option is enabled
if (target == null) {
target = gec.createNode(null, targetId);
}
} else {
continue;//Ignore this edge row, since no new nodes should be created.
}
}
if (typeColumn != null) {
type = reader.get(typeColumn);
//Undirected if indicated correctly, otherwise always directed:
if (type != null) {
directed = !type.equalsIgnoreCase("undirected");
} else {
directed = true;
}
} else {
directed = true;//Directed by default when not indicated
}
//Prepare the correct edge to assign the attributes:
if (idColumn != null) {
id = reader.get(idColumn);
if (id == null || id.isEmpty()) {
edge = gec.createEdge(source, target, directed);//id null or empty, assign one
} else {
edge = gec.createEdge(id, source, target, directed);
if (edge == null) {//Edge with that id already in graph
edge = gec.createEdge(source, target, directed);
}
}
} else {
edge = gec.createEdge(source, target, directed);
}
if (edge != null) {//Edge could be created because it does not already exist:
//Assign attributes to the current edge:
edgeAttributes = edge.getEdgeData().getAttributes();
for (AttributeColumn column : columnsList) {
setAttributeValue(reader.get(columnHeaders.get(column)), edgeAttributes, column);
}
} else {
//Do not ignore repeated edge, instead increase edge weight
edge = graph.getEdge(source, target);
if (edge == null) {
//Not from source to target but undirected and reverse?
edge = graph.getEdge(target, source);
if (edge != null && edge.isDirected()) {
edge = null;
}
}
if (edge != null) {
//Increase edge weight with specified weight (if specified), else increase by 1:
String weight = reader.get(columnHeaders.get(edgesTable.getColumn(PropertiesColumn.EDGE_WEIGHT.getIndex())));
if (weight != null) {
try {
Float weightFloat = Float.parseFloat(weight);
edge.getEdgeData().getAttributes().setValue(PropertiesColumn.EDGE_WEIGHT.getIndex(), edge.getWeight() + weightFloat);
} catch (NumberFormatException numberFormatException) {