style

Author: LorenzBuehmann

sansa-inference-common/src/main/scala/net/sansa_stack/inference/utils/GraphUtils.scala

@@ -27,7 +27,7 @@ import org.jgrapht.io._
 import org.openide.util.Lookup
 import scalax.collection.edge.LDiEdge
 
-import net.sansa_stack.inference.utils.graph.{EdgeEquivalenceComparator, LabeledEdge, NodeEquivalenceComparator};
+import net.sansa_stack.inference.utils.graph.{EdgeEquivalenceComparator, LabeledEdge, NodeEquivalenceComparator}
 
 /**
   * @author Lorenz Buehmann
@@ -188,12 +188,12 @@ object GraphUtils {
 
           // Gephi
           // Init a project - and therefore a workspace
-          val pc = Lookup.getDefault().lookup(classOf[ProjectController])
+          val pc = Lookup.getDefault.lookup(classOf[ProjectController])
           pc.newProject()
-          val workspace = pc.getCurrentWorkspace()
+          val workspace = pc.getCurrentWorkspace
 
           // Get controllers and models
-          val importController = Lookup.getDefault().lookup(classOf[ImportController])
+          val importController = Lookup.getDefault.lookup(classOf[ImportController])
 
           // export as GraphML
           val tmpFilename = "/tmp/temp-graph.graphml"
@@ -202,8 +202,8 @@ object GraphUtils {
           // Import file
           val file = new File(tmpFilename)
           val container = importController.importFile(file)
-          container.getLoader().setEdgeDefault(EdgeDirectionDefault.DIRECTED)   // Force DIRECTED
-          container.getLoader().setAllowAutoNode(false)  // Don't create missing nodes
+          container.getLoader.setEdgeDefault(EdgeDirectionDefault.DIRECTED)   // Force DIRECTED
+          container.getLoader.setAllowAutoNode(false)  // Don't create missing nodes
 
           // Append imported data to GraphAPI
           importController.process(container, new DefaultProcessor(), workspace)
@@ -213,7 +213,7 @@ object GraphUtils {
 
 
           // See if graph is well imported
-          val graphModel = Lookup.getDefault().lookup(classOf[GraphController]).getGraphModel
+          val graphModel = Lookup.getDefault.lookup(classOf[GraphController]).getGraphModel
           val g = graphModel.getDirectedGraph()
 
           // Run YifanHuLayout for 100 passes - The layout always takes the current visible view
@@ -223,23 +223,23 @@ object GraphUtils {
           layout.setOptimalDistance(200f)
 
           layout.initAlgo()
-          for (i <- 0 to 100 if layout.canAlgo()) {
+          for (i <- 0 to 100 if layout.canAlgo) {
             layout.goAlgo()
           }
           layout.endAlgo()
 
-          val model = Lookup.getDefault().lookup(classOf[PreviewController]).getModel()
-          model.getProperties().putValue(PreviewProperty.SHOW_NODE_LABELS, true)
-          model.getProperties().putValue(PreviewProperty.SHOW_EDGE_LABELS, true)
-          model.getProperties().putValue(PreviewProperty.EDGE_CURVED, false)
-          model.getProperties().putValue(PreviewProperty.EDGE_COLOR, new EdgeColor(java.awt.Color.GRAY))
-          model.getProperties().putValue(PreviewProperty.EDGE_THICKNESS, 0.1f)
-          model.getProperties().putValue(PreviewProperty.NODE_LABEL_FONT, model.getProperties().getFontValue(PreviewProperty.NODE_LABEL_FONT).deriveFont(8))
+          val model = Lookup.getDefault.lookup(classOf[PreviewController]).getModel()
+          model.getProperties.putValue(PreviewProperty.SHOW_NODE_LABELS, true)
+          model.getProperties.putValue(PreviewProperty.SHOW_EDGE_LABELS, true)
+          model.getProperties.putValue(PreviewProperty.EDGE_CURVED, false)
+          model.getProperties.putValue(PreviewProperty.EDGE_COLOR, new EdgeColor(java.awt.Color.GRAY))
+          model.getProperties.putValue(PreviewProperty.EDGE_THICKNESS, 0.1f)
+          model.getProperties.putValue(PreviewProperty.NODE_LABEL_FONT, model.getProperties.getFontValue(PreviewProperty.NODE_LABEL_FONT).deriveFont(8))
                 model.getProperties.putValue(Item.NODE_LABEL, "Vertex Label")
 
 
           // Export full graph
-          val ec = Lookup.getDefault().lookup(classOf[ExportController])
+          val ec = Lookup.getDefault.lookup(classOf[ExportController])
           //      ec.exportFile(new File("io_gexf.gexf"));
 
           // PDF Exporter config and export to Byte array
@@ -341,7 +341,7 @@ object GraphUtils {
         val label = e.label.asInstanceOf[Node]
         g.addVertex(s)
         g.addVertex(t)
-        g.addEdge(s, t, new LabeledEdge(s, t, label))
+        g.addEdge(s, t, LabeledEdge(s, t, label))
       }
 
       // In order to be able to export edge and node labels and IDs,