ComplexTopology.groovy

/*
 * © 2021. TU Dortmund University,
 * Institute of Energy Systems, Energy Efficiency and Energy Economics,
 * Research group Distribution grid planning and operation
 */
package edu.ie3.test.common

import static edu.ie3.test.common.SystemParticipantTestData.emptySystemParticipants

import edu.ie3.datamodel.graph.SubGridGate
import edu.ie3.datamodel.graph.SubGridTopologyGraph
import edu.ie3.datamodel.models.input.connector.ConnectorPort
import edu.ie3.datamodel.models.input.container.GraphicElements
import edu.ie3.datamodel.models.input.container.JointGridContainer
import edu.ie3.datamodel.models.input.container.RawGridElements
import edu.ie3.datamodel.models.input.container.SubGridContainer
import org.jgrapht.graph.DirectedMultigraph

class ComplexTopology extends GridTestData {
  public static final gridName = "complex_topology"

  private static final RawGridElements rawGrid = new RawGridElements(
  [
    nodeA,
    nodeB,
    nodeC,
    nodeD,
    nodeE,
    nodeF,
    nodeG
  ] as Set,
  [] as Set,
  [
    transformerBtoD,
    transformerBtoE,
    transformerCtoE,
    transformerCtoF,
    transformerCtoG
  ] as Set,
  [transformerAtoBtoC] as Set,
  [] as Set,
  [] as Set)

  public static final grid = new JointGridContainer(
  gridName,
  rawGrid,
  emptySystemParticipants,
  new GraphicElements(
  [] as Set,
  [] as Set))

  public static final HashMap<Integer, SubGridContainer> expectedSubGrids = new HashMap<>()

  public static final SubGridTopologyGraph expectedSubGridTopology

  static {
    expectedSubGrids.put(1, new SubGridContainer(
        gridName,
        1,
        new RawGridElements(
        [nodeA, nodeB, nodeC] as Set,
        [] as Set,
        [] as Set,
        [transformerAtoBtoC] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )
    expectedSubGrids.put(2, new SubGridContainer(
        gridName,
        2,
        new RawGridElements(
        [nodeA, nodeB, nodeC] as Set,
        [] as Set,
        [] as Set,
        [transformerAtoBtoC] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )
    expectedSubGrids.put(3, new SubGridContainer(
        gridName,
        3,
        new RawGridElements(
        [nodeA, nodeB, nodeC] as Set,
        [] as Set,
        [] as Set,
        [transformerAtoBtoC] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )
    expectedSubGrids.put(4, new SubGridContainer(
        gridName,
        4,
        new RawGridElements(
        [nodeB, nodeD] as Set,
        [] as Set,
        [transformerBtoD] as Set,
        [] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )
    expectedSubGrids.put(5, new SubGridContainer(
        gridName,
        5,
        new RawGridElements(
        [nodeB, nodeC, nodeE] as Set,
        [] as Set,
        [
          transformerBtoE,
          transformerCtoE
        ] as Set,
        [] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )
    expectedSubGrids.put(6, new SubGridContainer(
        gridName,
        6,
        new RawGridElements(
        [nodeC, nodeF, nodeG] as Set,
        [] as Set,
        [
          transformerCtoF,
          transformerCtoG
        ] as Set,
        [] as Set,
        [] as Set,
        [] as Set),
        emptySystemParticipants,
        new GraphicElements(
        [] as Set,
        [] as Set)
        )
        )

    DirectedMultigraph<SubGridContainer, SubGridGate> mutableGraph =
        new DirectedMultigraph<>(SubGridGate)
    /* Add all edges */
    expectedSubGrids.values().forEach({ subGrid -> mutableGraph.addVertex(subGrid) })

    mutableGraph.addEdge(expectedSubGrids.get(1), expectedSubGrids.get(2), SubGridGate.fromTransformer3W(transformerAtoBtoC, ConnectorPort.B))
    mutableGraph.addEdge(expectedSubGrids.get(1), expectedSubGrids.get(3), SubGridGate.fromTransformer3W(transformerAtoBtoC, ConnectorPort.C))
    mutableGraph.addEdge(expectedSubGrids.get(2), expectedSubGrids.get(4), SubGridGate.fromTransformer2W(transformerBtoD))
    mutableGraph.addEdge(expectedSubGrids.get(2), expectedSubGrids.get(5), SubGridGate.fromTransformer2W(transformerBtoE))
    mutableGraph.addEdge(expectedSubGrids.get(3), expectedSubGrids.get(5), SubGridGate.fromTransformer2W(transformerCtoE))
    mutableGraph.addEdge(expectedSubGrids.get(3), expectedSubGrids.get(6), SubGridGate.fromTransformer2W(transformerCtoF))
    mutableGraph.addEdge(expectedSubGrids.get(3), expectedSubGrids.get(6), SubGridGate.fromTransformer2W(transformerCtoG))
    expectedSubGridTopology = new SubGridTopologyGraph(mutableGraph)
  }
}