• Home
• com.github.mdr
• ascii-graphs_2.9.2
• 0.0.4
• source code
• DiagramToGraphConverter.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.mdr.ascii.graph.DiagramToGraphConverter.scala Maven / Gradle / Ivy

package com.github.mdr.ascii.graph

import com.github.mdr.ascii.diagram.Box
import com.github.mdr.ascii.diagram.Diagram
import com.github.mdr.ascii.util.Utils.makeMap

object DiagramToGraphConvertor {

  def toGraph(diagram: Diagram): Graph[String] = {
    val boxToVertexMap: Map[Box, String] = makeMap(diagram.childBoxes, _.text)

    val vertices = boxToVertexMap.values.toSet
    val edges =
      for {
        edge ← diagram.allEdges
        vertex1 ← boxToVertexMap.get(edge.box1)
        vertex2 ← boxToVertexMap.get(edge.box2)
      } yield {
        if (edge.hasArrow2)
          vertex1 -> vertex2
        else
          vertex2 -> vertex1
      }
    Graph(vertices, edges)
  }

}