scalax.collection.generic.edgeToString.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graph-core_2.13 Show documentation
Graph Core
The newest version!
package scalax.collection.generic

sealed protected trait EdgeToString { _: Edge[_] =>

  protected def nodesToStringSeparator: String

  protected def nodesToString: String = ends.iterator mkString nodesToStringSeparator

  /** Implementation in term of the protected method `nodesToString`. */
  override def toString: String = nodesToString
}

trait UnDiEdgeToString extends EdgeToString { _: AnyUnDiEdge[_] =>
  override protected def nodesToStringSeparator: String = " ~ "
}

trait DiEdgeToString extends EdgeToString { _: AnyDiEdge[_] =>
  override protected def nodesToStringSeparator: String = " ~> "
}

trait HyperEdgeToString extends EdgeToString { _: AnyHyperEdge[_] =>
  override protected def nodesToStringSeparator: String = " ~~ "
}

trait DiHyperEdgeToString extends EdgeToString { _: AnyDiHyperEdge[_] =>

  override protected def nodesToStringSeparator: String = " ~~> "

  override protected def nodesToString: String = {
    def part(nodes: IterableOnce[_]): String = s"""{${nodes.iterator mkString ", "}}"""
    s"${part(sources.iterator)}$nodesToStringSeparator${part(targets.iterator)}"
  }
}

sealed protected trait LEdgeToString extends EdgeToString { _: Edge[_] =>

  /** The part to be printed after the edge ends and before the label. */
  protected def labelSeparator: String = " :+ "

  /** The part to be printed after the edge ends and the label separator. */
  protected def labelToString: String

  /** Implemented in terms of the protected methods `nodesToString`, `labelSeparator` and  `labelToString`. */
  override def toString: String = s"$nodesToString$labelSeparator$labelToString"
}

trait LUnDiEdgeToString extends LEdgeToString with UnDiEdgeToString { this: AnyUnDiEdge[_] => }

trait LDiEdgeToString extends LEdgeToString with DiEdgeToString { this: AnyDiEdge[_] => }

trait LHyperEdgeToString extends LEdgeToString with HyperEdgeToString { this: AbstractHyperEdge[_] => }

trait LDiHyperEdgeToString extends LEdgeToString with DiHyperEdgeToString { this: AbstractDiHyperEdge[_] => }

/** Mix in this trait in your labeled edge class if you want to indicate in its `toString` representation
  * that the edge supports multigraphs. Once mixed in, the `labelSeparator` gets `:++`.
  */
trait MultiLEdgeToString { _: LEdgeToString with MultiEdge =>

  override protected def labelSeparator: String = " :++ "
}

sealed protected trait WEdgeToString extends LEdgeToString { _: Edge[_] =>

  override protected def labelSeparator: String = " % "

  /** The part to be printed after the edge ends and the label separator. */
  protected def labelToString: String = weight.toString
}

trait WUnDiEdgeToString extends WEdgeToString with UnDiEdgeToString { _: AnyUnDiEdge[_] => }

trait WDiEdgeToString extends WEdgeToString with DiEdgeToString { _: AnyDiEdge[_] => }

trait WHyperEdgeToString extends WEdgeToString with HyperEdgeToString { _: AnyHyperEdge[_] => }

trait WDiHyperEdgeToString extends WEdgeToString with DiHyperEdgeToString { _: AnyDiHyperEdge[_] => }

trait MultiWEdgeToString { _: WEdgeToString with MultiEdge =>

  override protected def labelSeparator: String = " %% "
}